Rebranding, refactoring, renaming, cleanup, updated docs
This commit is contained in:
@@ -1,12 +1,14 @@
|
||||
"""Helper functions for test mocking."""
|
||||
|
||||
from unittest.mock import Mock
|
||||
from providers.base import ModelCapabilities, ProviderType
|
||||
|
||||
from providers.base import ModelCapabilities, ProviderType, RangeTemperatureConstraint
|
||||
|
||||
|
||||
def create_mock_provider(model_name="gemini-2.0-flash-exp", max_tokens=1_048_576):
|
||||
"""Create a properly configured mock provider."""
|
||||
mock_provider = Mock()
|
||||
|
||||
|
||||
# Set up capabilities
|
||||
mock_capabilities = ModelCapabilities(
|
||||
provider=ProviderType.GOOGLE,
|
||||
@@ -17,14 +19,14 @@ def create_mock_provider(model_name="gemini-2.0-flash-exp", max_tokens=1_048_576
|
||||
supports_system_prompts=True,
|
||||
supports_streaming=True,
|
||||
supports_function_calling=True,
|
||||
temperature_range=(0.0, 2.0),
|
||||
temperature_constraint=RangeTemperatureConstraint(0.0, 2.0, 0.7),
|
||||
)
|
||||
|
||||
|
||||
mock_provider.get_capabilities.return_value = mock_capabilities
|
||||
mock_provider.get_provider_type.return_value = ProviderType.GOOGLE
|
||||
mock_provider.supports_thinking_mode.return_value = False
|
||||
mock_provider.validate_model_name.return_value = True
|
||||
|
||||
|
||||
# Set up generate_content response
|
||||
mock_response = Mock()
|
||||
mock_response.content = "Test response"
|
||||
@@ -33,7 +35,7 @@ def create_mock_provider(model_name="gemini-2.0-flash-exp", max_tokens=1_048_576
|
||||
mock_response.friendly_name = "Gemini"
|
||||
mock_response.provider = ProviderType.GOOGLE
|
||||
mock_response.metadata = {"finish_reason": "STOP"}
|
||||
|
||||
|
||||
mock_provider.generate_content.return_value = mock_response
|
||||
|
||||
|
||||
return mock_provider
|
||||
|
||||
Reference in New Issue
Block a user