fix: mistral tests

Author: jamescalam

semantic_router/index/qdrant.py

@@ -211,13 +211,13 @@ def _build_filter(self, base_filter: Optional[Any] = None) -> Optional[Any]:
 
         return models.Filter(must=[ns_condition, base_filter])
 
-    def _point_ids_for_utterances(self, routes_to_delete: dict) -> list[str]:
+    def _point_ids_for_utterances(self, routes_to_delete: dict) -> list[str | int]:
         """Compute deterministic point IDs for the given route→utterance mapping.
 
         Uses the same uuid5 scheme as ``add()`` so IDs are derived without
         a round-trip to Qdrant.
         """
-        ids = []
+        ids: list[str | int] = []
         for route, utterances in routes_to_delete.items():
             for utterance in utterances:
                 key = (

semantic_router/llms/mistral.py

@@ -76,9 +76,7 @@ def __call__(self, messages: List[Message]) -> str:
         """
         if self._client is None:
             raise ValueError("MistralAI client is not initialized.")
-        chat_messages = [
-            {"role": m.role, "content": m.content} for m in messages
-        ]
+        chat_messages = [{"role": m.role, "content": m.content} for m in messages]
         try:
             completion = self._client.chat.complete(
                 model=self.name,

tests/unit/llms/test_llm_mistral.py

@@ -8,7 +8,7 @@
 
 @pytest.fixture
 def mistralai_llm(mocker):
-    mocker.patch("mistralai.client.MistralClient")
+    mocker.patch("mistralai.Mistral")
     return MistralAILLM(mistralai_api_key="test_api_key")
 
 
@@ -48,7 +48,7 @@ def test_mistralai_llm_call_uninitialized_client(self, mistralai_llm):
 
     def test_mistralai_llm_init_exception(self, mocker):
         mocker.patch(
-            "mistralai.client.MistralClient",
+            "mistralai.Mistral",
             side_effect=Exception("Initialization error"),
         )
         with pytest.raises(ValueError) as e:
@@ -61,8 +61,8 @@ def test_mistralai_llm_call_success(self, mistralai_llm, mocker):
 
         mocker.patch("os.getenv", return_value="fake-api-key")
         mocker.patch.object(
-            mistralai_llm._client,
-            "chat",
+            mistralai_llm._client.chat,
+            "complete",
             return_value=mock_completion,
         )
         llm_input = [Message(role="user", content="test")]