eval-studio-client 1.0.0a1__py3-none-any.whl → 1.1.0a5__py3-none-any.whl

eval_studio_client/api/models/required_the_operation_to_update.py

@@ -1,7 +1,7 @@
 # coding: utf-8
 
 """
-    ai/h2o/eval_studio/v1/collection.proto
+    ai/h2o/eval_studio/v1/insight.proto
 
     No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 

eval_studio_client/api/models/required_the_test_case_to_update.py

@@ -1,7 +1,7 @@
 # coding: utf-8
 
 """
-    ai/h2o/eval_studio/v1/collection.proto
+    ai/h2o/eval_studio/v1/insight.proto
 
     No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
@@ -20,6 +20,7 @@ import json
 from datetime import datetime
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from eval_studio_client.api.models.v1_test_cases_generator import V1TestCasesGenerator
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -38,7 +39,10 @@ class RequiredTheTestCaseToUpdate(BaseModel):
     answer: Optional[StrictStr] = Field(default=None, description="Expected answer text. Model output.")
     constraints: Optional[List[StrictStr]] = Field(default=None, description="Constraints on the model output.")
     condition: Optional[StrictStr] = Field(default=None, description="Optional. Test case output condition, in a form of AIP-160 compliant filter expression.")
-    __properties: ClassVar[List[str]] = ["createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "parent", "prompt", "answer", "constraints", "condition"]
+    perturbed_by: Optional[List[StrictStr]] = Field(default=None, description="Output only. The list of perturbators applied to this test case.", alias="perturbedBy")
+    topics: Optional[List[StrictStr]] = Field(default=None, description="Output only. The list of topics used to generate this test case.")
+    generator: Optional[V1TestCasesGenerator] = None
+    __properties: ClassVar[List[str]] = ["createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "parent", "prompt", "answer", "constraints", "condition", "perturbedBy", "topics", "generator"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -76,6 +80,8 @@ class RequiredTheTestCaseToUpdate(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([
             "create_time",
@@ -84,6 +90,8 @@ class RequiredTheTestCaseToUpdate(BaseModel):
             "updater",
             "delete_time",
             "deleter",
+            "perturbed_by",
+            "topics",
         ])
 
         _dict = self.model_dump(
@@ -113,7 +121,10 @@ class RequiredTheTestCaseToUpdate(BaseModel):
             "prompt": obj.get("prompt"),
             "answer": obj.get("answer"),
             "constraints": obj.get("constraints"),
-            "condition": obj.get("condition")
+            "condition": obj.get("condition"),
+            "perturbedBy": obj.get("perturbedBy"),
+            "topics": obj.get("topics"),
+            "generator": obj.get("generator")
         })
         return _obj
 

eval_studio_client/api/models/required_the_test_to_update.py

@@ -1,7 +1,7 @@
 # coding: utf-8
 
 """
-    ai/h2o/eval_studio/v1/collection.proto
+    ai/h2o/eval_studio/v1/insight.proto
 
     No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 

eval_studio_client/api/models/required_the_updated_workflow.py

@@ -0,0 +1,160 @@
+# 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 datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from eval_studio_client.api.models.v1_workflow_node_artifacts import V1WorkflowNodeArtifacts
+from eval_studio_client.api.models.v1_workflow_type import V1WorkflowType
+from typing import Optional, Set
+from typing_extensions import Self
+
+class RequiredTheUpdatedWorkflow(BaseModel):
+    """
+    RequiredTheUpdatedWorkflow
+    """ # noqa: E501
+    display_name: Optional[StrictStr] = Field(default=None, description="Human-readable name of the Workflow.", alias="displayName")
+    description: Optional[StrictStr] = Field(default=None, description="Optional description of the Workflow.")
+    create_time: Optional[datetime] = Field(default=None, description="Output only. Immutable. Creation time of the Workflow.", alias="createTime")
+    creator: Optional[StrictStr] = Field(default=None, description="Output only. Immutable. Name of the user or service that requested creation of the Workflow.")
+    update_time: Optional[datetime] = Field(default=None, description="Output only. Optional. Last update time of the Workflow.", alias="updateTime")
+    updater: Optional[StrictStr] = Field(default=None, description="Output only. Name of the user or service that requested update of the Workflow.")
+    delete_time: Optional[datetime] = Field(default=None, description="Output only. Optional. Deletion time of the Workflow.", alias="deleteTime")
+    deleter: Optional[StrictStr] = Field(default=None, description="Output only. Name of the user or service that requested deletion of the Workflow.")
+    type: Optional[V1WorkflowType] = None
+    model: Optional[StrictStr] = Field(default=None, description="Immutable. Resource name of the Model associated with this Workflow.")
+    nodes: Optional[List[StrictStr]] = Field(default=None, description="Output only. List of the WorkflowNodes in the Workflow.")
+    edges: Optional[List[StrictStr]] = Field(default=None, description="Output only. List of the WorkflowEdges in the Workflow.")
+    outputs: Optional[Dict[str, Dict[str, Any]]] = Field(default=None, description="Output only. Optional. Computed outputs of all the WorkflowNodes in the Workflow.")
+    output_artifacts: Optional[Dict[str, V1WorkflowNodeArtifacts]] = Field(default=None, description="Output only. Optional. List of the WorkflowNodeArtifacts produces by all the WorkflowNodes in the Workflow.", alias="outputArtifacts")
+    llm_model: Optional[StrictStr] = Field(default=None, description="Immutable. LLM Model to use.", alias="llmModel")
+    model_parameters: Optional[StrictStr] = Field(default=None, description="Optional. Immutable. Model parameter overrides in JSON format.", alias="modelParameters")
+    document: Optional[StrictStr] = Field(default=None, description="The resource name of a Document.")
+    h2ogpte_collection: Optional[StrictStr] = Field(default=None, description="Existing h2oGPTe collection.", alias="h2ogpteCollection")
+    cloned_from_workflow: Optional[StrictStr] = Field(default=None, description="Optional. Output only. The Workflow that this Workflow was cloned from.", alias="clonedFromWorkflow")
+    __properties: ClassVar[List[str]] = ["displayName", "description", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "type", "model", "nodes", "edges", "outputs", "outputArtifacts", "llmModel", "modelParameters", "document", "h2ogpteCollection", "clonedFromWorkflow"]
+
+    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 RequiredTheUpdatedWorkflow 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.
+        * 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.
+        * 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.
+        * OpenAPI `readOnly` fields are excluded.
+        """
+        excluded_fields: Set[str] = set([
+            "create_time",
+            "creator",
+            "update_time",
+            "updater",
+            "delete_time",
+            "deleter",
+            "nodes",
+            "edges",
+            "outputs",
+            "output_artifacts",
+            "cloned_from_workflow",
+        ])
+
+        _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 value in output_artifacts (dict)
+        _field_dict = {}
+        if self.output_artifacts:
+            for _key in self.output_artifacts:
+                if self.output_artifacts[_key]:
+                    _field_dict[_key] = self.output_artifacts[_key].to_dict()
+            _dict['outputArtifacts'] = _field_dict
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of RequiredTheUpdatedWorkflow from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "displayName": obj.get("displayName"),
+            "description": obj.get("description"),
+            "createTime": obj.get("createTime"),
+            "creator": obj.get("creator"),
+            "updateTime": obj.get("updateTime"),
+            "updater": obj.get("updater"),
+            "deleteTime": obj.get("deleteTime"),
+            "deleter": obj.get("deleter"),
+            "type": obj.get("type"),
+            "model": obj.get("model"),
+            "nodes": obj.get("nodes"),
+            "edges": obj.get("edges"),
+            "outputs": obj.get("outputs"),
+            "outputArtifacts": dict(
+                (_k, V1WorkflowNodeArtifacts.from_dict(_v))
+                for _k, _v in obj["outputArtifacts"].items()
+            )
+            if obj.get("outputArtifacts") is not None
+            else None,
+            "llmModel": obj.get("llmModel"),
+            "modelParameters": obj.get("modelParameters"),
+            "document": obj.get("document"),
+            "h2ogpteCollection": obj.get("h2ogpteCollection"),
+            "clonedFromWorkflow": obj.get("clonedFromWorkflow")
+        })
+        return _obj
+
+

eval_studio_client/api/models/required_the_updated_workflow_node.py

@@ -0,0 +1,152 @@
+# 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 datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, 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_status import V1WorkflowNodeStatus
+from eval_studio_client.api.models.v1_workflow_node_type import V1WorkflowNodeType
+from typing import Optional, Set
+from typing_extensions import Self
+
+class RequiredTheUpdatedWorkflowNode(BaseModel):
+    """
+    RequiredTheUpdatedWorkflowNode
+    """ # noqa: E501
+    parent: Optional[StrictStr] = Field(default=None, description="Output only. Immutable. Resource name of the parent Workflow in format of `workflows/{workflow_id}/`.")
+    display_name: Optional[StrictStr] = Field(default=None, description="Human-readable name of the WorkflowNode.", alias="displayName")
+    description: Optional[StrictStr] = Field(default=None, description="Optional description of the WorkflowNode.")
+    create_time: Optional[datetime] = Field(default=None, description="Output only. Immutable. Creation time of the WorkflowNode.", alias="createTime")
+    creator: Optional[StrictStr] = Field(default=None, description="Output only. Immutable. Name of the user or service that requested creation of the WorkflowNode.")
+    update_time: Optional[datetime] = Field(default=None, description="Output only. Optional. Last update time of the WorkflowNode.", alias="updateTime")
+    updater: Optional[StrictStr] = Field(default=None, description="Output only. Name of the user or service that requested update of the WorkflowNode.")
+    delete_time: Optional[datetime] = Field(default=None, description="Output only. Optional. Deletion time of the WorkflowNode.", alias="deleteTime")
+    deleter: Optional[StrictStr] = Field(default=None, description="Output only. Name of the user or service that requested deletion of the WorkflowNode.")
+    type: Optional[V1WorkflowNodeType] = None
+    parameters: Optional[Dict[str, Any]] = Field(default=None, description="User-given parameters for the WorkflowNode.")
+    outputs: Optional[Dict[str, Any]] = Field(default=None, description="Output only. Computed outputs of the WorkflowNode.")
+    output_artifacts: Optional[List[V1WorkflowNodeArtifact]] = Field(default=None, description="Output only. Optional. List of the WorkflowNodeArtifacts produces by the WorkflowNode.", alias="outputArtifacts")
+    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]] = ["parent", "displayName", "description", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "type", "parameters", "outputs", "outputArtifacts", "status", "attributes", "processedByOperation"]
+
+    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 RequiredTheUpdatedWorkflowNode 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.
+        * 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.
+        * 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([
+            "parent",
+            "create_time",
+            "creator",
+            "update_time",
+            "updater",
+            "delete_time",
+            "deleter",
+            "outputs",
+            "output_artifacts",
+            "processed_by_operation",
+        ])
+
+        _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 output_artifacts (list)
+        _items = []
+        if self.output_artifacts:
+            for _item in self.output_artifacts:
+                if _item:
+                    _items.append(_item.to_dict())
+            _dict['outputArtifacts'] = _items
+        # override the default output from pydantic by calling `to_dict()` of attributes
+        if self.attributes:
+            _dict['attributes'] = self.attributes.to_dict()
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of RequiredTheUpdatedWorkflowNode from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "parent": obj.get("parent"),
+            "displayName": obj.get("displayName"),
+            "description": obj.get("description"),
+            "createTime": obj.get("createTime"),
+            "creator": obj.get("creator"),
+            "updateTime": obj.get("updateTime"),
+            "updater": obj.get("updater"),
+            "deleteTime": obj.get("deleteTime"),
+            "deleter": obj.get("deleter"),
+            "type": obj.get("type"),
+            "parameters": obj.get("parameters"),
+            "outputs": obj.get("outputs"),
+            "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")
+        })
+        return _obj
+
+

eval_studio_client/api/models/rpc_status.py

@@ -1,7 +1,7 @@
 # coding: utf-8
 
 """
-    ai/h2o/eval_studio/v1/collection.proto
+    ai/h2o/eval_studio/v1/insight.proto
 
     No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 

eval_studio_client/api/models/test_case_service_batch_delete_test_cases_request.py

@@ -1,7 +1,7 @@
 # coding: utf-8
 
 """
-    ai/h2o/eval_studio/v1/collection.proto
+    ai/h2o/eval_studio/v1/insight.proto
 
     No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 

eval_studio_client/api/models/test_service_clone_test_request.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, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class TestServiceCloneTestRequest(BaseModel):
+    """
+    TestServiceCloneTestRequest
+    """ # noqa: E501
+    new_test_display_name: Optional[StrictStr] = Field(default=None, description="Optional. Name of the newly created test.", alias="newTestDisplayName")
+    new_test_description: Optional[StrictStr] = Field(default=None, description="Optional. Description of the newly created Test.", alias="newTestDescription")
+    __properties: ClassVar[List[str]] = ["newTestDisplayName", "newTestDescription"]
+
+    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 TestServiceCloneTestRequest 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 TestServiceCloneTestRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "newTestDisplayName": obj.get("newTestDisplayName"),
+            "newTestDescription": obj.get("newTestDescription")
+        })
+        return _obj
+
+

eval_studio_client/api/models/test_service_generate_test_cases_request.py

@@ -1,7 +1,7 @@
 # coding: utf-8
 
 """
-    ai/h2o/eval_studio/v1/collection.proto
+    ai/h2o/eval_studio/v1/insight.proto
 
     No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
 
@@ -31,8 +31,10 @@ class TestServiceGenerateTestCasesRequest(BaseModel):
     model: Optional[StrictStr] = Field(default=None, description="Optional. The Model to use for generating TestCases. If not specified, the default RAG h2oGPTe will be used. Error is returned, if no default model is specified and this field is not set.")
     base_llm_model: Optional[StrictStr] = Field(default=None, description="Optional. The base LLM model to use for generating the prompts. Selected automatically if not specified.", alias="baseLlmModel")
     generators: Optional[List[V1TestCasesGenerator]] = Field(default=None, description="Optional. Generators to use for generation. If not specified, all generators are selected.")
-    h2ogpte_collection_id: Optional[StrictStr] = Field(default=None, description="Optional. The ID of the h2oGPTe collection to use. If empty, new temporary collection will be created.", alias="h2ogpteCollectionId")
-    __properties: ClassVar[List[str]] = ["count", "model", "baseLlmModel", "generators", "h2ogpteCollectionId"]
+    h2ogpte_collection_id: Optional[StrictStr] = Field(default=None, description="Optional. ID of the h2oGPTe collection to use. If provided, documents referenced by Test and any specified chunks are ignored. This field is required if Test does not reference any documents and no chunks are provided. If this field is left empty, a temporary collection will be created.", alias="h2ogpteCollectionId")
+    topics: Optional[List[StrictStr]] = Field(default=None, description="Optional. Topics to generate questions for. If not specified, use document summarization as topic generation.")
+    chunks: Optional[List[StrictStr]] = Field(default=None, description="Optional. The list of chunks to use for generation. If set, the Documents assigned to the Test and h2ogpte_collection_id are ignored.")
+    __properties: ClassVar[List[str]] = ["count", "model", "baseLlmModel", "generators", "h2ogpteCollectionId", "topics", "chunks"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -89,7 +91,9 @@ class TestServiceGenerateTestCasesRequest(BaseModel):
             "model": obj.get("model"),
             "baseLlmModel": obj.get("baseLlmModel"),
             "generators": obj.get("generators"),
-            "h2ogpteCollectionId": obj.get("h2ogpteCollectionId")
+            "h2ogpteCollectionId": obj.get("h2ogpteCollectionId"),
+            "topics": obj.get("topics"),
+            "chunks": obj.get("chunks")
         })
         return _obj
 

eval_studio_client/api/models/test_service_import_test_cases_from_library_request.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
+
+
+from __future__ import annotations
+import pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class TestServiceImportTestCasesFromLibraryRequest(BaseModel):
+    """
+    TestServiceImportTestCasesFromLibraryRequest
+    """ # noqa: E501
+    operation: Optional[StrictStr] = Field(default=None, description="Required. The Operation processing this prompt library retrieval process.")
+    test_suite_url: Optional[StrictStr] = Field(default=None, description="Required. The URL of the library test suite to get TestCases from (sample).", alias="testSuiteUrl")
+    count: Optional[StrictInt] = Field(default=None, description="Required. The number of TestCases to get from the library.")
+    test_document_urls: Optional[List[StrictStr]] = Field(default=None, description="Optional. The list of target Test corpus document URLs to skip when returning library TestCases corpus.", alias="testDocumentUrls")
+    __properties: ClassVar[List[str]] = ["operation", "testSuiteUrl", "count", "testDocumentUrls"]
+
+    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 TestServiceImportTestCasesFromLibraryRequest 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 TestServiceImportTestCasesFromLibraryRequest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "operation": obj.get("operation"),
+            "testSuiteUrl": obj.get("testSuiteUrl"),
+            "count": obj.get("count"),
+            "testDocumentUrls": obj.get("testDocumentUrls")
+        })
+        return _obj
+
+