mixpeek 0.6.1__py3-none-any.whl → 0.6.2__py3-none-any.whl

mixpeek/parse/client.py

@@ -9,96 +9,15 @@ from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.jsonable_encoder import jsonable_encoder
 from ..core.remove_none_from_dict import remove_none_from_dict
 from ..core.request_options import RequestOptions
-from ..errors.bad_request_error import BadRequestError
-from ..errors.forbidden_error import ForbiddenError
-from ..errors.internal_server_error import InternalServerError
-from ..errors.not_found_error import NotFoundError
-from ..errors.unauthorized_error import UnauthorizedError
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..types.audio_params import AudioParams
-from ..types.csv_params import CsvParams
-from ..types.error_response import ErrorResponse
-from ..types.html_params import HtmlParams
-from ..types.http_validation_error import HttpValidationError
-from ..types.image_params import ImageParams
-from ..types.pdf_params import PdfParams
-from ..types.ppt_params import PptParams
-from ..types.pptx_params import PptxParams
-from ..types.txt_params import TxtParams
-from ..types.video_params import VideoParams
-from ..types.xlsx_params import XlsxParams
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
 
 
 class ParseClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def parse_file(
-        self,
-        *,
-        file_url: typing.Optional[str] = OMIT,
-        contents: typing.Optional[str] = OMIT,
-        should_chunk: typing.Optional[bool] = OMIT,
-        clean_text: typing.Optional[bool] = OMIT,
-        max_characters_per_chunk: typing.Optional[int] = OMIT,
-        extract_tags: typing.Optional[bool] = OMIT,
-        summarize: typing.Optional[bool] = OMIT,
-        pdf_settings: typing.Optional[PdfParams] = OMIT,
-        html_settings: typing.Optional[HtmlParams] = OMIT,
-        csv_settings: typing.Optional[CsvParams] = OMIT,
-        ppt_settings: typing.Optional[PptParams] = OMIT,
-        pptx_settings: typing.Optional[PptxParams] = OMIT,
-        xlsx_settings: typing.Optional[XlsxParams] = OMIT,
-        txt_settings: typing.Optional[TxtParams] = OMIT,
-        audio_settings: typing.Optional[AudioParams] = OMIT,
-        image_settings: typing.Optional[ImageParams] = OMIT,
-        video_settings: typing.Optional[VideoParams] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Any:
+    def parse_file(self, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
         Parameters:
-            - file_url: typing.Optional[str].
-
-            - contents: typing.Optional[str].
-
-            - should_chunk: typing.Optional[bool].
-
-            - clean_text: typing.Optional[bool].
-
-            - max_characters_per_chunk: typing.Optional[int].
-
-            - extract_tags: typing.Optional[bool].
-
-            - summarize: typing.Optional[bool].
-
-            - pdf_settings: typing.Optional[PdfParams].
-
-            - html_settings: typing.Optional[HtmlParams].
-
-            - csv_settings: typing.Optional[CsvParams].
-
-            - ppt_settings: typing.Optional[PptParams].
-
-            - pptx_settings: typing.Optional[PptxParams].
-
-            - xlsx_settings: typing.Optional[XlsxParams].
-
-            - txt_settings: typing.Optional[TxtParams].
-
-            - audio_settings: typing.Optional[AudioParams].
-
-            - image_settings: typing.Optional[ImageParams].
-
-            - video_settings: typing.Optional[VideoParams].
-
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from mixpeek.client import Mixpeek
@@ -107,57 +26,18 @@ class ParseClient:
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.parse.parse_file()
         """
-        _request: typing.Dict[str, typing.Any] = {}
-        if file_url is not OMIT:
-            _request["file_url"] = file_url
-        if contents is not OMIT:
-            _request["contents"] = contents
-        if should_chunk is not OMIT:
-            _request["should_chunk"] = should_chunk
-        if clean_text is not OMIT:
-            _request["clean_text"] = clean_text
-        if max_characters_per_chunk is not OMIT:
-            _request["max_characters_per_chunk"] = max_characters_per_chunk
-        if extract_tags is not OMIT:
-            _request["extract_tags"] = extract_tags
-        if summarize is not OMIT:
-            _request["summarize"] = summarize
-        if pdf_settings is not OMIT:
-            _request["pdf_settings"] = pdf_settings
-        if html_settings is not OMIT:
-            _request["html_settings"] = html_settings
-        if csv_settings is not OMIT:
-            _request["csv_settings"] = csv_settings
-        if ppt_settings is not OMIT:
-            _request["ppt_settings"] = ppt_settings
-        if pptx_settings is not OMIT:
-            _request["pptx_settings"] = pptx_settings
-        if xlsx_settings is not OMIT:
-            _request["xlsx_settings"] = xlsx_settings
-        if txt_settings is not OMIT:
-            _request["txt_settings"] = txt_settings
-        if audio_settings is not OMIT:
-            _request["audio_settings"] = audio_settings
-        if image_settings is not OMIT:
-            _request["image_settings"] = image_settings
-        if video_settings is not OMIT:
-            _request["video_settings"] = video_settings
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "parsers"),
             params=jsonable_encoder(
                 request_options.get("additional_query_parameters") if request_options is not None else None
             ),
-            json=jsonable_encoder(_request)
-            if request_options is None or request_options.get("additional_body_parameters") is None
-            else {
-                **jsonable_encoder(_request),
-                **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
-            },
+            json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))
+            if request_options is not None
+            else None,
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {
@@ -173,19 +53,7 @@ class ParseClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.Any, _response.json())  # type: ignore
-        if _response.status_code == 400:
-            raise BadRequestError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
-        if _response.status_code == 401:
-            raise UnauthorizedError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
-        if _response.status_code == 403:
-            raise ForbiddenError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
-        if _response.status_code == 404:
-            raise NotFoundError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
-        if _response.status_code == 500:
-            raise InternalServerError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
+            return
         try:
             _response_json = _response.json()
         except JSONDecodeError:
@@ -197,64 +65,9 @@ class AsyncParseClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def parse_file(
-        self,
-        *,
-        file_url: typing.Optional[str] = OMIT,
-        contents: typing.Optional[str] = OMIT,
-        should_chunk: typing.Optional[bool] = OMIT,
-        clean_text: typing.Optional[bool] = OMIT,
-        max_characters_per_chunk: typing.Optional[int] = OMIT,
-        extract_tags: typing.Optional[bool] = OMIT,
-        summarize: typing.Optional[bool] = OMIT,
-        pdf_settings: typing.Optional[PdfParams] = OMIT,
-        html_settings: typing.Optional[HtmlParams] = OMIT,
-        csv_settings: typing.Optional[CsvParams] = OMIT,
-        ppt_settings: typing.Optional[PptParams] = OMIT,
-        pptx_settings: typing.Optional[PptxParams] = OMIT,
-        xlsx_settings: typing.Optional[XlsxParams] = OMIT,
-        txt_settings: typing.Optional[TxtParams] = OMIT,
-        audio_settings: typing.Optional[AudioParams] = OMIT,
-        image_settings: typing.Optional[ImageParams] = OMIT,
-        video_settings: typing.Optional[VideoParams] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Any:
+    async def parse_file(self, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
         Parameters:
-            - file_url: typing.Optional[str].
-
-            - contents: typing.Optional[str].
-
-            - should_chunk: typing.Optional[bool].
-
-            - clean_text: typing.Optional[bool].
-
-            - max_characters_per_chunk: typing.Optional[int].
-
-            - extract_tags: typing.Optional[bool].
-
-            - summarize: typing.Optional[bool].
-
-            - pdf_settings: typing.Optional[PdfParams].
-
-            - html_settings: typing.Optional[HtmlParams].
-
-            - csv_settings: typing.Optional[CsvParams].
-
-            - ppt_settings: typing.Optional[PptParams].
-
-            - pptx_settings: typing.Optional[PptxParams].
-
-            - xlsx_settings: typing.Optional[XlsxParams].
-
-            - txt_settings: typing.Optional[TxtParams].
-
-            - audio_settings: typing.Optional[AudioParams].
-
-            - image_settings: typing.Optional[ImageParams].
-
-            - video_settings: typing.Optional[VideoParams].
-
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
         from mixpeek.client import AsyncMixpeek
@@ -263,57 +76,18 @@ class AsyncParseClient:
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         await client.parse.parse_file()
         """
-        _request: typing.Dict[str, typing.Any] = {}
-        if file_url is not OMIT:
-            _request["file_url"] = file_url
-        if contents is not OMIT:
-            _request["contents"] = contents
-        if should_chunk is not OMIT:
-            _request["should_chunk"] = should_chunk
-        if clean_text is not OMIT:
-            _request["clean_text"] = clean_text
-        if max_characters_per_chunk is not OMIT:
-            _request["max_characters_per_chunk"] = max_characters_per_chunk
-        if extract_tags is not OMIT:
-            _request["extract_tags"] = extract_tags
-        if summarize is not OMIT:
-            _request["summarize"] = summarize
-        if pdf_settings is not OMIT:
-            _request["pdf_settings"] = pdf_settings
-        if html_settings is not OMIT:
-            _request["html_settings"] = html_settings
-        if csv_settings is not OMIT:
-            _request["csv_settings"] = csv_settings
-        if ppt_settings is not OMIT:
-            _request["ppt_settings"] = ppt_settings
-        if pptx_settings is not OMIT:
-            _request["pptx_settings"] = pptx_settings
-        if xlsx_settings is not OMIT:
-            _request["xlsx_settings"] = xlsx_settings
-        if txt_settings is not OMIT:
-            _request["txt_settings"] = txt_settings
-        if audio_settings is not OMIT:
-            _request["audio_settings"] = audio_settings
-        if image_settings is not OMIT:
-            _request["image_settings"] = image_settings
-        if video_settings is not OMIT:
-            _request["video_settings"] = video_settings
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "parsers"),
             params=jsonable_encoder(
                 request_options.get("additional_query_parameters") if request_options is not None else None
             ),
-            json=jsonable_encoder(_request)
-            if request_options is None or request_options.get("additional_body_parameters") is None
-            else {
-                **jsonable_encoder(_request),
-                **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
-            },
+            json=jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))
+            if request_options is not None
+            else None,
             headers=jsonable_encoder(
                 remove_none_from_dict(
                     {
@@ -329,19 +103,7 @@ class AsyncParseClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.Any, _response.json())  # type: ignore
-        if _response.status_code == 400:
-            raise BadRequestError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
-        if _response.status_code == 401:
-            raise UnauthorizedError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
-        if _response.status_code == 403:
-            raise ForbiddenError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
-        if _response.status_code == 404:
-            raise NotFoundError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
-        if _response.status_code == 500:
-            raise InternalServerError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
+            return
         try:
             _response_json = _response.json()
         except JSONDecodeError:

mixpeek/pipelines/client.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
 import typing
 import urllib.parse
 from json.decoder import JSONDecodeError
@@ -16,12 +15,9 @@ from ..errors.internal_server_error import InternalServerError
 from ..errors.not_found_error import NotFoundError
 from ..errors.unauthorized_error import UnauthorizedError
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..types.connection import Connection
-from ..types.destination_schema import DestinationSchema
 from ..types.error_response import ErrorResponse
 from ..types.http_validation_error import HttpValidationError
-from ..types.pipeline_response import PipelineResponse
-from ..types.source_schema import SourceSchema
+from ..types.source_destination_mapping import SourceDestinationMapping
 
 try:
     import pydantic.v1 as pydantic  # type: ignore
@@ -49,7 +45,6 @@ class PipelinesClient:
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.pipelines.invoke(
             pipeline_id="pipeline_id",
@@ -103,87 +98,52 @@ class PipelinesClient:
     def create(
         self,
         *,
-        connection_id: str,
-        pipeline_id: typing.Optional[str] = OMIT,
-        connection: typing.Optional[Connection] = OMIT,
-        source: SourceSchema,
-        destination: DestinationSchema,
+        source_destination_mappings: typing.Sequence[SourceDestinationMapping],
         metadata: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
-        enabled: typing.Optional[bool] = OMIT,
-        last_run: typing.Optional[dt.datetime] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> PipelineResponse:
+    ) -> typing.Any:
         """
         Parameters:
-            - connection_id: str.
-
-            - pipeline_id: typing.Optional[str].
-
-            - connection: typing.Optional[Connection].
-
-            - source: SourceSchema.
-
-            - destination: DestinationSchema.
+            - source_destination_mappings: typing.Sequence[SourceDestinationMapping].
 
             - metadata: typing.Optional[typing.Dict[str, typing.Any]].
 
-            - enabled: typing.Optional[bool].
-
-            - last_run: typing.Optional[dt.datetime].
-
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
-        from mixpeek import DestinationSchema, FieldSchema, SourceSchema
+        from mixpeek import Destination, Source, SourceDestinationMapping
         from mixpeek.client import Mixpeek
 
         client = Mixpeek(
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.pipelines.create(
-            connection_id="connection_id",
-            source=SourceSchema(
-                filters={},
-                on_operation=["on_operation"],
-                field=FieldSchema(
-                    name="name",
-                    type="url",
-                ),
-            ),
-            destination=DestinationSchema(
-                collection="collection",
-                new_field_name="new_field_name",
-                new_embeddings="new_embeddings",
-            ),
+            source_destination_mappings=[
+                SourceDestinationMapping(
+                    embedding_model="embedding_model",
+                    source=Source(
+                        field="field",
+                        type="file_url",
+                        settings={},
+                    ),
+                    destination=Destination(
+                        collection="collection",
+                        field="field",
+                        embedding="embedding",
+                    ),
+                )
+            ],
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"source": source, "destination": destination}
-        if pipeline_id is not OMIT:
-            _request["pipeline_id"] = pipeline_id
-        if connection is not OMIT:
-            _request["connection"] = connection
+        _request: typing.Dict[str, typing.Any] = {"source_destination_mappings": source_destination_mappings}
         if metadata is not OMIT:
             _request["metadata"] = metadata
-        if enabled is not OMIT:
-            _request["enabled"] = enabled
-        if last_run is not OMIT:
-            _request["last_run"] = last_run
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "pipelines"),
             params=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        "connection_id": connection_id,
-                        **(
-                            request_options.get("additional_query_parameters", {})
-                            if request_options is not None
-                            else {}
-                        ),
-                    }
-                )
+                request_options.get("additional_query_parameters") if request_options is not None else None
             ),
             json=jsonable_encoder(_request)
             if request_options is None or request_options.get("additional_body_parameters") is None
@@ -206,7 +166,7 @@ class PipelinesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(PipelineResponse, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.Any, _response.json())  # type: ignore
         if _response.status_code == 400:
             raise BadRequestError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
         if _response.status_code == 401:
@@ -240,7 +200,6 @@ class PipelinesClient:
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.pipelines.get_status(
             task_id="task_id",
@@ -306,7 +265,6 @@ class AsyncPipelinesClient:
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         await client.pipelines.invoke(
             pipeline_id="pipeline_id",
@@ -360,87 +318,52 @@ class AsyncPipelinesClient:
     async def create(
         self,
         *,
-        connection_id: str,
-        pipeline_id: typing.Optional[str] = OMIT,
-        connection: typing.Optional[Connection] = OMIT,
-        source: SourceSchema,
-        destination: DestinationSchema,
+        source_destination_mappings: typing.Sequence[SourceDestinationMapping],
         metadata: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
-        enabled: typing.Optional[bool] = OMIT,
-        last_run: typing.Optional[dt.datetime] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> PipelineResponse:
+    ) -> typing.Any:
         """
         Parameters:
-            - connection_id: str.
-
-            - pipeline_id: typing.Optional[str].
-
-            - connection: typing.Optional[Connection].
-
-            - source: SourceSchema.
-
-            - destination: DestinationSchema.
+            - source_destination_mappings: typing.Sequence[SourceDestinationMapping].
 
             - metadata: typing.Optional[typing.Dict[str, typing.Any]].
 
-            - enabled: typing.Optional[bool].
-
-            - last_run: typing.Optional[dt.datetime].
-
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
-        from mixpeek import DestinationSchema, FieldSchema, SourceSchema
+        from mixpeek import Destination, Source, SourceDestinationMapping
         from mixpeek.client import AsyncMixpeek
 
         client = AsyncMixpeek(
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         await client.pipelines.create(
-            connection_id="connection_id",
-            source=SourceSchema(
-                filters={},
-                on_operation=["on_operation"],
-                field=FieldSchema(
-                    name="name",
-                    type="url",
-                ),
-            ),
-            destination=DestinationSchema(
-                collection="collection",
-                new_field_name="new_field_name",
-                new_embeddings="new_embeddings",
-            ),
+            source_destination_mappings=[
+                SourceDestinationMapping(
+                    embedding_model="embedding_model",
+                    source=Source(
+                        field="field",
+                        type="file_url",
+                        settings={},
+                    ),
+                    destination=Destination(
+                        collection="collection",
+                        field="field",
+                        embedding="embedding",
+                    ),
+                )
+            ],
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"source": source, "destination": destination}
-        if pipeline_id is not OMIT:
-            _request["pipeline_id"] = pipeline_id
-        if connection is not OMIT:
-            _request["connection"] = connection
+        _request: typing.Dict[str, typing.Any] = {"source_destination_mappings": source_destination_mappings}
         if metadata is not OMIT:
             _request["metadata"] = metadata
-        if enabled is not OMIT:
-            _request["enabled"] = enabled
-        if last_run is not OMIT:
-            _request["last_run"] = last_run
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "pipelines"),
             params=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        "connection_id": connection_id,
-                        **(
-                            request_options.get("additional_query_parameters", {})
-                            if request_options is not None
-                            else {}
-                        ),
-                    }
-                )
+                request_options.get("additional_query_parameters") if request_options is not None else None
             ),
             json=jsonable_encoder(_request)
             if request_options is None or request_options.get("additional_body_parameters") is None
@@ -463,7 +386,7 @@ class AsyncPipelinesClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(PipelineResponse, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.Any, _response.json())  # type: ignore
         if _response.status_code == 400:
             raise BadRequestError(pydantic.parse_obj_as(ErrorResponse, _response.json()))  # type: ignore
         if _response.status_code == 401:
@@ -497,7 +420,6 @@ class AsyncPipelinesClient:
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         await client.pipelines.get_status(
             task_id="task_id",

mixpeek/storage/client.py

@@ -39,7 +39,6 @@ class StorageClient:
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.storage.test_connection()
         """
@@ -94,7 +93,6 @@ class StorageClient:
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.storage.sample_database()
         """
@@ -154,7 +152,6 @@ class AsyncStorageClient:
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         await client.storage.test_connection()
         """
@@ -209,7 +206,6 @@ class AsyncStorageClient:
             authorization="YOUR_AUTHORIZATION",
             index_id="YOUR_INDEX_ID",
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         await client.storage.sample_database()
         """