fix; ci

Author: mahiro72

tests/models/test_cohere.py

@@ -62,10 +62,20 @@ def test_init():
     assert m.base_url == 'https://api.cohere.com'
 
 
+@dataclass
+class MockClientWrapper:
+    def get_base_url(self) -> str:
+        return 'https://api.cohere.com'
+
+
 @dataclass
 class MockAsyncClientV2:
     completions: MockChatResponse | Sequence[MockChatResponse] | None = None
     index = 0
+    _client_wrapper: MockClientWrapper = None  # type: ignore
+
+    def __post_init__(self):
+        self._client_wrapper = MockClientWrapper()
 
     @classmethod
     def create_mock(cls, completions: MockChatResponse | Sequence[MockChatResponse]) -> AsyncClientV2:

tests/models/test_groq.py

@@ -97,6 +97,7 @@ class MockGroq:
     completions: MockChatCompletion | Sequence[MockChatCompletion] | None = None
     stream: Sequence[MockChatCompletionChunk] | Sequence[Sequence[MockChatCompletionChunk]] | None = None
     index: int = 0
+    base_url: str = 'https://api.groq.com/openai/v1'
 
     @cached_property
     def chat(self) -> Any:

tests/models/test_huggingface.py

@@ -80,6 +80,7 @@ class MockHuggingFace:
     stream: Sequence[MockStreamEvent] | Sequence[Sequence[MockStreamEvent]] | None = None
     index: int = 0
     chat_completion_kwargs: list[dict[str, Any]] = field(default_factory=list)
+    model: str = 'hf-model'
 
     @cached_property
     def chat(self) -> Any:

tests/models/test_mistral.py

@@ -69,12 +69,22 @@
 ]
 
 
+@dataclass
+class MockSdkConfiguration:
+    def get_server_details(self) -> tuple[str, ...]:
+        return ('https://api.mistral.ai/v1',)
+
+
 @dataclass
 class MockMistralAI:
     completions: MockChatCompletion | Sequence[MockChatCompletion] | None = None
     stream: Sequence[MockCompletionEvent] | Sequence[Sequence[MockCompletionEvent]] | None = None
     index: int = 0
 
+    @cached_property
+    def sdk_configuration(self) -> MockSdkConfiguration:
+        return MockSdkConfiguration()
+
     @cached_property
     def chat(self) -> Any:
         if self.stream: