eval-studio-client 1.1.6a3__py3-none-any.whl → 1.2.0a2__py3-none-any.whl

eval_studio_client/api/models/v1_role.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 V1Role(str, Enum):
+    """
+    Roles are used to define the access level of a user to a specific resource.   - ROLE_UNSPECIFIED: Unspecified role. This is used when the role is not set or not applicable.  - ROLE_READER: Reader role. This role allows read-only access to the resource.  - ROLE_WRITER: Writer role. This role allows read and write access to the resource. This does not allow to manage access to the resource or to delete it.
+    """
+
+    """
+    allowed enum values
+    """
+    ROLE_UNSPECIFIED = 'ROLE_UNSPECIFIED'
+    ROLE_READER = 'ROLE_READER'
+    ROLE_WRITER = 'ROLE_WRITER'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of V1Role from a JSON string"""
+        return cls(json.loads(json_str))
+
+

eval_studio_client/api/models/v1_role_binding.py

@@ -0,0 +1,92 @@
+# 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 eval_studio_client.api.models.v1_role import V1Role
+from typing import Optional, Set
+from typing_extensions import Self
+
+class V1RoleBinding(BaseModel):
+    """
+    RoleBinding is used to bind a user to a specific role for a specific resource.
+    """ # noqa: E501
+    subject: Optional[StrictStr] = Field(default=None, description="The subject to which the role is assigned.")
+    role: Optional[V1Role] = None
+    resource: Optional[StrictStr] = Field(default=None, description="The resource to which the role is assigned.")
+    __properties: ClassVar[List[str]] = ["subject", "role", "resource"]
+
+    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 V1RoleBinding 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 V1RoleBinding from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "subject": obj.get("subject"),
+            "role": obj.get("role"),
+            "resource": obj.get("resource")
+        })
+        return _obj
+
+

eval_studio_client/api/models/v1_stats.py

@@ -0,0 +1,99 @@
+# 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 V1Stats(BaseModel):
+    """
+    Stats represents statistics about the Eval Studio instance, jobs and utilization.
+    """ # noqa: E501
+    topic_modeling_pending_jobs: Optional[StrictStr] = Field(default=None, description="Number of pending jobs in the topic modeling queue. It's marked as optional to always be part of the response, even when the value is zero.", alias="topicModelingPendingJobs")
+    test_validation_pending_jobs: Optional[StrictStr] = Field(default=None, description="Number of pending jobs in the test validation queue. It's marked as optional to always be part of the response, even when the value is zero.", alias="testValidationPendingJobs")
+    __properties: ClassVar[List[str]] = ["topicModelingPendingJobs", "testValidationPendingJobs"]
+
+    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 V1Stats 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,
+        )
+        # set to None if topic_modeling_pending_jobs (nullable) is None
+        # and model_fields_set contains the field
+        if self.topic_modeling_pending_jobs is None and "topic_modeling_pending_jobs" in self.model_fields_set:
+            _dict['topicModelingPendingJobs'] = None
+
+        # set to None if test_validation_pending_jobs (nullable) is None
+        # and model_fields_set contains the field
+        if self.test_validation_pending_jobs is None and "test_validation_pending_jobs" in self.model_fields_set:
+            _dict['testValidationPendingJobs'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of V1Stats from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "topicModelingPendingJobs": obj.get("topicModelingPendingJobs"),
+            "testValidationPendingJobs": obj.get("testValidationPendingJobs")
+        })
+        return _obj
+
+

eval_studio_client/api/models/v1_workflow_node.py

@@ -18,10 +18,11 @@ import re  # noqa: F401
 import json
 
 from datetime import datetime
-from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from eval_studio_client.api.models.v1_workflow_node_artifact import V1WorkflowNodeArtifact
 from eval_studio_client.api.models.v1_workflow_node_attributes import V1WorkflowNodeAttributes
+from eval_studio_client.api.models.v1_workflow_node_result_status import V1WorkflowNodeResultStatus
 from eval_studio_client.api.models.v1_workflow_node_status import V1WorkflowNodeStatus
 from eval_studio_client.api.models.v1_workflow_node_type import V1WorkflowNodeType
 from typing import Optional, Set
@@ -48,7 +49,9 @@ class V1WorkflowNode(BaseModel):
     status: Optional[V1WorkflowNodeStatus] = None
     attributes: Optional[V1WorkflowNodeAttributes] = None
     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")
-    __properties: ClassVar[List[str]] = ["name", "parent", "displayName", "description", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "type", "parameters", "outputs", "outputArtifacts", "status", "attributes", "processedByOperation"]
+    result_status: Optional[V1WorkflowNodeResultStatus] = Field(default=None, alias="resultStatus")
+    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.")
+    __properties: ClassVar[List[str]] = ["name", "parent", "displayName", "description", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "type", "parameters", "outputs", "outputArtifacts", "status", "attributes", "processedByOperation", "resultStatus", "stale"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -91,6 +94,7 @@ class V1WorkflowNode(BaseModel):
         * 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",
@@ -104,6 +108,7 @@ class V1WorkflowNode(BaseModel):
             "outputs",
             "output_artifacts",
             "processed_by_operation",
+            "stale",
         ])
 
         _dict = self.model_dump(
@@ -149,7 +154,9 @@ class V1WorkflowNode(BaseModel):
             "outputArtifacts": [V1WorkflowNodeArtifact.from_dict(_item) for _item in obj["outputArtifacts"]] if obj.get("outputArtifacts") is not None else None,
             "status": obj.get("status"),
             "attributes": V1WorkflowNodeAttributes.from_dict(obj["attributes"]) if obj.get("attributes") is not None else None,
-            "processedByOperation": obj.get("processedByOperation")
+            "processedByOperation": obj.get("processedByOperation"),
+            "resultStatus": obj.get("resultStatus"),
+            "stale": obj.get("stale")
         })
         return _obj
 

eval_studio_client/api/models/v1_workflow_node_result_status.py

@@ -0,0 +1,40 @@
+# 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 V1WorkflowNodeResultStatus(str, Enum):
+    """
+    WorkflowNodeResultStatus represents the result status of the finished computation of the node.   - WORKFLOW_NODE_RESULT_STATUS_UNSPECIFIED: Unspecified status.  - WORKFLOW_NODE_RESULT_STATUS_NONE: The process has a no result yet (no computation was done).  - WORKFLOW_NODE_RESULT_STATUS_SUCCESS: The process succeeded with a valid computation.  - WORKFLOW_NODE_RESULT_STATUS_WARNING: The process ends up with a result and warning.  - WORKFLOW_NODE_RESULT_STATUS_ERROR: The process failed and no result was produced.
+    """
+
+    """
+    allowed enum values
+    """
+    WORKFLOW_NODE_RESULT_STATUS_UNSPECIFIED = 'WORKFLOW_NODE_RESULT_STATUS_UNSPECIFIED'
+    WORKFLOW_NODE_RESULT_STATUS_NONE = 'WORKFLOW_NODE_RESULT_STATUS_NONE'
+    WORKFLOW_NODE_RESULT_STATUS_SUCCESS = 'WORKFLOW_NODE_RESULT_STATUS_SUCCESS'
+    WORKFLOW_NODE_RESULT_STATUS_WARNING = 'WORKFLOW_NODE_RESULT_STATUS_WARNING'
+    WORKFLOW_NODE_RESULT_STATUS_ERROR = 'WORKFLOW_NODE_RESULT_STATUS_ERROR'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of V1WorkflowNodeResultStatus from a JSON string"""
+        return cls(json.loads(json_str))
+
+

eval_studio_client/api/models/v1_workflow_result_artifact_type.py

@@ -0,0 +1,40 @@
+# 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 V1WorkflowResultArtifactType(str, Enum):
+    """
+    WorkflowResultArtifactType enum representing the types of artifacts that can be produced by a Workflow result.
+    """
+
+    """
+    allowed enum values
+    """
+    WORKFLOW_RESULT_ARTIFACT_TYPE_UNSPECIFIED = 'WORKFLOW_RESULT_ARTIFACT_TYPE_UNSPECIFIED'
+    WORKFLOW_RESULT_ARTIFACT_TYPE_REPORT = 'WORKFLOW_RESULT_ARTIFACT_TYPE_REPORT'
+    WORKFLOW_RESULT_ARTIFACT_TYPE_CORPUS_PATCH = 'WORKFLOW_RESULT_ARTIFACT_TYPE_CORPUS_PATCH'
+    WORKFLOW_RESULT_ARTIFACT_TYPE_SYSTEM_PROMPT_PATCH = 'WORKFLOW_RESULT_ARTIFACT_TYPE_SYSTEM_PROMPT_PATCH'
+    WORKFLOW_RESULT_ARTIFACT_TYPE_GUARDRAIL_CONFIG = 'WORKFLOW_RESULT_ARTIFACT_TYPE_GUARDRAIL_CONFIG'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of V1WorkflowResultArtifactType from a JSON string"""
+        return cls(json.loads(json_str))
+
+

eval_studio_client/api/models/workflow_service_revoke_workflow_access_request.py

@@ -0,0 +1,90 @@
+# 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 eval_studio_client.api.models.v1_role import V1Role
+from typing import Optional, Set
+from typing_extensions import Self
+
+class WorkflowServiceRevokeWorkflowAccessRequest(BaseModel):
+    """
+    WorkflowServiceRevokeWorkflowAccessRequest
+    """ # noqa: E501
+    subject: Optional[StrictStr] = Field(default=None, description="Required. The subject to revoke access from.")
+    role: Optional[V1Role] = None
+    __properties: ClassVar[List[str]] = ["subject", "role"]
+
+    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 WorkflowServiceRevokeWorkflowAccessRequest 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 WorkflowServiceRevokeWorkflowAccessRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "subject": obj.get("subject"),
+            "role": obj.get("role")
+        })
+        return _obj
+
+

eval_studio_client/api/test/test_dashboard_service_api.py

@@ -56,18 +56,46 @@ class TestDashboardServiceApi(unittest.TestCase):
         """
         pass
 
+    def test_dashboard_service_grant_dashboard_access(self) -> None:
+        """Test case for dashboard_service_grant_dashboard_access
+
+        GrantDashboardAccess grants access to a Dashboard to a subject with a specified role.
+        """
+        pass
+
+    def test_dashboard_service_list_dashboard_access(self) -> None:
+        """Test case for dashboard_service_list_dashboard_access
+
+        ListDashboardAccess lists access to a Dashboard.
+        """
+        pass
+
     def test_dashboard_service_list_dashboards(self) -> None:
         """Test case for dashboard_service_list_dashboards
 
         """
         pass
 
+    def test_dashboard_service_list_dashboards_shared_with_me(self) -> None:
+        """Test case for dashboard_service_list_dashboards_shared_with_me
+
+        ListDashboardsSharedWithMe lists Dashboards shared with the authenticated user.
+        """
+        pass
+
     def test_dashboard_service_list_most_recent_dashboards(self) -> None:
         """Test case for dashboard_service_list_most_recent_dashboards
 
         """
         pass
 
+    def test_dashboard_service_revoke_dashboard_access(self) -> None:
+        """Test case for dashboard_service_revoke_dashboard_access
+
+        RevokeDashboardAccess revokes access to a Dashboard from a subject.
+        """
+        pass
+
     def test_dashboard_service_update_dashboard(self) -> None:
         """Test case for dashboard_service_update_dashboard
 

eval_studio_client/api/test/test_info_service_api.py

@@ -32,6 +32,12 @@ class TestInfoServiceApi(unittest.TestCase):
         """
         pass
 
+    def test_info_service_get_stats(self) -> None:
+        """Test case for info_service_get_stats
+
+        """
+        pass
+
 
 if __name__ == '__main__':
     unittest.main()

eval_studio_client/api/test/test_required_the_updated_workflow_node.py

@@ -65,7 +65,9 @@ class TestRequiredTheUpdatedWorkflowNode(unittest.TestCase):
                 status = 'WORKFLOW_NODE_STATUS_UNSPECIFIED',
                 attributes = eval_studio_client.api.models.v1_workflow_node_attributes.v1WorkflowNodeAttributes(
                     can_delete = True, ),
-                processed_by_operation = ''
+                processed_by_operation = '',
+                result_status = 'WORKFLOW_NODE_RESULT_STATUS_UNSPECIFIED',
+                stale = True
             )
         else:
             return RequiredTheUpdatedWorkflowNode(

eval_studio_client/api/test/test_test_service_api.py

@@ -74,6 +74,12 @@ class TestTestServiceApi(unittest.TestCase):
         """
         pass
 
+    def test_test_service_grant_test_access(self) -> None:
+        """Test case for test_service_grant_test_access
+
+        """
+        pass
+
     def test_test_service_import_test_cases_from_library(self) -> None:
         """Test case for test_service_import_test_cases_from_library
 
@@ -86,6 +92,12 @@ class TestTestServiceApi(unittest.TestCase):
         """
         pass
 
+    def test_test_service_list_test_access(self) -> None:
+        """Test case for test_service_list_test_access
+
+        """
+        pass
+
     def test_test_service_list_test_case_library_items(self) -> None:
         """Test case for test_service_list_test_case_library_items
 
@@ -98,6 +110,12 @@ class TestTestServiceApi(unittest.TestCase):
         """
         pass
 
+    def test_test_service_list_tests_shared_with_me(self) -> None:
+        """Test case for test_service_list_tests_shared_with_me
+
+        """
+        pass
+
     def test_test_service_perturb_test(self) -> None:
         """Test case for test_service_perturb_test
 
@@ -110,6 +128,12 @@ class TestTestServiceApi(unittest.TestCase):
         """
         pass
 
+    def test_test_service_revoke_test_access(self) -> None:
+        """Test case for test_service_revoke_test_access
+
+        """
+        pass
+
     def test_test_service_update_test(self) -> None:
         """Test case for test_service_update_test
 

eval_studio_client/api/test/test_test_service_grant_test_access_request.py

@@ -0,0 +1,52 @@
+# 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
+
+
+import unittest
+
+from eval_studio_client.api.models.test_service_grant_test_access_request import TestServiceGrantTestAccessRequest
+
+class TestTestServiceGrantTestAccessRequest(unittest.TestCase):
+    """TestServiceGrantTestAccessRequest unit test stubs"""
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def make_instance(self, include_optional) -> TestServiceGrantTestAccessRequest:
+        """Test TestServiceGrantTestAccessRequest
+            include_option is a boolean, when False only required
+            params are included, when True both required and
+            optional params are included """
+        # uncomment below to create an instance of `TestServiceGrantTestAccessRequest`
+        """
+        model = TestServiceGrantTestAccessRequest()
+        if include_optional:
+            return TestServiceGrantTestAccessRequest(
+                subject = '',
+                role = 'ROLE_UNSPECIFIED'
+            )
+        else:
+            return TestServiceGrantTestAccessRequest(
+        )
+        """
+
+    def testTestServiceGrantTestAccessRequest(self):
+        """Test TestServiceGrantTestAccessRequest"""
+        # inst_req_only = self.make_instance(include_optional=False)
+        # inst_req_and_optional = self.make_instance(include_optional=True)
+
+if __name__ == '__main__':
+    unittest.main()