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

orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py

@@ -1,259 +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.pretrained_embedding_model_name import PretrainedEmbeddingModelName
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="CreateLabeledMemorysetRequest")
-
-
-@_attrs_define
-class CreateLabeledMemorysetRequest:
-    """
-    Attributes:
-        name (str):
-        datasource_label_column (str):
-        datasource_value_column (str):
-        datasource_id (Union[None, Unset, str]):
-        datasource_name (Union[None, Unset, str]):
-        datasource_source_id_column (Union[None, Unset, str]):
-        pretrained_embedding_model_name (Union[None, PretrainedEmbeddingModelName, Unset]):
-        finetuned_embedding_model_id (Union[None, Unset, str]):
-        finetuned_embedding_model_name (Union[None, Unset, str]):
-        max_seq_length_override (Union[None, Unset, int]):
-        label_names (Union[Unset, List[str]]):
-    """
-
-    name: str
-    datasource_label_column: str
-    datasource_value_column: str
-    datasource_id: Union[None, Unset, str] = UNSET
-    datasource_name: Union[None, Unset, str] = UNSET
-    datasource_source_id_column: Union[None, Unset, str] = UNSET
-    pretrained_embedding_model_name: Union[None, PretrainedEmbeddingModelName, Unset] = UNSET
-    finetuned_embedding_model_id: Union[None, Unset, str] = UNSET
-    finetuned_embedding_model_name: Union[None, Unset, str] = UNSET
-    max_seq_length_override: Union[None, Unset, int] = UNSET
-    label_names: Union[Unset, List[str]] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        name = self.name
-
-        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
-
-        datasource_source_id_column: Union[None, Unset, str]
-        if isinstance(self.datasource_source_id_column, Unset):
-            datasource_source_id_column = UNSET
-        else:
-            datasource_source_id_column = self.datasource_source_id_column
-
-        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
-
-        finetuned_embedding_model_name: Union[None, Unset, str]
-        if isinstance(self.finetuned_embedding_model_name, Unset):
-            finetuned_embedding_model_name = UNSET
-        else:
-            finetuned_embedding_model_name = self.finetuned_embedding_model_name
-
-        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
-
-        label_names: Union[Unset, List[str]] = UNSET
-        if not isinstance(self.label_names, Unset):
-            label_names = self.label_names
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "name": name,
-                "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 datasource_source_id_column is not UNSET:
-            field_dict["datasource_source_id_column"] = datasource_source_id_column
-        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 finetuned_embedding_model_name is not UNSET:
-            field_dict["finetuned_embedding_model_name"] = finetuned_embedding_model_name
-        if max_seq_length_override is not UNSET:
-            field_dict["max_seq_length_override"] = max_seq_length_override
-        if label_names is not UNSET:
-            field_dict["label_names"] = label_names
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        name = d.pop("name")
-
-        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_datasource_source_id_column(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_source_id_column = _parse_datasource_source_id_column(d.pop("datasource_source_id_column", UNSET))
-
-        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_finetuned_embedding_model_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)
-
-        finetuned_embedding_model_name = _parse_finetuned_embedding_model_name(
-            d.pop("finetuned_embedding_model_name", 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))
-
-        label_names = cast(List[str], d.pop("label_names", UNSET))
-
-        create_labeled_memoryset_request = cls(
-            name=name,
-            datasource_label_column=datasource_label_column,
-            datasource_value_column=datasource_value_column,
-            datasource_id=datasource_id,
-            datasource_name=datasource_name,
-            datasource_source_id_column=datasource_source_id_column,
-            pretrained_embedding_model_name=pretrained_embedding_model_name,
-            finetuned_embedding_model_id=finetuned_embedding_model_id,
-            finetuned_embedding_model_name=finetuned_embedding_model_name,
-            max_seq_length_override=max_seq_length_override,
-            label_names=label_names,
-        )
-
-        create_labeled_memoryset_request.additional_properties = d
-        return create_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/create_rac_model_request.py

@@ -1,209 +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 ..models.rac_head_type import RACHeadType
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="CreateRACModelRequest")
-
-
-@_attrs_define
-class CreateRACModelRequest:
-    """
-    Attributes:
-        name (str):
-        memoryset_id (Union[None, Unset, str]):
-        memoryset_name (Union[None, Unset, str]):
-        head_type (Union[Unset, RACHeadType]):
-        weigh_memories (Union[None, Unset, bool]):
-        min_memory_weight (Union[None, Unset, float]):
-        memory_lookup_count (Union[None, Unset, int]):
-        num_classes (Union[None, Unset, int]):
-    """
-
-    name: str
-    memoryset_id: Union[None, Unset, str] = UNSET
-    memoryset_name: Union[None, Unset, str] = UNSET
-    head_type: Union[Unset, RACHeadType] = UNSET
-    weigh_memories: Union[None, Unset, bool] = UNSET
-    min_memory_weight: Union[None, Unset, float] = UNSET
-    memory_lookup_count: Union[None, Unset, int] = UNSET
-    num_classes: 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
-
-        memoryset_id: Union[None, Unset, str]
-        if isinstance(self.memoryset_id, Unset):
-            memoryset_id = UNSET
-        else:
-            memoryset_id = self.memoryset_id
-
-        memoryset_name: Union[None, Unset, str]
-        if isinstance(self.memoryset_name, Unset):
-            memoryset_name = UNSET
-        else:
-            memoryset_name = self.memoryset_name
-
-        head_type: Union[Unset, str] = UNSET
-        if not isinstance(self.head_type, Unset):
-            head_type = self.head_type.value
-
-        weigh_memories: Union[None, Unset, bool]
-        if isinstance(self.weigh_memories, Unset):
-            weigh_memories = UNSET
-        else:
-            weigh_memories = self.weigh_memories
-
-        min_memory_weight: Union[None, Unset, float]
-        if isinstance(self.min_memory_weight, Unset):
-            min_memory_weight = UNSET
-        else:
-            min_memory_weight = self.min_memory_weight
-
-        memory_lookup_count: Union[None, Unset, int]
-        if isinstance(self.memory_lookup_count, Unset):
-            memory_lookup_count = UNSET
-        else:
-            memory_lookup_count = self.memory_lookup_count
-
-        num_classes: Union[None, Unset, int]
-        if isinstance(self.num_classes, Unset):
-            num_classes = UNSET
-        else:
-            num_classes = self.num_classes
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "name": name,
-            }
-        )
-        if memoryset_id is not UNSET:
-            field_dict["memoryset_id"] = memoryset_id
-        if memoryset_name is not UNSET:
-            field_dict["memoryset_name"] = memoryset_name
-        if head_type is not UNSET:
-            field_dict["head_type"] = head_type
-        if weigh_memories is not UNSET:
-            field_dict["weigh_memories"] = weigh_memories
-        if min_memory_weight is not UNSET:
-            field_dict["min_memory_weight"] = min_memory_weight
-        if memory_lookup_count is not UNSET:
-            field_dict["memory_lookup_count"] = memory_lookup_count
-        if num_classes is not UNSET:
-            field_dict["num_classes"] = num_classes
-
-        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_memoryset_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_id = _parse_memoryset_id(d.pop("memoryset_id", UNSET))
-
-        def _parse_memoryset_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)
-
-        memoryset_name = _parse_memoryset_name(d.pop("memoryset_name", UNSET))
-
-        _head_type = d.pop("head_type", UNSET)
-        head_type: Union[Unset, RACHeadType]
-        if isinstance(_head_type, Unset):
-            head_type = UNSET
-        else:
-            head_type = RACHeadType(_head_type)
-
-        def _parse_weigh_memories(data: object) -> Union[None, Unset, bool]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, bool], data)
-
-        weigh_memories = _parse_weigh_memories(d.pop("weigh_memories", UNSET))
-
-        def _parse_min_memory_weight(data: object) -> Union[None, Unset, float]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, float], data)
-
-        min_memory_weight = _parse_min_memory_weight(d.pop("min_memory_weight", UNSET))
-
-        def _parse_memory_lookup_count(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)
-
-        memory_lookup_count = _parse_memory_lookup_count(d.pop("memory_lookup_count", UNSET))
-
-        def _parse_num_classes(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)
-
-        num_classes = _parse_num_classes(d.pop("num_classes", UNSET))
-
-        create_rac_model_request = cls(
-            name=name,
-            memoryset_id=memoryset_id,
-            memoryset_name=memoryset_name,
-            head_type=head_type,
-            weigh_memories=weigh_memories,
-            min_memory_weight=min_memory_weight,
-            memory_lookup_count=memory_lookup_count,
-            num_classes=num_classes,
-        )
-
-        create_rac_model_request.additional_properties = d
-        return create_rac_model_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/datasource_metadata.py

@@ -1,142 +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 TYPE_CHECKING, Any, List, Type, TypeVar
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-from dateutil.parser import isoparse
-
-if TYPE_CHECKING:
-    from ..models.column_info import ColumnInfo
-
-
-T = TypeVar("T", bound="DatasourceMetadata")
-
-
-@_attrs_define
-class DatasourceMetadata:
-    """
-    Attributes:
-        id (str):
-        org_id (str):
-        name (str):
-        storage_path (str):
-        length (int):
-        columns (List['ColumnInfo']):
-        created_at (datetime.datetime):
-        updated_at (datetime.datetime):
-    """
-
-    id: str
-    org_id: str
-    name: str
-    storage_path: str
-    length: int
-    columns: List["ColumnInfo"]
-    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]:
-        id = self.id
-
-        org_id = self.org_id
-
-        name = self.name
-
-        storage_path = self.storage_path
-
-        length = self.length
-
-        columns = []
-        for columns_item_data in self.columns:
-            columns_item = columns_item_data.to_dict()
-            columns.append(columns_item)
-
-        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(
-            {
-                "id": id,
-                "org_id": org_id,
-                "name": name,
-                "storage_path": storage_path,
-                "length": length,
-                "columns": columns,
-                "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.column_info import ColumnInfo
-
-        d = src_dict.copy()
-        id = d.pop("id")
-
-        org_id = d.pop("org_id")
-
-        name = d.pop("name")
-
-        storage_path = d.pop("storage_path")
-
-        length = d.pop("length")
-
-        columns = []
-        _columns = d.pop("columns")
-        for columns_item_data in _columns:
-            columns_item = ColumnInfo.from_dict(columns_item_data)
-
-            columns.append(columns_item)
-
-        created_at = isoparse(d.pop("created_at"))
-
-        updated_at = isoparse(d.pop("updated_at"))
-
-        datasource_metadata = cls(
-            id=id,
-            org_id=org_id,
-            name=name,
-            storage_path=storage_path,
-            length=length,
-            columns=columns,
-            created_at=created_at,
-            updated_at=updated_at,
-        )
-
-        datasource_metadata.additional_properties = d
-        return datasource_metadata
-
-    @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/delete_memories_request.py

@@ -1,70 +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, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-T = TypeVar("T", bound="DeleteMemoriesRequest")
-
-
-@_attrs_define
-class DeleteMemoriesRequest:
-    """
-    Attributes:
-        memory_ids (List[str]):
-    """
-
-    memory_ids: List[str]
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        memory_ids = self.memory_ids
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "memory_ids": memory_ids,
-            }
-        )
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        memory_ids = cast(List[str], d.pop("memory_ids"))
-
-        delete_memories_request = cls(
-            memory_ids=memory_ids,
-        )
-
-        delete_memories_request.additional_properties = d
-        return delete_memories_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