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

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)

orca_sdk/_generated_api_client/models/embedding_model_result.py

@@ -1,123 +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, 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.analyze_neighbor_labels_result import AnalyzeNeighborLabelsResult
-
-
-T = TypeVar("T", bound="EmbeddingModelResult")
-
-
-@_attrs_define
-class EmbeddingModelResult:
-    """
-    Attributes:
-        embedding_model_name (str):
-        embedding_model_path (str):
-        analysis_result (AnalyzeNeighborLabelsResult):
-        memoryset_name (Union[None, Unset, str]):
-        is_finetuned (Union[Unset, bool]):  Default: False.
-    """
-
-    embedding_model_name: str
-    embedding_model_path: str
-    analysis_result: "AnalyzeNeighborLabelsResult"
-    memoryset_name: Union[None, Unset, str] = UNSET
-    is_finetuned: Union[Unset, bool] = False
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        embedding_model_name = self.embedding_model_name
-
-        embedding_model_path = self.embedding_model_path
-
-        analysis_result = self.analysis_result.to_dict()
-
-        memoryset_name: Union[None, Unset, str]
-        if isinstance(self.memoryset_name, Unset):
-            memoryset_name = UNSET
-        else:
-            memoryset_name = self.memoryset_name
-
-        is_finetuned = self.is_finetuned
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "embedding_model_name": embedding_model_name,
-                "embedding_model_path": embedding_model_path,
-                "analysis_result": analysis_result,
-            }
-        )
-        if memoryset_name is not UNSET:
-            field_dict["memoryset_name"] = memoryset_name
-        if is_finetuned is not UNSET:
-            field_dict["is_finetuned"] = is_finetuned
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.analyze_neighbor_labels_result import AnalyzeNeighborLabelsResult
-
-        d = src_dict.copy()
-        embedding_model_name = d.pop("embedding_model_name")
-
-        embedding_model_path = d.pop("embedding_model_path")
-
-        analysis_result = AnalyzeNeighborLabelsResult.from_dict(d.pop("analysis_result"))
-
-        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))
-
-        is_finetuned = d.pop("is_finetuned", UNSET)
-
-        embedding_model_result = cls(
-            embedding_model_name=embedding_model_name,
-            embedding_model_path=embedding_model_path,
-            analysis_result=analysis_result,
-            memoryset_name=memoryset_name,
-            is_finetuned=is_finetuned,
-        )
-
-        embedding_model_result.additional_properties = d
-        return embedding_model_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/evaluation_response.py

@@ -1,153 +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.classification_metrics import ClassificationMetrics
-    from ..models.regression_metrics import RegressionMetrics
-
-
-T = TypeVar("T", bound="EvaluationResponse")
-
-
-@_attrs_define
-class EvaluationResponse:
-    """
-    Attributes:
-        task_id (str):
-        org_id (str):
-        status (TaskStatus): Status of task in the task queue
-        result (Union['ClassificationMetrics', 'RegressionMetrics', None]):
-        created_at (datetime.datetime):
-        updated_at (datetime.datetime):
-    """
-
-    task_id: str
-    org_id: str
-    status: TaskStatus
-    result: Union["ClassificationMetrics", "RegressionMetrics", None]
-    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.classification_metrics import ClassificationMetrics
-        from ..models.regression_metrics import RegressionMetrics
-
-        task_id = self.task_id
-
-        org_id = self.org_id
-
-        status = self.status.value if isinstance(self.status, Enum) else self.status
-
-        result: Union[Dict[str, Any], None]
-        if isinstance(self.result, ClassificationMetrics):
-            result = self.result.to_dict()
-        elif isinstance(self.result, RegressionMetrics):
-            result = self.result.to_dict()
-        else:
-            result = self.result
-
-        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,
-                "status": status,
-                "result": result,
-                "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.classification_metrics import ClassificationMetrics
-        from ..models.regression_metrics import RegressionMetrics
-
-        d = src_dict.copy()
-        task_id = d.pop("task_id")
-
-        org_id = d.pop("org_id")
-
-        status = TaskStatus(d.pop("status"))
-
-        def _parse_result(data: object) -> Union["ClassificationMetrics", "RegressionMetrics", None]:
-            if data is None:
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                result_type_0 = ClassificationMetrics.from_dict(data)
-
-                return result_type_0
-            except:  # noqa: E722
-                pass
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                result_type_1 = RegressionMetrics.from_dict(data)
-
-                return result_type_1
-            except:  # noqa: E722
-                pass
-            return cast(Union["ClassificationMetrics", "RegressionMetrics", None], data)
-
-        result = _parse_result(d.pop("result"))
-
-        created_at = isoparse(d.pop("created_at"))
-
-        updated_at = isoparse(d.pop("updated_at"))
-
-        evaluation_response = cls(
-            task_id=task_id,
-            org_id=org_id,
-            status=status,
-            result=result,
-            created_at=created_at,
-            updated_at=updated_at,
-        )
-
-        evaluation_response.additional_properties = d
-        return 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/evaluation_response_classification_metrics.py

@@ -1,140 +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.classification_metrics import ClassificationMetrics
-
-
-T = TypeVar("T", bound="EvaluationResponseClassificationMetrics")
-
-
-@_attrs_define
-class EvaluationResponseClassificationMetrics:
-    """
-    Attributes:
-        task_id (str):
-        org_id (str):
-        status (TaskStatus): Status of task in the task queue
-        result (Union['ClassificationMetrics', None]):
-        created_at (datetime.datetime):
-        updated_at (datetime.datetime):
-    """
-
-    task_id: str
-    org_id: str
-    status: TaskStatus
-    result: Union["ClassificationMetrics", None]
-    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.classification_metrics import ClassificationMetrics
-
-        task_id = self.task_id
-
-        org_id = self.org_id
-
-        status = self.status.value if isinstance(self.status, Enum) else self.status
-
-        result: Union[Dict[str, Any], None]
-        if isinstance(self.result, ClassificationMetrics):
-            result = self.result.to_dict()
-        else:
-            result = self.result
-
-        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,
-                "status": status,
-                "result": result,
-                "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.classification_metrics import ClassificationMetrics
-
-        d = src_dict.copy()
-        task_id = d.pop("task_id")
-
-        org_id = d.pop("org_id")
-
-        status = TaskStatus(d.pop("status"))
-
-        def _parse_result(data: object) -> Union["ClassificationMetrics", None]:
-            if data is None:
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                result_type_0 = ClassificationMetrics.from_dict(data)
-
-                return result_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["ClassificationMetrics", None], data)
-
-        result = _parse_result(d.pop("result"))
-
-        created_at = isoparse(d.pop("created_at"))
-
-        updated_at = isoparse(d.pop("updated_at"))
-
-        evaluation_response_classification_metrics = cls(
-            task_id=task_id,
-            org_id=org_id,
-            status=status,
-            result=result,
-            created_at=created_at,
-            updated_at=updated_at,
-        )
-
-        evaluation_response_classification_metrics.additional_properties = d
-        return evaluation_response_classification_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