mistralai 1.8.0__py3-none-any.whl → 1.8.1__py3-none-any.whl

mistralai/_version.py

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "mistralai"
-__version__: str = "1.8.0"
+__version__: str = "1.8.1"
 __openapi_doc_version__: str = "1.0.0"
 __gen_version__: str = "2.548.6"
-__user_agent__: str = "speakeasy-sdk/python 1.8.0 2.548.6 1.0.0 mistralai"
+__user_agent__: str = "speakeasy-sdk/python 1.8.1 2.548.6 1.0.0 mistralai"
 
 try:
     if __package__ is not None:

mistralai/beta.py

@@ -8,7 +8,9 @@ from mistralai.mistral_agents import MistralAgents
 
 class Beta(BaseSDK):
     conversations: Conversations
+    r"""(beta) Converstations API"""
     agents: MistralAgents
+    r"""(beta) Agents API"""
 
     def __init__(self, sdk_config: SDKConfiguration) -> None:
         BaseSDK.__init__(self, sdk_config)

mistralai/conversations.py

@@ -34,8 +34,9 @@ if typing.TYPE_CHECKING:
 # endregion imports
 
 
-
 class Conversations(BaseSDK):
+    r"""(beta) Converstations API"""
+
     # region sdk-class-body
     # Custom run code allowing client side execution of code
 
@@ -146,7 +147,9 @@ class Conversations(BaseSDK):
             completion_args=completion_args,
         )
 
-        async def run_generator() -> AsyncGenerator[Union[RunResultEvents, RunResult], None]:
+        async def run_generator() -> (
+            AsyncGenerator[Union[RunResultEvents, RunResult], None]
+        ):
             current_entries = input_entries
             while True:
                 received_event_tracker: defaultdict[

mistralai/embeddings.py

@@ -18,6 +18,8 @@ class Embeddings(BaseSDK):
         inputs: Union[
             models.EmbeddingRequestInputs, models.EmbeddingRequestInputsTypedDict
         ],
+        output_dimension: OptionalNullable[int] = UNSET,
+        output_dtype: Optional[models.EmbeddingDtype] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -29,6 +31,8 @@ class Embeddings(BaseSDK):
 
         :param model: ID of the model to use.
         :param inputs: Text to embed.
+        :param output_dimension: The dimension of the output embeddings.
+        :param output_dtype:
         :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
@@ -47,6 +51,8 @@ class Embeddings(BaseSDK):
         request = models.EmbeddingRequest(
             model=model,
             inputs=inputs,
+            output_dimension=output_dimension,
+            output_dtype=output_dtype,
         )
 
         req = self._build_request(
@@ -125,6 +131,8 @@ class Embeddings(BaseSDK):
         inputs: Union[
             models.EmbeddingRequestInputs, models.EmbeddingRequestInputsTypedDict
         ],
+        output_dimension: OptionalNullable[int] = UNSET,
+        output_dtype: Optional[models.EmbeddingDtype] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -136,6 +144,8 @@ class Embeddings(BaseSDK):
 
         :param model: ID of the model to use.
         :param inputs: Text to embed.
+        :param output_dimension: The dimension of the output embeddings.
+        :param output_dtype:
         :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
@@ -154,6 +164,8 @@ class Embeddings(BaseSDK):
         request = models.EmbeddingRequest(
             model=model,
             inputs=inputs,
+            output_dimension=output_dimension,
+            output_dtype=output_dtype,
         )
 
         req = self._build_request_async(

mistralai/mistral_agents.py

@@ -9,6 +9,8 @@ from typing import Any, List, Mapping, Optional, Union
 
 
 class MistralAgents(BaseSDK):
+    r"""(beta) Agents API"""
+
     def create(
         self,
         *,

mistralai/models/__init__.py

@@ -347,6 +347,7 @@ from .documenturlchunk import (
     DocumentURLChunkType,
     DocumentURLChunkTypedDict,
 )
+from .embeddingdtype import EmbeddingDtype
 from .embeddingrequest import (
     EmbeddingRequest,
     EmbeddingRequestInputs,
@@ -964,6 +965,7 @@ __all__ = [
     "DocumentURLChunk",
     "DocumentURLChunkType",
     "DocumentURLChunkTypedDict",
+    "EmbeddingDtype",
     "EmbeddingRequest",
     "EmbeddingRequestInputs",
     "EmbeddingRequestInputsTypedDict",

mistralai/models/chatcompletionresponse.py

@@ -4,8 +4,8 @@ from __future__ import annotations
 from .chatcompletionchoice import ChatCompletionChoice, ChatCompletionChoiceTypedDict
 from .usageinfo import UsageInfo, UsageInfoTypedDict
 from mistralai.types import BaseModel
-from typing import List, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import List
+from typing_extensions import TypedDict
 
 
 class ChatCompletionResponseTypedDict(TypedDict):
@@ -13,8 +13,8 @@ class ChatCompletionResponseTypedDict(TypedDict):
     object: str
     model: str
     usage: UsageInfoTypedDict
-    created: NotRequired[int]
-    choices: NotRequired[List[ChatCompletionChoiceTypedDict]]
+    created: int
+    choices: List[ChatCompletionChoiceTypedDict]
 
 
 class ChatCompletionResponse(BaseModel):
@@ -26,6 +26,6 @@ class ChatCompletionResponse(BaseModel):
 
     usage: UsageInfo
 
-    created: Optional[int] = None
+    created: int
 
-    choices: Optional[List[ChatCompletionChoice]] = None
+    choices: List[ChatCompletionChoice]

mistralai/models/embeddingdtype.py

@@ -0,0 +1,7 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from typing import Literal
+
+
+EmbeddingDtype = Literal["float", "int8", "uint8", "binary", "ubinary"]

mistralai/models/embeddingrequest.py

@@ -1,10 +1,12 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from mistralai.types import BaseModel
+from .embeddingdtype import EmbeddingDtype
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 import pydantic
-from typing import List, Union
-from typing_extensions import Annotated, TypeAliasType, TypedDict
+from pydantic import model_serializer
+from typing import List, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
 EmbeddingRequestInputsTypedDict = TypeAliasType(
@@ -22,6 +24,9 @@ class EmbeddingRequestTypedDict(TypedDict):
     r"""ID of the model to use."""
     inputs: EmbeddingRequestInputsTypedDict
     r"""Text to embed."""
+    output_dimension: NotRequired[Nullable[int]]
+    r"""The dimension of the output embeddings."""
+    output_dtype: NotRequired[EmbeddingDtype]
 
 
 class EmbeddingRequest(BaseModel):
@@ -30,3 +35,38 @@ class EmbeddingRequest(BaseModel):
 
     inputs: Annotated[EmbeddingRequestInputs, pydantic.Field(alias="input")]
     r"""Text to embed."""
+
+    output_dimension: OptionalNullable[int] = UNSET
+    r"""The dimension of the output embeddings."""
+
+    output_dtype: Optional[EmbeddingDtype] = None
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["output_dimension", "output_dtype"]
+        nullable_fields = ["output_dimension"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in self.model_fields.items():
+            k = f.alias or n
+            val = serialized.get(k)
+            serialized.pop(k, None)
+
+            optional_nullable = k in optional_fields and k in nullable_fields
+            is_set = (
+                self.__pydantic_fields_set__.intersection({n})
+                or k in null_default_fields
+            )  # pylint: disable=no-member
+
+            if val is not None and val != UNSET_SENTINEL:
+                m[k] = val
+            elif val != UNSET_SENTINEL and (
+                not k in optional_fields or (optional_nullable and is_set)
+            ):
+                m[k] = val
+
+        return m

mistralai/models/fimcompletionresponse.py

@@ -4,8 +4,8 @@ from __future__ import annotations
 from .chatcompletionchoice import ChatCompletionChoice, ChatCompletionChoiceTypedDict
 from .usageinfo import UsageInfo, UsageInfoTypedDict
 from mistralai.types import BaseModel
-from typing import List, Optional
-from typing_extensions import NotRequired, TypedDict
+from typing import List
+from typing_extensions import TypedDict
 
 
 class FIMCompletionResponseTypedDict(TypedDict):
@@ -13,8 +13,8 @@ class FIMCompletionResponseTypedDict(TypedDict):
     object: str
     model: str
     usage: UsageInfoTypedDict
-    created: NotRequired[int]
-    choices: NotRequired[List[ChatCompletionChoiceTypedDict]]
+    created: int
+    choices: List[ChatCompletionChoiceTypedDict]
 
 
 class FIMCompletionResponse(BaseModel):
@@ -26,6 +26,6 @@ class FIMCompletionResponse(BaseModel):
 
     usage: UsageInfo
 
-    created: Optional[int] = None
+    created: int
 
-    choices: Optional[List[ChatCompletionChoice]] = None
+    choices: List[ChatCompletionChoice]

{mistralai-1.8.0.dist-info → mistralai-1.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mistralai
-Version: 1.8.0
+Version: 1.8.1
 Summary: Python Client SDK for the Mistral AI API.
 Author: Mistral
 Requires-Python: >=3.9

{mistralai-1.8.0.dist-info → mistralai-1.8.1.dist-info}/RECORD

@@ -139,17 +139,17 @@ 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=z3AUFDpRJHj2m3h5PklvUeEcGohY0cfph4jL6-nGIzs,2812
-mistralai/_version.py,sha256=2h0W-7sYb3UbVvUO0AVfyy7sRfFO6mJfM7TomiOVI9I,460
+mistralai/_version.py,sha256=ov7SFKb_NFEoxMwoKD6j1BNiqtY8jnujwybVOhD60gY,460
 mistralai/agents.py,sha256=o_apyuwiDzxv-U252T84ynAHCb5fn1q7MMXqrZ4oHLo,32652
 mistralai/async_client.py,sha256=KUdYxIIqoD6L7vB0EGwUR6lQ0NK5iCTHjnLVR9CVcJY,355
 mistralai/basesdk.py,sha256=GsU5bp8O5fBCl34tKxaYmeYSIIM971eAPeFBBC_BpFo,12191
 mistralai/batch.py,sha256=YN4D0Duwrap9Ysmp_lRpADYp1Znay7THE_z8ERGvDds,501
-mistralai/beta.py,sha256=ZoBpHRinJu_LlNM7q91HC7huXmzKVpgGLL_bRdXjqBg,662
+mistralai/beta.py,sha256=wiOFmXiXNKeRMOEdgeIwVqAWg6E_3sqHc_ojwU9oZoY,728
 mistralai/chat.py,sha256=1XVVVvDi726bq6HXCur6-dsmFfzQAEpEWbKT_3sTZ4A,40549
 mistralai/classifiers.py,sha256=Cbrb6X_eq3-Yz5ZhWkOsFDTGbo3hkgh-vtIEQmU_UdI,33776
 mistralai/client.py,sha256=hrPg-LciKMKiascF0WbRRmqQyCv1lb2yDh6j-aaKVNo,509
-mistralai/conversations.py,sha256=2RTDBKRHPXGtTobMonJUnVWy9zP3i3aVRWE4pcKdF-Y,109256
-mistralai/embeddings.py,sha256=Tq5ZEo6CR0ktQCdxX7mhLy5CfzI5c8mtkuadIVtsWgM,8644
+mistralai/conversations.py,sha256=53wWkEymYuoiuGpKwdoAVe5drJYTKT3JsuEQSt59uOA,109317
+mistralai/embeddings.py,sha256=SHG4s0yKwCdKhYKGXKdzPXmcD4_WXI9BBnrM7cSI3gw,9258
 mistralai/extra/README.md,sha256=BTS9fy0ijkiUP7ZVoFQ7FVBxHtXIXqucYZyy_ucFjo4,1739
 mistralai/extra/__init__.py,sha256=8DsU_omYYadqcwlmBOoakBwkWKcSohwLmtB8v-jkn2M,392
 mistralai/extra/exceptions.py,sha256=4EEygCfdsniYiroHEFVSVDqerQZkpRG027mlJXvMqns,428
@@ -175,9 +175,9 @@ mistralai/fim.py,sha256=UMx-bFYbaSyANZug8VrCabHsqePdiHoYQO1YMKB2MvY,27935
 mistralai/fine_tuning.py,sha256=UENQqfE054VEsAYxdruV-TBLFIFfO-joXNznH08GUvE,477
 mistralai/httpclient.py,sha256=lC-YQ7q4yiJGKElxBeb3aZnr-4aYxjgEpZ6roeXYlyg,4318
 mistralai/jobs.py,sha256=1DZE14ad348Vg82VHhLRyXhu7SIh8_KgWXc_jP2oFIA,46767
-mistralai/mistral_agents.py,sha256=txjs5YVDV4u5sEHI6WFPcE3KGXb0I2Il3aJS6Eq9iCc,45543
+mistralai/mistral_agents.py,sha256=FrIUGExYpwz3EyMOetLKsD-LtcsoB1xrWR86zp1WwzE,45573
 mistralai/mistral_jobs.py,sha256=EQHFFxFkkx6XvPX-9S8TRZvVSOLUL7z91cg56J8pskQ,31114
-mistralai/models/__init__.py,sha256=3ZtDRtA2WtDv6pOqLkBFkI2M1WmJV7hc4A6ihqudzrw,43330
+mistralai/models/__init__.py,sha256=h-8plap2U1shoIGyNyeNpQmSH5GvIsVZfoT4wC_4Fcc,43395
 mistralai/models/agent.py,sha256=dx_uXYDNWoENtgK4InGqs1A9eCw1xDphMPCF_Pv43VQ,4185
 mistralai/models/agentconversation.py,sha256=WaPQn3pGiD2c6bNaP7Ya1akVQt54aK7qDtAEnHAjwrk,2086
 mistralai/models/agentcreationrequest.py,sha256=u9WM95ywRmDqnwx0yfEOT1FnJbiPY8lNtju71d3ix4A,3952
@@ -212,7 +212,7 @@ mistralai/models/builtinconnectors.py,sha256=cX1M7Q_2tsWeuH-lKWomXED7xN7Du6BJKvY
 mistralai/models/chatclassificationrequest.py,sha256=PmU036oOlGqfd75hNESDUJiN4uJNYguACoCt6CzBC2M,534
 mistralai/models/chatcompletionchoice.py,sha256=6iIFLZj2KYx0HFfzS3-E3sNXG6mPEAlDyXxIA5iZI_U,849
 mistralai/models/chatcompletionrequest.py,sha256=6Innwpi7UnKmyauATOJForAVvW0tkSnbjsiQOOp5OKg,9777
-mistralai/models/chatcompletionresponse.py,sha256=sLE-_Bx9W5rH2-HE2fBWPVbJbmBWx_jSY2mJ3KBEn6w,792
+mistralai/models/chatcompletionresponse.py,sha256=px0hjCapAtTP50u36hiQdPcC9X6LU81Nq5aJ3AlofjM,709
 mistralai/models/chatcompletionstreamrequest.py,sha256=0NFa_nMMRmHU66Hsgu1Zm4fggT0AzvW_imrkyZ4sUxc,9465
 mistralai/models/chatmoderationrequest.py,sha256=x1eAoxx_GhaxqGRe4wsqNaUi59K39HQakkedLJVUVD8,2236
 mistralai/models/checkpointout.py,sha256=A2kXS8-VT_1lbg3brifVjZD6tXdsET8vLqBm2a-yXgA,1109
@@ -256,7 +256,8 @@ mistralai/models/deletemodelout.py,sha256=W_crO0WtksoKUgq5s9Yh8zS8RxSuyKYQCBt1i8
 mistralai/models/deltamessage.py,sha256=7NtvEjdmBOl86rwOx7x2fcCCJSzIF8K6-eu-G9Wr9PI,1939
 mistralai/models/documentlibrarytool.py,sha256=EN50sX8wgfAw7mF6W8BkOwKyqRvEzGvHgINn-evuCcg,654
 mistralai/models/documenturlchunk.py,sha256=j3JB_Cy1eIRY7fTJe8AvQrdrLEA6xsJcM1l9_a1Sh68,1704
-mistralai/models/embeddingrequest.py,sha256=5GUp8OUrIAcoSdhJLO8Ue45_kGuzVaCqz85g0ZQx3gM,864
+mistralai/models/embeddingdtype.py,sha256=c7L-PKhBgPVPZeMGuMub0ZOs0MdxMbpW2ebE0t7oEpU,209
+mistralai/models/embeddingrequest.py,sha256=BE3L0ImjkdmzwOSUQnC_vCsnvyeQqWmqRw4xgzX6vJc,2258
 mistralai/models/embeddingresponse.py,sha256=te6E_LYEzRjHJ9QREmsFp5PeNP2J_8ALVjyb1T20pNA,663
 mistralai/models/embeddingresponsedata.py,sha256=fJ3mrZqyBBBE40a6iegOJX3DVDfgyMRq23ByeGSTLFk,534
 mistralai/models/eventout.py,sha256=TouRJeISBLphMTPHfgSOpuoOmbGDVohPOrdgHyExMpw,1633
@@ -270,7 +271,7 @@ mistralai/models/files_api_routes_upload_fileop.py,sha256=gIGH5xcPryWYkj1FmNv_0-
 mistralai/models/fileschema.py,sha256=n_IjCdNOrC2fuzkv75wJn01XvqGTmPK3JqAFSHaOiMA,2597
 mistralai/models/filesignedurl.py,sha256=VwvuhzhJulAB99Qxz6zr-2F1aINosAfaSxU0IhytDSU,289
 mistralai/models/fimcompletionrequest.py,sha256=wWDCkQ_PMnjB8DrIuIvVJlPGqQtTpVDHt4p7xJ204Ug,6565
-mistralai/models/fimcompletionresponse.py,sha256=_QwzRuL3KuKkyrA4Fxp366SW0H0EzOA7f4FLkWLm-PM,790
+mistralai/models/fimcompletionresponse.py,sha256=qNgb2WFVgkaW7Isfkk2Aol1gTV9UkhQomcDAhoPDeYw,707
 mistralai/models/fimcompletionstreamrequest.py,sha256=fxuR8FDOWMwIqlYU9ttAfGeRdVgTz4l2k26_OEfxelg,5944
 mistralai/models/finetuneablemodeltype.py,sha256=XmTpXeQU8AINnn1kVmXldFUauCaEnRtJNFAXUTVb6RQ,197
 mistralai/models/ftclassifierlossfunction.py,sha256=ApQB8ssAh2yE26-CljxPO7Jc5lxq3OoBPR4rUp-Td9U,203
@@ -390,7 +391,7 @@ mistralai/utils/serializers.py,sha256=EGH40Pgp3sSK9uM4PxL7_SYzSHtmo-Uy6QIE5xLVg6
 mistralai/utils/url.py,sha256=BgGPgcTA6MRK4bF8fjP2dUopN3NzEzxWMXPBVg8NQUA,5254
 mistralai/utils/values.py,sha256=CcaCXEa3xHhkUDROyXZocN8f0bdITftv9Y0P9lTf0YM,3517
 mistralai/version.py,sha256=iosXhlXclBwBqlADFKEilxAC2wWKbtuBKi87AmPi7s8,196
-mistralai-1.8.0.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
-mistralai-1.8.0.dist-info/METADATA,sha256=dL-LRZiJrDkGgzpt53FZDtT7zka1PQIEIWGX4sD_Ezo,33379
-mistralai-1.8.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-mistralai-1.8.0.dist-info/RECORD,,
+mistralai-1.8.1.dist-info/LICENSE,sha256=rUtQ_9GD0OyLPlb-2uWVdfE87hzudMRmsW-tS-0DK-0,11340
+mistralai-1.8.1.dist-info/METADATA,sha256=zvxBN6ahY13VTUqmneFHFNXCrEK7Fl5nlhKFGkj_PKw,33379
+mistralai-1.8.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+mistralai-1.8.1.dist-info/RECORD,,