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

orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py

@@ -1,107 +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
-
-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="PretrainedEmbeddingModelMetadata")
-
-
-@_attrs_define
-class PretrainedEmbeddingModelMetadata:
-    """
-    Attributes:
-        embedding_dim (int):
-        max_seq_length (int):
-        uses_context (bool):
-        name (PretrainedEmbeddingModelName): Names of pretrained embedding models that are supported by OrcaCloud
-        experimental (Union[Unset, bool]):  Default: False.
-    """
-
-    embedding_dim: int
-    max_seq_length: int
-    uses_context: bool
-    name: PretrainedEmbeddingModelName
-    experimental: Union[Unset, bool] = False
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        embedding_dim = self.embedding_dim
-
-        max_seq_length = self.max_seq_length
-
-        uses_context = self.uses_context
-
-        name = self.name.value if isinstance(self.name, Enum) else self.name
-
-        experimental = self.experimental
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "embedding_dim": embedding_dim,
-                "max_seq_length": max_seq_length,
-                "uses_context": uses_context,
-                "name": name,
-            }
-        )
-        if experimental is not UNSET:
-            field_dict["experimental"] = experimental
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        embedding_dim = d.pop("embedding_dim")
-
-        max_seq_length = d.pop("max_seq_length")
-
-        uses_context = d.pop("uses_context")
-
-        name = PretrainedEmbeddingModelName(d.pop("name"))
-
-        experimental = d.pop("experimental", UNSET)
-
-        pretrained_embedding_model_metadata = cls(
-            embedding_dim=embedding_dim,
-            max_seq_length=max_seq_length,
-            uses_context=uses_context,
-            name=name,
-            experimental=experimental,
-        )
-
-        pretrained_embedding_model_metadata.additional_properties = d
-        return pretrained_embedding_model_metadata
-
-    @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/pretrained_embedding_model_name.py

@@ -1,17 +0,0 @@
-from enum import Enum
-
-
-class PretrainedEmbeddingModelName(str, Enum):
-    BGE_BASE = "BGE_BASE"
-    CDE_SMALL = "CDE_SMALL"
-    CLIP_BASE = "CLIP_BASE"
-    DISTILBERT = "DISTILBERT"
-    E5_LARGE = "E5_LARGE"
-    GIST_LARGE = "GIST_LARGE"
-    GTE_BASE = "GTE_BASE"
-    GTE_SMALL = "GTE_SMALL"
-    MXBAI_LARGE = "MXBAI_LARGE"
-    QWEN2_1_5B = "QWEN2_1_5B"
-
-    def __str__(self) -> str:
-        return str(self.value)

orca_sdk/_generated_api_client/models/rac_head_type.py

@@ -1,11 +0,0 @@
-from enum import Enum
-
-
-class RACHeadType(str, Enum):
-    BMMOE = "BMMOE"
-    FF = "FF"
-    KNN = "KNN"
-    MMOE = "MMOE"
-
-    def __str__(self) -> str:
-        return str(self.value)

orca_sdk/_generated_api_client/models/rar_head_type.py

@@ -1,8 +0,0 @@
-from enum import Enum
-
-
-class RARHeadType(str, Enum):
-    MMOE = "MMOE"
-
-    def __str__(self) -> str:
-        return str(self.value)

orca_sdk/_generated_api_client/models/regression_evaluation_request.py

@@ -1,148 +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="RegressionEvaluationRequest")
-
-
-@_attrs_define
-class RegressionEvaluationRequest:
-    """
-    Attributes:
-        datasource_id (str):
-        datasource_score_column (str):
-        datasource_value_column (str):
-        memoryset_override_id (Union[None, Unset, str]):
-        record_telemetry (Union[Unset, bool]):  Default: False.
-        telemetry_tags (Union[List[str], None, Unset]):
-    """
-
-    datasource_id: str
-    datasource_score_column: str
-    datasource_value_column: str
-    memoryset_override_id: Union[None, Unset, str] = UNSET
-    record_telemetry: Union[Unset, bool] = False
-    telemetry_tags: Union[List[str], None, Unset] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        datasource_id = self.datasource_id
-
-        datasource_score_column = self.datasource_score_column
-
-        datasource_value_column = self.datasource_value_column
-
-        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
-
-        record_telemetry = self.record_telemetry
-
-        telemetry_tags: Union[List[str], None, Unset]
-        if isinstance(self.telemetry_tags, Unset):
-            telemetry_tags = UNSET
-        elif isinstance(self.telemetry_tags, list):
-            telemetry_tags = self.telemetry_tags
-
-        else:
-            telemetry_tags = self.telemetry_tags
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "datasource_id": datasource_id,
-                "datasource_score_column": datasource_score_column,
-                "datasource_value_column": datasource_value_column,
-            }
-        )
-        if memoryset_override_id is not UNSET:
-            field_dict["memoryset_override_id"] = memoryset_override_id
-        if record_telemetry is not UNSET:
-            field_dict["record_telemetry"] = record_telemetry
-        if telemetry_tags is not UNSET:
-            field_dict["telemetry_tags"] = telemetry_tags
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        datasource_id = d.pop("datasource_id")
-
-        datasource_score_column = d.pop("datasource_score_column")
-
-        datasource_value_column = d.pop("datasource_value_column")
-
-        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))
-
-        record_telemetry = d.pop("record_telemetry", UNSET)
-
-        def _parse_telemetry_tags(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()
-                telemetry_tags_type_0 = cast(List[str], data)
-
-                return telemetry_tags_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union[List[str], None, Unset], data)
-
-        telemetry_tags = _parse_telemetry_tags(d.pop("telemetry_tags", UNSET))
-
-        regression_evaluation_request = cls(
-            datasource_id=datasource_id,
-            datasource_score_column=datasource_score_column,
-            datasource_value_column=datasource_value_column,
-            memoryset_override_id=memoryset_override_id,
-            record_telemetry=record_telemetry,
-            telemetry_tags=telemetry_tags,
-        )
-
-        regression_evaluation_request.additional_properties = d
-        return regression_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/regression_metrics.py

@@ -1,172 +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, 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="RegressionMetrics")
-
-
-@_attrs_define
-class RegressionMetrics:
-    """
-    Attributes:
-        mse (float):
-        rmse (float):
-        mae (float):
-        r2 (float):
-        explained_variance (float):
-        loss (float):
-        anomaly_score_mean (Union[None, Unset, float]):
-        anomaly_score_median (Union[None, Unset, float]):
-        anomaly_score_variance (Union[None, Unset, float]):
-    """
-
-    mse: float
-    rmse: float
-    mae: float
-    r2: float
-    explained_variance: float
-    loss: float
-    anomaly_score_mean: Union[None, Unset, float] = UNSET
-    anomaly_score_median: Union[None, Unset, float] = UNSET
-    anomaly_score_variance: Union[None, Unset, float] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        mse = self.mse
-
-        rmse = self.rmse
-
-        mae = self.mae
-
-        r2 = self.r2
-
-        explained_variance = self.explained_variance
-
-        loss = self.loss
-
-        anomaly_score_mean: Union[None, Unset, float]
-        if isinstance(self.anomaly_score_mean, Unset):
-            anomaly_score_mean = UNSET
-        else:
-            anomaly_score_mean = self.anomaly_score_mean
-
-        anomaly_score_median: Union[None, Unset, float]
-        if isinstance(self.anomaly_score_median, Unset):
-            anomaly_score_median = UNSET
-        else:
-            anomaly_score_median = self.anomaly_score_median
-
-        anomaly_score_variance: Union[None, Unset, float]
-        if isinstance(self.anomaly_score_variance, Unset):
-            anomaly_score_variance = UNSET
-        else:
-            anomaly_score_variance = self.anomaly_score_variance
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "mse": mse,
-                "rmse": rmse,
-                "mae": mae,
-                "r2": r2,
-                "explained_variance": explained_variance,
-                "loss": loss,
-            }
-        )
-        if anomaly_score_mean is not UNSET:
-            field_dict["anomaly_score_mean"] = anomaly_score_mean
-        if anomaly_score_median is not UNSET:
-            field_dict["anomaly_score_median"] = anomaly_score_median
-        if anomaly_score_variance is not UNSET:
-            field_dict["anomaly_score_variance"] = anomaly_score_variance
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        mse = d.pop("mse")
-
-        rmse = d.pop("rmse")
-
-        mae = d.pop("mae")
-
-        r2 = d.pop("r2")
-
-        explained_variance = d.pop("explained_variance")
-
-        loss = d.pop("loss")
-
-        def _parse_anomaly_score_mean(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)
-
-        anomaly_score_mean = _parse_anomaly_score_mean(d.pop("anomaly_score_mean", UNSET))
-
-        def _parse_anomaly_score_median(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)
-
-        anomaly_score_median = _parse_anomaly_score_median(d.pop("anomaly_score_median", UNSET))
-
-        def _parse_anomaly_score_variance(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)
-
-        anomaly_score_variance = _parse_anomaly_score_variance(d.pop("anomaly_score_variance", UNSET))
-
-        regression_metrics = cls(
-            mse=mse,
-            rmse=rmse,
-            mae=mae,
-            r2=r2,
-            explained_variance=explained_variance,
-            loss=loss,
-            anomaly_score_mean=anomaly_score_mean,
-            anomaly_score_median=anomaly_score_median,
-            anomaly_score_variance=anomaly_score_variance,
-        )
-
-        regression_metrics.additional_properties = d
-        return regression_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/regression_model_metadata.py

@@ -1,177 +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 Any, 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.rar_head_type import RARHeadType
-
-T = TypeVar("T", bound="RegressionModelMetadata")
-
-
-@_attrs_define
-class RegressionModelMetadata:
-    """
-    Attributes:
-        id (str):
-        org_id (str):
-        name (str):
-        description (Union[None, str]):
-        version (int):
-        memoryset_id (str):
-        memory_lookup_count (int):
-        storage_path (str):
-        memoryset_collection_name (str):
-        created_at (datetime.datetime):
-        updated_at (datetime.datetime):
-        locked (bool):
-        head_type (RARHeadType):
-    """
-
-    id: str
-    org_id: str
-    name: str
-    description: Union[None, str]
-    version: int
-    memoryset_id: str
-    memory_lookup_count: int
-    storage_path: str
-    memoryset_collection_name: str
-    created_at: datetime.datetime
-    updated_at: datetime.datetime
-    locked: bool
-    head_type: RARHeadType
-    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
-
-        description: Union[None, str]
-        description = self.description
-
-        version = self.version
-
-        memoryset_id = self.memoryset_id
-
-        memory_lookup_count = self.memory_lookup_count
-
-        storage_path = self.storage_path
-
-        memoryset_collection_name = self.memoryset_collection_name
-
-        created_at = self.created_at.isoformat()
-
-        updated_at = self.updated_at.isoformat()
-
-        locked = self.locked
-
-        head_type = self.head_type.value if isinstance(self.head_type, Enum) else self.head_type
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "id": id,
-                "org_id": org_id,
-                "name": name,
-                "description": description,
-                "version": version,
-                "memoryset_id": memoryset_id,
-                "memory_lookup_count": memory_lookup_count,
-                "storage_path": storage_path,
-                "memoryset_collection_name": memoryset_collection_name,
-                "created_at": created_at,
-                "updated_at": updated_at,
-                "locked": locked,
-                "head_type": head_type,
-            }
-        )
-
-        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_description(data: object) -> Union[None, str]:
-            if data is None:
-                return data
-            return cast(Union[None, str], data)
-
-        description = _parse_description(d.pop("description"))
-
-        version = d.pop("version")
-
-        memoryset_id = d.pop("memoryset_id")
-
-        memory_lookup_count = d.pop("memory_lookup_count")
-
-        storage_path = d.pop("storage_path")
-
-        memoryset_collection_name = d.pop("memoryset_collection_name")
-
-        created_at = isoparse(d.pop("created_at"))
-
-        updated_at = isoparse(d.pop("updated_at"))
-
-        locked = d.pop("locked")
-
-        head_type = RARHeadType(d.pop("head_type"))
-
-        regression_model_metadata = cls(
-            id=id,
-            org_id=org_id,
-            name=name,
-            description=description,
-            version=version,
-            memoryset_id=memoryset_id,
-            memory_lookup_count=memory_lookup_count,
-            storage_path=storage_path,
-            memoryset_collection_name=memoryset_collection_name,
-            created_at=created_at,
-            updated_at=updated_at,
-            locked=locked,
-            head_type=head_type,
-        )
-
-        regression_model_metadata.additional_properties = d
-        return regression_model_metadata
-
-    @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