mistralai 1.9.10__py3-none-any.whl → 1.10.0__py3-none-any.whl

mistralai/mistral_jobs.py

@@ -6,6 +6,7 @@ from mistralai import models, utils
 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
 
 
@@ -104,26 +105,15 @@ class MistralJobs(BaseSDK):
         )
 
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.BatchJobsOut)
+            return unmarshal_json_response(models.BatchJobsOut, 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.status_code, http_res_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.status_code, http_res_text, http_res
-            )
-
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = utils.stream_to_text(http_res)
-        raise models.SDKError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+
+        raise models.SDKError("Unexpected response received", http_res)
 
     async def list_async(
         self,
@@ -219,26 +209,15 @@ class MistralJobs(BaseSDK):
         )
 
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.BatchJobsOut)
+            return unmarshal_json_response(models.BatchJobsOut, 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.status_code, http_res_text, 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.status_code, http_res_text, http_res
-            )
-
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = await utils.stream_to_text_async(http_res)
-        raise models.SDKError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+
+        raise models.SDKError("Unexpected response received", http_res)
 
     def create(
         self,
@@ -258,12 +237,12 @@ class MistralJobs(BaseSDK):
 
         Create a new batch job, it will be queued for processing.
 
-        :param input_files:
+        :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 endpoint:
-        :param model:
-        :param agent_id:
-        :param metadata:
-        :param timeout_hours:
+        :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
@@ -331,26 +310,15 @@ class MistralJobs(BaseSDK):
         )
 
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.BatchJobOut)
+            return unmarshal_json_response(models.BatchJobOut, 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.status_code, http_res_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.status_code, http_res_text, http_res
-            )
-
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = utils.stream_to_text(http_res)
-        raise models.SDKError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+
+        raise models.SDKError("Unexpected response received", http_res)
 
     async def create_async(
         self,
@@ -370,12 +338,12 @@ class MistralJobs(BaseSDK):
 
         Create a new batch job, it will be queued for processing.
 
-        :param input_files:
+        :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 endpoint:
-        :param model:
-        :param agent_id:
-        :param metadata:
-        :param timeout_hours:
+        :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
@@ -443,26 +411,15 @@ class MistralJobs(BaseSDK):
         )
 
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.BatchJobOut)
+            return unmarshal_json_response(models.BatchJobOut, 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.status_code, http_res_text, 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.status_code, http_res_text, http_res
-            )
-
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = await utils.stream_to_text_async(http_res)
-        raise models.SDKError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+
+        raise models.SDKError("Unexpected response received", http_res)
 
     def get(
         self,
@@ -537,26 +494,15 @@ class MistralJobs(BaseSDK):
         )
 
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.BatchJobOut)
+            return unmarshal_json_response(models.BatchJobOut, 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.status_code, http_res_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.status_code, http_res_text, http_res
-            )
-
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = utils.stream_to_text(http_res)
-        raise models.SDKError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+
+        raise models.SDKError("Unexpected response received", http_res)
 
     async def get_async(
         self,
@@ -631,26 +577,15 @@ class MistralJobs(BaseSDK):
         )
 
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.BatchJobOut)
+            return unmarshal_json_response(models.BatchJobOut, 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.status_code, http_res_text, 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.status_code, http_res_text, http_res
-            )
-
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = await utils.stream_to_text_async(http_res)
-        raise models.SDKError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+
+        raise models.SDKError("Unexpected response received", http_res)
 
     def cancel(
         self,
@@ -725,26 +660,15 @@ class MistralJobs(BaseSDK):
         )
 
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.BatchJobOut)
+            return unmarshal_json_response(models.BatchJobOut, 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.status_code, http_res_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.status_code, http_res_text, http_res
-            )
-
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = utils.stream_to_text(http_res)
-        raise models.SDKError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+
+        raise models.SDKError("Unexpected response received", http_res)
 
     async def cancel_async(
         self,
@@ -819,23 +743,12 @@ class MistralJobs(BaseSDK):
         )
 
         if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.BatchJobOut)
+            return unmarshal_json_response(models.BatchJobOut, 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.status_code, http_res_text, 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.status_code, http_res_text, http_res
-            )
-
-        content_type = http_res.headers.get("Content-Type")
-        http_res_text = await utils.stream_to_text_async(http_res)
-        raise models.SDKError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
+            raise models.SDKError("API error occurred", http_res, http_res_text)
+
+        raise models.SDKError("Unexpected response received", http_res)

mistralai/models/__init__.py

@@ -1,7 +1,10 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
+from .mistralerror import MistralError
 from typing import TYPE_CHECKING
 from importlib import import_module
+import builtins
+import sys
 
 if TYPE_CHECKING:
     from .agent import (
@@ -38,6 +41,10 @@ if TYPE_CHECKING:
         AgentHandoffStartedEventType,
         AgentHandoffStartedEventTypedDict,
     )
+    from .agents_api_v1_agents_deleteop import (
+        AgentsAPIV1AgentsDeleteRequest,
+        AgentsAPIV1AgentsDeleteRequestTypedDict,
+    )
     from .agents_api_v1_agents_getop import (
         AgentsAPIV1AgentsGetRequest,
         AgentsAPIV1AgentsGetRequestTypedDict,
@@ -62,6 +69,10 @@ if TYPE_CHECKING:
         AgentsAPIV1ConversationsAppendRequest,
         AgentsAPIV1ConversationsAppendRequestTypedDict,
     )
+    from .agents_api_v1_conversations_deleteop import (
+        AgentsAPIV1ConversationsDeleteRequest,
+        AgentsAPIV1ConversationsDeleteRequestTypedDict,
+    )
     from .agents_api_v1_conversations_getop import (
         AgentsAPIV1ConversationsGetRequest,
         AgentsAPIV1ConversationsGetRequestTypedDict,
@@ -371,7 +382,12 @@ if TYPE_CHECKING:
     )
     from .documentout import DocumentOut, DocumentOutTypedDict
     from .documenttextcontent import DocumentTextContent, DocumentTextContentTypedDict
-    from .documentupdatein import DocumentUpdateIn, DocumentUpdateInTypedDict
+    from .documentupdatein import (
+        Attributes,
+        AttributesTypedDict,
+        DocumentUpdateIn,
+        DocumentUpdateInTypedDict,
+    )
     from .documenturlchunk import (
         DocumentURLChunk,
         DocumentURLChunkType,
@@ -713,11 +729,19 @@ if TYPE_CHECKING:
     from .modellist import Data, DataTypedDict, ModelList, ModelListTypedDict
     from .moderationobject import ModerationObject, ModerationObjectTypedDict
     from .moderationresponse import ModerationResponse, ModerationResponseTypedDict
+    from .no_response_error import NoResponseError
     from .ocrimageobject import OCRImageObject, OCRImageObjectTypedDict
     from .ocrpagedimensions import OCRPageDimensions, OCRPageDimensionsTypedDict
     from .ocrpageobject import OCRPageObject, OCRPageObjectTypedDict
-    from .ocrrequest import Document, DocumentTypedDict, OCRRequest, OCRRequestTypedDict
+    from .ocrrequest import (
+        Document,
+        DocumentTypedDict,
+        OCRRequest,
+        OCRRequestTypedDict,
+        TableFormat,
+    )
     from .ocrresponse import OCRResponse, OCRResponseTypedDict
+    from .ocrtableobject import Format, OCRTableObject, OCRTableObjectTypedDict
     from .ocrusageinfo import OCRUsageInfo, OCRUsageInfoTypedDict
     from .outputcontentchunks import OutputContentChunks, OutputContentChunksTypedDict
     from .paginationinfo import PaginationInfo, PaginationInfoTypedDict
@@ -728,6 +752,7 @@ if TYPE_CHECKING:
         ReferenceChunkType,
         ReferenceChunkTypedDict,
     )
+    from .requestsource import RequestSource
     from .responsedoneevent import (
         ResponseDoneEvent,
         ResponseDoneEventType,
@@ -745,6 +770,7 @@ if TYPE_CHECKING:
         ResponseStartedEventType,
         ResponseStartedEventTypedDict,
     )
+    from .responsevalidationerror import ResponseValidationError
     from .retrieve_model_v1_models_model_id_getop import (
         RetrieveModelV1ModelsModelIDGetRequest,
         RetrieveModelV1ModelsModelIDGetRequestTypedDict,
@@ -787,15 +813,21 @@ if TYPE_CHECKING:
     from .toolchoiceenum import ToolChoiceEnum
     from .toolexecutiondeltaevent import (
         ToolExecutionDeltaEvent,
+        ToolExecutionDeltaEventName,
+        ToolExecutionDeltaEventNameTypedDict,
         ToolExecutionDeltaEventType,
         ToolExecutionDeltaEventTypedDict,
     )
     from .toolexecutiondoneevent import (
         ToolExecutionDoneEvent,
+        ToolExecutionDoneEventName,
+        ToolExecutionDoneEventNameTypedDict,
         ToolExecutionDoneEventType,
         ToolExecutionDoneEventTypedDict,
     )
     from .toolexecutionentry import (
+        Name,
+        NameTypedDict,
         ToolExecutionEntry,
         ToolExecutionEntryObject,
         ToolExecutionEntryType,
@@ -803,6 +835,8 @@ if TYPE_CHECKING:
     )
     from .toolexecutionstartedevent import (
         ToolExecutionStartedEvent,
+        ToolExecutionStartedEventName,
+        ToolExecutionStartedEventNameTypedDict,
         ToolExecutionStartedEventType,
         ToolExecutionStartedEventTypedDict,
     )
@@ -923,6 +957,8 @@ __all__ = [
     "AgentUpdateRequestTools",
     "AgentUpdateRequestToolsTypedDict",
     "AgentUpdateRequestTypedDict",
+    "AgentsAPIV1AgentsDeleteRequest",
+    "AgentsAPIV1AgentsDeleteRequestTypedDict",
     "AgentsAPIV1AgentsGetRequest",
     "AgentsAPIV1AgentsGetRequestTypedDict",
     "AgentsAPIV1AgentsListRequest",
@@ -935,6 +971,8 @@ __all__ = [
     "AgentsAPIV1ConversationsAppendRequestTypedDict",
     "AgentsAPIV1ConversationsAppendStreamRequest",
     "AgentsAPIV1ConversationsAppendStreamRequestTypedDict",
+    "AgentsAPIV1ConversationsDeleteRequest",
+    "AgentsAPIV1ConversationsDeleteRequestTypedDict",
     "AgentsAPIV1ConversationsGetRequest",
     "AgentsAPIV1ConversationsGetRequestTypedDict",
     "AgentsAPIV1ConversationsGetResponseV1ConversationsGet",
@@ -975,6 +1013,8 @@ __all__ = [
     "AssistantMessageContentTypedDict",
     "AssistantMessageRole",
     "AssistantMessageTypedDict",
+    "Attributes",
+    "AttributesTypedDict",
     "AudioChunk",
     "AudioChunkType",
     "AudioChunkTypedDict",
@@ -1206,6 +1246,7 @@ __all__ = [
     "FilesAPIRoutesUploadFileMultiPartBodyParamsTypedDict",
     "FineTuneableModelType",
     "FinishReason",
+    "Format",
     "Function",
     "FunctionCall",
     "FunctionCallEntry",
@@ -1391,6 +1432,7 @@ __all__ = [
     "MessagesTypedDict",
     "MetricOut",
     "MetricOutTypedDict",
+    "MistralError",
     "MistralPromptMode",
     "ModelCapabilities",
     "ModelCapabilitiesTypedDict",
@@ -1406,6 +1448,9 @@ __all__ = [
     "ModerationObjectTypedDict",
     "ModerationResponse",
     "ModerationResponseTypedDict",
+    "Name",
+    "NameTypedDict",
+    "NoResponseError",
     "OCRImageObject",
     "OCRImageObjectTypedDict",
     "OCRPageDimensions",
@@ -1416,6 +1461,8 @@ __all__ = [
     "OCRRequestTypedDict",
     "OCRResponse",
     "OCRResponseTypedDict",
+    "OCRTableObject",
+    "OCRTableObjectTypedDict",
     "OCRUsageInfo",
     "OCRUsageInfoTypedDict",
     "Object",
@@ -1437,6 +1484,7 @@ __all__ = [
     "ReferenceChunkTypedDict",
     "Repositories",
     "RepositoriesTypedDict",
+    "RequestSource",
     "Response1",
     "Response1TypedDict",
     "ResponseBody",
@@ -1453,6 +1501,7 @@ __all__ = [
     "ResponseStartedEvent",
     "ResponseStartedEventType",
     "ResponseStartedEventTypedDict",
+    "ResponseValidationError",
     "RetrieveFileOut",
     "RetrieveFileOutTypedDict",
     "RetrieveModelV1ModelsModelIDGetRequest",
@@ -1482,6 +1531,7 @@ __all__ = [
     "SystemMessageContentChunksTypedDict",
     "SystemMessageContentTypedDict",
     "SystemMessageTypedDict",
+    "TableFormat",
     "TextChunk",
     "TextChunkType",
     "TextChunkTypedDict",
@@ -1498,9 +1548,13 @@ __all__ = [
     "ToolChoiceEnum",
     "ToolChoiceTypedDict",
     "ToolExecutionDeltaEvent",
+    "ToolExecutionDeltaEventName",
+    "ToolExecutionDeltaEventNameTypedDict",
     "ToolExecutionDeltaEventType",
     "ToolExecutionDeltaEventTypedDict",
     "ToolExecutionDoneEvent",
+    "ToolExecutionDoneEventName",
+    "ToolExecutionDoneEventNameTypedDict",
     "ToolExecutionDoneEventType",
     "ToolExecutionDoneEventTypedDict",
     "ToolExecutionEntry",
@@ -1508,6 +1562,8 @@ __all__ = [
     "ToolExecutionEntryType",
     "ToolExecutionEntryTypedDict",
     "ToolExecutionStartedEvent",
+    "ToolExecutionStartedEventName",
+    "ToolExecutionStartedEventNameTypedDict",
     "ToolExecutionStartedEventType",
     "ToolExecutionStartedEventTypedDict",
     "ToolFileChunk",
@@ -1604,6 +1660,8 @@ _dynamic_imports: dict[str, str] = {
     "AgentHandoffStartedEvent": ".agenthandoffstartedevent",
     "AgentHandoffStartedEventType": ".agenthandoffstartedevent",
     "AgentHandoffStartedEventTypedDict": ".agenthandoffstartedevent",
+    "AgentsAPIV1AgentsDeleteRequest": ".agents_api_v1_agents_deleteop",
+    "AgentsAPIV1AgentsDeleteRequestTypedDict": ".agents_api_v1_agents_deleteop",
     "AgentsAPIV1AgentsGetRequest": ".agents_api_v1_agents_getop",
     "AgentsAPIV1AgentsGetRequestTypedDict": ".agents_api_v1_agents_getop",
     "AgentsAPIV1AgentsListRequest": ".agents_api_v1_agents_listop",
@@ -1616,6 +1674,8 @@ _dynamic_imports: dict[str, str] = {
     "AgentsAPIV1ConversationsAppendStreamRequestTypedDict": ".agents_api_v1_conversations_append_streamop",
     "AgentsAPIV1ConversationsAppendRequest": ".agents_api_v1_conversations_appendop",
     "AgentsAPIV1ConversationsAppendRequestTypedDict": ".agents_api_v1_conversations_appendop",
+    "AgentsAPIV1ConversationsDeleteRequest": ".agents_api_v1_conversations_deleteop",
+    "AgentsAPIV1ConversationsDeleteRequestTypedDict": ".agents_api_v1_conversations_deleteop",
     "AgentsAPIV1ConversationsGetRequest": ".agents_api_v1_conversations_getop",
     "AgentsAPIV1ConversationsGetRequestTypedDict": ".agents_api_v1_conversations_getop",
     "AgentsAPIV1ConversationsGetResponseV1ConversationsGet": ".agents_api_v1_conversations_getop",
@@ -1852,6 +1912,8 @@ _dynamic_imports: dict[str, str] = {
     "DocumentOutTypedDict": ".documentout",
     "DocumentTextContent": ".documenttextcontent",
     "DocumentTextContentTypedDict": ".documenttextcontent",
+    "Attributes": ".documentupdatein",
+    "AttributesTypedDict": ".documentupdatein",
     "DocumentUpdateIn": ".documentupdatein",
     "DocumentUpdateInTypedDict": ".documentupdatein",
     "DocumentURLChunk": ".documenturlchunk",
@@ -2107,6 +2169,7 @@ _dynamic_imports: dict[str, str] = {
     "ModerationObjectTypedDict": ".moderationobject",
     "ModerationResponse": ".moderationresponse",
     "ModerationResponseTypedDict": ".moderationresponse",
+    "NoResponseError": ".no_response_error",
     "OCRImageObject": ".ocrimageobject",
     "OCRImageObjectTypedDict": ".ocrimageobject",
     "OCRPageDimensions": ".ocrpagedimensions",
@@ -2117,8 +2180,12 @@ _dynamic_imports: dict[str, str] = {
     "DocumentTypedDict": ".ocrrequest",
     "OCRRequest": ".ocrrequest",
     "OCRRequestTypedDict": ".ocrrequest",
+    "TableFormat": ".ocrrequest",
     "OCRResponse": ".ocrresponse",
     "OCRResponseTypedDict": ".ocrresponse",
+    "Format": ".ocrtableobject",
+    "OCRTableObject": ".ocrtableobject",
+    "OCRTableObjectTypedDict": ".ocrtableobject",
     "OCRUsageInfo": ".ocrusageinfo",
     "OCRUsageInfoTypedDict": ".ocrusageinfo",
     "OutputContentChunks": ".outputcontentchunks",
@@ -2132,6 +2199,7 @@ _dynamic_imports: dict[str, str] = {
     "ReferenceChunk": ".referencechunk",
     "ReferenceChunkType": ".referencechunk",
     "ReferenceChunkTypedDict": ".referencechunk",
+    "RequestSource": ".requestsource",
     "ResponseDoneEvent": ".responsedoneevent",
     "ResponseDoneEventType": ".responsedoneevent",
     "ResponseDoneEventTypedDict": ".responsedoneevent",
@@ -2144,6 +2212,7 @@ _dynamic_imports: dict[str, str] = {
     "ResponseStartedEvent": ".responsestartedevent",
     "ResponseStartedEventType": ".responsestartedevent",
     "ResponseStartedEventTypedDict": ".responsestartedevent",
+    "ResponseValidationError": ".responsevalidationerror",
     "RetrieveModelV1ModelsModelIDGetRequest": ".retrieve_model_v1_models_model_id_getop",
     "RetrieveModelV1ModelsModelIDGetRequestTypedDict": ".retrieve_model_v1_models_model_id_getop",
     "RetrieveModelV1ModelsModelIDGetResponseRetrieveModelV1ModelsModelIDGet": ".retrieve_model_v1_models_model_id_getop",
@@ -2187,16 +2256,24 @@ _dynamic_imports: dict[str, str] = {
     "ToolChoiceTypedDict": ".toolchoice",
     "ToolChoiceEnum": ".toolchoiceenum",
     "ToolExecutionDeltaEvent": ".toolexecutiondeltaevent",
+    "ToolExecutionDeltaEventName": ".toolexecutiondeltaevent",
+    "ToolExecutionDeltaEventNameTypedDict": ".toolexecutiondeltaevent",
     "ToolExecutionDeltaEventType": ".toolexecutiondeltaevent",
     "ToolExecutionDeltaEventTypedDict": ".toolexecutiondeltaevent",
     "ToolExecutionDoneEvent": ".toolexecutiondoneevent",
+    "ToolExecutionDoneEventName": ".toolexecutiondoneevent",
+    "ToolExecutionDoneEventNameTypedDict": ".toolexecutiondoneevent",
     "ToolExecutionDoneEventType": ".toolexecutiondoneevent",
     "ToolExecutionDoneEventTypedDict": ".toolexecutiondoneevent",
+    "Name": ".toolexecutionentry",
+    "NameTypedDict": ".toolexecutionentry",
     "ToolExecutionEntry": ".toolexecutionentry",
     "ToolExecutionEntryObject": ".toolexecutionentry",
     "ToolExecutionEntryType": ".toolexecutionentry",
     "ToolExecutionEntryTypedDict": ".toolexecutionentry",
     "ToolExecutionStartedEvent": ".toolexecutionstartedevent",
+    "ToolExecutionStartedEventName": ".toolexecutionstartedevent",
+    "ToolExecutionStartedEventNameTypedDict": ".toolexecutionstartedevent",
     "ToolExecutionStartedEventType": ".toolexecutionstartedevent",
     "ToolExecutionStartedEventTypedDict": ".toolexecutionstartedevent",
     "ToolFileChunk": ".toolfilechunk",
@@ -2268,6 +2345,18 @@ _dynamic_imports: dict[str, str] = {
 }
 
 
+def dynamic_import(modname, retries=3):
+    for attempt in range(retries):
+        try:
+            return import_module(modname, __package__)
+        except KeyError:
+            # Clear any half-initialized module and retry
+            sys.modules.pop(modname, None)
+            if attempt == retries - 1:
+                break
+    raise KeyError(f"Failed to import module '{modname}' after {retries} attempts")
+
+
 def __getattr__(attr_name: str) -> object:
     module_name = _dynamic_imports.get(attr_name)
     if module_name is None:
@@ -2276,7 +2365,7 @@ def __getattr__(attr_name: str) -> object:
         )
 
     try:
-        module = import_module(module_name, __package__)
+        module = dynamic_import(module_name)
         result = getattr(module, attr_name)
         return result
     except ImportError as e:
@@ -2290,5 +2379,5 @@ def __getattr__(attr_name: str) -> object:
 
 
 def __dir__():
-    lazy_attrs = list(_dynamic_imports.keys())
-    return sorted(lazy_attrs)
+    lazy_attrs = builtins.list(_dynamic_imports.keys())
+    return builtins.sorted(lazy_attrs)