feat: add stream_timeout_ms to SDKConfiguration and Mistral.__init__ [AIR-1092]

Author: Tanmay Kale

src/mistralai/client/agents.py

@@ -462,6 +462,8 @@ def stream(
         """
         base_url = None
         url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.stream_timeout_ms
         if timeout_ms is None:
             timeout_ms = self.sdk_configuration.timeout_ms
 
@@ -656,6 +658,8 @@ async def stream_async(
         """
         base_url = None
         url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.stream_timeout_ms
         if timeout_ms is None:
             timeout_ms = self.sdk_configuration.timeout_ms
 

src/mistralai/client/chat.py

@@ -571,6 +571,8 @@ def stream(
         """
         base_url = None
         url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.stream_timeout_ms
         if timeout_ms is None:
             timeout_ms = self.sdk_configuration.timeout_ms
 
@@ -774,6 +776,8 @@ async def stream_async(
         """
         base_url = None
         url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.stream_timeout_ms
         if timeout_ms is None:
             timeout_ms = self.sdk_configuration.timeout_ms
 

src/mistralai/client/fim.py

@@ -329,6 +329,8 @@ def stream(
         """
         base_url = None
         url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.stream_timeout_ms
         if timeout_ms is None:
             timeout_ms = self.sdk_configuration.timeout_ms
 
@@ -471,6 +473,8 @@ async def stream_async(
         """
         base_url = None
         url_variables = None
+        if timeout_ms is None:
+            timeout_ms = self.sdk_configuration.stream_timeout_ms
         if timeout_ms is None:
             timeout_ms = self.sdk_configuration.timeout_ms
 

src/mistralai/client/sdk.py

@@ -81,6 +81,7 @@ def __init__(
         async_client: Optional[AsyncHttpClient] = None,
         retry_config: OptionalNullable[RetryConfig] = UNSET,
         timeout_ms: Optional[int] = None,
+        stream_timeout_ms: Optional[int] = None,
         debug_logger: Optional[Logger] = None,
     ) -> None:
         r"""Instantiates the SDK configuring it with the provided parameters.
@@ -93,6 +94,7 @@ def __init__(
         :param async_client: The Async HTTP client to use for all asynchronous methods
         :param retry_config: The retry configuration to use for all supported methods
         :param timeout_ms: Optional request timeout applied to each operation in milliseconds
+        :param stream_timeout_ms: Optional idle-read timeout for SSE streaming operations in milliseconds (default 20 000 ms). Overrides timeout_ms for streaming calls only.
         """
         client_supplied = True
         if client is None:
@@ -140,6 +142,7 @@ def __init__(
                 server=server,
                 retry_config=retry_config,
                 timeout_ms=timeout_ms,
+                stream_timeout_ms=stream_timeout_ms,
                 debug_logger=debug_logger,
             ),
             parent_ref=self,

src/mistralai/client/sdkconfiguration.py

@@ -47,6 +47,7 @@ class SDKConfiguration:
     user_agent: str = __user_agent__
     retry_config: OptionalNullable[RetryConfig] = Field(default_factory=lambda: UNSET)
     timeout_ms: Optional[int] = None
+    stream_timeout_ms: Optional[int] = None
 
     def get_server_details(self) -> Tuple[str, Dict[str, str]]:
         if self.server_url is not None and self.server_url: