eval-studio-client 1.1.6a3__py3-none-any.whl → 1.2.0a2__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.
Files changed (97) hide show
  1. eval_studio_client/api/__init__.py +20 -0
  2. eval_studio_client/api/api/__init__.py +1 -0
  3. eval_studio_client/api/api/dashboard_service_api.py +1134 -73
  4. eval_studio_client/api/api/info_service_api.py +240 -0
  5. eval_studio_client/api/api/test_service_api.py +1233 -172
  6. eval_studio_client/api/api/workflow_result_service_api.py +1057 -0
  7. eval_studio_client/api/api/workflow_service_api.py +1125 -64
  8. eval_studio_client/api/docs/DashboardServiceApi.md +272 -0
  9. eval_studio_client/api/docs/InfoServiceApi.md +63 -0
  10. eval_studio_client/api/docs/RequiredTheUpdatedWorkflowNode.md +2 -0
  11. eval_studio_client/api/docs/TestServiceApi.md +268 -0
  12. eval_studio_client/api/docs/TestServiceGrantTestAccessRequest.md +30 -0
  13. eval_studio_client/api/docs/TestServiceRevokeTestAccessRequest.md +30 -0
  14. eval_studio_client/api/docs/V1GetStatsResponse.md +29 -0
  15. eval_studio_client/api/docs/V1GetWorkflowResultCorpusPatchResponse.md +29 -0
  16. eval_studio_client/api/docs/V1GetWorkflowResultReportResponse.md +29 -0
  17. eval_studio_client/api/docs/V1GetWorkflowResultSummaryResponse.md +32 -0
  18. eval_studio_client/api/docs/V1GetWorkflowResultSystemPromptPatchResponse.md +29 -0
  19. eval_studio_client/api/docs/V1Info.md +1 -0
  20. eval_studio_client/api/docs/V1ListDashboardAccessResponse.md +29 -0
  21. eval_studio_client/api/docs/V1ListDashboardsSharedWithMeResponse.md +29 -0
  22. eval_studio_client/api/docs/V1ListTestAccessResponse.md +29 -0
  23. eval_studio_client/api/docs/V1ListTestsSharedWithMeResponse.md +29 -0
  24. eval_studio_client/api/docs/V1ListWorkflowAccessResponse.md +29 -0
  25. eval_studio_client/api/docs/V1ListWorkflowsSharedWithMeResponse.md +29 -0
  26. eval_studio_client/api/docs/V1Role.md +12 -0
  27. eval_studio_client/api/docs/V1RoleBinding.md +32 -0
  28. eval_studio_client/api/docs/V1Stats.md +31 -0
  29. eval_studio_client/api/docs/V1WorkflowNode.md +2 -0
  30. eval_studio_client/api/docs/V1WorkflowNodeResultStatus.md +12 -0
  31. eval_studio_client/api/docs/V1WorkflowResultArtifactType.md +12 -0
  32. eval_studio_client/api/docs/WorkflowResultServiceApi.md +280 -0
  33. eval_studio_client/api/docs/WorkflowServiceApi.md +277 -5
  34. eval_studio_client/api/docs/WorkflowServiceRevokeWorkflowAccessRequest.md +30 -0
  35. eval_studio_client/api/models/__init__.py +19 -0
  36. eval_studio_client/api/models/required_the_updated_workflow_node.py +10 -3
  37. eval_studio_client/api/models/test_service_grant_test_access_request.py +90 -0
  38. eval_studio_client/api/models/test_service_revoke_test_access_request.py +90 -0
  39. eval_studio_client/api/models/v1_get_stats_response.py +91 -0
  40. eval_studio_client/api/models/v1_get_workflow_result_corpus_patch_response.py +87 -0
  41. eval_studio_client/api/models/v1_get_workflow_result_report_response.py +87 -0
  42. eval_studio_client/api/models/v1_get_workflow_result_summary_response.py +94 -0
  43. eval_studio_client/api/models/v1_get_workflow_result_system_prompt_patch_response.py +87 -0
  44. eval_studio_client/api/models/v1_info.py +4 -2
  45. eval_studio_client/api/models/v1_list_dashboard_access_response.py +95 -0
  46. eval_studio_client/api/models/v1_list_dashboards_shared_with_me_response.py +95 -0
  47. eval_studio_client/api/models/v1_list_test_access_response.py +95 -0
  48. eval_studio_client/api/models/v1_list_tests_shared_with_me_response.py +95 -0
  49. eval_studio_client/api/models/v1_list_workflow_access_response.py +95 -0
  50. eval_studio_client/api/models/v1_list_workflows_shared_with_me_response.py +95 -0
  51. eval_studio_client/api/models/v1_role.py +38 -0
  52. eval_studio_client/api/models/v1_role_binding.py +92 -0
  53. eval_studio_client/api/models/v1_stats.py +99 -0
  54. eval_studio_client/api/models/v1_workflow_node.py +10 -3
  55. eval_studio_client/api/models/v1_workflow_node_result_status.py +40 -0
  56. eval_studio_client/api/models/v1_workflow_result_artifact_type.py +40 -0
  57. eval_studio_client/api/models/workflow_service_revoke_workflow_access_request.py +90 -0
  58. eval_studio_client/api/test/test_dashboard_service_api.py +28 -0
  59. eval_studio_client/api/test/test_info_service_api.py +6 -0
  60. eval_studio_client/api/test/test_required_the_updated_workflow_node.py +3 -1
  61. eval_studio_client/api/test/test_test_service_api.py +24 -0
  62. eval_studio_client/api/test/test_test_service_grant_test_access_request.py +52 -0
  63. eval_studio_client/api/test/test_test_service_revoke_test_access_request.py +52 -0
  64. eval_studio_client/api/test/test_v1_batch_get_workflow_nodes_response.py +3 -1
  65. eval_studio_client/api/test/test_v1_create_workflow_node_response.py +3 -1
  66. eval_studio_client/api/test/test_v1_delete_workflow_node_response.py +3 -1
  67. eval_studio_client/api/test/test_v1_get_info_response.py +2 -1
  68. eval_studio_client/api/test/test_v1_get_stats_response.py +53 -0
  69. eval_studio_client/api/test/test_v1_get_workflow_node_response.py +3 -1
  70. eval_studio_client/api/test/test_v1_get_workflow_result_corpus_patch_response.py +51 -0
  71. eval_studio_client/api/test/test_v1_get_workflow_result_report_response.py +51 -0
  72. eval_studio_client/api/test/test_v1_get_workflow_result_summary_response.py +58 -0
  73. eval_studio_client/api/test/test_v1_get_workflow_result_system_prompt_patch_response.py +51 -0
  74. eval_studio_client/api/test/test_v1_info.py +2 -1
  75. eval_studio_client/api/test/test_v1_init_workflow_node_response.py +3 -1
  76. eval_studio_client/api/test/test_v1_list_dashboard_access_response.py +56 -0
  77. eval_studio_client/api/test/test_v1_list_dashboards_shared_with_me_response.py +69 -0
  78. eval_studio_client/api/test/test_v1_list_test_access_response.py +56 -0
  79. eval_studio_client/api/test/test_v1_list_tests_shared_with_me_response.py +70 -0
  80. eval_studio_client/api/test/test_v1_list_workflow_access_response.py +56 -0
  81. eval_studio_client/api/test/test_v1_list_workflow_dependencies_response.py +3 -1
  82. eval_studio_client/api/test/test_v1_list_workflows_shared_with_me_response.py +95 -0
  83. eval_studio_client/api/test/test_v1_reset_workflow_node_response.py +3 -1
  84. eval_studio_client/api/test/test_v1_role.py +33 -0
  85. eval_studio_client/api/test/test_v1_role_binding.py +53 -0
  86. eval_studio_client/api/test/test_v1_stats.py +52 -0
  87. eval_studio_client/api/test/test_v1_update_workflow_node_response.py +3 -1
  88. eval_studio_client/api/test/test_v1_workflow_node.py +3 -1
  89. eval_studio_client/api/test/test_v1_workflow_node_result_status.py +33 -0
  90. eval_studio_client/api/test/test_v1_workflow_result_artifact_type.py +33 -0
  91. eval_studio_client/api/test/test_workflow_result_service_api.py +59 -0
  92. eval_studio_client/api/test/test_workflow_service_api.py +28 -0
  93. eval_studio_client/api/test/test_workflow_service_revoke_workflow_access_request.py +52 -0
  94. eval_studio_client/gen/openapiv2/eval_studio.swagger.json +891 -15
  95. {eval_studio_client-1.1.6a3.dist-info → eval_studio_client-1.2.0a2.dist-info}/METADATA +1 -1
  96. {eval_studio_client-1.1.6a3.dist-info → eval_studio_client-1.2.0a2.dist-info}/RECORD +97 -37
  97. {eval_studio_client-1.1.6a3.dist-info → eval_studio_client-1.2.0a2.dist-info}/WHEEL +0 -0
@@ -33,10 +33,12 @@ from eval_studio_client.api.models.rpc_status import RpcStatus
33
33
  from eval_studio_client.api.models.test_case_service_batch_delete_test_cases_request import TestCaseServiceBatchDeleteTestCasesRequest
34
34
  from eval_studio_client.api.models.test_service_clone_test_request import TestServiceCloneTestRequest
35
35
  from eval_studio_client.api.models.test_service_generate_test_cases_request import TestServiceGenerateTestCasesRequest
36
+ from eval_studio_client.api.models.test_service_grant_test_access_request import TestServiceGrantTestAccessRequest
36
37
  from eval_studio_client.api.models.test_service_import_test_cases_from_library_request import TestServiceImportTestCasesFromLibraryRequest
37
38
  from eval_studio_client.api.models.test_service_list_test_case_library_items_request import TestServiceListTestCaseLibraryItemsRequest
38
39
  from eval_studio_client.api.models.test_service_perturb_test_in_place_request import TestServicePerturbTestInPlaceRequest
39
40
  from eval_studio_client.api.models.test_service_perturb_test_request import TestServicePerturbTestRequest
41
+ from eval_studio_client.api.models.test_service_revoke_test_access_request import TestServiceRevokeTestAccessRequest
40
42
  from eval_studio_client.api.models.v1_abort_operation_response import V1AbortOperationResponse
41
43
  from eval_studio_client.api.models.v1_batch_create_leaderboards_request import V1BatchCreateLeaderboardsRequest
42
44
  from eval_studio_client.api.models.v1_batch_create_leaderboards_response import V1BatchCreateLeaderboardsResponse
@@ -124,12 +126,17 @@ from eval_studio_client.api.models.v1_get_model_response import V1GetModelRespon
124
126
  from eval_studio_client.api.models.v1_get_operation_progress_by_parent_response import V1GetOperationProgressByParentResponse
125
127
  from eval_studio_client.api.models.v1_get_operation_response import V1GetOperationResponse
126
128
  from eval_studio_client.api.models.v1_get_perturbator_response import V1GetPerturbatorResponse
129
+ from eval_studio_client.api.models.v1_get_stats_response import V1GetStatsResponse
127
130
  from eval_studio_client.api.models.v1_get_test_case_response import V1GetTestCaseResponse
128
131
  from eval_studio_client.api.models.v1_get_test_class_response import V1GetTestClassResponse
129
132
  from eval_studio_client.api.models.v1_get_test_response import V1GetTestResponse
130
133
  from eval_studio_client.api.models.v1_get_workflow_node_prerequisites_response import V1GetWorkflowNodePrerequisitesResponse
131
134
  from eval_studio_client.api.models.v1_get_workflow_node_response import V1GetWorkflowNodeResponse
132
135
  from eval_studio_client.api.models.v1_get_workflow_response import V1GetWorkflowResponse
136
+ from eval_studio_client.api.models.v1_get_workflow_result_corpus_patch_response import V1GetWorkflowResultCorpusPatchResponse
137
+ from eval_studio_client.api.models.v1_get_workflow_result_report_response import V1GetWorkflowResultReportResponse
138
+ from eval_studio_client.api.models.v1_get_workflow_result_summary_response import V1GetWorkflowResultSummaryResponse
139
+ from eval_studio_client.api.models.v1_get_workflow_result_system_prompt_patch_response import V1GetWorkflowResultSystemPromptPatchResponse
133
140
  from eval_studio_client.api.models.v1_import_evaluation_request import V1ImportEvaluationRequest
134
141
  from eval_studio_client.api.models.v1_import_leaderboard_request import V1ImportLeaderboardRequest
135
142
  from eval_studio_client.api.models.v1_import_leaderboard_response import V1ImportLeaderboardResponse
@@ -154,7 +161,9 @@ from eval_studio_client.api.models.v1_leaderboard_status import V1LeaderboardSta
154
161
  from eval_studio_client.api.models.v1_leaderboard_type import V1LeaderboardType
155
162
  from eval_studio_client.api.models.v1_leaderboard_view import V1LeaderboardView
156
163
  from eval_studio_client.api.models.v1_list_base_models_response import V1ListBaseModelsResponse
164
+ from eval_studio_client.api.models.v1_list_dashboard_access_response import V1ListDashboardAccessResponse
157
165
  from eval_studio_client.api.models.v1_list_dashboards_response import V1ListDashboardsResponse
166
+ from eval_studio_client.api.models.v1_list_dashboards_shared_with_me_response import V1ListDashboardsSharedWithMeResponse
158
167
  from eval_studio_client.api.models.v1_list_documents_response import V1ListDocumentsResponse
159
168
  from eval_studio_client.api.models.v1_list_evaluators_response import V1ListEvaluatorsResponse
160
169
  from eval_studio_client.api.models.v1_list_llm_models_response import V1ListLLMModelsResponse
@@ -169,13 +178,17 @@ from eval_studio_client.api.models.v1_list_operations_response import V1ListOper
169
178
  from eval_studio_client.api.models.v1_list_perturbators_response import V1ListPerturbatorsResponse
170
179
  from eval_studio_client.api.models.v1_list_prompt_library_items_response import V1ListPromptLibraryItemsResponse
171
180
  from eval_studio_client.api.models.v1_list_rag_collections_response import V1ListRAGCollectionsResponse
181
+ from eval_studio_client.api.models.v1_list_test_access_response import V1ListTestAccessResponse
172
182
  from eval_studio_client.api.models.v1_list_test_case_library_items_response import V1ListTestCaseLibraryItemsResponse
173
183
  from eval_studio_client.api.models.v1_list_test_case_relationships_response import V1ListTestCaseRelationshipsResponse
174
184
  from eval_studio_client.api.models.v1_list_test_cases_response import V1ListTestCasesResponse
175
185
  from eval_studio_client.api.models.v1_list_test_classes_response import V1ListTestClassesResponse
176
186
  from eval_studio_client.api.models.v1_list_tests_response import V1ListTestsResponse
187
+ from eval_studio_client.api.models.v1_list_tests_shared_with_me_response import V1ListTestsSharedWithMeResponse
188
+ from eval_studio_client.api.models.v1_list_workflow_access_response import V1ListWorkflowAccessResponse
177
189
  from eval_studio_client.api.models.v1_list_workflow_dependencies_response import V1ListWorkflowDependenciesResponse
178
190
  from eval_studio_client.api.models.v1_list_workflows_response import V1ListWorkflowsResponse
191
+ from eval_studio_client.api.models.v1_list_workflows_shared_with_me_response import V1ListWorkflowsSharedWithMeResponse
179
192
  from eval_studio_client.api.models.v1_metric_score import V1MetricScore
180
193
  from eval_studio_client.api.models.v1_metric_scores import V1MetricScores
181
194
  from eval_studio_client.api.models.v1_model import V1Model
@@ -193,6 +206,9 @@ from eval_studio_client.api.models.v1_prompt_library_item import V1PromptLibrary
193
206
  from eval_studio_client.api.models.v1_repeated_context import V1RepeatedContext
194
207
  from eval_studio_client.api.models.v1_repeated_string import V1RepeatedString
195
208
  from eval_studio_client.api.models.v1_reset_workflow_node_response import V1ResetWorkflowNodeResponse
209
+ from eval_studio_client.api.models.v1_role import V1Role
210
+ from eval_studio_client.api.models.v1_role_binding import V1RoleBinding
211
+ from eval_studio_client.api.models.v1_stats import V1Stats
196
212
  from eval_studio_client.api.models.v1_test import V1Test
197
213
  from eval_studio_client.api.models.v1_test_case import V1TestCase
198
214
  from eval_studio_client.api.models.v1_test_case_relationship import V1TestCaseRelationship
@@ -220,8 +236,11 @@ from eval_studio_client.api.models.v1_workflow_node import V1WorkflowNode
220
236
  from eval_studio_client.api.models.v1_workflow_node_artifact import V1WorkflowNodeArtifact
221
237
  from eval_studio_client.api.models.v1_workflow_node_artifacts import V1WorkflowNodeArtifacts
222
238
  from eval_studio_client.api.models.v1_workflow_node_attributes import V1WorkflowNodeAttributes
239
+ from eval_studio_client.api.models.v1_workflow_node_result_status import V1WorkflowNodeResultStatus
223
240
  from eval_studio_client.api.models.v1_workflow_node_status import V1WorkflowNodeStatus
224
241
  from eval_studio_client.api.models.v1_workflow_node_type import V1WorkflowNodeType
225
242
  from eval_studio_client.api.models.v1_workflow_node_view import V1WorkflowNodeView
243
+ from eval_studio_client.api.models.v1_workflow_result_artifact_type import V1WorkflowResultArtifactType
226
244
  from eval_studio_client.api.models.v1_workflow_type import V1WorkflowType
227
245
  from eval_studio_client.api.models.workflow_service_clone_workflow_request import WorkflowServiceCloneWorkflowRequest
246
+ from eval_studio_client.api.models.workflow_service_revoke_workflow_access_request import WorkflowServiceRevokeWorkflowAccessRequest
@@ -18,10 +18,11 @@ import re # noqa: F401
18
18
  import json
19
19
 
20
20
  from datetime import datetime
21
- from pydantic import BaseModel, ConfigDict, Field, StrictStr
21
+ from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
22
22
  from typing import Any, ClassVar, Dict, List, Optional
23
23
  from eval_studio_client.api.models.v1_workflow_node_artifact import V1WorkflowNodeArtifact
24
24
  from eval_studio_client.api.models.v1_workflow_node_attributes import V1WorkflowNodeAttributes
25
+ from eval_studio_client.api.models.v1_workflow_node_result_status import V1WorkflowNodeResultStatus
25
26
  from eval_studio_client.api.models.v1_workflow_node_status import V1WorkflowNodeStatus
26
27
  from eval_studio_client.api.models.v1_workflow_node_type import V1WorkflowNodeType
27
28
  from typing import Optional, Set
@@ -47,7 +48,9 @@ class RequiredTheUpdatedWorkflowNode(BaseModel):
47
48
  status: Optional[V1WorkflowNodeStatus] = None
48
49
  attributes: Optional[V1WorkflowNodeAttributes] = None
49
50
  processed_by_operation: Optional[StrictStr] = Field(default=None, description="Output only. Optional. Resource name of the latest Operation that has processed or is currently processing this WorkflowNode.", alias="processedByOperation")
50
- __properties: ClassVar[List[str]] = ["parent", "displayName", "description", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "type", "parameters", "outputs", "outputArtifacts", "status", "attributes", "processedByOperation"]
51
+ result_status: Optional[V1WorkflowNodeResultStatus] = Field(default=None, alias="resultStatus")
52
+ stale: Optional[StrictBool] = Field(default=None, description="Output only. The stale field marks whether the internal result is outdated and need to be checked for validity.")
53
+ __properties: ClassVar[List[str]] = ["parent", "displayName", "description", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "type", "parameters", "outputs", "outputArtifacts", "status", "attributes", "processedByOperation", "resultStatus", "stale"]
51
54
 
52
55
  model_config = ConfigDict(
53
56
  populate_by_name=True,
@@ -89,6 +92,7 @@ class RequiredTheUpdatedWorkflowNode(BaseModel):
89
92
  * OpenAPI `readOnly` fields are excluded.
90
93
  * OpenAPI `readOnly` fields are excluded.
91
94
  * OpenAPI `readOnly` fields are excluded.
95
+ * OpenAPI `readOnly` fields are excluded.
92
96
  """
93
97
  excluded_fields: Set[str] = set([
94
98
  "parent",
@@ -101,6 +105,7 @@ class RequiredTheUpdatedWorkflowNode(BaseModel):
101
105
  "outputs",
102
106
  "output_artifacts",
103
107
  "processed_by_operation",
108
+ "stale",
104
109
  ])
105
110
 
106
111
  _dict = self.model_dump(
@@ -145,7 +150,9 @@ class RequiredTheUpdatedWorkflowNode(BaseModel):
145
150
  "outputArtifacts": [V1WorkflowNodeArtifact.from_dict(_item) for _item in obj["outputArtifacts"]] if obj.get("outputArtifacts") is not None else None,
146
151
  "status": obj.get("status"),
147
152
  "attributes": V1WorkflowNodeAttributes.from_dict(obj["attributes"]) if obj.get("attributes") is not None else None,
148
- "processedByOperation": obj.get("processedByOperation")
153
+ "processedByOperation": obj.get("processedByOperation"),
154
+ "resultStatus": obj.get("resultStatus"),
155
+ "stale": obj.get("stale")
149
156
  })
150
157
  return _obj
151
158
 
@@ -0,0 +1,90 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ ai/h2o/eval_studio/v1/insight.proto
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: version not set
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
21
+ from typing import Any, ClassVar, Dict, List, Optional
22
+ from eval_studio_client.api.models.v1_role import V1Role
23
+ from typing import Optional, Set
24
+ from typing_extensions import Self
25
+
26
+ class TestServiceGrantTestAccessRequest(BaseModel):
27
+ """
28
+ TestServiceGrantTestAccessRequest
29
+ """ # noqa: E501
30
+ subject: Optional[StrictStr] = Field(default=None, description="Required. The subject to grant access to.")
31
+ role: Optional[V1Role] = None
32
+ __properties: ClassVar[List[str]] = ["subject", "role"]
33
+
34
+ model_config = ConfigDict(
35
+ populate_by_name=True,
36
+ validate_assignment=True,
37
+ protected_namespaces=(),
38
+ )
39
+
40
+
41
+ def to_str(self) -> str:
42
+ """Returns the string representation of the model using alias"""
43
+ return pprint.pformat(self.model_dump(by_alias=True))
44
+
45
+ def to_json(self) -> str:
46
+ """Returns the JSON representation of the model using alias"""
47
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
48
+ return json.dumps(self.to_dict())
49
+
50
+ @classmethod
51
+ def from_json(cls, json_str: str) -> Optional[Self]:
52
+ """Create an instance of TestServiceGrantTestAccessRequest from a JSON string"""
53
+ return cls.from_dict(json.loads(json_str))
54
+
55
+ def to_dict(self) -> Dict[str, Any]:
56
+ """Return the dictionary representation of the model using alias.
57
+
58
+ This has the following differences from calling pydantic's
59
+ `self.model_dump(by_alias=True)`:
60
+
61
+ * `None` is only added to the output dict for nullable fields that
62
+ were set at model initialization. Other fields with value `None`
63
+ are ignored.
64
+ """
65
+ excluded_fields: Set[str] = set([
66
+ ])
67
+
68
+ _dict = self.model_dump(
69
+ by_alias=True,
70
+ exclude=excluded_fields,
71
+ exclude_none=True,
72
+ )
73
+ return _dict
74
+
75
+ @classmethod
76
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
77
+ """Create an instance of TestServiceGrantTestAccessRequest from a dict"""
78
+ if obj is None:
79
+ return None
80
+
81
+ if not isinstance(obj, dict):
82
+ return cls.model_validate(obj)
83
+
84
+ _obj = cls.model_validate({
85
+ "subject": obj.get("subject"),
86
+ "role": obj.get("role")
87
+ })
88
+ return _obj
89
+
90
+
@@ -0,0 +1,90 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ ai/h2o/eval_studio/v1/insight.proto
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: version not set
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
21
+ from typing import Any, ClassVar, Dict, List, Optional
22
+ from eval_studio_client.api.models.v1_role import V1Role
23
+ from typing import Optional, Set
24
+ from typing_extensions import Self
25
+
26
+ class TestServiceRevokeTestAccessRequest(BaseModel):
27
+ """
28
+ TestServiceRevokeTestAccessRequest
29
+ """ # noqa: E501
30
+ subject: Optional[StrictStr] = Field(default=None, description="Required. The subject to revoke access to.")
31
+ role: Optional[V1Role] = None
32
+ __properties: ClassVar[List[str]] = ["subject", "role"]
33
+
34
+ model_config = ConfigDict(
35
+ populate_by_name=True,
36
+ validate_assignment=True,
37
+ protected_namespaces=(),
38
+ )
39
+
40
+
41
+ def to_str(self) -> str:
42
+ """Returns the string representation of the model using alias"""
43
+ return pprint.pformat(self.model_dump(by_alias=True))
44
+
45
+ def to_json(self) -> str:
46
+ """Returns the JSON representation of the model using alias"""
47
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
48
+ return json.dumps(self.to_dict())
49
+
50
+ @classmethod
51
+ def from_json(cls, json_str: str) -> Optional[Self]:
52
+ """Create an instance of TestServiceRevokeTestAccessRequest from a JSON string"""
53
+ return cls.from_dict(json.loads(json_str))
54
+
55
+ def to_dict(self) -> Dict[str, Any]:
56
+ """Return the dictionary representation of the model using alias.
57
+
58
+ This has the following differences from calling pydantic's
59
+ `self.model_dump(by_alias=True)`:
60
+
61
+ * `None` is only added to the output dict for nullable fields that
62
+ were set at model initialization. Other fields with value `None`
63
+ are ignored.
64
+ """
65
+ excluded_fields: Set[str] = set([
66
+ ])
67
+
68
+ _dict = self.model_dump(
69
+ by_alias=True,
70
+ exclude=excluded_fields,
71
+ exclude_none=True,
72
+ )
73
+ return _dict
74
+
75
+ @classmethod
76
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
77
+ """Create an instance of TestServiceRevokeTestAccessRequest from a dict"""
78
+ if obj is None:
79
+ return None
80
+
81
+ if not isinstance(obj, dict):
82
+ return cls.model_validate(obj)
83
+
84
+ _obj = cls.model_validate({
85
+ "subject": obj.get("subject"),
86
+ "role": obj.get("role")
87
+ })
88
+ return _obj
89
+
90
+
@@ -0,0 +1,91 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ ai/h2o/eval_studio/v1/insight.proto
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: version not set
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
21
+ from typing import Any, ClassVar, Dict, List, Optional
22
+ from eval_studio_client.api.models.v1_stats import V1Stats
23
+ from typing import Optional, Set
24
+ from typing_extensions import Self
25
+
26
+ class V1GetStatsResponse(BaseModel):
27
+ """
28
+ V1GetStatsResponse
29
+ """ # noqa: E501
30
+ stats: Optional[V1Stats] = None
31
+ __properties: ClassVar[List[str]] = ["stats"]
32
+
33
+ model_config = ConfigDict(
34
+ populate_by_name=True,
35
+ validate_assignment=True,
36
+ protected_namespaces=(),
37
+ )
38
+
39
+
40
+ def to_str(self) -> str:
41
+ """Returns the string representation of the model using alias"""
42
+ return pprint.pformat(self.model_dump(by_alias=True))
43
+
44
+ def to_json(self) -> str:
45
+ """Returns the JSON representation of the model using alias"""
46
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
47
+ return json.dumps(self.to_dict())
48
+
49
+ @classmethod
50
+ def from_json(cls, json_str: str) -> Optional[Self]:
51
+ """Create an instance of V1GetStatsResponse from a JSON string"""
52
+ return cls.from_dict(json.loads(json_str))
53
+
54
+ def to_dict(self) -> Dict[str, Any]:
55
+ """Return the dictionary representation of the model using alias.
56
+
57
+ This has the following differences from calling pydantic's
58
+ `self.model_dump(by_alias=True)`:
59
+
60
+ * `None` is only added to the output dict for nullable fields that
61
+ were set at model initialization. Other fields with value `None`
62
+ are ignored.
63
+ """
64
+ excluded_fields: Set[str] = set([
65
+ ])
66
+
67
+ _dict = self.model_dump(
68
+ by_alias=True,
69
+ exclude=excluded_fields,
70
+ exclude_none=True,
71
+ )
72
+ # override the default output from pydantic by calling `to_dict()` of stats
73
+ if self.stats:
74
+ _dict['stats'] = self.stats.to_dict()
75
+ return _dict
76
+
77
+ @classmethod
78
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
79
+ """Create an instance of V1GetStatsResponse from a dict"""
80
+ if obj is None:
81
+ return None
82
+
83
+ if not isinstance(obj, dict):
84
+ return cls.model_validate(obj)
85
+
86
+ _obj = cls.model_validate({
87
+ "stats": V1Stats.from_dict(obj["stats"]) if obj.get("stats") is not None else None
88
+ })
89
+ return _obj
90
+
91
+
@@ -0,0 +1,87 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ ai/h2o/eval_studio/v1/insight.proto
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: version not set
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
21
+ from typing import Any, ClassVar, Dict, List, Optional
22
+ from typing import Optional, Set
23
+ from typing_extensions import Self
24
+
25
+ class V1GetWorkflowResultCorpusPatchResponse(BaseModel):
26
+ """
27
+ V1GetWorkflowResultCorpusPatchResponse
28
+ """ # noqa: E501
29
+ patch_html: Optional[StrictStr] = Field(default=None, description="The corpus patch workflow result content as a HTML string.", alias="patchHtml")
30
+ __properties: ClassVar[List[str]] = ["patchHtml"]
31
+
32
+ model_config = ConfigDict(
33
+ populate_by_name=True,
34
+ validate_assignment=True,
35
+ protected_namespaces=(),
36
+ )
37
+
38
+
39
+ def to_str(self) -> str:
40
+ """Returns the string representation of the model using alias"""
41
+ return pprint.pformat(self.model_dump(by_alias=True))
42
+
43
+ def to_json(self) -> str:
44
+ """Returns the JSON representation of the model using alias"""
45
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
46
+ return json.dumps(self.to_dict())
47
+
48
+ @classmethod
49
+ def from_json(cls, json_str: str) -> Optional[Self]:
50
+ """Create an instance of V1GetWorkflowResultCorpusPatchResponse from a JSON string"""
51
+ return cls.from_dict(json.loads(json_str))
52
+
53
+ def to_dict(self) -> Dict[str, Any]:
54
+ """Return the dictionary representation of the model using alias.
55
+
56
+ This has the following differences from calling pydantic's
57
+ `self.model_dump(by_alias=True)`:
58
+
59
+ * `None` is only added to the output dict for nullable fields that
60
+ were set at model initialization. Other fields with value `None`
61
+ are ignored.
62
+ """
63
+ excluded_fields: Set[str] = set([
64
+ ])
65
+
66
+ _dict = self.model_dump(
67
+ by_alias=True,
68
+ exclude=excluded_fields,
69
+ exclude_none=True,
70
+ )
71
+ return _dict
72
+
73
+ @classmethod
74
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
75
+ """Create an instance of V1GetWorkflowResultCorpusPatchResponse from a dict"""
76
+ if obj is None:
77
+ return None
78
+
79
+ if not isinstance(obj, dict):
80
+ return cls.model_validate(obj)
81
+
82
+ _obj = cls.model_validate({
83
+ "patchHtml": obj.get("patchHtml")
84
+ })
85
+ return _obj
86
+
87
+
@@ -0,0 +1,87 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ ai/h2o/eval_studio/v1/insight.proto
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: version not set
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
21
+ from typing import Any, ClassVar, Dict, List, Optional
22
+ from typing import Optional, Set
23
+ from typing_extensions import Self
24
+
25
+ class V1GetWorkflowResultReportResponse(BaseModel):
26
+ """
27
+ V1GetWorkflowResultReportResponse
28
+ """ # noqa: E501
29
+ report_html: Optional[StrictStr] = Field(default=None, description="The report workflow result content as a HTML string.", alias="reportHtml")
30
+ __properties: ClassVar[List[str]] = ["reportHtml"]
31
+
32
+ model_config = ConfigDict(
33
+ populate_by_name=True,
34
+ validate_assignment=True,
35
+ protected_namespaces=(),
36
+ )
37
+
38
+
39
+ def to_str(self) -> str:
40
+ """Returns the string representation of the model using alias"""
41
+ return pprint.pformat(self.model_dump(by_alias=True))
42
+
43
+ def to_json(self) -> str:
44
+ """Returns the JSON representation of the model using alias"""
45
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
46
+ return json.dumps(self.to_dict())
47
+
48
+ @classmethod
49
+ def from_json(cls, json_str: str) -> Optional[Self]:
50
+ """Create an instance of V1GetWorkflowResultReportResponse from a JSON string"""
51
+ return cls.from_dict(json.loads(json_str))
52
+
53
+ def to_dict(self) -> Dict[str, Any]:
54
+ """Return the dictionary representation of the model using alias.
55
+
56
+ This has the following differences from calling pydantic's
57
+ `self.model_dump(by_alias=True)`:
58
+
59
+ * `None` is only added to the output dict for nullable fields that
60
+ were set at model initialization. Other fields with value `None`
61
+ are ignored.
62
+ """
63
+ excluded_fields: Set[str] = set([
64
+ ])
65
+
66
+ _dict = self.model_dump(
67
+ by_alias=True,
68
+ exclude=excluded_fields,
69
+ exclude_none=True,
70
+ )
71
+ return _dict
72
+
73
+ @classmethod
74
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
75
+ """Create an instance of V1GetWorkflowResultReportResponse from a dict"""
76
+ if obj is None:
77
+ return None
78
+
79
+ if not isinstance(obj, dict):
80
+ return cls.model_validate(obj)
81
+
82
+ _obj = cls.model_validate({
83
+ "reportHtml": obj.get("reportHtml")
84
+ })
85
+ return _obj
86
+
87
+
@@ -0,0 +1,94 @@
1
+ # coding: utf-8
2
+
3
+ """
4
+ ai/h2o/eval_studio/v1/insight.proto
5
+
6
+ No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
7
+
8
+ The version of the OpenAPI document: version not set
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
21
+ from typing import Any, ClassVar, Dict, List, Optional
22
+ from eval_studio_client.api.models.v1_workflow_result_artifact_type import V1WorkflowResultArtifactType
23
+ from typing import Optional, Set
24
+ from typing_extensions import Self
25
+
26
+ class V1GetWorkflowResultSummaryResponse(BaseModel):
27
+ """
28
+ V1GetWorkflowResultSummaryResponse
29
+ """ # noqa: E501
30
+ intro: Optional[StrictStr] = Field(default=None, description="The 3x3x3 workflow summary: 3 summary sentences + 3 bullets with most serious highlights + 3 recommended actions sentences.")
31
+ bullets: Optional[List[StrictStr]] = None
32
+ actions: Optional[StrictStr] = None
33
+ artifact_types: Optional[List[V1WorkflowResultArtifactType]] = Field(default=None, alias="artifactTypes")
34
+ __properties: ClassVar[List[str]] = ["intro", "bullets", "actions", "artifactTypes"]
35
+
36
+ model_config = ConfigDict(
37
+ populate_by_name=True,
38
+ validate_assignment=True,
39
+ protected_namespaces=(),
40
+ )
41
+
42
+
43
+ def to_str(self) -> str:
44
+ """Returns the string representation of the model using alias"""
45
+ return pprint.pformat(self.model_dump(by_alias=True))
46
+
47
+ def to_json(self) -> str:
48
+ """Returns the JSON representation of the model using alias"""
49
+ # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
50
+ return json.dumps(self.to_dict())
51
+
52
+ @classmethod
53
+ def from_json(cls, json_str: str) -> Optional[Self]:
54
+ """Create an instance of V1GetWorkflowResultSummaryResponse from a JSON string"""
55
+ return cls.from_dict(json.loads(json_str))
56
+
57
+ def to_dict(self) -> Dict[str, Any]:
58
+ """Return the dictionary representation of the model using alias.
59
+
60
+ This has the following differences from calling pydantic's
61
+ `self.model_dump(by_alias=True)`:
62
+
63
+ * `None` is only added to the output dict for nullable fields that
64
+ were set at model initialization. Other fields with value `None`
65
+ are ignored.
66
+ """
67
+ excluded_fields: Set[str] = set([
68
+ ])
69
+
70
+ _dict = self.model_dump(
71
+ by_alias=True,
72
+ exclude=excluded_fields,
73
+ exclude_none=True,
74
+ )
75
+ return _dict
76
+
77
+ @classmethod
78
+ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
79
+ """Create an instance of V1GetWorkflowResultSummaryResponse from a dict"""
80
+ if obj is None:
81
+ return None
82
+
83
+ if not isinstance(obj, dict):
84
+ return cls.model_validate(obj)
85
+
86
+ _obj = cls.model_validate({
87
+ "intro": obj.get("intro"),
88
+ "bullets": obj.get("bullets"),
89
+ "actions": obj.get("actions"),
90
+ "artifactTypes": obj.get("artifactTypes")
91
+ })
92
+ return _obj
93
+
94
+