mistralai 1.5.1__py3-none-any.whl → 1.5.2rc1__py3-none-any.whl

mistralai/files.py

@@ -44,6 +44,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesUploadFileMultiPartBodyParams(
             file=utils.get_pydantic_model(file, models.File),
@@ -83,6 +85,7 @@ class Files(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_upload_file",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -148,6 +151,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesUploadFileMultiPartBodyParams(
             file=utils.get_pydantic_model(file, models.File),
@@ -187,6 +192,7 @@ class Files(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_upload_file",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -256,6 +262,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesListFilesRequest(
             page=page,
@@ -292,6 +300,7 @@ class Files(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_list_files",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -361,6 +370,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesListFilesRequest(
             page=page,
@@ -397,6 +408,7 @@ class Files(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_list_files",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -456,6 +468,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesRetrieveFileRequest(
             file_id=file_id,
@@ -487,6 +501,7 @@ class Files(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_retrieve_file",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -546,6 +561,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesRetrieveFileRequest(
             file_id=file_id,
@@ -577,6 +594,7 @@ class Files(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_retrieve_file",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -636,6 +654,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesDeleteFileRequest(
             file_id=file_id,
@@ -667,6 +687,7 @@ class Files(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_delete_file",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -726,6 +747,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesDeleteFileRequest(
             file_id=file_id,
@@ -757,6 +780,7 @@ class Files(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_delete_file",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -816,6 +840,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesDownloadFileRequest(
             file_id=file_id,
@@ -847,6 +873,7 @@ class Files(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_download_file",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -907,6 +934,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesDownloadFileRequest(
             file_id=file_id,
@@ -938,6 +967,7 @@ class Files(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_download_file",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -998,6 +1028,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesGetSignedURLRequest(
             file_id=file_id,
@@ -1030,6 +1062,7 @@ class Files(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_get_signed_url",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -1089,6 +1122,8 @@ class Files(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FilesAPIRoutesGetSignedURLRequest(
             file_id=file_id,
@@ -1121,6 +1156,7 @@ class Files(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="files_api_routes_get_signed_url",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(

mistralai/fim.py

@@ -60,6 +60,8 @@ class Fim(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FIMCompletionRequest(
             model=model,
@@ -103,6 +105,7 @@ class Fim(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="fim_completion_v1_fim_completions_post",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -114,12 +117,14 @@ class Fim(BaseSDK):
             retry_config=retry_config,
         )
 
-        data: Any = None
+        response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
             return utils.unmarshal_json(http_res.text, models.FIMCompletionResponse)
         if utils.match_response(http_res, "422", "application/json"):
-            data = utils.unmarshal_json(http_res.text, models.HTTPValidationErrorData)
-            raise models.HTTPValidationError(data=data)
+            response_data = utils.unmarshal_json(
+                http_res.text, models.HTTPValidationErrorData
+            )
+            raise models.HTTPValidationError(data=response_data)
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.SDKError(
@@ -189,6 +194,8 @@ class Fim(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FIMCompletionRequest(
             model=model,
@@ -232,6 +239,7 @@ class Fim(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="fim_completion_v1_fim_completions_post",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -243,12 +251,14 @@ class Fim(BaseSDK):
             retry_config=retry_config,
         )
 
-        data: Any = None
+        response_data: Any = None
         if utils.match_response(http_res, "200", "application/json"):
             return utils.unmarshal_json(http_res.text, models.FIMCompletionResponse)
         if utils.match_response(http_res, "422", "application/json"):
-            data = utils.unmarshal_json(http_res.text, models.HTTPValidationErrorData)
-            raise models.HTTPValidationError(data=data)
+            response_data = utils.unmarshal_json(
+                http_res.text, models.HTTPValidationErrorData
+            )
+            raise models.HTTPValidationError(data=response_data)
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.SDKError(
@@ -318,6 +328,8 @@ class Fim(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FIMCompletionStreamRequest(
             model=model,
@@ -361,6 +373,7 @@ class Fim(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="stream_fim",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -373,7 +386,7 @@ class Fim(BaseSDK):
             retry_config=retry_config,
         )
 
-        data: Any = None
+        response_data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
             return eventstreaming.EventStream(
                 http_res,
@@ -382,8 +395,10 @@ class Fim(BaseSDK):
             )
         if utils.match_response(http_res, "422", "application/json"):
             http_res_text = utils.stream_to_text(http_res)
-            data = utils.unmarshal_json(http_res_text, models.HTTPValidationErrorData)
-            raise models.HTTPValidationError(data=data)
+            response_data = utils.unmarshal_json(
+                http_res_text, models.HTTPValidationErrorData
+            )
+            raise models.HTTPValidationError(data=response_data)
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = utils.stream_to_text(http_res)
             raise models.SDKError(
@@ -453,6 +468,8 @@ class Fim(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.FIMCompletionStreamRequest(
             model=model,
@@ -496,6 +513,7 @@ class Fim(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="stream_fim",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -508,7 +526,7 @@ class Fim(BaseSDK):
             retry_config=retry_config,
         )
 
-        data: Any = None
+        response_data: Any = None
         if utils.match_response(http_res, "200", "text/event-stream"):
             return eventstreaming.EventStreamAsync(
                 http_res,
@@ -517,8 +535,10 @@ class Fim(BaseSDK):
             )
         if utils.match_response(http_res, "422", "application/json"):
             http_res_text = await utils.stream_to_text_async(http_res)
-            data = utils.unmarshal_json(http_res_text, models.HTTPValidationErrorData)
-            raise models.HTTPValidationError(data=data)
+            response_data = utils.unmarshal_json(
+                http_res_text, models.HTTPValidationErrorData
+            )
+            raise models.HTTPValidationError(data=response_data)
         if utils.match_response(http_res, "4XX", "*"):
             http_res_text = await utils.stream_to_text_async(http_res)
             raise models.SDKError(

mistralai/httpclient.py

@@ -94,7 +94,9 @@ class ClientOwner(Protocol):
 def close_clients(
     owner: ClientOwner,
     sync_client: Union[HttpClient, None],
+    sync_client_supplied: bool,
     async_client: Union[AsyncHttpClient, None],
+    async_client_supplied: bool,
 ) -> None:
     """
     A finalizer function that is meant to be used with weakref.finalize to close
@@ -107,13 +109,13 @@ def close_clients(
     owner.client = None
     owner.async_client = None
 
-    if sync_client is not None:
+    if sync_client is not None and not sync_client_supplied:
         try:
             sync_client.close()
         except Exception:
             pass
 
-    if async_client is not None:
+    if async_client is not None and not async_client_supplied:
         is_async = False
         try:
             asyncio.get_running_loop()

mistralai/jobs.py

@@ -52,6 +52,8 @@ class Jobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesFineTuningGetFineTuningJobsRequest(
             page=page,
@@ -91,6 +93,7 @@ class Jobs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_get_fine_tuning_jobs",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -166,6 +169,8 @@ class Jobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesFineTuningGetFineTuningJobsRequest(
             page=page,
@@ -205,6 +210,7 @@ class Jobs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_get_fine_tuning_jobs",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -290,6 +296,8 @@ class Jobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobIn(
             model=model,
@@ -339,6 +347,7 @@ class Jobs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_create_fine_tuning_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -426,6 +435,8 @@ class Jobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobIn(
             model=model,
@@ -475,6 +486,7 @@ class Jobs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_create_fine_tuning_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -536,6 +548,8 @@ class Jobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesFineTuningGetFineTuningJobRequest(
             job_id=job_id,
@@ -567,6 +581,7 @@ class Jobs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_get_fine_tuning_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -626,6 +641,8 @@ class Jobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesFineTuningGetFineTuningJobRequest(
             job_id=job_id,
@@ -657,6 +674,7 @@ class Jobs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_get_fine_tuning_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -716,6 +734,8 @@ class Jobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesFineTuningCancelFineTuningJobRequest(
             job_id=job_id,
@@ -747,6 +767,7 @@ class Jobs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_cancel_fine_tuning_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -806,6 +827,8 @@ class Jobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesFineTuningCancelFineTuningJobRequest(
             job_id=job_id,
@@ -837,6 +860,7 @@ class Jobs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_cancel_fine_tuning_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -896,6 +920,8 @@ class Jobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesFineTuningStartFineTuningJobRequest(
             job_id=job_id,
@@ -927,6 +953,7 @@ class Jobs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_start_fine_tuning_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -986,6 +1013,8 @@ class Jobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesFineTuningStartFineTuningJobRequest(
             job_id=job_id,
@@ -1017,6 +1046,7 @@ class Jobs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_fine_tuning_start_fine_tuning_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(

mistralai/mistral_jobs.py

@@ -48,6 +48,8 @@ class MistralJobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesBatchGetBatchJobsRequest(
             page=page,
@@ -85,6 +87,7 @@ class MistralJobs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_get_batch_jobs",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -156,6 +159,8 @@ class MistralJobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesBatchGetBatchJobsRequest(
             page=page,
@@ -193,6 +198,7 @@ class MistralJobs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_get_batch_jobs",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -260,6 +266,8 @@ class MistralJobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.BatchJobIn(
             input_files=input_files,
@@ -298,6 +306,7 @@ class MistralJobs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_create_batch_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -365,6 +374,8 @@ class MistralJobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.BatchJobIn(
             input_files=input_files,
@@ -403,6 +414,7 @@ class MistralJobs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_create_batch_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -462,6 +474,8 @@ class MistralJobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesBatchGetBatchJobRequest(
             job_id=job_id,
@@ -493,6 +507,7 @@ class MistralJobs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_get_batch_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -552,6 +567,8 @@ class MistralJobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesBatchGetBatchJobRequest(
             job_id=job_id,
@@ -583,6 +600,7 @@ class MistralJobs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_get_batch_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -642,6 +660,8 @@ class MistralJobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesBatchCancelBatchJobRequest(
             job_id=job_id,
@@ -673,6 +693,7 @@ class MistralJobs(BaseSDK):
 
         http_res = self.do_request(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_cancel_batch_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(
@@ -732,6 +753,8 @@ class MistralJobs(BaseSDK):
 
         if server_url is not None:
             base_url = server_url
+        else:
+            base_url = self._get_url(base_url, url_variables)
 
         request = models.JobsAPIRoutesBatchCancelBatchJobRequest(
             job_id=job_id,
@@ -763,6 +786,7 @@ class MistralJobs(BaseSDK):
 
         http_res = await self.do_request_async(
             hook_ctx=HookContext(
+                base_url=base_url or "",
                 operation_id="jobs_api_routes_batch_cancel_batch_job",
                 oauth2_scopes=[],
                 security_source=get_security_from_env(

mistralai/models/__init__.py

@@ -115,7 +115,11 @@ from .detailedjobout import (
     DetailedJobOutStatus,
     DetailedJobOutTypedDict,
 )
-from .documenturlchunk import DocumentURLChunk, DocumentURLChunkTypedDict
+from .documenturlchunk import (
+    DocumentURLChunk,
+    DocumentURLChunkType,
+    DocumentURLChunkTypedDict,
+)
 from .embeddingrequest import (
     EmbeddingRequest,
     EmbeddingRequestTypedDict,
@@ -455,6 +459,7 @@ __all__ = [
     "Document",
     "DocumentTypedDict",
     "DocumentURLChunk",
+    "DocumentURLChunkType",
     "DocumentURLChunkTypedDict",
     "EmbeddingRequest",
     "EmbeddingRequestTypedDict",