mistralai 1.9.1__py3-none-any.whl → 1.9.3__py3-none-any.whl

mistralai/models/libraryinupdate.py

@@ -0,0 +1,47 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing_extensions import NotRequired, TypedDict
+
+
+class LibraryInUpdateTypedDict(TypedDict):
+    name: NotRequired[Nullable[str]]
+    description: NotRequired[Nullable[str]]
+
+
+class LibraryInUpdate(BaseModel):
+    name: OptionalNullable[str] = UNSET
+
+    description: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["name", "description"]
+        nullable_fields = ["name", "description"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in type(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/libraryout.py

@@ -0,0 +1,107 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from datetime import datetime
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing_extensions import NotRequired, TypedDict
+
+
+class LibraryOutTypedDict(TypedDict):
+    id: str
+    name: str
+    created_at: datetime
+    updated_at: datetime
+    owner_id: str
+    owner_type: str
+    total_size: int
+    nb_documents: int
+    chunk_size: Nullable[int]
+    emoji: NotRequired[Nullable[str]]
+    description: NotRequired[Nullable[str]]
+    generated_name: NotRequired[Nullable[str]]
+    generated_description: NotRequired[Nullable[str]]
+    explicit_user_members_count: NotRequired[Nullable[int]]
+    explicit_workspace_members_count: NotRequired[Nullable[int]]
+    org_sharing_role: NotRequired[Nullable[str]]
+
+
+class LibraryOut(BaseModel):
+    id: str
+
+    name: str
+
+    created_at: datetime
+
+    updated_at: datetime
+
+    owner_id: str
+
+    owner_type: str
+
+    total_size: int
+
+    nb_documents: int
+
+    chunk_size: Nullable[int]
+
+    emoji: OptionalNullable[str] = UNSET
+
+    description: OptionalNullable[str] = UNSET
+
+    generated_name: OptionalNullable[str] = UNSET
+
+    generated_description: OptionalNullable[str] = UNSET
+
+    explicit_user_members_count: OptionalNullable[int] = UNSET
+
+    explicit_workspace_members_count: OptionalNullable[int] = UNSET
+
+    org_sharing_role: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "emoji",
+            "description",
+            "generated_name",
+            "generated_description",
+            "explicit_user_members_count",
+            "explicit_workspace_members_count",
+            "org_sharing_role",
+        ]
+        nullable_fields = [
+            "chunk_size",
+            "emoji",
+            "description",
+            "generated_name",
+            "generated_description",
+            "explicit_user_members_count",
+            "explicit_workspace_members_count",
+            "org_sharing_role",
+        ]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in type(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/listdocumentout.py

@@ -0,0 +1,19 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .documentout import DocumentOut, DocumentOutTypedDict
+from .paginationinfo import PaginationInfo, PaginationInfoTypedDict
+from mistralai.types import BaseModel
+from typing import List
+from typing_extensions import TypedDict
+
+
+class ListDocumentOutTypedDict(TypedDict):
+    pagination: PaginationInfoTypedDict
+    data: List[DocumentOutTypedDict]
+
+
+class ListDocumentOut(BaseModel):
+    pagination: PaginationInfo
+
+    data: List[DocumentOut]

mistralai/models/listlibraryout.py

@@ -0,0 +1,15 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .libraryout import LibraryOut, LibraryOutTypedDict
+from mistralai.types import BaseModel
+from typing import List
+from typing_extensions import TypedDict
+
+
+class ListLibraryOutTypedDict(TypedDict):
+    data: List[LibraryOutTypedDict]
+
+
+class ListLibraryOut(BaseModel):
+    data: List[LibraryOut]

mistralai/models/listsharingout.py

@@ -0,0 +1,15 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .sharingout import SharingOut, SharingOutTypedDict
+from mistralai.types import BaseModel
+from typing import List
+from typing_extensions import TypedDict
+
+
+class ListSharingOutTypedDict(TypedDict):
+    data: List[SharingOutTypedDict]
+
+
+class ListSharingOut(BaseModel):
+    data: List[SharingOut]

mistralai/models/messageinputentry.py

@@ -14,7 +14,7 @@ from typing_extensions import NotRequired, TypeAliasType, TypedDict
 
 Object = Literal["entry"]
 
-Type = Literal["message.input"]
+MessageInputEntryType = Literal["message.input"]
 
 MessageInputEntryRole = Literal["assistant", "user"]
 
@@ -35,10 +35,11 @@ class MessageInputEntryTypedDict(TypedDict):
     role: MessageInputEntryRole
     content: MessageInputEntryContentTypedDict
     object: NotRequired[Object]
-    type: NotRequired[Type]
+    type: NotRequired[MessageInputEntryType]
     created_at: NotRequired[datetime]
     completed_at: NotRequired[Nullable[datetime]]
     id: NotRequired[str]
+    prefix: NotRequired[bool]
 
 
 class MessageInputEntry(BaseModel):
@@ -50,7 +51,7 @@ class MessageInputEntry(BaseModel):
 
     object: Optional[Object] = "entry"
 
-    type: Optional[Type] = "message.input"
+    type: Optional[MessageInputEntryType] = "message.input"
 
     created_at: Optional[datetime] = None
 
@@ -58,9 +59,18 @@ class MessageInputEntry(BaseModel):
 
     id: Optional[str] = None
 
+    prefix: Optional[bool] = False
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["object", "type", "created_at", "completed_at", "id"]
+        optional_fields = [
+            "object",
+            "type",
+            "created_at",
+            "completed_at",
+            "id",
+            "prefix",
+        ]
         nullable_fields = ["completed_at"]
         null_default_fields = []
 

mistralai/models/paginationinfo.py

@@ -0,0 +1,25 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class PaginationInfoTypedDict(TypedDict):
+    total_items: int
+    total_pages: int
+    current_page: int
+    page_size: int
+    has_more: bool
+
+
+class PaginationInfo(BaseModel):
+    total_items: int
+
+    total_pages: int
+
+    current_page: int
+
+    page_size: int
+
+    has_more: bool

mistralai/models/processingstatusout.py

@@ -0,0 +1,16 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class ProcessingStatusOutTypedDict(TypedDict):
+    document_id: str
+    processing_status: str
+
+
+class ProcessingStatusOut(BaseModel):
+    document_id: str
+
+    processing_status: str

mistralai/models/retrievefileout.py

@@ -28,6 +28,8 @@ class RetrieveFileOutTypedDict(TypedDict):
     source: Source
     deleted: bool
     num_lines: NotRequired[Nullable[int]]
+    mimetype: NotRequired[Nullable[str]]
+    signature: NotRequired[Nullable[str]]
 
 
 class RetrieveFileOut(BaseModel):
@@ -56,10 +58,14 @@ class RetrieveFileOut(BaseModel):
 
     num_lines: OptionalNullable[int] = UNSET
 
+    mimetype: OptionalNullable[str] = UNSET
+
+    signature: OptionalNullable[str] = UNSET
+
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["num_lines"]
-        nullable_fields = ["num_lines"]
+        optional_fields = ["num_lines", "mimetype", "signature"]
+        nullable_fields = ["num_lines", "mimetype", "signature"]
         null_default_fields = []
 
         serialized = handler(self)

mistralai/models/shareenum.py

@@ -0,0 +1,8 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import UnrecognizedStr
+from typing import Literal, Union
+
+
+ShareEnum = Union[Literal["Viewer", "Editor"], UnrecognizedStr]

mistralai/models/sharingdelete.py

@@ -0,0 +1,26 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .entitytype import EntityType
+from mistralai.types import BaseModel
+from mistralai.utils import validate_open_enum
+from pydantic.functional_validators import PlainValidator
+from typing_extensions import Annotated, TypedDict
+
+
+class SharingDeleteTypedDict(TypedDict):
+    org_id: str
+    share_with_uuid: str
+    r"""The id of the entity (user, workspace or organization) to share with"""
+    share_with_type: EntityType
+    r"""The type of entity, used to share a library."""
+
+
+class SharingDelete(BaseModel):
+    org_id: str
+
+    share_with_uuid: str
+    r"""The id of the entity (user, workspace or organization) to share with"""
+
+    share_with_type: Annotated[EntityType, PlainValidator(validate_open_enum(False))]
+    r"""The type of entity, used to share a library."""

mistralai/models/sharingin.py

@@ -0,0 +1,30 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .entitytype import EntityType
+from .shareenum import ShareEnum
+from mistralai.types import BaseModel
+from mistralai.utils import validate_open_enum
+from pydantic.functional_validators import PlainValidator
+from typing_extensions import Annotated, TypedDict
+
+
+class SharingInTypedDict(TypedDict):
+    org_id: str
+    level: ShareEnum
+    share_with_uuid: str
+    r"""The id of the entity (user, workspace or organization) to share with"""
+    share_with_type: EntityType
+    r"""The type of entity, used to share a library."""
+
+
+class SharingIn(BaseModel):
+    org_id: str
+
+    level: Annotated[ShareEnum, PlainValidator(validate_open_enum(False))]
+
+    share_with_uuid: str
+    r"""The id of the entity (user, workspace or organization) to share with"""
+
+    share_with_type: Annotated[EntityType, PlainValidator(validate_open_enum(False))]
+    r"""The type of entity, used to share a library."""

mistralai/models/sharingout.py

@@ -0,0 +1,59 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from pydantic import model_serializer
+from typing_extensions import NotRequired, TypedDict
+
+
+class SharingOutTypedDict(TypedDict):
+    library_id: str
+    org_id: str
+    role: str
+    share_with_type: str
+    share_with_uuid: str
+    user_id: NotRequired[Nullable[str]]
+
+
+class SharingOut(BaseModel):
+    library_id: str
+
+    org_id: str
+
+    role: str
+
+    share_with_type: str
+
+    share_with_uuid: str
+
+    user_id: OptionalNullable[str] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["user_id"]
+        nullable_fields = ["user_id"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in type(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/ssetypes.py

@@ -10,6 +10,7 @@ SSETypes = Literal[
     "conversation.response.error",
     "message.output.delta",
     "tool.execution.started",
+    "tool.execution.delta",
     "tool.execution.done",
     "agent.handoff.started",
     "agent.handoff.done",

mistralai/models/thinkchunk.py

@@ -0,0 +1,35 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .referencechunk import ReferenceChunk, ReferenceChunkTypedDict
+from .textchunk import TextChunk, TextChunkTypedDict
+from mistralai.types import BaseModel
+from typing import List, Literal, Optional, Union
+from typing_extensions import NotRequired, TypeAliasType, TypedDict
+
+
+ThinkingTypedDict = TypeAliasType(
+    "ThinkingTypedDict", Union[ReferenceChunkTypedDict, TextChunkTypedDict]
+)
+
+
+Thinking = TypeAliasType("Thinking", Union[ReferenceChunk, TextChunk])
+
+
+ThinkChunkType = Literal["thinking"]
+
+
+class ThinkChunkTypedDict(TypedDict):
+    thinking: List[ThinkingTypedDict]
+    closed: NotRequired[bool]
+    r"""Whether the thinking chunk is closed or not. Currently only used for prefixing."""
+    type: NotRequired[ThinkChunkType]
+
+
+class ThinkChunk(BaseModel):
+    thinking: List[Thinking]
+
+    closed: Optional[bool] = None
+    r"""Whether the thinking chunk is closed or not. Currently only used for prefixing."""
+
+    type: Optional[ThinkChunkType] = "thinking"

mistralai/models/timestampgranularity.py

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

mistralai/models/toolcall.py

@@ -3,18 +3,28 @@
 from __future__ import annotations
 from .functioncall import FunctionCall, FunctionCallTypedDict
 from .tooltypes import ToolTypes
-from mistralai.types import BaseModel
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
 from mistralai.utils import validate_open_enum
+from pydantic import model_serializer
 from pydantic.functional_validators import PlainValidator
 from typing import Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
 
 
+class MetadataTypedDict(TypedDict):
+    pass
+
+
+class Metadata(BaseModel):
+    pass
+
+
 class ToolCallTypedDict(TypedDict):
     function: FunctionCallTypedDict
     id: NotRequired[str]
     type: NotRequired[ToolTypes]
     index: NotRequired[int]
+    metadata: NotRequired[Nullable[MetadataTypedDict]]
 
 
 class ToolCall(BaseModel):
@@ -27,3 +37,35 @@ class ToolCall(BaseModel):
     )
 
     index: Optional[int] = 0
+
+    metadata: OptionalNullable[Metadata] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["id", "type", "index", "metadata"]
+        nullable_fields = ["metadata"]
+        null_default_fields = []
+
+        serialized = handler(self)
+
+        m = {}
+
+        for n, f in type(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/toolexecutiondeltaevent.py

@@ -0,0 +1,34 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .builtinconnectors import BuiltInConnectors
+from datetime import datetime
+from mistralai.types import BaseModel
+from typing import Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+ToolExecutionDeltaEventType = Literal["tool.execution.delta"]
+
+
+class ToolExecutionDeltaEventTypedDict(TypedDict):
+    id: str
+    name: BuiltInConnectors
+    arguments: str
+    type: NotRequired[ToolExecutionDeltaEventType]
+    created_at: NotRequired[datetime]
+    output_index: NotRequired[int]
+
+
+class ToolExecutionDeltaEvent(BaseModel):
+    id: str
+
+    name: BuiltInConnectors
+
+    arguments: str
+
+    type: Optional[ToolExecutionDeltaEventType] = "tool.execution.delta"
+
+    created_at: Optional[datetime] = None
+
+    output_index: Optional[int] = 0

mistralai/models/toolexecutionentry.py

@@ -16,6 +16,7 @@ ToolExecutionEntryType = Literal["tool.execution"]
 
 class ToolExecutionEntryTypedDict(TypedDict):
     name: BuiltInConnectors
+    arguments: str
     object: NotRequired[ToolExecutionEntryObject]
     type: NotRequired[ToolExecutionEntryType]
     created_at: NotRequired[datetime]
@@ -27,6 +28,8 @@ class ToolExecutionEntryTypedDict(TypedDict):
 class ToolExecutionEntry(BaseModel):
     name: BuiltInConnectors
 
+    arguments: str
+
     object: Optional[ToolExecutionEntryObject] = "entry"
 
     type: Optional[ToolExecutionEntryType] = "tool.execution"

mistralai/models/toolexecutionstartedevent.py

@@ -14,6 +14,7 @@ ToolExecutionStartedEventType = Literal["tool.execution.started"]
 class ToolExecutionStartedEventTypedDict(TypedDict):
     id: str
     name: BuiltInConnectors
+    arguments: str
     type: NotRequired[ToolExecutionStartedEventType]
     created_at: NotRequired[datetime]
     output_index: NotRequired[int]
@@ -24,6 +25,8 @@ class ToolExecutionStartedEvent(BaseModel):
 
     name: BuiltInConnectors
 
+    arguments: str
+
     type: Optional[ToolExecutionStartedEventType] = "tool.execution.started"
 
     created_at: Optional[datetime] = None

mistralai/models/toolreferencechunk.py

@@ -16,7 +16,8 @@ class ToolReferenceChunkTypedDict(TypedDict):
     title: str
     type: NotRequired[ToolReferenceChunkType]
     url: NotRequired[Nullable[str]]
-    source: NotRequired[Nullable[str]]
+    favicon: NotRequired[Nullable[str]]
+    description: NotRequired[Nullable[str]]
 
 
 class ToolReferenceChunk(BaseModel):
@@ -28,12 +29,14 @@ class ToolReferenceChunk(BaseModel):
 
     url: OptionalNullable[str] = UNSET
 
-    source: OptionalNullable[str] = UNSET
+    favicon: OptionalNullable[str] = UNSET
+
+    description: OptionalNullable[str] = UNSET
 
     @model_serializer(mode="wrap")
     def serialize_model(self, handler):
-        optional_fields = ["type", "url", "source"]
-        nullable_fields = ["url", "source"]
+        optional_fields = ["type", "url", "favicon", "description"]
+        nullable_fields = ["url", "favicon", "description"]
         null_default_fields = []
 
         serialized = handler(self)