eval-studio-client 1.2.0a1__py3-none-any.whl → 1.2.0a3__py3-none-any.whl

eval_studio_client/api/models/v1_list_workflows_shared_with_me_response.py

@@ -0,0 +1,95 @@
+# 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_workflow import V1Workflow
+from typing import Optional, Set
+from typing_extensions import Self
+
+class V1ListWorkflowsSharedWithMeResponse(BaseModel):
+    """
+    V1ListWorkflowsSharedWithMeResponse
+    """ # noqa: E501
+    workflows: Optional[List[V1Workflow]] = Field(default=None, description="The Workflows that match the request.")
+    __properties: ClassVar[List[str]] = ["workflows"]
+
+    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 V1ListWorkflowsSharedWithMeResponse 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 workflows (list)
+        _items = []
+        if self.workflows:
+            for _item in self.workflows:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['workflows'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of V1ListWorkflowsSharedWithMeResponse from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "workflows": [V1Workflow.from_dict(_item) for _item in obj["workflows"]] if obj.get("workflows") is not None else None
+        })
+        return _obj
+
+

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,106 @@
+# 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")
+    failure_clustering_pending_jobs: Optional[StrictStr] = Field(default=None, description="Number of pending jobs in the failure clustering queue. It's marked as optional to always be part of the response, even when the value is zero.", alias="failureClusteringPendingJobs")
+    __properties: ClassVar[List[str]] = ["topicModelingPendingJobs", "testValidationPendingJobs", "failureClusteringPendingJobs"]
+
+    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
+
+        # set to None if failure_clustering_pending_jobs (nullable) is None
+        # and model_fields_set contains the field
+        if self.failure_clustering_pending_jobs is None and "failure_clustering_pending_jobs" in self.model_fields_set:
+            _dict['failureClusteringPendingJobs'] = 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"),
+            "failureClusteringPendingJobs": obj.get("failureClusteringPendingJobs")
+        })
+        return _obj
+
+

eval_studio_client/api/models/v1_workflow_node_type.py

@@ -35,6 +35,7 @@ class V1WorkflowNodeType(str, Enum):
     WORKFLOW_NODE_TYPE_MARGINAL_ANALYSIS = 'WORKFLOW_NODE_TYPE_MARGINAL_ANALYSIS'
     WORKFLOW_NODE_TYPE_VALIDATION = 'WORKFLOW_NODE_TYPE_VALIDATION'
     WORKFLOW_NODE_TYPE_ADVERSARIAL_INPUTS = 'WORKFLOW_NODE_TYPE_ADVERSARIAL_INPUTS'
+    WORKFLOW_NODE_TYPE_FAILURE_CLUSTERING = 'WORKFLOW_NODE_TYPE_FAILURE_CLUSTERING'
 
     @classmethod
     def from_json(cls, json_str: str) -> Self:

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/v1_workflow_result_report_format.py

@@ -0,0 +1,37 @@
+# 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 V1WorkflowResultReportFormat(str, Enum):
+    """
+    V1WorkflowResultReportFormat
+    """
+
+    """
+    allowed enum values
+    """
+    WORKFLOW_RESULT_REPORT_FORMAT_UNSPECIFIED = 'WORKFLOW_RESULT_REPORT_FORMAT_UNSPECIFIED'
+    WORKFLOW_RESULT_REPORT_FORMAT_HTML = 'WORKFLOW_RESULT_REPORT_FORMAT_HTML'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of V1WorkflowResultReportFormat 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_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()