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

mistralai/agents.py

@@ -69,7 +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:
+        :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
@@ -140,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=[],
@@ -238,7 +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:
+        :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
@@ -309,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=[],
@@ -409,7 +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:
+        :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
@@ -480,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=[],
@@ -586,7 +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:
+        :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
@@ -657,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

@@ -3,6 +3,7 @@
 from .basesdk import BaseSDK
 from .sdkconfiguration import SDKConfiguration
 from mistralai.conversations import Conversations
+from mistralai.libraries import Libraries
 from mistralai.mistral_agents import MistralAgents
 
 
@@ -11,6 +12,8 @@ class Beta(BaseSDK):
     r"""(beta) Conversations API"""
     agents: MistralAgents
     r"""(beta) Agents API"""
+    libraries: Libraries
+    r"""(beta) Libraries API for indexing documents to enhance agent capabilities."""
 
     def __init__(self, sdk_config: SDKConfiguration) -> None:
         BaseSDK.__init__(self, sdk_config)
@@ -20,3 +23,4 @@ class Beta(BaseSDK):
     def _init_sdks(self):
         self.conversations = Conversations(self.sdk_configuration)
         self.agents = MistralAgents(self.sdk_configuration)
+        self.libraries = Libraries(self.sdk_configuration)

mistralai/chat.py

@@ -148,7 +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:
+        :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
@@ -221,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=[],
@@ -316,7 +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:
+        :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
@@ -389,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=[],
@@ -494,7 +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:
+        :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
@@ -569,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=[],
@@ -680,7 +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:
+        :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
@@ -755,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=[],

mistralai/conversations.py

@@ -313,6 +313,7 @@ class Conversations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_start",
                 oauth2_scopes=[],
@@ -451,6 +452,7 @@ class Conversations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_start",
                 oauth2_scopes=[],
@@ -553,6 +555,7 @@ class Conversations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_list",
                 oauth2_scopes=[],
@@ -655,6 +658,7 @@ class Conversations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_list",
                 oauth2_scopes=[],
@@ -708,7 +712,7 @@ class Conversations(BaseSDK):
 
         Given a conversation_id retrieve a conversation entity with its attributes.
 
-        :param conversation_id:
+        :param conversation_id: ID of the conversation from which we are fetching 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
@@ -754,6 +758,7 @@ class Conversations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_get",
                 oauth2_scopes=[],
@@ -810,7 +815,7 @@ class Conversations(BaseSDK):
 
         Given a conversation_id retrieve a conversation entity with its attributes.
 
-        :param conversation_id:
+        :param conversation_id: ID of the conversation from which we are fetching 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
@@ -856,6 +861,7 @@ class Conversations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_get",
                 oauth2_scopes=[],
@@ -988,6 +994,7 @@ class Conversations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_append",
                 oauth2_scopes=[],
@@ -1117,6 +1124,7 @@ class Conversations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_append",
                 oauth2_scopes=[],
@@ -1170,7 +1178,7 @@ class Conversations(BaseSDK):
 
         Given a conversation_id retrieve all the entries belonging to that conversation. The entries are sorted in the order they were appended, those can be messages, connectors or function_call.
 
-        :param conversation_id:
+        :param conversation_id: ID of the conversation from which we are fetching entries.
         :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
@@ -1216,6 +1224,7 @@ class Conversations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_history",
                 oauth2_scopes=[],
@@ -1269,7 +1278,7 @@ class Conversations(BaseSDK):
 
         Given a conversation_id retrieve all the entries belonging to that conversation. The entries are sorted in the order they were appended, those can be messages, connectors or function_call.
 
-        :param conversation_id:
+        :param conversation_id: ID of the conversation from which we are fetching entries.
         :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
@@ -1315,6 +1324,7 @@ class Conversations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_history",
                 oauth2_scopes=[],
@@ -1368,7 +1378,7 @@ class Conversations(BaseSDK):
 
         Given a conversation_id retrieve all the messages belonging to that conversation. This is similar to retrieving all entries except we filter the messages only.
 
-        :param conversation_id:
+        :param conversation_id: ID of the conversation from which we are fetching messages.
         :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
@@ -1414,6 +1424,7 @@ class Conversations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_messages",
                 oauth2_scopes=[],
@@ -1467,7 +1478,7 @@ class Conversations(BaseSDK):
 
         Given a conversation_id retrieve all the messages belonging to that conversation. This is similar to retrieving all entries except we filter the messages only.
 
-        :param conversation_id:
+        :param conversation_id: ID of the conversation from which we are fetching messages.
         :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
@@ -1513,6 +1524,7 @@ class Conversations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_messages",
                 oauth2_scopes=[],
@@ -1576,7 +1588,7 @@ class Conversations(BaseSDK):
 
         Given a conversation_id and an id, recreate a conversation from this point and run completion. A new conversation is returned with the new entries returned.
 
-        :param conversation_id:
+        :param conversation_id: ID of the original conversation which is being restarted.
         :param inputs:
         :param from_entry_id:
         :param stream:
@@ -1605,10 +1617,10 @@ class Conversations(BaseSDK):
                 stream=stream,
                 store=store,
                 handoff_execution=handoff_execution,
-                from_entry_id=from_entry_id,
                 completion_args=utils.get_pydantic_model(
                     completion_args, Optional[models.CompletionArgs]
                 ),
+                from_entry_id=from_entry_id,
             ),
         )
 
@@ -1645,6 +1657,7 @@ class Conversations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_restart",
                 oauth2_scopes=[],
@@ -1708,7 +1721,7 @@ class Conversations(BaseSDK):
 
         Given a conversation_id and an id, recreate a conversation from this point and run completion. A new conversation is returned with the new entries returned.
 
-        :param conversation_id:
+        :param conversation_id: ID of the original conversation which is being restarted.
         :param inputs:
         :param from_entry_id:
         :param stream:
@@ -1737,10 +1750,10 @@ class Conversations(BaseSDK):
                 stream=stream,
                 store=store,
                 handoff_execution=handoff_execution,
-                from_entry_id=from_entry_id,
                 completion_args=utils.get_pydantic_model(
                     completion_args, Optional[models.CompletionArgs]
                 ),
+                from_entry_id=from_entry_id,
             ),
         )
 
@@ -1777,6 +1790,7 @@ class Conversations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_restart",
                 oauth2_scopes=[],
@@ -1922,6 +1936,7 @@ class Conversations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_start_stream",
                 oauth2_scopes=[],
@@ -2072,6 +2087,7 @@ class Conversations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_start_stream",
                 oauth2_scopes=[],
@@ -2206,6 +2222,7 @@ class Conversations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_append_stream",
                 oauth2_scopes=[],
@@ -2340,6 +2357,7 @@ class Conversations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_append_stream",
                 oauth2_scopes=[],
@@ -2408,7 +2426,7 @@ class Conversations(BaseSDK):
 
         Given a conversation_id and an id, recreate a conversation from this point and run completion. A new conversation is returned with the new entries returned.
 
-        :param conversation_id:
+        :param conversation_id: ID of the original conversation which is being restarted.
         :param inputs:
         :param from_entry_id:
         :param stream:
@@ -2437,10 +2455,10 @@ class Conversations(BaseSDK):
                 stream=stream,
                 store=store,
                 handoff_execution=handoff_execution,
-                from_entry_id=from_entry_id,
                 completion_args=utils.get_pydantic_model(
                     completion_args, Optional[models.CompletionArgs]
                 ),
+                from_entry_id=from_entry_id,
             ),
         )
 
@@ -2477,6 +2495,7 @@ class Conversations(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_restart_stream",
                 oauth2_scopes=[],
@@ -2545,7 +2564,7 @@ class Conversations(BaseSDK):
 
         Given a conversation_id and an id, recreate a conversation from this point and run completion. A new conversation is returned with the new entries returned.
 
-        :param conversation_id:
+        :param conversation_id: ID of the original conversation which is being restarted.
         :param inputs:
         :param from_entry_id:
         :param stream:
@@ -2574,10 +2593,10 @@ class Conversations(BaseSDK):
                 stream=stream,
                 store=store,
                 handoff_execution=handoff_execution,
-                from_entry_id=from_entry_id,
                 completion_args=utils.get_pydantic_model(
                     completion_args, Optional[models.CompletionArgs]
                 ),
+                from_entry_id=from_entry_id,
             ),
         )
 
@@ -2614,6 +2633,7 @@ class Conversations(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                config=self.sdk_configuration,
                 base_url=base_url or "",
                 operation_id="agents_api_v1_conversations_restart_stream",
                 oauth2_scopes=[],