mistralai 1.2.0__tar.gz → 1.2.1__tar.gz

{mistralai-1.2.0 → mistralai-1.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mistralai
-Version: 1.2.0
+Version: 1.2.1
 Summary: Python Client SDK for the Mistral AI API.
 Home-page: https://github.com/mistralai/client-python.git
 Author: Mistral

{mistralai-1.2.0 → mistralai-1.2.1}/packages/mistralai_gcp/src/mistralai_gcp/sdk.py

@@ -1,7 +1,7 @@
 """Code generated by Speakeasy (https://speakeasyapi.dev). DO NOT EDIT."""
 
 import json
-from typing import Optional, Union
+from typing import Optional, Tuple, Union
 
 import google.auth
 import google.auth.credentials
@@ -20,6 +20,19 @@ from .sdkconfiguration import SDKConfiguration
 from .utils.logger import Logger, NoOpLogger
 from .utils.retries import RetryConfig
 
+LEGACY_MODEL_ID_FORMAT = {
+    "codestral-2405": "codestral@2405",
+    "mistral-large-2407": "mistral-large@2407",
+    "mistral-nemo-2407": "mistral-nemo@2407",
+}
+
+def get_model_info(model: str) -> Tuple[str,str]:
+    # if the model requiers the legacy fomat, use it, else do nothing.
+    model_id = LEGACY_MODEL_ID_FORMAT.get(model, model)
+    model = "-".join(model.split("-")[:-1])
+    return model, model_id
+
+
 
 class MistralGoogleCloud(BaseSDK):
     r"""Mistral AI API: Our Chat Completion and Embeddings APIs specification. Create your account on [La Plateforme](https://console.mistral.ai) to get access and read the [docs](https://docs.mistral.ai) to learn how to use it."""
@@ -140,28 +153,24 @@ class GoogleCloudBeforeRequestHook(BeforeRequestHook):
     def before_request(
         self, hook_ctx, request: httpx.Request
     ) -> Union[httpx.Request, Exception]:
-        # The goal of this function is to template in the region, project, model, and model_version into the URL path
+        # The goal of this function is to template in the region, project and model into the URL path
         # We do this here so that the API remains more user-friendly
-        model = None
-        model_version = None
+        model_id = None
         new_content = None
         if request.content:
             parsed = json.loads(request.content.decode("utf-8"))
             model_raw = parsed.get("model")
-            model = "-".join(model_raw.split("-")[:-1])
-            model_version = model_raw.split("-")[-1]
-            parsed["model"] = model
+            model_name, model_id = get_model_info(model_raw)
+            parsed["model"] = model_name
             new_content = json.dumps(parsed).encode("utf-8")
 
-        if model == "":
+        if model_id == "":
             raise models.SDKError("model must be provided")
 
-        if model_version is None:
-            raise models.SDKError("model_version must be provided")
 
         stream = "streamRawPredict" in request.url.path
         specifier = "streamRawPredict" if stream else "rawPredict"
-        url = f"/v1/projects/{self.project_id}/locations/{self.region}/publishers/mistralai/models/{model}@{model_version}:{specifier}"
+        url = f"/v1/projects/{self.project_id}/locations/{self.region}/publishers/mistralai/models/{model_id}:{specifier}"
 
         headers = dict(request.headers)
         # Delete content-length header as it will need to be recalculated

{mistralai-1.2.0 → mistralai-1.2.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mistralai"
-version = "1.2.0"
+version = "1.2.1"
 description = "Python Client SDK for the Mistral AI API."
 authors = ["Mistral"]
 readme = "README-PYPI.md"

{mistralai-1.2.0 → mistralai-1.2.1}/src/mistralai/_version.py

@@ -3,7 +3,7 @@
 import importlib.metadata
 
 __title__: str = "mistralai"
-__version__: str = "1.2.0"
+__version__: str = "1.2.1"
 
 try:
     if __package__ is not None:

{mistralai-1.2.0 → mistralai-1.2.1}/src/mistralai/sdkconfiguration.py

@@ -28,9 +28,9 @@ class SDKConfiguration:
     server: Optional[str] = ""
     language: str = "python"
     openapi_doc_version: str = "0.0.2"
-    sdk_version: str = "1.2.0"
+    sdk_version: str = "1.2.1"
     gen_version: str = "2.452.0"
-    user_agent: str = "speakeasy-sdk/python 1.2.0 2.452.0 0.0.2 mistralai"
+    user_agent: str = "speakeasy-sdk/python 1.2.1 2.452.0 0.0.2 mistralai"
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None