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

mistralai/models/agent.py

@@ -0,0 +1,129 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .codeinterpretertool import CodeInterpreterTool, CodeInterpreterToolTypedDict
+from .completionargs import CompletionArgs, CompletionArgsTypedDict
+from .documentlibrarytool import DocumentLibraryTool, DocumentLibraryToolTypedDict
+from .functiontool import FunctionTool, FunctionToolTypedDict
+from .imagegenerationtool import ImageGenerationTool, ImageGenerationToolTypedDict
+from .websearchpremiumtool import WebSearchPremiumTool, WebSearchPremiumToolTypedDict
+from .websearchtool import WebSearchTool, WebSearchToolTypedDict
+from datetime import datetime
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from mistralai.utils import get_discriminator
+from pydantic import Discriminator, Tag, model_serializer
+from typing import List, Literal, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
+
+
+AgentToolsTypedDict = TypeAliasType(
+    "AgentToolsTypedDict",
+    Union[
+        WebSearchToolTypedDict,
+        WebSearchPremiumToolTypedDict,
+        CodeInterpreterToolTypedDict,
+        ImageGenerationToolTypedDict,
+        FunctionToolTypedDict,
+        DocumentLibraryToolTypedDict,
+    ],
+)
+
+
+AgentTools = Annotated[
+    Union[
+        Annotated[CodeInterpreterTool, Tag("code_interpreter")],
+        Annotated[DocumentLibraryTool, Tag("document_library")],
+        Annotated[FunctionTool, Tag("function")],
+        Annotated[ImageGenerationTool, Tag("image_generation")],
+        Annotated[WebSearchTool, Tag("web_search")],
+        Annotated[WebSearchPremiumTool, Tag("web_search_premium")],
+    ],
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]
+
+
+AgentObject = Literal["agent"]
+
+
+class AgentTypedDict(TypedDict):
+    model: str
+    name: str
+    id: str
+    version: int
+    created_at: datetime
+    updated_at: datetime
+    instructions: NotRequired[Nullable[str]]
+    r"""Instruction prompt the model will follow during the conversation."""
+    tools: NotRequired[List[AgentToolsTypedDict]]
+    r"""List of tools which are available to the model during the conversation."""
+    completion_args: NotRequired[CompletionArgsTypedDict]
+    r"""White-listed arguments from the completion API"""
+    description: NotRequired[Nullable[str]]
+    handoffs: NotRequired[Nullable[List[str]]]
+    object: NotRequired[AgentObject]
+
+
+class Agent(BaseModel):
+    model: str
+
+    name: str
+
+    id: str
+
+    version: int
+
+    created_at: datetime
+
+    updated_at: datetime
+
+    instructions: OptionalNullable[str] = UNSET
+    r"""Instruction prompt the model will follow during the conversation."""
+
+    tools: Optional[List[AgentTools]] = None
+    r"""List of tools which are available to the model during the conversation."""
+
+    completion_args: Optional[CompletionArgs] = None
+    r"""White-listed arguments from the completion API"""
+
+    description: OptionalNullable[str] = UNSET
+
+    handoffs: OptionalNullable[List[str]] = UNSET
+
+    object: Optional[AgentObject] = "agent"
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "instructions",
+            "tools",
+            "completion_args",
+            "description",
+            "handoffs",
+            "object",
+        ]
+        nullable_fields = ["instructions", "description", "handoffs"]
+        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/agentconversation.py

@@ -0,0 +1,71 @@
+"""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 import Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+AgentConversationObject = Literal["conversation"]
+
+
+class AgentConversationTypedDict(TypedDict):
+    id: str
+    created_at: datetime
+    updated_at: datetime
+    agent_id: str
+    name: NotRequired[Nullable[str]]
+    r"""Name given to the conversation."""
+    description: NotRequired[Nullable[str]]
+    r"""Description of the what the conversation is about."""
+    object: NotRequired[AgentConversationObject]
+
+
+class AgentConversation(BaseModel):
+    id: str
+
+    created_at: datetime
+
+    updated_at: datetime
+
+    agent_id: str
+
+    name: OptionalNullable[str] = UNSET
+    r"""Name given to the conversation."""
+
+    description: OptionalNullable[str] = UNSET
+    r"""Description of the what the conversation is about."""
+
+    object: Optional[AgentConversationObject] = "conversation"
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["name", "description", "object"]
+        nullable_fields = ["name", "description"]
+        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/agentcreationrequest.py

@@ -0,0 +1,109 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .codeinterpretertool import CodeInterpreterTool, CodeInterpreterToolTypedDict
+from .completionargs import CompletionArgs, CompletionArgsTypedDict
+from .documentlibrarytool import DocumentLibraryTool, DocumentLibraryToolTypedDict
+from .functiontool import FunctionTool, FunctionToolTypedDict
+from .imagegenerationtool import ImageGenerationTool, ImageGenerationToolTypedDict
+from .websearchpremiumtool import WebSearchPremiumTool, WebSearchPremiumToolTypedDict
+from .websearchtool import WebSearchTool, WebSearchToolTypedDict
+from mistralai.types import BaseModel, Nullable, OptionalNullable, UNSET, UNSET_SENTINEL
+from mistralai.utils import get_discriminator
+from pydantic import Discriminator, Tag, model_serializer
+from typing import List, Optional, Union
+from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
+
+
+AgentCreationRequestToolsTypedDict = TypeAliasType(
+    "AgentCreationRequestToolsTypedDict",
+    Union[
+        WebSearchToolTypedDict,
+        WebSearchPremiumToolTypedDict,
+        CodeInterpreterToolTypedDict,
+        ImageGenerationToolTypedDict,
+        FunctionToolTypedDict,
+        DocumentLibraryToolTypedDict,
+    ],
+)
+
+
+AgentCreationRequestTools = Annotated[
+    Union[
+        Annotated[CodeInterpreterTool, Tag("code_interpreter")],
+        Annotated[DocumentLibraryTool, Tag("document_library")],
+        Annotated[FunctionTool, Tag("function")],
+        Annotated[ImageGenerationTool, Tag("image_generation")],
+        Annotated[WebSearchTool, Tag("web_search")],
+        Annotated[WebSearchPremiumTool, Tag("web_search_premium")],
+    ],
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]
+
+
+class AgentCreationRequestTypedDict(TypedDict):
+    model: str
+    name: str
+    instructions: NotRequired[Nullable[str]]
+    r"""Instruction prompt the model will follow during the conversation."""
+    tools: NotRequired[List[AgentCreationRequestToolsTypedDict]]
+    r"""List of tools which are available to the model during the conversation."""
+    completion_args: NotRequired[CompletionArgsTypedDict]
+    r"""White-listed arguments from the completion API"""
+    description: NotRequired[Nullable[str]]
+    handoffs: NotRequired[Nullable[List[str]]]
+
+
+class AgentCreationRequest(BaseModel):
+    model: str
+
+    name: str
+
+    instructions: OptionalNullable[str] = UNSET
+    r"""Instruction prompt the model will follow during the conversation."""
+
+    tools: Optional[List[AgentCreationRequestTools]] = None
+    r"""List of tools which are available to the model during the conversation."""
+
+    completion_args: Optional[CompletionArgs] = None
+    r"""White-listed arguments from the completion API"""
+
+    description: OptionalNullable[str] = UNSET
+
+    handoffs: OptionalNullable[List[str]] = UNSET
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = [
+            "instructions",
+            "tools",
+            "completion_args",
+            "description",
+            "handoffs",
+        ]
+        nullable_fields = ["instructions", "description", "handoffs"]
+        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/agenthandoffdoneevent.py

@@ -0,0 +1,33 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from datetime import datetime
+from mistralai.types import BaseModel
+from typing import Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+AgentHandoffDoneEventType = Literal["agent.handoff.done"]
+
+
+class AgentHandoffDoneEventTypedDict(TypedDict):
+    id: str
+    next_agent_id: str
+    next_agent_name: str
+    type: NotRequired[AgentHandoffDoneEventType]
+    created_at: NotRequired[datetime]
+    output_index: NotRequired[int]
+
+
+class AgentHandoffDoneEvent(BaseModel):
+    id: str
+
+    next_agent_id: str
+
+    next_agent_name: str
+
+    type: Optional[AgentHandoffDoneEventType] = "agent.handoff.done"
+
+    created_at: Optional[datetime] = None
+
+    output_index: Optional[int] = 0

mistralai/models/agenthandoffentry.py

@@ -0,0 +1,75 @@
+"""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 import Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+AgentHandoffEntryObject = Literal["entry"]
+
+AgentHandoffEntryType = Literal["agent.handoff"]
+
+
+class AgentHandoffEntryTypedDict(TypedDict):
+    previous_agent_id: str
+    previous_agent_name: str
+    next_agent_id: str
+    next_agent_name: str
+    object: NotRequired[AgentHandoffEntryObject]
+    type: NotRequired[AgentHandoffEntryType]
+    created_at: NotRequired[datetime]
+    completed_at: NotRequired[Nullable[datetime]]
+    id: NotRequired[str]
+
+
+class AgentHandoffEntry(BaseModel):
+    previous_agent_id: str
+
+    previous_agent_name: str
+
+    next_agent_id: str
+
+    next_agent_name: str
+
+    object: Optional[AgentHandoffEntryObject] = "entry"
+
+    type: Optional[AgentHandoffEntryType] = "agent.handoff"
+
+    created_at: Optional[datetime] = None
+
+    completed_at: OptionalNullable[datetime] = UNSET
+
+    id: Optional[str] = None
+
+    @model_serializer(mode="wrap")
+    def serialize_model(self, handler):
+        optional_fields = ["object", "type", "created_at", "completed_at", "id"]
+        nullable_fields = ["completed_at"]
+        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/agenthandoffstartedevent.py

@@ -0,0 +1,33 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from datetime import datetime
+from mistralai.types import BaseModel
+from typing import Literal, Optional
+from typing_extensions import NotRequired, TypedDict
+
+
+AgentHandoffStartedEventType = Literal["agent.handoff.started"]
+
+
+class AgentHandoffStartedEventTypedDict(TypedDict):
+    id: str
+    previous_agent_id: str
+    previous_agent_name: str
+    type: NotRequired[AgentHandoffStartedEventType]
+    created_at: NotRequired[datetime]
+    output_index: NotRequired[int]
+
+
+class AgentHandoffStartedEvent(BaseModel):
+    id: str
+
+    previous_agent_id: str
+
+    previous_agent_name: str
+
+    type: Optional[AgentHandoffStartedEventType] = "agent.handoff.started"
+
+    created_at: Optional[datetime] = None
+
+    output_index: Optional[int] = 0

mistralai/models/agents_api_v1_agents_getop.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 mistralai.utils import FieldMetadata, PathParamMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class AgentsAPIV1AgentsGetRequestTypedDict(TypedDict):
+    agent_id: str
+
+
+class AgentsAPIV1AgentsGetRequest(BaseModel):
+    agent_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]

mistralai/models/agents_api_v1_agents_listop.py

@@ -0,0 +1,24 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, QueryParamMetadata
+from typing import Optional
+from typing_extensions import Annotated, NotRequired, TypedDict
+
+
+class AgentsAPIV1AgentsListRequestTypedDict(TypedDict):
+    page: NotRequired[int]
+    page_size: NotRequired[int]
+
+
+class AgentsAPIV1AgentsListRequest(BaseModel):
+    page: Annotated[
+        Optional[int],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = 0
+
+    page_size: Annotated[
+        Optional[int],
+        FieldMetadata(query=QueryParamMetadata(style="form", explode=True)),
+    ] = 20

mistralai/models/agents_api_v1_agents_update_versionop.py

@@ -0,0 +1,21 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata, QueryParamMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class AgentsAPIV1AgentsUpdateVersionRequestTypedDict(TypedDict):
+    agent_id: str
+    version: int
+
+
+class AgentsAPIV1AgentsUpdateVersionRequest(BaseModel):
+    agent_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    version: Annotated[
+        int, FieldMetadata(query=QueryParamMetadata(style="form", explode=True))
+    ]

mistralai/models/agents_api_v1_agents_updateop.py

@@ -0,0 +1,23 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .agentupdaterequest import AgentUpdateRequest, AgentUpdateRequestTypedDict
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata, RequestMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class AgentsAPIV1AgentsUpdateRequestTypedDict(TypedDict):
+    agent_id: str
+    agent_update_request: AgentUpdateRequestTypedDict
+
+
+class AgentsAPIV1AgentsUpdateRequest(BaseModel):
+    agent_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+    agent_update_request: Annotated[
+        AgentUpdateRequest,
+        FieldMetadata(request=RequestMetadata(media_type="application/json")),
+    ]

mistralai/models/agents_api_v1_conversations_append_streamop.py

@@ -0,0 +1,28 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .conversationappendstreamrequest import (
+    ConversationAppendStreamRequest,
+    ConversationAppendStreamRequestTypedDict,
+)
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata, RequestMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class AgentsAPIV1ConversationsAppendStreamRequestTypedDict(TypedDict):
+    conversation_id: str
+    r"""ID of the conversation to which we append entries."""
+    conversation_append_stream_request: ConversationAppendStreamRequestTypedDict
+
+
+class AgentsAPIV1ConversationsAppendStreamRequest(BaseModel):
+    conversation_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+    r"""ID of the conversation to which we append entries."""
+
+    conversation_append_stream_request: Annotated[
+        ConversationAppendStreamRequest,
+        FieldMetadata(request=RequestMetadata(media_type="application/json")),
+    ]

mistralai/models/agents_api_v1_conversations_appendop.py

@@ -0,0 +1,28 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .conversationappendrequest import (
+    ConversationAppendRequest,
+    ConversationAppendRequestTypedDict,
+)
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata, RequestMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class AgentsAPIV1ConversationsAppendRequestTypedDict(TypedDict):
+    conversation_id: str
+    r"""ID of the conversation to which we append entries."""
+    conversation_append_request: ConversationAppendRequestTypedDict
+
+
+class AgentsAPIV1ConversationsAppendRequest(BaseModel):
+    conversation_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+    r"""ID of the conversation to which we append entries."""
+
+    conversation_append_request: Annotated[
+        ConversationAppendRequest,
+        FieldMetadata(request=RequestMetadata(media_type="application/json")),
+    ]

mistralai/models/agents_api_v1_conversations_getop.py

@@ -0,0 +1,33 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from .agentconversation import AgentConversation, AgentConversationTypedDict
+from .modelconversation import ModelConversation, ModelConversationTypedDict
+from mistralai.types import BaseModel
+from mistralai.utils import FieldMetadata, PathParamMetadata
+from typing import Union
+from typing_extensions import Annotated, TypeAliasType, TypedDict
+
+
+class AgentsAPIV1ConversationsGetRequestTypedDict(TypedDict):
+    conversation_id: str
+
+
+class AgentsAPIV1ConversationsGetRequest(BaseModel):
+    conversation_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]
+
+
+AgentsAPIV1ConversationsGetResponseV1ConversationsGetTypedDict = TypeAliasType(
+    "AgentsAPIV1ConversationsGetResponseV1ConversationsGetTypedDict",
+    Union[AgentConversationTypedDict, ModelConversationTypedDict],
+)
+r"""Successful Response"""
+
+
+AgentsAPIV1ConversationsGetResponseV1ConversationsGet = TypeAliasType(
+    "AgentsAPIV1ConversationsGetResponseV1ConversationsGet",
+    Union[AgentConversation, ModelConversation],
+)
+r"""Successful Response"""

mistralai/models/agents_api_v1_conversations_historyop.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 mistralai.utils import FieldMetadata, PathParamMetadata
+from typing_extensions import Annotated, TypedDict
+
+
+class AgentsAPIV1ConversationsHistoryRequestTypedDict(TypedDict):
+    conversation_id: str
+
+
+class AgentsAPIV1ConversationsHistoryRequest(BaseModel):
+    conversation_id: Annotated[
+        str, FieldMetadata(path=PathParamMetadata(style="simple", explode=False))
+    ]