eval-studio-client 0.7.0__py3-none-any.whl

eval_studio_client/api/models/v1alpha_perturbator_configuration.py

@@ -0,0 +1,92 @@
+# coding: utf-8
+
+"""
+    ai/h2o/eval_studio/v1alpha/collection.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.v1alpha_perturbator_intensity import V1alphaPerturbatorIntensity
+from typing import Optional, Set
+from typing_extensions import Self
+
+class V1alphaPerturbatorConfiguration(BaseModel):
+    """
+    PerturbatorConfiguration represents the configuration of a perturbator to use during the perturbation process.
+    """ # noqa: E501
+    name: Optional[StrictStr] = None
+    intensity: Optional[V1alphaPerturbatorIntensity] = None
+    params: Optional[StrictStr] = Field(default=None, description="Optional. The parameters to pass to the perturbator.")
+    __properties: ClassVar[List[str]] = ["name", "intensity", "params"]
+
+    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 V1alphaPerturbatorConfiguration 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 V1alphaPerturbatorConfiguration from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "name": obj.get("name"),
+            "intensity": obj.get("intensity"),
+            "params": obj.get("params")
+        })
+        return _obj
+
+

eval_studio_client/api/models/v1alpha_perturbator_intensity.py

@@ -0,0 +1,39 @@
+# coding: utf-8
+
+"""
+    ai/h2o/eval_studio/v1alpha/collection.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 V1alphaPerturbatorIntensity(str, Enum):
+    """
+    V1alphaPerturbatorIntensity
+    """
+
+    """
+    allowed enum values
+    """
+    PERTURBATOR_INTENSITY_UNSPECIFIED = 'PERTURBATOR_INTENSITY_UNSPECIFIED'
+    PERTURBATOR_INTENSITY_LOW = 'PERTURBATOR_INTENSITY_LOW'
+    PERTURBATOR_INTENSITY_MEDIUM = 'PERTURBATOR_INTENSITY_MEDIUM'
+    PERTURBATOR_INTENSITY_HIGH = 'PERTURBATOR_INTENSITY_HIGH'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of V1alphaPerturbatorIntensity from a JSON string"""
+        return cls(json.loads(json_str))
+
+

eval_studio_client/api/models/v1alpha_problem_and_action.py

@@ -0,0 +1,129 @@
+# coding: utf-8
+
+"""
+    ai/h2o/eval_studio/v1alpha/collection.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 V1alphaProblemAndAction(BaseModel):
+    """
+    V1alphaProblemAndAction
+    """ # noqa: E501
+    description: Optional[StrictStr] = Field(default=None, description="Output only. Problem description.")
+    severity: Optional[StrictStr] = Field(default=None, description="Output only. Problem severity.")
+    problem_type: Optional[StrictStr] = Field(default=None, description="Output only. Problem type.", alias="problemType")
+    problem_attrs: Optional[Dict[str, StrictStr]] = Field(default=None, description="Output only. Problem attributes.", alias="problemAttrs")
+    actions_description: Optional[StrictStr] = Field(default=None, description="Output only. Description of actions to mitigate the problem.", alias="actionsDescription")
+    explainer_id: Optional[StrictStr] = Field(default=None, description="Output only. ID of the explainer which identified the problem.", alias="explainerId")
+    explainer_name: Optional[StrictStr] = Field(default=None, description="Output only. Display name of the explainer which identified the problem.", alias="explainerName")
+    explanation_type: Optional[StrictStr] = Field(default=None, description="Output only. Type of the explanation which can clarify the problem.", alias="explanationType")
+    explanation_name: Optional[StrictStr] = Field(default=None, description="Output only. Name of the explanation which can clarify the problem.", alias="explanationName")
+    explanation_mime: Optional[StrictStr] = Field(default=None, description="Output only. Media type of the explanation which can clarify the problem.", alias="explanationMime")
+    resources: Optional[List[StrictStr]] = Field(default=None, description="Output only. Problem resources.")
+    __properties: ClassVar[List[str]] = ["description", "severity", "problemType", "problemAttrs", "actionsDescription", "explainerId", "explainerName", "explanationType", "explanationName", "explanationMime", "resources"]
+
+    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 V1alphaProblemAndAction 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([
+            "description",
+            "severity",
+            "problem_type",
+            "problem_attrs",
+            "actions_description",
+            "explainer_id",
+            "explainer_name",
+            "explanation_type",
+            "explanation_name",
+            "explanation_mime",
+            "resources",
+        ])
+
+        _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 V1alphaProblemAndAction from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "description": obj.get("description"),
+            "severity": obj.get("severity"),
+            "problemType": obj.get("problemType"),
+            "problemAttrs": obj.get("problemAttrs"),
+            "actionsDescription": obj.get("actionsDescription"),
+            "explainerId": obj.get("explainerId"),
+            "explainerName": obj.get("explainerName"),
+            "explanationType": obj.get("explanationType"),
+            "explanationName": obj.get("explanationName"),
+            "explanationMime": obj.get("explanationMime"),
+            "resources": obj.get("resources")
+        })
+        return _obj
+
+

eval_studio_client/api/models/v1alpha_test.py

@@ -0,0 +1,126 @@
+# coding: utf-8
+
+"""
+    ai/h2o/eval_studio/v1alpha/collection.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, StrictBool, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from typing import Optional, Set
+from typing_extensions import Self
+
+class V1alphaTest(BaseModel):
+    """
+    V1alphaTest
+    """ # noqa: E501
+    name: Optional[StrictStr] = None
+    create_time: Optional[datetime] = Field(default=None, description="Output only. Timestamp when the Test was created.", alias="createTime")
+    creator: Optional[StrictStr] = Field(default=None, description="Output only. Name of the user or service that requested creation of the Test.")
+    update_time: Optional[datetime] = Field(default=None, description="Output only. Optional. Timestamp when the Test was last updated.", alias="updateTime")
+    updater: Optional[StrictStr] = Field(default=None, description="Output only. Optional. Name of the user or service that requested update of the Test.")
+    delete_time: Optional[datetime] = Field(default=None, description="Output only. Optional. Set when the Test is deleted. When set, Test should be considered as deleted.", alias="deleteTime")
+    deleter: Optional[StrictStr] = Field(default=None, description="Output only. Optional. Name of the user or service that requested deletion of the Test.")
+    display_name: Optional[StrictStr] = Field(default=None, description="Human readable name of the Test.", alias="displayName")
+    description: Optional[StrictStr] = Field(default=None, description="Optional. Arbitrary description of the Test.")
+    documents: Optional[List[StrictStr]] = Field(default=None, description="Immutable. Resource names of Documents assigned to the Test.")
+    tags: Optional[List[StrictStr]] = Field(default=None, description="Tags assigned to the Test.")
+    demo: Optional[StrictBool] = Field(default=None, description="Output only. Whether the Test is a demo resource or not. Demo resources are read only.")
+    __properties: ClassVar[List[str]] = ["name", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "displayName", "description", "documents", "tags", "demo"]
+
+    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 V1alphaTest 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.
+        """
+        excluded_fields: Set[str] = set([
+            "name",
+            "create_time",
+            "creator",
+            "update_time",
+            "updater",
+            "delete_time",
+            "deleter",
+            "demo",
+        ])
+
+        _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 V1alphaTest from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "name": obj.get("name"),
+            "createTime": obj.get("createTime"),
+            "creator": obj.get("creator"),
+            "updateTime": obj.get("updateTime"),
+            "updater": obj.get("updater"),
+            "deleteTime": obj.get("deleteTime"),
+            "deleter": obj.get("deleter"),
+            "displayName": obj.get("displayName"),
+            "description": obj.get("description"),
+            "documents": obj.get("documents"),
+            "tags": obj.get("tags"),
+            "demo": obj.get("demo")
+        })
+        return _obj
+
+

eval_studio_client/api/models/v1alpha_test_case.py

@@ -0,0 +1,124 @@
+# coding: utf-8
+
+"""
+    ai/h2o/eval_studio/v1alpha/collection.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 typing import Optional, Set
+from typing_extensions import Self
+
+class V1alphaTestCase(BaseModel):
+    """
+    V1alphaTestCase
+    """ # noqa: E501
+    name: Optional[StrictStr] = None
+    create_time: Optional[datetime] = Field(default=None, description="Output only. Timestamp when the Test was created.", alias="createTime")
+    creator: Optional[StrictStr] = Field(default=None, description="Output only. Name of the user or service that requested creation of the TestCase.")
+    update_time: Optional[datetime] = Field(default=None, description="Output only. Optional. Timestamp when the TestCase was last updated.", alias="updateTime")
+    updater: Optional[StrictStr] = Field(default=None, description="Output only. Optional. Name of the user or service that requested update of the TestCase.")
+    delete_time: Optional[datetime] = Field(default=None, description="Output only. Optional. Set when the TestCase is deleted. When set TestCase should be considered as deleted.", alias="deleteTime")
+    deleter: Optional[StrictStr] = Field(default=None, description="Output only. Optional. Name of the user or service that requested deletion of the TestCase.")
+    parent: Optional[StrictStr] = Field(default=None, description="Parent Test resource name. e.g.: \"tests/<UUID>\".")
+    prompt: Optional[StrictStr] = Field(default=None, description="Prompt text. Model input.")
+    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]] = ["name", "createTime", "creator", "updateTime", "updater", "deleteTime", "deleter", "parent", "prompt", "answer", "constraints", "condition"]
+
+    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 V1alphaTestCase 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.
+        """
+        excluded_fields: Set[str] = set([
+            "name",
+            "create_time",
+            "creator",
+            "update_time",
+            "updater",
+            "delete_time",
+            "deleter",
+        ])
+
+        _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 V1alphaTestCase from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "name": obj.get("name"),
+            "createTime": obj.get("createTime"),
+            "creator": obj.get("creator"),
+            "updateTime": obj.get("updateTime"),
+            "updater": obj.get("updater"),
+            "deleteTime": obj.get("deleteTime"),
+            "deleter": obj.get("deleter"),
+            "parent": obj.get("parent"),
+            "prompt": obj.get("prompt"),
+            "answer": obj.get("answer"),
+            "constraints": obj.get("constraints"),
+            "condition": obj.get("condition")
+        })
+        return _obj
+
+

eval_studio_client/api/models/v1alpha_test_case_relationship.py

@@ -0,0 +1,91 @@
+# coding: utf-8
+
+"""
+    ai/h2o/eval_studio/v1alpha/collection.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 V1alphaTestCaseRelationship(BaseModel):
+    """
+    V1alphaTestCaseRelationship
+    """ # noqa: E501
+    source: Optional[StrictStr] = Field(default=None, description="Required. Source test case.")
+    target: Optional[StrictStr] = Field(default=None, description="Required. Target test case.")
+    type: Optional[StrictStr] = Field(default=None, description="Required. Type of the relationship.")
+    __properties: ClassVar[List[str]] = ["source", "target", "type"]
+
+    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 V1alphaTestCaseRelationship 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 V1alphaTestCaseRelationship from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "source": obj.get("source"),
+            "target": obj.get("target"),
+            "type": obj.get("type")
+        })
+        return _obj
+
+