rapidata 2.28.5__py3-none-any.whl → 2.29.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of rapidata might be problematic. Click here for more details.

Files changed (84) hide show
  1. rapidata/__init__.py +1 -1
  2. rapidata/api_client/__init__.py +41 -8
  3. rapidata/api_client/api/__init__.py +1 -0
  4. rapidata/api_client/api/benchmark_api.py +2766 -0
  5. rapidata/api_client/api/customer_rapid_api.py +332 -1
  6. rapidata/api_client/api/dataset_api.py +16 -0
  7. rapidata/api_client/api/leaderboard_api.py +1392 -270
  8. rapidata/api_client/models/__init__.py +40 -8
  9. rapidata/api_client/models/add_campaign_model.py +3 -3
  10. rapidata/api_client/models/and_user_filter_model.py +106 -0
  11. rapidata/api_client/models/and_user_filter_model_filters_inner.py +282 -0
  12. rapidata/api_client/models/benchmark_query_result.py +94 -0
  13. rapidata/api_client/models/benchmark_query_result_paged_result.py +105 -0
  14. rapidata/api_client/models/boost_leaderboard_model.py +89 -0
  15. rapidata/api_client/models/create_benchmark_model.py +87 -0
  16. rapidata/api_client/models/create_benchmark_participant_model.py +87 -0
  17. rapidata/api_client/models/create_benchmark_participant_result.py +89 -0
  18. rapidata/api_client/models/create_benchmark_result.py +87 -0
  19. rapidata/api_client/models/create_datapoint_result.py +4 -16
  20. rapidata/api_client/models/create_leaderboard_model.py +18 -2
  21. rapidata/api_client/models/create_leaderboard_result.py +5 -3
  22. rapidata/api_client/models/create_order_model.py +3 -3
  23. rapidata/api_client/models/file_asset_input.py +104 -0
  24. rapidata/api_client/models/file_asset_input1.py +104 -0
  25. rapidata/api_client/models/file_asset_input1_file.py +168 -0
  26. rapidata/api_client/models/file_asset_input2.py +104 -0
  27. rapidata/api_client/models/file_asset_input_file.py +182 -0
  28. rapidata/api_client/models/form_file_wrapper.py +120 -0
  29. rapidata/api_client/models/get_benchmark_by_id_query.py +96 -0
  30. rapidata/api_client/models/get_benchmark_by_id_query_result.py +94 -0
  31. rapidata/api_client/models/get_benchmark_by_id_query_result_paged_result.py +105 -0
  32. rapidata/api_client/models/get_benchmark_by_id_result.py +94 -0
  33. rapidata/api_client/models/get_participant_by_id_result.py +6 -26
  34. rapidata/api_client/models/get_standing_by_id_result.py +113 -0
  35. rapidata/api_client/models/local_file_wrapper.py +120 -0
  36. rapidata/api_client/models/multi_asset_input.py +110 -0
  37. rapidata/api_client/models/multi_asset_input1.py +110 -0
  38. rapidata/api_client/models/multi_asset_input1_assets_inner.py +170 -0
  39. rapidata/api_client/models/multi_asset_input2.py +110 -0
  40. rapidata/api_client/models/multi_asset_input_assets_inner.py +170 -0
  41. rapidata/api_client/models/not_user_filter_model.py +3 -3
  42. rapidata/api_client/models/or_user_filter_model.py +3 -3
  43. rapidata/api_client/models/participant_by_benchmark.py +102 -0
  44. rapidata/api_client/models/participant_by_benchmark_paged_result.py +105 -0
  45. rapidata/api_client/models/participant_status.py +1 -4
  46. rapidata/api_client/models/potential_validation_rapid.py +103 -0
  47. rapidata/api_client/models/potential_validation_rapid_paged_result.py +105 -0
  48. rapidata/api_client/models/potential_validation_rapid_truth.py +280 -0
  49. rapidata/api_client/models/prompt_asset_metadata_input.py +3 -3
  50. rapidata/api_client/models/prompt_asset_metadata_input_asset.py +170 -0
  51. rapidata/api_client/models/prompt_by_benchmark_result.py +92 -0
  52. rapidata/api_client/models/prompt_by_benchmark_result_paged_result.py +105 -0
  53. rapidata/api_client/models/prompt_metadata_input.py +5 -3
  54. rapidata/api_client/models/proxy_file_wrapper.py +114 -0
  55. rapidata/api_client/models/query_validation_model.py +97 -0
  56. rapidata/api_client/models/standing_by_leaderboard.py +113 -0
  57. rapidata/api_client/models/standing_by_leaderboard_paged_result.py +105 -0
  58. rapidata/api_client/models/standing_status.py +38 -0
  59. rapidata/api_client/models/stream_file_wrapper.py +116 -0
  60. rapidata/api_client/models/submit_prompt_model.py +89 -0
  61. rapidata/api_client/models/text_asset_input.py +100 -0
  62. rapidata/api_client/models/transcription_metadata_input.py +5 -3
  63. rapidata/api_client/models/zip_entry_file_wrapper.py +120 -0
  64. rapidata/api_client_README.md +62 -16
  65. rapidata/rapidata_client/benchmark/leaderboard/__init__.py +0 -0
  66. rapidata/rapidata_client/benchmark/leaderboard/rapidata_leaderboard.py +62 -0
  67. rapidata/rapidata_client/benchmark/rapidata_benchmark.py +227 -0
  68. rapidata/rapidata_client/benchmark/rapidata_benchmark_manager.py +83 -0
  69. rapidata/rapidata_client/filter/not_filter.py +2 -2
  70. rapidata/rapidata_client/filter/or_filter.py +2 -2
  71. rapidata/rapidata_client/metadata/__init__.py +1 -0
  72. rapidata/rapidata_client/metadata/_media_asset_metadata.py +8 -1
  73. rapidata/rapidata_client/metadata/_prompt_identifier_metadata.py +15 -0
  74. rapidata/rapidata_client/order/_rapidata_dataset.py +6 -6
  75. rapidata/rapidata_client/order/_rapidata_order_builder.py +4 -4
  76. rapidata/rapidata_client/rapidata_client.py +3 -3
  77. rapidata/service/openapi_service.py +5 -0
  78. {rapidata-2.28.5.dist-info → rapidata-2.29.0.dist-info}/METADATA +1 -1
  79. {rapidata-2.28.5.dist-info → rapidata-2.29.0.dist-info}/RECORD +82 -35
  80. rapidata/rapidata_client/leaderboard/rapidata_leaderboard.py +0 -127
  81. rapidata/rapidata_client/leaderboard/rapidata_leaderboard_manager.py +0 -92
  82. /rapidata/rapidata_client/{leaderboard → benchmark}/__init__.py +0 -0
  83. {rapidata-2.28.5.dist-info → rapidata-2.29.0.dist-info}/LICENSE +0 -0
  84. {rapidata-2.28.5.dist-info → rapidata-2.29.0.dist-info}/WHEEL +0 -0
@@ -17,8 +17,8 @@ import pprint
17
17
  import re # noqa: F401
18
18
  import json
19
19
 
20
- from pydantic import BaseModel, ConfigDict, Field, StrictFloat, StrictInt, StrictStr, field_validator
21
- from typing import Any, ClassVar, Dict, List, Optional, Union
20
+ from pydantic import BaseModel, ConfigDict, Field, StrictStr
21
+ from typing import Any, ClassVar, Dict, List
22
22
  from typing import Optional, Set
23
23
  from typing_extensions import Self
24
24
 
@@ -28,20 +28,9 @@ class GetParticipantByIdResult(BaseModel):
28
28
  """ # noqa: E501
29
29
  id: StrictStr
30
30
  name: StrictStr
31
- leaderboard_id: StrictStr = Field(alias="leaderboardId")
31
+ benchmark_id: StrictStr = Field(alias="benchmarkId")
32
32
  dataset_id: StrictStr = Field(alias="datasetId")
33
- status: StrictStr
34
- score: Optional[Union[StrictFloat, StrictInt]] = None
35
- wins: StrictInt
36
- total_matches: StrictInt = Field(alias="totalMatches")
37
- __properties: ClassVar[List[str]] = ["id", "name", "leaderboardId", "datasetId", "status", "score", "wins", "totalMatches"]
38
-
39
- @field_validator('status')
40
- def status_validate_enum(cls, value):
41
- """Validates the enum"""
42
- if value not in set(['Created', 'Queued', 'Running', 'Completed', 'Failed']):
43
- raise ValueError("must be one of enum values ('Created', 'Queued', 'Running', 'Completed', 'Failed')")
44
- return value
33
+ __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "datasetId"]
45
34
 
46
35
  model_config = ConfigDict(
47
36
  populate_by_name=True,
@@ -82,11 +71,6 @@ class GetParticipantByIdResult(BaseModel):
82
71
  exclude=excluded_fields,
83
72
  exclude_none=True,
84
73
  )
85
- # set to None if score (nullable) is None
86
- # and model_fields_set contains the field
87
- if self.score is None and "score" in self.model_fields_set:
88
- _dict['score'] = None
89
-
90
74
  return _dict
91
75
 
92
76
  @classmethod
@@ -101,12 +85,8 @@ class GetParticipantByIdResult(BaseModel):
101
85
  _obj = cls.model_validate({
102
86
  "id": obj.get("id"),
103
87
  "name": obj.get("name"),
104
- "leaderboardId": obj.get("leaderboardId"),
105
- "datasetId": obj.get("datasetId"),
106
- "status": obj.get("status"),
107
- "score": obj.get("score"),
108
- "wins": obj.get("wins"),
109
- "totalMatches": obj.get("totalMatches")
88
+ "benchmarkId": obj.get("benchmarkId"),
89
+ "datasetId": obj.get("datasetId")
110
90
  })
111
91
  return _obj
112
92
 
@@ -0,0 +1,113 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Rapidata.Dataset
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: v1
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import pprint
17
+ import re # noqa: F401
18
+ import json
19
+
20
+ from pydantic import BaseModel, ConfigDict, Field, StrictFloat, StrictInt, StrictStr, field_validator
21
+ from typing import Any, ClassVar, Dict, List, Optional, Union
22
+ from typing import Optional, Set
23
+ from typing_extensions import Self
24
+
25
+ class GetStandingByIdResult(BaseModel):
26
+ """
27
+ GetStandingByIdResult
28
+ """ # noqa: E501
29
+ id: StrictStr
30
+ name: StrictStr
31
+ benchmark_id: StrictStr = Field(alias="benchmarkId")
32
+ dataset_id: StrictStr = Field(alias="datasetId")
33
+ status: StrictStr
34
+ score: Optional[Union[StrictFloat, StrictInt]] = None
35
+ wins: StrictInt
36
+ total_matches: StrictInt = Field(alias="totalMatches")
37
+ __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "datasetId", "status", "score", "wins", "totalMatches"]
38
+
39
+ @field_validator('status')
40
+ def status_validate_enum(cls, value):
41
+ """Validates the enum"""
42
+ if value not in set(['Created', 'Active', 'Idle']):
43
+ raise ValueError("must be one of enum values ('Created', 'Active', 'Idle')")
44
+ return value
45
+
46
+ model_config = ConfigDict(
47
+ populate_by_name=True,
48
+ validate_assignment=True,
49
+ protected_namespaces=(),
50
+ )
51
+
52
+
53
+ def to_str(self) -> str:
54
+ """Returns the string representation of the model using alias"""
55
+ return pprint.pformat(self.model_dump(by_alias=True))
56
+
57
+ def to_json(self) -> str:
58
+ """Returns the JSON representation of the model using alias"""
59
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
60
+ return json.dumps(self.to_dict())
61
+
62
+ @classmethod
63
+ def from_json(cls, json_str: str) -> Optional[Self]:
64
+ """Create an instance of GetStandingByIdResult from a JSON string"""
65
+ return cls.from_dict(json.loads(json_str))
66
+
67
+ def to_dict(self) -> Dict[str, Any]:
68
+ """Return the dictionary representation of the model using alias.
69
+
70
+ This has the following differences from calling pydantic's
71
+ `self.model_dump(by_alias=True)`:
72
+
73
+ * `None` is only added to the output dict for nullable fields that
74
+ were set at model initialization. Other fields with value `None`
75
+ are ignored.
76
+ """
77
+ excluded_fields: Set[str] = set([
78
+ ])
79
+
80
+ _dict = self.model_dump(
81
+ by_alias=True,
82
+ exclude=excluded_fields,
83
+ exclude_none=True,
84
+ )
85
+ # set to None if score (nullable) is None
86
+ # and model_fields_set contains the field
87
+ if self.score is None and "score" in self.model_fields_set:
88
+ _dict['score'] = None
89
+
90
+ return _dict
91
+
92
+ @classmethod
93
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
94
+ """Create an instance of GetStandingByIdResult from a dict"""
95
+ if obj is None:
96
+ return None
97
+
98
+ if not isinstance(obj, dict):
99
+ return cls.model_validate(obj)
100
+
101
+ _obj = cls.model_validate({
102
+ "id": obj.get("id"),
103
+ "name": obj.get("name"),
104
+ "benchmarkId": obj.get("benchmarkId"),
105
+ "datasetId": obj.get("datasetId"),
106
+ "status": obj.get("status"),
107
+ "score": obj.get("score"),
108
+ "wins": obj.get("wins"),
109
+ "totalMatches": obj.get("totalMatches")
110
+ })
111
+ return _obj
112
+
113
+
@@ -0,0 +1,120 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Rapidata.Dataset
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: v1
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import pprint
17
+ import re # noqa: F401
18
+ import json
19
+
20
+ from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, StrictStr, field_validator
21
+ from typing import Any, ClassVar, Dict, List, Optional
22
+ from typing import Optional, Set
23
+ from typing_extensions import Self
24
+
25
+ class LocalFileWrapper(BaseModel):
26
+ """
27
+ LocalFileWrapper
28
+ """ # noqa: E501
29
+ t: StrictStr = Field(description="Discriminator value for LocalFileWrapper", alias="_t")
30
+ name: Optional[StrictStr] = None
31
+ content_length: Optional[StrictInt] = Field(default=None, alias="contentLength")
32
+ content_type: Optional[StrictStr] = Field(default=None, alias="contentType")
33
+ is_in_memory: Optional[StrictBool] = Field(default=None, alias="isInMemory")
34
+ __properties: ClassVar[List[str]] = ["_t", "name", "contentLength", "contentType", "isInMemory"]
35
+
36
+ @field_validator('t')
37
+ def t_validate_enum(cls, value):
38
+ """Validates the enum"""
39
+ if value not in set(['LocalFileWrapper']):
40
+ raise ValueError("must be one of enum values ('LocalFileWrapper')")
41
+ return value
42
+
43
+ model_config = ConfigDict(
44
+ populate_by_name=True,
45
+ validate_assignment=True,
46
+ protected_namespaces=(),
47
+ )
48
+
49
+
50
+ def to_str(self) -> str:
51
+ """Returns the string representation of the model using alias"""
52
+ return pprint.pformat(self.model_dump(by_alias=True))
53
+
54
+ def to_json(self) -> str:
55
+ """Returns the JSON representation of the model using alias"""
56
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
57
+ return json.dumps(self.to_dict())
58
+
59
+ @classmethod
60
+ def from_json(cls, json_str: str) -> Optional[Self]:
61
+ """Create an instance of LocalFileWrapper from a JSON string"""
62
+ return cls.from_dict(json.loads(json_str))
63
+
64
+ def to_dict(self) -> Dict[str, Any]:
65
+ """Return the dictionary representation of the model using alias.
66
+
67
+ This has the following differences from calling pydantic's
68
+ `self.model_dump(by_alias=True)`:
69
+
70
+ * `None` is only added to the output dict for nullable fields that
71
+ were set at model initialization. Other fields with value `None`
72
+ are ignored.
73
+ * OpenAPI `readOnly` fields are excluded.
74
+ * OpenAPI `readOnly` fields are excluded.
75
+ * OpenAPI `readOnly` fields are excluded.
76
+ * OpenAPI `readOnly` fields are excluded.
77
+ """
78
+ excluded_fields: Set[str] = set([
79
+ "name",
80
+ "content_length",
81
+ "content_type",
82
+ "is_in_memory",
83
+ ])
84
+
85
+ _dict = self.model_dump(
86
+ by_alias=True,
87
+ exclude=excluded_fields,
88
+ exclude_none=True,
89
+ )
90
+ # set to None if content_length (nullable) is None
91
+ # and model_fields_set contains the field
92
+ if self.content_length is None and "content_length" in self.model_fields_set:
93
+ _dict['contentLength'] = None
94
+
95
+ # set to None if content_type (nullable) is None
96
+ # and model_fields_set contains the field
97
+ if self.content_type is None and "content_type" in self.model_fields_set:
98
+ _dict['contentType'] = None
99
+
100
+ return _dict
101
+
102
+ @classmethod
103
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
104
+ """Create an instance of LocalFileWrapper from a dict"""
105
+ if obj is None:
106
+ return None
107
+
108
+ if not isinstance(obj, dict):
109
+ return cls.model_validate(obj)
110
+
111
+ _obj = cls.model_validate({
112
+ "_t": obj.get("_t") if obj.get("_t") is not None else 'LocalFileWrapper',
113
+ "name": obj.get("name"),
114
+ "contentLength": obj.get("contentLength"),
115
+ "contentType": obj.get("contentType"),
116
+ "isInMemory": obj.get("isInMemory")
117
+ })
118
+ return _obj
119
+
120
+
@@ -0,0 +1,110 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Rapidata.Dataset
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: v1
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import pprint
17
+ import re # noqa: F401
18
+ import json
19
+
20
+ from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
21
+ from typing import Any, ClassVar, Dict, List, Optional
22
+ from typing import Optional, Set
23
+ from typing_extensions import Self
24
+
25
+ class MultiAssetInput(BaseModel):
26
+ """
27
+ MultiAssetInput
28
+ """ # noqa: E501
29
+ t: StrictStr = Field(description="Discriminator value for MultiAssetInput", alias="_t")
30
+ assets: List[MultiAssetInputAssetsInner]
31
+ identifier: Optional[StrictStr] = None
32
+ __properties: ClassVar[List[str]] = ["_t", "assets", "identifier"]
33
+
34
+ @field_validator('t')
35
+ def t_validate_enum(cls, value):
36
+ """Validates the enum"""
37
+ if value not in set(['MultiAssetInput']):
38
+ raise ValueError("must be one of enum values ('MultiAssetInput')")
39
+ return value
40
+
41
+ model_config = ConfigDict(
42
+ populate_by_name=True,
43
+ validate_assignment=True,
44
+ protected_namespaces=(),
45
+ )
46
+
47
+
48
+ def to_str(self) -> str:
49
+ """Returns the string representation of the model using alias"""
50
+ return pprint.pformat(self.model_dump(by_alias=True))
51
+
52
+ def to_json(self) -> str:
53
+ """Returns the JSON representation of the model using alias"""
54
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
55
+ return json.dumps(self.to_dict())
56
+
57
+ @classmethod
58
+ def from_json(cls, json_str: str) -> Optional[Self]:
59
+ """Create an instance of MultiAssetInput from a JSON string"""
60
+ return cls.from_dict(json.loads(json_str))
61
+
62
+ def to_dict(self) -> Dict[str, Any]:
63
+ """Return the dictionary representation of the model using alias.
64
+
65
+ This has the following differences from calling pydantic's
66
+ `self.model_dump(by_alias=True)`:
67
+
68
+ * `None` is only added to the output dict for nullable fields that
69
+ were set at model initialization. Other fields with value `None`
70
+ are ignored.
71
+ * OpenAPI `readOnly` fields are excluded.
72
+ """
73
+ excluded_fields: Set[str] = set([
74
+ "identifier",
75
+ ])
76
+
77
+ _dict = self.model_dump(
78
+ by_alias=True,
79
+ exclude=excluded_fields,
80
+ exclude_none=True,
81
+ )
82
+ # override the default output from pydantic by calling `to_dict()` of each item in assets (list)
83
+ _items = []
84
+ if self.assets:
85
+ for _item_assets in self.assets:
86
+ if _item_assets:
87
+ _items.append(_item_assets.to_dict())
88
+ _dict['assets'] = _items
89
+ return _dict
90
+
91
+ @classmethod
92
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
93
+ """Create an instance of MultiAssetInput from a dict"""
94
+ if obj is None:
95
+ return None
96
+
97
+ if not isinstance(obj, dict):
98
+ return cls.model_validate(obj)
99
+
100
+ _obj = cls.model_validate({
101
+ "_t": obj.get("_t") if obj.get("_t") is not None else 'MultiAssetInput',
102
+ "assets": [MultiAssetInputAssetsInner.from_dict(_item) for _item in obj["assets"]] if obj.get("assets") is not None else None,
103
+ "identifier": obj.get("identifier")
104
+ })
105
+ return _obj
106
+
107
+ from rapidata.api_client.models.multi_asset_input_assets_inner import MultiAssetInputAssetsInner
108
+ # TODO: Rewrite to not use raise_errors
109
+ MultiAssetInput.model_rebuild(raise_errors=False)
110
+
@@ -0,0 +1,110 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Rapidata.Dataset
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: v1
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import pprint
17
+ import re # noqa: F401
18
+ import json
19
+
20
+ from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
21
+ from typing import Any, ClassVar, Dict, List, Optional
22
+ from typing import Optional, Set
23
+ from typing_extensions import Self
24
+
25
+ class MultiAssetInput1(BaseModel):
26
+ """
27
+ MultiAssetInput1
28
+ """ # noqa: E501
29
+ t: StrictStr = Field(description="Discriminator value for MultiAssetInput", alias="_t")
30
+ assets: List[MultiAssetInput1AssetsInner]
31
+ identifier: Optional[StrictStr] = None
32
+ __properties: ClassVar[List[str]] = ["_t", "assets", "identifier"]
33
+
34
+ @field_validator('t')
35
+ def t_validate_enum(cls, value):
36
+ """Validates the enum"""
37
+ if value not in set(['MultiAssetInput']):
38
+ raise ValueError("must be one of enum values ('MultiAssetInput')")
39
+ return value
40
+
41
+ model_config = ConfigDict(
42
+ populate_by_name=True,
43
+ validate_assignment=True,
44
+ protected_namespaces=(),
45
+ )
46
+
47
+
48
+ def to_str(self) -> str:
49
+ """Returns the string representation of the model using alias"""
50
+ return pprint.pformat(self.model_dump(by_alias=True))
51
+
52
+ def to_json(self) -> str:
53
+ """Returns the JSON representation of the model using alias"""
54
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
55
+ return json.dumps(self.to_dict())
56
+
57
+ @classmethod
58
+ def from_json(cls, json_str: str) -> Optional[Self]:
59
+ """Create an instance of MultiAssetInput1 from a JSON string"""
60
+ return cls.from_dict(json.loads(json_str))
61
+
62
+ def to_dict(self) -> Dict[str, Any]:
63
+ """Return the dictionary representation of the model using alias.
64
+
65
+ This has the following differences from calling pydantic's
66
+ `self.model_dump(by_alias=True)`:
67
+
68
+ * `None` is only added to the output dict for nullable fields that
69
+ were set at model initialization. Other fields with value `None`
70
+ are ignored.
71
+ * OpenAPI `readOnly` fields are excluded.
72
+ """
73
+ excluded_fields: Set[str] = set([
74
+ "identifier",
75
+ ])
76
+
77
+ _dict = self.model_dump(
78
+ by_alias=True,
79
+ exclude=excluded_fields,
80
+ exclude_none=True,
81
+ )
82
+ # override the default output from pydantic by calling `to_dict()` of each item in assets (list)
83
+ _items = []
84
+ if self.assets:
85
+ for _item_assets in self.assets:
86
+ if _item_assets:
87
+ _items.append(_item_assets.to_dict())
88
+ _dict['assets'] = _items
89
+ return _dict
90
+
91
+ @classmethod
92
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
93
+ """Create an instance of MultiAssetInput1 from a dict"""
94
+ if obj is None:
95
+ return None
96
+
97
+ if not isinstance(obj, dict):
98
+ return cls.model_validate(obj)
99
+
100
+ _obj = cls.model_validate({
101
+ "_t": obj.get("_t") if obj.get("_t") is not None else 'MultiAssetInput',
102
+ "assets": [MultiAssetInput1AssetsInner.from_dict(_item) for _item in obj["assets"]] if obj.get("assets") is not None else None,
103
+ "identifier": obj.get("identifier")
104
+ })
105
+ return _obj
106
+
107
+ from rapidata.api_client.models.multi_asset_input1_assets_inner import MultiAssetInput1AssetsInner
108
+ # TODO: Rewrite to not use raise_errors
109
+ MultiAssetInput1.model_rebuild(raise_errors=False)
110
+
@@ -0,0 +1,170 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ Rapidata.Dataset
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: v1
9
+ Generated by OpenAPI Generator (https://openapi-generator.tech)
10
+
11
+ Do not edit the class manually.
12
+ """ # noqa: E501
13
+
14
+
15
+ from __future__ import annotations
16
+ import json
17
+ import pprint
18
+ from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
19
+ from typing import Any, List, Optional
20
+ from rapidata.api_client.models.file_asset_input1 import FileAssetInput1
21
+ from rapidata.api_client.models.text_asset_input import TextAssetInput
22
+ from rapidata.api_client.models.url_asset_input import UrlAssetInput
23
+ from pydantic import StrictStr, Field
24
+ from typing import Union, List, Set, Optional, Dict
25
+ from typing_extensions import Literal, Self
26
+
27
+ MULTIASSETINPUT1ASSETSINNER_ONE_OF_SCHEMAS = ["FileAssetInput1", "MultiAssetInput1", "TextAssetInput", "UrlAssetInput"]
28
+
29
+ class MultiAssetInput1AssetsInner(BaseModel):
30
+ """
31
+ MultiAssetInput1AssetsInner
32
+ """
33
+ # data type: FileAssetInput1
34
+ oneof_schema_1_validator: Optional[FileAssetInput1] = None
35
+ # data type: MultiAssetInput1
36
+ oneof_schema_2_validator: Optional[MultiAssetInput1] = None
37
+ # data type: TextAssetInput
38
+ oneof_schema_3_validator: Optional[TextAssetInput] = None
39
+ # data type: UrlAssetInput
40
+ oneof_schema_4_validator: Optional[UrlAssetInput] = None
41
+ actual_instance: Optional[Union[FileAssetInput1, MultiAssetInput1, TextAssetInput, UrlAssetInput]] = None
42
+ one_of_schemas: Set[str] = { "FileAssetInput1", "MultiAssetInput1", "TextAssetInput", "UrlAssetInput" }
43
+
44
+ model_config = ConfigDict(
45
+ validate_assignment=True,
46
+ protected_namespaces=(),
47
+ )
48
+
49
+
50
+ discriminator_value_class_map: Dict[str, str] = {
51
+ }
52
+
53
+ def __init__(self, *args, **kwargs) -> None:
54
+ if args:
55
+ if len(args) > 1:
56
+ raise ValueError("If a position argument is used, only 1 is allowed to set `actual_instance`")
57
+ if kwargs:
58
+ raise ValueError("If a position argument is used, keyword arguments cannot be used.")
59
+ super().__init__(actual_instance=args[0])
60
+ else:
61
+ super().__init__(**kwargs)
62
+
63
+ @field_validator('actual_instance')
64
+ def actual_instance_must_validate_oneof(cls, v):
65
+ instance = MultiAssetInput1AssetsInner.model_construct()
66
+ error_messages = []
67
+ match = 0
68
+ # validate data type: FileAssetInput1
69
+ if not isinstance(v, FileAssetInput1):
70
+ error_messages.append(f"Error! Input type `{type(v)}` is not `FileAssetInput1`")
71
+ else:
72
+ match += 1
73
+ # validate data type: MultiAssetInput1
74
+ if not isinstance(v, MultiAssetInput1):
75
+ error_messages.append(f"Error! Input type `{type(v)}` is not `MultiAssetInput1`")
76
+ else:
77
+ match += 1
78
+ # validate data type: TextAssetInput
79
+ if not isinstance(v, TextAssetInput):
80
+ error_messages.append(f"Error! Input type `{type(v)}` is not `TextAssetInput`")
81
+ else:
82
+ match += 1
83
+ # validate data type: UrlAssetInput
84
+ if not isinstance(v, UrlAssetInput):
85
+ error_messages.append(f"Error! Input type `{type(v)}` is not `UrlAssetInput`")
86
+ else:
87
+ match += 1
88
+ if match > 1:
89
+ # more than 1 match
90
+ raise ValueError("Multiple matches found when setting `actual_instance` in MultiAssetInput1AssetsInner with oneOf schemas: FileAssetInput1, MultiAssetInput1, TextAssetInput, UrlAssetInput. Details: " + ", ".join(error_messages))
91
+ elif match == 0:
92
+ # no match
93
+ raise ValueError("No match found when setting `actual_instance` in MultiAssetInput1AssetsInner with oneOf schemas: FileAssetInput1, MultiAssetInput1, TextAssetInput, UrlAssetInput. Details: " + ", ".join(error_messages))
94
+ else:
95
+ return v
96
+
97
+ @classmethod
98
+ def from_dict(cls, obj: Union[str, Dict[str, Any]]) -> Self:
99
+ return cls.from_json(json.dumps(obj))
100
+
101
+ @classmethod
102
+ def from_json(cls, json_str: str) -> Self:
103
+ """Returns the object represented by the json string"""
104
+ instance = cls.model_construct()
105
+ error_messages = []
106
+ match = 0
107
+
108
+ # deserialize data into FileAssetInput1
109
+ try:
110
+ instance.actual_instance = FileAssetInput1.from_json(json_str)
111
+ match += 1
112
+ except (ValidationError, ValueError) as e:
113
+ error_messages.append(str(e))
114
+ # deserialize data into MultiAssetInput1
115
+ try:
116
+ instance.actual_instance = MultiAssetInput1.from_json(json_str)
117
+ match += 1
118
+ except (ValidationError, ValueError) as e:
119
+ error_messages.append(str(e))
120
+ # deserialize data into TextAssetInput
121
+ try:
122
+ instance.actual_instance = TextAssetInput.from_json(json_str)
123
+ match += 1
124
+ except (ValidationError, ValueError) as e:
125
+ error_messages.append(str(e))
126
+ # deserialize data into UrlAssetInput
127
+ try:
128
+ instance.actual_instance = UrlAssetInput.from_json(json_str)
129
+ match += 1
130
+ except (ValidationError, ValueError) as e:
131
+ error_messages.append(str(e))
132
+
133
+ if match > 1:
134
+ # more than 1 match
135
+ raise ValueError("Multiple matches found when deserializing the JSON string into MultiAssetInput1AssetsInner with oneOf schemas: FileAssetInput1, MultiAssetInput1, TextAssetInput, UrlAssetInput. Details: " + ", ".join(error_messages))
136
+ elif match == 0:
137
+ # no match
138
+ raise ValueError("No match found when deserializing the JSON string into MultiAssetInput1AssetsInner with oneOf schemas: FileAssetInput1, MultiAssetInput1, TextAssetInput, UrlAssetInput. Details: " + ", ".join(error_messages))
139
+ else:
140
+ return instance
141
+
142
+ def to_json(self) -> str:
143
+ """Returns the JSON representation of the actual instance"""
144
+ if self.actual_instance is None:
145
+ return "null"
146
+
147
+ if hasattr(self.actual_instance, "to_json") and callable(self.actual_instance.to_json):
148
+ return self.actual_instance.to_json()
149
+ else:
150
+ return json.dumps(self.actual_instance)
151
+
152
+ def to_dict(self) -> Optional[Union[Dict[str, Any], FileAssetInput1, MultiAssetInput1, TextAssetInput, UrlAssetInput]]:
153
+ """Returns the dict representation of the actual instance"""
154
+ if self.actual_instance is None:
155
+ return None
156
+
157
+ if hasattr(self.actual_instance, "to_dict") and callable(self.actual_instance.to_dict):
158
+ return self.actual_instance.to_dict()
159
+ else:
160
+ # primitive type
161
+ return self.actual_instance
162
+
163
+ def to_str(self) -> str:
164
+ """Returns the string representation of the actual instance"""
165
+ return pprint.pformat(self.model_dump())
166
+
167
+ from rapidata.api_client.models.multi_asset_input1 import MultiAssetInput1
168
+ # TODO: Rewrite to not use raise_errors
169
+ MultiAssetInput1AssetsInner.model_rebuild(raise_errors=False)
170
+