rapidata 2.35.1__py3-none-any.whl → 3.0.0a8__py3-none-any.whl

rapidata/api_client/models/read_bridge_token_keys_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)
@@ -17,7 +17,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
@@ -26,21 +26,13 @@ class ReadBridgeTokenKeysResult(BaseModel):
     """
     ReadBridgeTokenKeysResult
     """ # noqa: E501
-    t: StrictStr = Field(description="Discriminator value for ReadBridgeTokenKeysResult", alias="_t")
-    access_token: Optional[StrictStr] = Field(default=None, alias="accessToken")
-    expires_in: Optional[StrictInt] = Field(default=None, alias="expiresIn")
-    refresh_token: Optional[StrictStr] = Field(default=None, alias="refreshToken")
-    id_token: Optional[StrictStr] = Field(default=None, alias="idToken")
-    token_type: Optional[StrictStr] = Field(default=None, alias="tokenType")
-    scope: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["_t", "accessToken", "expiresIn", "refreshToken", "idToken", "tokenType", "scope"]
-
-    @field_validator('t')
-    def t_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(['ReadBridgeTokenKeysResult']):
-            raise ValueError("must be one of enum values ('ReadBridgeTokenKeysResult')")
-        return value
+    access_token: Optional[StrictStr] = Field(alias="accessToken")
+    expires_in: StrictInt = Field(alias="expiresIn")
+    refresh_token: Optional[StrictStr] = Field(alias="refreshToken")
+    id_token: Optional[StrictStr] = Field(alias="idToken")
+    token_type: Optional[StrictStr] = Field(alias="tokenType")
+    scope: Optional[StrictStr]
+    __properties: ClassVar[List[str]] = ["accessToken", "expiresIn", "refreshToken", "idToken", "tokenType", "scope"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -86,11 +78,6 @@ class ReadBridgeTokenKeysResult(BaseModel):
         if self.access_token is None and "access_token" in self.model_fields_set:
             _dict['accessToken'] = None
 
-        # set to None if expires_in (nullable) is None
-        # and model_fields_set contains the field
-        if self.expires_in is None and "expires_in" in self.model_fields_set:
-            _dict['expiresIn'] = None
-
         # set to None if refresh_token (nullable) is None
         # and model_fields_set contains the field
         if self.refresh_token is None and "refresh_token" in self.model_fields_set:
@@ -123,7 +110,6 @@ class ReadBridgeTokenKeysResult(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "_t": obj.get("_t") if obj.get("_t") is not None else 'ReadBridgeTokenKeysResult',
             "accessToken": obj.get("accessToken"),
             "expiresIn": obj.get("expiresIn"),
             "refreshToken": obj.get("refreshToken"),

rapidata/api_client/models/reference_asset_input.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, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ReferenceAssetInput(BaseModel):
+    """
+    ReferenceAssetInput
+    """ # noqa: E501
+    t: StrictStr = Field(description="Discriminator value for ReferenceAssetInput", alias="_t")
+    identifier: Optional[StrictStr] = None
+    reference: StrictStr
+    __properties: ClassVar[List[str]] = ["_t", "identifier", "reference"]
+
+    @field_validator('t')
+    def t_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['ReferenceAssetInput']):
+            raise ValueError("must be one of enum values ('ReferenceAssetInput')")
+        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 ReferenceAssetInput 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.
+        * OpenAPI `readOnly` fields are excluded.
+        """
+        excluded_fields: Set[str] = set([
+            "identifier",
+        ])
+
+        _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 ReferenceAssetInput 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 'ReferenceAssetInput',
+            "identifier": obj.get("identifier"),
+            "reference": obj.get("reference")
+        })
+        return _obj
+
+

rapidata/api_client/models/report_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)
@@ -17,28 +17,22 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.rapid_issue import RapidIssue
 from typing import Optional, Set
 from typing_extensions import Self
 
 class ReportModel(BaseModel):
     """
-    The model for reporting an issue with a rapid.
+    ReportModel
     """ # noqa: E501
-    issue: StrictStr
-    message: Optional[StrictStr] = Field(default=None, description="An optional message typed by the user.")
-    dump: Optional[StrictStr] = Field(default=None, description="A dump, that the frontend defines and can read again.")
-    source: Optional[StrictStr] = Field(default=None, description="An optional identifier where the report originated from.")
+    issue: RapidIssue
+    message: Optional[StrictStr] = None
+    dump: Optional[StrictStr] = None
+    source: Optional[StrictStr] = None
     __properties: ClassVar[List[str]] = ["issue", "message", "dump", "source"]
 
-    @field_validator('issue')
-    def issue_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(['Other', 'CannotSubmit', 'NoAsset', 'Inappropriate', 'NoCorrectOption', 'WrongLanguage', 'DoNotUnderstand', 'DoNotCare', 'NotSeeOptionsOrMediaDidntLoad', 'MyAnswerIsCorrect']):
-            raise ValueError("must be one of enum values ('Other', 'CannotSubmit', 'NoAsset', 'Inappropriate', 'NoCorrectOption', 'WrongLanguage', 'DoNotUnderstand', 'DoNotCare', 'NotSeeOptionsOrMediaDidntLoad', 'MyAnswerIsCorrect')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/response_count_filter.py

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

rapidata/api_client/models/response_count_user_filter_model.py

@@ -19,6 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List
+from rapidata.api_client.models.comparison_operator import ComparisonOperator
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -29,7 +30,7 @@ class ResponseCountUserFilterModel(BaseModel):
     t: StrictStr = Field(description="Discriminator value for ResponseCountFilter", alias="_t")
     response_count: StrictInt = Field(alias="responseCount")
     dimension: StrictStr
-    operator: StrictStr
+    operator: ComparisonOperator
     __properties: ClassVar[List[str]] = ["_t", "responseCount", "dimension", "operator"]
 
     @field_validator('t')
@@ -39,13 +40,6 @@ class ResponseCountUserFilterModel(BaseModel):
             raise ValueError("must be one of enum values ('ResponseCountFilter')")
         return value
 
-    @field_validator('operator')
-    def operator_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(['Equal', 'NotEqual', 'LessThan', 'LessThanOrEqual', 'GreaterThan', 'GreaterThanOrEqual']):
-            raise ValueError("must be one of enum values ('Equal', 'NotEqual', 'LessThan', 'LessThanOrEqual', 'GreaterThan', 'GreaterThanOrEqual')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/retrieval_mode.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/root_filter.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)
@@ -17,9 +17,10 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict
 from typing import Any, ClassVar, Dict, List, Optional
 from rapidata.api_client.models.filter import Filter
+from rapidata.api_client.models.logic_operator import LogicOperator
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -28,19 +29,9 @@ class RootFilter(BaseModel):
     RootFilter
     """ # noqa: E501
     filters: Optional[List[Filter]] = None
-    logic: Optional[StrictStr] = None
+    logic: Optional[LogicOperator] = None
     __properties: ClassVar[List[str]] = ["filters", "logic"]
 
-    @field_validator('logic')
-    def logic_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        if value not in set(['And', 'Or', 'Not']):
-            raise ValueError("must be one of enum values ('And', 'Or', 'Not')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/run_status.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/runs_by_leaderboard_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,8 +18,9 @@ import re  # noqa: F401
 import json
 
 from datetime import datetime
-from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.run_status import RunStatus
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -29,19 +30,12 @@ class RunsByLeaderboardResult(BaseModel):
     """ # noqa: E501
     id: StrictStr
     name: StrictStr
-    status: StrictStr
+    status: RunStatus
     created_at: datetime = Field(alias="createdAt")
     owner_mail: StrictStr = Field(alias="ownerMail")
-    order_id: Optional[StrictStr] = Field(default=None, alias="orderId")
+    order_id: Optional[StrictStr] = Field(alias="orderId")
     __properties: ClassVar[List[str]] = ["id", "name", "status", "createdAt", "ownerMail", "orderId"]
 
-    @field_validator('status')
-    def status_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(['Queued', 'Running', 'Completed', 'Failed']):
-            raise ValueError("must be one of enum values ('Queued', 'Running', 'Completed', 'Failed')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/sample_by_identifier.py

@@ -0,0 +1,125 @@
+# 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 datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.i_asset_model import IAssetModel
+from typing import Optional, Set
+from typing_extensions import Self
+
+class SampleByIdentifier(BaseModel):
+    """
+    SampleByIdentifier
+    """ # noqa: E501
+    id: StrictStr
+    identifier: StrictStr
+    participant_id: StrictStr = Field(alias="participantId")
+    participant_name: StrictStr = Field(alias="participantName")
+    asset: IAssetModel
+    prompt: Optional[StrictStr] = None
+    prompt_asset: Optional[IAssetModel] = Field(default=None, alias="promptAsset")
+    tags: List[StrictStr]
+    created_at: Optional[datetime] = Field(default=None, alias="createdAt")
+    owner_id: Optional[StrictStr] = Field(default=None, alias="ownerId")
+    owner_mail: StrictStr = Field(alias="ownerMail")
+    __properties: ClassVar[List[str]] = ["id", "identifier", "participantId", "participantName", "asset", "prompt", "promptAsset", "tags", "createdAt", "ownerId", "ownerMail"]
+
+    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 SampleByIdentifier 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 asset
+        if self.asset:
+            _dict['asset'] = self.asset.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of prompt_asset
+        if self.prompt_asset:
+            _dict['promptAsset'] = self.prompt_asset.to_dict()
+        # set to None if prompt (nullable) is None
+        # and model_fields_set contains the field
+        if self.prompt is None and "prompt" in self.model_fields_set:
+            _dict['prompt'] = None
+
+        # set to None if prompt_asset (nullable) is None
+        # and model_fields_set contains the field
+        if self.prompt_asset is None and "prompt_asset" in self.model_fields_set:
+            _dict['promptAsset'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of SampleByIdentifier from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "id": obj.get("id"),
+            "identifier": obj.get("identifier"),
+            "participantId": obj.get("participantId"),
+            "participantName": obj.get("participantName"),
+            "asset": IAssetModel.from_dict(obj["asset"]) if obj.get("asset") is not None else None,
+            "prompt": obj.get("prompt"),
+            "promptAsset": IAssetModel.from_dict(obj["promptAsset"]) if obj.get("promptAsset") is not None else None,
+            "tags": obj.get("tags"),
+            "createdAt": obj.get("createdAt"),
+            "ownerId": obj.get("ownerId"),
+            "ownerMail": obj.get("ownerMail")
+        })
+        return _obj
+
+