rapidata 2.40.6__py3-none-any.whl → 2.41.0__py3-none-any.whl

rapidata/api_client/models/get_evaluation_workflow_results_model.py

@@ -0,0 +1,114 @@
+# 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
+from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.page_info import PageInfo
+from rapidata.api_client.models.rapid_state import RapidState
+from rapidata.api_client.models.sort_criterion import SortCriterion
+from typing import Optional, Set
+from typing_extensions import Self
+
+class GetEvaluationWorkflowResultsModel(BaseModel):
+    """
+    Model for getting the overview of a simple workflow result.
+    """ # noqa: E501
+    page: Optional[PageInfo] = None
+    states: Optional[List[RapidState]] = Field(default=None, description="An optional list of states to filter the rapids by.")
+    sort_criteria: Optional[List[SortCriterion]] = Field(default=None, description="A list of criteria to sort the results by.", alias="sortCriteria")
+    __properties: ClassVar[List[str]] = ["page", "states", "sortCriteria"]
+
+    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 GetEvaluationWorkflowResultsModel 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 page
+        if self.page:
+            _dict['page'] = self.page.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in sort_criteria (list)
+        _items = []
+        if self.sort_criteria:
+            for _item_sort_criteria in self.sort_criteria:
+                if _item_sort_criteria:
+                    _items.append(_item_sort_criteria.to_dict())
+            _dict['sortCriteria'] = _items
+        # set to None if states (nullable) is None
+        # and model_fields_set contains the field
+        if self.states is None and "states" in self.model_fields_set:
+            _dict['states'] = None
+
+        # set to None if sort_criteria (nullable) is None
+        # and model_fields_set contains the field
+        if self.sort_criteria is None and "sort_criteria" in self.model_fields_set:
+            _dict['sortCriteria'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of GetEvaluationWorkflowResultsModel from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "page": PageInfo.from_dict(obj["page"]) if obj.get("page") is not None else None,
+            "states": obj.get("states"),
+            "sortCriteria": [SortCriterion.from_dict(_item) for _item in obj["sortCriteria"]] if obj.get("sortCriteria") is not None else None
+        })
+        return _obj
+
+

rapidata/api_client/models/get_participant_by_id_result.py

@@ -29,8 +29,7 @@ class GetParticipantByIdResult(BaseModel):
     id: StrictStr
     name: StrictStr
     benchmark_id: StrictStr = Field(alias="benchmarkId")
-    dataset_id: StrictStr = Field(alias="datasetId")
-    __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "datasetId"]
+    __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -85,8 +84,7 @@ class GetParticipantByIdResult(BaseModel):
         _obj = cls.model_validate({
             "id": obj.get("id"),
             "name": obj.get("name"),
-            "benchmarkId": obj.get("benchmarkId"),
-            "datasetId": obj.get("datasetId")
+            "benchmarkId": obj.get("benchmarkId")
         })
         return _obj
 

rapidata/api_client/models/get_standing_by_id_result.py

@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
-from typing import Any, ClassVar, Dict, List, Optional
+from typing import Any, ClassVar, Dict, List
 from rapidata.api_client.models.standing_status import StandingStatus
 from typing import Optional, Set
 from typing_extensions import Self
@@ -30,10 +30,9 @@ class GetStandingByIdResult(BaseModel):
     id: StrictStr
     name: StrictStr
     benchmark_id: StrictStr = Field(alias="benchmarkId")
-    dataset_id: Optional[StrictStr] = Field(default=None, alias="datasetId")
     status: StandingStatus
     is_disabled: StrictBool = Field(alias="isDisabled")
-    __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "datasetId", "status", "isDisabled"]
+    __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "status", "isDisabled"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -74,11 +73,6 @@ class GetStandingByIdResult(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
@@ -94,7 +88,6 @@ class GetStandingByIdResult(BaseModel):
             "id": obj.get("id"),
             "name": obj.get("name"),
             "benchmarkId": obj.get("benchmarkId"),
-            "datasetId": obj.get("datasetId"),
             "status": obj.get("status"),
             "isDisabled": obj.get("isDisabled")
         })

rapidata/api_client/models/labeling_selection.py

@@ -28,10 +28,11 @@ class LabelingSelection(BaseModel):
     LabelingSelection
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for LabelingSelection", alias="_t")
-    amount: StrictInt
+    amount: Optional[StrictInt] = None
+    effort_budget: Optional[StrictInt] = Field(default=None, 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", "amount", "retrievalMode", "maxIterations"]
+    __properties: ClassVar[List[str]] = ["_t", "amount", "effortBudget", "retrievalMode", "maxIterations"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -79,6 +80,11 @@ class LabelingSelection(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # set to None if amount (nullable) is None
+        # and model_fields_set contains the field
+        if self.amount is None and "amount" in self.model_fields_set:
+            _dict['amount'] = None
+
         # 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:
@@ -98,6 +104,7 @@ class LabelingSelection(BaseModel):
         _obj = cls.model_validate({
             "_t": obj.get("_t") if obj.get("_t") is not None else 'LabelingSelection',
             "amount": obj.get("amount"),
+            "effortBudget": obj.get("effortBudget"),
             "retrievalMode": obj.get("retrievalMode"),
             "maxIterations": obj.get("maxIterations")
         })

rapidata/api_client/models/participant_by_benchmark.py

@@ -30,9 +30,8 @@ class ParticipantByBenchmark(BaseModel):
     id: StrictStr
     name: StrictStr
     benchmark_id: StrictStr = Field(alias="benchmarkId")
-    dataset_id: StrictStr = Field(alias="datasetId")
     status: ParticipantStatus
-    __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "datasetId", "status"]
+    __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "status"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -88,7 +87,6 @@ class ParticipantByBenchmark(BaseModel):
             "id": obj.get("id"),
             "name": obj.get("name"),
             "benchmarkId": obj.get("benchmarkId"),
-            "datasetId": obj.get("datasetId"),
             "status": obj.get("status")
         })
         return _obj

rapidata/api_client/models/standing_by_benchmark.py

@@ -0,0 +1,113 @@
+# 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, StrictBool, StrictFloat, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional, Union
+from rapidata.api_client.models.confidence_interval import ConfidenceInterval
+from rapidata.api_client.models.standing_status import StandingStatus
+from typing import Optional, Set
+from typing_extensions import Self
+
+class StandingByBenchmark(BaseModel):
+    """
+    StandingByBenchmark
+    """ # noqa: E501
+    id: StrictStr
+    name: StrictStr
+    benchmark_id: StrictStr = Field(alias="benchmarkId")
+    status: StandingStatus
+    score: Optional[Union[StrictFloat, StrictInt]] = None
+    wins: Union[StrictFloat, StrictInt]
+    total_matches: Union[StrictFloat, StrictInt] = Field(alias="totalMatches")
+    is_disabled: StrictBool = Field(alias="isDisabled")
+    confidence_interval: Optional[ConfidenceInterval] = Field(default=None, alias="confidenceInterval")
+    __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "status", "score", "wins", "totalMatches", "isDisabled", "confidenceInterval"]
+
+    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 StandingByBenchmark 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 confidence_interval
+        if self.confidence_interval:
+            _dict['confidenceInterval'] = self.confidence_interval.to_dict()
+        # set to None if score (nullable) is None
+        # and model_fields_set contains the field
+        if self.score is None and "score" in self.model_fields_set:
+            _dict['score'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of StandingByBenchmark 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"),
+            "name": obj.get("name"),
+            "benchmarkId": obj.get("benchmarkId"),
+            "status": obj.get("status"),
+            "score": obj.get("score"),
+            "wins": obj.get("wins"),
+            "totalMatches": obj.get("totalMatches"),
+            "isDisabled": obj.get("isDisabled"),
+            "confidenceInterval": ConfidenceInterval.from_dict(obj["confidenceInterval"]) if obj.get("confidenceInterval") is not None else None
+        })
+        return _obj
+
+

rapidata/api_client/models/standing_by_leaderboard.py

@@ -31,14 +31,13 @@ class StandingByLeaderboard(BaseModel):
     id: StrictStr
     name: StrictStr
     leaderboard_id: StrictStr = Field(alias="leaderboardId")
-    dataset_id: Optional[StrictStr] = Field(default=None, alias="datasetId")
     status: StandingStatus
     score: Optional[Union[StrictFloat, StrictInt]] = None
     wins: Union[StrictFloat, StrictInt]
     total_matches: Union[StrictFloat, StrictInt] = Field(alias="totalMatches")
     is_disabled: StrictBool = Field(alias="isDisabled")
     confidence_interval: Optional[ConfidenceInterval] = Field(default=None, alias="confidenceInterval")
-    __properties: ClassVar[List[str]] = ["id", "name", "leaderboardId", "datasetId", "status", "score", "wins", "totalMatches", "isDisabled", "confidenceInterval"]
+    __properties: ClassVar[List[str]] = ["id", "name", "leaderboardId", "status", "score", "wins", "totalMatches", "isDisabled", "confidenceInterval"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -102,7 +101,6 @@ class StandingByLeaderboard(BaseModel):
             "id": obj.get("id"),
             "name": obj.get("name"),
             "leaderboardId": obj.get("leaderboardId"),
-            "datasetId": obj.get("datasetId"),
             "status": obj.get("status"),
             "score": obj.get("score"),
             "wins": obj.get("wins"),

rapidata/api_client/models/standings_by_benchmark_result.py

@@ -0,0 +1,95 @@
+# 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
+from typing import Any, ClassVar, Dict, List
+from rapidata.api_client.models.standing_by_benchmark import StandingByBenchmark
+from typing import Optional, Set
+from typing_extensions import Self
+
+class StandingsByBenchmarkResult(BaseModel):
+    """
+    StandingsByBenchmarkResult
+    """ # noqa: E501
+    items: List[StandingByBenchmark]
+    __properties: ClassVar[List[str]] = ["items"]
+
+    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 StandingsByBenchmarkResult 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 each item in items (list)
+        _items = []
+        if self.items:
+            for _item_items in self.items:
+                if _item_items:
+                    _items.append(_item_items.to_dict())
+            _dict['items'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of StandingsByBenchmarkResult from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "items": [StandingByBenchmark.from_dict(_item) for _item in obj["items"]] if obj.get("items") is not None else None
+        })
+        return _obj
+
+

rapidata/api_client/models/sticky_state.py

@@ -29,6 +29,7 @@ class StickyState(str, Enum):
     NONE = 'None'
     TEMPORARY = 'Temporary'
     PERMANENT = 'Permanent'
+    PASSIVE = 'Passive'
 
     @classmethod
     def from_json(cls, json_str: str) -> Self:

rapidata/api_client/models/update_benchmark_model.py

@@ -17,7 +17,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
@@ -27,7 +27,8 @@ class UpdateBenchmarkModel(BaseModel):
     The model used to update a benchmark.
     """ # noqa: E501
     name: Optional[StrictStr] = None
-    __properties: ClassVar[List[str]] = ["name"]
+    is_public: Optional[StrictBool] = Field(default=None, alias="isPublic")
+    __properties: ClassVar[List[str]] = ["name", "isPublic"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -80,7 +81,8 @@ class UpdateBenchmarkModel(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "name": obj.get("name")
+            "name": obj.get("name"),
+            "isPublic": obj.get("isPublic")
         })
         return _obj
 

rapidata/api_client/models/update_priority_model.py

@@ -0,0 +1,87 @@
+# 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
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+
+class UpdatePriorityModel(BaseModel):
+    """
+    The model to update the campaign priority.
+    """ # noqa: E501
+    priority: StrictInt = Field(description="The new priority value for the campaign.")
+    __properties: ClassVar[List[str]] = ["priority"]
+
+    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 UpdatePriorityModel 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 UpdatePriorityModel from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "priority": obj.get("priority")
+        })
+        return _obj
+
+

rapidata/api_client/models/validation_set_model.py

@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from datetime import datetime
-from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List
 from typing import Optional, Set
 from typing_extensions import Self
@@ -29,11 +29,38 @@ class ValidationSetModel(BaseModel):
     """ # noqa: E501
     id: StrictStr
     name: StrictStr
+    asset_type: List[StrictStr] = Field(alias="assetType")
+    modality: List[StrictStr]
+    prompt_type: List[StrictStr] = Field(alias="promptType")
     is_public: StrictBool = Field(alias="isPublic")
     owner_id: StrictStr = Field(alias="ownerId")
     owner_mail: StrictStr = Field(alias="ownerMail")
     created_at: datetime = Field(alias="createdAt")
-    __properties: ClassVar[List[str]] = ["id", "name", "isPublic", "ownerId", "ownerMail", "createdAt"]
+    __properties: ClassVar[List[str]] = ["id", "name", "assetType", "modality", "promptType", "isPublic", "ownerId", "ownerMail", "createdAt"]
+
+    @field_validator('asset_type')
+    def asset_type_validate_enum(cls, value):
+        """Validates the enum"""
+        for i in value:
+            if i not in set(['None', 'Image', 'Video', 'Audio', 'Text']):
+                raise ValueError("each list item must be one of ('None', 'Image', 'Video', 'Audio', 'Text')")
+        return value
+
+    @field_validator('modality')
+    def modality_validate_enum(cls, value):
+        """Validates the enum"""
+        for i in value:
+            if i not in set(['None', 'BoundingBox', 'Classify', 'Compare', 'FreeText', 'Line', 'Locate', 'NamedEntity', 'Polygon', 'Scrub', 'Transcription']):
+                raise ValueError("each list item must be one of ('None', 'BoundingBox', 'Classify', 'Compare', 'FreeText', 'Line', 'Locate', 'NamedEntity', 'Polygon', 'Scrub', 'Transcription')")
+        return value
+
+    @field_validator('prompt_type')
+    def prompt_type_validate_enum(cls, value):
+        """Validates the enum"""
+        for i in value:
+            if i not in set(['None', 'Text', 'Asset']):
+                raise ValueError("each list item must be one of ('None', 'Text', 'Asset')")
+        return value
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -88,6 +115,9 @@ class ValidationSetModel(BaseModel):
         _obj = cls.model_validate({
             "id": obj.get("id"),
             "name": obj.get("name"),
+            "assetType": obj.get("assetType"),
+            "modality": obj.get("modality"),
+            "promptType": obj.get("promptType"),
             "isPublic": obj.get("isPublic"),
             "ownerId": obj.get("ownerId"),
             "ownerMail": obj.get("ownerMail"),