orca-sdk 0.1.0__py3-none-any.whl

orca_sdk/_generated_api_client/models/classification_evaluation_result.py

@@ -0,0 +1,114 @@
+"""
+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, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ClassificationEvaluationResult")
+
+
+@_attrs_define
+class ClassificationEvaluationResult:
+    """
+    Attributes:
+        f1_score (float):
+        roc_auc (Union[None, float]):
+        pr_auc (Union[None, float]):
+        accuracy (float):
+        loss (float):
+    """
+
+    f1_score: float
+    roc_auc: Union[None, float]
+    pr_auc: Union[None, float]
+    accuracy: float
+    loss: float
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        f1_score = self.f1_score
+
+        roc_auc: Union[None, float]
+        roc_auc = self.roc_auc
+
+        pr_auc: Union[None, float]
+        pr_auc = self.pr_auc
+
+        accuracy = self.accuracy
+
+        loss = self.loss
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "f1_score": f1_score,
+                "roc_auc": roc_auc,
+                "pr_auc": pr_auc,
+                "accuracy": accuracy,
+                "loss": loss,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        f1_score = d.pop("f1_score")
+
+        def _parse_roc_auc(data: object) -> Union[None, float]:
+            if data is None:
+                return data
+            return cast(Union[None, float], data)
+
+        roc_auc = _parse_roc_auc(d.pop("roc_auc"))
+
+        def _parse_pr_auc(data: object) -> Union[None, float]:
+            if data is None:
+                return data
+            return cast(Union[None, float], data)
+
+        pr_auc = _parse_pr_auc(d.pop("pr_auc"))
+
+        accuracy = d.pop("accuracy")
+
+        loss = d.pop("loss")
+
+        classification_evaluation_result = cls(
+            f1_score=f1_score,
+            roc_auc=roc_auc,
+            pr_auc=pr_auc,
+            accuracy=accuracy,
+            loss=loss,
+        )
+
+        classification_evaluation_result.additional_properties = d
+        return classification_evaluation_result
+
+    @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/clone_labeled_memoryset_request.py

@@ -0,0 +1,150 @@
+"""
+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 enum import Enum
+from typing import Any, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.pretrained_embedding_model_name import PretrainedEmbeddingModelName
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="CloneLabeledMemorysetRequest")
+
+
+@_attrs_define
+class CloneLabeledMemorysetRequest:
+    """
+    Attributes:
+        name (str):
+        pretrained_embedding_model_name (Union[None, PretrainedEmbeddingModelName, Unset]):
+        finetuned_embedding_model_id (Union[None, Unset, str]):
+        max_seq_length_override (Union[None, Unset, int]):
+    """
+
+    name: str
+    pretrained_embedding_model_name: Union[None, PretrainedEmbeddingModelName, Unset] = UNSET
+    finetuned_embedding_model_id: Union[None, Unset, str] = UNSET
+    max_seq_length_override: Union[None, Unset, int] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        name = self.name
+
+        pretrained_embedding_model_name: Union[None, Unset, str]
+        if isinstance(self.pretrained_embedding_model_name, Unset):
+            pretrained_embedding_model_name = UNSET
+        elif isinstance(self.pretrained_embedding_model_name, PretrainedEmbeddingModelName):
+            pretrained_embedding_model_name = (
+                self.pretrained_embedding_model_name.value
+                if isinstance(self.pretrained_embedding_model_name, Enum)
+                else self.pretrained_embedding_model_name
+            )
+        else:
+            pretrained_embedding_model_name = self.pretrained_embedding_model_name
+
+        finetuned_embedding_model_id: Union[None, Unset, str]
+        if isinstance(self.finetuned_embedding_model_id, Unset):
+            finetuned_embedding_model_id = UNSET
+        else:
+            finetuned_embedding_model_id = self.finetuned_embedding_model_id
+
+        max_seq_length_override: Union[None, Unset, int]
+        if isinstance(self.max_seq_length_override, Unset):
+            max_seq_length_override = UNSET
+        else:
+            max_seq_length_override = self.max_seq_length_override
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "name": name,
+            }
+        )
+        if pretrained_embedding_model_name is not UNSET:
+            field_dict["pretrained_embedding_model_name"] = pretrained_embedding_model_name
+        if finetuned_embedding_model_id is not UNSET:
+            field_dict["finetuned_embedding_model_id"] = finetuned_embedding_model_id
+        if max_seq_length_override is not UNSET:
+            field_dict["max_seq_length_override"] = max_seq_length_override
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        name = d.pop("name")
+
+        def _parse_pretrained_embedding_model_name(data: object) -> Union[None, PretrainedEmbeddingModelName, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                pretrained_embedding_model_name_type_0 = PretrainedEmbeddingModelName(data)
+
+                return pretrained_embedding_model_name_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, PretrainedEmbeddingModelName, Unset], data)
+
+        pretrained_embedding_model_name = _parse_pretrained_embedding_model_name(
+            d.pop("pretrained_embedding_model_name", UNSET)
+        )
+
+        def _parse_finetuned_embedding_model_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)
+
+        finetuned_embedding_model_id = _parse_finetuned_embedding_model_id(d.pop("finetuned_embedding_model_id", UNSET))
+
+        def _parse_max_seq_length_override(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_override = _parse_max_seq_length_override(d.pop("max_seq_length_override", UNSET))
+
+        clone_labeled_memoryset_request = cls(
+            name=name,
+            pretrained_embedding_model_name=pretrained_embedding_model_name,
+            finetuned_embedding_model_id=finetuned_embedding_model_id,
+            max_seq_length_override=max_seq_length_override,
+        )
+
+        clone_labeled_memoryset_request.additional_properties = d
+        return clone_labeled_memoryset_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/column_info.py

@@ -0,0 +1,114 @@
+"""
+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 enum import Enum
+from typing import Any, List, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.column_type import ColumnType
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="ColumnInfo")
+
+
+@_attrs_define
+class ColumnInfo:
+    """Information about a column in a datasource
+
+    Attributes:
+        name (str):
+        type (ColumnType): The type of a column in a datasource
+        enum_options (Union[List[str], None, Unset]):
+    """
+
+    name: str
+    type: ColumnType
+    enum_options: Union[List[str], None, Unset] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        name = self.name
+
+        type = self.type.value if isinstance(self.type, Enum) else self.type
+
+        enum_options: Union[List[str], None, Unset]
+        if isinstance(self.enum_options, Unset):
+            enum_options = UNSET
+        elif isinstance(self.enum_options, list):
+            enum_options = self.enum_options
+
+        else:
+            enum_options = self.enum_options
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "name": name,
+                "type": type,
+            }
+        )
+        if enum_options is not UNSET:
+            field_dict["enum_options"] = enum_options
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        name = d.pop("name")
+
+        type = ColumnType(d.pop("type"))
+
+        def _parse_enum_options(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()
+                enum_options_type_0 = cast(List[str], data)
+
+                return enum_options_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[List[str], None, Unset], data)
+
+        enum_options = _parse_enum_options(d.pop("enum_options", UNSET))
+
+        column_info = cls(
+            name=name,
+            type=type,
+            enum_options=enum_options,
+        )
+
+        column_info.additional_properties = d
+        return column_info
+
+    @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/column_type.py

@@ -0,0 +1,14 @@
+from enum import Enum
+
+
+class ColumnType(str, Enum):
+    BOOL = "BOOL"
+    ENUM = "ENUM"
+    FLOAT = "FLOAT"
+    IMAGE = "IMAGE"
+    INT = "INT"
+    OTHER = "OTHER"
+    STRING = "STRING"
+
+    def __str__(self) -> str:
+        return str(self.value)

orca_sdk/_generated_api_client/models/conflict_error_response.py

@@ -0,0 +1,80 @@
+"""
+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, Literal, Type, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ConflictErrorResponse")
+
+
+@_attrs_define
+class ConflictErrorResponse:
+    """
+    Attributes:
+        status_code (Literal[409]):
+        constraint (str):
+    """
+
+    status_code: Literal[409]
+    constraint: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        status_code = self.status_code
+
+        constraint = self.constraint
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "status_code": status_code,
+                "constraint": constraint,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        status_code = cast(Literal[409], d.pop("status_code"))
+        if status_code != 409:
+            raise ValueError(f"status_code must match const 409, got '{status_code}'")
+
+        constraint = d.pop("constraint")
+
+        conflict_error_response = cls(
+            status_code=status_code,
+            constraint=constraint,
+        )
+
+        conflict_error_response.additional_properties = d
+        return conflict_error_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/create_api_key_request.py

@@ -0,0 +1,99 @@
+"""
+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, 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="CreateApiKeyRequest")
+
+
+@_attrs_define
+class CreateApiKeyRequest:
+    """
+    Attributes:
+        id (Union[Unset, str]):
+        name (Union[Unset, str]):  Default: 'default'.
+        created_by (Union[None, Unset, str]):
+    """
+
+    id: Union[Unset, str] = UNSET
+    name: Union[Unset, str] = "default"
+    created_by: Union[None, Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = self.id
+
+        name = self.name
+
+        created_by: Union[None, Unset, str]
+        if isinstance(self.created_by, Unset):
+            created_by = UNSET
+        else:
+            created_by = self.created_by
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if id is not UNSET:
+            field_dict["id"] = id
+        if name is not UNSET:
+            field_dict["name"] = name
+        if created_by is not UNSET:
+            field_dict["created_by"] = created_by
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        id = d.pop("id", UNSET)
+
+        name = d.pop("name", UNSET)
+
+        def _parse_created_by(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)
+
+        created_by = _parse_created_by(d.pop("created_by", UNSET))
+
+        create_api_key_request = cls(
+            id=id,
+            name=name,
+            created_by=created_by,
+        )
+
+        create_api_key_request.additional_properties = d
+        return create_api_key_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/create_api_key_response.py

@@ -0,0 +1,126 @@
+"""
+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 typing import Any, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+T = TypeVar("T", bound="CreateApiKeyResponse")
+
+
+@_attrs_define
+class CreateApiKeyResponse:
+    """
+    Attributes:
+        id (str):
+        org_id (str):
+        name (str):
+        created_by (Union[None, str]):
+        created_at (datetime.datetime):
+        updated_at (datetime.datetime):
+        api_key (str):
+    """
+
+    id: str
+    org_id: str
+    name: str
+    created_by: Union[None, str]
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    api_key: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        id = self.id
+
+        org_id = self.org_id
+
+        name = self.name
+
+        created_by: Union[None, str]
+        created_by = self.created_by
+
+        created_at = self.created_at.isoformat()
+
+        updated_at = self.updated_at.isoformat()
+
+        api_key = self.api_key
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "id": id,
+                "org_id": org_id,
+                "name": name,
+                "created_by": created_by,
+                "created_at": created_at,
+                "updated_at": updated_at,
+                "api_key": api_key,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        id = d.pop("id")
+
+        org_id = d.pop("org_id")
+
+        name = d.pop("name")
+
+        def _parse_created_by(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        created_by = _parse_created_by(d.pop("created_by"))
+
+        created_at = isoparse(d.pop("created_at"))
+
+        updated_at = isoparse(d.pop("updated_at"))
+
+        api_key = d.pop("api_key")
+
+        create_api_key_response = cls(
+            id=id,
+            org_id=org_id,
+            name=name,
+            created_by=created_by,
+            created_at=created_at,
+            updated_at=updated_at,
+            api_key=api_key,
+        )
+
+        create_api_key_response.additional_properties = d
+        return create_api_key_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