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

orca_sdk/_generated_api_client/api/__init__.py

@@ -112,6 +112,9 @@ from .memoryset.potential_duplicate_groups_memoryset_name_or_id_potential_duplic
 from .memoryset.query_memoryset_memoryset_name_or_id_memories_post import (
     sync as query_memoryset,
 )
+from .memoryset.suggest_cascading_edits_memoryset_name_or_id_memory_memory_id_cascading_edits_post import (
+    sync as suggest_cascading_edits,
+)
 from .memoryset.update_memories_gpu_memoryset_name_or_id_memories_patch import (
     sync as update_memories_gpu,
 )
@@ -225,6 +228,7 @@ __all__ = [
     "predict_gpu",
     "query_memoryset",
     "record_prediction_feedback",
+    "suggest_cascading_edits",
     "update_memories_gpu",
     "update_memory_gpu",
     "update_memoryset",

orca_sdk/_generated_api_client/api/memoryset/suggest_cascading_edits_memoryset_name_or_id_memory_memory_id_cascading_edits_post.py

@@ -0,0 +1,233 @@
+"""
+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/endpoint_module.py.jinja
+
+The main changes are:
+- Update the API call responses to either return the successful response type or raise an error by:
+    - Updating the _parse_response function to raise an error if the response status code is not in the 2xx range
+- Inject a client into every method via a context manager
+"""
+
+from http import HTTPStatus
+from typing import Any, Dict, List
+
+import httpx
+
+from ...client import _client_context
+from ...errors import get_error_for_response
+from ...models.cascade_edit_suggestions_request import CascadeEditSuggestionsRequest
+from ...models.cascading_edit_suggestion import CascadingEditSuggestion
+from ...types import Response
+
+
+def _get_kwargs(
+    name_or_id: str,
+    memory_id: str,
+    *,
+    body: CascadeEditSuggestionsRequest,
+) -> dict[str, Any]:
+    headers: Dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": f"/memoryset/{name_or_id}/memory/{memory_id}/cascading_edits",
+    }
+
+    _body = body.to_dict()
+
+    _kwargs["json"] = _body
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(*, response: httpx.Response) -> List["CascadingEditSuggestion"]:
+    if response.status_code == 200:
+        response_200 = []
+        _response_200 = response.json()
+        for response_200_item_data in _response_200:
+            response_200_item = CascadingEditSuggestion.from_dict(response_200_item_data)
+
+            response_200.append(response_200_item)
+
+        return response_200
+    if response.status_code == 401:
+        raise get_error_for_response(response)
+    if response.status_code == 500:
+        raise get_error_for_response(response)
+    if response.status_code == 503:
+        raise get_error_for_response(response)
+    if response.status_code == 404:
+        raise get_error_for_response(response)
+    else:
+        raise get_error_for_response(response)
+
+
+def _build_response(*, response: httpx.Response) -> Response[List["CascadingEditSuggestion"]]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(response=response),
+    )
+
+
+def sync_detailed(
+    name_or_id: str,
+    memory_id: str,
+    *,
+    body: CascadeEditSuggestionsRequest,
+) -> Response[List["CascadingEditSuggestion"]]:
+    """Suggest Cascading Edits
+
+     Suggest nearby neighbors who might be mislabeled based on a memory's label change.
+
+    Args:
+        name_or_id: The name or ID of the memoryset.
+        memory_id: The ID of the memory to base suggestions on.
+        auth_context: The authentication context containing the organization ID.
+        request: The request containing parameters for generating suggestions.
+
+    Args:
+        name_or_id (str):
+        memory_id (str):
+        body (CascadeEditSuggestionsRequest):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[List['CascadingEditSuggestion']]
+    """
+
+    kwargs = _get_kwargs(
+        name_or_id=name_or_id,
+        memory_id=memory_id,
+        body=body,
+    )
+
+    with _client_context() as client:
+        response = client.get_httpx_client().request(
+            **kwargs,
+        )
+
+        return _build_response(response=response)
+
+
+def sync(
+    name_or_id: str,
+    memory_id: str,
+    *,
+    body: CascadeEditSuggestionsRequest,
+) -> List["CascadingEditSuggestion"]:
+    """Suggest Cascading Edits
+
+     Suggest nearby neighbors who might be mislabeled based on a memory's label change.
+
+    Args:
+        name_or_id: The name or ID of the memoryset.
+        memory_id: The ID of the memory to base suggestions on.
+        auth_context: The authentication context containing the organization ID.
+        request: The request containing parameters for generating suggestions.
+
+    Args:
+        name_or_id (str):
+        memory_id (str):
+        body (CascadeEditSuggestionsRequest):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        List['CascadingEditSuggestion']
+    """
+
+    return sync_detailed(
+        name_or_id=name_or_id,
+        memory_id=memory_id,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    name_or_id: str,
+    memory_id: str,
+    *,
+    body: CascadeEditSuggestionsRequest,
+) -> Response[List["CascadingEditSuggestion"]]:
+    """Suggest Cascading Edits
+
+     Suggest nearby neighbors who might be mislabeled based on a memory's label change.
+
+    Args:
+        name_or_id: The name or ID of the memoryset.
+        memory_id: The ID of the memory to base suggestions on.
+        auth_context: The authentication context containing the organization ID.
+        request: The request containing parameters for generating suggestions.
+
+    Args:
+        name_or_id (str):
+        memory_id (str):
+        body (CascadeEditSuggestionsRequest):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[List['CascadingEditSuggestion']]
+    """
+
+    kwargs = _get_kwargs(
+        name_or_id=name_or_id,
+        memory_id=memory_id,
+        body=body,
+    )
+
+    with _client_context() as client:
+        response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(response=response)
+
+
+async def asyncio(
+    name_or_id: str,
+    memory_id: str,
+    *,
+    body: CascadeEditSuggestionsRequest,
+) -> List["CascadingEditSuggestion"]:
+    """Suggest Cascading Edits
+
+     Suggest nearby neighbors who might be mislabeled based on a memory's label change.
+
+    Args:
+        name_or_id: The name or ID of the memoryset.
+        memory_id: The ID of the memory to base suggestions on.
+        auth_context: The authentication context containing the organization ID.
+        request: The request containing parameters for generating suggestions.
+
+    Args:
+        name_or_id (str):
+        memory_id (str):
+        body (CascadeEditSuggestionsRequest):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        List['CascadingEditSuggestion']
+    """
+
+    return (
+        await asyncio_detailed(
+            name_or_id=name_or_id,
+            memory_id=memory_id,
+            body=body,
+        )
+    ).parsed

orca_sdk/_generated_api_client/models/__init__.py

@@ -6,6 +6,8 @@ from .api_key_metadata_scope_item import ApiKeyMetadataScopeItem
 from .base_label_prediction_result import BaseLabelPredictionResult
 from .base_model import BaseModel
 from .body_create_datasource_datasource_post import BodyCreateDatasourceDatasourcePost
+from .cascade_edit_suggestions_request import CascadeEditSuggestionsRequest
+from .cascading_edit_suggestion import CascadingEditSuggestion
 from .classification_evaluation_result import ClassificationEvaluationResult
 from .clone_labeled_memoryset_request import CloneLabeledMemorysetRequest
 from .cluster_metrics import ClusterMetrics
@@ -118,6 +120,8 @@ __all__ = (
     "BaseLabelPredictionResult",
     "BaseModel",
     "BodyCreateDatasourceDatasourcePost",
+    "CascadeEditSuggestionsRequest",
+    "CascadingEditSuggestion",
     "ClassificationEvaluationResult",
     "CloneLabeledMemorysetRequest",
     "ClusterMetrics",

orca_sdk/_generated_api_client/models/cascade_edit_suggestions_request.py

@@ -0,0 +1,154 @@
+"""
+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
+
+from ..types import UNSET, Unset
+
+T = TypeVar("T", bound="CascadeEditSuggestionsRequest")
+
+
+@_attrs_define
+class CascadeEditSuggestionsRequest:
+    """
+    Attributes:
+        old_label (int):
+        new_label (int):
+        max_neighbors (Union[Unset, int]):  Default: 50.
+        max_validation_neighbors (Union[Unset, int]):  Default: 10.
+        similarity_threshold (Union[None, Unset, float]):
+        only_if_has_old_label (Union[Unset, bool]):  Default: True.
+        exclude_if_new_label (Union[Unset, bool]):  Default: True.
+        suggestion_cooldown_time (Union[Unset, float]):  Default: 86400.0.
+        label_confirmation_cooldown_time (Union[Unset, float]):  Default: 604800.0.
+    """
+
+    old_label: int
+    new_label: int
+    max_neighbors: Union[Unset, int] = 50
+    max_validation_neighbors: Union[Unset, int] = 10
+    similarity_threshold: Union[None, Unset, float] = UNSET
+    only_if_has_old_label: Union[Unset, bool] = True
+    exclude_if_new_label: Union[Unset, bool] = True
+    suggestion_cooldown_time: Union[Unset, float] = 86400.0
+    label_confirmation_cooldown_time: Union[Unset, float] = 604800.0
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        old_label = self.old_label
+
+        new_label = self.new_label
+
+        max_neighbors = self.max_neighbors
+
+        max_validation_neighbors = self.max_validation_neighbors
+
+        similarity_threshold: Union[None, Unset, float]
+        if isinstance(self.similarity_threshold, Unset):
+            similarity_threshold = UNSET
+        else:
+            similarity_threshold = self.similarity_threshold
+
+        only_if_has_old_label = self.only_if_has_old_label
+
+        exclude_if_new_label = self.exclude_if_new_label
+
+        suggestion_cooldown_time = self.suggestion_cooldown_time
+
+        label_confirmation_cooldown_time = self.label_confirmation_cooldown_time
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "old_label": old_label,
+                "new_label": new_label,
+            }
+        )
+        if max_neighbors is not UNSET:
+            field_dict["max_neighbors"] = max_neighbors
+        if max_validation_neighbors is not UNSET:
+            field_dict["max_validation_neighbors"] = max_validation_neighbors
+        if similarity_threshold is not UNSET:
+            field_dict["similarity_threshold"] = similarity_threshold
+        if only_if_has_old_label is not UNSET:
+            field_dict["only_if_has_old_label"] = only_if_has_old_label
+        if exclude_if_new_label is not UNSET:
+            field_dict["exclude_if_new_label"] = exclude_if_new_label
+        if suggestion_cooldown_time is not UNSET:
+            field_dict["suggestion_cooldown_time"] = suggestion_cooldown_time
+        if label_confirmation_cooldown_time is not UNSET:
+            field_dict["label_confirmation_cooldown_time"] = label_confirmation_cooldown_time
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        d = src_dict.copy()
+        old_label = d.pop("old_label")
+
+        new_label = d.pop("new_label")
+
+        max_neighbors = d.pop("max_neighbors", UNSET)
+
+        max_validation_neighbors = d.pop("max_validation_neighbors", UNSET)
+
+        def _parse_similarity_threshold(data: object) -> Union[None, Unset, float]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, float], data)
+
+        similarity_threshold = _parse_similarity_threshold(d.pop("similarity_threshold", UNSET))
+
+        only_if_has_old_label = d.pop("only_if_has_old_label", UNSET)
+
+        exclude_if_new_label = d.pop("exclude_if_new_label", UNSET)
+
+        suggestion_cooldown_time = d.pop("suggestion_cooldown_time", UNSET)
+
+        label_confirmation_cooldown_time = d.pop("label_confirmation_cooldown_time", UNSET)
+
+        cascade_edit_suggestions_request = cls(
+            old_label=old_label,
+            new_label=new_label,
+            max_neighbors=max_neighbors,
+            max_validation_neighbors=max_validation_neighbors,
+            similarity_threshold=similarity_threshold,
+            only_if_has_old_label=only_if_has_old_label,
+            exclude_if_new_label=exclude_if_new_label,
+            suggestion_cooldown_time=suggestion_cooldown_time,
+            label_confirmation_cooldown_time=label_confirmation_cooldown_time,
+        )
+
+        cascade_edit_suggestions_request.additional_properties = d
+        return cascade_edit_suggestions_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/cascading_edit_suggestion.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 TYPE_CHECKING, Any, Type, TypeVar
+
+from attrs import define as _attrs_define
+from attrs import field as _attrs_field
+
+if TYPE_CHECKING:
+    from ..models.labeled_memory_lookup import LabeledMemoryLookup
+
+
+T = TypeVar("T", bound="CascadingEditSuggestion")
+
+
+@_attrs_define
+class CascadingEditSuggestion:
+    """
+    Attributes:
+        neighbor (LabeledMemoryLookup): Single labeled memory lookup result.
+        suggested_label (int):
+        lookup_score (float):
+    """
+
+    neighbor: "LabeledMemoryLookup"
+    suggested_label: int
+    lookup_score: float
+    additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
+
+    def to_dict(self) -> dict[str, Any]:
+        neighbor = self.neighbor.to_dict()
+
+        suggested_label = self.suggested_label
+
+        lookup_score = self.lookup_score
+
+        field_dict: dict[str, Any] = {}
+        field_dict.update(self.additional_properties)
+        field_dict.update(
+            {
+                "neighbor": neighbor,
+                "suggested_label": suggested_label,
+                "lookup_score": lookup_score,
+            }
+        )
+
+        return field_dict
+
+    @classmethod
+    def from_dict(cls: Type[T], src_dict: dict[str, Any]) -> T:
+        from ..models.labeled_memory_lookup import LabeledMemoryLookup
+
+        d = src_dict.copy()
+        neighbor = LabeledMemoryLookup.from_dict(d.pop("neighbor"))
+
+        suggested_label = d.pop("suggested_label")
+
+        lookup_score = d.pop("lookup_score")
+
+        cascading_edit_suggestion = cls(
+            neighbor=neighbor,
+            suggested_label=suggested_label,
+            lookup_score=lookup_score,
+        )
+
+        cascading_edit_suggestion.additional_properties = d
+        return cascading_edit_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/classification_evaluation_result.py

@@ -15,6 +15,8 @@ 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 ..types import UNSET, Unset
+
 if TYPE_CHECKING:
     from ..models.precision_recall_curve import PrecisionRecallCurve
     from ..models.roc_curve import ROCCurve
@@ -32,6 +34,9 @@ class ClassificationEvaluationResult:
         loss (float):
         precision_recall_curve (Union['PrecisionRecallCurve', None]):
         roc_curve (Union['ROCCurve', None]):
+        anomaly_score_mean (Union[None, Unset, float]):
+        anomaly_score_median (Union[None, Unset, float]):
+        anomaly_score_variance (Union[None, Unset, float]):
     """
 
     f1_score: float
@@ -39,6 +44,9 @@ class ClassificationEvaluationResult:
     loss: float
     precision_recall_curve: Union["PrecisionRecallCurve", None]
     roc_curve: Union["ROCCurve", None]
+    anomaly_score_mean: Union[None, Unset, float] = UNSET
+    anomaly_score_median: Union[None, Unset, float] = UNSET
+    anomaly_score_variance: Union[None, Unset, float] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
     def to_dict(self) -> dict[str, Any]:
@@ -63,6 +71,24 @@ class ClassificationEvaluationResult:
         else:
             roc_curve = self.roc_curve
 
+        anomaly_score_mean: Union[None, Unset, float]
+        if isinstance(self.anomaly_score_mean, Unset):
+            anomaly_score_mean = UNSET
+        else:
+            anomaly_score_mean = self.anomaly_score_mean
+
+        anomaly_score_median: Union[None, Unset, float]
+        if isinstance(self.anomaly_score_median, Unset):
+            anomaly_score_median = UNSET
+        else:
+            anomaly_score_median = self.anomaly_score_median
+
+        anomaly_score_variance: Union[None, Unset, float]
+        if isinstance(self.anomaly_score_variance, Unset):
+            anomaly_score_variance = UNSET
+        else:
+            anomaly_score_variance = self.anomaly_score_variance
+
         field_dict: dict[str, Any] = {}
         field_dict.update(self.additional_properties)
         field_dict.update(
@@ -74,6 +100,12 @@ class ClassificationEvaluationResult:
                 "roc_curve": roc_curve,
             }
         )
+        if anomaly_score_mean is not UNSET:
+            field_dict["anomaly_score_mean"] = anomaly_score_mean
+        if anomaly_score_median is not UNSET:
+            field_dict["anomaly_score_median"] = anomaly_score_median
+        if anomaly_score_variance is not UNSET:
+            field_dict["anomaly_score_variance"] = anomaly_score_variance
 
         return field_dict
 
@@ -119,12 +151,42 @@ class ClassificationEvaluationResult:
 
         roc_curve = _parse_roc_curve(d.pop("roc_curve"))
 
+        def _parse_anomaly_score_mean(data: object) -> Union[None, Unset, float]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, float], data)
+
+        anomaly_score_mean = _parse_anomaly_score_mean(d.pop("anomaly_score_mean", UNSET))
+
+        def _parse_anomaly_score_median(data: object) -> Union[None, Unset, float]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, float], data)
+
+        anomaly_score_median = _parse_anomaly_score_median(d.pop("anomaly_score_median", UNSET))
+
+        def _parse_anomaly_score_variance(data: object) -> Union[None, Unset, float]:
+            if data is None:
+                return data
+            if isinstance(data, Unset):
+                return data
+            return cast(Union[None, Unset, float], data)
+
+        anomaly_score_variance = _parse_anomaly_score_variance(d.pop("anomaly_score_variance", UNSET))
+
         classification_evaluation_result = cls(
             f1_score=f1_score,
             accuracy=accuracy,
             loss=loss,
             precision_recall_curve=precision_recall_curve,
             roc_curve=roc_curve,
+            anomaly_score_mean=anomaly_score_mean,
+            anomaly_score_median=anomaly_score_median,
+            anomaly_score_variance=anomaly_score_variance,
         )
 
         classification_evaluation_result.additional_properties = d

orca_sdk/_generated_api_client/models/filter_item_field_type_0_item.py

@@ -3,6 +3,7 @@ from enum import Enum
 
 class FilterItemFieldType0Item(str, Enum):
     CREATED_AT = "created_at"
+    EDITED_AT = "edited_at"
     LABEL = "label"
     MEMORY_ID = "memory_id"
     METADATA = "metadata"

orca_sdk/_generated_api_client/models/label_prediction_memory_lookup.py

@@ -38,6 +38,7 @@ class LabelPredictionMemoryLookup:
         memory_version (int):
         created_at (datetime.datetime):
         updated_at (datetime.datetime):
+        edited_at (datetime.datetime):
         metrics (MemoryMetrics):
         label (int):
         label_name (Union[None, str]):
@@ -54,6 +55,7 @@ class LabelPredictionMemoryLookup:
     memory_version: int
     created_at: datetime.datetime
     updated_at: datetime.datetime
+    edited_at: datetime.datetime
     metrics: "MemoryMetrics"
     label: int
     label_name: Union[None, str]
@@ -81,6 +83,8 @@ class LabelPredictionMemoryLookup:
 
         updated_at = self.updated_at.isoformat()
 
+        edited_at = self.edited_at.isoformat()
+
         metrics = self.metrics.to_dict()
 
         label = self.label
@@ -106,6 +110,7 @@ class LabelPredictionMemoryLookup:
                 "memory_version": memory_version,
                 "created_at": created_at,
                 "updated_at": updated_at,
+                "edited_at": edited_at,
                 "metrics": metrics,
                 "label": label,
                 "label_name": label_name,
@@ -148,6 +153,8 @@ class LabelPredictionMemoryLookup:
 
         updated_at = isoparse(d.pop("updated_at"))
 
+        edited_at = isoparse(d.pop("edited_at"))
+
         metrics = MemoryMetrics.from_dict(d.pop("metrics"))
 
         label = d.pop("label")
@@ -174,6 +181,7 @@ class LabelPredictionMemoryLookup:
             memory_version=memory_version,
             created_at=created_at,
             updated_at=updated_at,
+            edited_at=edited_at,
             metrics=metrics,
             label=label,
             label_name=label_name,