orca-sdk 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

orca_sdk/_generated_api_client/models/memoryset_analysis_request.py

@@ -1,83 +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.memoryset_analysis_request_type import MemorysetAnalysisRequestType
-from ..types import UNSET, Unset
-
-T = TypeVar("T", bound="MemorysetAnalysisRequest")
-
-
-@_attrs_define
-class MemorysetAnalysisRequest:
-    """
-    Attributes:
-        type (MemorysetAnalysisRequestType):
-        neighbor_count (Union[Unset, int]):  Default: 10.
-    """
-
-    type: MemorysetAnalysisRequestType
-    neighbor_count: Union[Unset, int] = 10
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        type = self.type.value if isinstance(self.type, Enum) else self.type
-
-        neighbor_count = self.neighbor_count
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "type": type,
-            }
-        )
-        if neighbor_count is not UNSET:
-            field_dict["neighbor_count"] = neighbor_count
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        type = MemorysetAnalysisRequestType(d.pop("type"))
-
-        neighbor_count = d.pop("neighbor_count", UNSET)
-
-        memoryset_analysis_request = cls(
-            type=type,
-            neighbor_count=neighbor_count,
-        )
-
-        memoryset_analysis_request.additional_properties = d
-        return memoryset_analysis_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/memoryset_analysis_request_type.py

@@ -1,9 +0,0 @@
-from enum import Enum
-
-
-class MemorysetAnalysisRequestType(str, Enum):
-    ANALYZE_DUPLICATE_MEMORIES = "ANALYZE_DUPLICATE_MEMORIES"
-    ANALYZE_MEMORY_NEIGHBOR_LABELS = "ANALYZE_MEMORY_NEIGHBOR_LABELS"
-
-    def __str__(self) -> str:
-        return str(self.value)

orca_sdk/_generated_api_client/models/memoryset_analysis_response.py

@@ -1,180 +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.memoryset_analysis_response_type import MemorysetAnalysisResponseType
-from ..models.task_status import TaskStatus
-
-if TYPE_CHECKING:
-    from ..models.analyze_neighbor_labels_result import AnalyzeNeighborLabelsResult
-    from ..models.find_duplicates_analysis_result import FindDuplicatesAnalysisResult
-    from ..models.memoryset_analysis_response_config import MemorysetAnalysisResponseConfig
-
-
-T = TypeVar("T", bound="MemorysetAnalysisResponse")
-
-
-@_attrs_define
-class MemorysetAnalysisResponse:
-    """
-    Attributes:
-        task_id (str):
-        org_id (str):
-        memoryset_id (str):
-        type (MemorysetAnalysisResponseType):
-        status (TaskStatus): Status of task in the task queue
-        config (MemorysetAnalysisResponseConfig):
-        result (Union['AnalyzeNeighborLabelsResult', 'FindDuplicatesAnalysisResult', None]):
-        created_at (datetime.datetime):
-        updated_at (datetime.datetime):
-    """
-
-    task_id: str
-    org_id: str
-    memoryset_id: str
-    type: MemorysetAnalysisResponseType
-    status: TaskStatus
-    config: "MemorysetAnalysisResponseConfig"
-    result: Union["AnalyzeNeighborLabelsResult", "FindDuplicatesAnalysisResult", 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.analyze_neighbor_labels_result import AnalyzeNeighborLabelsResult
-        from ..models.find_duplicates_analysis_result import FindDuplicatesAnalysisResult
-
-        task_id = self.task_id
-
-        org_id = self.org_id
-
-        memoryset_id = self.memoryset_id
-
-        type = self.type.value if isinstance(self.type, Enum) else self.type
-
-        status = self.status.value if isinstance(self.status, Enum) else self.status
-
-        config = self.config.to_dict()
-
-        result: Union[Dict[str, Any], None]
-        if isinstance(self.result, AnalyzeNeighborLabelsResult):
-            result = self.result.to_dict()
-        elif isinstance(self.result, FindDuplicatesAnalysisResult):
-            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,
-                "memoryset_id": memoryset_id,
-                "type": type,
-                "status": status,
-                "config": config,
-                "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.analyze_neighbor_labels_result import AnalyzeNeighborLabelsResult
-        from ..models.find_duplicates_analysis_result import FindDuplicatesAnalysisResult
-        from ..models.memoryset_analysis_response_config import MemorysetAnalysisResponseConfig
-
-        d = src_dict.copy()
-        task_id = d.pop("task_id")
-
-        org_id = d.pop("org_id")
-
-        memoryset_id = d.pop("memoryset_id")
-
-        type = MemorysetAnalysisResponseType(d.pop("type"))
-
-        status = TaskStatus(d.pop("status"))
-
-        config = MemorysetAnalysisResponseConfig.from_dict(d.pop("config"))
-
-        def _parse_result(data: object) -> Union["AnalyzeNeighborLabelsResult", "FindDuplicatesAnalysisResult", None]:
-            if data is None:
-                return data
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                result_type_0 = AnalyzeNeighborLabelsResult.from_dict(data)
-
-                return result_type_0
-            except:  # noqa: E722
-                pass
-            try:
-                if not isinstance(data, dict):
-                    raise TypeError()
-                result_type_1 = FindDuplicatesAnalysisResult.from_dict(data)
-
-                return result_type_1
-            except:  # noqa: E722
-                pass
-            return cast(Union["AnalyzeNeighborLabelsResult", "FindDuplicatesAnalysisResult", None], data)
-
-        result = _parse_result(d.pop("result"))
-
-        created_at = isoparse(d.pop("created_at"))
-
-        updated_at = isoparse(d.pop("updated_at"))
-
-        memoryset_analysis_response = cls(
-            task_id=task_id,
-            org_id=org_id,
-            memoryset_id=memoryset_id,
-            type=type,
-            status=status,
-            config=config,
-            result=result,
-            created_at=created_at,
-            updated_at=updated_at,
-        )
-
-        memoryset_analysis_response.additional_properties = d
-        return memoryset_analysis_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/memoryset_analysis_response_config.py

@@ -1,66 +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
-
-T = TypeVar("T", bound="MemorysetAnalysisResponseConfig")
-
-
-@_attrs_define
-class MemorysetAnalysisResponseConfig:
-    """ """
-
-    additional_properties: dict[str, Union[float, int, str]] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        field_dict: dict[str, Any] = {}
-        for prop_name, prop in self.additional_properties.items():
-            field_dict[prop_name] = prop
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        memoryset_analysis_response_config = cls()
-
-        additional_properties = {}
-        for prop_name, prop_dict in d.items():
-
-            def _parse_additional_property(data: object) -> Union[float, int, str]:
-                return cast(Union[float, int, str], data)
-
-            additional_property = _parse_additional_property(prop_dict)
-
-            additional_properties[prop_name] = additional_property
-
-        memoryset_analysis_response_config.additional_properties = additional_properties
-        return memoryset_analysis_response_config
-
-    @property
-    def additional_keys(self) -> list[str]:
-        return list(self.additional_properties.keys())
-
-    def __getitem__(self, key: str) -> Union[float, int, str]:
-        return self.additional_properties[key]
-
-    def __setitem__(self, key: str, value: Union[float, int, str]) -> 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_analysis_response_type.py

@@ -1,9 +0,0 @@
-from enum import Enum
-
-
-class MemorysetAnalysisResponseType(str, Enum):
-    ANALYZE_DUPLICATE_MEMORIES = "ANALYZE_DUPLICATE_MEMORIES"
-    ANALYZE_MEMORY_NEIGHBOR_LABELS = "ANALYZE_MEMORY_NEIGHBOR_LABELS"
-
-    def __str__(self) -> str:
-        return str(self.value)

orca_sdk/_generated_api_client/models/not_found_error_response.py

@@ -1,100 +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, Literal, Type, TypeVar, Union, cast
-
-from attrs import define as _attrs_define
-from attrs import field as _attrs_field
-
-from ..models.not_found_error_response_resource_type_0 import NotFoundErrorResponseResourceType0
-
-T = TypeVar("T", bound="NotFoundErrorResponse")
-
-
-@_attrs_define
-class NotFoundErrorResponse:
-    """
-    Attributes:
-        status_code (Literal[404]):
-        resource (Union[None, NotFoundErrorResponseResourceType0]):
-    """
-
-    status_code: Literal[404]
-    resource: Union[None, NotFoundErrorResponseResourceType0]
-    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
-
-    def to_dict(self) -> dict[str, Any]:
-        status_code = self.status_code
-
-        resource: Union[None, str]
-        if isinstance(self.resource, NotFoundErrorResponseResourceType0):
-            resource = self.resource.value if isinstance(self.resource, Enum) else self.resource
-        else:
-            resource = self.resource
-
-        field_dict: dict[str, Any] = {}
-        field_dict.update(self.additional_properties)
-        field_dict.update(
-            {
-                "status_code": status_code,
-                "resource": resource,
-            }
-        )
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        status_code = cast(Literal[404], d.pop("status_code"))
-        if status_code != 404:
-            raise ValueError(f"status_code must match const 404, got '{status_code}'")
-
-        def _parse_resource(data: object) -> Union[None, NotFoundErrorResponseResourceType0]:
-            if data is None:
-                return data
-            try:
-                if not isinstance(data, str):
-                    raise TypeError()
-                resource_type_0 = NotFoundErrorResponseResourceType0(data)
-
-                return resource_type_0
-            except:  # noqa: E722
-                pass
-            return cast(Union[None, NotFoundErrorResponseResourceType0], data)
-
-        resource = _parse_resource(d.pop("resource"))
-
-        not_found_error_response = cls(
-            status_code=status_code,
-            resource=resource,
-        )
-
-        not_found_error_response.additional_properties = d
-        return not_found_error_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/not_found_error_response_resource_type_0.py

@@ -1,20 +0,0 @@
-from enum import Enum
-
-
-class NotFoundErrorResponseResourceType0(str, Enum):
-    ANALYSIS = "analysis"
-    API_KEY = "api_key"
-    CLASSIFICATION_MODEL = "classification_model"
-    DATASOURCE = "datasource"
-    EVALUATION = "evaluation"
-    FEEDBACK_CATEGORY = "feedback_category"
-    FINETUNED_EMBEDDING_MODEL = "finetuned_embedding_model"
-    MEMORY = "memory"
-    MEMORYSET = "memoryset"
-    ORG = "org"
-    PREDICTION = "prediction"
-    PRETRAINED_EMBEDDING_MODEL = "pretrained_embedding_model"
-    TASK = "task"
-
-    def __str__(self) -> str:
-        return str(self.value)

orca_sdk/_generated_api_client/models/prediction_feedback.py

@@ -1,157 +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.feedback_type import FeedbackType
-
-T = TypeVar("T", bound="PredictionFeedback")
-
-
-@_attrs_define
-class PredictionFeedback:
-    """
-    Attributes:
-        prediction_id (str):
-        category_name (str):
-        value (Union[bool, float]):
-        comment (Union[None, str]):
-        id (str):
-        org_id (str):
-        category_id (str):
-        category_type (FeedbackType):
-        created_at (datetime.datetime):
-        updated_at (datetime.datetime):
-    """
-
-    prediction_id: str
-    category_name: str
-    value: Union[bool, float]
-    comment: Union[None, str]
-    id: str
-    org_id: str
-    category_id: str
-    category_type: FeedbackType
-    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]:
-        prediction_id = self.prediction_id
-
-        category_name = self.category_name
-
-        value: Union[bool, float]
-        value = self.value
-
-        comment: Union[None, str]
-        comment = self.comment
-
-        id = self.id
-
-        org_id = self.org_id
-
-        category_id = self.category_id
-
-        category_type = self.category_type.value if isinstance(self.category_type, Enum) else self.category_type
-
-        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(
-            {
-                "prediction_id": prediction_id,
-                "category_name": category_name,
-                "value": value,
-                "comment": comment,
-                "id": id,
-                "org_id": org_id,
-                "category_id": category_id,
-                "category_type": category_type,
-                "created_at": created_at,
-                "updated_at": updated_at,
-            }
-        )
-
-        return field_dict
-
-    @classmethod
-    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
-        d = src_dict.copy()
-        prediction_id = d.pop("prediction_id")
-
-        category_name = d.pop("category_name")
-
-        def _parse_value(data: object) -> Union[bool, float]:
-            return cast(Union[bool, float], data)
-
-        value = _parse_value(d.pop("value"))
-
-        def _parse_comment(data: object) -> Union[None, str]:
-            if data is None:
-                return data
-            return cast(Union[None, str], data)
-
-        comment = _parse_comment(d.pop("comment"))
-
-        id = d.pop("id")
-
-        org_id = d.pop("org_id")
-
-        category_id = d.pop("category_id")
-
-        category_type = FeedbackType(d.pop("category_type"))
-
-        created_at = isoparse(d.pop("created_at"))
-
-        updated_at = isoparse(d.pop("updated_at"))
-
-        prediction_feedback = cls(
-            prediction_id=prediction_id,
-            category_name=category_name,
-            value=value,
-            comment=comment,
-            id=id,
-            org_id=org_id,
-            category_id=category_id,
-            category_type=category_type,
-            created_at=created_at,
-            updated_at=updated_at,
-        )
-
-        prediction_feedback.additional_properties = d
-        return prediction_feedback
-
-    @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