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

orca_sdk/_generated_api_client/models/{clone_labeled_memoryset_request.py → clone_memoryset_request.py}

@@ -19,11 +19,11 @@ from attrs import field as _attrs_field
 from ..models.pretrained_embedding_model_name import PretrainedEmbeddingModelName
 from ..types import UNSET, Unset
 
-T = TypeVar("T", bound="CloneLabeledMemorysetRequest")
+T = TypeVar("T", bound="CloneMemorysetRequest")
 
 
 @_attrs_define
-class CloneLabeledMemorysetRequest:
+class CloneMemorysetRequest:
     """
     Attributes:
         name (str):
@@ -142,7 +142,7 @@ class CloneLabeledMemorysetRequest:
 
         max_seq_length_override = _parse_max_seq_length_override(d.pop("max_seq_length_override", UNSET))
 
-        clone_labeled_memoryset_request = cls(
+        clone_memoryset_request = cls(
             name=name,
             description=description,
             pretrained_embedding_model_name=pretrained_embedding_model_name,
@@ -150,8 +150,8 @@ class CloneLabeledMemorysetRequest:
             max_seq_length_override=max_seq_length_override,
         )
 
-        clone_labeled_memoryset_request.additional_properties = d
-        return clone_labeled_memoryset_request
+        clone_memoryset_request.additional_properties = d
+        return clone_memoryset_request
 
     @property
     def additional_keys(self) -> list[str]:

orca_sdk/_generated_api_client/models/column_info.py

@@ -30,11 +30,13 @@ class ColumnInfo:
         name (str):
         type (ColumnType): The type of a column in a datasource
         enum_options (Union[List[str], None, Unset]):
+        int_values (Union[List[int], None, Unset]):
     """
 
     name: str
     type: ColumnType
     enum_options: Union[List[str], None, Unset] = UNSET
+    int_values: Union[List[int], None, Unset] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -51,6 +53,15 @@ class ColumnInfo:
         else:
             enum_options = self.enum_options
 
+        int_values: Union[List[int], None, Unset]
+        if isinstance(self.int_values, Unset):
+            int_values = UNSET
+        elif isinstance(self.int_values, list):
+            int_values = self.int_values
+
+        else:
+            int_values = self.int_values
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -61,6 +72,8 @@ class ColumnInfo:
         )
         if enum_options is not UNSET:
             field_dict["enum_options"] = enum_options
+        if int_values is not UNSET:
+            field_dict["int_values"] = int_values
 
         return field_dict
 
@@ -88,10 +101,28 @@ class ColumnInfo:
 
         enum_options = _parse_enum_options(d.pop("enum_options", UNSET))
 
+        def _parse_int_values(data: object) -> Union[List[int], None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                int_values_type_0 = cast(List[int], data)
+
+                return int_values_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[List[int], None, Unset], data)
+
+        int_values = _parse_int_values(d.pop("int_values", UNSET))
+
         column_info = cls(
             name=name,
             type=type,
             enum_options=enum_options,
+            int_values=int_values,
         )
 
         column_info.additional_properties = d

orca_sdk/_generated_api_client/models/count_predictions_request.py

@@ -0,0 +1,195 @@
+"""
+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 typing import 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 ..types import UNSET, Unset
+
+T = TypeVar("T", bound="CountPredictionsRequest")
+
+
+@_attrs_define
+class CountPredictionsRequest:
+    """
+    Attributes:
+        model_id (Union[None, Unset, str]):
+        tag (Union[None, Unset, str]):
+        prediction_ids (Union[List[str], None, Unset]):
+        start_timestamp (Union[None, Unset, datetime.datetime]):
+        end_timestamp (Union[None, Unset, datetime.datetime]):
+    """
+
+    model_id: Union[None, Unset, str] = UNSET
+    tag: Union[None, Unset, str] = UNSET
+    prediction_ids: Union[List[str], None, Unset] = UNSET
+    start_timestamp: Union[None, Unset, datetime.datetime] = UNSET
+    end_timestamp: Union[None, Unset, datetime.datetime] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        model_id: Union[None, Unset, str]
+        if isinstance(self.model_id, Unset):
+            model_id = UNSET
+        else:
+            model_id = self.model_id
+
+        tag: Union[None, Unset, str]
+        if isinstance(self.tag, Unset):
+            tag = UNSET
+        else:
+            tag = self.tag
+
+        prediction_ids: Union[List[str], None, Unset]
+        if isinstance(self.prediction_ids, Unset):
+            prediction_ids = UNSET
+        elif isinstance(self.prediction_ids, list):
+            prediction_ids = self.prediction_ids
+
+        else:
+            prediction_ids = self.prediction_ids
+
+        start_timestamp: Union[None, Unset, str]
+        if isinstance(self.start_timestamp, Unset):
+            start_timestamp = UNSET
+        elif isinstance(self.start_timestamp, datetime.datetime):
+            start_timestamp = self.start_timestamp.isoformat()
+        else:
+            start_timestamp = self.start_timestamp
+
+        end_timestamp: Union[None, Unset, str]
+        if isinstance(self.end_timestamp, Unset):
+            end_timestamp = UNSET
+        elif isinstance(self.end_timestamp, datetime.datetime):
+            end_timestamp = self.end_timestamp.isoformat()
+        else:
+            end_timestamp = self.end_timestamp
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update({})
+        if model_id is not UNSET:
+            field_dict["model_id"] = model_id
+        if tag is not UNSET:
+            field_dict["tag"] = tag
+        if prediction_ids is not UNSET:
+            field_dict["prediction_ids"] = prediction_ids
+        if start_timestamp is not UNSET:
+            field_dict["start_timestamp"] = start_timestamp
+        if end_timestamp is not UNSET:
+            field_dict["end_timestamp"] = end_timestamp
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+
+        def _parse_model_id(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)
+
+        model_id = _parse_model_id(d.pop("model_id", UNSET))
+
+        def _parse_tag(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)
+
+        tag = _parse_tag(d.pop("tag", UNSET))
+
+        def _parse_prediction_ids(data: object) -> Union[List[str], None, Unset]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, list):
+                    raise TypeError()
+                prediction_ids_type_0 = cast(List[str], data)
+
+                return prediction_ids_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[List[str], None, Unset], data)
+
+        prediction_ids = _parse_prediction_ids(d.pop("prediction_ids", UNSET))
+
+        def _parse_start_timestamp(data: object) -> Union[None, Unset, datetime.datetime]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                start_timestamp_type_0 = isoparse(data)
+
+                return start_timestamp_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, Unset, datetime.datetime], data)
+
+        start_timestamp = _parse_start_timestamp(d.pop("start_timestamp", UNSET))
+
+        def _parse_end_timestamp(data: object) -> Union[None, Unset, datetime.datetime]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            try:
+                if not isinstance(data, str):
+                    raise TypeError()
+                end_timestamp_type_0 = isoparse(data)
+
+                return end_timestamp_type_0
+            except:  # noqa: E722
+                pass
+            return cast(Union[None, Unset, datetime.datetime], data)
+
+        end_timestamp = _parse_end_timestamp(d.pop("end_timestamp", UNSET))
+
+        count_predictions_request = cls(
+            model_id=model_id,
+            tag=tag,
+            prediction_ids=prediction_ids,
+            start_timestamp=start_timestamp,
+            end_timestamp=end_timestamp,
+        )
+
+        count_predictions_request.additional_properties = d
+        return count_predictions_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/{create_rac_model_request.py → create_classification_model_request.py}

@@ -18,55 +18,49 @@ from attrs import field as _attrs_field
 from ..models.rac_head_type import RACHeadType
 from ..types import UNSET, Unset
 
-T = TypeVar("T", bound="CreateRACModelRequest")
+T = TypeVar("T", bound="CreateClassificationModelRequest")
 
 
 @_attrs_define
-class CreateRACModelRequest:
+class CreateClassificationModelRequest:
     """
     Attributes:
         name (str):
+        memoryset_id (str):
         description (Union[None, Unset, str]):
-        memoryset_id (Union[None, Unset, str]):
-        memoryset_name (Union[None, Unset, str]):
+        memory_lookup_count (Union[None, Unset, int]):
         head_type (Union[Unset, RACHeadType]):
         weigh_memories (Union[None, Unset, bool]):
         min_memory_weight (Union[None, Unset, float]):
-        memory_lookup_count (Union[None, Unset, int]):
         num_classes (Union[None, Unset, int]):
     """
 
     name: str
+    memoryset_id: str
     description: Union[None, Unset, str] = UNSET
-    memoryset_id: Union[None, Unset, str] = UNSET
-    memoryset_name: Union[None, Unset, str] = UNSET
+    memory_lookup_count: Union[None, Unset, int] = UNSET
     head_type: Union[Unset, RACHeadType] = UNSET
     weigh_memories: Union[None, Unset, bool] = UNSET
     min_memory_weight: Union[None, Unset, float] = UNSET
-    memory_lookup_count: Union[None, Unset, int] = UNSET
     num_classes: Union[None, Unset, int] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
         name = self.name
 
+        memoryset_id = self.memoryset_id
+
         description: Union[None, Unset, str]
         if isinstance(self.description, Unset):
             description = UNSET
         else:
             description = self.description
 
-        memoryset_id: Union[None, Unset, str]
-        if isinstance(self.memoryset_id, Unset):
-            memoryset_id = UNSET
-        else:
-            memoryset_id = self.memoryset_id
-
-        memoryset_name: Union[None, Unset, str]
-        if isinstance(self.memoryset_name, Unset):
-            memoryset_name = UNSET
+        memory_lookup_count: Union[None, Unset, int]
+        if isinstance(self.memory_lookup_count, Unset):
+            memory_lookup_count = UNSET
         else:
-            memoryset_name = self.memoryset_name
+            memory_lookup_count = self.memory_lookup_count
 
         head_type: Union[Unset, str] = UNSET
         if not isinstance(self.head_type, Unset):
@@ -84,12 +78,6 @@ class CreateRACModelRequest:
         else:
             min_memory_weight = self.min_memory_weight
 
-        memory_lookup_count: Union[None, Unset, int]
-        if isinstance(self.memory_lookup_count, Unset):
-            memory_lookup_count = UNSET
-        else:
-            memory_lookup_count = self.memory_lookup_count
-
         num_classes: Union[None, Unset, int]
         if isinstance(self.num_classes, Unset):
             num_classes = UNSET
@@ -101,22 +89,19 @@ class CreateRACModelRequest:
         field_dict.update(
             {
                 "name": name,
+                "memoryset_id": memoryset_id,
             }
         )
         if description is not UNSET:
             field_dict["description"] = description
-        if memoryset_id is not UNSET:
-            field_dict["memoryset_id"] = memoryset_id
-        if memoryset_name is not UNSET:
-            field_dict["memoryset_name"] = memoryset_name
+        if memory_lookup_count is not UNSET:
+            field_dict["memory_lookup_count"] = memory_lookup_count
         if head_type is not UNSET:
             field_dict["head_type"] = head_type
         if weigh_memories is not UNSET:
             field_dict["weigh_memories"] = weigh_memories
         if min_memory_weight is not UNSET:
             field_dict["min_memory_weight"] = min_memory_weight
-        if memory_lookup_count is not UNSET:
-            field_dict["memory_lookup_count"] = memory_lookup_count
         if num_classes is not UNSET:
             field_dict["num_classes"] = num_classes
 
@@ -127,6 +112,8 @@ class CreateRACModelRequest:
         d = src_dict.copy()
         name = d.pop("name")
 
+        memoryset_id = d.pop("memoryset_id")
+
         def _parse_description(data: object) -> Union[None, Unset, str]:
             if data is None:
                 return data
@@ -136,23 +123,14 @@ class CreateRACModelRequest:
 
         description = _parse_description(d.pop("description", UNSET))
 
-        def _parse_memoryset_id(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)
-
-        memoryset_id = _parse_memoryset_id(d.pop("memoryset_id", UNSET))
-
-        def _parse_memoryset_name(data: object) -> Union[None, Unset, str]:
+        def _parse_memory_lookup_count(data: object) -> Union[None, Unset, int]:
             if data is None:
                 return data
             if isinstance(data, Unset):
                 return data
-            return cast(Union[None, Unset, str], data)
+            return cast(Union[None, Unset, int], data)
 
-        memoryset_name = _parse_memoryset_name(d.pop("memoryset_name", UNSET))
+        memory_lookup_count = _parse_memory_lookup_count(d.pop("memory_lookup_count", UNSET))
 
         _head_type = d.pop("head_type", UNSET)
         head_type: Union[Unset, RACHeadType]
@@ -179,15 +157,6 @@ class CreateRACModelRequest:
 
         min_memory_weight = _parse_min_memory_weight(d.pop("min_memory_weight", UNSET))
 
-        def _parse_memory_lookup_count(data: object) -> Union[None, Unset, int]:
-            if data is None:
-                return data
-            if isinstance(data, Unset):
-                return data
-            return cast(Union[None, Unset, int], data)
-
-        memory_lookup_count = _parse_memory_lookup_count(d.pop("memory_lookup_count", UNSET))
-
         def _parse_num_classes(data: object) -> Union[None, Unset, int]:
             if data is None:
                 return data
@@ -197,20 +166,19 @@ class CreateRACModelRequest:
 
         num_classes = _parse_num_classes(d.pop("num_classes", UNSET))
 
-        create_rac_model_request = cls(
+        create_classification_model_request = cls(
             name=name,
-            description=description,
             memoryset_id=memoryset_id,
-            memoryset_name=memoryset_name,
+            description=description,
+            memory_lookup_count=memory_lookup_count,
             head_type=head_type,
             weigh_memories=weigh_memories,
             min_memory_weight=min_memory_weight,
-            memory_lookup_count=memory_lookup_count,
             num_classes=num_classes,
         )
 
-        create_rac_model_request.additional_properties = d
-        return create_rac_model_request
+        create_classification_model_request.additional_properties = d
+        return create_classification_model_request
 
     @property
     def additional_keys(self) -> list[str]: