mistralai 1.3.1__tar.gz → 1.4.0__tar.gz

{mistralai-1.3.1 → mistralai-1.4.0}/PKG-INFO

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: mistralai
-Version: 1.3.1
+Version: 1.4.0
 Summary: Python Client SDK for the Mistral AI API.
-Home-page: https://github.com/mistralai/client-python.git
 Author: Mistral
 Requires-Python: >=3.8
 Classifier: Programming Language :: Python :: 3
@@ -118,9 +117,7 @@ with Mistral(
             "content": "Who is the best French painter? Answer in one short sentence.",
             "role": "user",
         },
-    ])
-
-    assert res is not None
+    ], stream=False)
 
     # Handle response
     print(res)
@@ -145,9 +142,7 @@ async def main():
                 "content": "Who is the best French painter? Answer in one short sentence.",
                 "role": "user",
             },
-        ])
-
-        assert res is not None
+        ], stream=False)
 
         # Handle response
         print(res)
@@ -173,8 +168,6 @@ with Mistral(
         "content": open("example.file", "rb"),
     })
 
-    assert res is not None
-
     # Handle response
     print(res)
 ```
@@ -198,8 +191,6 @@ async def main():
             "content": open("example.file", "rb"),
         })
 
-        assert res is not None
-
         # Handle response
         print(res)
 
@@ -224,9 +215,7 @@ with Mistral(
             "content": "Who is the best French painter? Answer in one short sentence.",
             "role": "user",
         },
-    ], agent_id="<value>")
-
-    assert res is not None
+    ], agent_id="<id>", stream=False)
 
     # Handle response
     print(res)
@@ -251,9 +240,7 @@ async def main():
                 "content": "Who is the best French painter? Answer in one short sentence.",
                 "role": "user",
             },
-        ], agent_id="<value>")
-
-        assert res is not None
+        ], agent_id="<id>", stream=False)
 
         # Handle response
         print(res)
@@ -277,9 +264,7 @@ with Mistral(
     res = mistral.embeddings.create(inputs=[
         "Embed this sentence.",
         "As well as this one.",
-    ], model="Wrangler")
-
-    assert res is not None
+    ], model="mistral-embed")
 
     # Handle response
     print(res)
@@ -302,9 +287,7 @@ async def main():
         res = await mistral.embeddings.create_async(inputs=[
             "Embed this sentence.",
             "As well as this one.",
-        ], model="Wrangler")
-
-        assert res is not None
+        ], model="mistral-embed")
 
         # Handle response
         print(res)
@@ -506,9 +489,7 @@ with Mistral(
             "content": "Who is the best French painter? Answer in one short sentence.",
             "role": "user",
         },
-    ])
-
-    assert res is not None
+    ], stream=True)
 
     with res as event_stream:
         for event in event_stream:
@@ -545,8 +526,6 @@ with Mistral(
         "content": open("example.file", "rb"),
     })
 
-    assert res is not None
-
     # Handle response
     print(res)
 
@@ -571,8 +550,6 @@ with Mistral(
     res = mistral.models.list(,
         RetryConfig("backoff", BackoffStrategy(1, 50, 1.1, 100), False))
 
-    assert res is not None
-
     # Handle response
     print(res)
 
@@ -591,8 +568,6 @@ with Mistral(
 
     res = mistral.models.list()
 
-    assert res is not None
-
     # Handle response
     print(res)
 
@@ -634,8 +609,6 @@ with Mistral(
 
         res = mistral.models.list()
 
-        assert res is not None
-
         # Handle response
         print(res)
 
@@ -672,8 +645,6 @@ with Mistral(
 
     res = mistral.models.list()
 
-    assert res is not None
-
     # Handle response
     print(res)
 
@@ -693,8 +664,6 @@ with Mistral(
 
     res = mistral.models.list()
 
-    assert res is not None
-
     # Handle response
     print(res)
 
@@ -804,8 +773,6 @@ with Mistral(
 
     res = mistral.models.list()
 
-    assert res is not None
-
     # Handle response
     print(res)
 

{mistralai-1.3.1 → mistralai-1.4.0}/README-PYPI.md

@@ -92,9 +92,7 @@ with Mistral(
             "content": "Who is the best French painter? Answer in one short sentence.",
             "role": "user",
         },
-    ])
-
-    assert res is not None
+    ], stream=False)
 
     # Handle response
     print(res)
@@ -119,9 +117,7 @@ async def main():
                 "content": "Who is the best French painter? Answer in one short sentence.",
                 "role": "user",
             },
-        ])
-
-        assert res is not None
+        ], stream=False)
 
         # Handle response
         print(res)
@@ -147,8 +143,6 @@ with Mistral(
         "content": open("example.file", "rb"),
     })
 
-    assert res is not None
-
     # Handle response
     print(res)
 ```
@@ -172,8 +166,6 @@ async def main():
             "content": open("example.file", "rb"),
         })
 
-        assert res is not None
-
         # Handle response
         print(res)
 
@@ -198,9 +190,7 @@ with Mistral(
             "content": "Who is the best French painter? Answer in one short sentence.",
             "role": "user",
         },
-    ], agent_id="<value>")
-
-    assert res is not None
+    ], agent_id="<id>", stream=False)
 
     # Handle response
     print(res)
@@ -225,9 +215,7 @@ async def main():
                 "content": "Who is the best French painter? Answer in one short sentence.",
                 "role": "user",
             },
-        ], agent_id="<value>")
-
-        assert res is not None
+        ], agent_id="<id>", stream=False)
 
         # Handle response
         print(res)
@@ -251,9 +239,7 @@ with Mistral(
     res = mistral.embeddings.create(inputs=[
         "Embed this sentence.",
         "As well as this one.",
-    ], model="Wrangler")
-
-    assert res is not None
+    ], model="mistral-embed")
 
     # Handle response
     print(res)
@@ -276,9 +262,7 @@ async def main():
         res = await mistral.embeddings.create_async(inputs=[
             "Embed this sentence.",
             "As well as this one.",
-        ], model="Wrangler")
-
-        assert res is not None
+        ], model="mistral-embed")
 
         # Handle response
         print(res)
@@ -480,9 +464,7 @@ with Mistral(
             "content": "Who is the best French painter? Answer in one short sentence.",
             "role": "user",
         },
-    ])
-
-    assert res is not None
+    ], stream=True)
 
     with res as event_stream:
         for event in event_stream:
@@ -519,8 +501,6 @@ with Mistral(
         "content": open("example.file", "rb"),
     })
 
-    assert res is not None
-
     # Handle response
     print(res)
 
@@ -545,8 +525,6 @@ with Mistral(
     res = mistral.models.list(,
         RetryConfig("backoff", BackoffStrategy(1, 50, 1.1, 100), False))
 
-    assert res is not None
-
     # Handle response
     print(res)
 
@@ -565,8 +543,6 @@ with Mistral(
 
     res = mistral.models.list()
 
-    assert res is not None
-
     # Handle response
     print(res)
 
@@ -608,8 +584,6 @@ with Mistral(
 
         res = mistral.models.list()
 
-        assert res is not None
-
         # Handle response
         print(res)
 
@@ -646,8 +620,6 @@ with Mistral(
 
     res = mistral.models.list()
 
-    assert res is not None
-
     # Handle response
     print(res)
 
@@ -667,8 +639,6 @@ with Mistral(
 
     res = mistral.models.list()
 
-    assert res is not None
-
     # Handle response
     print(res)
 
@@ -778,8 +748,6 @@ with Mistral(
 
     res = mistral.models.list()
 
-    assert res is not None
-
     # Handle response
     print(res)
 

{mistralai-1.3.1/packages/mistralai_azure/src/mistralai_azure → mistralai-1.4.0/packages/mistralai_gcp/src/mistralai_gcp}/_hooks/custom_user_agent.py

@@ -19,4 +19,4 @@ class CustomUserAgentHook(BeforeRequestHook):
             PREFIX + current.split(" ")[1]
         )
 
-        return request
+        return request

{mistralai-1.3.1 → mistralai-1.4.0}/packages/mistralai_gcp/src/mistralai_gcp/sdk.py

@@ -30,8 +30,7 @@ def get_model_info(model: str) -> Tuple[str, str]:
     # if the model requiers the legacy fomat, use it, else do nothing.
     if model in LEGACY_MODEL_ID_FORMAT:
         return "-".join(model.split("-")[:-1]), LEGACY_MODEL_ID_FORMAT[model]
-    else:
-        return model, model
+    return model, model
 
 
 

{mistralai-1.3.1 → mistralai-1.4.0}/pyproject.toml

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

mistralai-1.4.0/src/mistralai/__init__.py

@@ -0,0 +1,18 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from ._version import (
+    __title__,
+    __version__,
+    __openapi_doc_version__,
+    __gen_version__,
+    __user_agent__,
+)
+from .sdk import *
+from .sdkconfiguration import *
+from .models import *
+
+
+VERSION: str = __version__
+OPENAPI_DOC_VERSION = __openapi_doc_version__
+SPEAKEASY_GENERATOR_VERSION = __gen_version__
+USER_AGENT = __user_agent__

{mistralai-1.3.1 → mistralai-1.4.0}/src/mistralai/_version.py

@@ -3,7 +3,10 @@
 import importlib.metadata
 
 __title__: str = "mistralai"
-__version__: str = "1.3.1"
+__version__: str = "1.4.0"
+__openapi_doc_version__: str = "0.0.2"
+__gen_version__: str = "2.493.32"
+__user_agent__: str = "speakeasy-sdk/python 1.4.0 2.493.32 0.0.2 mistralai"
 
 try:
     if __package__ is not None:

{mistralai-1.3.1 → mistralai-1.4.0}/src/mistralai/agents.py

@@ -43,11 +43,14 @@ class Agents(BaseSDK):
         presence_penalty: Optional[float] = None,
         frequency_penalty: Optional[float] = None,
         n: OptionalNullable[int] = UNSET,
+        prediction: Optional[
+            Union[models.Prediction, models.PredictionTypedDict]
+        ] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> Optional[models.ChatCompletionResponse]:
+    ) -> models.ChatCompletionResponse:
         r"""Agents Completion
 
         :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content.
@@ -62,6 +65,7 @@ class Agents(BaseSDK):
         :param presence_penalty: presence_penalty determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative.
         :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
         :param n: Number of completions to return for each request, input tokens are only billed once.
+        :param prediction:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -93,6 +97,9 @@ class Agents(BaseSDK):
             presence_penalty=presence_penalty,
             frequency_penalty=frequency_penalty,
             n=n,
+            prediction=utils.get_pydantic_model(
+                prediction, Optional[models.Prediction]
+            ),
             agent_id=agent_id,
         )
 
@@ -138,13 +145,16 @@ class Agents(BaseSDK):
 
         data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(
-                http_res.text, Optional[models.ChatCompletionResponse]
-            )
+            return utils.unmarshal_json(http_res.text, models.ChatCompletionResponse)
         if utils.match_response(http_res, "422", "application/json"):
             data = utils.unmarshal_json(http_res.text, models.HTTPValidationErrorData)
             raise models.HTTPValidationError(data=data)
-        if utils.match_response(http_res, ["4XX", "5XX"], "*"):
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.SDKError(
                 "API error occurred", http_res.status_code, http_res_text, http_res
@@ -191,11 +201,14 @@ class Agents(BaseSDK):
         presence_penalty: Optional[float] = None,
         frequency_penalty: Optional[float] = None,
         n: OptionalNullable[int] = UNSET,
+        prediction: Optional[
+            Union[models.Prediction, models.PredictionTypedDict]
+        ] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> Optional[models.ChatCompletionResponse]:
+    ) -> models.ChatCompletionResponse:
         r"""Agents Completion
 
         :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content.
@@ -210,6 +223,7 @@ class Agents(BaseSDK):
         :param presence_penalty: presence_penalty determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative.
         :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
         :param n: Number of completions to return for each request, input tokens are only billed once.
+        :param prediction:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -241,6 +255,9 @@ class Agents(BaseSDK):
             presence_penalty=presence_penalty,
             frequency_penalty=frequency_penalty,
             n=n,
+            prediction=utils.get_pydantic_model(
+                prediction, Optional[models.Prediction]
+            ),
             agent_id=agent_id,
         )
 
@@ -286,13 +303,16 @@ class Agents(BaseSDK):
 
         data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(
-                http_res.text, Optional[models.ChatCompletionResponse]
-            )
+            return utils.unmarshal_json(http_res.text, models.ChatCompletionResponse)
         if utils.match_response(http_res, "422", "application/json"):
             data = utils.unmarshal_json(http_res.text, models.HTTPValidationErrorData)
             raise models.HTTPValidationError(data=data)
-        if utils.match_response(http_res, ["4XX", "5XX"], "*"):
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.SDKError(
                 "API error occurred", http_res.status_code, http_res_text, http_res
@@ -339,11 +359,14 @@ class Agents(BaseSDK):
         presence_penalty: Optional[float] = None,
         frequency_penalty: Optional[float] = None,
         n: OptionalNullable[int] = UNSET,
+        prediction: Optional[
+            Union[models.Prediction, models.PredictionTypedDict]
+        ] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> Optional[eventstreaming.EventStream[models.CompletionEvent]]:
+    ) -> eventstreaming.EventStream[models.CompletionEvent]:
         r"""Stream Agents completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -360,6 +383,7 @@ class Agents(BaseSDK):
         :param presence_penalty: presence_penalty determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative.
         :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
         :param n: Number of completions to return for each request, input tokens are only billed once.
+        :param prediction:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -391,6 +415,9 @@ class Agents(BaseSDK):
             presence_penalty=presence_penalty,
             frequency_penalty=frequency_penalty,
             n=n,
+            prediction=utils.get_pydantic_model(
+                prediction, Optional[models.Prediction]
+            ),
             agent_id=agent_id,
         )
 
@@ -446,7 +473,12 @@ class Agents(BaseSDK):
             http_res_text = utils.stream_to_text(http_res)
             data = utils.unmarshal_json(http_res_text, models.HTTPValidationErrorData)
             raise models.HTTPValidationError(data=data)
-        if utils.match_response(http_res, ["4XX", "5XX"], "*"):
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.SDKError(
                 "API error occurred", http_res.status_code, http_res_text, http_res
@@ -493,11 +525,14 @@ class Agents(BaseSDK):
         presence_penalty: Optional[float] = None,
         frequency_penalty: Optional[float] = None,
         n: OptionalNullable[int] = UNSET,
+        prediction: Optional[
+            Union[models.Prediction, models.PredictionTypedDict]
+        ] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
         http_headers: Optional[Mapping[str, str]] = None,
-    ) -> Optional[eventstreaming.EventStreamAsync[models.CompletionEvent]]:
+    ) -> eventstreaming.EventStreamAsync[models.CompletionEvent]:
         r"""Stream Agents completion
 
         Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON.
@@ -514,6 +549,7 @@ class Agents(BaseSDK):
         :param presence_penalty: presence_penalty determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative.
         :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition.
         :param n: Number of completions to return for each request, input tokens are only billed once.
+        :param prediction:
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
         :param timeout_ms: Override the default request timeout configuration for this method in milliseconds
@@ -545,6 +581,9 @@ class Agents(BaseSDK):
             presence_penalty=presence_penalty,
             frequency_penalty=frequency_penalty,
             n=n,
+            prediction=utils.get_pydantic_model(
+                prediction, Optional[models.Prediction]
+            ),
             agent_id=agent_id,
         )
 
@@ -600,7 +639,12 @@ class Agents(BaseSDK):
             http_res_text = await utils.stream_to_text_async(http_res)
             data = utils.unmarshal_json(http_res_text, models.HTTPValidationErrorData)
             raise models.HTTPValidationError(data=data)
-        if utils.match_response(http_res, ["4XX", "5XX"], "*"):
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError(
+                "API error occurred", http_res.status_code, http_res_text, http_res
+            )
+        if utils.match_response(http_res, "5XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.SDKError(
                 "API error occurred", http_res.status_code, http_res_text, http_res