rapidata 2.24.0__py3-none-any.whl → 3.0.1__py3-none-any.whl

rapidata/api_client/models/client_model.py

@@ -0,0 +1,181 @@
+# coding: utf-8
+
+"""
+    Rapidata Asset API
+
+    The API for the Rapidata Asset service
+
+    The version of the OpenAPI document: v1
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.json_web_key_set import JsonWebKeySet
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ClientModel(BaseModel):
+    """
+    ClientModel
+    """ # noqa: E501
+    client_id: StrictStr
+    client_secret: Optional[StrictStr]
+    client_id_issued_at: StrictInt
+    client_secret_expires_at: StrictInt
+    redirect_uris: Optional[List[List[StrictStr]]]
+    grant_types: Optional[List[List[StrictStr]]]
+    response_types: Optional[List[List[StrictStr]]]
+    client_name: Optional[StrictStr]
+    client_uri: Optional[StrictStr]
+    logo_uri: Optional[StrictStr]
+    scope: Optional[StrictStr]
+    contacts: List[StrictStr]
+    tos_uri: Optional[StrictStr]
+    policy_uri: Optional[StrictStr]
+    jwks_uri: Optional[StrictStr]
+    jwks: Optional[JsonWebKeySet]
+    __properties: ClassVar[List[str]] = ["client_id", "client_secret", "client_id_issued_at", "client_secret_expires_at", "redirect_uris", "grant_types", "response_types", "client_name", "client_uri", "logo_uri", "scope", "contacts", "tos_uri", "policy_uri", "jwks_uri", "jwks"]
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of ClientModel from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of jwks
+        if self.jwks:
+            _dict['jwks'] = self.jwks.to_dict()
+        # set to None if client_secret (nullable) is None
+        # and model_fields_set contains the field
+        if self.client_secret is None and "client_secret" in self.model_fields_set:
+            _dict['client_secret'] = None
+
+        # set to None if redirect_uris (nullable) is None
+        # and model_fields_set contains the field
+        if self.redirect_uris is None and "redirect_uris" in self.model_fields_set:
+            _dict['redirect_uris'] = None
+
+        # set to None if grant_types (nullable) is None
+        # and model_fields_set contains the field
+        if self.grant_types is None and "grant_types" in self.model_fields_set:
+            _dict['grant_types'] = None
+
+        # set to None if response_types (nullable) is None
+        # and model_fields_set contains the field
+        if self.response_types is None and "response_types" in self.model_fields_set:
+            _dict['response_types'] = None
+
+        # set to None if client_name (nullable) is None
+        # and model_fields_set contains the field
+        if self.client_name is None and "client_name" in self.model_fields_set:
+            _dict['client_name'] = None
+
+        # set to None if client_uri (nullable) is None
+        # and model_fields_set contains the field
+        if self.client_uri is None and "client_uri" in self.model_fields_set:
+            _dict['client_uri'] = None
+
+        # set to None if logo_uri (nullable) is None
+        # and model_fields_set contains the field
+        if self.logo_uri is None and "logo_uri" in self.model_fields_set:
+            _dict['logo_uri'] = None
+
+        # set to None if scope (nullable) is None
+        # and model_fields_set contains the field
+        if self.scope is None and "scope" in self.model_fields_set:
+            _dict['scope'] = None
+
+        # set to None if tos_uri (nullable) is None
+        # and model_fields_set contains the field
+        if self.tos_uri is None and "tos_uri" in self.model_fields_set:
+            _dict['tos_uri'] = None
+
+        # set to None if policy_uri (nullable) is None
+        # and model_fields_set contains the field
+        if self.policy_uri is None and "policy_uri" in self.model_fields_set:
+            _dict['policy_uri'] = None
+
+        # set to None if jwks_uri (nullable) is None
+        # and model_fields_set contains the field
+        if self.jwks_uri is None and "jwks_uri" in self.model_fields_set:
+            _dict['jwks_uri'] = None
+
+        # set to None if jwks (nullable) is None
+        # and model_fields_set contains the field
+        if self.jwks is None and "jwks" in self.model_fields_set:
+            _dict['jwks'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ClientModel from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "client_id": obj.get("client_id"),
+            "client_secret": obj.get("client_secret"),
+            "client_id_issued_at": obj.get("client_id_issued_at"),
+            "client_secret_expires_at": obj.get("client_secret_expires_at"),
+            "redirect_uris": obj.get("redirect_uris"),
+            "grant_types": obj.get("grant_types"),
+            "response_types": obj.get("response_types"),
+            "client_name": obj.get("client_name"),
+            "client_uri": obj.get("client_uri"),
+            "logo_uri": obj.get("logo_uri"),
+            "scope": obj.get("scope"),
+            "contacts": obj.get("contacts"),
+            "tos_uri": obj.get("tos_uri"),
+            "policy_uri": obj.get("policy_uri"),
+            "jwks_uri": obj.get("jwks_uri"),
+            "jwks": JsonWebKeySet.from_dict(obj["jwks"]) if obj.get("jwks") is not None else None
+        })
+        return _obj
+
+

rapidata/api_client/models/clients_query_result.py

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -27,9 +27,9 @@ class ClientsQueryResult(BaseModel):
     """
     ClientsQueryResult
     """ # noqa: E501
-    client_id: Optional[StrictStr] = Field(default=None, alias="clientId")
-    display_name: Optional[StrictStr] = Field(default=None, alias="displayName")
-    created_at: Optional[datetime] = Field(default=None, alias="createdAt")
+    client_id: Optional[StrictStr] = Field(alias="clientId")
+    display_name: Optional[StrictStr] = Field(alias="displayName")
+    created_at: Optional[datetime] = Field(alias="createdAt")
     __properties: ClassVar[List[str]] = ["clientId", "displayName", "createdAt"]
 
     model_config = ConfigDict(

rapidata/api_client/models/clone_order_model.py

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -24,11 +24,10 @@ from typing_extensions import Self
 
 class CloneOrderModel(BaseModel):
     """
-    The model for cloning an existing order.
+    CloneOrderModel
     """ # noqa: E501
-    order_name: StrictStr = Field(description="The name of the new order", alias="orderName")
-    order_to_clone_id: StrictStr = Field(description="The id of the order to clone", alias="orderToCloneId")
-    __properties: ClassVar[List[str]] = ["orderName", "orderToCloneId"]
+    order_name: StrictStr = Field(alias="orderName")
+    __properties: ClassVar[List[str]] = ["orderName"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -81,8 +80,7 @@ class CloneOrderModel(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "orderName": obj.get("orderName"),
-            "orderToCloneId": obj.get("orderToCloneId")
+            "orderName": obj.get("orderName")
         })
         return _obj
 

rapidata/api_client/models/clone_order_result.py

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -26,7 +26,7 @@ class CloneOrderResult(BaseModel):
     """
     CloneOrderResult
     """ # noqa: E501
-    dataset_id: StrictStr = Field(alias="datasetId")
+    dataset_id: Optional[StrictStr] = Field(default=None, alias="datasetId")
     order_id: StrictStr = Field(alias="orderId")
     __properties: ClassVar[List[str]] = ["datasetId", "orderId"]
 
@@ -69,6 +69,11 @@ class CloneOrderResult(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # set to None if dataset_id (nullable) is None
+        # and model_fields_set contains the field
+        if self.dataset_id is None and "dataset_id" in self.model_fields_set:
+            _dict['datasetId'] = None
+
         return _dict
 
     @classmethod

rapidata/api_client/models/compare_rapid_blueprint.py

@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -28,7 +28,8 @@ class CompareRapidBlueprint(BaseModel):
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for CompareBlueprint", alias="_t")
     criteria: StrictStr
-    __properties: ClassVar[List[str]] = ["_t", "criteria"]
+    index_identifiers: Optional[List[StrictStr]] = Field(default=None, alias="indexIdentifiers")
+    __properties: ClassVar[List[str]] = ["_t", "criteria", "indexIdentifiers"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -89,7 +90,8 @@ class CompareRapidBlueprint(BaseModel):
 
         _obj = cls.model_validate({
             "_t": obj.get("_t") if obj.get("_t") is not None else 'CompareBlueprint',
-            "criteria": obj.get("criteria")
+            "criteria": obj.get("criteria"),
+            "indexIdentifiers": obj.get("indexIdentifiers")
         })
         return _obj
 

rapidata/api_client/models/compare_rapid_blueprint1.py

@@ -0,0 +1,96 @@
+# coding: utf-8
+
+"""
+    Rapidata.Dataset
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: v1
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+
+class CompareRapidBlueprint1(BaseModel):
+    """
+    CompareRapidBlueprint1
+    """ # noqa: E501
+    t: StrictStr = Field(description="Discriminator value for CompareBlueprint", alias="_t")
+    criteria: StrictStr
+    __properties: ClassVar[List[str]] = ["_t", "criteria"]
+
+    @field_validator('t')
+    def t_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['CompareBlueprint']):
+            raise ValueError("must be one of enum values ('CompareBlueprint')")
+        return value
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of CompareRapidBlueprint1 from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of CompareRapidBlueprint1 from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "_t": obj.get("_t") if obj.get("_t") is not None else 'CompareBlueprint',
+            "criteria": obj.get("criteria")
+        })
+        return _obj
+
+

rapidata/api_client/models/compare_result.py

@@ -27,10 +27,9 @@ class CompareResult(BaseModel):
     CompareResult
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for CompareResult", alias="_t")
-    winner_id: Optional[StrictStr] = Field(default=None, alias="winnerId")
     winners: Optional[List[StrictStr]] = None
     rapid_id: StrictStr = Field(alias="rapidId")
-    __properties: ClassVar[List[str]] = ["_t", "winnerId", "winners", "rapidId"]
+    __properties: ClassVar[List[str]] = ["_t", "winners", "rapidId"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -69,10 +68,8 @@ class CompareResult(BaseModel):
         * `None` is only added to the output dict for nullable fields that
           were set at model initialization. Other fields with value `None`
           are ignored.
-        * OpenAPI `readOnly` fields are excluded.
         """
         excluded_fields: Set[str] = set([
-            "winner_id",
         ])
 
         _dict = self.model_dump(
@@ -80,11 +77,6 @@ class CompareResult(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # set to None if winner_id (nullable) is None
-        # and model_fields_set contains the field
-        if self.winner_id is None and "winner_id" in self.model_fields_set:
-            _dict['winnerId'] = None
-
         return _dict
 
     @classmethod
@@ -98,7 +90,6 @@ class CompareResult(BaseModel):
 
         _obj = cls.model_validate({
             "_t": obj.get("_t") if obj.get("_t") is not None else 'CompareResult',
-            "winnerId": obj.get("winnerId"),
             "winners": obj.get("winners"),
             "rapidId": obj.get("rapidId")
         })

rapidata/api_client/models/compare_workflow_config.py

@@ -19,10 +19,11 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
-from rapidata.api_client.models.compare_workflow_config_metadata_value import CompareWorkflowConfigMetadataValue
+from rapidata.api_client.models.compare_workflow_config_context_asset import CompareWorkflowConfigContextAsset
 from rapidata.api_client.models.compare_workflow_config_pair_maker_config import CompareWorkflowConfigPairMakerConfig
 from rapidata.api_client.models.compare_workflow_model1_referee import CompareWorkflowModel1Referee
 from rapidata.api_client.models.elo_config import EloConfig
+from rapidata.api_client.models.feature_flag import FeatureFlag
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -32,12 +33,13 @@ class CompareWorkflowConfig(BaseModel):
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for CompareWorkflowConfig", alias="_t")
     criteria: StrictStr
+    referee: CompareWorkflowModel1Referee
+    context: Optional[StrictStr] = None
+    context_asset: Optional[CompareWorkflowConfigContextAsset] = Field(default=None, alias="contextAsset")
     elo_config: Optional[EloConfig] = Field(default=None, alias="eloConfig")
     pair_maker_config: Optional[CompareWorkflowConfigPairMakerConfig] = Field(default=None, alias="pairMakerConfig")
-    referee: CompareWorkflowModel1Referee
-    target_country_codes: List[StrictStr] = Field(alias="targetCountryCodes")
-    metadata: Dict[str, CompareWorkflowConfigMetadataValue]
-    __properties: ClassVar[List[str]] = ["_t", "criteria", "eloConfig", "pairMakerConfig", "referee", "targetCountryCodes", "metadata"]
+    feature_flags: Optional[List[FeatureFlag]] = Field(default=None, alias="featureFlags")
+    __properties: ClassVar[List[str]] = ["_t", "criteria", "referee", "context", "contextAsset", "eloConfig", "pairMakerConfig", "featureFlags"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -85,22 +87,35 @@ class CompareWorkflowConfig(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of referee
+        if self.referee:
+            _dict['referee'] = self.referee.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of context_asset
+        if self.context_asset:
+            _dict['contextAsset'] = self.context_asset.to_dict()
         # override the default output from pydantic by calling `to_dict()` of elo_config
         if self.elo_config:
             _dict['eloConfig'] = self.elo_config.to_dict()
         # override the default output from pydantic by calling `to_dict()` of pair_maker_config
         if self.pair_maker_config:
             _dict['pairMakerConfig'] = self.pair_maker_config.to_dict()
-        # override the default output from pydantic by calling `to_dict()` of referee
-        if self.referee:
-            _dict['referee'] = self.referee.to_dict()
-        # override the default output from pydantic by calling `to_dict()` of each value in metadata (dict)
-        _field_dict = {}
-        if self.metadata:
-            for _key_metadata in self.metadata:
-                if self.metadata[_key_metadata]:
-                    _field_dict[_key_metadata] = self.metadata[_key_metadata].to_dict()
-            _dict['metadata'] = _field_dict
+        # override the default output from pydantic by calling `to_dict()` of each item in feature_flags (list)
+        _items = []
+        if self.feature_flags:
+            for _item_feature_flags in self.feature_flags:
+                if _item_feature_flags:
+                    _items.append(_item_feature_flags.to_dict())
+            _dict['featureFlags'] = _items
+        # set to None if context (nullable) is None
+        # and model_fields_set contains the field
+        if self.context is None and "context" in self.model_fields_set:
+            _dict['context'] = None
+
+        # set to None if context_asset (nullable) is None
+        # and model_fields_set contains the field
+        if self.context_asset is None and "context_asset" in self.model_fields_set:
+            _dict['contextAsset'] = None
+
         return _dict
 
     @classmethod
@@ -115,16 +130,12 @@ class CompareWorkflowConfig(BaseModel):
         _obj = cls.model_validate({
             "_t": obj.get("_t") if obj.get("_t") is not None else 'CompareWorkflowConfig',
             "criteria": obj.get("criteria"),
+            "referee": CompareWorkflowModel1Referee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
+            "context": obj.get("context"),
+            "contextAsset": CompareWorkflowConfigContextAsset.from_dict(obj["contextAsset"]) if obj.get("contextAsset") is not None else None,
             "eloConfig": EloConfig.from_dict(obj["eloConfig"]) if obj.get("eloConfig") is not None else None,
             "pairMakerConfig": CompareWorkflowConfigPairMakerConfig.from_dict(obj["pairMakerConfig"]) if obj.get("pairMakerConfig") is not None else None,
-            "referee": CompareWorkflowModel1Referee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
-            "targetCountryCodes": obj.get("targetCountryCodes"),
-            "metadata": dict(
-                (_k, CompareWorkflowConfigMetadataValue.from_dict(_v))
-                for _k, _v in obj["metadata"].items()
-            )
-            if obj.get("metadata") is not None
-            else None
+            "featureFlags": [FeatureFlag.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None
         })
         return _obj