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

orca_sdk/_generated_api_client/models/action_recommendation.py

@@ -0,0 +1,82 @@
+"""
+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
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..models.action_recommendation_action import ActionRecommendationAction
+
+T = TypeVar("T", bound="ActionRecommendation")
+
+
+@_attrs_define
+class ActionRecommendation:
+    """Recommended actions to take based on the prediction and explanation
+
+    Attributes:
+        action (ActionRecommendationAction): The recommended action to take
+        rationale (str): Explanation for why this action was recommended
+    """
+
+    action: ActionRecommendationAction
+    rationale: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        action = self.action.value if isinstance(self.action, Enum) else self.action
+
+        rationale = self.rationale
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "action": action,
+                "rationale": rationale,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        action = ActionRecommendationAction(d.pop("action"))
+
+        rationale = d.pop("rationale")
+
+        action_recommendation = cls(
+            action=action,
+            rationale=rationale,
+        )
+
+        action_recommendation.additional_properties = d
+        return action_recommendation
+
+    @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/action_recommendation_action.py

@@ -0,0 +1,11 @@
+from enum import Enum
+
+
+class ActionRecommendationAction(str, Enum):
+    ADD_MEMORIES = "add_memories"
+    DETECT_MISLABELS = "detect_mislabels"
+    FINETUNING = "finetuning"
+    REMOVE_DUPLICATES = "remove_duplicates"
+
+    def __str__(self) -> str:
+        return str(self.value)

orca_sdk/_generated_api_client/models/add_memory_recommendations.py

@@ -0,0 +1,85 @@
+"""
+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.add_memory_suggestion import AddMemorySuggestion
+
+
+T = TypeVar("T", bound="AddMemoryRecommendations")
+
+
+@_attrs_define
+class AddMemoryRecommendations:
+    """Recommendations for memories to add
+
+    Attributes:
+        memories (List['AddMemorySuggestion']):
+    """
+
+    memories: List["AddMemorySuggestion"]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        memories = []
+        for memories_item_data in self.memories:
+            memories_item = memories_item_data.to_dict()
+            memories.append(memories_item)
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "memories": memories,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.add_memory_suggestion import AddMemorySuggestion
+
+        d = src_dict.copy()
+        memories = []
+        _memories = d.pop("memories")
+        for memories_item_data in _memories:
+            memories_item = AddMemorySuggestion.from_dict(memories_item_data)
+
+            memories.append(memories_item)
+
+        add_memory_recommendations = cls(
+            memories=memories,
+        )
+
+        add_memory_recommendations.additional_properties = d
+        return add_memory_recommendations
+
+    @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/add_memory_suggestion.py

@@ -0,0 +1,79 @@
+"""
+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
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="AddMemorySuggestion")
+
+
+@_attrs_define
+class AddMemorySuggestion:
+    """A suggested memory to add to the memory store
+
+    Attributes:
+        value (str):
+        label (str):
+    """
+
+    value: str
+    label: str
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        value = self.value
+
+        label = self.label
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "value": value,
+                "label": label,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        value = d.pop("value")
+
+        label = d.pop("label")
+
+        add_memory_suggestion = cls(
+            value=value,
+            label=label,
+        )
+
+        add_memory_suggestion.additional_properties = d
+        return add_memory_suggestion
+
+    @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/body_create_datasource_from_files_datasource_upload_post.py

@@ -0,0 +1,145 @@
+"""
+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 json
+from io import BytesIO
+from typing import Any, List, Tuple, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+from ..types import UNSET, File, Unset
+
+T = TypeVar("T", bound="BodyCreateDatasourceFromFilesDatasourceUploadPost")
+
+
+@_attrs_define
+class BodyCreateDatasourceFromFilesDatasourceUploadPost:
+    """
+    Attributes:
+        files (List[File]):
+        name (str):
+        description (Union[None, Unset, str]):
+    """
+
+    files: List[File]
+    name: str
+    description: Union[None, Unset, str] = UNSET
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        files = []
+        for files_item_data in self.files:
+            files_item = files_item_data.to_tuple()
+
+            files.append(files_item)
+
+        name = self.name
+
+        description: Union[None, Unset, str]
+        if isinstance(self.description, Unset):
+            description = UNSET
+        else:
+            description = self.description
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "files": files,
+                "name": name,
+            }
+        )
+        if description is not UNSET:
+            field_dict["description"] = description
+
+        return field_dict
+
+    def to_multipart(self) -> dict[str, Any]:
+        _temp_files = []
+        for files_item_data in self.files:
+            files_item = files_item_data.to_tuple()
+
+            _temp_files.append(files_item)
+        files = (None, json.dumps(_temp_files).encode(), "application/json")
+
+        name = (None, str(self.name).encode(), "text/plain")
+
+        description: Union[Tuple[None, bytes, str], Unset]
+
+        if isinstance(self.description, Unset):
+            description = UNSET
+        elif isinstance(self.description, str):
+            description = (None, str(self.description).encode(), "text/plain")
+        else:
+            description = (None, str(self.description).encode(), "text/plain")
+
+        field_dict: dict[str, Any] = {}
+        for prop_name, prop in self.additional_properties.items():
+            field_dict[prop_name] = (None, str(prop).encode(), "text/plain")
+
+        field_dict.update(
+            {
+                "files": files,
+                "name": name,
+            }
+        )
+        if description is not UNSET:
+            field_dict["description"] = description
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        files = []
+        _files = d.pop("files")
+        for files_item_data in _files:
+            files_item = File(payload=BytesIO(files_item_data))
+
+            files.append(files_item)
+
+        name = d.pop("name")
+
+        def _parse_description(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)
+
+        description = _parse_description(d.pop("description", UNSET))
+
+        body_create_datasource_from_files_datasource_upload_post = cls(
+            files=files,
+            name=name,
+            description=description,
+        )
+
+        body_create_datasource_from_files_datasource_upload_post.additional_properties = d
+        return body_create_datasource_from_files_datasource_upload_post
+
+    @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/class_representatives.py

@@ -0,0 +1,92 @@
+"""
+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, List, Type, TypeVar, Union, cast
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+T = TypeVar("T", bound="ClassRepresentatives")
+
+
+@_attrs_define
+class ClassRepresentatives:
+    """
+    Attributes:
+        label (int):
+        label_name (Union[None, str]):
+        representative_memory_ids (List[str]):
+    """
+
+    label: int
+    label_name: Union[None, str]
+    representative_memory_ids: List[str]
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        label = self.label
+
+        label_name: Union[None, str]
+        label_name = self.label_name
+
+        representative_memory_ids = self.representative_memory_ids
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "label": label,
+                "label_name": label_name,
+                "representative_memory_ids": representative_memory_ids,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        label = d.pop("label")
+
+        def _parse_label_name(data: object) -> Union[None, str]:
+            if data is None:
+                return data
+            return cast(Union[None, str], data)
+
+        label_name = _parse_label_name(d.pop("label_name"))
+
+        representative_memory_ids = cast(List[str], d.pop("representative_memory_ids"))
+
+        class_representatives = cls(
+            label=label,
+            label_name=label_name,
+            representative_memory_ids=representative_memory_ids,
+        )
+
+        class_representatives.additional_properties = d
+        return class_representatives
+
+    @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/classification_model_metadata.py

@@ -31,6 +31,7 @@ class ClassificationModelMetadata:
         org_id (str):
         name (str):
         description (Union[None, str]):
+        notes (Union[None, str]):
         version (int):
         memoryset_id (str):
         memory_lookup_count (int):
@@ -49,6 +50,7 @@ class ClassificationModelMetadata:
     org_id: str
     name: str
     description: Union[None, str]
+    notes: Union[None, str]
     version: int
     memoryset_id: str
     memory_lookup_count: int
@@ -73,6 +75,9 @@ class ClassificationModelMetadata:
         description: Union[None, str]
         description = self.description
 
+        notes: Union[None, str]
+        notes = self.notes
+
         version = self.version
 
         memoryset_id = self.memoryset_id
@@ -107,6 +112,7 @@ class ClassificationModelMetadata:
                 "org_id": org_id,
                 "name": name,
                 "description": description,
+                "notes": notes,
                 "version": version,
                 "memoryset_id": memoryset_id,
                 "memory_lookup_count": memory_lookup_count,
@@ -140,6 +146,13 @@ class ClassificationModelMetadata:
 
         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")
@@ -179,6 +192,7 @@ class ClassificationModelMetadata:
             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/clone_memoryset_request.py

@@ -28,16 +28,20 @@ class CloneMemorysetRequest:
     Attributes:
         name (str):
         description (Union[None, Unset, str]):
+        notes (Union[None, Unset, str]):
         pretrained_embedding_model_name (Union[None, PretrainedEmbeddingModelName, Unset]):
         finetuned_embedding_model_id (Union[None, Unset, str]):
         max_seq_length_override (Union[None, Unset, int]):
+        prompt (Union[None, Unset, str]):
     """
 
     name: str
     description: Union[None, Unset, str] = UNSET
+    notes: Union[None, Unset, str] = UNSET
     pretrained_embedding_model_name: Union[None, PretrainedEmbeddingModelName, Unset] = UNSET
     finetuned_embedding_model_id: Union[None, Unset, str] = UNSET
     max_seq_length_override: Union[None, Unset, int] = UNSET
+    prompt: Union[None, Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -49,6 +53,12 @@ class CloneMemorysetRequest:
         else:
             description = self.description
 
+        notes: Union[None, Unset, str]
+        if isinstance(self.notes, Unset):
+            notes = UNSET
+        else:
+            notes = self.notes
+
         pretrained_embedding_model_name: Union[None, Unset, str]
         if isinstance(self.pretrained_embedding_model_name, Unset):
             pretrained_embedding_model_name = UNSET
@@ -73,6 +83,12 @@ class CloneMemorysetRequest:
         else:
             max_seq_length_override = self.max_seq_length_override
 
+        prompt: Union[None, Unset, str]
+        if isinstance(self.prompt, Unset):
+            prompt = UNSET
+        else:
+            prompt = self.prompt
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -82,12 +98,16 @@ class CloneMemorysetRequest:
         )
         if description is not UNSET:
             field_dict["description"] = description
+        if notes is not UNSET:
+            field_dict["notes"] = notes
         if pretrained_embedding_model_name is not UNSET:
             field_dict["pretrained_embedding_model_name"] = pretrained_embedding_model_name
         if finetuned_embedding_model_id is not UNSET:
             field_dict["finetuned_embedding_model_id"] = finetuned_embedding_model_id
         if max_seq_length_override is not UNSET:
             field_dict["max_seq_length_override"] = max_seq_length_override
+        if prompt is not UNSET:
+            field_dict["prompt"] = prompt
 
         return field_dict
 
@@ -105,6 +125,15 @@ class CloneMemorysetRequest:
 
         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))
+
         def _parse_pretrained_embedding_model_name(data: object) -> Union[None, PretrainedEmbeddingModelName, Unset]:
             if data is None:
                 return data
@@ -142,12 +171,23 @@ class CloneMemorysetRequest:
 
         max_seq_length_override = _parse_max_seq_length_override(d.pop("max_seq_length_override", UNSET))
 
+        def _parse_prompt(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)
+
+        prompt = _parse_prompt(d.pop("prompt", UNSET))
+
         clone_memoryset_request = cls(
             name=name,
             description=description,
+            notes=notes,
             pretrained_embedding_model_name=pretrained_embedding_model_name,
             finetuned_embedding_model_id=finetuned_embedding_model_id,
             max_seq_length_override=max_seq_length_override,
+            prompt=prompt,
         )
 
         clone_memoryset_request.additional_properties = d

orca_sdk/_generated_api_client/models/constraint_violation_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.constraint_violation_error_response_status_code import ConstraintViolationErrorResponseStatusCode
+
 T = TypeVar("T", bound="ConstraintViolationErrorResponse")
 
 
@@ -22,16 +25,16 @@ T = TypeVar("T", bound="ConstraintViolationErrorResponse")
 class ConstraintViolationErrorResponse:
     """
     Attributes:
-        status_code (Literal[409]):
+        status_code (ConstraintViolationErrorResponseStatusCode):
         constraint (str):
     """
 
-    status_code: Literal[409]
+    status_code: ConstraintViolationErrorResponseStatusCode
     constraint: 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
 
         constraint = self.constraint
 
@@ -49,9 +52,7 @@ class ConstraintViolationErrorResponse:
     @classmethod
     def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
         d = src_dict.copy()
-        status_code = cast(Literal[409], d.pop("status_code"))
-        if status_code != 409:
-            raise ValueError(f"status_code must match const 409, got '{status_code}'")
+        status_code = ConstraintViolationErrorResponseStatusCode(d.pop("status_code"))
 
         constraint = d.pop("constraint")