mixpeek 0.20.20__py3-none-any.whl → 0.21.1__py3-none-any.whl

mixpeek/namespaces.py

@@ -13,7 +13,13 @@ class Namespaces(BaseSDK):
         self,
         *,
         namespace_name: str,
-        embedding_models: List[str],
+        description: OptionalNullable[str] = UNSET,
+        feature_extractors: Optional[
+            Union[
+                List[models.BasicFeatureExtractor],
+                List[models.BasicFeatureExtractorTypedDict],
+            ]
+        ] = None,
         payload_indexes: OptionalNullable[
             Union[
                 List[models.PayloadIndexConfig],
@@ -30,7 +36,8 @@ class Namespaces(BaseSDK):
         Creates a new namespace with the specified configuration
 
         :param namespace_name: Name of the namespace to create
-        :param embedding_models: List of vector indexes to be used within this namespace. Must be one of: 'image', 'openai-clip-vit-base-patch32', 'multimodal', 'vertex-multimodal', 'text', 'baai-bge-m3', 'keyword', 'naver-splade-v3'
+        :param description: Description of the namespace
+        :param feature_extractors: List of feature extractors to use
         :param payload_indexes: List of payload index configurations
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -49,7 +56,10 @@ class Namespaces(BaseSDK):
 
         request = models.CreateNamespaceRequest(
             namespace_name=namespace_name,
-            embedding_models=embedding_models,
+            description=description,
+            feature_extractors=utils.get_pydantic_model(
+                feature_extractors, Optional[List[models.BasicFeatureExtractor]]
+            ),
             payload_indexes=utils.get_pydantic_model(
                 payload_indexes, OptionalNullable[List[models.PayloadIndexConfig]]
             ),
@@ -140,7 +150,13 @@ class Namespaces(BaseSDK):
         self,
         *,
         namespace_name: str,
-        embedding_models: List[str],
+        description: OptionalNullable[str] = UNSET,
+        feature_extractors: Optional[
+            Union[
+                List[models.BasicFeatureExtractor],
+                List[models.BasicFeatureExtractorTypedDict],
+            ]
+        ] = None,
         payload_indexes: OptionalNullable[
             Union[
                 List[models.PayloadIndexConfig],
@@ -157,7 +173,8 @@ class Namespaces(BaseSDK):
         Creates a new namespace with the specified configuration
 
         :param namespace_name: Name of the namespace to create
-        :param embedding_models: List of vector indexes to be used within this namespace. Must be one of: 'image', 'openai-clip-vit-base-patch32', 'multimodal', 'vertex-multimodal', 'text', 'baai-bge-m3', 'keyword', 'naver-splade-v3'
+        :param description: Description of the namespace
+        :param feature_extractors: List of feature extractors to use
         :param payload_indexes: List of payload index configurations
         :param retries: Override the default retry configuration for this method
         :param server_url: Override the default server URL for this method
@@ -176,7 +193,10 @@ class Namespaces(BaseSDK):
 
         request = models.CreateNamespaceRequest(
             namespace_name=namespace_name,
-            embedding_models=embedding_models,
+            description=description,
+            feature_extractors=utils.get_pydantic_model(
+                feature_extractors, Optional[List[models.BasicFeatureExtractor]]
+            ),
             payload_indexes=utils.get_pydantic_model(
                 payload_indexes, OptionalNullable[List[models.PayloadIndexConfig]]
             ),
@@ -693,272 +713,6 @@ class Namespaces(BaseSDK):
             http_res,
         )
 
-    def patch_namespace_v1_namespaces_namespace_patch(
-        self,
-        *,
-        namespace: str,
-        namespace_name: OptionalNullable[str] = UNSET,
-        payload_indexes: OptionalNullable[
-            Union[
-                List[models.PayloadIndexConfig],
-                List[models.PayloadIndexConfigTypedDict],
-            ]
-        ] = UNSET,
-        retries: OptionalNullable[utils.RetryConfig] = UNSET,
-        server_url: Optional[str] = None,
-        timeout_ms: Optional[int] = None,
-        http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.NamespaceResponse:
-        r"""Partially Update Namespace
-
-        Updates specific fields of an existing namespace
-
-        :param namespace: Either the namespace name or namespace ID
-        :param namespace_name: Name of the namespace to update
-        :param payload_indexes: Updated list of payload index configurations
-        :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
-        :param http_headers: Additional headers to set or replace on requests.
-        """
-        base_url = None
-        url_variables = None
-        if timeout_ms is None:
-            timeout_ms = self.sdk_configuration.timeout_ms
-
-        if server_url is not None:
-            base_url = server_url
-        else:
-            base_url = self._get_url(base_url, url_variables)
-
-        request = models.PatchNamespaceV1NamespacesNamespacePatchRequest(
-            namespace=namespace,
-            update_namespace_request=models.UpdateNamespaceRequest(
-                namespace_name=namespace_name,
-                payload_indexes=utils.get_pydantic_model(
-                    payload_indexes, OptionalNullable[List[models.PayloadIndexConfig]]
-                ),
-            ),
-        )
-
-        req = self._build_request(
-            method="PATCH",
-            path="/v1/namespaces/{namespace}",
-            base_url=base_url,
-            url_variables=url_variables,
-            request=request,
-            request_body_required=True,
-            request_has_path_params=True,
-            request_has_query_params=True,
-            user_agent_header="user-agent",
-            accept_header_value="application/json",
-            http_headers=http_headers,
-            security=self.sdk_configuration.security,
-            get_serialized_body=lambda: utils.serialize_request_body(
-                request.update_namespace_request,
-                False,
-                False,
-                "json",
-                models.UpdateNamespaceRequest,
-            ),
-            timeout_ms=timeout_ms,
-        )
-
-        if retries == UNSET:
-            if self.sdk_configuration.retry_config is not UNSET:
-                retries = self.sdk_configuration.retry_config
-
-        retry_config = None
-        if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
-
-        http_res = self.do_request(
-            hook_ctx=HookContext(
-                base_url=base_url or "",
-                operation_id="patch_namespace_v1_namespaces__namespace__patch",
-                oauth2_scopes=[],
-                security_source=get_security_from_env(
-                    self.sdk_configuration.security, models.Security
-                ),
-            ),
-            request=req,
-            error_status_codes=["400", "401", "403", "404", "422", "4XX", "500", "5XX"],
-            retry_config=retry_config,
-        )
-
-        response_data: Any = None
-        if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.NamespaceResponse)
-        if utils.match_response(
-            http_res, ["400", "401", "403", "404"], "application/json"
-        ):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.ErrorResponseData
-            )
-            raise models.ErrorResponse(data=response_data)
-        if utils.match_response(http_res, "422", "application/json"):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.HTTPValidationErrorData
-            )
-            raise models.HTTPValidationError(data=response_data)
-        if utils.match_response(http_res, "500", "application/json"):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.ErrorResponseData
-            )
-            raise models.ErrorResponse(data=response_data)
-        if utils.match_response(http_res, "4XX", "*"):
-            http_res_text = utils.stream_to_text(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
-        if utils.match_response(http_res, "5XX", "*"):
-            http_res_text = utils.stream_to_text(http_res)
-            raise models.APIError(
-                "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.APIError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
-
-    async def patch_namespace_v1_namespaces_namespace_patch_async(
-        self,
-        *,
-        namespace: str,
-        namespace_name: OptionalNullable[str] = UNSET,
-        payload_indexes: OptionalNullable[
-            Union[
-                List[models.PayloadIndexConfig],
-                List[models.PayloadIndexConfigTypedDict],
-            ]
-        ] = UNSET,
-        retries: OptionalNullable[utils.RetryConfig] = UNSET,
-        server_url: Optional[str] = None,
-        timeout_ms: Optional[int] = None,
-        http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.NamespaceResponse:
-        r"""Partially Update Namespace
-
-        Updates specific fields of an existing namespace
-
-        :param namespace: Either the namespace name or namespace ID
-        :param namespace_name: Name of the namespace to update
-        :param payload_indexes: Updated list of payload index configurations
-        :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
-        :param http_headers: Additional headers to set or replace on requests.
-        """
-        base_url = None
-        url_variables = None
-        if timeout_ms is None:
-            timeout_ms = self.sdk_configuration.timeout_ms
-
-        if server_url is not None:
-            base_url = server_url
-        else:
-            base_url = self._get_url(base_url, url_variables)
-
-        request = models.PatchNamespaceV1NamespacesNamespacePatchRequest(
-            namespace=namespace,
-            update_namespace_request=models.UpdateNamespaceRequest(
-                namespace_name=namespace_name,
-                payload_indexes=utils.get_pydantic_model(
-                    payload_indexes, OptionalNullable[List[models.PayloadIndexConfig]]
-                ),
-            ),
-        )
-
-        req = self._build_request_async(
-            method="PATCH",
-            path="/v1/namespaces/{namespace}",
-            base_url=base_url,
-            url_variables=url_variables,
-            request=request,
-            request_body_required=True,
-            request_has_path_params=True,
-            request_has_query_params=True,
-            user_agent_header="user-agent",
-            accept_header_value="application/json",
-            http_headers=http_headers,
-            security=self.sdk_configuration.security,
-            get_serialized_body=lambda: utils.serialize_request_body(
-                request.update_namespace_request,
-                False,
-                False,
-                "json",
-                models.UpdateNamespaceRequest,
-            ),
-            timeout_ms=timeout_ms,
-        )
-
-        if retries == UNSET:
-            if self.sdk_configuration.retry_config is not UNSET:
-                retries = self.sdk_configuration.retry_config
-
-        retry_config = None
-        if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
-
-        http_res = await self.do_request_async(
-            hook_ctx=HookContext(
-                base_url=base_url or "",
-                operation_id="patch_namespace_v1_namespaces__namespace__patch",
-                oauth2_scopes=[],
-                security_source=get_security_from_env(
-                    self.sdk_configuration.security, models.Security
-                ),
-            ),
-            request=req,
-            error_status_codes=["400", "401", "403", "404", "422", "4XX", "500", "5XX"],
-            retry_config=retry_config,
-        )
-
-        response_data: Any = None
-        if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.NamespaceResponse)
-        if utils.match_response(
-            http_res, ["400", "401", "403", "404"], "application/json"
-        ):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.ErrorResponseData
-            )
-            raise models.ErrorResponse(data=response_data)
-        if utils.match_response(http_res, "422", "application/json"):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.HTTPValidationErrorData
-            )
-            raise models.HTTPValidationError(data=response_data)
-        if utils.match_response(http_res, "500", "application/json"):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.ErrorResponseData
-            )
-            raise models.ErrorResponse(data=response_data)
-        if utils.match_response(http_res, "4XX", "*"):
-            http_res_text = await utils.stream_to_text_async(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
-        if utils.match_response(http_res, "5XX", "*"):
-            http_res_text = await utils.stream_to_text_async(http_res)
-            raise models.APIError(
-                "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.APIError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
-
     def update(
         self,
         *,
@@ -1446,211 +1200,3 @@ class Namespaces(BaseSDK):
             http_res_text,
             http_res,
         )
-
-    def list_models(
-        self,
-        *,
-        retries: OptionalNullable[utils.RetryConfig] = UNSET,
-        server_url: Optional[str] = None,
-        timeout_ms: Optional[int] = None,
-        http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.AvailableModelsResponse:
-        r"""List Available Models
-
-        Returns all available models and their configurations, scoped to the organization
-
-        :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
-        :param http_headers: Additional headers to set or replace on requests.
-        """
-        base_url = None
-        url_variables = None
-        if timeout_ms is None:
-            timeout_ms = self.sdk_configuration.timeout_ms
-
-        if server_url is not None:
-            base_url = server_url
-        else:
-            base_url = self._get_url(base_url, url_variables)
-        req = self._build_request(
-            method="GET",
-            path="/v1/namespaces/models",
-            base_url=base_url,
-            url_variables=url_variables,
-            request=None,
-            request_body_required=False,
-            request_has_path_params=False,
-            request_has_query_params=True,
-            user_agent_header="user-agent",
-            accept_header_value="application/json",
-            http_headers=http_headers,
-            security=self.sdk_configuration.security,
-            timeout_ms=timeout_ms,
-        )
-
-        if retries == UNSET:
-            if self.sdk_configuration.retry_config is not UNSET:
-                retries = self.sdk_configuration.retry_config
-
-        retry_config = None
-        if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
-
-        http_res = self.do_request(
-            hook_ctx=HookContext(
-                base_url=base_url or "",
-                operation_id="list_available_models_v1_namespaces_models_get",
-                oauth2_scopes=[],
-                security_source=get_security_from_env(
-                    self.sdk_configuration.security, models.Security
-                ),
-            ),
-            request=req,
-            error_status_codes=["400", "401", "403", "404", "422", "4XX", "500", "5XX"],
-            retry_config=retry_config,
-        )
-
-        response_data: Any = None
-        if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.AvailableModelsResponse)
-        if utils.match_response(
-            http_res, ["400", "401", "403", "404"], "application/json"
-        ):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.ErrorResponseData
-            )
-            raise models.ErrorResponse(data=response_data)
-        if utils.match_response(http_res, "422", "application/json"):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.HTTPValidationErrorData
-            )
-            raise models.HTTPValidationError(data=response_data)
-        if utils.match_response(http_res, "500", "application/json"):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.ErrorResponseData
-            )
-            raise models.ErrorResponse(data=response_data)
-        if utils.match_response(http_res, "4XX", "*"):
-            http_res_text = utils.stream_to_text(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
-        if utils.match_response(http_res, "5XX", "*"):
-            http_res_text = utils.stream_to_text(http_res)
-            raise models.APIError(
-                "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.APIError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )
-
-    async def list_models_async(
-        self,
-        *,
-        retries: OptionalNullable[utils.RetryConfig] = UNSET,
-        server_url: Optional[str] = None,
-        timeout_ms: Optional[int] = None,
-        http_headers: Optional[Mapping[str, str]] = None,
-    ) -> models.AvailableModelsResponse:
-        r"""List Available Models
-
-        Returns all available models and their configurations, scoped to the organization
-
-        :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
-        :param http_headers: Additional headers to set or replace on requests.
-        """
-        base_url = None
-        url_variables = None
-        if timeout_ms is None:
-            timeout_ms = self.sdk_configuration.timeout_ms
-
-        if server_url is not None:
-            base_url = server_url
-        else:
-            base_url = self._get_url(base_url, url_variables)
-        req = self._build_request_async(
-            method="GET",
-            path="/v1/namespaces/models",
-            base_url=base_url,
-            url_variables=url_variables,
-            request=None,
-            request_body_required=False,
-            request_has_path_params=False,
-            request_has_query_params=True,
-            user_agent_header="user-agent",
-            accept_header_value="application/json",
-            http_headers=http_headers,
-            security=self.sdk_configuration.security,
-            timeout_ms=timeout_ms,
-        )
-
-        if retries == UNSET:
-            if self.sdk_configuration.retry_config is not UNSET:
-                retries = self.sdk_configuration.retry_config
-
-        retry_config = None
-        if isinstance(retries, utils.RetryConfig):
-            retry_config = (retries, ["429", "500", "502", "503", "504"])
-
-        http_res = await self.do_request_async(
-            hook_ctx=HookContext(
-                base_url=base_url or "",
-                operation_id="list_available_models_v1_namespaces_models_get",
-                oauth2_scopes=[],
-                security_source=get_security_from_env(
-                    self.sdk_configuration.security, models.Security
-                ),
-            ),
-            request=req,
-            error_status_codes=["400", "401", "403", "404", "422", "4XX", "500", "5XX"],
-            retry_config=retry_config,
-        )
-
-        response_data: Any = None
-        if utils.match_response(http_res, "200", "application/json"):
-            return utils.unmarshal_json(http_res.text, models.AvailableModelsResponse)
-        if utils.match_response(
-            http_res, ["400", "401", "403", "404"], "application/json"
-        ):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.ErrorResponseData
-            )
-            raise models.ErrorResponse(data=response_data)
-        if utils.match_response(http_res, "422", "application/json"):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.HTTPValidationErrorData
-            )
-            raise models.HTTPValidationError(data=response_data)
-        if utils.match_response(http_res, "500", "application/json"):
-            response_data = utils.unmarshal_json(
-                http_res.text, models.ErrorResponseData
-            )
-            raise models.ErrorResponse(data=response_data)
-        if utils.match_response(http_res, "4XX", "*"):
-            http_res_text = await utils.stream_to_text_async(http_res)
-            raise models.APIError(
-                "API error occurred", http_res.status_code, http_res_text, http_res
-            )
-        if utils.match_response(http_res, "5XX", "*"):
-            http_res_text = await utils.stream_to_text_async(http_res)
-            raise models.APIError(
-                "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.APIError(
-            f"Unexpected response received (code: {http_res.status_code}, type: {content_type})",
-            http_res.status_code,
-            http_res_text,
-            http_res,
-        )