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

mistralai/_hooks/types.py

@@ -3,10 +3,12 @@
 from abc import ABC, abstractmethod
 import httpx
 from mistralai.httpclient import HttpClient
+from mistralai.sdkconfiguration import SDKConfiguration
 from typing import Any, Callable, List, Optional, Tuple, Union
 
 
 class HookContext:
+    config: SDKConfiguration
     base_url: str
     operation_id: str
     oauth2_scopes: Optional[List[str]] = None
@@ -14,11 +16,13 @@ class HookContext:
 
     def __init__(
         self,
+        config: SDKConfiguration,
         base_url: str,
         operation_id: str,
         oauth2_scopes: Optional[List[str]],
         security_source: Optional[Union[Any, Callable[[], Any]]],
     ):
+        self.config = config
         self.base_url = base_url
         self.operation_id = operation_id
         self.oauth2_scopes = oauth2_scopes
@@ -28,6 +32,7 @@ class HookContext:
 class BeforeRequestContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,
@@ -38,6 +43,7 @@ class BeforeRequestContext(HookContext):
 class AfterSuccessContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,
@@ -48,6 +54,7 @@ class AfterSuccessContext(HookContext):
 class AfterErrorContext(HookContext):
     def __init__(self, hook_ctx: HookContext):
         super().__init__(
+            hook_ctx.config,
             hook_ctx.base_url,
             hook_ctx.operation_id,
             hook_ctx.oauth2_scopes,

mistralai/_version.py

@@ -3,10 +3,10 @@
 import importlib.metadata
 
 __title__: str = "mistralai"
-__version__: str = "1.8.1"
+__version__: str = "1.9.1"
 __openapi_doc_version__: str = "1.0.0"
-__gen_version__: str = "2.548.6"
-__user_agent__: str = "speakeasy-sdk/python 1.8.1 2.548.6 1.0.0 mistralai"
+__gen_version__: str = "2.634.2"
+__user_agent__: str = "speakeasy-sdk/python 1.9.1 2.634.2 1.0.0 mistralai"
 
 try:
     if __package__ is not None:

mistralai/agents.py

@@ -47,6 +47,7 @@ class Agents(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -68,6 +69,7 @@ class Agents(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :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
@@ -105,6 +107,7 @@ class Agents(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             agent_id=agent_id,
         )
 
@@ -137,6 +140,7 @@ class Agents(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_completion_v1_agents_completions_post",
                 oauth2_scopes=[],
@@ -213,6 +217,7 @@ class Agents(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -234,6 +239,7 @@ class Agents(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :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
@@ -271,6 +277,7 @@ class Agents(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             agent_id=agent_id,
         )
 
@@ -303,6 +310,7 @@ class Agents(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_completion_v1_agents_completions_post",
                 oauth2_scopes=[],
@@ -379,6 +387,7 @@ class Agents(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -402,6 +411,7 @@ class Agents(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :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
@@ -439,6 +449,7 @@ class Agents(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             agent_id=agent_id,
         )
 
@@ -471,6 +482,7 @@ class Agents(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="stream_agents",
                 oauth2_scopes=[],
@@ -553,6 +565,7 @@ class Agents(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
         timeout_ms: Optional[int] = None,
@@ -576,6 +589,7 @@ class Agents(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :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
@@ -613,6 +627,7 @@ class Agents(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             agent_id=agent_id,
         )
 
@@ -645,6 +660,7 @@ class Agents(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="stream_agents",
                 oauth2_scopes=[],

mistralai/basesdk.py

@@ -218,12 +218,12 @@ class BaseSDK:
         client = self.sdk_configuration.client
         logger = self.sdk_configuration.debug_logger
 
+        hooks = self.sdk_configuration.__dict__["_hooks"]
+
         def do():
             http_res = None
             try:
-                req = self.sdk_configuration.get_hooks().before_request(
-                    BeforeRequestContext(hook_ctx), request
-                )
+                req = hooks.before_request(BeforeRequestContext(hook_ctx), request)
                 logger.debug(
                     "Request:\nMethod: %s\nURL: %s\nHeaders: %s\nBody: %s",
                     req.method,
@@ -237,9 +237,7 @@ class BaseSDK:
 
                 http_res = client.send(req, stream=stream)
             except Exception as e:
-                _, e = self.sdk_configuration.get_hooks().after_error(
-                    AfterErrorContext(hook_ctx), None, e
-                )
+                _, e = hooks.after_error(AfterErrorContext(hook_ctx), None, e)
                 if e is not None:
                     logger.debug("Request Exception", exc_info=True)
                     raise e
@@ -257,7 +255,7 @@ class BaseSDK:
             )
 
             if utils.match_status_codes(error_status_codes, http_res.status_code):
-                result, err = self.sdk_configuration.get_hooks().after_error(
+                result, err = hooks.after_error(
                     AfterErrorContext(hook_ctx), http_res, None
                 )
                 if err is not None:
@@ -277,9 +275,7 @@ class BaseSDK:
             http_res = do()
 
         if not utils.match_status_codes(error_status_codes, http_res.status_code):
-            http_res = self.sdk_configuration.get_hooks().after_success(
-                AfterSuccessContext(hook_ctx), http_res
-            )
+            http_res = hooks.after_success(AfterSuccessContext(hook_ctx), http_res)
 
         return http_res
 
@@ -294,12 +290,12 @@ class BaseSDK:
         client = self.sdk_configuration.async_client
         logger = self.sdk_configuration.debug_logger
 
+        hooks = self.sdk_configuration.__dict__["_hooks"]
+
         async def do():
             http_res = None
             try:
-                req = self.sdk_configuration.get_hooks().before_request(
-                    BeforeRequestContext(hook_ctx), request
-                )
+                req = hooks.before_request(BeforeRequestContext(hook_ctx), request)
                 logger.debug(
                     "Request:\nMethod: %s\nURL: %s\nHeaders: %s\nBody: %s",
                     req.method,
@@ -313,9 +309,7 @@ class BaseSDK:
 
                 http_res = await client.send(req, stream=stream)
             except Exception as e:
-                _, e = self.sdk_configuration.get_hooks().after_error(
-                    AfterErrorContext(hook_ctx), None, e
-                )
+                _, e = hooks.after_error(AfterErrorContext(hook_ctx), None, e)
                 if e is not None:
                     logger.debug("Request Exception", exc_info=True)
                     raise e
@@ -333,7 +327,7 @@ class BaseSDK:
             )
 
             if utils.match_status_codes(error_status_codes, http_res.status_code):
-                result, err = self.sdk_configuration.get_hooks().after_error(
+                result, err = hooks.after_error(
                     AfterErrorContext(hook_ctx), http_res, None
                 )
                 if err is not None:
@@ -355,8 +349,6 @@ class BaseSDK:
             http_res = await do()
 
         if not utils.match_status_codes(error_status_codes, http_res.status_code):
-            http_res = self.sdk_configuration.get_hooks().after_success(
-                AfterSuccessContext(hook_ctx), http_res
-            )
+            http_res = hooks.after_success(AfterSuccessContext(hook_ctx), http_res)
 
         return http_res

mistralai/beta.py

@@ -8,7 +8,7 @@ from mistralai.mistral_agents import MistralAgents
 
 class Beta(BaseSDK):
     conversations: Conversations
-    r"""(beta) Converstations API"""
+    r"""(beta) Conversations API"""
     agents: MistralAgents
     r"""(beta) Agents API"""
 

mistralai/chat.py

@@ -123,6 +123,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -147,6 +148,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -186,6 +188,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -218,6 +221,7 @@ class Chat(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="chat_completion_v1_chat_completions_post",
                 oauth2_scopes=[],
@@ -288,6 +292,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -312,6 +317,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -351,6 +357,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -383,6 +390,7 @@ class Chat(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="chat_completion_v1_chat_completions_post",
                 oauth2_scopes=[],
@@ -461,6 +469,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -487,6 +496,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -528,6 +538,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -560,6 +571,7 @@ class Chat(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="stream_chat",
                 oauth2_scopes=[],
@@ -644,6 +656,7 @@ class Chat(BaseSDK):
             Union[models.Prediction, models.PredictionTypedDict]
         ] = None,
         parallel_tool_calls: Optional[bool] = None,
+        prompt_mode: OptionalNullable[models.MistralPromptMode] = UNSET,
         safe_prompt: Optional[bool] = None,
         retries: OptionalNullable[utils.RetryConfig] = UNSET,
         server_url: Optional[str] = None,
@@ -670,6 +683,7 @@ class Chat(BaseSDK):
         :param n: Number of completions to return for each request, input tokens are only billed once.
         :param prediction:
         :param parallel_tool_calls:
+        :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used.
         :param safe_prompt: Whether to inject a safety prompt before all conversations.
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -711,6 +725,7 @@ class Chat(BaseSDK):
                 prediction, Optional[models.Prediction]
             ),
             parallel_tool_calls=parallel_tool_calls,
+            prompt_mode=prompt_mode,
             safe_prompt=safe_prompt,
         )
 
@@ -743,6 +758,7 @@ class Chat(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="stream_chat",
                 oauth2_scopes=[],

mistralai/classifiers.py

@@ -77,6 +77,7 @@ class Classifiers(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="moderations_v1_moderations_post",
                 oauth2_scopes=[],
@@ -183,6 +184,7 @@ class Classifiers(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="moderations_v1_moderations_post",
                 oauth2_scopes=[],
@@ -289,6 +291,7 @@ class Classifiers(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="chat_moderations_v1_chat_moderations_post",
                 oauth2_scopes=[],
@@ -395,6 +398,7 @@ class Classifiers(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="chat_moderations_v1_chat_moderations_post",
                 oauth2_scopes=[],
@@ -501,6 +505,7 @@ class Classifiers(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="classifications_v1_classifications_post",
                 oauth2_scopes=[],
@@ -607,6 +612,7 @@ class Classifiers(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="classifications_v1_classifications_post",
                 oauth2_scopes=[],
@@ -710,6 +716,7 @@ class Classifiers(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="chat_classifications_v1_chat_classifications_post",
                 oauth2_scopes=[],
@@ -813,6 +820,7 @@ class Classifiers(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="chat_classifications_v1_chat_classifications_post",
                 oauth2_scopes=[],