orca-sdk 0.0.78__py3-none-any.whl

orca_sdk/_generated_api_client/models/classification_evaluation_result.py

@@ -0,0 +1,147 @@
+"""
+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 TYPE_CHECKING, Any, Dict, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+    from ..models.precision_recall_curve import PrecisionRecallCurve
+    from ..models.roc_curve import ROCCurve
+
+
+T = TypeVar("T", bound="ClassificationEvaluationResult")
+
+
+@_attrs_define
+class ClassificationEvaluationResult:
+    """
+    Attributes:
+        f1_score (float):
+        accuracy (float):
+        loss (float):
+        precision_recall_curve (Union['PrecisionRecallCurve', None]):
+        roc_curve (Union['ROCCurve', None]):
+    """
+
+    f1_score: float
+    accuracy: float
+    loss: float
+    precision_recall_curve: Union["PrecisionRecallCurve", None]
+    roc_curve: Union["ROCCurve", None]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.precision_recall_curve import PrecisionRecallCurve
+        from ..models.roc_curve import ROCCurve
+
+        f1_score = self.f1_score
+
+        accuracy = self.accuracy
+
+        loss = self.loss
+
+        precision_recall_curve: Union[Dict[str, Any], None]
+        if isinstance(self.precision_recall_curve, PrecisionRecallCurve):
+            precision_recall_curve = self.precision_recall_curve.to_dict()
+        else:
+            precision_recall_curve = self.precision_recall_curve
+
+        roc_curve: Union[Dict[str, Any], None]
+        if isinstance(self.roc_curve, ROCCurve):
+            roc_curve = self.roc_curve.to_dict()
+        else:
+            roc_curve = self.roc_curve
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "f1_score": f1_score,
+                "accuracy": accuracy,
+                "loss": loss,
+                "precision_recall_curve": precision_recall_curve,
+                "roc_curve": roc_curve,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.precision_recall_curve import PrecisionRecallCurve
+        from ..models.roc_curve import ROCCurve
+
+        d = src_dict.copy()
+        f1_score = d.pop("f1_score")
+
+        accuracy = d.pop("accuracy")
+
+        loss = d.pop("loss")
+
+        def _parse_precision_recall_curve(data: object) -> Union["PrecisionRecallCurve", None]:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                precision_recall_curve_type_0 = PrecisionRecallCurve.from_dict(data)
+
+                return precision_recall_curve_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["PrecisionRecallCurve", None], data)
+
+        precision_recall_curve = _parse_precision_recall_curve(d.pop("precision_recall_curve"))
+
+        def _parse_roc_curve(data: object) -> Union["ROCCurve", None]:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                roc_curve_type_0 = ROCCurve.from_dict(data)
+
+                return roc_curve_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["ROCCurve", None], data)
+
+        roc_curve = _parse_roc_curve(d.pop("roc_curve"))
+
+        classification_evaluation_result = cls(
+            f1_score=f1_score,
+            accuracy=accuracy,
+            loss=loss,
+            precision_recall_curve=precision_recall_curve,
+            roc_curve=roc_curve,
+        )
+
+        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,120 @@
+"""
+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.create_api_key_request_scope_item import CreateApiKeyRequestScopeItem
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="CreateApiKeyRequest")
+
+
+@_attrs_define
+class CreateApiKeyRequest:
+    """
+    Attributes:
+        scope (List[CreateApiKeyRequestScopeItem]):
+        id (Union[Unset, str]):
+        name (Union[Unset, str]):  Default: 'default'.
+        created_by (Union[None, Unset, str]):
+    """
+
+    scope: List[CreateApiKeyRequestScopeItem]
+    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]:
+        scope = []
+        for scope_item_data in self.scope:
+            scope_item = scope_item_data.value if isinstance(scope_item_data, Enum) else scope_item_data
+            scope.append(scope_item)
+
+        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(
+            {
+                "scope": scope,
+            }
+        )
+        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()
+        scope = []
+        _scope = d.pop("scope")
+        for scope_item_data in _scope:
+            scope_item = CreateApiKeyRequestScopeItem(scope_item_data)
+
+            scope.append(scope_item)
+
+        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(
+            scope=scope,
+            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_request_scope_item.py

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