orca-sdk 0.1.0__py3-none-any.whl

orca_sdk/_generated_api_client/models/rac_model_metadata.py

@@ -0,0 +1,191 @@
+"""
+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.rac_head_type import RACHeadType
+
+T = TypeVar("T", bound="RACModelMetadata")
+
+
+@_attrs_define
+class RACModelMetadata:
+    """
+    Attributes:
+        id (str):
+        org_id (str):
+        name (str):
+        version (int):
+        num_classes (int):
+        head_type (RACHeadType):
+        memoryset_id (str):
+        memory_lookup_count (int):
+        weigh_memories (Union[None, bool]):
+        min_memory_weight (Union[None, float]):
+        storage_path (str):
+        memoryset_collection_name (str):
+        created_at (datetime.datetime):
+        updated_at (datetime.datetime):
+    """
+
+    id: str
+    org_id: str
+    name: str
+    version: int
+    num_classes: int
+    head_type: RACHeadType
+    memoryset_id: str
+    memory_lookup_count: int
+    weigh_memories: Union[None, bool]
+    min_memory_weight: Union[None, float]
+    storage_path: str
+    memoryset_collection_name: str
+    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]:
+        id = self.id
+
+        org_id = self.org_id
+
+        name = self.name
+
+        version = self.version
+
+        num_classes = self.num_classes
+
+        head_type = self.head_type.value if isinstance(self.head_type, Enum) else self.head_type
+
+        memoryset_id = self.memoryset_id
+
+        memory_lookup_count = self.memory_lookup_count
+
+        weigh_memories: Union[None, bool]
+        weigh_memories = self.weigh_memories
+
+        min_memory_weight: Union[None, float]
+        min_memory_weight = self.min_memory_weight
+
+        storage_path = self.storage_path
+
+        memoryset_collection_name = self.memoryset_collection_name
+
+        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(
+            {
+                "id": id,
+                "org_id": org_id,
+                "name": name,
+                "version": version,
+                "num_classes": num_classes,
+                "head_type": head_type,
+                "memoryset_id": memoryset_id,
+                "memory_lookup_count": memory_lookup_count,
+                "weigh_memories": weigh_memories,
+                "min_memory_weight": min_memory_weight,
+                "storage_path": storage_path,
+                "memoryset_collection_name": memoryset_collection_name,
+                "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()
+        id = d.pop("id")
+
+        org_id = d.pop("org_id")
+
+        name = d.pop("name")
+
+        version = d.pop("version")
+
+        num_classes = d.pop("num_classes")
+
+        head_type = RACHeadType(d.pop("head_type"))
+
+        memoryset_id = d.pop("memoryset_id")
+
+        memory_lookup_count = d.pop("memory_lookup_count")
+
+        def _parse_weigh_memories(data: object) -> Union[None, bool]:
+            if data is None:
+                return data
+            return cast(Union[None, bool], data)
+
+        weigh_memories = _parse_weigh_memories(d.pop("weigh_memories"))
+
+        def _parse_min_memory_weight(data: object) -> Union[None, float]:
+            if data is None:
+                return data
+            return cast(Union[None, float], data)
+
+        min_memory_weight = _parse_min_memory_weight(d.pop("min_memory_weight"))
+
+        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"))
+
+        rac_model_metadata = cls(
+            id=id,
+            org_id=org_id,
+            name=name,
+            version=version,
+            num_classes=num_classes,
+            head_type=head_type,
+            memoryset_id=memoryset_id,
+            memory_lookup_count=memory_lookup_count,
+            weigh_memories=weigh_memories,
+            min_memory_weight=min_memory_weight,
+            storage_path=storage_path,
+            memoryset_collection_name=memoryset_collection_name,
+            created_at=created_at,
+            updated_at=updated_at,
+        )
+
+        rac_model_metadata.additional_properties = d
+        return rac_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/service_unavailable_error_response.py

@@ -0,0 +1,80 @@
+"""
+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, Literal, Type, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ServiceUnavailableErrorResponse")
+
+
+@_attrs_define
+class ServiceUnavailableErrorResponse:
+    """
+    Attributes:
+        status_code (Literal[503]):
+        service (str):
+    """
+
+    status_code: Literal[503]
+    service: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        status_code = self.status_code
+
+        service = self.service
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "status_code": status_code,
+                "service": service,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        status_code = cast(Literal[503], d.pop("status_code"))
+        if status_code != 503:
+            raise ValueError(f"status_code must match const 503, got '{status_code}'")
+
+        service = d.pop("service")
+
+        service_unavailable_error_response = cls(
+            status_code=status_code,
+            service=service,
+        )
+
+        service_unavailable_error_response.additional_properties = d
+        return service_unavailable_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/task.py

@@ -0,0 +1,198 @@
+"""
+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.base_model import BaseModel
+
+
+T = TypeVar("T", bound="Task")
+
+
+@_attrs_define
+class Task:
+    """
+    Attributes:
+        status (TaskStatus): Status of task in the task queue
+        steps_total (Union[None, int]):
+        steps_completed (Union[None, int]):
+        exception (Union[None, str]):
+        updated_at (datetime.datetime):
+        created_at (datetime.datetime):
+        id (str):
+        org_id (str):
+        type (str):
+        payload (BaseModel):
+        result (Union['BaseModel', None]):
+    """
+
+    status: TaskStatus
+    steps_total: Union[None, int]
+    steps_completed: Union[None, int]
+    exception: Union[None, str]
+    updated_at: datetime.datetime
+    created_at: datetime.datetime
+    id: str
+    org_id: str
+    type: str
+    payload: "BaseModel"
+    result: Union["BaseModel", None]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        from ..models.base_model import BaseModel
+
+        status = self.status.value if isinstance(self.status, Enum) else self.status
+
+        steps_total: Union[None, int]
+        steps_total = self.steps_total
+
+        steps_completed: Union[None, int]
+        steps_completed = self.steps_completed
+
+        exception: Union[None, str]
+        exception = self.exception
+
+        updated_at = self.updated_at.isoformat()
+
+        created_at = self.created_at.isoformat()
+
+        id = self.id
+
+        org_id = self.org_id
+
+        type = self.type
+
+        payload = self.payload.to_dict()
+
+        result: Union[Dict[str, Any], None]
+        if isinstance(self.result, BaseModel):
+            result = self.result.to_dict()
+        else:
+            result = self.result
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "status": status,
+                "steps_total": steps_total,
+                "steps_completed": steps_completed,
+                "exception": exception,
+                "updated_at": updated_at,
+                "created_at": created_at,
+                "id": id,
+                "org_id": org_id,
+                "type": type,
+                "payload": payload,
+                "result": result,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.base_model import BaseModel
+
+        d = src_dict.copy()
+        status = TaskStatus(d.pop("status"))
+
+        def _parse_steps_total(data: object) -> Union[None, int]:
+            if data is None:
+                return data
+            return cast(Union[None, int], data)
+
+        steps_total = _parse_steps_total(d.pop("steps_total"))
+
+        def _parse_steps_completed(data: object) -> Union[None, int]:
+            if data is None:
+                return data
+            return cast(Union[None, int], data)
+
+        steps_completed = _parse_steps_completed(d.pop("steps_completed"))
+
+        def _parse_exception(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        exception = _parse_exception(d.pop("exception"))
+
+        updated_at = isoparse(d.pop("updated_at"))
+
+        created_at = isoparse(d.pop("created_at"))
+
+        id = d.pop("id")
+
+        org_id = d.pop("org_id")
+
+        type = d.pop("type")
+
+        payload = BaseModel.from_dict(d.pop("payload"))
+
+        def _parse_result(data: object) -> Union["BaseModel", None]:
+            if data is None:
+                return data
+            try:
+                if not isinstance(data, dict):
+                    raise TypeError()
+                result_type_0 = BaseModel.from_dict(data)
+
+                return result_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union["BaseModel", None], data)
+
+        result = _parse_result(d.pop("result"))
+
+        task = cls(
+            status=status,
+            steps_total=steps_total,
+            steps_completed=steps_completed,
+            exception=exception,
+            updated_at=updated_at,
+            created_at=created_at,
+            id=id,
+            org_id=org_id,
+            type=type,
+            payload=payload,
+            result=result,
+        )
+
+        task.additional_properties = d
+        return task
+
+    @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/task_status.py

@@ -0,0 +1,14 @@
+from enum import Enum
+
+
+class TaskStatus(str, Enum):
+    ABORTED = "ABORTED"
+    ABORTING = "ABORTING"
+    COMPLETED = "COMPLETED"
+    DISPATCHED = "DISPATCHED"
+    FAILED = "FAILED"
+    INITIALIZED = "INITIALIZED"
+    PROCESSING = "PROCESSING"
+
+    def __str__(self) -> str:
+        return str(self.value)

orca_sdk/_generated_api_client/models/task_status_info.py

@@ -0,0 +1,133 @@
+"""
+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.task_status import TaskStatus
+
+T = TypeVar("T", bound="TaskStatusInfo")
+
+
+@_attrs_define
+class TaskStatusInfo:
+    """
+    Attributes:
+        status (TaskStatus): Status of task in the task queue
+        steps_total (Union[None, int]):
+        steps_completed (Union[None, int]):
+        exception (Union[None, str]):
+        updated_at (datetime.datetime):
+        created_at (datetime.datetime):
+    """
+
+    status: TaskStatus
+    steps_total: Union[None, int]
+    steps_completed: Union[None, int]
+    exception: Union[None, str]
+    updated_at: datetime.datetime
+    created_at: datetime.datetime
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        status = self.status.value if isinstance(self.status, Enum) else self.status
+
+        steps_total: Union[None, int]
+        steps_total = self.steps_total
+
+        steps_completed: Union[None, int]
+        steps_completed = self.steps_completed
+
+        exception: Union[None, str]
+        exception = self.exception
+
+        updated_at = self.updated_at.isoformat()
+
+        created_at = self.created_at.isoformat()
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "status": status,
+                "steps_total": steps_total,
+                "steps_completed": steps_completed,
+                "exception": exception,
+                "updated_at": updated_at,
+                "created_at": created_at,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        status = TaskStatus(d.pop("status"))
+
+        def _parse_steps_total(data: object) -> Union[None, int]:
+            if data is None:
+                return data
+            return cast(Union[None, int], data)
+
+        steps_total = _parse_steps_total(d.pop("steps_total"))
+
+        def _parse_steps_completed(data: object) -> Union[None, int]:
+            if data is None:
+                return data
+            return cast(Union[None, int], data)
+
+        steps_completed = _parse_steps_completed(d.pop("steps_completed"))
+
+        def _parse_exception(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        exception = _parse_exception(d.pop("exception"))
+
+        updated_at = isoparse(d.pop("updated_at"))
+
+        created_at = isoparse(d.pop("created_at"))
+
+        task_status_info = cls(
+            status=status,
+            steps_total=steps_total,
+            steps_completed=steps_completed,
+            exception=exception,
+            updated_at=updated_at,
+            created_at=created_at,
+        )
+
+        task_status_info.additional_properties = d
+        return task_status_info
+
+    @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/unauthenticated_error_response.py

@@ -0,0 +1,72 @@
+"""
+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, Literal, Type, TypeVar, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="UnauthenticatedErrorResponse")
+
+
+@_attrs_define
+class UnauthenticatedErrorResponse:
+    """
+    Attributes:
+        status_code (Literal[401]):
+    """
+
+    status_code: Literal[401]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        status_code = self.status_code
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "status_code": status_code,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        status_code = cast(Literal[401], d.pop("status_code"))
+        if status_code != 401:
+            raise ValueError(f"status_code must match const 401, got '{status_code}'")
+
+        unauthenticated_error_response = cls(
+            status_code=status_code,
+        )
+
+        unauthenticated_error_response.additional_properties = d
+        return unauthenticated_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