eval-studio-client 1.1.0a5__py3-none-any.whl → 1.1.0a7__py3-none-any.whl

eval_studio_client/api/models/v1_dashboard.py

@@ -21,6 +21,7 @@ from datetime import datetime
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from eval_studio_client.api.models.v1_dashboard_status import V1DashboardStatus
+from eval_studio_client.api.models.v1_dashboard_type import V1DashboardType
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -41,7 +42,8 @@ class V1Dashboard(BaseModel):
     leaderboards: Optional[List[StrictStr]] = Field(default=None, description="Immutable. Resource names of the Leaderboards used in this Dashboard.")
     create_operation: Optional[StrictStr] = Field(default=None, description="Output only. Operation resource name that created this Dashboard.", alias="createOperation")
     demo: Optional[StrictBool] = Field(default=None, description="Output only. Whether the Dashboard is a demo resource or not. Demo resources are read only.")
-    __properties: ClassVar[List[str]] = ["name", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "displayName", "description", "status", "leaderboards", "createOperation", "demo"]
+    type: Optional[V1DashboardType] = None
+    __properties: ClassVar[List[str]] = ["name", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "displayName", "description", "status", "leaderboards", "createOperation", "demo", "type"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -124,7 +126,8 @@ class V1Dashboard(BaseModel):
             "status": obj.get("status"),
             "leaderboards": obj.get("leaderboards"),
             "createOperation": obj.get("createOperation"),
-            "demo": obj.get("demo")
+            "demo": obj.get("demo"),
+            "type": obj.get("type")
         })
         return _obj
 

eval_studio_client/api/models/v1_dashboard_type.py

@@ -0,0 +1,38 @@
+# coding: utf-8
+
+"""
+    ai/h2o/eval_studio/v1/insight.proto
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: version not set
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+from enum import Enum
+from typing_extensions import Self
+
+
+class V1DashboardType(str, Enum):
+    """
+     - DASHBOARD_TYPE_UNSPECIFIED: Unspecified type.  - DASHBOARD_TYPE_STANDALONE: Standalone dashboard.  - DASHBOARD_TYPE_WORKFLOW: Dashboard is part of a dashboard group.
+    """
+
+    """
+    allowed enum values
+    """
+    DASHBOARD_TYPE_UNSPECIFIED = 'DASHBOARD_TYPE_UNSPECIFIED'
+    DASHBOARD_TYPE_STANDALONE = 'DASHBOARD_TYPE_STANDALONE'
+    DASHBOARD_TYPE_WORKFLOW = 'DASHBOARD_TYPE_WORKFLOW'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of V1DashboardType from a JSON string"""
+        return cls(json.loads(json_str))
+
+

eval_studio_client/api/models/{generated_questions_validation_service_validate_generated_questions_request.py → v1_dependency_list.py}

@@ -19,17 +19,17 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
-from eval_studio_client.api.models.v1_generated_test_case import V1GeneratedTestCase
+from eval_studio_client.api.models.v1_workflow_dependency import V1WorkflowDependency
 from typing import Optional, Set
 from typing_extensions import Self
 
-class GeneratedQuestionsValidationServiceValidateGeneratedQuestionsRequest(BaseModel):
+class V1DependencyList(BaseModel):
     """
-    GeneratedQuestionsValidationServiceValidateGeneratedQuestionsRequest
+    V1DependencyList
     """ # noqa: E501
-    operation: Optional[StrictStr] = Field(default=None, description="Required. The Operation processing this question validation process.")
-    test_cases: Optional[List[V1GeneratedTestCase]] = Field(default=None, description="Required. Generated Test Cases, i.e., Test cases with context that was used for their generation.", alias="testCases")
-    __properties: ClassVar[List[str]] = ["operation", "testCases"]
+    node: Optional[StrictStr] = Field(default=None, description="The name of the Workflow Node to retrieve dependencies for.")
+    dependencies: Optional[List[V1WorkflowDependency]] = Field(default=None, description="The names of the Workflow Node dependencies.")
+    __properties: ClassVar[List[str]] = ["node", "dependencies"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -49,7 +49,7 @@ class GeneratedQuestionsValidationServiceValidateGeneratedQuestionsRequest(BaseM
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of GeneratedQuestionsValidationServiceValidateGeneratedQuestionsRequest from a JSON string"""
+        """Create an instance of V1DependencyList from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -70,18 +70,18 @@ class GeneratedQuestionsValidationServiceValidateGeneratedQuestionsRequest(BaseM
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of each item in test_cases (list)
+        # override the default output from pydantic by calling `to_dict()` of each item in dependencies (list)
         _items = []
-        if self.test_cases:
-            for _item in self.test_cases:
+        if self.dependencies:
+            for _item in self.dependencies:
                 if _item:
                     _items.append(_item.to_dict())
-            _dict['testCases'] = _items
+            _dict['dependencies'] = _items
         return _dict
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of GeneratedQuestionsValidationServiceValidateGeneratedQuestionsRequest from a dict"""
+        """Create an instance of V1DependencyList from a dict"""
         if obj is None:
             return None
 
@@ -89,8 +89,8 @@ class GeneratedQuestionsValidationServiceValidateGeneratedQuestionsRequest(BaseM
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "operation": obj.get("operation"),
-            "testCases": [V1GeneratedTestCase.from_dict(_item) for _item in obj["testCases"]] if obj.get("testCases") is not None else None
+            "node": obj.get("node"),
+            "dependencies": [V1WorkflowDependency.from_dict(_item) for _item in obj["dependencies"]] if obj.get("dependencies") is not None else None
         })
         return _obj
 

eval_studio_client/api/models/v1_evaluator.py

@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from datetime import datetime
-from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictFloat, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional, Union
 from typing_extensions import Annotated
 from eval_studio_client.api.models.v1_evaluator_parameter import V1EvaluatorParameter
@@ -47,7 +47,9 @@ class V1Evaluator(BaseModel):
     brief_description: Optional[StrictStr] = Field(default=None, description="Optional. Short preview of the Evaluator's description.", alias="briefDescription")
     enabled: Optional[StrictBool] = Field(default=None, description="Output only. Whether this Evaluator can be used for creating evaluations. Evaluator might be disabled because it has some external requirements that are not met.")
     tagline: Optional[StrictStr] = Field(default=None, description="Output only. Tagline is a short (single-line) and high-level description of the evaluator.")
-    __properties: ClassVar[List[str]] = ["name", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "displayName", "description", "content", "mimeType", "filename", "identifier", "tags", "parameters", "briefDescription", "enabled", "tagline"]
+    primary_metric: Optional[StrictStr] = Field(default=None, description="Output only. Optional. The name of the primary metric.", alias="primaryMetric")
+    primary_metric_default_threshold: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, description="Output only. Optional. Default threshold of the primary metric. Value must be ignored if primary_metric is invalid.", alias="primaryMetricDefaultThreshold")
+    __properties: ClassVar[List[str]] = ["name", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "displayName", "description", "content", "mimeType", "filename", "identifier", "tags", "parameters", "briefDescription", "enabled", "tagline", "primaryMetric", "primaryMetricDefaultThreshold"]
 
     @field_validator('content')
     def content_validate_regular_expression(cls, value):
@@ -98,6 +100,8 @@ class V1Evaluator(BaseModel):
         * OpenAPI `readOnly` fields are excluded.
         * OpenAPI `readOnly` fields are excluded.
         * OpenAPI `readOnly` fields are excluded.
+        * OpenAPI `readOnly` fields are excluded.
+        * OpenAPI `readOnly` fields are excluded.
         """
         excluded_fields: Set[str] = set([
             "name",
@@ -109,6 +113,8 @@ class V1Evaluator(BaseModel):
             "deleter",
             "enabled",
             "tagline",
+            "primary_metric",
+            "primary_metric_default_threshold",
         ])
 
         _dict = self.model_dump(
@@ -152,7 +158,9 @@ class V1Evaluator(BaseModel):
             "parameters": [V1EvaluatorParameter.from_dict(_item) for _item in obj["parameters"]] if obj.get("parameters") is not None else None,
             "briefDescription": obj.get("briefDescription"),
             "enabled": obj.get("enabled"),
-            "tagline": obj.get("tagline")
+            "tagline": obj.get("tagline"),
+            "primaryMetric": obj.get("primaryMetric"),
+            "primaryMetricDefaultThreshold": obj.get("primaryMetricDefaultThreshold")
         })
         return _obj
 

eval_studio_client/api/models/v1_get_guardrails_configuration_response.py

@@ -0,0 +1,87 @@
+# coding: utf-8
+
+"""
+    ai/h2o/eval_studio/v1/insight.proto
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: version not set
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class V1GetGuardrailsConfigurationResponse(BaseModel):
+    """
+    V1GetGuardrailsConfigurationResponse
+    """ # noqa: E501
+    guardrails_configuration_json: Optional[StrictStr] = Field(default=None, description="The guardrails configuration for the Workflow requested in JSON format. This configuration is usable by the guardrails client as is.", alias="guardrailsConfigurationJson")
+    __properties: ClassVar[List[str]] = ["guardrailsConfigurationJson"]
+
+    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 V1GetGuardrailsConfigurationResponse 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 V1GetGuardrailsConfigurationResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "guardrailsConfigurationJson": obj.get("guardrailsConfigurationJson")
+        })
+        return _obj
+
+

eval_studio_client/api/models/v1_leaderboard_report_model.py

@@ -33,10 +33,8 @@ class V1LeaderboardReportModel(BaseModel):
     collection_name: Optional[StrictStr] = Field(default=None, description="Optional. Collection name.", alias="collectionName")
     llm_model_name: Optional[StrictStr] = Field(default=None, description="Output only. LLM model name.", alias="llmModelName")
     documents: Optional[List[StrictStr]] = Field(default=None, description="Output only. List of documents.")
-    model_cfg: Optional[Dict[str, Any]] = Field(default=None, description="Output only. Model configuration.", alias="modelCfg")
     key: Optional[StrictStr] = Field(default=None, description="Output only. Model key.")
-    llm_model_meta: Optional[Dict[str, Any]] = Field(default=None, description="Output only. LLM model metadata - recursive dictionaries with an additional info like performance stats.", alias="llmModelMeta")
-    __properties: ClassVar[List[str]] = ["connection", "modelType", "name", "collectionId", "collectionName", "llmModelName", "documents", "modelCfg", "key", "llmModelMeta"]
+    __properties: ClassVar[List[str]] = ["connection", "modelType", "name", "collectionId", "collectionName", "llmModelName", "documents", "key"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -74,8 +72,6 @@ class V1LeaderboardReportModel(BaseModel):
         * OpenAPI `readOnly` fields are excluded.
         * OpenAPI `readOnly` fields are excluded.
         * OpenAPI `readOnly` fields are excluded.
-        * OpenAPI `readOnly` fields are excluded.
-        * OpenAPI `readOnly` fields are excluded.
         """
         excluded_fields: Set[str] = set([
             "connection",
@@ -83,9 +79,7 @@ class V1LeaderboardReportModel(BaseModel):
             "name",
             "llm_model_name",
             "documents",
-            "model_cfg",
             "key",
-            "llm_model_meta",
         ])
 
         _dict = self.model_dump(
@@ -112,9 +106,7 @@ class V1LeaderboardReportModel(BaseModel):
             "collectionName": obj.get("collectionName"),
             "llmModelName": obj.get("llmModelName"),
             "documents": obj.get("documents"),
-            "modelCfg": obj.get("modelCfg"),
-            "key": obj.get("key"),
-            "llmModelMeta": obj.get("llmModelMeta")
+            "key": obj.get("key")
         })
         return _obj
 

eval_studio_client/api/models/v1_list_workflow_dependencies_response.py

@@ -0,0 +1,105 @@
+# coding: utf-8
+
+"""
+    ai/h2o/eval_studio/v1/insight.proto
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: version not set
+    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 eval_studio_client.api.models.v1_dependency_list import V1DependencyList
+from eval_studio_client.api.models.v1_workflow_node import V1WorkflowNode
+from typing import Optional, Set
+from typing_extensions import Self
+
+class V1ListWorkflowDependenciesResponse(BaseModel):
+    """
+    V1ListWorkflowDependenciesResponse
+    """ # noqa: E501
+    nodes: Optional[List[V1WorkflowNode]] = Field(default=None, description="The list of the WorkflowNodes related to requested workflow.")
+    dependencies: Optional[List[V1DependencyList]] = Field(default=None, description="The dependency map for the workflow.")
+    __properties: ClassVar[List[str]] = ["nodes", "dependencies"]
+
+    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 V1ListWorkflowDependenciesResponse 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 nodes (list)
+        _items = []
+        if self.nodes:
+            for _item in self.nodes:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['nodes'] = _items
+        # override the default output from pydantic by calling `to_dict()` of each item in dependencies (list)
+        _items = []
+        if self.dependencies:
+            for _item in self.dependencies:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['dependencies'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of V1ListWorkflowDependenciesResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "nodes": [V1WorkflowNode.from_dict(_item) for _item in obj["nodes"]] if obj.get("nodes") is not None else None,
+            "dependencies": [V1DependencyList.from_dict(_item) for _item in obj["dependencies"]] if obj.get("dependencies") is not None else None
+        })
+        return _obj
+
+

eval_studio_client/api/models/{v1_generated_test_case.py → v1_repeated_context.py}

@@ -17,20 +17,18 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, ConfigDict
 from typing import Any, ClassVar, Dict, List, Optional
 from eval_studio_client.api.models.v1_context import V1Context
-from eval_studio_client.api.models.v1_test_case import V1TestCase
 from typing import Optional, Set
 from typing_extensions import Self
 
-class V1GeneratedTestCase(BaseModel):
+class V1RepeatedContext(BaseModel):
     """
-    V1GeneratedTestCase
+    V1RepeatedContext
     """ # noqa: E501
-    test_case: Optional[V1TestCase] = Field(default=None, alias="testCase")
-    context: Optional[List[V1Context]] = Field(default=None, description="Context used for test_case generation.")
-    __properties: ClassVar[List[str]] = ["testCase", "context"]
+    content: Optional[List[V1Context]] = None
+    __properties: ClassVar[List[str]] = ["content"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -50,7 +48,7 @@ class V1GeneratedTestCase(BaseModel):
 
     @classmethod
     def from_json(cls, json_str: str) -> Optional[Self]:
-        """Create an instance of V1GeneratedTestCase from a JSON string"""
+        """Create an instance of V1RepeatedContext from a JSON string"""
         return cls.from_dict(json.loads(json_str))
 
     def to_dict(self) -> Dict[str, Any]:
@@ -71,21 +69,18 @@ class V1GeneratedTestCase(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of test_case
-        if self.test_case:
-            _dict['testCase'] = self.test_case.to_dict()
-        # override the default output from pydantic by calling `to_dict()` of each item in context (list)
+        # override the default output from pydantic by calling `to_dict()` of each item in content (list)
         _items = []
-        if self.context:
-            for _item in self.context:
+        if self.content:
+            for _item in self.content:
                 if _item:
                     _items.append(_item.to_dict())
-            _dict['context'] = _items
+            _dict['content'] = _items
         return _dict
 
     @classmethod
     def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
-        """Create an instance of V1GeneratedTestCase from a dict"""
+        """Create an instance of V1RepeatedContext from a dict"""
         if obj is None:
             return None
 
@@ -93,8 +88,7 @@ class V1GeneratedTestCase(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "testCase": V1TestCase.from_dict(obj["testCase"]) if obj.get("testCase") is not None else None,
-            "context": [V1Context.from_dict(_item) for _item in obj["context"]] if obj.get("context") is not None else None
+            "content": [V1Context.from_dict(_item) for _item in obj["content"]] if obj.get("content") is not None else None
         })
         return _obj
 

eval_studio_client/api/models/v1_test.py

@@ -20,6 +20,7 @@ import json
 from datetime import datetime
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from eval_studio_client.api.models.v1_test_type import V1TestType
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -39,7 +40,8 @@ class V1Test(BaseModel):
     documents: Optional[List[StrictStr]] = Field(default=None, description="Immutable. Resource names of Documents assigned to the Test.")
     tags: Optional[List[StrictStr]] = Field(default=None, description="Tags assigned to the Test.")
     demo: Optional[StrictBool] = Field(default=None, description="Output only. Whether the Test is a demo resource or not. Demo resources are read only.")
-    __properties: ClassVar[List[str]] = ["name", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "displayName", "description", "documents", "tags", "demo"]
+    type: Optional[V1TestType] = None
+    __properties: ClassVar[List[str]] = ["name", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "displayName", "description", "documents", "tags", "demo", "type"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -119,7 +121,8 @@ class V1Test(BaseModel):
             "description": obj.get("description"),
             "documents": obj.get("documents"),
             "tags": obj.get("tags"),
-            "demo": obj.get("demo")
+            "demo": obj.get("demo"),
+            "type": obj.get("type")
         })
         return _obj
 

eval_studio_client/api/models/v1_test_type.py

@@ -0,0 +1,38 @@
+# coding: utf-8
+
+"""
+    ai/h2o/eval_studio/v1/insight.proto
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: version not set
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+from enum import Enum
+from typing_extensions import Self
+
+
+class V1TestType(str, Enum):
+    """
+     - TEST_TYPE_UNSPECIFIED: Unspecified type.  - TEST_TYPE_STANDALONE: Standalone test.  - TEST_TYPE_WORKFLOW: Test is part of a dashboard.
+    """
+
+    """
+    allowed enum values
+    """
+    TEST_TYPE_UNSPECIFIED = 'TEST_TYPE_UNSPECIFIED'
+    TEST_TYPE_STANDALONE = 'TEST_TYPE_STANDALONE'
+    TEST_TYPE_WORKFLOW = 'TEST_TYPE_WORKFLOW'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of V1TestType from a JSON string"""
+        return cls(json.loads(json_str))
+
+

eval_studio_client/api/models/v1_workflow_dependency.py

@@ -0,0 +1,89 @@
+# coding: utf-8
+
+"""
+    ai/h2o/eval_studio/v1/insight.proto
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: version not set
+    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, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class V1WorkflowDependency(BaseModel):
+    """
+    V1WorkflowDependency
+    """ # noqa: E501
+    name: Optional[StrictStr] = Field(default=None, description="Required. The name of the Workflow Dependency node.")
+    optional: Optional[StrictBool] = Field(default=None, description="Optional. Whether the dependency is optional.")
+    __properties: ClassVar[List[str]] = ["name", "optional"]
+
+    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 V1WorkflowDependency 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 V1WorkflowDependency from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "name": obj.get("name"),
+            "optional": obj.get("optional")
+        })
+        return _obj
+
+