orca-sdk 0.0.78__py3-none-any.whl

orca_sdk/_generated_api_client/models/create_api_key_response.py

@@ -0,0 +1,145 @@
+"""
+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 Any, List, 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.create_api_key_response_scope_item import CreateApiKeyResponseScopeItem
+
+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):
+        scope (List[CreateApiKeyResponseScopeItem]):
+        api_key (str):
+    """
+
+    id: str
+    org_id: str
+    name: str
+    created_by: Union[None, str]
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    scope: List[CreateApiKeyResponseScopeItem]
+    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()
+
+        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)
+
+        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,
+                "scope": scope,
+                "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"))
+
+        scope = []
+        _scope = d.pop("scope")
+        for scope_item_data in _scope:
+            scope_item = CreateApiKeyResponseScopeItem(scope_item_data)
+
+            scope.append(scope_item)
+
+        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,
+            scope=scope,
+            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

orca_sdk/_generated_api_client/models/create_api_key_response_scope_item.py

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

orca_sdk/_generated_api_client/models/create_labeled_memoryset_request.py

@@ -0,0 +1,279 @@
+"""
+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[List[str], None, Unset]):
+    """
+
+    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[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
+
+        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[List[str], None, Unset]
+        if isinstance(self.label_names, Unset):
+            label_names = UNSET
+        elif isinstance(self.label_names, list):
+            label_names = self.label_names
+
+        else:
+            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))
+
+        def _parse_label_names(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()
+                label_names_type_0 = cast(List[str], data)
+
+                return label_names_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[List[str], None, Unset], data)
+
+        label_names = _parse_label_names(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

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