orca-sdk 0.0.97__py3-none-any.whl → 0.0.98__py3-none-any.whl

orca_sdk/_generated_api_client/models/classification_prediction_request.py

@@ -1,220 +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 TYPE_CHECKING, Any, Dict, List, Type, TypeVar, Union, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-if TYPE_CHECKING:
-    from ..models.filter_item import FilterItem
-
-
-T = TypeVar("T", bound="ClassificationPredictionRequest")
-
-
-@_attrs_define
-class ClassificationPredictionRequest:
-    """
-    Attributes:
-        input_values (List[str]):
-        expected_labels (Union[List[int], None, Unset]):
-        filters (Union[Unset, List['FilterItem']]):
-        tags (Union[Unset, List[str]]):
-        memoryset_override_id (Union[None, Unset, str]):
-        save_telemetry (Union[Unset, bool]):  Default: True.
-        save_telemetry_synchronously (Union[Unset, bool]):  Default: False.
-    """
-
-    input_values: List[str]
-    expected_labels: Union[List[int], None, Unset] = UNSET
-    filters: Union[Unset, List["FilterItem"]] = UNSET
-    tags: Union[Unset, List[str]] = UNSET
-    memoryset_override_id: Union[None, Unset, str] = UNSET
-    save_telemetry: Union[Unset, bool] = True
-    save_telemetry_synchronously: Union[Unset, bool] = False
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        input_values: List[str]
-        if isinstance(self.input_values, list):
-            input_values = self.input_values
-
-        expected_labels: Union[List[int], None, Unset]
-        if isinstance(self.expected_labels, Unset):
-            expected_labels = UNSET
-        elif isinstance(self.expected_labels, list):
-            expected_labels = self.expected_labels
-
-        else:
-            expected_labels = self.expected_labels
-
-        filters: Union[Unset, List[Dict[str, Any]]] = UNSET
-        if not isinstance(self.filters, Unset):
-            filters = []
-            for filters_item_data in self.filters:
-                filters_item = filters_item_data.to_dict()
-                filters.append(filters_item)
-
-        tags: Union[Unset, List[str]] = UNSET
-        if not isinstance(self.tags, Unset):
-            tags = self.tags
-
-        memoryset_override_id: Union[None, Unset, str]
-        if isinstance(self.memoryset_override_id, Unset):
-            memoryset_override_id = UNSET
-        else:
-            memoryset_override_id = self.memoryset_override_id
-
-        save_telemetry = self.save_telemetry
-
-        save_telemetry_synchronously = self.save_telemetry_synchronously
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "input_values": input_values,
-            }
-        )
-        if expected_labels is not UNSET:
-            field_dict["expected_labels"] = expected_labels
-        if filters is not UNSET:
-            field_dict["filters"] = filters
-        if tags is not UNSET:
-            field_dict["tags"] = tags
-        if memoryset_override_id is not UNSET:
-            field_dict["memoryset_override_id"] = memoryset_override_id
-        if save_telemetry is not UNSET:
-            field_dict["save_telemetry"] = save_telemetry
-        if save_telemetry_synchronously is not UNSET:
-            field_dict["save_telemetry_synchronously"] = save_telemetry_synchronously
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.filter_item import FilterItem
-
-        d = src_dict.copy()
-
-        def _parse_input_values(data: object) -> List[str]:
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                input_values_type_0 = cast(List[str], data)
-
-                return input_values_type_0
-            except:  # noqa: E722
-                pass
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                input_values_type_1 = cast(List[str], data)
-
-                return input_values_type_1
-            except:  # noqa: E722
-                pass
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                input_values_type_2 = cast(List[str], data)
-
-                return input_values_type_2
-            except:  # noqa: E722
-                pass
-            if not isinstance(data, list):
-                raise TypeError()
-            input_values_type_3 = []
-            _input_values_type_3 = data
-            for input_values_type_3_item_data in _input_values_type_3:
-
-                def _parse_input_values_type_3_item(data: object) -> str:
-                    return cast(str, data)
-
-                input_values_type_3_item = _parse_input_values_type_3_item(input_values_type_3_item_data)
-
-                input_values_type_3.append(input_values_type_3_item)
-
-            return input_values_type_3
-
-        input_values = _parse_input_values(d.pop("input_values"))
-
-        def _parse_expected_labels(data: object) -> Union[List[int], None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                expected_labels_type_0 = cast(List[int], data)
-
-                return expected_labels_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union[List[int], None, Unset], data)
-
-        expected_labels = _parse_expected_labels(d.pop("expected_labels", UNSET))
-
-        filters = []
-        _filters = d.pop("filters", UNSET)
-        for filters_item_data in _filters or []:
-            filters_item = FilterItem.from_dict(filters_item_data)
-
-            filters.append(filters_item)
-
-        tags = cast(List[str], d.pop("tags", UNSET))
-
-        def _parse_memoryset_override_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_override_id = _parse_memoryset_override_id(d.pop("memoryset_override_id", UNSET))
-
-        save_telemetry = d.pop("save_telemetry", UNSET)
-
-        save_telemetry_synchronously = d.pop("save_telemetry_synchronously", UNSET)
-
-        classification_prediction_request = cls(
-            input_values=input_values,
-            expected_labels=expected_labels,
-            filters=filters,
-            tags=tags,
-            memoryset_override_id=memoryset_override_id,
-            save_telemetry=save_telemetry,
-            save_telemetry_synchronously=save_telemetry_synchronously,
-        )
-
-        classification_prediction_request.additional_properties = d
-        return classification_prediction_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/clone_memoryset_request.py

@@ -1,210 +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="CloneMemorysetRequest")
-
-
-@_attrs_define
-class CloneMemorysetRequest:
-    """
-    Attributes:
-        name (str):
-        description (Union[None, Unset, str]):
-        notes (Union[None, Unset, 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]):
-        prompt (Union[None, Unset, str]):
-    """
-
-    name: str
-    description: Union[None, Unset, str] = UNSET
-    notes: Union[None, Unset, str] = UNSET
-    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
-    prompt: Union[None, Unset, str] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        name = self.name
-
-        description: Union[None, Unset, str]
-        if isinstance(self.description, Unset):
-            description = UNSET
-        else:
-            description = self.description
-
-        notes: Union[None, Unset, str]
-        if isinstance(self.notes, Unset):
-            notes = UNSET
-        else:
-            notes = self.notes
-
-        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
-
-        prompt: Union[None, Unset, str]
-        if isinstance(self.prompt, Unset):
-            prompt = UNSET
-        else:
-            prompt = self.prompt
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "name": name,
-            }
-        )
-        if description is not UNSET:
-            field_dict["description"] = description
-        if notes is not UNSET:
-            field_dict["notes"] = notes
-        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
-        if prompt is not UNSET:
-            field_dict["prompt"] = prompt
-
-        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_description(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)
-
-        description = _parse_description(d.pop("description", UNSET))
-
-        def _parse_notes(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)
-
-        notes = _parse_notes(d.pop("notes", 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_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_prompt(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)
-
-        prompt = _parse_prompt(d.pop("prompt", UNSET))
-
-        clone_memoryset_request = cls(
-            name=name,
-            description=description,
-            notes=notes,
-            pretrained_embedding_model_name=pretrained_embedding_model_name,
-            finetuned_embedding_model_id=finetuned_embedding_model_id,
-            max_seq_length_override=max_seq_length_override,
-            prompt=prompt,
-        )
-
-        clone_memoryset_request.additional_properties = d
-        return clone_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/cluster_metrics.py

@@ -1,78 +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
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-T = TypeVar("T", bound="ClusterMetrics")
-
-
-@_attrs_define
-class ClusterMetrics:
-    """
-    Attributes:
-        cluster (int):
-        memory_count (int):
-    """
-
-    cluster: int
-    memory_count: int
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        cluster = self.cluster
-
-        memory_count = self.memory_count
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "cluster": cluster,
-                "memory_count": memory_count,
-            }
-        )
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        cluster = d.pop("cluster")
-
-        memory_count = d.pop("memory_count")
-
-        cluster_metrics = cls(
-            cluster=cluster,
-            memory_count=memory_count,
-        )
-
-        cluster_metrics.additional_properties = d
-        return cluster_metrics
-
-    @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,145 +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]):
-        int_values (Union[List[int], None, Unset]):
-    """
-
-    name: str
-    type: ColumnType
-    enum_options: Union[List[str], None, Unset] = UNSET
-    int_values: Union[List[int], 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
-
-        int_values: Union[List[int], None, Unset]
-        if isinstance(self.int_values, Unset):
-            int_values = UNSET
-        elif isinstance(self.int_values, list):
-            int_values = self.int_values
-
-        else:
-            int_values = self.int_values
-
-        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
-        if int_values is not UNSET:
-            field_dict["int_values"] = int_values
-
-        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))
-
-        def _parse_int_values(data: object) -> Union[List[int], None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                int_values_type_0 = cast(List[int], data)
-
-                return int_values_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union[List[int], None, Unset], data)
-
-        int_values = _parse_int_values(d.pop("int_values", UNSET))
-
-        column_info = cls(
-            name=name,
-            type=type,
-            enum_options=enum_options,
-            int_values=int_values,
-        )
-
-        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)