Rebranding, refactoring, renaming, cleanup, updated docs
This commit is contained in:
Fahad
@@ -1,11 +1,11 @@
|
||||
"""Tests for auto mode functionality"""
|
||||
|
||||
import os
|
||||
import pytest
|
||||
from unittest.mock import patch, Mock
|
||||
import importlib
|
||||
import os
|
||||
from unittest.mock import patch
|
||||
|
||||
import pytest
|
||||
|
||||
from mcp.types import TextContent
|
||||
from tools.analyze import AnalyzeTool
|
||||
|
||||
|
||||
@@ -16,23 +16,24 @@ class TestAutoMode:
|
||||
"""Test that auto mode is detected correctly"""
|
||||
# Save original
|
||||
original = os.environ.get("DEFAULT_MODEL", "")
|
||||
|
||||
|
||||
try:
|
||||
# Test auto mode
|
||||
os.environ["DEFAULT_MODEL"] = "auto"
|
||||
import config
|
||||
|
||||
importlib.reload(config)
|
||||
|
||||
|
||||
assert config.DEFAULT_MODEL == "auto"
|
||||
assert config.IS_AUTO_MODE is True
|
||||
|
||||
|
||||
# Test non-auto mode
|
||||
os.environ["DEFAULT_MODEL"] = "pro"
|
||||
importlib.reload(config)
|
||||
|
||||
|
||||
assert config.DEFAULT_MODEL == "pro"
|
||||
assert config.IS_AUTO_MODE is False
|
||||
|
||||
|
||||
finally:
|
||||
# Restore
|
||||
if original:
|
||||
@@ -44,7 +45,7 @@ class TestAutoMode:
|
||||
def test_model_capabilities_descriptions(self):
|
||||
"""Test that model capabilities are properly defined"""
|
||||
from config import MODEL_CAPABILITIES_DESC
|
||||
|
||||
|
||||
# Check all expected models are present
|
||||
expected_models = ["flash", "pro", "o3", "o3-mini"]
|
||||
for model in expected_models:
|
||||
@@ -56,25 +57,26 @@ class TestAutoMode:
|
||||
"""Test that tool schemas require model in auto mode"""
|
||||
# Save original
|
||||
original = os.environ.get("DEFAULT_MODEL", "")
|
||||
|
||||
|
||||
try:
|
||||
# Enable auto mode
|
||||
os.environ["DEFAULT_MODEL"] = "auto"
|
||||
import config
|
||||
|
||||
importlib.reload(config)
|
||||
|
||||
|
||||
tool = AnalyzeTool()
|
||||
schema = tool.get_input_schema()
|
||||
|
||||
|
||||
# Model should be required
|
||||
assert "model" in schema["required"]
|
||||
|
||||
|
||||
# Model field should have detailed descriptions
|
||||
model_schema = schema["properties"]["model"]
|
||||
assert "enum" in model_schema
|
||||
assert "flash" in model_schema["enum"]
|
||||
assert "Choose the best model" in model_schema["description"]
|
||||
|
||||
|
||||
finally:
|
||||
# Restore
|
||||
if original:
|
||||
@@ -88,10 +90,10 @@ class TestAutoMode:
|
||||
# This test uses the default from conftest.py which sets non-auto mode
|
||||
tool = AnalyzeTool()
|
||||
schema = tool.get_input_schema()
|
||||
|
||||
|
||||
# Model should not be required
|
||||
assert "model" not in schema["required"]
|
||||
|
||||
|
||||
# Model field should have simpler description
|
||||
model_schema = schema["properties"]["model"]
|
||||
assert "enum" not in model_schema
|
||||
@@ -102,29 +104,27 @@ class TestAutoMode:
|
||||
"""Test that auto mode enforces model parameter"""
|
||||
# Save original
|
||||
original = os.environ.get("DEFAULT_MODEL", "")
|
||||
|
||||
|
||||
try:
|
||||
# Enable auto mode
|
||||
os.environ["DEFAULT_MODEL"] = "auto"
|
||||
import config
|
||||
|
||||
importlib.reload(config)
|
||||
|
||||
|
||||
tool = AnalyzeTool()
|
||||
|
||||
|
||||
# Mock the provider to avoid real API calls
|
||||
with patch.object(tool, 'get_model_provider') as mock_provider:
|
||||
with patch.object(tool, "get_model_provider"):
|
||||
# Execute without model parameter
|
||||
result = await tool.execute({
|
||||
"files": ["/tmp/test.py"],
|
||||
"prompt": "Analyze this"
|
||||
})
|
||||
|
||||
result = await tool.execute({"files": ["/tmp/test.py"], "prompt": "Analyze this"})
|
||||
|
||||
# Should get error
|
||||
assert len(result) == 1
|
||||
response = result[0].text
|
||||
assert "error" in response
|
||||
assert "Model parameter is required" in response
|
||||
|
||||
|
||||
finally:
|
||||
# Restore
|
||||
if original:
|
||||
@@ -136,45 +136,57 @@ class TestAutoMode:
|
||||
def test_model_field_schema_generation(self):
|
||||
"""Test the get_model_field_schema method"""
|
||||
from tools.base import BaseTool
|
||||
|
||||
|
||||
# Create a minimal concrete tool for testing
|
||||
class TestTool(BaseTool):
|
||||
def get_name(self): return "test"
|
||||
def get_description(self): return "test"
|
||||
def get_input_schema(self): return {}
|
||||
def get_system_prompt(self): return ""
|
||||
def get_request_model(self): return None
|
||||
async def prepare_prompt(self, request): return ""
|
||||
|
||||
def get_name(self):
|
||||
return "test"
|
||||
|
||||
def get_description(self):
|
||||
return "test"
|
||||
|
||||
def get_input_schema(self):
|
||||
return {}
|
||||
|
||||
def get_system_prompt(self):
|
||||
return ""
|
||||
|
||||
def get_request_model(self):
|
||||
return None
|
||||
|
||||
async def prepare_prompt(self, request):
|
||||
return ""
|
||||
|
||||
tool = TestTool()
|
||||
|
||||
|
||||
# Save original
|
||||
original = os.environ.get("DEFAULT_MODEL", "")
|
||||
|
||||
|
||||
try:
|
||||
# Test auto mode
|
||||
os.environ["DEFAULT_MODEL"] = "auto"
|
||||
os.environ["DEFAULT_MODEL"] = "auto"
|
||||
import config
|
||||
|
||||
importlib.reload(config)
|
||||
|
||||
|
||||
schema = tool.get_model_field_schema()
|
||||
assert "enum" in schema
|
||||
assert all(model in schema["enum"] for model in ["flash", "pro", "o3"])
|
||||
assert "Choose the best model" in schema["description"]
|
||||
|
||||
|
||||
# Test normal mode
|
||||
os.environ["DEFAULT_MODEL"] = "pro"
|
||||
importlib.reload(config)
|
||||
|
||||
|
||||
schema = tool.get_model_field_schema()
|
||||
assert "enum" not in schema
|
||||
assert "Available:" in schema["description"]
|
||||
assert "'pro'" in schema["description"]
|
||||
|
||||
|
||||
finally:
|
||||
# Restore
|
||||
if original:
|
||||
os.environ["DEFAULT_MODEL"] = original
|
||||
else:
|
||||
os.environ.pop("DEFAULT_MODEL", None)
|
||||
importlib.reload(config)
|
||||
importlib.reload(config)
|
||||
|
||||
Reference in New Issue
Block a user