orca-sdk 0.1.10__py3-none-any.whl → 0.1.11__py3-none-any.whl

orca_sdk/async_client.py

@@ -29,9 +29,7 @@ from httpx import (
     AsyncBaseTransport,
     AsyncClient,
     AsyncHTTPTransport,
-    BaseTransport,
     Headers,
-    HTTPTransport,
     Limits,
     Proxy,
     Request,
@@ -211,12 +209,13 @@ class DatasetFilterItem(TypedDict):
     value: Any
 
 
-class DeleteMemoriesRequest(TypedDict):
-    memory_ids: list[str]
+class DeleteMemoriesResponse(TypedDict):
+    deleted_count: int
 
 
 class DeleteMemorysetsRequest(TypedDict):
     memoryset_ids: list[str]
+    cascade: NotRequired[bool]
 
 
 class EmbedRequest(TypedDict):
@@ -292,7 +291,20 @@ class FilterItem(TypedDict):
         ]
     )
     op: Literal["==", "!=", ">", ">=", "<", "<=", "in", "not in", "like", "contains all", "contains any"]
-    value: str | int | float | bool | list[str | None] | list[int] | list[float] | list[bool] | None
+    value: (
+        str
+        | int
+        | float
+        | bool
+        | list[None]
+        | list[str]
+        | list[str | None]
+        | list[int]
+        | list[int | None]
+        | list[float]
+        | list[bool]
+        | None
+    )
     transform: NotRequired[Literal["length"]]
 
 
@@ -355,6 +367,13 @@ class LabelPercentage(TypedDict):
     percentage: float
 
 
+class LabeledBatchMemoryUpdatePatch(TypedDict):
+    metadata: NotRequired[dict[str, str | int | float | bool | None] | None]
+    source_id: NotRequired[str | None]
+    partition_id: NotRequired[str | None]
+    label: NotRequired[int | None]
+
+
 class LabeledExample(TypedDict):
     text: str
     label_name: str
@@ -373,8 +392,6 @@ class ListMemoriesRequest(TypedDict):
     offset: NotRequired[int]
     limit: NotRequired[int]
     filters: NotRequired[list[FilterItem]]
-    partition_id: NotRequired[str | None]
-    partition_filter_mode: NotRequired[Literal["ignore_partitions", "include_global", "exclude_global", "only_global"]]
 
 
 class LookupRequest(TypedDict):
@@ -526,6 +543,7 @@ class NotFoundErrorResponse(TypedDict):
             "api_key",
             "datasource",
             "memoryset",
+            "predictive_model",
             "classification_model",
             "regression_model",
             "prediction",
@@ -730,6 +748,13 @@ class ScorePredictionWithMemoriesAndFeedback(TypedDict):
     feedbacks: list[PredictionFeedback]
 
 
+class ScoredBatchMemoryUpdatePatch(TypedDict):
+    metadata: NotRequired[dict[str, str | int | float | bool | None] | None]
+    source_id: NotRequired[str | None]
+    partition_id: NotRequired[str | None]
+    score: NotRequired[float | None]
+
+
 class ScoredMemory(TypedDict):
     value: str | bytes
     embedding: list[float]
@@ -835,6 +860,10 @@ class UnauthorizedErrorResponse(TypedDict):
     reason: str
 
 
+class UpdateMemoriesResponse(TypedDict):
+    updated_count: int
+
+
 class UpdateOrgPlanRequest(TypedDict):
     tier: Literal["FREE", "PRO", "ENTERPRISE", "CANCELLED"]
 
@@ -885,6 +914,7 @@ class GetMemorysetByNameOrIdParams(TypedDict):
 
 class DeleteMemorysetByNameOrIdParams(TypedDict):
     name_or_id: str
+    cascade: NotRequired[bool]
 
 
 class PostGpuMemorysetByNameOrIdLookupParams(TypedDict):
@@ -965,6 +995,7 @@ class GetFinetunedEmbeddingModelByNameOrIdParams(TypedDict):
 
 class DeleteFinetunedEmbeddingModelByNameOrIdParams(TypedDict):
     name_or_id: str
+    cascade: NotRequired[bool]
 
 
 class PostGpuFinetunedEmbeddingModelByNameOrIdEmbeddingParams(TypedDict):
@@ -1058,6 +1089,20 @@ class GetDatasourceByNameOrIdDownloadParams(TypedDict):
     """
 
 
+class GetClassificationModelParams(TypedDict):
+    memoryset_name_or_id: NotRequired[str | None]
+    """
+    Filter by memoryset name or ID
+    """
+
+
+class GetRegressionModelParams(TypedDict):
+    memoryset_name_or_id: NotRequired[str | None]
+    """
+    Filter by memoryset name or ID
+    """
+
+
 class PatchClassificationModelByNameOrIdParams(TypedDict):
     name_or_id: str
 
@@ -1082,6 +1127,13 @@ class DeleteRegressionModelByNameOrIdParams(TypedDict):
     name_or_id: str
 
 
+class GetPredictiveModelParams(TypedDict):
+    memoryset_name_or_id: NotRequired[str | None]
+    """
+    Filter by memoryset name or ID
+    """
+
+
 class PostGpuClassificationModelByNameOrIdPredictionParams(TypedDict):
     name_or_id: str
 
@@ -1404,6 +1456,11 @@ class DatasourceMetadata(TypedDict):
     updated_at: str
 
 
+class DeleteMemoriesRequest(TypedDict):
+    memory_ids: NotRequired[list[str] | None]
+    filters: NotRequired[list[FilterItem] | None]
+
+
 class EmbeddingEvaluationResponse(TypedDict):
     job_id: str
     org_id: str
@@ -1704,7 +1761,10 @@ class WorkerInfo(TypedDict):
 PatchGpuMemorysetByNameOrIdMemoryRequest: TypeAlias = LabeledMemoryUpdate | ScoredMemoryUpdate
 
 
-PatchGpuMemorysetByNameOrIdMemoriesRequest: TypeAlias = list[LabeledMemoryUpdate] | list[ScoredMemoryUpdate]
+class BatchMemoryUpdateRequest(TypedDict):
+    updates: NotRequired[list[LabeledMemoryUpdate | ScoredMemoryUpdate] | None]
+    filters: NotRequired[list[FilterItem] | None]
+    patch: NotRequired[LabeledBatchMemoryUpdatePatch | ScoredBatchMemoryUpdatePatch | None]
 
 
 class CascadingEditSuggestion(TypedDict):
@@ -1992,7 +2052,7 @@ class OrcaAsyncClient(AsyncClient):
         self,
         path: Literal["/memoryset"],
         *,
-        params: GetMemorysetParams,
+        params: GetMemorysetParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2048,7 +2108,7 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[list[LabeledMemory]] | list[list[ScoredMemory]]:
+    ) -> list[list[LabeledMemory]] | list[list[ScoredMemory]] | None:
         pass
 
     @overload
@@ -2326,7 +2386,7 @@ class OrcaAsyncClient(AsyncClient):
         self,
         path: Literal["/classification_model"],
         *,
-        params: None = None,
+        params: GetClassificationModelParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2342,7 +2402,7 @@ class OrcaAsyncClient(AsyncClient):
         self,
         path: Literal["/regression_model"],
         *,
-        params: None = None,
+        params: GetRegressionModelParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2390,7 +2450,7 @@ class OrcaAsyncClient(AsyncClient):
         self,
         path: Literal["/predictive_model"],
         *,
-        params: None = None,
+        params: GetPredictiveModelParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2502,7 +2562,7 @@ class OrcaAsyncClient(AsyncClient):
         self,
         path: Literal["/job"],
         *,
-        params: GetJobParams,
+        params: GetJobParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2518,7 +2578,7 @@ class OrcaAsyncClient(AsyncClient):
         self,
         path: Literal["/worker"],
         *,
-        params: GetWorkerParams,
+        params: GetWorkerParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -3150,7 +3210,7 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> None:
+    ) -> DeleteMemoriesResponse:
         pass
 
     @overload
@@ -3867,7 +3927,7 @@ class OrcaAsyncClient(AsyncClient):
         path: Literal["/gpu/memoryset/{name_or_id}/memories"],
         *,
         params: PatchGpuMemorysetByNameOrIdMemoriesParams,
-        json: PatchGpuMemorysetByNameOrIdMemoriesRequest,
+        json: BatchMemoryUpdateRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3878,7 +3938,7 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[LabeledMemory] | list[ScoredMemory]:
+    ) -> UpdateMemoriesResponse:
         pass
 
     @overload

orca_sdk/classification_model.py

@@ -279,7 +279,7 @@ class ClassificationModel:
             List of handles to all classification models in the OrcaCloud
         """
         client = OrcaClient._resolve_client()
-        return [cls(metadata) for metadata in client.GET("/classification_model")]
+        return [cls(metadata) for metadata in client.GET("/classification_model", params={})]
 
     @classmethod
     def drop(cls, name_or_id: str, if_not_exists: DropMode = "error"):

orca_sdk/classification_model_test.py

@@ -108,6 +108,14 @@ def test_list_models_unauthorized(unauthorized_client, classification_model: Cla
         assert ClassificationModel.all() == []
 
 
+def test_memoryset_classification_models_property(
+    classification_model: ClassificationModel, readonly_memoryset: LabeledMemoryset
+):
+    models = readonly_memoryset.classification_models
+    assert len(models) > 0
+    assert any(model.id == classification_model.id for model in models)
+
+
 def test_update_model_attributes(classification_model: ClassificationModel):
     classification_model.description = "New description"
     assert classification_model.description == "New description"
@@ -162,6 +170,29 @@ def test_delete_memoryset_before_model_constraint_violation(hf_dataset):
         LabeledMemoryset.drop(memoryset.id)
 
 
+def test_delete_memoryset_with_model_cascade(hf_dataset):
+    """Test that cascade=False prevents deletion and cascade=True allows it."""
+    memoryset = LabeledMemoryset.from_hf_dataset("test_memoryset_cascade_delete", hf_dataset)
+    model = ClassificationModel.create("test_model_cascade_delete", memoryset)
+
+    # Verify model exists
+    assert ClassificationModel.open(model.name) is not None
+
+    # Without cascade, deletion should fail
+    with pytest.raises(RuntimeError):
+        LabeledMemoryset.drop(memoryset.id, cascade=False)
+
+    # Model should still exist
+    assert ClassificationModel.exists(model.name)
+
+    # With cascade, deletion should succeed
+    LabeledMemoryset.drop(memoryset.id, cascade=True)
+
+    # Model should be deleted along with the memoryset
+    assert not ClassificationModel.exists(model.name)
+    assert not LabeledMemoryset.exists(memoryset.name)
+
+
 @pytest.mark.parametrize("data_type", ["dataset", "datasource"])
 def test_evaluate(classification_model, eval_datasource: Datasource, eval_dataset: Dataset, data_type):
     result = (
@@ -683,28 +714,27 @@ def test_predict_with_memoryset_update(writable_memoryset: LabeledMemoryset):
         num_classes=2,
         memory_lookup_count=3,
     )
-
-    prediction = model.predict("Do you love soup?")
-    assert prediction.label == 0
-    assert prediction.label_name == "soup"
-
-    # insert new memories
-    writable_memoryset.insert(
-        [
-            {"value": "Do you love soup?", "label": 1, "key": "g1"},
-            {"value": "Do you love soup for dinner?", "label": 1, "key": "g2"},
-            {"value": "Do you love crackers?", "label": 1, "key": "g2"},
-            {"value": "Do you love broth?", "label": 1, "key": "g2"},
-            {"value": "Do you love chicken soup?", "label": 1, "key": "g2"},
-            {"value": "Do you love chicken soup for dinner?", "label": 1, "key": "g2"},
-            {"value": "Do you love chicken soup for dinner?", "label": 1, "key": "g2"},
-        ],
-    )
-    prediction = model.predict("Do you love soup?")
-    assert prediction.label == 1
-    assert prediction.label_name == "cats"
-
-    ClassificationModel.drop("test_predict_with_memoryset_update")
+    try:
+        prediction = model.predict("Do you love soup?", partition_filter_mode="ignore_partitions")
+        assert prediction.label == 0
+        assert prediction.label_name == "soup"
+        # insert new memories
+        writable_memoryset.insert(
+            [
+                {"value": "Do you love soup?", "label": 1, "key": "g1"},
+                {"value": "Do you love soup for dinner?", "label": 1, "key": "g2"},
+                {"value": "Do you love crackers?", "label": 1, "key": "g2"},
+                {"value": "Do you love broth?", "label": 1, "key": "g2"},
+                {"value": "Do you love chicken soup?", "label": 1, "key": "g2"},
+                {"value": "Do you love chicken soup for dinner?", "label": 1, "key": "g2"},
+                {"value": "Do you love chicken soup for dinner?", "label": 1, "key": "g2"},
+            ],
+        )
+        prediction = model.predict("Do you love soup?")
+        assert prediction.label == 1
+        assert prediction.label_name == "cats"
+    finally:
+        ClassificationModel.drop("test_predict_with_memoryset_update")
 
 
 def test_last_prediction_with_batch(classification_model: ClassificationModel):
@@ -828,6 +858,23 @@ def test_predict_with_prompt(classification_model: ClassificationModel):
     assert prediction_without_prompt.label is not None
 
 
+def test_predict_with_empty_partition(fully_partitioned_classification_resources):
+    datasource, memoryset, classification_model = fully_partitioned_classification_resources
+
+    assert memoryset.length == 15
+
+    with pytest.raises(RuntimeError, match="lookup failed to return the correct number of memories"):
+        classification_model.predict("i love cats", partition_filter_mode="only_global")
+
+    with pytest.raises(RuntimeError, match="lookup failed to return the correct number of memories"):
+        classification_model.predict(
+            "i love cats", partition_filter_mode="exclude_global", partition_id="p_does_not_exist"
+        )
+
+    with pytest.raises(RuntimeError, match="lookup failed to return the correct number of memories"):
+        classification_model.evaluate(datasource, partition_filter_mode="only_global")
+
+
 @pytest.mark.asyncio
 async def test_predict_async_single(classification_model: ClassificationModel, label_names: list[str]):
     """Test async prediction with a single value"""

orca_sdk/client.py

@@ -209,12 +209,13 @@ class DatasetFilterItem(TypedDict):
     value: Any
 
 
-class DeleteMemoriesRequest(TypedDict):
-    memory_ids: list[str]
+class DeleteMemoriesResponse(TypedDict):
+    deleted_count: int
 
 
 class DeleteMemorysetsRequest(TypedDict):
     memoryset_ids: list[str]
+    cascade: NotRequired[bool]
 
 
 class EmbedRequest(TypedDict):
@@ -290,7 +291,20 @@ class FilterItem(TypedDict):
         ]
     )
     op: Literal["==", "!=", ">", ">=", "<", "<=", "in", "not in", "like", "contains all", "contains any"]
-    value: str | int | float | bool | list[str | None] | list[int] | list[float] | list[bool] | None
+    value: (
+        str
+        | int
+        | float
+        | bool
+        | list[None]
+        | list[str]
+        | list[str | None]
+        | list[int]
+        | list[int | None]
+        | list[float]
+        | list[bool]
+        | None
+    )
     transform: NotRequired[Literal["length"]]
 
 
@@ -353,6 +367,13 @@ class LabelPercentage(TypedDict):
     percentage: float
 
 
+class LabeledBatchMemoryUpdatePatch(TypedDict):
+    metadata: NotRequired[dict[str, str | int | float | bool | None] | None]
+    source_id: NotRequired[str | None]
+    partition_id: NotRequired[str | None]
+    label: NotRequired[int | None]
+
+
 class LabeledExample(TypedDict):
     text: str
     label_name: str
@@ -371,8 +392,6 @@ class ListMemoriesRequest(TypedDict):
     offset: NotRequired[int]
     limit: NotRequired[int]
     filters: NotRequired[list[FilterItem]]
-    partition_id: NotRequired[str | None]
-    partition_filter_mode: NotRequired[Literal["ignore_partitions", "include_global", "exclude_global", "only_global"]]
 
 
 class LookupRequest(TypedDict):
@@ -524,6 +543,7 @@ class NotFoundErrorResponse(TypedDict):
             "api_key",
             "datasource",
             "memoryset",
+            "predictive_model",
             "classification_model",
             "regression_model",
             "prediction",
@@ -728,6 +748,13 @@ class ScorePredictionWithMemoriesAndFeedback(TypedDict):
     feedbacks: list[PredictionFeedback]
 
 
+class ScoredBatchMemoryUpdatePatch(TypedDict):
+    metadata: NotRequired[dict[str, str | int | float | bool | None] | None]
+    source_id: NotRequired[str | None]
+    partition_id: NotRequired[str | None]
+    score: NotRequired[float | None]
+
+
 class ScoredMemory(TypedDict):
     value: str | bytes
     embedding: list[float]
@@ -833,6 +860,10 @@ class UnauthorizedErrorResponse(TypedDict):
     reason: str
 
 
+class UpdateMemoriesResponse(TypedDict):
+    updated_count: int
+
+
 class UpdateOrgPlanRequest(TypedDict):
     tier: Literal["FREE", "PRO", "ENTERPRISE", "CANCELLED"]
 
@@ -883,6 +914,7 @@ class GetMemorysetByNameOrIdParams(TypedDict):
 
 class DeleteMemorysetByNameOrIdParams(TypedDict):
     name_or_id: str
+    cascade: NotRequired[bool]
 
 
 class PostGpuMemorysetByNameOrIdLookupParams(TypedDict):
@@ -963,6 +995,7 @@ class GetFinetunedEmbeddingModelByNameOrIdParams(TypedDict):
 
 class DeleteFinetunedEmbeddingModelByNameOrIdParams(TypedDict):
     name_or_id: str
+    cascade: NotRequired[bool]
 
 
 class PostGpuFinetunedEmbeddingModelByNameOrIdEmbeddingParams(TypedDict):
@@ -1056,6 +1089,20 @@ class GetDatasourceByNameOrIdDownloadParams(TypedDict):
     """
 
 
+class GetClassificationModelParams(TypedDict):
+    memoryset_name_or_id: NotRequired[str | None]
+    """
+    Filter by memoryset name or ID
+    """
+
+
+class GetRegressionModelParams(TypedDict):
+    memoryset_name_or_id: NotRequired[str | None]
+    """
+    Filter by memoryset name or ID
+    """
+
+
 class PatchClassificationModelByNameOrIdParams(TypedDict):
     name_or_id: str
 
@@ -1080,6 +1127,13 @@ class DeleteRegressionModelByNameOrIdParams(TypedDict):
     name_or_id: str
 
 
+class GetPredictiveModelParams(TypedDict):
+    memoryset_name_or_id: NotRequired[str | None]
+    """
+    Filter by memoryset name or ID
+    """
+
+
 class PostGpuClassificationModelByNameOrIdPredictionParams(TypedDict):
     name_or_id: str
 
@@ -1402,6 +1456,11 @@ class DatasourceMetadata(TypedDict):
     updated_at: str
 
 
+class DeleteMemoriesRequest(TypedDict):
+    memory_ids: NotRequired[list[str] | None]
+    filters: NotRequired[list[FilterItem] | None]
+
+
 class EmbeddingEvaluationResponse(TypedDict):
     job_id: str
     org_id: str
@@ -1702,7 +1761,10 @@ class WorkerInfo(TypedDict):
 PatchGpuMemorysetByNameOrIdMemoryRequest: TypeAlias = LabeledMemoryUpdate | ScoredMemoryUpdate
 
 
-PatchGpuMemorysetByNameOrIdMemoriesRequest: TypeAlias = list[LabeledMemoryUpdate] | list[ScoredMemoryUpdate]
+class BatchMemoryUpdateRequest(TypedDict):
+    updates: NotRequired[list[LabeledMemoryUpdate | ScoredMemoryUpdate] | None]
+    filters: NotRequired[list[FilterItem] | None]
+    patch: NotRequired[LabeledBatchMemoryUpdatePatch | ScoredBatchMemoryUpdatePatch | None]
 
 
 class CascadingEditSuggestion(TypedDict):
@@ -1990,7 +2052,7 @@ class OrcaClient(Client):
         self,
         path: Literal["/memoryset"],
         *,
-        params: GetMemorysetParams,
+        params: GetMemorysetParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2046,7 +2108,7 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[list[LabeledMemory]] | list[list[ScoredMemory]]:
+    ) -> list[list[LabeledMemory]] | list[list[ScoredMemory]] | None:
         pass
 
     @overload
@@ -2324,7 +2386,7 @@ class OrcaClient(Client):
         self,
         path: Literal["/classification_model"],
         *,
-        params: None = None,
+        params: GetClassificationModelParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2340,7 +2402,7 @@ class OrcaClient(Client):
         self,
         path: Literal["/regression_model"],
         *,
-        params: None = None,
+        params: GetRegressionModelParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2388,7 +2450,7 @@ class OrcaClient(Client):
         self,
         path: Literal["/predictive_model"],
         *,
-        params: None = None,
+        params: GetPredictiveModelParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2500,7 +2562,7 @@ class OrcaClient(Client):
         self,
         path: Literal["/job"],
         *,
-        params: GetJobParams,
+        params: GetJobParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2516,7 +2578,7 @@ class OrcaClient(Client):
         self,
         path: Literal["/worker"],
         *,
-        params: GetWorkerParams,
+        params: GetWorkerParams | None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -3148,7 +3210,7 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> None:
+    ) -> DeleteMemoriesResponse:
         pass
 
     @overload
@@ -3865,7 +3927,7 @@ class OrcaClient(Client):
         path: Literal["/gpu/memoryset/{name_or_id}/memories"],
         *,
         params: PatchGpuMemorysetByNameOrIdMemoriesParams,
-        json: PatchGpuMemorysetByNameOrIdMemoriesRequest,
+        json: BatchMemoryUpdateRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3876,7 +3938,7 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[LabeledMemory] | list[ScoredMemory]:
+    ) -> UpdateMemoriesResponse:
         pass
 
     @overload