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

mistralai/models/libraryin.py

@@ -0,0 +1,50 @@
+"""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 LibraryInTypedDict(TypedDict):
+    name: str
+    description: NotRequired[Nullable[str]]
+    chunk_size: NotRequired[Nullable[int]]
+
+
+class LibraryIn(BaseModel):
+    name: str
+
+    description: OptionalNullable[str] = UNSET
+
+    chunk_size: OptionalNullable[int] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["description", "chunk_size"]
+        nullable_fields = ["description", "chunk_size"]
+        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/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/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/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)

{mistralai-1.9.1.dist-info → mistralai-1.9.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: mistralai
-Version: 1.9.1
+Version: 1.9.2
 Summary: Python Client SDK for the Mistral AI API.
 Author: Mistral
 Requires-Python: >=3.9
@@ -495,6 +495,33 @@ The documentation for the GCP SDK is available [here](https://github.com/mistral
 * [append_stream](https://github.com/mistralai/client-python/blob/master/docs/sdks/conversations/README.md#append_stream) - Append new entries to an existing conversation.
 * [restart_stream](https://github.com/mistralai/client-python/blob/master/docs/sdks/conversations/README.md#restart_stream) - Restart a conversation starting from a given entry.
 
+#### [beta.libraries](https://github.com/mistralai/client-python/blob/master/docs/sdks/libraries/README.md)
+
+* [list](https://github.com/mistralai/client-python/blob/master/docs/sdks/libraries/README.md#list) - List all libraries you have access to.
+* [create](https://github.com/mistralai/client-python/blob/master/docs/sdks/libraries/README.md#create) - Create a new Library.
+* [get](https://github.com/mistralai/client-python/blob/master/docs/sdks/libraries/README.md#get) - Detailed information about a specific Library.
+* [delete](https://github.com/mistralai/client-python/blob/master/docs/sdks/libraries/README.md#delete) - Delete a library and all of it's document.
+* [update](https://github.com/mistralai/client-python/blob/master/docs/sdks/libraries/README.md#update) - Update a library.
+
+#### [beta.libraries.accesses](https://github.com/mistralai/client-python/blob/master/docs/sdks/accesses/README.md)
+
+* [list](https://github.com/mistralai/client-python/blob/master/docs/sdks/accesses/README.md#list) - List all of the access to this library.
+* [update_or_create](https://github.com/mistralai/client-python/blob/master/docs/sdks/accesses/README.md#update_or_create) - Create or update an access level.
+* [delete](https://github.com/mistralai/client-python/blob/master/docs/sdks/accesses/README.md#delete) - Delete an access level.
+
+#### [beta.libraries.documents](https://github.com/mistralai/client-python/blob/master/docs/sdks/documents/README.md)
+
+* [list](https://github.com/mistralai/client-python/blob/master/docs/sdks/documents/README.md#list) - List document in a given library.
+* [upload](https://github.com/mistralai/client-python/blob/master/docs/sdks/documents/README.md#upload) - Upload a new document.
+* [get](https://github.com/mistralai/client-python/blob/master/docs/sdks/documents/README.md#get) - Retrieve the metadata of a specific document.
+* [update](https://github.com/mistralai/client-python/blob/master/docs/sdks/documents/README.md#update) - Update the metadata of a specific document.
+* [delete](https://github.com/mistralai/client-python/blob/master/docs/sdks/documents/README.md#delete) - Delete a document.
+* [text_content](https://github.com/mistralai/client-python/blob/master/docs/sdks/documents/README.md#text_content) - Retrieve the text content of a specific document.
+* [status](https://github.com/mistralai/client-python/blob/master/docs/sdks/documents/README.md#status) - Retrieve the processing status of a specific document.
+* [get_signed_url](https://github.com/mistralai/client-python/blob/master/docs/sdks/documents/README.md#get_signed_url) - Retrieve the signed URL of a specific document.
+* [extracted_text_signed_url](https://github.com/mistralai/client-python/blob/master/docs/sdks/documents/README.md#extracted_text_signed_url) - Retrieve the signed URL of text extracted from a given document.
+* [reprocess](https://github.com/mistralai/client-python/blob/master/docs/sdks/documents/README.md#reprocess) - Reprocess a document.
+
 ### [chat](https://github.com/mistralai/client-python/blob/master/docs/sdks/chat/README.md)
 
 * [complete](https://github.com/mistralai/client-python/blob/master/docs/sdks/chat/README.md#complete) - Chat Completion
@@ -614,7 +641,7 @@ with Mistral(
     api_key=os.getenv("MISTRAL_API_KEY", ""),
 ) as mistral:
 
-    res = mistral.files.upload(file={
+    res = mistral.beta.libraries.documents.upload(library_id="a02150d9-5ee0-4877-b62c-28b1fcdf3b76", file={
         "file_name": "example.file",
         "content": open("example.file", "rb"),
     })