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

orca_sdk/_generated_api_client/models/org_plan.py

@@ -0,0 +1,99 @@
+"""
+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
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+from dateutil.parser import isoparse
+
+from ..models.org_plan_tier import OrgPlanTier
+
+T = TypeVar("T", bound="OrgPlan")
+
+
+@_attrs_define
+class OrgPlan:
+    """
+    Attributes:
+        org_id (str):
+        created_at (datetime.datetime):
+        updated_at (datetime.datetime):
+        tier (OrgPlanTier):
+    """
+
+    org_id: str
+    created_at: datetime.datetime
+    updated_at: datetime.datetime
+    tier: OrgPlanTier
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        org_id = self.org_id
+
+        created_at = self.created_at.isoformat()
+
+        updated_at = self.updated_at.isoformat()
+
+        tier = self.tier.value if isinstance(self.tier, Enum) else self.tier
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "org_id": org_id,
+                "created_at": created_at,
+                "updated_at": updated_at,
+                "tier": tier,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        org_id = d.pop("org_id")
+
+        created_at = isoparse(d.pop("created_at"))
+
+        updated_at = isoparse(d.pop("updated_at"))
+
+        tier = OrgPlanTier(d.pop("tier"))
+
+        org_plan = cls(
+            org_id=org_id,
+            created_at=created_at,
+            updated_at=updated_at,
+            tier=tier,
+        )
+
+        org_plan.additional_properties = d
+        return org_plan
+
+    @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/org_plan_tier.py

@@ -0,0 +1,11 @@
+from enum import Enum
+
+
+class OrgPlanTier(str, Enum):
+    CANCELLED = "CANCELLED"
+    ENTERPRISE = "ENTERPRISE"
+    FREE = "FREE"
+    PRO = "PRO"
+
+    def __str__(self) -> str:
+        return str(self.value)

orca_sdk/_generated_api_client/models/paginated_task.py

@@ -0,0 +1,108 @@
+"""
+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, List, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+    from ..models.task import Task
+
+
+T = TypeVar("T", bound="PaginatedTask")
+
+
+@_attrs_define
+class PaginatedTask:
+    """
+    Attributes:
+        items (List['Task']):
+        total (int):
+        offset (int):
+        limit (int):
+    """
+
+    items: List["Task"]
+    total: int
+    offset: int
+    limit: int
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        items = []
+        for items_item_data in self.items:
+            items_item = items_item_data.to_dict()
+            items.append(items_item)
+
+        total = self.total
+
+        offset = self.offset
+
+        limit = self.limit
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "items": items,
+                "total": total,
+                "offset": offset,
+                "limit": limit,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.task import Task
+
+        d = src_dict.copy()
+        items = []
+        _items = d.pop("items")
+        for items_item_data in _items:
+            items_item = Task.from_dict(items_item_data)
+
+            items.append(items_item)
+
+        total = d.pop("total")
+
+        offset = d.pop("offset")
+
+        limit = d.pop("limit")
+
+        paginated_task = cls(
+            items=items,
+            total=total,
+            offset=offset,
+            limit=limit,
+        )
+
+        paginated_task.additional_properties = d
+        return paginated_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/predictive_model_update.py

@@ -25,10 +25,12 @@ class PredictiveModelUpdate:
     """
     Attributes:
         description (Union[None, Unset, str]):
+        notes (Union[None, Unset, str]):
         locked (Union[Unset, bool]):
     """
 
     description: Union[None, Unset, str] = UNSET
+    notes: Union[None, Unset, str] = UNSET
     locked: Union[Unset, bool] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -39,6 +41,12 @@ class PredictiveModelUpdate:
         else:
             description = self.description
 
+        notes: Union[None, Unset, str]
+        if isinstance(self.notes, Unset):
+            notes = UNSET
+        else:
+            notes = self.notes
+
         locked = self.locked
 
         field_dict: dict[str, Any] = {}
@@ -46,6 +54,8 @@ class PredictiveModelUpdate:
         field_dict.update({})
         if description is not UNSET:
             field_dict["description"] = description
+        if notes is not UNSET:
+            field_dict["notes"] = notes
         if locked is not UNSET:
             field_dict["locked"] = locked
 
@@ -64,10 +74,20 @@ class PredictiveModelUpdate:
 
         description = _parse_description(d.pop("description", UNSET))
 
+        def _parse_notes(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)
+
+        notes = _parse_notes(d.pop("notes", UNSET))
+
         locked = d.pop("locked", UNSET)
 
         predictive_model_update = cls(
             description=description,
+            notes=notes,
             locked=locked,
         )
 

orca_sdk/_generated_api_client/models/pretrained_embedding_model_metadata.py

@@ -30,6 +30,7 @@ class PretrainedEmbeddingModelMetadata:
         max_seq_length (int):
         uses_context (bool):
         name (PretrainedEmbeddingModelName): Names of pretrained embedding models that are supported by OrcaCloud
+        supports_instructions (bool):
         experimental (Union[Unset, bool]):  Default: False.
     """
 
@@ -37,6 +38,7 @@ class PretrainedEmbeddingModelMetadata:
     max_seq_length: int
     uses_context: bool
     name: PretrainedEmbeddingModelName
+    supports_instructions: bool
     experimental: Union[Unset, bool] = False
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -49,6 +51,8 @@ class PretrainedEmbeddingModelMetadata:
 
         name = self.name.value if isinstance(self.name, Enum) else self.name
 
+        supports_instructions = self.supports_instructions
+
         experimental = self.experimental
 
         field_dict: dict[str, Any] = {}
@@ -59,6 +63,7 @@ class PretrainedEmbeddingModelMetadata:
                 "max_seq_length": max_seq_length,
                 "uses_context": uses_context,
                 "name": name,
+                "supports_instructions": supports_instructions,
             }
         )
         if experimental is not UNSET:
@@ -77,6 +82,8 @@ class PretrainedEmbeddingModelMetadata:
 
         name = PretrainedEmbeddingModelName(d.pop("name"))
 
+        supports_instructions = d.pop("supports_instructions")
+
         experimental = d.pop("experimental", UNSET)
 
         pretrained_embedding_model_metadata = cls(
@@ -84,6 +91,7 @@ class PretrainedEmbeddingModelMetadata:
             max_seq_length=max_seq_length,
             uses_context=uses_context,
             name=name,
+            supports_instructions=supports_instructions,
             experimental=experimental,
         )
 

orca_sdk/_generated_api_client/models/regression_model_metadata.py

@@ -31,6 +31,7 @@ class RegressionModelMetadata:
         org_id (str):
         name (str):
         description (Union[None, str]):
+        notes (Union[None, str]):
         version (int):
         memoryset_id (str):
         memory_lookup_count (int):
@@ -46,6 +47,7 @@ class RegressionModelMetadata:
     org_id: str
     name: str
     description: Union[None, str]
+    notes: Union[None, str]
     version: int
     memoryset_id: str
     memory_lookup_count: int
@@ -67,6 +69,9 @@ class RegressionModelMetadata:
         description: Union[None, str]
         description = self.description
 
+        notes: Union[None, str]
+        notes = self.notes
+
         version = self.version
 
         memoryset_id = self.memoryset_id
@@ -93,6 +98,7 @@ class RegressionModelMetadata:
                 "org_id": org_id,
                 "name": name,
                 "description": description,
+                "notes": notes,
                 "version": version,
                 "memoryset_id": memoryset_id,
                 "memory_lookup_count": memory_lookup_count,
@@ -123,6 +129,13 @@ class RegressionModelMetadata:
 
         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"))
+
         version = d.pop("version")
 
         memoryset_id = d.pop("memoryset_id")
@@ -146,6 +159,7 @@ class RegressionModelMetadata:
             org_id=org_id,
             name=name,
             description=description,
+            notes=notes,
             version=version,
             memoryset_id=memoryset_id,
             memory_lookup_count=memory_lookup_count,

orca_sdk/_generated_api_client/models/scored_memory_update.py

@@ -33,16 +33,16 @@ class ScoredMemoryUpdate:
         value (Union[Unset, str]):
         metadata (Union['ScoredMemoryUpdateMetadataType0', None, Unset]):
         source_id (Union[None, Unset, str]):
-        score (Union[Unset, float]):
         metrics (Union['MemoryMetrics', None, Unset]):
+        score (Union[Unset, float]):
     """
 
     memory_id: str
     value: Union[Unset, str] = UNSET
     metadata: Union["ScoredMemoryUpdateMetadataType0", None, Unset] = UNSET
     source_id: Union[None, Unset, str] = UNSET
-    score: Union[Unset, float] = UNSET
     metrics: Union["MemoryMetrics", None, Unset] = UNSET
+    score: Union[Unset, float] = UNSET
 
     def to_dict(self) -> dict[str, Any]:
         from ..models.memory_metrics import MemoryMetrics
@@ -70,8 +70,6 @@ class ScoredMemoryUpdate:
         else:
             source_id = self.source_id
 
-        score = self.score
-
         metrics: Union[Dict[str, Any], None, Unset]
         if isinstance(self.metrics, Unset):
             metrics = UNSET
@@ -80,6 +78,8 @@ class ScoredMemoryUpdate:
         else:
             metrics = self.metrics
 
+        score = self.score
+
         field_dict: dict[str, Any] = {}
         field_dict.update(
             {
@@ -92,10 +92,10 @@ class ScoredMemoryUpdate:
             field_dict["metadata"] = metadata
         if source_id is not UNSET:
             field_dict["source_id"] = source_id
-        if score is not UNSET:
-            field_dict["score"] = score
         if metrics is not UNSET:
             field_dict["metrics"] = metrics
+        if score is not UNSET:
+            field_dict["score"] = score
 
         return field_dict
 
@@ -140,8 +140,6 @@ class ScoredMemoryUpdate:
 
         source_id = _parse_source_id(d.pop("source_id", UNSET))
 
-        score = d.pop("score", UNSET)
-
         def _parse_metrics(data: object) -> Union["MemoryMetrics", None, Unset]:
             if data is None:
                 return data
@@ -159,13 +157,15 @@ class ScoredMemoryUpdate:
 
         metrics = _parse_metrics(d.pop("metrics", UNSET))
 
+        score = d.pop("score", UNSET)
+
         scored_memory_update = cls(
             memory_id=memory_id,
             value=value,
             metadata=metadata,
             source_id=source_id,
-            score=score,
             metrics=metrics,
+            score=score,
         )
 
         return scored_memory_update

orca_sdk/_generated_api_client/models/service_unavailable_error_response.py

@@ -10,11 +10,14 @@ The main change is:
 
 # flake8: noqa: C901
 
-from typing import Any, Literal, Type, TypeVar, cast
+from enum import Enum
+from typing import Any, Type, TypeVar
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
+from ..models.service_unavailable_error_response_status_code import ServiceUnavailableErrorResponseStatusCode
+
 T = TypeVar("T", bound="ServiceUnavailableErrorResponse")
 
 
@@ -22,16 +25,16 @@ T = TypeVar("T", bound="ServiceUnavailableErrorResponse")
 class ServiceUnavailableErrorResponse:
     """
     Attributes:
-        status_code (Literal[503]):
+        status_code (ServiceUnavailableErrorResponseStatusCode):
         service (str):
     """
 
-    status_code: Literal[503]
+    status_code: ServiceUnavailableErrorResponseStatusCode
     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
+        status_code = self.status_code.value if isinstance(self.status_code, Enum) else self.status_code
 
         service = self.service
 
@@ -49,9 +52,7 @@ class ServiceUnavailableErrorResponse:
     @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}'")
+        status_code = ServiceUnavailableErrorResponseStatusCode(d.pop("status_code"))
 
         service = d.pop("service")
 

orca_sdk/_generated_api_client/models/service_unavailable_error_response_status_code.py

@@ -0,0 +1,8 @@
+from enum import IntEnum
+
+
+class ServiceUnavailableErrorResponseStatusCode(IntEnum):
+    VALUE_503 = 503
+
+    def __str__(self) -> str:
+        return str(self.value)

orca_sdk/_generated_api_client/models/telemetry_field_type_0_item_type_0.py

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

orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_0.py

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

orca_sdk/_generated_api_client/models/telemetry_field_type_1_item_type_1.py

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

orca_sdk/_generated_api_client/models/telemetry_filter_item.py

@@ -11,12 +11,15 @@ The main change is:
 # flake8: noqa: C901
 
 from enum import Enum
-from typing import Any, List, Literal, Type, TypeVar, Union, cast
+from typing import Any, List, Type, TypeVar, Union, cast
 
 from attrs import define as _attrs_define
 from attrs import field as _attrs_field
 
+from ..models.telemetry_field_type_0_item_type_0 import TelemetryFieldType0ItemType0
 from ..models.telemetry_field_type_0_item_type_2 import TelemetryFieldType0ItemType2
+from ..models.telemetry_field_type_1_item_type_0 import TelemetryFieldType1ItemType0
+from ..models.telemetry_field_type_1_item_type_1 import TelemetryFieldType1ItemType1
 from ..models.telemetry_filter_item_op import TelemetryFilterItemOp
 
 T = TypeVar("T", bound="TelemetryFilterItem")
@@ -26,27 +29,33 @@ T = TypeVar("T", bound="TelemetryFilterItem")
 class TelemetryFilterItem:
     """
     Attributes:
-        field (Union[List[Union[Literal['count'], Literal['lookup']]], List[Union[Literal['feedback_metrics'],
-            TelemetryFieldType0ItemType2, str]]]):
+        field (Union[List[Union[TelemetryFieldType0ItemType0, TelemetryFieldType0ItemType2, str]],
+            List[Union[TelemetryFieldType1ItemType0, TelemetryFieldType1ItemType1]]]):
         op (TelemetryFilterItemOp):
         value (Union[List[float], List[int], float, int]):
     """
 
     field: Union[
-        List[Union[Literal["count"], Literal["lookup"]]],
-        List[Union[Literal["feedback_metrics"], TelemetryFieldType0ItemType2, str]],
+        List[Union[TelemetryFieldType0ItemType0, TelemetryFieldType0ItemType2, str]],
+        List[Union[TelemetryFieldType1ItemType0, TelemetryFieldType1ItemType1]],
     ]
     op: TelemetryFilterItemOp
     value: Union[List[float], List[int], float, int]
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
-        field: Union[List[Union[Literal["count"], Literal["lookup"]]], List[Union[Literal["feedback_metrics"], str]]]
+        field: List[str]
         if isinstance(self.field, list):
             field = []
             for componentsschemas_telemetry_field_type_0_item_data in self.field:
-                componentsschemas_telemetry_field_type_0_item: Union[Literal["feedback_metrics"], str]
-                if isinstance(componentsschemas_telemetry_field_type_0_item_data, TelemetryFieldType0ItemType2):
+                componentsschemas_telemetry_field_type_0_item: str
+                if isinstance(componentsschemas_telemetry_field_type_0_item_data, TelemetryFieldType0ItemType0):
+                    componentsschemas_telemetry_field_type_0_item = (
+                        componentsschemas_telemetry_field_type_0_item_data.value
+                        if isinstance(componentsschemas_telemetry_field_type_0_item_data, Enum)
+                        else componentsschemas_telemetry_field_type_0_item_data
+                    )
+                elif isinstance(componentsschemas_telemetry_field_type_0_item_data, TelemetryFieldType0ItemType2):
                     componentsschemas_telemetry_field_type_0_item = (
                         componentsschemas_telemetry_field_type_0_item_data.value
                         if isinstance(componentsschemas_telemetry_field_type_0_item_data, Enum)
@@ -84,8 +93,8 @@ class TelemetryFilterItem:
         def _parse_field(
             data: object,
         ) -> Union[
-            List[Union[Literal["count"], Literal["lookup"]]],
-            List[Union[Literal["feedback_metrics"], TelemetryFieldType0ItemType2, str]],
+            List[Union[TelemetryFieldType0ItemType0, TelemetryFieldType0ItemType2, str]],
+            List[Union[TelemetryFieldType1ItemType0, TelemetryFieldType1ItemType1]],
         ]:
             try:
                 if not isinstance(data, list):
@@ -96,13 +105,15 @@ class TelemetryFilterItem:
 
                     def _parse_componentsschemas_telemetry_field_type_0_item(
                         data: object,
-                    ) -> Union[Literal["feedback_metrics"], TelemetryFieldType0ItemType2, str]:
-                        componentsschemas_telemetry_field_type_0_item_type_0 = cast(Literal["feedback_metrics"], data)
-                        if componentsschemas_telemetry_field_type_0_item_type_0 != "feedback_metrics":
-                            raise ValueError(
-                                f"/components/schemas/TelemetryField_type_0_item_type_0 must match const 'feedback_metrics', got '{componentsschemas_telemetry_field_type_0_item_type_0}'"
-                            )
-                        return componentsschemas_telemetry_field_type_0_item_type_0
+                    ) -> Union[TelemetryFieldType0ItemType0, TelemetryFieldType0ItemType2, str]:
+                        try:
+                            if not isinstance(data, str):
+                                raise TypeError()
+                            componentsschemas_telemetry_field_type_0_item_type_0 = TelemetryFieldType0ItemType0(data)
+
+                            return componentsschemas_telemetry_field_type_0_item_type_0
+                        except:  # noqa: E722
+                            pass
                         try:
                             if not isinstance(data, str):
                                 raise TypeError()
@@ -111,7 +122,7 @@ class TelemetryFilterItem:
                             return componentsschemas_telemetry_field_type_0_item_type_2
                         except:  # noqa: E722
                             pass
-                        return cast(Union[Literal["feedback_metrics"], TelemetryFieldType0ItemType2, str], data)
+                        return cast(Union[TelemetryFieldType0ItemType0, TelemetryFieldType0ItemType2, str], data)
 
                     componentsschemas_telemetry_field_type_0_item = (
                         _parse_componentsschemas_telemetry_field_type_0_item(
@@ -132,18 +143,19 @@ class TelemetryFilterItem:
 
                 def _parse_componentsschemas_telemetry_field_type_1_item(
                     data: object,
-                ) -> Union[Literal["count"], Literal["lookup"]]:
-                    componentsschemas_telemetry_field_type_1_item_type_0 = cast(Literal["lookup"], data)
-                    if componentsschemas_telemetry_field_type_1_item_type_0 != "lookup":
-                        raise ValueError(
-                            f"/components/schemas/TelemetryField_type_1_item_type_0 must match const 'lookup', got '{componentsschemas_telemetry_field_type_1_item_type_0}'"
-                        )
-                    return componentsschemas_telemetry_field_type_1_item_type_0
-                    componentsschemas_telemetry_field_type_1_item_type_1 = cast(Literal["count"], data)
-                    if componentsschemas_telemetry_field_type_1_item_type_1 != "count":
-                        raise ValueError(
-                            f"/components/schemas/TelemetryField_type_1_item_type_1 must match const 'count', got '{componentsschemas_telemetry_field_type_1_item_type_1}'"
-                        )
+                ) -> Union[TelemetryFieldType1ItemType0, TelemetryFieldType1ItemType1]:
+                    try:
+                        if not isinstance(data, str):
+                            raise TypeError()
+                        componentsschemas_telemetry_field_type_1_item_type_0 = TelemetryFieldType1ItemType0(data)
+
+                        return componentsschemas_telemetry_field_type_1_item_type_0
+                    except:  # noqa: E722
+                        pass
+                    if not isinstance(data, str):
+                        raise TypeError()
+                    componentsschemas_telemetry_field_type_1_item_type_1 = TelemetryFieldType1ItemType1(data)
+
                     return componentsschemas_telemetry_field_type_1_item_type_1
 
                 componentsschemas_telemetry_field_type_1_item = _parse_componentsschemas_telemetry_field_type_1_item(