mistralai 1.8.0__tar.gz → 1.8.1__tar.gz

{mistralai-1.8.0 → mistralai-1.8.1}/PKG-INFO

@@ -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 → mistralai-1.8.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mistralai"
-version = "1.8.0"
+version = "1.8.1"
 description = "Python Client SDK for the Mistral AI API."
 authors = [{ name = "Mistral" },]
 readme = "README-PYPI.md"
@@ -69,7 +69,13 @@ module = "typing_inspect"
 ignore_missing_imports = true
 
 [[tool.mypy.overrides]]
-module = "jsonpath"
+module = [
+    "jsonpath.*",
+    "typing_inspect.*",
+    "authlib.*",
+    "mcp.*",
+    "griffe.*"
+]
 ignore_missing_imports = true
 
 [tool.pyright]

{mistralai-1.8.0 → mistralai-1.8.1}/src/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-1.8.0 → mistralai-1.8.1}/src/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-1.8.0 → mistralai-1.8.1}/src/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-1.8.0 → mistralai-1.8.1}/src/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-1.8.0 → mistralai-1.8.1}/src/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-1.8.0 → mistralai-1.8.1}/src/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-1.8.0 → mistralai-1.8.1}/src/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-1.8.1/src/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-1.8.1/src/mistralai/models/embeddingrequest.py

@@ -0,0 +1,72 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .embeddingdtype import EmbeddingDtype
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+import pydantic
+from pydantic import model_serializer
+from typing import List, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
+
+
+EmbeddingRequestInputsTypedDict = TypeAliasType(
+    "EmbeddingRequestInputsTypedDict", Union[str, List[str]]
+)
+r"""Text to embed."""
+
+
+EmbeddingRequestInputs = TypeAliasType("EmbeddingRequestInputs", Union[str, List[str]])
+r"""Text to embed."""
+
+
+class EmbeddingRequestTypedDict(TypedDict):
+    model: str
+    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):
+    model: str
+    r"""ID of the model to use."""
+
+    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-1.8.0 → mistralai-1.8.1}/src/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/src/mistralai/models/embeddingrequest.py

@@ -1,32 +0,0 @@
-"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
-
-from __future__ import annotations
-from mistralai.types import BaseModel
-import pydantic
-from typing import List, Union
-from typing_extensions import Annotated, TypeAliasType, TypedDict
-
-
-EmbeddingRequestInputsTypedDict = TypeAliasType(
-    "EmbeddingRequestInputsTypedDict", Union[str, List[str]]
-)
-r"""Text to embed."""
-
-
-EmbeddingRequestInputs = TypeAliasType("EmbeddingRequestInputs", Union[str, List[str]])
-r"""Text to embed."""
-
-
-class EmbeddingRequestTypedDict(TypedDict):
-    model: str
-    r"""ID of the model to use."""
-    inputs: EmbeddingRequestInputsTypedDict
-    r"""Text to embed."""
-
-
-class EmbeddingRequest(BaseModel):
-    model: str
-    r"""ID of the model to use."""
-
-    inputs: Annotated[EmbeddingRequestInputs, pydantic.Field(alias="input")]
-    r"""Text to embed."""