orca-sdk 0.1.0__py3-none-any.whl

orca_sdk/_generated_api_client/models/memoryset_analysis_request.py

@@ -0,0 +1,83 @@
+"""
+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

@@ -0,0 +1,9 @@
+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

@@ -0,0 +1,180 @@
+"""
+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

@@ -0,0 +1,66 @@
+"""
+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

@@ -0,0 +1,9 @@
+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

@@ -0,0 +1,100 @@
+"""
+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

@@ -0,0 +1,20 @@
+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

@@ -0,0 +1,157 @@
+"""
+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