mistralai 1.9.11__py3-none-any.whl → 1.10.1__py3-none-any.whl

mistralai/mistral_agents.py

@@ -6,7 +6,7 @@ from mistralai._hooks import HookContext
 from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import get_security_from_env
 from mistralai.utils.unmarshal_json_response import unmarshal_json_response
-from typing import Any, List, Mapping, Optional, Union
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class MistralAgents(BaseSDK):
@@ -29,6 +29,7 @@ class MistralAgents(BaseSDK):
         ] = None,
         description: OptionalNullable[str] = UNSET,
         handoffs: OptionalNullable[List[str]] = UNSET,
+        metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -45,6 +46,7 @@ class MistralAgents(BaseSDK):
         :param completion_args: White-listed arguments from the completion API
         :param description:
         :param handoffs:
+        :param metadata:
         :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
@@ -72,6 +74,7 @@ class MistralAgents(BaseSDK):
             name=name,
             description=description,
             handoffs=handoffs,
+            metadata=metadata,
         )
 
         req = self._build_request(
@@ -150,6 +153,7 @@ class MistralAgents(BaseSDK):
         ] = None,
         description: OptionalNullable[str] = UNSET,
         handoffs: OptionalNullable[List[str]] = UNSET,
+        metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -166,6 +170,7 @@ class MistralAgents(BaseSDK):
         :param completion_args: White-listed arguments from the completion API
         :param description:
         :param handoffs:
+        :param metadata:
         :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
@@ -193,6 +198,7 @@ class MistralAgents(BaseSDK):
             name=name,
             description=description,
             handoffs=handoffs,
+            metadata=metadata,
         )
 
         req = self._build_request_async(
@@ -259,6 +265,11 @@ class MistralAgents(BaseSDK):
         *,
         page: Optional[int] = 0,
         page_size: Optional[int] = 20,
+        deployment_chat: OptionalNullable[bool] = UNSET,
+        sources: OptionalNullable[List[models.RequestSource]] = UNSET,
+        name: OptionalNullable[str] = UNSET,
+        id: OptionalNullable[str] = UNSET,
+        metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -270,6 +281,11 @@ class MistralAgents(BaseSDK):
 
         :param page:
         :param page_size:
+        :param deployment_chat:
+        :param sources:
+        :param name:
+        :param id:
+        :param metadata:
         :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
@@ -288,6 +304,11 @@ class MistralAgents(BaseSDK):
         request = models.AgentsAPIV1AgentsListRequest(
             page=page,
             page_size=page_size,
+            deployment_chat=deployment_chat,
+            sources=sources,
+            name=name,
+            id=id,
+            metadata=metadata,
         )
 
         req = self._build_request(
@@ -351,6 +372,11 @@ class MistralAgents(BaseSDK):
         *,
         page: Optional[int] = 0,
         page_size: Optional[int] = 20,
+        deployment_chat: OptionalNullable[bool] = UNSET,
+        sources: OptionalNullable[List[models.RequestSource]] = UNSET,
+        name: OptionalNullable[str] = UNSET,
+        id: OptionalNullable[str] = UNSET,
+        metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -362,6 +388,11 @@ class MistralAgents(BaseSDK):
 
         :param page:
         :param page_size:
+        :param deployment_chat:
+        :param sources:
+        :param name:
+        :param id:
+        :param metadata:
         :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
@@ -380,6 +411,11 @@ class MistralAgents(BaseSDK):
         request = models.AgentsAPIV1AgentsListRequest(
             page=page,
             page_size=page_size,
+            deployment_chat=deployment_chat,
+            sources=sources,
+            name=name,
+            id=id,
+            metadata=metadata,
         )
 
         req = self._build_request_async(
@@ -442,6 +478,7 @@ class MistralAgents(BaseSDK):
         self,
         *,
         agent_id: str,
+        agent_version: OptionalNullable[int] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -452,6 +489,7 @@ class MistralAgents(BaseSDK):
         Given an agent retrieve an agent entity with its attributes.
 
         :param agent_id:
+        :param agent_version:
         :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
@@ -469,6 +507,7 @@ class MistralAgents(BaseSDK):
 
         request = models.AgentsAPIV1AgentsGetRequest(
             agent_id=agent_id,
+            agent_version=agent_version,
         )
 
         req = self._build_request(
@@ -531,6 +570,7 @@ class MistralAgents(BaseSDK):
         self,
         *,
         agent_id: str,
+        agent_version: OptionalNullable[int] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -541,6 +581,7 @@ class MistralAgents(BaseSDK):
         Given an agent retrieve an agent entity with its attributes.
 
         :param agent_id:
+        :param agent_version:
         :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
@@ -558,6 +599,7 @@ class MistralAgents(BaseSDK):
 
         request = models.AgentsAPIV1AgentsGetRequest(
             agent_id=agent_id,
+            agent_version=agent_version,
         )
 
         req = self._build_request_async(
@@ -634,6 +676,8 @@ class MistralAgents(BaseSDK):
         name: OptionalNullable[str] = UNSET,
         description: OptionalNullable[str] = UNSET,
         handoffs: OptionalNullable[List[str]] = UNSET,
+        deployment_chat: OptionalNullable[bool] = UNSET,
+        metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -651,6 +695,8 @@ class MistralAgents(BaseSDK):
         :param name:
         :param description:
         :param handoffs:
+        :param deployment_chat:
+        :param metadata:
         :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
@@ -680,6 +726,8 @@ class MistralAgents(BaseSDK):
                 name=name,
                 description=description,
                 handoffs=handoffs,
+                deployment_chat=deployment_chat,
+                metadata=metadata,
             ),
         )
 
@@ -764,6 +812,8 @@ class MistralAgents(BaseSDK):
         name: OptionalNullable[str] = UNSET,
         description: OptionalNullable[str] = UNSET,
         handoffs: OptionalNullable[List[str]] = UNSET,
+        deployment_chat: OptionalNullable[bool] = UNSET,
+        metadata: OptionalNullable[Dict[str, Any]] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -781,6 +831,8 @@ class MistralAgents(BaseSDK):
         :param name:
         :param description:
         :param handoffs:
+        :param deployment_chat:
+        :param metadata:
         :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
@@ -810,6 +862,8 @@ class MistralAgents(BaseSDK):
                 name=name,
                 description=description,
                 handoffs=handoffs,
+                deployment_chat=deployment_chat,
+                metadata=metadata,
             ),
         )
 
@@ -876,6 +930,180 @@ class MistralAgents(BaseSDK):
 
         raise models.SDKError("Unexpected response received", http_res)
 
+    def delete(
+        self,
+        *,
+        agent_id: str,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ):
+        r"""Delete an agent entity.
+
+        :param agent_id:
+        :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
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.AgentsAPIV1AgentsDeleteRequest(
+            agent_id=agent_id,
+        )
+
+        req = self._build_request(
+            method="DELETE",
+            path="/v1/agents/{agent_id}",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=False,
+            request_has_path_params=True,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = self.do_request(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="agents_api_v1_agents_delete",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["422", "4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        response_data: Any = None
+        if utils.match_response(http_res, "204", "*"):
+            return
+        if utils.match_response(http_res, "422", "application/json"):
+            response_data = unmarshal_json_response(
+                models.HTTPValidationErrorData, http_res
+            )
+            raise models.HTTPValidationError(response_data, http_res)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = utils.stream_to_text(http_res)
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+
+        raise models.SDKError("Unexpected response received", http_res)
+
+    async def delete_async(
+        self,
+        *,
+        agent_id: str,
+        retries: OptionalNullable[utils.RetryConfig] = UNSET,
+        server_url: Optional[str] = None,
+        timeout_ms: Optional[int] = None,
+        http_headers: Optional[Mapping[str, str]] = None,
+    ):
+        r"""Delete an agent entity.
+
+        :param agent_id:
+        :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
+        :param http_headers: Additional headers to set or replace on requests.
+        """
+        base_url = None
+        url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.timeout_ms
+
+        if server_url is not None:
+            base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
+
+        request = models.AgentsAPIV1AgentsDeleteRequest(
+            agent_id=agent_id,
+        )
+
+        req = self._build_request_async(
+            method="DELETE",
+            path="/v1/agents/{agent_id}",
+            base_url=base_url,
+            url_variables=url_variables,
+            request=request,
+            request_body_required=False,
+            request_has_path_params=True,
+            request_has_query_params=True,
+            user_agent_header="user-agent",
+            accept_header_value="application/json",
+            http_headers=http_headers,
+            security=self.sdk_configuration.security,
+            timeout_ms=timeout_ms,
+        )
+
+        if retries == UNSET:
+            if self.sdk_configuration.retry_config is not UNSET:
+                retries = self.sdk_configuration.retry_config
+
+        retry_config = None
+        if isinstance(retries, utils.RetryConfig):
+            retry_config = (retries, ["429", "500", "502", "503", "504"])
+
+        http_res = await self.do_request_async(
+            hook_ctx=HookContext(
+                config=self.sdk_configuration,
+                base_url=base_url or "",
+                operation_id="agents_api_v1_agents_delete",
+                oauth2_scopes=[],
+                security_source=get_security_from_env(
+                    self.sdk_configuration.security, models.Security
+                ),
+            ),
+            request=req,
+            error_status_codes=["422", "4XX", "5XX"],
+            retry_config=retry_config,
+        )
+
+        response_data: Any = None
+        if utils.match_response(http_res, "204", "*"):
+            return
+        if utils.match_response(http_res, "422", "application/json"):
+            response_data = unmarshal_json_response(
+                models.HTTPValidationErrorData, http_res
+            )
+            raise models.HTTPValidationError(response_data, http_res)
+        if utils.match_response(http_res, "4XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+        if utils.match_response(http_res, "5XX", "*"):
+            http_res_text = await utils.stream_to_text_async(http_res)
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+
+        raise models.SDKError("Unexpected response received", http_res)
+
     def update_version(
         self,
         *,

mistralai/mistral_jobs.py

@@ -7,7 +7,7 @@ from mistralai._hooks import HookContext
 from mistralai.types import OptionalNullable, UNSET
 from mistralai.utils import get_security_from_env
 from mistralai.utils.unmarshal_json_response import unmarshal_json_response
-from typing import Any, Dict, List, Mapping, Optional
+from typing import Any, Dict, List, Mapping, Optional, Union
 
 
 class MistralJobs(BaseSDK):
@@ -222,8 +222,11 @@ class MistralJobs(BaseSDK):
     def create(
         self,
         *,
-        input_files: List[str],
         endpoint: models.APIEndpoint,
+        input_files: OptionalNullable[List[str]] = UNSET,
+        requests: OptionalNullable[
+            Union[List[models.BatchRequest], List[models.BatchRequestTypedDict]]
+        ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, str]] = UNSET,
@@ -237,12 +240,13 @@ class MistralJobs(BaseSDK):
 
         Create a new batch job, it will be queued for processing.
 
-        :param input_files:
         :param endpoint:
-        :param model:
-        :param agent_id:
-        :param metadata:
-        :param timeout_hours:
+        :param input_files: The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```
+        :param requests:
+        :param model: The model to be used for batch inference.
+        :param agent_id: In case you want to use a specific agent from the **deprecated** agents api for batch inference, you can specify the agent ID here.
+        :param metadata: The metadata of your choice to be associated with the batch inference job.
+        :param timeout_hours: The timeout in hours for the batch inference job.
         :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
@@ -260,6 +264,9 @@ class MistralJobs(BaseSDK):
 
         request = models.BatchJobIn(
             input_files=input_files,
+            requests=utils.get_pydantic_model(
+                requests, OptionalNullable[List[models.BatchRequest]]
+            ),
             endpoint=endpoint,
             model=model,
             agent_id=agent_id,
@@ -323,8 +330,11 @@ class MistralJobs(BaseSDK):
     async def create_async(
         self,
         *,
-        input_files: List[str],
         endpoint: models.APIEndpoint,
+        input_files: OptionalNullable[List[str]] = UNSET,
+        requests: OptionalNullable[
+            Union[List[models.BatchRequest], List[models.BatchRequestTypedDict]]
+        ] = UNSET,
         model: OptionalNullable[str] = UNSET,
         agent_id: OptionalNullable[str] = UNSET,
         metadata: OptionalNullable[Dict[str, str]] = UNSET,
@@ -338,12 +348,13 @@ class MistralJobs(BaseSDK):
 
         Create a new batch job, it will be queued for processing.
 
-        :param input_files:
         :param endpoint:
-        :param model:
-        :param agent_id:
-        :param metadata:
-        :param timeout_hours:
+        :param input_files: The list of input files to be used for batch inference, these files should be `jsonl` files, containing the input data corresponding to the bory request for the batch inference in a \"body\" field. An example of such file is the following: ```json {\"custom_id\": \"0\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French cheese?\"}]}} {\"custom_id\": \"1\", \"body\": {\"max_tokens\": 100, \"messages\": [{\"role\": \"user\", \"content\": \"What is the best French wine?\"}]}} ```
+        :param requests:
+        :param model: The model to be used for batch inference.
+        :param agent_id: In case you want to use a specific agent from the **deprecated** agents api for batch inference, you can specify the agent ID here.
+        :param metadata: The metadata of your choice to be associated with the batch inference job.
+        :param timeout_hours: The timeout in hours for the batch inference job.
         :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
@@ -361,6 +372,9 @@ class MistralJobs(BaseSDK):
 
         request = models.BatchJobIn(
             input_files=input_files,
+            requests=utils.get_pydantic_model(
+                requests, OptionalNullable[List[models.BatchRequest]]
+            ),
             endpoint=endpoint,
             model=model,
             agent_id=agent_id,
@@ -425,6 +439,7 @@ class MistralJobs(BaseSDK):
         self,
         *,
         job_id: str,
+        inline: OptionalNullable[bool] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -434,7 +449,11 @@ class MistralJobs(BaseSDK):
 
         Get a batch job details by its UUID.
 
+        Args:
+        inline: If True, return results inline in the response.
+
         :param job_id:
+        :param inline:
         :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
@@ -452,6 +471,7 @@ class MistralJobs(BaseSDK):
 
         request = models.JobsAPIRoutesBatchGetBatchJobRequest(
             job_id=job_id,
+            inline=inline,
         )
 
         req = self._build_request(
@@ -508,6 +528,7 @@ class MistralJobs(BaseSDK):
         self,
         *,
         job_id: str,
+        inline: OptionalNullable[bool] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -517,7 +538,11 @@ class MistralJobs(BaseSDK):
 
         Get a batch job details by its UUID.
 
+        Args:
+        inline: If True, return results inline in the response.
+
         :param job_id:
+        :param inline:
         :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
@@ -535,6 +560,7 @@ class MistralJobs(BaseSDK):
 
         request = models.JobsAPIRoutesBatchGetBatchJobRequest(
             job_id=job_id,
+            inline=inline,
         )
 
         req = self._build_request_async(