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

orca_sdk/_generated_api_client/models/classification_evaluation_result.py

@@ -1,114 +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, 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

@@ -1,150 +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 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

@@ -1,114 +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 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

@@ -1,14 +0,0 @@
-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

@@ -1,80 +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, 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

@@ -1,99 +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, 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

@@ -1,126 +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 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