Merge pull request #665 from aurelio-labs/james/version-updates

fix: version updates

Author: jamescalam

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "semantic-router"
-version = "0.1.14"
+version = "0.1.15"
 description = "Super fast semantic router for AI decision making"
 authors = [{ name = "Aurelio AI", email = "hello@aurelio.ai" }]
 requires-python = ">=3.9,<3.14"
@@ -18,7 +18,8 @@ dependencies = [
     "aiohttp>=3.10.11,<4",
     "tornado>=6.4.2,<7",
     "urllib3>=1.26,<3",
-    "litellm>=1.61.3",
+    # https://www.cve.org/CVERecord?id=CVE-2026-42208 explains >=1.83.7 
+    "litellm>=1.83.7",
     "openai>=1.10.0,<3.0.0",
 ]
 
@@ -38,7 +39,7 @@ vision = [
     "pillow>=10.2.0,<11.0.0 ; python_version < '3.13'",
     "torch>=2.6.0 ; python_version < '3.13'"
 ]
-mistralai = ["mistralai>=0.0.12,<0.1.0"]
+mistralai = ["mistralai>=1.0.0,<2.0.0"]
 qdrant = ["qdrant-client>=1.11.1,<2"]
 google = ["google-cloud-aiplatform>=1.45.0,<2"]
 bedrock = [

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

@@ -13,7 +13,6 @@ class MistralAILLM(BaseLLM):
     """LLM for MistralAI. Requires a MistralAI API key from https://console.mistral.ai/api-keys/"""
 
     _client: Any = PrivateAttr()
-    _mistralai: Any = PrivateAttr()
 
     def __init__(
         self,
@@ -36,7 +35,7 @@ def __init__(
         if name is None:
             name = EncoderDefault.MISTRAL.value["language_model"]
         super().__init__(name=name)
-        self._client, self._mistralai = self._initialize_client(mistralai_api_key)
+        self._client = self._initialize_client(mistralai_api_key)
         self.temperature = temperature
         self.max_tokens = max_tokens
 
@@ -46,11 +45,10 @@ def _initialize_client(self, api_key):
         :param api_key: The MistralAI API key.
         :type api_key: Optional[str]
         :return: The MistralAI client.
-        :rtype: MistralClient
+        :rtype: Mistral
         """
         try:
-            import mistralai
-            from mistralai.client import MistralClient
+            from mistralai import Mistral
         except ImportError:
             raise ImportError(
                 "Please install MistralAI to use MistralAI LLM. "
@@ -61,12 +59,12 @@ def _initialize_client(self, api_key):
         if api_key is None:
             raise ValueError("MistralAI API key cannot be 'None'.")
         try:
-            client = MistralClient(api_key=api_key)
+            client = Mistral(api_key=api_key)
         except Exception as e:
             raise ValueError(
                 f"MistralAI API client failed to initialize. Error: {e}"
             ) from e
-        return client, mistralai
+        return client
 
     def __call__(self, messages: List[Message]) -> str:
         """Call the MistralAILLM.
@@ -78,14 +76,9 @@ def __call__(self, messages: List[Message]) -> str:
         """
         if self._client is None:
             raise ValueError("MistralAI client is not initialized.")
-        chat_messages = [
-            self._mistralai.models.chat_completion.ChatMessage(
-                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(
+            completion = self._client.chat.complete(
                 model=self.name,
                 messages=chat_messages,
                 temperature=self.temperature,

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")]