orca-sdk 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

orca_sdk/_generated_api_client/models/embed_request.py

@@ -1,127 +0,0 @@
-"""
-This file is generated by the openapi-python-client tool via the generate_api_client.py script
-
-It is a customized template from the openapi-python-client tool's default template:
-https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
-
-The main change is:
-- Fix typing issues
-"""
-
-# flake8: noqa: C901
-
-from typing import Any, List, Type, TypeVar, Union, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="EmbedRequest")
-
-
-@_attrs_define
-class EmbedRequest:
-    """
-    Attributes:
-        values (List[str]):
-        max_seq_length (Union[None, Unset, int]):
-    """
-
-    values: List[str]
-    max_seq_length: Union[None, Unset, int] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        values: List[str]
-        if isinstance(self.values, list):
-            values = self.values
-
-        max_seq_length: Union[None, Unset, int]
-        if isinstance(self.max_seq_length, Unset):
-            max_seq_length = UNSET
-        else:
-            max_seq_length = self.max_seq_length
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "values": values,
-            }
-        )
-        if max_seq_length is not UNSET:
-            field_dict["max_seq_length"] = max_seq_length
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-
-        def _parse_values(data: object) -> List[str]:
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                values_type_0 = cast(List[str], data)
-
-                return values_type_0
-            except:  # noqa: E722
-                pass
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                values_type_1 = cast(List[str], data)
-
-                return values_type_1
-            except:  # noqa: E722
-                pass
-            if not isinstance(data, list):
-                raise TypeError()
-            values_type_2 = []
-            _values_type_2 = data
-            for values_type_2_item_data in _values_type_2:
-
-                def _parse_values_type_2_item(data: object) -> str:
-                    return cast(str, data)
-
-                values_type_2_item = _parse_values_type_2_item(values_type_2_item_data)
-
-                values_type_2.append(values_type_2_item)
-
-            return values_type_2
-
-        values = _parse_values(d.pop("values"))
-
-        def _parse_max_seq_length(data: object) -> Union[None, Unset, int]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, int], data)
-
-        max_seq_length = _parse_max_seq_length(d.pop("max_seq_length", UNSET))
-
-        embed_request = cls(
-            values=values,
-            max_seq_length=max_seq_length,
-        )
-
-        embed_request.additional_properties = d
-        return embed_request
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Any:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Any) -> None:
-        self.additional_properties[key] = value
-
-    def __delitem__(self, key: str) -> None:
-        del self.additional_properties[key]
-
-    def __contains__(self, key: str) -> bool:
-        return key in self.additional_properties

orca_sdk/_generated_api_client/models/embedding_finetuning_method.py

@@ -1,9 +0,0 @@
-from enum import Enum
-
-
-class EmbeddingFinetuningMethod(str, Enum):
-    BATCH_TRIPLET_LOSS = "batch_triplet_loss"
-    CLASSIFICATION = "classification"
-
-    def __str__(self) -> str:
-        return str(self.value)

orca_sdk/_generated_api_client/models/evaluation_request.py

@@ -1,180 +0,0 @@
-"""
-This file is generated by the openapi-python-client tool via the generate_api_client.py script
-
-It is a customized template from the openapi-python-client tool's default template:
-https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
-
-The main change is:
-- Fix typing issues
-"""
-
-# flake8: noqa: C901
-
-from typing import Any, List, Type, TypeVar, Union, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="EvaluationRequest")
-
-
-@_attrs_define
-class EvaluationRequest:
-    """
-    Attributes:
-        datasource_label_column (str):
-        datasource_value_column (str):
-        datasource_id (Union[None, Unset, str]):
-        datasource_name (Union[None, Unset, str]):
-        memoryset_override_id (Union[None, Unset, str]):
-        record_telemetry (Union[Unset, bool]):  Default: False.
-        telemetry_tags (Union[List[str], None, Unset]):
-    """
-
-    datasource_label_column: str
-    datasource_value_column: str
-    datasource_id: Union[None, Unset, str] = UNSET
-    datasource_name: Union[None, Unset, str] = UNSET
-    memoryset_override_id: Union[None, Unset, str] = UNSET
-    record_telemetry: Union[Unset, bool] = False
-    telemetry_tags: Union[List[str], None, Unset] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        datasource_label_column = self.datasource_label_column
-
-        datasource_value_column = self.datasource_value_column
-
-        datasource_id: Union[None, Unset, str]
-        if isinstance(self.datasource_id, Unset):
-            datasource_id = UNSET
-        else:
-            datasource_id = self.datasource_id
-
-        datasource_name: Union[None, Unset, str]
-        if isinstance(self.datasource_name, Unset):
-            datasource_name = UNSET
-        else:
-            datasource_name = self.datasource_name
-
-        memoryset_override_id: Union[None, Unset, str]
-        if isinstance(self.memoryset_override_id, Unset):
-            memoryset_override_id = UNSET
-        else:
-            memoryset_override_id = self.memoryset_override_id
-
-        record_telemetry = self.record_telemetry
-
-        telemetry_tags: Union[List[str], None, Unset]
-        if isinstance(self.telemetry_tags, Unset):
-            telemetry_tags = UNSET
-        elif isinstance(self.telemetry_tags, list):
-            telemetry_tags = self.telemetry_tags
-
-        else:
-            telemetry_tags = self.telemetry_tags
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "datasource_label_column": datasource_label_column,
-                "datasource_value_column": datasource_value_column,
-            }
-        )
-        if datasource_id is not UNSET:
-            field_dict["datasource_id"] = datasource_id
-        if datasource_name is not UNSET:
-            field_dict["datasource_name"] = datasource_name
-        if memoryset_override_id is not UNSET:
-            field_dict["memoryset_override_id"] = memoryset_override_id
-        if record_telemetry is not UNSET:
-            field_dict["record_telemetry"] = record_telemetry
-        if telemetry_tags is not UNSET:
-            field_dict["telemetry_tags"] = telemetry_tags
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        datasource_label_column = d.pop("datasource_label_column")
-
-        datasource_value_column = d.pop("datasource_value_column")
-
-        def _parse_datasource_id(data: object) -> Union[None, Unset, str]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, str], data)
-
-        datasource_id = _parse_datasource_id(d.pop("datasource_id", UNSET))
-
-        def _parse_datasource_name(data: object) -> Union[None, Unset, str]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, str], data)
-
-        datasource_name = _parse_datasource_name(d.pop("datasource_name", UNSET))
-
-        def _parse_memoryset_override_id(data: object) -> Union[None, Unset, str]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, str], data)
-
-        memoryset_override_id = _parse_memoryset_override_id(d.pop("memoryset_override_id", UNSET))
-
-        record_telemetry = d.pop("record_telemetry", UNSET)
-
-        def _parse_telemetry_tags(data: object) -> Union[List[str], None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                telemetry_tags_type_0 = cast(List[str], data)
-
-                return telemetry_tags_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union[List[str], None, Unset], data)
-
-        telemetry_tags = _parse_telemetry_tags(d.pop("telemetry_tags", UNSET))
-
-        evaluation_request = cls(
-            datasource_label_column=datasource_label_column,
-            datasource_value_column=datasource_value_column,
-            datasource_id=datasource_id,
-            datasource_name=datasource_name,
-            memoryset_override_id=memoryset_override_id,
-            record_telemetry=record_telemetry,
-            telemetry_tags=telemetry_tags,
-        )
-
-        evaluation_request.additional_properties = d
-        return evaluation_request
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Any:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Any) -> None:
-        self.additional_properties[key] = value
-
-    def __delitem__(self, key: str) -> None:
-        del self.additional_properties[key]
-
-    def __contains__(self, key: str) -> bool:
-        return key in self.additional_properties

orca_sdk/_generated_api_client/models/evaluation_response.py

@@ -1,140 +0,0 @@
-"""
-This file is generated by the openapi-python-client tool via the generate_api_client.py script
-
-It is a customized template from the openapi-python-client tool's default template:
-https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
-
-The main change is:
-- Fix typing issues
-"""
-
-# flake8: noqa: C901
-
-import datetime
-from enum import Enum
-from typing import TYPE_CHECKING, Any, Dict, Type, TypeVar, Union, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-from dateutil.parser import isoparse
-
-from ..models.task_status import TaskStatus
-
-if TYPE_CHECKING:
-    from ..models.classification_evaluation_result import ClassificationEvaluationResult
-
-
-T = TypeVar("T", bound="EvaluationResponse")
-
-
-@_attrs_define
-class EvaluationResponse:
-    """
-    Attributes:
-        task_id (str):
-        org_id (str):
-        status (TaskStatus): Status of task in the task queue
-        result (Union['ClassificationEvaluationResult', None]):
-        created_at (datetime.datetime):
-        updated_at (datetime.datetime):
-    """
-
-    task_id: str
-    org_id: str
-    status: TaskStatus
-    result: Union["ClassificationEvaluationResult", None]
-    created_at: datetime.datetime
-    updated_at: datetime.datetime
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        from ..models.classification_evaluation_result import ClassificationEvaluationResult
-
-        task_id = self.task_id
-
-        org_id = self.org_id
-
-        status = self.status.value if isinstance(self.status, Enum) else self.status
-
-        result: Union[Dict[str, Any], None]
-        if isinstance(self.result, ClassificationEvaluationResult):
-            result = self.result.to_dict()
-        else:
-            result = self.result
-
-        created_at = self.created_at.isoformat()
-
-        updated_at = self.updated_at.isoformat()
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "task_id": task_id,
-                "org_id": org_id,
-                "status": status,
-                "result": result,
-                "created_at": created_at,
-                "updated_at": updated_at,
-            }
-        )
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.classification_evaluation_result import ClassificationEvaluationResult
-
-        d = src_dict.copy()
-        task_id = d.pop("task_id")
-
-        org_id = d.pop("org_id")
-
-        status = TaskStatus(d.pop("status"))
-
-        def _parse_result(data: object) -> Union["ClassificationEvaluationResult", None]:
-            if data is None:
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                result_type_0 = ClassificationEvaluationResult.from_dict(data)
-
-                return result_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["ClassificationEvaluationResult", None], data)
-
-        result = _parse_result(d.pop("result"))
-
-        created_at = isoparse(d.pop("created_at"))
-
-        updated_at = isoparse(d.pop("updated_at"))
-
-        evaluation_response = cls(
-            task_id=task_id,
-            org_id=org_id,
-            status=status,
-            result=result,
-            created_at=created_at,
-            updated_at=updated_at,
-        )
-
-        evaluation_response.additional_properties = d
-        return evaluation_response
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Any:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Any) -> None:
-        self.additional_properties[key] = value
-
-    def __delitem__(self, key: str) -> None:
-        del self.additional_properties[key]
-
-    def __contains__(self, key: str) -> bool:
-        return key in self.additional_properties

orca_sdk/_generated_api_client/models/feedback_type.py

@@ -1,9 +0,0 @@
-from enum import Enum
-
-
-class FeedbackType(str, Enum):
-    BINARY = "BINARY"
-    CONTINUOUS = "CONTINUOUS"
-
-    def __str__(self) -> str:
-        return str(self.value)

orca_sdk/_generated_api_client/models/field_validation_error.py

@@ -1,103 +0,0 @@
-"""
-This file is generated by the openapi-python-client tool via the generate_api_client.py script
-
-It is a customized template from the openapi-python-client tool's default template:
-https://github.com/openapi-generators/openapi-python-client/blob/861ef5622f10fc96d240dc9becb0edf94e61446c/openapi_python_client/templates/model.py.jinja
-
-The main change is:
-- Fix typing issues
-"""
-
-# flake8: noqa: C901
-
-from typing import Any, List, Type, TypeVar, Union, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="FieldValidationError")
-
-
-@_attrs_define
-class FieldValidationError:
-    """Validation error for a field that mirrors pydantic.ValidationError
-
-    Attributes:
-        loc (List[Union[int, str]]):
-        msg (str):
-        type (Union[Unset, str]):  Default: 'custom'.
-    """
-
-    loc: List[Union[int, str]]
-    msg: str
-    type: Union[Unset, str] = "custom"
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        loc = []
-        for loc_item_data in self.loc:
-            loc_item: Union[int, str]
-            loc_item = loc_item_data
-            loc.append(loc_item)
-
-        msg = self.msg
-
-        type = self.type
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "loc": loc,
-                "msg": msg,
-            }
-        )
-        if type is not UNSET:
-            field_dict["type"] = type
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        loc = []
-        _loc = d.pop("loc")
-        for loc_item_data in _loc:
-
-            def _parse_loc_item(data: object) -> Union[int, str]:
-                return cast(Union[int, str], data)
-
-            loc_item = _parse_loc_item(loc_item_data)
-
-            loc.append(loc_item)
-
-        msg = d.pop("msg")
-
-        type = d.pop("type", UNSET)
-
-        field_validation_error = cls(
-            loc=loc,
-            msg=msg,
-            type=type,
-        )
-
-        field_validation_error.additional_properties = d
-        return field_validation_error
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Any:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Any) -> None:
-        self.additional_properties[key] = value
-
-    def __delitem__(self, key: str) -> None:
-        del self.additional_properties[key]
-
-    def __contains__(self, key: str) -> bool:
-        return key in self.additional_properties