refactor: cleanup, remove unused method

This commit is contained in:
Fahad
2025-10-03 23:26:22 +04:00
parent 5b0601bea8
commit 4968e1f7bc
2 changed files with 0 additions and 61 deletions

View File

@@ -389,47 +389,3 @@ class TestThinkingModes:
# Reload config and clear registry
importlib.reload(config)
ModelProviderRegistry._instance = None
def test_thinking_budget_mapping(self):
"""Test that thinking modes map to correct budget values"""
from tools.shared.base_tool import BaseTool
# Create a simple test tool
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 "test"
def get_request_model(self):
return None
async def prepare_prompt(self, request):
return "test"
# Test dynamic budget calculation for Flash 2.5
from providers.gemini import GeminiModelProvider
provider = GeminiModelProvider(api_key="test-key")
flash_model = "gemini-2.5-flash"
flash_max_tokens = 24576
expected_budgets = {
"minimal": int(flash_max_tokens * 0.005), # 123
"low": int(flash_max_tokens * 0.08), # 1966
"medium": int(flash_max_tokens * 0.33), # 8110
"high": int(flash_max_tokens * 0.67), # 16465
"max": int(flash_max_tokens * 1.0), # 24576
}
# Check each mode using the helper method
for mode, expected_budget in expected_budgets.items():
actual_budget = provider.get_thinking_budget(flash_model, mode)
assert actual_budget == expected_budget, f"Mode {mode}: expected {expected_budget}, got {actual_budget}"