mistralai 1.2.0__py3-none-any.whl → 1.2.1__py3-none-any.whl

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/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
 

{mistralai-1.2.0.dist-info → mistralai-1.2.1.dist-info}/METADATA

@@ -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.dist-info → mistralai-1.2.1.dist-info}/RECORD

@@ -103,7 +103,7 @@ mistralai_gcp/models/usageinfo.py,sha256=Uo2LJB58JMzlrmnfMUQnDxiMCINMS63ejp-sbOq
 mistralai_gcp/models/usermessage.py,sha256=Wv_yQOP6K81KZZ13Jn0N4i7nM8qh3yuPO0t0pgq-JHQ,1693
 mistralai_gcp/models/validationerror.py,sha256=nPH4v5nEe7Vy-l4fYft3vvfb67yhGYHhoAUFDkupEZg,462
 mistralai_gcp/py.typed,sha256=zrp19r0G21lr2yRiMC0f8MFkQFGj9wMpSbboePMg8KM,59
-mistralai_gcp/sdk.py,sha256=WkfTdRfXdLVqJXNvQG53N8ewRQDj3w-3dGztWaxR2Og,6506
+mistralai_gcp/sdk.py,sha256=qmgp6E5LTBMnNN3mc1aDzmnHnvRQNTR7JEJgjBkYGWg,6730
 mistralai_gcp/sdkconfiguration.py,sha256=tl3dHaj9Bflp_NXy8vZUMVOyeo1FjFRccgYcJKOM0rk,1700
 mistralai_gcp/types/__init__.py,sha256=RArOwSgeeTIva6h-4ttjXwMUeCkz10nAFBL9D-QljI4,377
 mistralai_gcp/types/basemodel.py,sha256=PexI39iKiOkIlobB8Ueo0yn8PLHp6_wb-WO-zelNDZY,1170
@@ -130,7 +130,7 @@ mistralai/_hooks/deprecation_warning.py,sha256=eyEOf7-o9uqqNWJnufD2RXp3dYrGV4in9
 mistralai/_hooks/registration.py,sha256=ML0W-XbE4WYdJ4eGks_XxF2aLCJTaIWjQATFGzFwvyU,861
 mistralai/_hooks/sdkhooks.py,sha256=s-orhdvnV89TmI3QiPC2LWQtYeM9RrsG1CTll-fYZmQ,2559
 mistralai/_hooks/types.py,sha256=vUkTVk_TSaK10aEj368KYWvnd4T5EsawixMcTro_UHc,2570
-mistralai/_version.py,sha256=2mDc-6ys7Z2_9DG49D98FdOTezgZbeAhnySMQPuquT0,313
+mistralai/_version.py,sha256=j99Wka-bmcP4xPwygSV8i6CYX38lJNUdWGnDWwKDOFM,313
 mistralai/agents.py,sha256=EYdKVTBsW4QeH6afr5CrWFbEL6K1wzCeZw2l5DeBdTw,28771
 mistralai/async_client.py,sha256=KUdYxIIqoD6L7vB0EGwUR6lQ0NK5iCTHjnLVR9CVcJY,355
 mistralai/basesdk.py,sha256=MQIeNInArcIcDYgSy67EDf77ykys9JwjTlnboqdpEJ8,11273
@@ -251,7 +251,7 @@ mistralai/models/wandbintegrationout.py,sha256=C0HpS8jJGnACs7eWnuIq0qJEroIUAbjkv
 mistralai/models_.py,sha256=Kj5U6ODBgKiKs93Miaq11dlV3MDvaz--nI8ucvZVark,39797
 mistralai/py.typed,sha256=zrp19r0G21lr2yRiMC0f8MFkQFGj9wMpSbboePMg8KM,59
 mistralai/sdk.py,sha256=JBssVtpwLDcni_Q-IHNs62Zz8-ZxFldiOj3e42-Cb8k,5007
-mistralai/sdkconfiguration.py,sha256=klmP_ccBSxSvYtRpvkGJgtdk6HP2_qwTwUuovI-vIP4,1688
+mistralai/sdkconfiguration.py,sha256=BWK7DpPHmxpvpWIdtqbgi5_xBxDM3Ra-EOHCmd4IzhA,1688
 mistralai/types/__init__.py,sha256=RArOwSgeeTIva6h-4ttjXwMUeCkz10nAFBL9D-QljI4,377
 mistralai/types/basemodel.py,sha256=PexI39iKiOkIlobB8Ueo0yn8PLHp6_wb-WO-zelNDZY,1170
 mistralai/utils/__init__.py,sha256=8npwwHS-7zjVrbkzBGO-Uk4GkjC240PCleMbgPK1Axs,2418
@@ -270,7 +270,7 @@ mistralai/utils/serializers.py,sha256=BSJT7kBOkNBFyP7KREyMoe14JGbgijD1M6AXFMbdmc
 mistralai/utils/url.py,sha256=BgGPgcTA6MRK4bF8fjP2dUopN3NzEzxWMXPBVg8NQUA,5254
 mistralai/utils/values.py,sha256=_89YXPTI_BU6SXJBzFR4pIzTCBPQW9tsOTN1jeBBIDs,3428
 mistralai/version.py,sha256=iosXhlXclBwBqlADFKEilxAC2wWKbtuBKi87AmPi7s8,196
-mistralai-1.2.0.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
-mistralai-1.2.0.dist-info/METADATA,sha256=pXGl4n99vLAB5MMcYiiXae_5wU1TZx0nM2puSZAaThM,26504
-mistralai-1.2.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-mistralai-1.2.0.dist-info/RECORD,,
+mistralai-1.2.1.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
+mistralai-1.2.1.dist-info/METADATA,sha256=Rjjd8iWB2Dx-zQeu4cN-dgmQYeQNSpk3sxFlf_iYYOY,26504
+mistralai-1.2.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+mistralai-1.2.1.dist-info/RECORD,,

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