rapidata 2.27.2__py3-none-any.whl → 2.44.3__py3-none-any.whl

rapidata/api_client/models/demographic_filter.py

@@ -0,0 +1,100 @@
+# 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, StrictInt, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class DemographicFilter(BaseModel):
+    """
+    DemographicFilter
+    """ # noqa: E501
+    t: StrictStr = Field(description="Discriminator value for DemographicFilter", alias="_t")
+    identifier: StrictStr
+    values: List[StrictStr]
+    execution_order: Optional[StrictInt] = Field(default=None, alias="executionOrder")
+    __properties: ClassVar[List[str]] = ["_t", "identifier", "values", "executionOrder"]
+
+    @field_validator('t')
+    def t_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['DemographicFilter']):
+            raise ValueError("must be one of enum values ('DemographicFilter')")
+        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 DemographicFilter 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 DemographicFilter 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 'DemographicFilter',
+            "identifier": obj.get("identifier"),
+            "values": obj.get("values"),
+            "executionOrder": obj.get("executionOrder")
+        })
+        return _obj
+
+

rapidata/api_client/models/dynamic_client_registration_request.py

@@ -0,0 +1,160 @@
+# 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, 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 DynamicClientRegistrationRequest(BaseModel):
+    """
+    DynamicClientRegistrationRequest
+    """ # noqa: E501
+    redirect_uris: Optional[List[StrictStr]] = None
+    grant_types: Optional[List[StrictStr]] = None
+    response_types: Optional[List[StrictStr]] = None
+    client_id: Optional[StrictStr] = None
+    client_name: Optional[StrictStr] = None
+    client_uri: Optional[StrictStr] = None
+    logo_uri: Optional[StrictStr] = None
+    scope: Optional[StrictStr] = None
+    contacts: Optional[List[StrictStr]] = None
+    tos_uri: Optional[StrictStr] = None
+    policy_uri: Optional[StrictStr] = None
+    jwks_uri: Optional[StrictStr] = None
+    jwks: Optional[JsonWebKeySet] = None
+    __properties: ClassVar[List[str]] = ["redirect_uris", "grant_types", "response_types", "client_id", "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 DynamicClientRegistrationRequest 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_id (nullable) is None
+        # and model_fields_set contains the field
+        if self.client_id is None and "client_id" in self.model_fields_set:
+            _dict['client_id'] = 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 DynamicClientRegistrationRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "redirect_uris": obj.get("redirect_uris"),
+            "grant_types": obj.get("grant_types"),
+            "response_types": obj.get("response_types"),
+            "client_id": obj.get("client_id"),
+            "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/early_stopping_referee_model.py

@@ -24,11 +24,11 @@ from typing_extensions import Self
 
 class EarlyStoppingRefereeModel(BaseModel):
     """
-    The Early Stopping Referee can currently only be used for classification tasks and will stop once a certain threshold is reached or the maximum number of votes is collected.
+    EarlyStoppingRefereeModel
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for EarlyStoppingReferee", alias="_t")
-    max_votes: StrictInt = Field(description="The upper limit of votes that will be collected for each task.", alias="maxVotes")
-    threshold: Union[StrictFloat, StrictInt] = Field(description="The threshold that needs to be reached to stop collecting votes.")
+    max_votes: StrictInt = Field(alias="maxVotes")
+    threshold: Union[StrictFloat, StrictInt]
     __properties: ClassVar[List[str]] = ["_t", "maxVotes", "threshold"]
 
     @field_validator('t')

rapidata/api_client/models/effort_capped_selection.py

@@ -0,0 +1,106 @@
+# 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, StrictInt, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.retrieval_mode import RetrievalMode
+from typing import Optional, Set
+from typing_extensions import Self
+
+class EffortCappedSelection(BaseModel):
+    """
+    EffortCappedSelection
+    """ # noqa: E501
+    t: StrictStr = Field(description="Discriminator value for EffortCappedSelection", alias="_t")
+    effort_budget: StrictInt = Field(alias="effortBudget")
+    retrieval_mode: Optional[RetrievalMode] = Field(default=None, alias="retrievalMode")
+    max_iterations: Optional[StrictInt] = Field(default=None, alias="maxIterations")
+    __properties: ClassVar[List[str]] = ["_t", "effortBudget", "retrievalMode", "maxIterations"]
+
+    @field_validator('t')
+    def t_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['EffortCappedSelection']):
+            raise ValueError("must be one of enum values ('EffortCappedSelection')")
+        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 EffortCappedSelection 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,
+        )
+        # set to None if max_iterations (nullable) is None
+        # and model_fields_set contains the field
+        if self.max_iterations is None and "max_iterations" in self.model_fields_set:
+            _dict['maxIterations'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of EffortCappedSelection 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 'EffortCappedSelection',
+            "effortBudget": obj.get("effortBudget"),
+            "retrievalMode": obj.get("retrievalMode"),
+            "maxIterations": obj.get("maxIterations")
+        })
+        return _obj
+
+

rapidata/api_client/models/elo_config.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)

rapidata/api_client/models/elo_config_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,11 @@ from typing_extensions import Self
 
 class EloConfigModel(BaseModel):
     """
-    These parameters affect how the ELO algorithms behaves
+    EloConfigModel
     """ # noqa: E501
-    starting_elo: Optional[StrictInt] = Field(default=None, description="The starting ELO score for each datapoint.", alias="startingElo")
-    k_factor: Optional[StrictInt] = Field(default=None, description="The K-factor to use in the ELO calculation. Determines the maximum possible change in a player's Elo rating after a single match. Higher K-Factor values result in larger rating changes.", alias="kFactor")
-    scaling_factor: Optional[StrictInt] = Field(default=None, description="Scaling factor to use in the ELO calculation. Adjusts the sensitivity of the Elo rating system to differences in player ratings. It affects how much the rating changes based on the expected outcome versus the actual outcome.", alias="scalingFactor")
+    starting_elo: Optional[StrictInt] = Field(default=None, alias="startingElo")
+    k_factor: Optional[StrictInt] = Field(default=None, alias="kFactor")
+    scaling_factor: Optional[StrictInt] = Field(default=None, alias="scalingFactor")
     __properties: ClassVar[List[str]] = ["startingElo", "kFactor", "scalingFactor"]
 
     model_config = ConfigDict(

rapidata/api_client/models/entity_tag_header_value.py

@@ -0,0 +1,93 @@
+# 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, Field, StrictBool
+from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.string_segment import StringSegment
+from typing import Optional, Set
+from typing_extensions import Self
+
+class EntityTagHeaderValue(BaseModel):
+    """
+    EntityTagHeaderValue
+    """ # noqa: E501
+    tag: Optional[StringSegment] = None
+    is_weak: Optional[StrictBool] = Field(default=None, alias="isWeak")
+    __properties: ClassVar[List[str]] = ["tag", "isWeak"]
+
+    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 EntityTagHeaderValue 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 tag
+        if self.tag:
+            _dict['tag'] = self.tag.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of EntityTagHeaderValue from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "tag": StringSegment.from_dict(obj["tag"]) if obj.get("tag") is not None else None,
+            "isWeak": obj.get("isWeak")
+        })
+        return _obj
+
+

rapidata/api_client/models/evaluation_workflow_config.py

@@ -18,8 +18,9 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
 from rapidata.api_client.models.compare_workflow_model1_referee import CompareWorkflowModel1Referee
+from rapidata.api_client.models.feature_flag import FeatureFlag
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -31,7 +32,8 @@ class EvaluationWorkflowConfig(BaseModel):
     validation_set_id: StrictStr = Field(alias="validationSetId")
     referee: CompareWorkflowModel1Referee
     should_accept_incorrect: StrictBool = Field(alias="shouldAcceptIncorrect")
-    __properties: ClassVar[List[str]] = ["_t", "validationSetId", "referee", "shouldAcceptIncorrect"]
+    feature_flags: Optional[List[FeatureFlag]] = Field(default=None, alias="featureFlags")
+    __properties: ClassVar[List[str]] = ["_t", "validationSetId", "referee", "shouldAcceptIncorrect", "featureFlags"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -82,6 +84,13 @@ class EvaluationWorkflowConfig(BaseModel):
         # 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 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
         return _dict
 
     @classmethod
@@ -97,7 +106,8 @@ class EvaluationWorkflowConfig(BaseModel):
             "_t": obj.get("_t") if obj.get("_t") is not None else 'EvaluationWorkflowConfig',
             "validationSetId": obj.get("validationSetId"),
             "referee": CompareWorkflowModel1Referee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
-            "shouldAcceptIncorrect": obj.get("shouldAcceptIncorrect")
+            "shouldAcceptIncorrect": obj.get("shouldAcceptIncorrect"),
+            "featureFlags": [FeatureFlag.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None
         })
         return _obj
 

rapidata/api_client/models/evaluation_workflow_model.py

@@ -18,18 +18,20 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.feature_flag import FeatureFlag
 from typing import Optional, Set
 from typing_extensions import Self
 
 class EvaluationWorkflowModel(BaseModel):
     """
-    If the EvaluationWorkflow is chosen a validation set will be used as a source for all tasks. It functions similarly to the SimpleWorkflow as there is a 1:1 mapping between validation rapids and tasks.
+    EvaluationWorkflowModel
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for EvaluationWorkflow", alias="_t")
-    validation_set_id: StrictStr = Field(description="The Validation Set id is used to as a source for the tasks that will be sent to the user.", alias="validationSetId")
-    should_accept_incorrect: StrictBool = Field(description="Indicates if the user should get feedback on their answers if they answer wrong. If set to true the user will not notice that he was tested.", alias="shouldAcceptIncorrect")
-    __properties: ClassVar[List[str]] = ["_t", "validationSetId", "shouldAcceptIncorrect"]
+    validation_set_id: StrictStr = Field(alias="validationSetId")
+    should_accept_incorrect: StrictBool = Field(alias="shouldAcceptIncorrect")
+    feature_flags: Optional[List[FeatureFlag]] = Field(default=None, alias="featureFlags")
+    __properties: ClassVar[List[str]] = ["_t", "validationSetId", "shouldAcceptIncorrect", "featureFlags"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -77,6 +79,13 @@ class EvaluationWorkflowModel(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # 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
         return _dict
 
     @classmethod
@@ -91,7 +100,8 @@ class EvaluationWorkflowModel(BaseModel):
         _obj = cls.model_validate({
             "_t": obj.get("_t") if obj.get("_t") is not None else 'EvaluationWorkflow',
             "validationSetId": obj.get("validationSetId"),
-            "shouldAcceptIncorrect": obj.get("shouldAcceptIncorrect")
+            "shouldAcceptIncorrect": obj.get("shouldAcceptIncorrect"),
+            "featureFlags": [FeatureFlag.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None
         })
         return _obj