orca-sdk 0.0.102__py3-none-any.whl → 0.0.104__py3-none-any.whl

orca_sdk/client.py

@@ -9,7 +9,16 @@ import uuid
 from string import Formatter
 from typing import Any, Literal, Mapping, NotRequired, TypedDict, cast, overload
 
-from httpx import URL, USE_CLIENT_DEFAULT, Client, Headers, Request, Response, Timeout
+from httpx import (
+    URL,
+    USE_CLIENT_DEFAULT,
+    Client,
+    Headers,
+    HTTPTransport,
+    Request,
+    Response,
+    Timeout,
+)
 from httpx._client import UseClientDefault  # type: ignore
 from httpx._types import AuthTypes  # type: ignore
 from httpx._types import CookieTypes  # type: ignore
@@ -18,6 +27,7 @@ from httpx._types import HeaderTypes  # type: ignore
 from httpx._types import RequestContent  # type: ignore
 from httpx._types import RequestExtensions  # type: ignore
 from httpx._types import TimeoutTypes  # type: ignore
+from httpx_retries import Retry, RetryTransport
 
 
 class ActionRecommendation(TypedDict):
@@ -36,6 +46,10 @@ class AddMemorySuggestion(TypedDict):
     label_name: str
 
 
+class AliveResponse(TypedDict):
+    ok: bool
+
+
 class ApiKeyMetadata(TypedDict):
     id: str
     org_id: str
@@ -50,7 +64,7 @@ class BaseLabelPredictionResult(TypedDict):
     prediction_id: str | None
     confidence: float
     anomaly_score: float | None
-    label: int
+    label: int | None
     label_name: str | None
     logits: list[float]
 
@@ -63,11 +77,11 @@ class BaseScorePredictionResult(TypedDict):
     prediction_id: str | None
     confidence: float
     anomaly_score: float | None
-    score: float
+    score: float | None
 
 
 class CascadeEditSuggestionsRequest(TypedDict):
-    old_label: int
+    old_label: int | None
     new_label: int
     max_neighbors: NotRequired[int]
     max_validation_neighbors: NotRequired[int]
@@ -105,6 +119,13 @@ class ClassificationEvaluationRequest(TypedDict):
     telemetry_tags: NotRequired[list[str] | None]
 
 
+class CleanupResponse(TypedDict):
+    deleted_milvus_collections: list[str]
+    deleted_milvus_collections_count: int
+    deleted_blob_paths: list[str]
+    deleted_blob_paths_count: int
+
+
 class ClusterMetrics(TypedDict):
     cluster: int
     memory_count: int
@@ -201,6 +222,14 @@ class GetMemoriesRequest(TypedDict):
     memory_ids: list[str]
 
 
+class HealthyResponse(TypedDict):
+    ok: bool
+    checks: dict[str, bool]
+    durations: dict[str, int]
+    draining: bool
+    config: dict[str, str | float | int | bool | None]
+
+
 class InternalServerErrorResponse(TypedDict):
     status_code: NotRequired[int]
     message: str
@@ -208,7 +237,7 @@ class InternalServerErrorResponse(TypedDict):
 
 
 class LabelClassMetrics(TypedDict):
-    label: int
+    label: int | None
     label_name: NotRequired[str | None]
     average_lookup_score: float
     memory_count: int
@@ -224,7 +253,7 @@ class LabeledMemoryInsert(TypedDict):
     value: str | bytes
     metadata: NotRequired[dict[str, str | int | float | bool | None]]
     source_id: NotRequired[str | None]
-    label: int
+    label: int | None
 
 
 class ListMemoriesRequest(TypedDict):
@@ -255,7 +284,7 @@ class MemoryMetrics(TypedDict):
     embedding_2d: NotRequired[list]
     anomaly_score: NotRequired[float]
     neighbor_label_logits: NotRequired[list[float]]
-    neighbor_predicted_label: NotRequired[int]
+    neighbor_predicted_label: NotRequired[int | None]
     neighbor_predicted_label_ambiguity: NotRequired[float]
     neighbor_predicted_label_confidence: NotRequired[float]
     current_label_neighbor_confidence: NotRequired[float]
@@ -282,6 +311,7 @@ class MemorysetClassPatternsMetrics(TypedDict):
     variance_spread: float
     mean_uniformity: float
     variance_uniformity: float
+    updated_at: str
 
 
 class MemorysetClusterAnalysisConfig(TypedDict):
@@ -299,6 +329,7 @@ class MemorysetClusterMetrics(TypedDict):
     cluster_metrics: list[ClusterMetrics]
     num_outliers: int
     num_clusters: int
+    updated_at: str
 
 
 class MemorysetConceptAnalysisConfig(TypedDict):
@@ -321,6 +352,7 @@ class MemorysetDuplicateAnalysisConfig(TypedDict):
 class MemorysetDuplicateMetrics(TypedDict):
     num_duplicates: int
     num_potential_duplicates: int
+    updated_at: str
 
 
 class MemorysetLabelAnalysisConfig(TypedDict):
@@ -333,7 +365,9 @@ class MemorysetLabelMetrics(TypedDict):
     mean_neighbor_label_confidence: float
     mean_neighbor_label_entropy: float
     mean_neighbor_predicted_label_ambiguity: float
+    num_no_labeled_neighbors: int
     num_potential_mislabels: int
+    updated_at: str
 
 
 class MemorysetNeighborAnalysisConfig(TypedDict):
@@ -343,6 +377,7 @@ class MemorysetNeighborAnalysisConfig(TypedDict):
 
 class MemorysetNeighborMetrics(TypedDict):
     lookup_score_metrics: dict[str, LookupScoreMetrics]
+    updated_at: str
 
 
 class MemorysetProjectionAnalysisConfig(TypedDict):
@@ -351,7 +386,7 @@ class MemorysetProjectionAnalysisConfig(TypedDict):
 
 
 class MemorysetProjectionMetrics(TypedDict):
-    pass
+    updated_at: str
 
 
 class MemorysetUpdate(TypedDict):
@@ -359,6 +394,7 @@ class MemorysetUpdate(TypedDict):
     description: NotRequired[str | None]
     name: NotRequired[str]
     notes: NotRequired[str | None]
+    hidden: NotRequired[bool]
 
 
 class NotFoundErrorResponse(TypedDict):
@@ -477,6 +513,11 @@ class ROCCurve(TypedDict):
     true_positive_rates: list[float]
 
 
+class ReadyResponse(TypedDict):
+    ok: bool
+    draining: bool
+
+
 class RegressionEvaluationRequest(TypedDict):
     datasource_name_or_id: str
     memoryset_override_name_or_id: NotRequired[str | None]
@@ -487,6 +528,7 @@ class RegressionEvaluationRequest(TypedDict):
 
 
 class RegressionMetrics(TypedDict):
+    coverage: float
     mse: float
     rmse: float
     mae: float
@@ -539,7 +581,7 @@ class ScorePredictionMemoryLookup(TypedDict):
     edited_at: str
     metrics: MemoryMetrics
     lookup_score: float
-    score: float
+    score: float | None
     prediction_id: str
     attention_weight: float
 
@@ -548,7 +590,7 @@ class ScorePredictionWithMemoriesAndFeedback(TypedDict):
     prediction_id: str
     confidence: float
     anomaly_score: float | None
-    score: float
+    score: float | None
     timestamp: str
     input_value: str | bytes
     input_embedding: list[float]
@@ -575,7 +617,7 @@ class ScoredMemory(TypedDict):
     updated_at: str
     edited_at: str
     metrics: MemoryMetrics
-    score: float
+    score: float | None
 
 
 class ScoredMemoryInsert(TypedDict):
@@ -583,7 +625,7 @@ class ScoredMemoryInsert(TypedDict):
     value: str | bytes
     metadata: NotRequired[dict[str, str | int | float | bool | None]]
     source_id: NotRequired[str | None]
-    score: float
+    score: float | None
 
 
 class ScoredMemoryLookup(TypedDict):
@@ -598,7 +640,7 @@ class ScoredMemoryLookup(TypedDict):
     edited_at: str
     metrics: MemoryMetrics
     lookup_score: float
-    score: float
+    score: float | None
 
 
 class ScoredMemoryUpdate(TypedDict):
@@ -607,7 +649,7 @@ class ScoredMemoryUpdate(TypedDict):
     metadata: NotRequired[dict[str, str | int | float | bool | None] | None]
     source_id: NotRequired[str | None]
     metrics: NotRequired[MemoryMetrics | None]
-    score: NotRequired[float]
+    score: NotRequired[float | None]
 
 
 class ScoredMemoryWithFeedbackMetrics(TypedDict):
@@ -621,7 +663,7 @@ class ScoredMemoryWithFeedbackMetrics(TypedDict):
     updated_at: str
     edited_at: str
     metrics: MemoryMetrics
-    score: float
+    score: float | None
     feedback_metrics: dict[str, FeedbackMetrics]
     lookup_count: int
 
@@ -639,7 +681,7 @@ class SubConceptMetrics(TypedDict):
     memory_count: int
 
 
-TaskStatus = Literal["INITIALIZED", "DISPATCHED", "PROCESSING", "COMPLETED", "FAILED", "ABORTING", "ABORTED"]
+TaskStatus = Literal["INITIALIZED", "DISPATCHED", "WAITING", "PROCESSING", "COMPLETED", "FAILED", "ABORTING", "ABORTED"]
 
 
 class TaskStatusInfo(TypedDict):
@@ -697,6 +739,7 @@ class DeleteAuthApiKeyByNameOrIdParams(TypedDict):
 
 class GetMemorysetParams(TypedDict):
     type: NotRequired[MemoryType | None]
+    show_hidden: NotRequired[bool | None]
 
 
 class PostMemorysetByNameOrIdCloneParams(TypedDict):
@@ -1025,14 +1068,6 @@ class AddMemoryRecommendations(TypedDict):
     memories: list[AddMemorySuggestion]
 
 
-class AnalyzeNeighborLabelsResult(TypedDict):
-    label_metrics: list[LabelClassMetrics]
-    neighbor_prediction_accuracy: float
-    mean_neighbor_label_confidence: float
-    mean_neighbor_label_entropy: float
-    mean_neighbor_predicted_label_ambiguity: float
-
-
 class BootstrapClassificationModelRequest(TypedDict):
     model_description: str
     label_names: list[str]
@@ -1048,9 +1083,10 @@ class BootstrapClassificationModelResult(TypedDict):
 
 
 class ClassificationMetrics(TypedDict):
+    coverage: float
     f1_score: float
     accuracy: float
-    loss: float
+    loss: float | None
     anomaly_score_mean: NotRequired[float | None]
     anomaly_score_median: NotRequired[float | None]
     anomaly_score_variance: NotRequired[float | None]
@@ -1108,6 +1144,7 @@ class ColumnInfo(TypedDict):
     type: ColumnType
     enum_options: NotRequired[list[str] | None]
     int_values: NotRequired[list[int] | None]
+    contains_nones: NotRequired[bool]
 
 
 class ConceptMetrics(TypedDict):
@@ -1149,6 +1186,7 @@ class CreateMemorysetRequest(TypedDict):
     index_type: NotRequired[Literal["FLAT", "IVF_FLAT", "IVF_SQ8", "IVF_PQ", "HNSW", "DISKANN"]]
     index_params: NotRequired[dict[str, int | float | str]]
     prompt: NotRequired[str]
+    hidden: NotRequired[bool]
 
 
 class CreateRegressionModelRequest(TypedDict):
@@ -1212,7 +1250,7 @@ class EmbeddingEvaluationResponseUnionClassificationMetricsRegressionMetrics(Typ
 class EmbeddingModelResult(TypedDict):
     embedding_model_name: str
     embedding_model_path: str
-    analysis_result: AnalyzeNeighborLabelsResult
+    analysis_result: MemorysetLabelMetrics
     memoryset_name: NotRequired[str | None]
     is_finetuned: NotRequired[bool]
 
@@ -1291,7 +1329,7 @@ class LabelPredictionMemoryLookup(TypedDict):
     updated_at: str
     edited_at: str
     metrics: MemoryMetrics
-    label: int
+    label: int | None
     label_name: str | None
     lookup_score: float
     prediction_id: str
@@ -1302,7 +1340,7 @@ class LabelPredictionWithMemoriesAndFeedback(TypedDict):
     prediction_id: str
     confidence: float
     anomaly_score: float | None
-    label: int
+    label: int | None
     label_name: str | None
     logits: list[float]
     timestamp: str
@@ -1332,7 +1370,7 @@ class LabeledMemory(TypedDict):
     updated_at: str
     edited_at: str
     metrics: MemoryMetrics
-    label: int
+    label: int | None
     label_name: str | None
 
 
@@ -1347,7 +1385,7 @@ class LabeledMemoryLookup(TypedDict):
     updated_at: str
     edited_at: str
     metrics: MemoryMetrics
-    label: int
+    label: int | None
     label_name: str | None
     lookup_score: float
 
@@ -1358,7 +1396,7 @@ class LabeledMemoryUpdate(TypedDict):
     metadata: NotRequired[dict[str, str | int | float | bool | None] | None]
     source_id: NotRequired[str | None]
     metrics: NotRequired[MemoryMetrics | None]
-    label: NotRequired[int]
+    label: NotRequired[int | None]
 
 
 class LabeledMemoryWithFeedbackMetrics(TypedDict):
@@ -1372,7 +1410,7 @@ class LabeledMemoryWithFeedbackMetrics(TypedDict):
     updated_at: str
     edited_at: str
     metrics: MemoryMetrics
-    label: int
+    label: int | None
     label_name: str | None
     feedback_metrics: dict[str, FeedbackMetrics]
     lookup_count: int
@@ -1410,6 +1448,7 @@ class MemorysetAnalysisRequest(TypedDict):
 class MemorysetConceptMetrics(TypedDict):
     concepts: list[ConceptMetrics]
     num_outliers: int
+    updated_at: str
 
 
 class MemorysetMetrics(TypedDict):
@@ -1451,6 +1490,7 @@ class Task(TypedDict):
     type: str
     payload: BaseModel
     result: BaseModel | None
+    depends_on: NotRequired[list[str]]
 
 
 class TelemetryMemoriesRequest(TypedDict):
@@ -1514,6 +1554,7 @@ class MemorysetMetadata(TypedDict):
     database_uri: str | None
     document_prompt_override: str | None
     query_prompt_override: str | None
+    hidden: bool
 
 
 class PaginatedTask(TypedDict):
@@ -1563,9 +1604,9 @@ class OrcaClient(Client):
         return path_params, query_params
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/"],
+        path: Literal["/cleanup"],
         *,
         params: None = None,
         parse_as: Literal["json"] = "json",
@@ -1575,15 +1616,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Any:
+    ) -> CleanupResponse:
+        """Cleanup orphaned milvus collections and blobs"""
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/check/alive"],
+        path: Literal["/auth/api_key/{name_or_id}"],
         *,
-        params: None = None,
+        params: DeleteAuthApiKeyByNameOrIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1591,13 +1633,14 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Any:
+    ) -> None:
+        """Delete an API key by name or ID."""
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/check/ready"],
+        path: Literal["/auth/org"],
         *,
         params: None = None,
         parse_as: Literal["json"] = "json",
@@ -1607,15 +1650,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Any:
+    ) -> None:
+        """Deletes the org and all associated resources"""
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/auth/root"],
+        path: Literal["/memoryset/{name_or_id}"],
         *,
-        params: None = None,
+        params: DeleteMemorysetByNameOrIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1623,16 +1667,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> bool:
-        """Return true only when called with a valid root API key; otherwise 401 Unauthenticated."""
+    ) -> None:
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/auth/api_key"],
+        path: Literal["/memoryset/{name_or_id}/memory/{memory_id}"],
         *,
-        params: None = None,
+        params: DeleteMemorysetByNameOrIdMemoryByMemoryIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1640,16 +1683,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[ApiKeyMetadata]:
-        """List all API keys for the organization."""
+    ) -> None:
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/auth"],
+        path: Literal["/finetuned_embedding_model/{name_or_id}"],
         *,
-        params: None = None,
+        params: DeleteFinetunedEmbeddingModelByNameOrIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1657,16 +1699,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> bool:
-        """Returns true if the api key header is valid for the org (will be false for admin api key)"""
+    ) -> None:
+        """Delete a finetuned embedding model by name or ID."""
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/auth/org/plan"],
+        path: Literal["/datasource/{name_or_id}"],
         *,
-        params: None = None,
+        params: DeleteDatasourceByNameOrIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1674,16 +1716,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> OrgPlan:
-        """Get the organization plan."""
+    ) -> None:
+        """Delete a datasource by name or ID."""
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/memoryset"],
+        path: Literal["/classification_model/{name_or_id}"],
         *,
-        params: GetMemorysetParams,
+        params: DeleteClassificationModelByNameOrIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1691,15 +1733,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[MemorysetMetadata]:
+    ) -> None:
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/memoryset/{name_or_id}"],
+        path: Literal["/classification_model/{model_name_or_id}/evaluation/{task_id}"],
         *,
-        params: GetMemorysetByNameOrIdParams,
+        params: DeleteClassificationModelByModelNameOrIdEvaluationByTaskIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1707,15 +1749,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> MemorysetMetadata:
+    ) -> None:
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/memoryset/{name_or_id}/memory/{memory_id}"],
+        path: Literal["/regression_model/{name_or_id}"],
         *,
-        params: GetMemorysetByNameOrIdMemoryByMemoryIdParams,
+        params: DeleteRegressionModelByNameOrIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1723,15 +1765,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> LabeledMemory | ScoredMemory:
+    ) -> None:
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/memoryset/{name_or_id}/potential_duplicate_groups"],
+        path: Literal["/regression_model/{model_name_or_id}/evaluation/{task_id}"],
         *,
-        params: GetMemorysetByNameOrIdPotentialDuplicateGroupsParams,
+        params: DeleteRegressionModelByModelNameOrIdEvaluationByTaskIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1739,15 +1781,15 @@ 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]]:
+    ) -> None:
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/memoryset/{name_or_id}/analysis"],
+        path: Literal["/task/{task_id}/abort"],
         *,
-        params: GetMemorysetByNameOrIdAnalysisParams,
+        params: DeleteTaskByTaskIdAbortParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1755,15 +1797,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[MemorysetAnalysisResponse]:
+    ) -> None:
         pass
 
     @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/memoryset/{name_or_id}/analysis/{analysis_task_id}"],
+        path: Literal["/telemetry/feedback_category/{name_or_id}"],
         *,
-        params: GetMemorysetByNameOrIdAnalysisByAnalysisTaskIdParams,
+        params: DeleteTelemetryFeedbackCategoryByNameOrIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1771,32 +1813,46 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> MemorysetAnalysisResponse:
+    ) -> None:
+        """Delete a feedback category and all associated feedback records."""
         pass
 
-    @overload
-    def GET(
+    def DELETE(
         self,
-        path: Literal["/finetuned_embedding_model"],
+        path: str,
         *,
-        params: None = None,
-        parse_as: Literal["json"] = "json",
+        params: Mapping[str, Any] | None = None,
+        parse_as: Literal["json", "text"] | None = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
         auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[FinetunedEmbeddingModelMetadata]:
-        """List all finetuned embedding models for the organization."""
-        pass
+    ) -> Any:
+        path_params, query_params = self._parse_params(params or {}, path)
+        res = self.delete(
+            path.format(**path_params),
+            params=query_params,
+            headers=headers,
+            cookies=cookies,
+            auth=auth,
+            follow_redirects=follow_redirects,
+            timeout=timeout,
+            extensions=extensions,
+        ).raise_for_status()
+        return (
+            None
+            if res.status_code == 204
+            else res.json() if parse_as == "json" else res.text if parse_as == "text" else res.content
+        )
 
     @overload
     def GET(
         self,
-        path: Literal["/finetuned_embedding_model/{name_or_id}"],
+        path: Literal["/check/alive"],
         *,
-        params: GetFinetunedEmbeddingModelByNameOrIdParams,
+        params: None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1804,16 +1860,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> FinetunedEmbeddingModelMetadata:
-        """Get a finetuned embedding model by name or ID."""
+    ) -> AliveResponse:
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/finetuned_embedding_model/{name_or_id}/evaluation/{task_id}"],
+        path: Literal["/check/ready"],
         *,
-        params: GetFinetunedEmbeddingModelByNameOrIdEvaluationByTaskIdParams,
+        params: None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1821,16 +1876,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> EmbeddingEvaluationResponseUnionClassificationMetricsRegressionMetrics:
-        """Get evaluation results for a finetuned embedding model by task ID."""
+    ) -> ReadyResponse:
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/finetuned_embedding_model/{name_or_id}/evaluations"],
+        path: Literal["/gpu/check/healthy"],
         *,
-        params: GetFinetunedEmbeddingModelByNameOrIdEvaluationsParams,
+        params: None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1838,14 +1892,13 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[EmbeddingEvaluationResponseUnionClassificationMetricsRegressionMetrics]:
-        """List all evaluation results for a finetuned embedding model."""
+    ) -> HealthyResponse:
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/pretrained_embedding_model"],
+        path: Literal["/check/healthy"],
         *,
         params: None = None,
         parse_as: Literal["json"] = "json",
@@ -1855,16 +1908,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[PretrainedEmbeddingModelMetadata]:
-        """List all available pretrained embedding models."""
+    ) -> HealthyResponse:
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/pretrained_embedding_model/{model_name}"],
+        path: Literal["/gpu/config"],
         *,
-        params: GetPretrainedEmbeddingModelByModelNameParams,
+        params: None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1872,16 +1924,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> PretrainedEmbeddingModelMetadata:
-        """Get metadata for a specific pretrained embedding model."""
+    ) -> dict[str, str | float | int | bool | None]:
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/pretrained_embedding_model/{model_name}/evaluation/{task_id}"],
+        path: Literal["/config"],
         *,
-        params: GetPretrainedEmbeddingModelByModelNameEvaluationByTaskIdParams,
+        params: None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1889,50 +1940,47 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> EmbeddingEvaluationResponseUnionClassificationMetricsRegressionMetrics:
-        """Get evaluation results for a pretrained embedding model by task ID."""
+    ) -> dict[str, str | float | int | bool | None]:
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/pretrained_embedding_model/{model_name}/evaluations"],
+        path: Literal["/gpu/"],
         *,
-        params: GetPretrainedEmbeddingModelByModelNameEvaluationsParams,
-        parse_as: Literal["json"] = "json",
+        params: None = None,
+        parse_as: Literal["text"],
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
         auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[EmbeddingEvaluationResponseUnionClassificationMetricsRegressionMetrics]:
-        """List all evaluation results for a pretrained embedding model."""
+    ) -> str:
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/datasource"],
+        path: Literal["/"],
         *,
         params: None = None,
-        parse_as: Literal["json"] = "json",
+        parse_as: Literal["text"],
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
         auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[DatasourceMetadata]:
-        """List all datasources for the organization."""
+    ) -> str:
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/datasource/{name_or_id}"],
+        path: Literal["/auth/root"],
         *,
-        params: GetDatasourceByNameOrIdParams,
+        params: None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1940,16 +1988,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> DatasourceMetadata:
-        """Get a datasource by name or ID."""
+    ) -> bool:
+        """Return true only when called with a valid root API key; otherwise 401 Unauthenticated."""
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/datasource/{name_or_id}/embedding_evaluation"],
+        path: Literal["/auth/api_key"],
         *,
-        params: GetDatasourceByNameOrIdEmbeddingEvaluationParams,
+        params: None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1957,16 +2005,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[DatasourceEmbeddingEvaluationsResponse]:
-        """List embedding evaluation tasks for a datasource."""
+    ) -> list[ApiKeyMetadata]:
+        """List all API keys for the organization."""
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/datasource/{name_or_id}/embedding_evaluation/{task_id}"],
+        path: Literal["/auth"],
         *,
-        params: GetDatasourceByNameOrIdEmbeddingEvaluationByTaskIdParams,
+        params: None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1974,16 +2022,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> DatasourceEmbeddingEvaluationsResponse:
-        """Get an embedding evaluation task by ID."""
+    ) -> bool:
+        """Returns true if the api key header is valid for the org (will be false for admin api key)"""
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/datasource/{name_or_id}/download"],
+        path: Literal["/auth/org/plan"],
         *,
-        params: GetDatasourceByNameOrIdDownloadParams,
+        params: None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1991,25 +2039,342 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[dict[str, Any]]:
-        """Download datasource in the specified format."""
+    ) -> OrgPlan:
+        """Get the organization plan."""
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/datasource/{name_or_id}/download"],
+        path: Literal["/memoryset"],
         *,
-        params: GetDatasourceByNameOrIdDownloadParams,
-        parse_as: Literal["text"],
+        params: GetMemorysetParams,
+        parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
         auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> str:
-        """Download datasource in the specified format."""
+    ) -> list[MemorysetMetadata]:
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/memoryset/{name_or_id}"],
+        *,
+        params: GetMemorysetByNameOrIdParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> MemorysetMetadata:
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/memoryset/{name_or_id}/memory/{memory_id}"],
+        *,
+        params: GetMemorysetByNameOrIdMemoryByMemoryIdParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> LabeledMemory | ScoredMemory:
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/memoryset/{name_or_id}/potential_duplicate_groups"],
+        *,
+        params: GetMemorysetByNameOrIdPotentialDuplicateGroupsParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> list[list[LabeledMemory]] | list[list[ScoredMemory]]:
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/memoryset/{name_or_id}/analysis"],
+        *,
+        params: GetMemorysetByNameOrIdAnalysisParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> list[MemorysetAnalysisResponse]:
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/memoryset/{name_or_id}/analysis/{analysis_task_id}"],
+        *,
+        params: GetMemorysetByNameOrIdAnalysisByAnalysisTaskIdParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> MemorysetAnalysisResponse:
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/finetuned_embedding_model"],
+        *,
+        params: None = None,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> list[FinetunedEmbeddingModelMetadata]:
+        """List all finetuned embedding models for the organization."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/finetuned_embedding_model/{name_or_id}"],
+        *,
+        params: GetFinetunedEmbeddingModelByNameOrIdParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> FinetunedEmbeddingModelMetadata:
+        """Get a finetuned embedding model by name or ID."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/finetuned_embedding_model/{name_or_id}/evaluation/{task_id}"],
+        *,
+        params: GetFinetunedEmbeddingModelByNameOrIdEvaluationByTaskIdParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> EmbeddingEvaluationResponseUnionClassificationMetricsRegressionMetrics:
+        """Get evaluation results for a finetuned embedding model by task ID."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/finetuned_embedding_model/{name_or_id}/evaluations"],
+        *,
+        params: GetFinetunedEmbeddingModelByNameOrIdEvaluationsParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> list[EmbeddingEvaluationResponseUnionClassificationMetricsRegressionMetrics]:
+        """List all evaluation results for a finetuned embedding model."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/pretrained_embedding_model"],
+        *,
+        params: None = None,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> list[PretrainedEmbeddingModelMetadata]:
+        """List all available pretrained embedding models."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/pretrained_embedding_model/{model_name}"],
+        *,
+        params: GetPretrainedEmbeddingModelByModelNameParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> PretrainedEmbeddingModelMetadata:
+        """Get metadata for a specific pretrained embedding model."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/pretrained_embedding_model/{model_name}/evaluation/{task_id}"],
+        *,
+        params: GetPretrainedEmbeddingModelByModelNameEvaluationByTaskIdParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> EmbeddingEvaluationResponseUnionClassificationMetricsRegressionMetrics:
+        """Get evaluation results for a pretrained embedding model by task ID."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/pretrained_embedding_model/{model_name}/evaluations"],
+        *,
+        params: GetPretrainedEmbeddingModelByModelNameEvaluationsParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> list[EmbeddingEvaluationResponseUnionClassificationMetricsRegressionMetrics]:
+        """List all evaluation results for a pretrained embedding model."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/datasource"],
+        *,
+        params: None = None,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> list[DatasourceMetadata]:
+        """List all datasources for the organization."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/datasource/{name_or_id}"],
+        *,
+        params: GetDatasourceByNameOrIdParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> DatasourceMetadata:
+        """Get a datasource by name or ID."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/datasource/{name_or_id}/embedding_evaluation"],
+        *,
+        params: GetDatasourceByNameOrIdEmbeddingEvaluationParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> list[DatasourceEmbeddingEvaluationsResponse]:
+        """List embedding evaluation tasks for a datasource."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/datasource/{name_or_id}/embedding_evaluation/{task_id}"],
+        *,
+        params: GetDatasourceByNameOrIdEmbeddingEvaluationByTaskIdParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> DatasourceEmbeddingEvaluationsResponse:
+        """Get an embedding evaluation task by ID."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/datasource/{name_or_id}/download"],
+        *,
+        params: GetDatasourceByNameOrIdDownloadParams,
+        parse_as: Literal["json"] = "json",
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> list[dict[str, Any]]:
+        """Download datasource in the specified format."""
+        pass
+
+    @overload
+    def GET(
+        self,
+        path: Literal["/datasource/{name_or_id}/download"],
+        *,
+        params: GetDatasourceByNameOrIdDownloadParams,
+        parse_as: Literal["text"],
+        headers: HeaderTypes | None = None,
+        cookies: CookieTypes | None = None,
+        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
+        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
+        extensions: RequestExtensions | None = None,
+    ) -> str:
+        """Download datasource in the specified format."""
         pass
 
     @overload
@@ -2275,65 +2640,9 @@ class OrcaClient(Client):
     @overload
     def GET(
         self,
-        path: Literal["/telemetry/prediction/{prediction_id}/memory_suggestions"],
-        *,
-        params: GetTelemetryPredictionByPredictionIdMemorySuggestionsParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> AddMemoryRecommendations:
-        """
-        Generate synthetic memory suggestions to improve a specific prediction.
-
-        The returned suggestions have labels as string representations of integer indices
-        corresponding to the memoryset's label_names.
-        """
-        pass
-
-    @overload
-    def GET(
-        self,
-        path: Literal["/telemetry/feedback_category"],
-        *,
-        params: None = None,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> list[PredictionFeedbackCategory]:
-        """List all feedback categories for the organization."""
-        pass
-
-    @overload
-    def GET(
-        self,
-        path: Literal["/telemetry/feedback_category/{name_or_id}"],
-        *,
-        params: GetTelemetryFeedbackCategoryByNameOrIdParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> PredictionFeedbackCategory:
-        """Get a feedback category by name or ID."""
-        pass
-
-    @overload
-    def GET(
-        self,
-        path: Literal["/agents/bootstrap_classification_model/{task_id}"],
+        path: Literal["/telemetry/prediction/{prediction_id}/memory_suggestions"],
         *,
-        params: GetAgentsBootstrapClassificationModelByTaskIdParams,
+        params: GetTelemetryPredictionByPredictionIdMemorySuggestionsParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2341,14 +2650,19 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> BootstrapClassificationModelResponse:
-        """Get the status of a bootstrap classification model task"""
+    ) -> AddMemoryRecommendations:
+        """
+        Generate synthetic memory suggestions to improve a specific prediction.
+
+        The returned suggestions have labels as string representations of integer indices
+        corresponding to the memoryset's label_names.
+        """
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/gpu/"],
+        path: Literal["/telemetry/feedback_category"],
         *,
         params: None = None,
         parse_as: Literal["json"] = "json",
@@ -2358,15 +2672,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Any:
+    ) -> list[PredictionFeedbackCategory]:
+        """List all feedback categories for the organization."""
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/gpu/check/alive"],
+        path: Literal["/telemetry/feedback_category/{name_or_id}"],
         *,
-        params: None = None,
+        params: GetTelemetryFeedbackCategoryByNameOrIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2374,15 +2689,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Any:
+    ) -> PredictionFeedbackCategory:
+        """Get a feedback category by name or ID."""
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/gpu/check/ready"],
+        path: Literal["/agents/bootstrap_classification_model/{task_id}"],
         *,
-        params: None = None,
+        params: GetAgentsBootstrapClassificationModelByTaskIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2390,7 +2706,8 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Any:
+    ) -> BootstrapClassificationModelResponse:
+        """Get the status of a bootstrap classification model task"""
         pass
 
     def GET(
@@ -3104,233 +3421,6 @@ class OrcaClient(Client):
             else res.json() if parse_as == "json" else res.text if parse_as == "text" else res.content
         )
 
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/auth/api_key/{name_or_id}"],
-        *,
-        params: DeleteAuthApiKeyByNameOrIdParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        """Delete an API key by name or ID."""
-        pass
-
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/auth/org"],
-        *,
-        params: None = None,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        """Deletes the org and all associated resources"""
-        pass
-
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/memoryset/{name_or_id}"],
-        *,
-        params: DeleteMemorysetByNameOrIdParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        pass
-
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/memoryset/{name_or_id}/memory/{memory_id}"],
-        *,
-        params: DeleteMemorysetByNameOrIdMemoryByMemoryIdParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        pass
-
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/finetuned_embedding_model/{name_or_id}"],
-        *,
-        params: DeleteFinetunedEmbeddingModelByNameOrIdParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        """Delete a finetuned embedding model by name or ID."""
-        pass
-
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/datasource/{name_or_id}"],
-        *,
-        params: DeleteDatasourceByNameOrIdParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        """Delete a datasource by name or ID."""
-        pass
-
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/classification_model/{name_or_id}"],
-        *,
-        params: DeleteClassificationModelByNameOrIdParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        pass
-
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/classification_model/{model_name_or_id}/evaluation/{task_id}"],
-        *,
-        params: DeleteClassificationModelByModelNameOrIdEvaluationByTaskIdParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        pass
-
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/regression_model/{name_or_id}"],
-        *,
-        params: DeleteRegressionModelByNameOrIdParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        pass
-
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/regression_model/{model_name_or_id}/evaluation/{task_id}"],
-        *,
-        params: DeleteRegressionModelByModelNameOrIdEvaluationByTaskIdParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        pass
-
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/task/{task_id}/abort"],
-        *,
-        params: DeleteTaskByTaskIdAbortParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        pass
-
-    @overload
-    def DELETE(
-        self,
-        path: Literal["/telemetry/feedback_category/{name_or_id}"],
-        *,
-        params: DeleteTelemetryFeedbackCategoryByNameOrIdParams,
-        parse_as: Literal["json"] = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> None:
-        """Delete a feedback category and all associated feedback records."""
-        pass
-
-    def DELETE(
-        self,
-        path: str,
-        *,
-        params: Mapping[str, Any] | None = None,
-        parse_as: Literal["json", "text"] | None = "json",
-        headers: HeaderTypes | None = None,
-        cookies: CookieTypes | None = None,
-        auth: AuthTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
-        timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
-        extensions: RequestExtensions | None = None,
-    ) -> Any:
-        path_params, query_params = self._parse_params(params or {}, path)
-        res = self.delete(
-            path.format(**path_params),
-            params=query_params,
-            headers=headers,
-            cookies=cookies,
-            auth=auth,
-            follow_redirects=follow_redirects,
-            timeout=timeout,
-            extensions=extensions,
-        ).raise_for_status()
-        return (
-            None
-            if res.status_code == 204
-            else res.json() if parse_as == "json" else res.text if parse_as == "text" else res.content
-        )
-
     @overload
     def PUT(
         self,
@@ -3641,6 +3731,15 @@ logging.getLogger("httpx").setLevel(logging.WARNING)
 logging.getLogger("httpcore").setLevel(logging.ERROR)
 
 orca_api = OrcaClient(
+    transport=RetryTransport(
+        transport=HTTPTransport(),
+        retry=Retry(
+            total=5,
+            backoff_factor=0.5,
+            allowed_methods=["GET", "POST", "PUT", "PATCH", "DELETE"],
+            status_forcelist=[429, 500, 502, 503, 504],
+        ),
+    ),
     event_hooks={"request": [_instrument_request], "response": [_raise_error_for_response]},
     follow_redirects=True,
     timeout=Timeout(connect=3, read=20, write=10, pool=5),