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

orca_sdk/_generated_api_client/models/memoryset_metadata.py

@@ -1,333 +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, List, 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.memory_type import MemoryType
-from ..models.memoryset_metadata_index_type import MemorysetMetadataIndexType
-from ..models.pretrained_embedding_model_name import PretrainedEmbeddingModelName
-from ..models.task_status import TaskStatus
-
-if TYPE_CHECKING:
-    from ..models.memoryset_metadata_index_params import MemorysetMetadataIndexParams
-    from ..models.memoryset_metrics import MemorysetMetrics
-
-
-T = TypeVar("T", bound="MemorysetMetadata")
-
-
-@_attrs_define
-class MemorysetMetadata:
-    """
-    Attributes:
-        id (str):
-        org_id (str):
-        collection_name (str):
-        name (str):
-        description (Union[None, str]):
-        notes (Union[None, str]):
-        length (int):
-        pretrained_embedding_model_name (Union[None, PretrainedEmbeddingModelName]):
-        finetuned_embedding_model_id (Union[None, str]):
-        created_at (datetime.datetime):
-        updated_at (datetime.datetime):
-        memories_updated_at (datetime.datetime):
-        insertion_task_id (str):
-        insertion_status (TaskStatus): Status of task in the task queue
-        metrics (MemorysetMetrics):
-        memory_type (MemoryType):
-        label_names (Union[List[str], None]):
-        index_type (MemorysetMetadataIndexType):
-        index_params (MemorysetMetadataIndexParams):
-        database_uri (Union[None, str]):
-        document_prompt_override (Union[None, str]):
-        query_prompt_override (Union[None, str]):
-    """
-
-    id: str
-    org_id: str
-    collection_name: str
-    name: str
-    description: Union[None, str]
-    notes: Union[None, str]
-    length: int
-    pretrained_embedding_model_name: Union[None, PretrainedEmbeddingModelName]
-    finetuned_embedding_model_id: Union[None, str]
-    created_at: datetime.datetime
-    updated_at: datetime.datetime
-    memories_updated_at: datetime.datetime
-    insertion_task_id: str
-    insertion_status: TaskStatus
-    metrics: "MemorysetMetrics"
-    memory_type: MemoryType
-    label_names: Union[List[str], None]
-    index_type: MemorysetMetadataIndexType
-    index_params: "MemorysetMetadataIndexParams"
-    database_uri: Union[None, str]
-    document_prompt_override: Union[None, str]
-    query_prompt_override: Union[None, str]
-    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
-
-        collection_name = self.collection_name
-
-        name = self.name
-
-        description: Union[None, str]
-        description = self.description
-
-        notes: Union[None, str]
-        notes = self.notes
-
-        length = self.length
-
-        pretrained_embedding_model_name: Union[None, str]
-        if 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, str]
-        finetuned_embedding_model_id = self.finetuned_embedding_model_id
-
-        created_at = self.created_at.isoformat()
-
-        updated_at = self.updated_at.isoformat()
-
-        memories_updated_at = self.memories_updated_at.isoformat()
-
-        insertion_task_id = self.insertion_task_id
-
-        insertion_status = (
-            self.insertion_status.value if isinstance(self.insertion_status, Enum) else self.insertion_status
-        )
-
-        metrics = self.metrics.to_dict()
-
-        memory_type = self.memory_type.value if isinstance(self.memory_type, Enum) else self.memory_type
-
-        label_names: Union[List[str], None]
-        if isinstance(self.label_names, list):
-            label_names = self.label_names
-
-        else:
-            label_names = self.label_names
-
-        index_type = self.index_type.value if isinstance(self.index_type, Enum) else self.index_type
-
-        index_params = self.index_params.to_dict()
-
-        database_uri: Union[None, str]
-        database_uri = self.database_uri
-
-        document_prompt_override: Union[None, str]
-        document_prompt_override = self.document_prompt_override
-
-        query_prompt_override: Union[None, str]
-        query_prompt_override = self.query_prompt_override
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "id": id,
-                "org_id": org_id,
-                "collection_name": collection_name,
-                "name": name,
-                "description": description,
-                "notes": notes,
-                "length": length,
-                "pretrained_embedding_model_name": pretrained_embedding_model_name,
-                "finetuned_embedding_model_id": finetuned_embedding_model_id,
-                "created_at": created_at,
-                "updated_at": updated_at,
-                "memories_updated_at": memories_updated_at,
-                "insertion_task_id": insertion_task_id,
-                "insertion_status": insertion_status,
-                "metrics": metrics,
-                "memory_type": memory_type,
-                "label_names": label_names,
-                "index_type": index_type,
-                "index_params": index_params,
-                "database_uri": database_uri,
-                "document_prompt_override": document_prompt_override,
-                "query_prompt_override": query_prompt_override,
-            }
-        )
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.memoryset_metadata_index_params import MemorysetMetadataIndexParams
-        from ..models.memoryset_metrics import MemorysetMetrics
-
-        d = src_dict.copy()
-        id = d.pop("id")
-
-        org_id = d.pop("org_id")
-
-        collection_name = d.pop("collection_name")
-
-        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"))
-
-        def _parse_notes(data: object) -> Union[None, str]:
-            if data is None:
-                return data
-            return cast(Union[None, str], data)
-
-        notes = _parse_notes(d.pop("notes"))
-
-        length = d.pop("length")
-
-        def _parse_pretrained_embedding_model_name(data: object) -> Union[None, PretrainedEmbeddingModelName]:
-            if data is None:
-                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], data)
-
-        pretrained_embedding_model_name = _parse_pretrained_embedding_model_name(
-            d.pop("pretrained_embedding_model_name")
-        )
-
-        def _parse_finetuned_embedding_model_id(data: object) -> Union[None, str]:
-            if data is None:
-                return data
-            return cast(Union[None, str], data)
-
-        finetuned_embedding_model_id = _parse_finetuned_embedding_model_id(d.pop("finetuned_embedding_model_id"))
-
-        created_at = isoparse(d.pop("created_at"))
-
-        updated_at = isoparse(d.pop("updated_at"))
-
-        memories_updated_at = isoparse(d.pop("memories_updated_at"))
-
-        insertion_task_id = d.pop("insertion_task_id")
-
-        insertion_status = TaskStatus(d.pop("insertion_status"))
-
-        metrics = MemorysetMetrics.from_dict(d.pop("metrics"))
-
-        memory_type = MemoryType(d.pop("memory_type"))
-
-        def _parse_label_names(data: object) -> Union[List[str], None]:
-            if data is None:
-                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], data)
-
-        label_names = _parse_label_names(d.pop("label_names"))
-
-        index_type = MemorysetMetadataIndexType(d.pop("index_type"))
-
-        index_params = MemorysetMetadataIndexParams.from_dict(d.pop("index_params"))
-
-        def _parse_database_uri(data: object) -> Union[None, str]:
-            if data is None:
-                return data
-            return cast(Union[None, str], data)
-
-        database_uri = _parse_database_uri(d.pop("database_uri"))
-
-        def _parse_document_prompt_override(data: object) -> Union[None, str]:
-            if data is None:
-                return data
-            return cast(Union[None, str], data)
-
-        document_prompt_override = _parse_document_prompt_override(d.pop("document_prompt_override"))
-
-        def _parse_query_prompt_override(data: object) -> Union[None, str]:
-            if data is None:
-                return data
-            return cast(Union[None, str], data)
-
-        query_prompt_override = _parse_query_prompt_override(d.pop("query_prompt_override"))
-
-        memoryset_metadata = cls(
-            id=id,
-            org_id=org_id,
-            collection_name=collection_name,
-            name=name,
-            description=description,
-            notes=notes,
-            length=length,
-            pretrained_embedding_model_name=pretrained_embedding_model_name,
-            finetuned_embedding_model_id=finetuned_embedding_model_id,
-            created_at=created_at,
-            updated_at=updated_at,
-            memories_updated_at=memories_updated_at,
-            insertion_task_id=insertion_task_id,
-            insertion_status=insertion_status,
-            metrics=metrics,
-            memory_type=memory_type,
-            label_names=label_names,
-            index_type=index_type,
-            index_params=index_params,
-            database_uri=database_uri,
-            document_prompt_override=document_prompt_override,
-            query_prompt_override=query_prompt_override,
-        )
-
-        memoryset_metadata.additional_properties = d
-        return memoryset_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/memoryset_metadata_index_params.py

@@ -1,55 +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="MemorysetMetadataIndexParams")
-
-
-@_attrs_define
-class MemorysetMetadataIndexParams:
-    """ """
-
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        memoryset_metadata_index_params = cls()
-
-        memoryset_metadata_index_params.additional_properties = d
-        return memoryset_metadata_index_params
-
-    @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/memoryset_metadata_index_type.py

@@ -1,13 +0,0 @@
-from enum import Enum
-
-
-class MemorysetMetadataIndexType(str, Enum):
-    DISKANN = "DISKANN"
-    FLAT = "FLAT"
-    HNSW = "HNSW"
-    IVF_FLAT = "IVF_FLAT"
-    IVF_PQ = "IVF_PQ"
-    IVF_SQ8 = "IVF_SQ8"
-
-    def __str__(self) -> str:
-        return str(self.value)

orca_sdk/_generated_api_client/models/memoryset_metrics.py

@@ -1,265 +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, 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.memoryset_class_patterns_metrics import MemorysetClassPatternsMetrics
-    from ..models.memoryset_cluster_metrics import MemorysetClusterMetrics
-    from ..models.memoryset_duplicate_metrics import MemorysetDuplicateMetrics
-    from ..models.memoryset_label_metrics import MemorysetLabelMetrics
-    from ..models.memoryset_neighbor_metrics import MemorysetNeighborMetrics
-    from ..models.memoryset_projection_metrics import MemorysetProjectionMetrics
-
-
-T = TypeVar("T", bound="MemorysetMetrics")
-
-
-@_attrs_define
-class MemorysetMetrics:
-    """
-    Attributes:
-        neighbor (Union['MemorysetNeighborMetrics', None, Unset]):
-        label (Union['MemorysetLabelMetrics', None, Unset]):
-        duplicate (Union['MemorysetDuplicateMetrics', None, Unset]):
-        projection (Union['MemorysetProjectionMetrics', None, Unset]):
-        cluster (Union['MemorysetClusterMetrics', None, Unset]):
-        class_patterns (Union['MemorysetClassPatternsMetrics', None, Unset]):
-    """
-
-    neighbor: Union["MemorysetNeighborMetrics", None, Unset] = UNSET
-    label: Union["MemorysetLabelMetrics", None, Unset] = UNSET
-    duplicate: Union["MemorysetDuplicateMetrics", None, Unset] = UNSET
-    projection: Union["MemorysetProjectionMetrics", None, Unset] = UNSET
-    cluster: Union["MemorysetClusterMetrics", None, Unset] = UNSET
-    class_patterns: Union["MemorysetClassPatternsMetrics", None, Unset] = UNSET
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        from ..models.memoryset_class_patterns_metrics import MemorysetClassPatternsMetrics
-        from ..models.memoryset_cluster_metrics import MemorysetClusterMetrics
-        from ..models.memoryset_duplicate_metrics import MemorysetDuplicateMetrics
-        from ..models.memoryset_label_metrics import MemorysetLabelMetrics
-        from ..models.memoryset_neighbor_metrics import MemorysetNeighborMetrics
-        from ..models.memoryset_projection_metrics import MemorysetProjectionMetrics
-
-        neighbor: Union[Dict[str, Any], None, Unset]
-        if isinstance(self.neighbor, Unset):
-            neighbor = UNSET
-        elif isinstance(self.neighbor, MemorysetNeighborMetrics):
-            neighbor = self.neighbor.to_dict()
-        else:
-            neighbor = self.neighbor
-
-        label: Union[Dict[str, Any], None, Unset]
-        if isinstance(self.label, Unset):
-            label = UNSET
-        elif isinstance(self.label, MemorysetLabelMetrics):
-            label = self.label.to_dict()
-        else:
-            label = self.label
-
-        duplicate: Union[Dict[str, Any], None, Unset]
-        if isinstance(self.duplicate, Unset):
-            duplicate = UNSET
-        elif isinstance(self.duplicate, MemorysetDuplicateMetrics):
-            duplicate = self.duplicate.to_dict()
-        else:
-            duplicate = self.duplicate
-
-        projection: Union[Dict[str, Any], None, Unset]
-        if isinstance(self.projection, Unset):
-            projection = UNSET
-        elif isinstance(self.projection, MemorysetProjectionMetrics):
-            projection = self.projection.to_dict()
-        else:
-            projection = self.projection
-
-        cluster: Union[Dict[str, Any], None, Unset]
-        if isinstance(self.cluster, Unset):
-            cluster = UNSET
-        elif isinstance(self.cluster, MemorysetClusterMetrics):
-            cluster = self.cluster.to_dict()
-        else:
-            cluster = self.cluster
-
-        class_patterns: Union[Dict[str, Any], None, Unset]
-        if isinstance(self.class_patterns, Unset):
-            class_patterns = UNSET
-        elif isinstance(self.class_patterns, MemorysetClassPatternsMetrics):
-            class_patterns = self.class_patterns.to_dict()
-        else:
-            class_patterns = self.class_patterns
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update({})
-        if neighbor is not UNSET:
-            field_dict["neighbor"] = neighbor
-        if label is not UNSET:
-            field_dict["label"] = label
-        if duplicate is not UNSET:
-            field_dict["duplicate"] = duplicate
-        if projection is not UNSET:
-            field_dict["projection"] = projection
-        if cluster is not UNSET:
-            field_dict["cluster"] = cluster
-        if class_patterns is not UNSET:
-            field_dict["class_patterns"] = class_patterns
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        from ..models.memoryset_class_patterns_metrics import MemorysetClassPatternsMetrics
-        from ..models.memoryset_cluster_metrics import MemorysetClusterMetrics
-        from ..models.memoryset_duplicate_metrics import MemorysetDuplicateMetrics
-        from ..models.memoryset_label_metrics import MemorysetLabelMetrics
-        from ..models.memoryset_neighbor_metrics import MemorysetNeighborMetrics
-        from ..models.memoryset_projection_metrics import MemorysetProjectionMetrics
-
-        d = src_dict.copy()
-
-        def _parse_neighbor(data: object) -> Union["MemorysetNeighborMetrics", None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                neighbor_type_0 = MemorysetNeighborMetrics.from_dict(data)
-
-                return neighbor_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["MemorysetNeighborMetrics", None, Unset], data)
-
-        neighbor = _parse_neighbor(d.pop("neighbor", UNSET))
-
-        def _parse_label(data: object) -> Union["MemorysetLabelMetrics", None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                label_type_0 = MemorysetLabelMetrics.from_dict(data)
-
-                return label_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["MemorysetLabelMetrics", None, Unset], data)
-
-        label = _parse_label(d.pop("label", UNSET))
-
-        def _parse_duplicate(data: object) -> Union["MemorysetDuplicateMetrics", None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                duplicate_type_0 = MemorysetDuplicateMetrics.from_dict(data)
-
-                return duplicate_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["MemorysetDuplicateMetrics", None, Unset], data)
-
-        duplicate = _parse_duplicate(d.pop("duplicate", UNSET))
-
-        def _parse_projection(data: object) -> Union["MemorysetProjectionMetrics", None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                projection_type_0 = MemorysetProjectionMetrics.from_dict(data)
-
-                return projection_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["MemorysetProjectionMetrics", None, Unset], data)
-
-        projection = _parse_projection(d.pop("projection", UNSET))
-
-        def _parse_cluster(data: object) -> Union["MemorysetClusterMetrics", None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                cluster_type_0 = MemorysetClusterMetrics.from_dict(data)
-
-                return cluster_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["MemorysetClusterMetrics", None, Unset], data)
-
-        cluster = _parse_cluster(d.pop("cluster", UNSET))
-
-        def _parse_class_patterns(data: object) -> Union["MemorysetClassPatternsMetrics", None, Unset]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                class_patterns_type_0 = MemorysetClassPatternsMetrics.from_dict(data)
-
-                return class_patterns_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union["MemorysetClassPatternsMetrics", None, Unset], data)
-
-        class_patterns = _parse_class_patterns(d.pop("class_patterns", UNSET))
-
-        memoryset_metrics = cls(
-            neighbor=neighbor,
-            label=label,
-            duplicate=duplicate,
-            projection=projection,
-            cluster=cluster,
-            class_patterns=class_patterns,
-        )
-
-        memoryset_metrics.additional_properties = d
-        return memoryset_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