rapidata 1.7.0__py3-none-any.whl → 1.7.1__py3-none-any.whl

rapidata/api_client/api/identity_api.py

@@ -16,11 +16,11 @@ from pydantic import validate_call, Field, StrictFloat, StrictStr, StrictInt
 from typing import Any, Dict, List, Optional, Tuple, Union
 from typing_extensions import Annotated
 
-from pydantic import Field
+from pydantic import Field, StrictStr
 from typing import Optional
 from typing_extensions import Annotated
-from rapidata.api_client.models.create_client_model import CreateClientModel
-from rapidata.api_client.models.create_client_result import CreateClientResult
+from rapidata.api_client.models.create_bridge_token_result import CreateBridgeTokenResult
+from rapidata.api_client.models.read_bridge_token_keys_result import ReadBridgeTokenKeysResult
 from rapidata.api_client.models.register_temporary_customer_model import RegisterTemporaryCustomerModel
 
 from rapidata.api_client.api_client import ApiClient, RequestSerialized
@@ -42,9 +42,9 @@ class IdentityApi:
 
 
     @validate_call
-    def identity_create_client_post(
+    def identity_create_bridge_token_post(
         self,
-        create_client_model: Annotated[Optional[CreateClientModel], Field(description="The model for creating a new client.")] = None,
+        client_id: Annotated[Optional[StrictStr], Field(description="The client ID to create the keys for.")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -57,12 +57,12 @@ class IdentityApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> CreateClientResult:
-        """Creates a new client for the current customer.
+    ) -> CreateBridgeTokenResult:
+        """Creates a pair of read and write keys for a client.  The write key is used to store the authentication result.  The read key is used to retrieve the authentication result.
 
 
-        :param create_client_model: The model for creating a new client.
-        :type create_client_model: CreateClientModel
+        :param client_id: The client ID to create the keys for.
+        :type client_id: str
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -85,8 +85,8 @@ class IdentityApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._identity_create_client_post_serialize(
-            create_client_model=create_client_model,
+        _param = self._identity_create_bridge_token_post_serialize(
+            client_id=client_id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -94,7 +94,7 @@ class IdentityApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "CreateClientResult",
+            '200': "CreateBridgeTokenResult",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -108,9 +108,9 @@ class IdentityApi:
 
 
     @validate_call
-    def identity_create_client_post_with_http_info(
+    def identity_create_bridge_token_post_with_http_info(
         self,
-        create_client_model: Annotated[Optional[CreateClientModel], Field(description="The model for creating a new client.")] = None,
+        client_id: Annotated[Optional[StrictStr], Field(description="The client ID to create the keys for.")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -123,12 +123,12 @@ class IdentityApi:
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> ApiResponse[CreateClientResult]:
-        """Creates a new client for the current customer.
+    ) -> ApiResponse[CreateBridgeTokenResult]:
+        """Creates a pair of read and write keys for a client.  The write key is used to store the authentication result.  The read key is used to retrieve the authentication result.
 
 
-        :param create_client_model: The model for creating a new client.
-        :type create_client_model: CreateClientModel
+        :param client_id: The client ID to create the keys for.
+        :type client_id: str
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -151,8 +151,8 @@ class IdentityApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._identity_create_client_post_serialize(
-            create_client_model=create_client_model,
+        _param = self._identity_create_bridge_token_post_serialize(
+            client_id=client_id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -160,7 +160,7 @@ class IdentityApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "CreateClientResult",
+            '200': "CreateBridgeTokenResult",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -174,9 +174,9 @@ class IdentityApi:
 
 
     @validate_call
-    def identity_create_client_post_without_preload_content(
+    def identity_create_bridge_token_post_without_preload_content(
         self,
-        create_client_model: Annotated[Optional[CreateClientModel], Field(description="The model for creating a new client.")] = None,
+        client_id: Annotated[Optional[StrictStr], Field(description="The client ID to create the keys for.")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -190,11 +190,11 @@ class IdentityApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> RESTResponseType:
-        """Creates a new client for the current customer.
+        """Creates a pair of read and write keys for a client.  The write key is used to store the authentication result.  The read key is used to retrieve the authentication result.
 
 
-        :param create_client_model: The model for creating a new client.
-        :type create_client_model: CreateClientModel
+        :param client_id: The client ID to create the keys for.
+        :type client_id: str
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -217,8 +217,8 @@ class IdentityApi:
         :return: Returns the result object.
         """ # noqa: E501
 
-        _param = self._identity_create_client_post_serialize(
-            create_client_model=create_client_model,
+        _param = self._identity_create_bridge_token_post_serialize(
+            client_id=client_id,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -226,7 +226,7 @@ class IdentityApi:
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            '200': "CreateClientResult",
+            '200': "CreateBridgeTokenResult",
         }
         response_data = self.api_client.call_api(
             *_param,
@@ -235,9 +235,9 @@ class IdentityApi:
         return response_data.response
 
 
-    def _identity_create_client_post_serialize(
+    def _identity_create_bridge_token_post_serialize(
         self,
-        create_client_model,
+        client_id,
         _request_auth,
         _content_type,
         _headers,
@@ -260,11 +260,13 @@ class IdentityApi:
 
         # process the path parameters
         # process the query parameters
+        if client_id is not None:
+            
+            _query_params.append(('clientId', client_id))
+            
         # process the header parameters
         # process the form parameters
         # process the body parameter
-        if create_client_model is not None:
-            _body_params = create_client_model
 
 
         # set the HTTP header `Accept`
@@ -277,21 +279,6 @@ class IdentityApi:
                 ]
             )
 
-        # set the HTTP header `Content-Type`
-        if _content_type:
-            _header_params['Content-Type'] = _content_type
-        else:
-            _default_content_type = (
-                self.api_client.select_header_content_type(
-                    [
-                        'application/json', 
-                        'text/json', 
-                        'application/*+json'
-                    ]
-                )
-            )
-            if _default_content_type is not None:
-                _header_params['Content-Type'] = _default_content_type
 
         # authentication setting
         _auth_settings: List[str] = [
@@ -301,7 +288,276 @@ class IdentityApi:
 
         return self.api_client.param_serialize(
             method='POST',
-            resource_path='/Identity/CreateClient',
+            resource_path='/Identity/CreateBridgeToken',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+
+
+
+    @validate_call
+    def identity_read_bridge_token_get(
+        self,
+        read_key: Annotated[Optional[StrictStr], Field(description="The read key to read the bridge token keys for.")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ReadBridgeTokenKeysResult:
+        """Tries to read the bridge token keys for a given read key.  The read key is used to retrieve the authentication result written by the write key.
+
+
+        :param read_key: The read key to read the bridge token keys for.
+        :type read_key: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._identity_read_bridge_token_get_serialize(
+            read_key=read_key,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "ReadBridgeTokenKeysResult",
+            '202': None,
+            '404': "ProblemDetails",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    def identity_read_bridge_token_get_with_http_info(
+        self,
+        read_key: Annotated[Optional[StrictStr], Field(description="The read key to read the bridge token keys for.")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[ReadBridgeTokenKeysResult]:
+        """Tries to read the bridge token keys for a given read key.  The read key is used to retrieve the authentication result written by the write key.
+
+
+        :param read_key: The read key to read the bridge token keys for.
+        :type read_key: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._identity_read_bridge_token_get_serialize(
+            read_key=read_key,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "ReadBridgeTokenKeysResult",
+            '202': None,
+            '404': "ProblemDetails",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    def identity_read_bridge_token_get_without_preload_content(
+        self,
+        read_key: Annotated[Optional[StrictStr], Field(description="The read key to read the bridge token keys for.")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """Tries to read the bridge token keys for a given read key.  The read key is used to retrieve the authentication result written by the write key.
+
+
+        :param read_key: The read key to read the bridge token keys for.
+        :type read_key: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._identity_read_bridge_token_get_serialize(
+            read_key=read_key,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "ReadBridgeTokenKeysResult",
+            '202': None,
+            '404': "ProblemDetails",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _identity_read_bridge_token_get_serialize(
+        self,
+        read_key,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        # process the query parameters
+        if read_key is not None:
+            
+            _query_params.append(('readKey', read_key))
+            
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'text/plain', 
+                    'application/json', 
+                    'text/json'
+                ]
+            )
+
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'bearer', 
+            'oauth2'
+        ]
+
+        return self.api_client.param_serialize(
+            method='GET',
+            resource_path='/Identity/ReadBridgeToken',
             path_params=_path_params,
             query_params=_query_params,
             header_params=_header_params,

rapidata/api_client/api/order_api.py

@@ -31,7 +31,7 @@ from rapidata.api_client.models.get_order_by_id_result import GetOrderByIdResult
 from rapidata.api_client.models.get_order_results_result import GetOrderResultsResult
 from rapidata.api_client.models.get_public_orders_result import GetPublicOrdersResult
 from rapidata.api_client.models.order_model_paged_result import OrderModelPagedResult
-from rapidata.api_client.models.query_orders_model import QueryOrdersModel
+from rapidata.api_client.models.query_model import QueryModel
 from rapidata.api_client.models.unlock_order_result import UnlockOrderResult
 from rapidata.api_client.models.update_access_model import UpdateAccessModel
 from rapidata.api_client.models.update_order_model import UpdateOrderModel
@@ -2959,7 +2959,7 @@ class OrderApi:
     @validate_call
     def order_query_get(
         self,
-        request: Annotated[Optional[QueryOrdersModel], Field(description="The parameters for filtering, paging, and sorting")] = None,
+        request: Annotated[Optional[QueryModel], Field(description="The parameters for filtering, paging, and sorting")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2977,7 +2977,7 @@ class OrderApi:
 
 
         :param request: The parameters for filtering, paging, and sorting
-        :type request: QueryOrdersModel
+        :type request: QueryModel
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -3025,7 +3025,7 @@ class OrderApi:
     @validate_call
     def order_query_get_with_http_info(
         self,
-        request: Annotated[Optional[QueryOrdersModel], Field(description="The parameters for filtering, paging, and sorting")] = None,
+        request: Annotated[Optional[QueryModel], Field(description="The parameters for filtering, paging, and sorting")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -3043,7 +3043,7 @@ class OrderApi:
 
 
         :param request: The parameters for filtering, paging, and sorting
-        :type request: QueryOrdersModel
+        :type request: QueryModel
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -3091,7 +3091,7 @@ class OrderApi:
     @validate_call
     def order_query_get_without_preload_content(
         self,
-        request: Annotated[Optional[QueryOrdersModel], Field(description="The parameters for filtering, paging, and sorting")] = None,
+        request: Annotated[Optional[QueryModel], Field(description="The parameters for filtering, paging, and sorting")] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -3109,7 +3109,7 @@ class OrderApi:
 
 
         :param request: The parameters for filtering, paging, and sorting
-        :type request: QueryOrdersModel
+        :type request: QueryModel
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of

rapidata/api_client/models/__init__.py

@@ -44,6 +44,8 @@ from rapidata.api_client.models.classification_metadata import ClassificationMet
 from rapidata.api_client.models.classification_metadata_filter_config import ClassificationMetadataFilterConfig
 from rapidata.api_client.models.classification_metadata_model import ClassificationMetadataModel
 from rapidata.api_client.models.classify_payload import ClassifyPayload
+from rapidata.api_client.models.clients_query_result import ClientsQueryResult
+from rapidata.api_client.models.clients_query_result_paged_result import ClientsQueryResultPagedResult
 from rapidata.api_client.models.clone_dataset_model import CloneDatasetModel
 from rapidata.api_client.models.clone_order_model import CloneOrderModel
 from rapidata.api_client.models.clone_order_result import CloneOrderResult
@@ -65,6 +67,7 @@ from rapidata.api_client.models.count_classification_metadata_filter_config impo
 from rapidata.api_client.models.count_metadata import CountMetadata
 from rapidata.api_client.models.count_metadata_model import CountMetadataModel
 from rapidata.api_client.models.country_user_filter_model import CountryUserFilterModel
+from rapidata.api_client.models.create_bridge_token_result import CreateBridgeTokenResult
 from rapidata.api_client.models.create_client_model import CreateClientModel
 from rapidata.api_client.models.create_client_result import CreateClientResult
 from rapidata.api_client.models.create_complex_order_model import CreateComplexOrderModel
@@ -189,6 +192,7 @@ from rapidata.api_client.models.polygon_result import PolygonResult
 from rapidata.api_client.models.polygon_truth import PolygonTruth
 from rapidata.api_client.models.private_text_metadata_input import PrivateTextMetadataInput
 from rapidata.api_client.models.probabilistic_attach_category_referee_config import ProbabilisticAttachCategoryRefereeConfig
+from rapidata.api_client.models.problem_details import ProblemDetails
 from rapidata.api_client.models.prompt_metadata import PromptMetadata
 from rapidata.api_client.models.prompt_metadata_input import PromptMetadataInput
 from rapidata.api_client.models.prompt_metadata_model import PromptMetadataModel
@@ -196,7 +200,6 @@ from rapidata.api_client.models.public_order_model import PublicOrderModel
 from rapidata.api_client.models.public_text_metadata_input import PublicTextMetadataInput
 from rapidata.api_client.models.query_campaigns_model import QueryCampaignsModel
 from rapidata.api_client.models.query_model import QueryModel
-from rapidata.api_client.models.query_orders_model import QueryOrdersModel
 from rapidata.api_client.models.query_validation_rapids_result import QueryValidationRapidsResult
 from rapidata.api_client.models.query_validation_rapids_result_asset import QueryValidationRapidsResultAsset
 from rapidata.api_client.models.query_validation_rapids_result_paged_result import QueryValidationRapidsResultPagedResult
@@ -208,6 +211,7 @@ from rapidata.api_client.models.rapid_answer_result import RapidAnswerResult
 from rapidata.api_client.models.rapid_result_model import RapidResultModel
 from rapidata.api_client.models.rapid_result_model_result import RapidResultModelResult
 from rapidata.api_client.models.rapid_skipped_model import RapidSkippedModel
+from rapidata.api_client.models.read_bridge_token_keys_result import ReadBridgeTokenKeysResult
 from rapidata.api_client.models.register_temporary_customer_model import RegisterTemporaryCustomerModel
 from rapidata.api_client.models.root_filter import RootFilter
 from rapidata.api_client.models.send_completion_mail_step_model import SendCompletionMailStepModel