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

orca_sdk/_generated_api_client/models/embedding_evaluation_payload.py

@@ -1,187 +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[PretrainedEmbeddingModelName], 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[PretrainedEmbeddingModelName], 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 = (
-                    embedding_models_type_0_item_data.value
-                    if isinstance(embedding_models_type_0_item_data, Enum)
-                    else 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[PretrainedEmbeddingModelName], 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:
-                    embedding_models_type_0_item = PretrainedEmbeddingModelName(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[PretrainedEmbeddingModelName], 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,179 +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[PretrainedEmbeddingModelName], 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[PretrainedEmbeddingModelName], 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 = (
-                    embedding_models_type_0_item_data.value
-                    if isinstance(embedding_models_type_0_item_data, Enum)
-                    else 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[PretrainedEmbeddingModelName], 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:
-                    embedding_models_type_0_item = PretrainedEmbeddingModelName(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[PretrainedEmbeddingModelName], 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

orca_sdk/_generated_api_client/models/embedding_evaluation_response.py

@@ -1,158 +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 enum import Enum
-from typing import TYPE_CHECKING, Any, Dict, 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.task_status import TaskStatus
-
-if TYPE_CHECKING:
-    from ..models.embedding_evaluation_payload import EmbeddingEvaluationPayload
-    from ..models.embedding_evaluation_result import EmbeddingEvaluationResult
-
-
-T = TypeVar("T", bound="EmbeddingEvaluationResponse")
-
-
-@_attrs_define
-class EmbeddingEvaluationResponse:
-    """
-    Attributes:
-        task_id (str):
-        org_id (str):
-        datasource_id (str):
-        status (TaskStatus): Status of task in the task queue
-        result (Union['EmbeddingEvaluationResult', None]):
-        payload (EmbeddingEvaluationPayload):
-        created_at (datetime.datetime):
-        updated_at (datetime.datetime):
-    """
-
-    task_id: str
-    org_id: str
-    datasource_id: str
-    status: TaskStatus
-    result: Union["EmbeddingEvaluationResult", None]
-    payload: "EmbeddingEvaluationPayload"
-    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]:
-        from ..models.embedding_evaluation_result import EmbeddingEvaluationResult
-
-        task_id = self.task_id
-
-        org_id = self.org_id
-
-        datasource_id = self.datasource_id
-
-        status = self.status.value if isinstance(self.status, Enum) else self.status
-
-        result: Union[Dict[str, Any], None]
-        if isinstance(self.result, EmbeddingEvaluationResult):
-            result = self.result.to_dict()
-        else:
-            result = self.result
-
-        payload = self.payload.to_dict()
-
-        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(
-            {
-                "task_id": task_id,
-                "org_id": org_id,
-                "datasource_id": datasource_id,
-                "status": status,
-                "result": result,
-                "payload": payload,
-                "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.embedding_evaluation_payload import EmbeddingEvaluationPayload
-        from ..models.embedding_evaluation_result import EmbeddingEvaluationResult
-
-        d = src_dict.copy()
-        task_id = d.pop("task_id")
-
-        org_id = d.pop("org_id")
-
-        datasource_id = d.pop("datasource_id")
-
-        status = TaskStatus(d.pop("status"))
-
-        def _parse_result(data: object) -> Union["EmbeddingEvaluationResult", None]:
-            if data is None:
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                result_type_0 = EmbeddingEvaluationResult.from_dict(data)
-
-                return result_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["EmbeddingEvaluationResult", None], data)
-
-        result = _parse_result(d.pop("result"))
-
-        payload = EmbeddingEvaluationPayload.from_dict(d.pop("payload"))
-
-        created_at = isoparse(d.pop("created_at"))
-
-        updated_at = isoparse(d.pop("updated_at"))
-
-        embedding_evaluation_response = cls(
-            task_id=task_id,
-            org_id=org_id,
-            datasource_id=datasource_id,
-            status=status,
-            result=result,
-            payload=payload,
-            created_at=created_at,
-            updated_at=updated_at,
-        )
-
-        embedding_evaluation_response.additional_properties = d
-        return embedding_evaluation_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/embedding_evaluation_result.py

@@ -1,86 +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
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..types import UNSET, Unset
-
-if TYPE_CHECKING:
-    from ..models.embedding_model_result import EmbeddingModelResult
-
-
-T = TypeVar("T", bound="EmbeddingEvaluationResult")
-
-
-@_attrs_define
-class EmbeddingEvaluationResult:
-    """
-    Attributes:
-        evaluation_results (Union[Unset, List['EmbeddingModelResult']]):
-    """
-
-    evaluation_results: Union[Unset, List["EmbeddingModelResult"]] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        evaluation_results: Union[Unset, List[Dict[str, Any]]] = UNSET
-        if not isinstance(self.evaluation_results, Unset):
-            evaluation_results = []
-            for evaluation_results_item_data in self.evaluation_results:
-                evaluation_results_item = evaluation_results_item_data.to_dict()
-                evaluation_results.append(evaluation_results_item)
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if evaluation_results is not UNSET:
-            field_dict["evaluation_results"] = evaluation_results
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.embedding_model_result import EmbeddingModelResult
-
-        d = src_dict.copy()
-        evaluation_results = []
-        _evaluation_results = d.pop("evaluation_results", UNSET)
-        for evaluation_results_item_data in _evaluation_results or []:
-            evaluation_results_item = EmbeddingModelResult.from_dict(evaluation_results_item_data)
-
-            evaluation_results.append(evaluation_results_item)
-
-        embedding_evaluation_result = cls(
-            evaluation_results=evaluation_results,
-        )
-
-        embedding_evaluation_result.additional_properties = d
-        return embedding_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/embedding_finetuning_method.py

@@ -1,9 +0,0 @@
-from enum import Enum
-
-
-class EmbeddingFinetuningMethod(str, Enum):
-    BATCH_TRIPLET_LOSS = "batch_triplet_loss"
-    CLASSIFICATION = "classification"
-
-    def __str__(self) -> str:
-        return str(self.value)