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

orca_sdk/_generated_api_client/models/embed_request.py

@@ -1,155 +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, 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="EmbedRequest")
-
-
-@_attrs_define
-class EmbedRequest:
-    """
-    Attributes:
-        values (List[str]):
-        max_seq_length (Union[None, Unset, int]):
-        prompt (Union[None, Unset, str]):
-    """
-
-    values: List[str]
-    max_seq_length: 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]:
-        values: List[str]
-        if isinstance(self.values, list):
-            values = self.values
-
-        max_seq_length: Union[None, Unset, int]
-        if isinstance(self.max_seq_length, Unset):
-            max_seq_length = UNSET
-        else:
-            max_seq_length = self.max_seq_length
-
-        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(
-            {
-                "values": values,
-            }
-        )
-        if max_seq_length is not UNSET:
-            field_dict["max_seq_length"] = max_seq_length
-        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()
-
-        def _parse_values(data: object) -> List[str]:
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                values_type_0 = cast(List[str], data)
-
-                return values_type_0
-            except:  # noqa: E722
-                pass
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                values_type_1 = cast(List[str], data)
-
-                return values_type_1
-            except:  # noqa: E722
-                pass
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                values_type_2 = cast(List[str], data)
-
-                return values_type_2
-            except:  # noqa: E722
-                pass
-            if not isinstance(data, list):
-                raise TypeError()
-            values_type_3 = []
-            _values_type_3 = data
-            for values_type_3_item_data in _values_type_3:
-
-                def _parse_values_type_3_item(data: object) -> str:
-                    return cast(str, data)
-
-                values_type_3_item = _parse_values_type_3_item(values_type_3_item_data)
-
-                values_type_3.append(values_type_3_item)
-
-            return values_type_3
-
-        values = _parse_values(d.pop("values"))
-
-        def _parse_max_seq_length(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 = _parse_max_seq_length(d.pop("max_seq_length", 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))
-
-        embed_request = cls(
-            values=values,
-            max_seq_length=max_seq_length,
-            prompt=prompt,
-        )
-
-        embed_request.additional_properties = d
-        return embed_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/embedding_evaluation_payload.py

@@ -1,205 +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="EmbeddingEvaluationPayload")
-
-
-@_attrs_define
-class EmbeddingEvaluationPayload:
-    """
-    Attributes:
-        value_column (str):
-        label_column (str):
-        source_id_column (Union[None, str]):
-        datasource_id (str):
-        neighbor_count (Union[Unset, int]):  Default: 5.
-        label_names (Union[List[str], None, Unset]):
-        embedding_models (Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset]):
-    """
-
-    value_column: str
-    label_column: str
-    source_id_column: Union[None, str]
-    datasource_id: str
-    neighbor_count: Union[Unset, int] = 5
-    label_names: Union[List[str], None, Unset] = UNSET
-    embedding_models: Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        value_column = self.value_column
-
-        label_column = self.label_column
-
-        source_id_column: Union[None, str]
-        source_id_column = self.source_id_column
-
-        datasource_id = self.datasource_id
-
-        neighbor_count = self.neighbor_count
-
-        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
-
-        embedding_models: Union[List[str], None, Unset]
-        if isinstance(self.embedding_models, Unset):
-            embedding_models = UNSET
-        elif isinstance(self.embedding_models, list):
-            embedding_models = []
-            for embedding_models_type_0_item_data in self.embedding_models:
-                embedding_models_type_0_item: str
-                if isinstance(embedding_models_type_0_item_data, PretrainedEmbeddingModelName):
-                    embedding_models_type_0_item = (
-                        embedding_models_type_0_item_data.value
-                        if isinstance(embedding_models_type_0_item_data, Enum)
-                        else embedding_models_type_0_item_data
-                    )
-                else:
-                    embedding_models_type_0_item = embedding_models_type_0_item_data
-                embedding_models.append(embedding_models_type_0_item)
-
-        else:
-            embedding_models = self.embedding_models
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "value_column": value_column,
-                "label_column": label_column,
-                "source_id_column": source_id_column,
-                "datasource_id": datasource_id,
-            }
-        )
-        if neighbor_count is not UNSET:
-            field_dict["neighbor_count"] = neighbor_count
-        if label_names is not UNSET:
-            field_dict["label_names"] = label_names
-        if embedding_models is not UNSET:
-            field_dict["embedding_models"] = embedding_models
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        value_column = d.pop("value_column")
-
-        label_column = d.pop("label_column")
-
-        def _parse_source_id_column(data: object) -> Union[None, str]:
-            if data is None:
-                return data
-            return cast(Union[None, str], data)
-
-        source_id_column = _parse_source_id_column(d.pop("source_id_column"))
-
-        datasource_id = d.pop("datasource_id")
-
-        neighbor_count = d.pop("neighbor_count", 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))
-
-        def _parse_embedding_models(data: object) -> Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                embedding_models_type_0 = []
-                _embedding_models_type_0 = data
-                for embedding_models_type_0_item_data in _embedding_models_type_0:
-
-                    def _parse_embedding_models_type_0_item(data: object) -> Union[PretrainedEmbeddingModelName, str]:
-                        try:
-                            if not isinstance(data, str):
-                                raise TypeError()
-                            embedding_models_type_0_item_type_0 = PretrainedEmbeddingModelName(data)
-
-                            return embedding_models_type_0_item_type_0
-                        except:  # noqa: E722
-                            pass
-                        return cast(Union[PretrainedEmbeddingModelName, str], data)
-
-                    embedding_models_type_0_item = _parse_embedding_models_type_0_item(
-                        embedding_models_type_0_item_data
-                    )
-
-                    embedding_models_type_0.append(embedding_models_type_0_item)
-
-                return embedding_models_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset], data)
-
-        embedding_models = _parse_embedding_models(d.pop("embedding_models", UNSET))
-
-        embedding_evaluation_payload = cls(
-            value_column=value_column,
-            label_column=label_column,
-            source_id_column=source_id_column,
-            datasource_id=datasource_id,
-            neighbor_count=neighbor_count,
-            label_names=label_names,
-            embedding_models=embedding_models,
-        )
-
-        embedding_evaluation_payload.additional_properties = d
-        return embedding_evaluation_payload
-
-    @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/embedding_evaluation_request.py

@@ -1,197 +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="EmbeddingEvaluationRequest")
-
-
-@_attrs_define
-class EmbeddingEvaluationRequest:
-    """
-    Attributes:
-        value_column (str):
-        label_column (str):
-        source_id_column (Union[None, str]):
-        neighbor_count (Union[Unset, int]):  Default: 5.
-        label_names (Union[List[str], None, Unset]):
-        embedding_models (Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset]):
-    """
-
-    value_column: str
-    label_column: str
-    source_id_column: Union[None, str]
-    neighbor_count: Union[Unset, int] = 5
-    label_names: Union[List[str], None, Unset] = UNSET
-    embedding_models: Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        value_column = self.value_column
-
-        label_column = self.label_column
-
-        source_id_column: Union[None, str]
-        source_id_column = self.source_id_column
-
-        neighbor_count = self.neighbor_count
-
-        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
-
-        embedding_models: Union[List[str], None, Unset]
-        if isinstance(self.embedding_models, Unset):
-            embedding_models = UNSET
-        elif isinstance(self.embedding_models, list):
-            embedding_models = []
-            for embedding_models_type_0_item_data in self.embedding_models:
-                embedding_models_type_0_item: str
-                if isinstance(embedding_models_type_0_item_data, PretrainedEmbeddingModelName):
-                    embedding_models_type_0_item = (
-                        embedding_models_type_0_item_data.value
-                        if isinstance(embedding_models_type_0_item_data, Enum)
-                        else embedding_models_type_0_item_data
-                    )
-                else:
-                    embedding_models_type_0_item = embedding_models_type_0_item_data
-                embedding_models.append(embedding_models_type_0_item)
-
-        else:
-            embedding_models = self.embedding_models
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "value_column": value_column,
-                "label_column": label_column,
-                "source_id_column": source_id_column,
-            }
-        )
-        if neighbor_count is not UNSET:
-            field_dict["neighbor_count"] = neighbor_count
-        if label_names is not UNSET:
-            field_dict["label_names"] = label_names
-        if embedding_models is not UNSET:
-            field_dict["embedding_models"] = embedding_models
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        value_column = d.pop("value_column")
-
-        label_column = d.pop("label_column")
-
-        def _parse_source_id_column(data: object) -> Union[None, str]:
-            if data is None:
-                return data
-            return cast(Union[None, str], data)
-
-        source_id_column = _parse_source_id_column(d.pop("source_id_column"))
-
-        neighbor_count = d.pop("neighbor_count", 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))
-
-        def _parse_embedding_models(data: object) -> Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, list):
-                    raise TypeError()
-                embedding_models_type_0 = []
-                _embedding_models_type_0 = data
-                for embedding_models_type_0_item_data in _embedding_models_type_0:
-
-                    def _parse_embedding_models_type_0_item(data: object) -> Union[PretrainedEmbeddingModelName, str]:
-                        try:
-                            if not isinstance(data, str):
-                                raise TypeError()
-                            embedding_models_type_0_item_type_0 = PretrainedEmbeddingModelName(data)
-
-                            return embedding_models_type_0_item_type_0
-                        except:  # noqa: E722
-                            pass
-                        return cast(Union[PretrainedEmbeddingModelName, str], data)
-
-                    embedding_models_type_0_item = _parse_embedding_models_type_0_item(
-                        embedding_models_type_0_item_data
-                    )
-
-                    embedding_models_type_0.append(embedding_models_type_0_item)
-
-                return embedding_models_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union[List[Union[PretrainedEmbeddingModelName, str]], None, Unset], data)
-
-        embedding_models = _parse_embedding_models(d.pop("embedding_models", UNSET))
-
-        embedding_evaluation_request = cls(
-            value_column=value_column,
-            label_column=label_column,
-            source_id_column=source_id_column,
-            neighbor_count=neighbor_count,
-            label_names=label_names,
-            embedding_models=embedding_models,
-        )
-
-        embedding_evaluation_request.additional_properties = d
-        return embedding_evaluation_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