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

eval_studio_client/api/test/test_v1_list_workflow_dependencies_response.py

@@ -0,0 +1,93 @@
+# 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.v1_list_workflow_dependencies_response import V1ListWorkflowDependenciesResponse
+
+class TestV1ListWorkflowDependenciesResponse(unittest.TestCase):
+    """V1ListWorkflowDependenciesResponse unit test stubs"""
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def make_instance(self, include_optional) -> V1ListWorkflowDependenciesResponse:
+        """Test V1ListWorkflowDependenciesResponse
+            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 `V1ListWorkflowDependenciesResponse`
+        """
+        model = V1ListWorkflowDependenciesResponse()
+        if include_optional:
+            return V1ListWorkflowDependenciesResponse(
+                nodes = [
+                    eval_studio_client.api.models.v1_workflow_node.v1WorkflowNode(
+                        name = '', 
+                        parent = '', 
+                        display_name = '', 
+                        description = '', 
+                        create_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
+                        creator = '', 
+                        update_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
+                        updater = '', 
+                        delete_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
+                        deleter = '', 
+                        type = 'WORKFLOW_NODE_TYPE_UNSPECIFIED', 
+                        parameters = eval_studio_client.api.models.parameters.parameters(), 
+                        outputs = eval_studio_client.api.models.outputs.outputs(), 
+                        output_artifacts = [
+                            eval_studio_client.api.models.v1_workflow_node_artifact.v1WorkflowNodeArtifact(
+                                name = '', 
+                                parent = '', 
+                                display_name = '', 
+                                description = '', 
+                                create_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
+                                creator = '', 
+                                update_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
+                                updater = '', 
+                                delete_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
+                                deleter = '', 
+                                mime_type = '', )
+                            ], 
+                        status = 'WORKFLOW_NODE_STATUS_UNSPECIFIED', 
+                        attributes = eval_studio_client.api.models.v1_workflow_node_attributes.v1WorkflowNodeAttributes(
+                            can_delete = True, ), 
+                        processed_by_operation = '', )
+                    ],
+                dependencies = [
+                    eval_studio_client.api.models.v1_dependency_list.v1DependencyList(
+                        node = '', 
+                        dependencies = [
+                            eval_studio_client.api.models.v1_workflow_dependency.v1WorkflowDependency(
+                                name = '', 
+                                optional = True, )
+                            ], )
+                    ]
+            )
+        else:
+            return V1ListWorkflowDependenciesResponse(
+        )
+        """
+
+    def testV1ListWorkflowDependenciesResponse(self):
+        """Test V1ListWorkflowDependenciesResponse"""
+        # inst_req_only = self.make_instance(include_optional=False)
+        # inst_req_and_optional = self.make_instance(include_optional=True)
+
+if __name__ == '__main__':
+    unittest.main()

eval_studio_client/api/test/test_v1_perturb_test_in_place_response.py

@@ -51,7 +51,8 @@ class TestV1PerturbTestInPlaceResponse(unittest.TestCase):
                     tags = [
                         ''
                         ], 
-                    demo = True, )
+                    demo = True, 
+                    type = 'TEST_TYPE_UNSPECIFIED', )
             )
         else:
             return V1PerturbTestInPlaceResponse(

eval_studio_client/api/test/test_v1_perturb_test_response.py

@@ -51,7 +51,8 @@ class TestV1PerturbTestResponse(unittest.TestCase):
                     tags = [
                         ''
                         ], 
-                    demo = True, )
+                    demo = True, 
+                    type = 'TEST_TYPE_UNSPECIFIED', )
             )
         else:
             return V1PerturbTestResponse(

eval_studio_client/api/test/test_v1_repeated_context.py

@@ -0,0 +1,62 @@
+# 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.v1_repeated_context import V1RepeatedContext
+
+class TestV1RepeatedContext(unittest.TestCase):
+    """V1RepeatedContext unit test stubs"""
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def make_instance(self, include_optional) -> V1RepeatedContext:
+        """Test V1RepeatedContext
+            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 `V1RepeatedContext`
+        """
+        model = V1RepeatedContext()
+        if include_optional:
+            return V1RepeatedContext(
+                content = [
+                    eval_studio_client.api.models.v1_context.v1Context(
+                        collection_id = '', 
+                        chunk_id = 56, 
+                        score = 1.337, 
+                        content = '', 
+                        document_id = '', 
+                        document_name = '', 
+                        pages = '', 
+                        topic = '', 
+                        topic_id = 56, )
+                    ]
+            )
+        else:
+            return V1RepeatedContext(
+        )
+        """
+
+    def testV1RepeatedContext(self):
+        """Test V1RepeatedContext"""
+        # inst_req_only = self.make_instance(include_optional=False)
+        # inst_req_and_optional = self.make_instance(include_optional=True)
+
+if __name__ == '__main__':
+    unittest.main()

eval_studio_client/api/test/test_v1_test.py

@@ -50,7 +50,8 @@ class TestV1Test(unittest.TestCase):
                 tags = [
                     ''
                     ],
-                demo = True
+                demo = True,
+                type = 'TEST_TYPE_UNSPECIFIED'
             )
         else:
             return V1Test(

eval_studio_client/api/test/test_v1_test_type.py

@@ -0,0 +1,33 @@
+# 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.v1_test_type import V1TestType
+
+class TestV1TestType(unittest.TestCase):
+    """V1TestType unit test stubs"""
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def testV1TestType(self):
+        """Test V1TestType"""
+        # inst = V1TestType()
+
+if __name__ == '__main__':
+    unittest.main()

eval_studio_client/api/test/test_v1_update_dashboard_response.py

@@ -50,7 +50,8 @@ class TestV1UpdateDashboardResponse(unittest.TestCase):
                         ''
                         ], 
                     create_operation = '', 
-                    demo = True, )
+                    demo = True, 
+                    type = 'DASHBOARD_TYPE_UNSPECIFIED', )
             )
         else:
             return V1UpdateDashboardResponse(

eval_studio_client/api/test/test_v1_update_test_response.py

@@ -51,7 +51,8 @@ class TestV1UpdateTestResponse(unittest.TestCase):
                     tags = [
                         ''
                         ], 
-                    demo = True, )
+                    demo = True, 
+                    type = 'TEST_TYPE_UNSPECIFIED', )
             )
         else:
             return V1UpdateTestResponse(

eval_studio_client/api/test/test_v1_workflow.py

@@ -69,7 +69,8 @@ class TestV1Workflow(unittest.TestCase):
                                 updater = '', 
                                 delete_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                                 deleter = '', 
-                                mime_type = '', )
+                                mime_type = '', 
+                                type = '', )
                             ], )
                     },
                 llm_model = '',

eval_studio_client/api/test/test_v1_workflow_dependency.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.v1_workflow_dependency import V1WorkflowDependency
+
+class TestV1WorkflowDependency(unittest.TestCase):
+    """V1WorkflowDependency unit test stubs"""
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    def make_instance(self, include_optional) -> V1WorkflowDependency:
+        """Test V1WorkflowDependency
+            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 `V1WorkflowDependency`
+        """
+        model = V1WorkflowDependency()
+        if include_optional:
+            return V1WorkflowDependency(
+                name = '',
+                optional = True
+            )
+        else:
+            return V1WorkflowDependency(
+        )
+        """
+
+    def testV1WorkflowDependency(self):
+        """Test V1WorkflowDependency"""
+        # inst_req_only = self.make_instance(include_optional=False)
+        # inst_req_and_optional = self.make_instance(include_optional=True)
+
+if __name__ == '__main__':
+    unittest.main()

eval_studio_client/api/test/test_v1_workflow_node.py

@@ -60,7 +60,8 @@ class TestV1WorkflowNode(unittest.TestCase):
                         updater = '', 
                         delete_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                         deleter = '', 
-                        mime_type = '', )
+                        mime_type = '', 
+                        type = '', )
                     ],
                 status = 'WORKFLOW_NODE_STATUS_UNSPECIFIED',
                 attributes = eval_studio_client.api.models.v1_workflow_node_attributes.v1WorkflowNodeAttributes(

eval_studio_client/api/test/test_v1_workflow_node_artifact.py

@@ -45,7 +45,8 @@ class TestV1WorkflowNodeArtifact(unittest.TestCase):
                 updater = '',
                 delete_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'),
                 deleter = '',
-                mime_type = ''
+                mime_type = '',
+                type = ''
             )
         else:
             return V1WorkflowNodeArtifact(

eval_studio_client/api/test/test_v1_workflow_node_artifacts.py

@@ -47,7 +47,8 @@ class TestV1WorkflowNodeArtifacts(unittest.TestCase):
                         updater = '', 
                         delete_time = datetime.datetime.strptime('2013-10-20 19:20:30.00', '%Y-%m-%d %H:%M:%S.%f'), 
                         deleter = '', 
-                        mime_type = '', )
+                        mime_type = '', 
+                        type = '', )
                     ]
             )
         else:

eval_studio_client/api/test/test_workflow_service_api.py

@@ -54,6 +54,12 @@ class TestWorkflowServiceApi(unittest.TestCase):
         """
         pass
 
+    def test_workflow_service_get_guardrails_configuration(self) -> None:
+        """Test case for workflow_service_get_guardrails_configuration
+
+        """
+        pass
+
     def test_workflow_service_get_workflow(self) -> None:
         """Test case for workflow_service_get_workflow
 
@@ -61,6 +67,13 @@ class TestWorkflowServiceApi(unittest.TestCase):
         """
         pass
 
+    def test_workflow_service_list_workflow_dependencies(self) -> None:
+        """Test case for workflow_service_list_workflow_dependencies
+
+        ListWorkflowDependencies lists workflow nodes and map of the node dependencies.
+        """
+        pass
+
     def test_workflow_service_list_workflows(self) -> None:
         """Test case for workflow_service_list_workflows
 

eval_studio_client/client.py

@@ -18,7 +18,11 @@ class Client:
     """Main Eval Studio client class."""
 
     def __init__(
-        self, host: str, token_provider: Optional[h2o_authn.TokenProvider] = None
+        self,
+        host: str,
+        token_provider: Optional[h2o_authn.TokenProvider] = None,
+        verify_ssl: bool = True,
+        ssl_ca_cert: Optional[str] = None,
     ):
         """Initialize the client connection to Eval Studio host.
 
@@ -26,9 +30,14 @@ class Client:
             host: The Eval Studio host URL.
             token_provider: Optional TokenProvider used for authentication. If not
                 provided, the client will not set any authentication headers.
+            verify: Whether to verify SSL certificates. Defaults to True.
+            ssl_ca_cert: Path to a file of concatenated CA certificates
+                in PEM format. If not provided, the default system CA certificates will be used.
         """
         _host = host.rstrip("/")
-        _api_config = api.Configuration(host=_host)
+        _api_config = api.Configuration(host=_host, ssl_ca_cert=ssl_ca_cert)
+        _api_config.verify_ssl = verify_ssl
+
         self._api_client = api.ApiClient(configuration=_api_config)
         if token_provider is not None:
             self._api_client = _TokenApiClient(_api_config, token_provider.token)