orca-sdk 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

orca_sdk/async_client.py

@@ -135,6 +135,10 @@ class ClassificationEvaluationRequest(TypedDict):
     datasource_value_column: str
     record_telemetry: NotRequired[bool]
     telemetry_tags: NotRequired[list[str] | None]
+    subsample: NotRequired[int | float | None]
+    ignore_unlabeled: NotRequired[bool]
+    datasource_partition_column: NotRequired[str | None]
+    partition_filter_mode: NotRequired[Literal["ignore_partitions", "include_global", "exclude_global", "only_global"]]
 
 
 class CleanupResponse(TypedDict):
@@ -163,6 +167,7 @@ class CountPredictionsRequest(TypedDict):
     prediction_ids: NotRequired[list[str] | None]
     start_timestamp: NotRequired[str | None]
     end_timestamp: NotRequired[str | None]
+    memory_id: NotRequired[str | None]
 
 
 class CreateApiKeyRequest(TypedDict):
@@ -193,6 +198,12 @@ class CreateOrgPlanRequest(TypedDict):
     tier: Literal["FREE", "PRO", "ENTERPRISE", "CANCELLED"]
 
 
+class DatasetFilterItem(TypedDict):
+    field: str
+    op: Literal["==", "!=", ">", ">=", "<", "<=", "in", "not in", "like"]
+    value: Any
+
+
 class DeleteMemoriesRequest(TypedDict):
     memory_ids: list[str]
 
@@ -210,7 +221,7 @@ class EmbedRequest(TypedDict):
 class EmbeddingEvaluationRequest(TypedDict):
     datasource_name_or_id: str
     eval_datasource_name_or_id: NotRequired[str | None]
-    subsample: NotRequired[int | None]
+    subsample: NotRequired[int | float | None]
     datasource_value_column: NotRequired[str]
     datasource_label_column: NotRequired[str | None]
     datasource_score_column: NotRequired[str | None]
@@ -219,7 +230,7 @@ class EmbeddingEvaluationRequest(TypedDict):
     weigh_memories: NotRequired[bool]
 
 
-EmbeddingFinetuningMethod = Literal["classification", "batch_triplet_loss"]
+EmbeddingFinetuningMethod = Literal["classification", "regression", "batch_triplet_loss"]
 
 
 class FeedbackMetrics(TypedDict):
@@ -233,7 +244,19 @@ FeedbackType = Literal["CONTINUOUS", "BINARY"]
 class FilterItem(TypedDict):
     field: list
     op: Literal["==", "!=", ">", ">=", "<", "<=", "in", "not in", "like"]
-    value: str | int | float | bool | list[str] | list[int] | list[float] | list[bool] | None
+    value: str | int | float | bool | list[str | None] | list[int] | list[float] | list[bool] | None
+
+
+class GetDatasourceRowCountRequest(TypedDict):
+    filters: NotRequired[list[DatasetFilterItem]]
+
+
+class GetDatasourceRowsRequest(TypedDict):
+    filters: NotRequired[list[DatasetFilterItem]]
+    limit: NotRequired[int]
+    offset: NotRequired[int]
+    shuffle: NotRequired[bool]
+    shuffle_seed: NotRequired[int | None]
 
 
 class GetMemoriesRequest(TypedDict):
@@ -254,6 +277,18 @@ class InternalServerErrorResponse(TypedDict):
     request_id: str
 
 
+JobStatus = Literal["INITIALIZED", "DISPATCHED", "WAITING", "PROCESSING", "COMPLETED", "FAILED", "ABORTING", "ABORTED"]
+
+
+class JobStatusInfo(TypedDict):
+    status: JobStatus
+    steps_total: int | None
+    steps_completed: int | None
+    exception: str | None
+    updated_at: str
+    created_at: str
+
+
 class LabelClassMetrics(TypedDict):
     label: int | None
     label_name: NotRequired[str | None]
@@ -276,6 +311,7 @@ class LabeledMemoryInsert(TypedDict):
     value: str | bytes
     metadata: NotRequired[dict[str, str | int | float | bool | None]]
     source_id: NotRequired[str | None]
+    partition_id: NotRequired[str | None]
     label: int | None
 
 
@@ -283,12 +319,16 @@ 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):
     query: list[str]
     count: NotRequired[int]
     prompt: NotRequired[str | None]
+    partition_id: NotRequired[str | list[str | None] | None]
+    partition_filter_mode: NotRequired[Literal["ignore_partitions", "include_global", "exclude_global", "only_global"]]
 
 
 class LookupScoreMetrics(TypedDict):
@@ -340,8 +380,6 @@ class MemorysetClassPatternsMetrics(TypedDict):
 class MemorysetClusterAnalysisConfig(TypedDict):
     min_cluster_size: NotRequired[int | None]
     max_cluster_size: NotRequired[int | None]
-    clustering_method: NotRequired[Literal["density", "graph"]]
-    min_cluster_distance: NotRequired[float]
     partitioning_method: NotRequired[Literal["ng", "rb", "cpm"]]
     resolution: NotRequired[float | None]
     num_iterations: NotRequired[int]
@@ -370,6 +408,7 @@ class MemorysetConceptAnalysisConfig(TypedDict):
     use_generative_naming: NotRequired[bool]
     naming_examples_count: NotRequired[int]
     naming_counterexample_count: NotRequired[int]
+    primary_label_pct_threshold: NotRequired[float]
     seed: NotRequired[int]
 
 
@@ -439,7 +478,7 @@ class NotFoundErrorResponse(TypedDict):
             "memory",
             "evaluation",
             "analysis",
-            "task",
+            "job",
             "pretrained_embedding_model",
             "finetuned_embedding_model",
             "feedback_category",
@@ -553,6 +592,10 @@ class RegressionEvaluationRequest(TypedDict):
     datasource_value_column: str
     record_telemetry: NotRequired[bool]
     telemetry_tags: NotRequired[list[str] | None]
+    subsample: NotRequired[int | float | None]
+    ignore_unlabeled: NotRequired[bool]
+    datasource_partition_column: NotRequired[str | None]
+    partition_filter_mode: NotRequired[Literal["ignore_partitions", "include_global", "exclude_global", "only_global"]]
 
 
 class RegressionMetrics(TypedDict):
@@ -595,12 +638,16 @@ class RegressionPredictionRequest(TypedDict):
     prompt: NotRequired[str | None]
     use_lookup_cache: NotRequired[bool]
     consistency_level: NotRequired[Literal["Bounded", "Session", "Strong", "Eventual"] | None]
+    ignore_unlabeled: NotRequired[bool]
+    partition_ids: NotRequired[str | list[str | None] | None]
+    partition_filter_mode: NotRequired[Literal["ignore_partitions", "include_global", "exclude_global", "only_global"]]
 
 
 class ScorePredictionMemoryLookup(TypedDict):
     value: str | bytes
     embedding: list[float]
     source_id: str | None
+    partition_id: str | None
     metadata: dict[str, str | int | float | bool | None]
     memory_id: str
     memory_version: int
@@ -638,6 +685,7 @@ class ScoredMemory(TypedDict):
     value: str | bytes
     embedding: list[float]
     source_id: str | None
+    partition_id: str | None
     metadata: dict[str, str | int | float | bool | None]
     memory_id: str
     memory_version: int
@@ -653,6 +701,7 @@ class ScoredMemoryInsert(TypedDict):
     value: str | bytes
     metadata: NotRequired[dict[str, str | int | float | bool | None]]
     source_id: NotRequired[str | None]
+    partition_id: NotRequired[str | None]
     score: float | None
 
 
@@ -660,6 +709,7 @@ class ScoredMemoryLookup(TypedDict):
     value: str | bytes
     embedding: list[float]
     source_id: str | None
+    partition_id: str | None
     metadata: dict[str, str | int | float | bool | None]
     memory_id: str
     memory_version: int
@@ -676,6 +726,7 @@ class ScoredMemoryUpdate(TypedDict):
     value: NotRequired[str | bytes]
     metadata: NotRequired[dict[str, str | int | float | bool | None] | None]
     source_id: NotRequired[str | None]
+    partition_id: NotRequired[str | None]
     metrics: NotRequired[MemoryMetrics | None]
     score: NotRequired[float | None]
 
@@ -684,6 +735,7 @@ class ScoredMemoryWithFeedbackMetrics(TypedDict):
     value: str | bytes
     embedding: list[float]
     source_id: str | None
+    partition_id: str | None
     metadata: dict[str, str | int | float | bool | None]
     memory_id: str
     memory_version: int
@@ -709,18 +761,6 @@ class SubConceptMetrics(TypedDict):
     memory_count: int
 
 
-TaskStatus = Literal["INITIALIZED", "DISPATCHED", "WAITING", "PROCESSING", "COMPLETED", "FAILED", "ABORTING", "ABORTED"]
-
-
-class TaskStatusInfo(TypedDict):
-    status: TaskStatus
-    steps_total: int | None
-    steps_completed: int | None
-    exception: str | None
-    updated_at: str
-    created_at: str
-
-
 TelemetryField = list
 
 
@@ -793,6 +833,10 @@ class DeleteMemorysetByNameOrIdParams(TypedDict):
     name_or_id: str
 
 
+class PostGpuMemorysetByNameOrIdLookupParams(TypedDict):
+    name_or_id: str
+
+
 class GetMemorysetByNameOrIdMemoryByMemoryIdParams(TypedDict):
     name_or_id: str
     memory_id: str
@@ -825,20 +869,35 @@ class PostMemorysetByNameOrIdMemoriesDeleteParams(TypedDict):
     name_or_id: str
 
 
+class PatchGpuMemorysetByNameOrIdMemoryParams(TypedDict):
+    name_or_id: str
+
+
+class PostGpuMemorysetByNameOrIdMemoryParams(TypedDict):
+    name_or_id: str
+
+
+PostGpuMemorysetByNameOrIdMemoryRequest = list[LabeledMemoryInsert] | list[ScoredMemoryInsert]
+
+
+class PatchGpuMemorysetByNameOrIdMemoriesParams(TypedDict):
+    name_or_id: str
+
+
 class PostMemorysetByNameOrIdAnalysisParams(TypedDict):
     name_or_id: str
 
 
 class GetMemorysetByNameOrIdAnalysisParams(TypedDict):
     name_or_id: str
-    status: NotRequired[TaskStatus | None]
+    status: NotRequired[JobStatus | None]
     limit: NotRequired[int | None]
     offset: NotRequired[int | None]
 
 
-class GetMemorysetByNameOrIdAnalysisByAnalysisTaskIdParams(TypedDict):
+class GetMemorysetByNameOrIdAnalysisByAnalysisJobIdParams(TypedDict):
     name_or_id: str
-    analysis_task_id: str
+    analysis_job_id: str
 
 
 class PostMemorysetByNameOrIdMemoryByMemoryIdCascadingEditsParams(TypedDict):
@@ -854,34 +913,42 @@ class DeleteFinetunedEmbeddingModelByNameOrIdParams(TypedDict):
     name_or_id: str
 
 
-class PostFinetunedEmbeddingModelByNameOrIdEvaluationParams(TypedDict):
+class PostGpuFinetunedEmbeddingModelByNameOrIdEmbeddingParams(TypedDict):
     name_or_id: str
 
 
-class GetFinetunedEmbeddingModelByNameOrIdEvaluationByTaskIdParams(TypedDict):
-    name_or_id: str
-    task_id: str
+class GetPretrainedEmbeddingModelByModelNameParams(TypedDict):
+    model_name: PretrainedEmbeddingModelName
 
 
-class GetFinetunedEmbeddingModelByNameOrIdEvaluationsParams(TypedDict):
-    name_or_id: str
-    datasource: NotRequired[str | None]
-    value_column: NotRequired[str | None]
-    label_column: NotRequired[str | None]
-    score_column: NotRequired[str | None]
+class PostGpuPretrainedEmbeddingModelByModelNameEmbeddingParams(TypedDict):
+    model_name: PretrainedEmbeddingModelName
 
 
-class GetPretrainedEmbeddingModelByModelNameParams(TypedDict):
-    model_name: PretrainedEmbeddingModelName
+class PostFinetunedEmbeddingModelByNameOrIdEvaluationParams(TypedDict):
+    name_or_id: str
 
 
 class PostPretrainedEmbeddingModelByModelNameEvaluationParams(TypedDict):
     model_name: PretrainedEmbeddingModelName
 
 
-class GetPretrainedEmbeddingModelByModelNameEvaluationByTaskIdParams(TypedDict):
+class GetFinetunedEmbeddingModelByNameOrIdEvaluationByJobIdParams(TypedDict):
+    name_or_id: str
+    job_id: str
+
+
+class GetPretrainedEmbeddingModelByModelNameEvaluationByJobIdParams(TypedDict):
     model_name: PretrainedEmbeddingModelName
-    task_id: str
+    job_id: str
+
+
+class GetFinetunedEmbeddingModelByNameOrIdEvaluationsParams(TypedDict):
+    name_or_id: str
+    datasource: NotRequired[str | None]
+    value_column: NotRequired[str | None]
+    label_column: NotRequired[str | None]
+    score_column: NotRequired[str | None]
 
 
 class GetPretrainedEmbeddingModelByModelNameEvaluationsParams(TypedDict):
@@ -911,6 +978,14 @@ class DeleteDatasourceByNameOrIdParams(TypedDict):
     name_or_id: str
 
 
+class PostDatasourceByNameOrIdRowsParams(TypedDict):
+    name_or_id: str
+
+
+class PostDatasourceByNameOrIdRowsCountParams(TypedDict):
+    name_or_id: str
+
+
 class GetDatasourceByNameOrIdEmbeddingModelEvaluationsParams(TypedDict):
     name_or_id: str
     value_column: NotRequired[str | None]
@@ -941,36 +1016,42 @@ class DeleteClassificationModelByNameOrIdParams(TypedDict):
     name_or_id: str
 
 
-class PostClassificationModelByModelNameOrIdEvaluationParams(TypedDict):
-    model_name_or_id: str
+class PatchRegressionModelByNameOrIdParams(TypedDict):
+    name_or_id: str
 
 
-class GetClassificationModelByModelNameOrIdEvaluationParams(TypedDict):
-    model_name_or_id: str
+class GetRegressionModelByNameOrIdParams(TypedDict):
+    name_or_id: str
 
 
-class GetClassificationModelByModelNameOrIdEvaluationByTaskIdParams(TypedDict):
-    model_name_or_id: str
-    task_id: str
+class DeleteRegressionModelByNameOrIdParams(TypedDict):
+    name_or_id: str
 
 
-class DeleteClassificationModelByModelNameOrIdEvaluationByTaskIdParams(TypedDict):
-    model_name_or_id: str
-    task_id: str
+class PostGpuClassificationModelByNameOrIdPredictionParams(TypedDict):
+    name_or_id: str
 
 
-class PatchRegressionModelByNameOrIdParams(TypedDict):
+class PostClassificationModelByNameOrIdPredictionParams(TypedDict):
     name_or_id: str
 
 
-class GetRegressionModelByNameOrIdParams(TypedDict):
+class PostGpuRegressionModelByNameOrIdPredictionParams(TypedDict):
     name_or_id: str
 
 
-class DeleteRegressionModelByNameOrIdParams(TypedDict):
+class PostRegressionModelByNameOrIdPredictionParams(TypedDict):
     name_or_id: str
 
 
+class PostClassificationModelByModelNameOrIdEvaluationParams(TypedDict):
+    model_name_or_id: str
+
+
+class GetClassificationModelByModelNameOrIdEvaluationParams(TypedDict):
+    model_name_or_id: str
+
+
 class PostRegressionModelByModelNameOrIdEvaluationParams(TypedDict):
     model_name_or_id: str
 
@@ -979,26 +1060,36 @@ class GetRegressionModelByModelNameOrIdEvaluationParams(TypedDict):
     model_name_or_id: str
 
 
-class GetRegressionModelByModelNameOrIdEvaluationByTaskIdParams(TypedDict):
+class GetClassificationModelByModelNameOrIdEvaluationByJobIdParams(TypedDict):
     model_name_or_id: str
-    task_id: str
+    job_id: str
 
 
-class DeleteRegressionModelByModelNameOrIdEvaluationByTaskIdParams(TypedDict):
+class DeleteClassificationModelByModelNameOrIdEvaluationByJobIdParams(TypedDict):
     model_name_or_id: str
-    task_id: str
+    job_id: str
 
 
-class GetTaskByTaskIdParams(TypedDict):
-    task_id: str
+class GetRegressionModelByModelNameOrIdEvaluationByJobIdParams(TypedDict):
+    model_name_or_id: str
+    job_id: str
 
 
-class GetTaskByTaskIdStatusParams(TypedDict):
-    task_id: str
+class DeleteRegressionModelByModelNameOrIdEvaluationByJobIdParams(TypedDict):
+    model_name_or_id: str
+    job_id: str
+
+
+class GetJobByJobIdParams(TypedDict):
+    job_id: str
+
 
+class GetJobByJobIdStatusParams(TypedDict):
+    job_id: str
 
-class GetTaskParams(TypedDict):
-    status: NotRequired[TaskStatus | list[TaskStatus] | None]
+
+class GetJobParams(TypedDict):
+    status: NotRequired[JobStatus | list[JobStatus] | None]
     type: NotRequired[str | list[str] | None]
     limit: NotRequired[int | None]
     offset: NotRequired[int]
@@ -1006,8 +1097,8 @@ class GetTaskParams(TypedDict):
     end_timestamp: NotRequired[str | None]
 
 
-class DeleteTaskByTaskIdAbortParams(TypedDict):
-    task_id: str
+class DeleteJobByJobIdAbortParams(TypedDict):
+    job_id: str
 
 
 class GetWorkerParams(TypedDict):
@@ -1063,43 +1154,8 @@ class DeleteTelemetryFeedbackCategoryByNameOrIdParams(TypedDict):
 PutTelemetryPredictionFeedbackRequest = list[PredictionFeedbackRequest]
 
 
-class GetAgentsBootstrapClassificationModelByTaskIdParams(TypedDict):
-    task_id: str
-
-
-class PostGpuMemorysetByNameOrIdLookupParams(TypedDict):
-    name_or_id: str
-
-
-class PatchGpuMemorysetByNameOrIdMemoryParams(TypedDict):
-    name_or_id: str
-
-
-class PostGpuMemorysetByNameOrIdMemoryParams(TypedDict):
-    name_or_id: str
-
-
-PostGpuMemorysetByNameOrIdMemoryRequest = list[LabeledMemoryInsert] | list[ScoredMemoryInsert]
-
-
-class PatchGpuMemorysetByNameOrIdMemoriesParams(TypedDict):
-    name_or_id: str
-
-
-class PostGpuClassificationModelByNameOrIdPredictionParams(TypedDict):
-    name_or_id: str
-
-
-class PostGpuRegressionModelByNameOrIdPredictionParams(TypedDict):
-    name_or_id: str
-
-
-class PostGpuFinetunedEmbeddingModelByNameOrIdEmbeddingParams(TypedDict):
-    name_or_id: str
-
-
-class PostGpuPretrainedEmbeddingModelByModelNameEmbeddingParams(TypedDict):
-    model_name: PretrainedEmbeddingModelName
+class GetAgentsBootstrapClassificationModelByJobIdParams(TypedDict):
+    job_id: str
 
 
 class FieldValidationError(TypedDict):
@@ -1171,6 +1227,9 @@ class ClassificationPredictionRequest(TypedDict):
     prompt: NotRequired[str | None]
     use_lookup_cache: NotRequired[bool]
     consistency_level: NotRequired[Literal["Bounded", "Session", "Strong", "Eventual"] | None]
+    ignore_unlabeled: NotRequired[bool]
+    partition_ids: NotRequired[str | list[str | None] | None]
+    partition_filter_mode: NotRequired[Literal["ignore_partitions", "include_global", "exclude_global", "only_global"]]
 
 
 class CloneMemorysetRequest(TypedDict):
@@ -1187,6 +1246,7 @@ class ColumnInfo(TypedDict):
     name: str
     type: ColumnType
     enum_options: NotRequired[list[str] | None]
+    string_values: NotRequired[list[str] | None]
     int_values: NotRequired[list[int] | None]
     contains_nones: NotRequired[bool]
 
@@ -1223,6 +1283,7 @@ class CreateMemorysetRequest(TypedDict):
     datasource_score_column: NotRequired[str | None]
     datasource_value_column: str
     datasource_source_id_column: NotRequired[str | None]
+    datasource_partition_id_column: NotRequired[str | None]
     remove_duplicates: NotRequired[bool]
     pretrained_embedding_model_name: NotRequired[PretrainedEmbeddingModelName | None]
     finetuned_embedding_model_name_or_id: NotRequired[str | None]
@@ -1233,6 +1294,8 @@ class CreateMemorysetRequest(TypedDict):
     prompt: NotRequired[str]
     hidden: NotRequired[bool]
     batch_size: NotRequired[int]
+    subsample: NotRequired[int | float | None]
+    memory_type: NotRequired[MemoryType]
 
 
 class CreateRegressionModelRequest(TypedDict):
@@ -1257,48 +1320,52 @@ class DatasourceMetadata(TypedDict):
 
 
 class EmbeddingEvaluationResponse(TypedDict):
-    task_id: str
+    job_id: str
     org_id: str
     finetuned_embedding_model_id: str | None
     pretrained_embedding_model_name: PretrainedEmbeddingModelName | None
     datasource_id: str
-    subsample: int | None
+    subsample: int | float | None
     datasource_value_column: str
     datasource_label_column: NotRequired[str | None]
     datasource_score_column: NotRequired[str | None]
     neighbor_count: int
     weigh_memories: bool
-    status: TaskStatus
+    status: JobStatus
     result: ClassificationMetrics | RegressionMetrics | None
     created_at: str
     updated_at: str
+    task_id: str
 
 
 class EvaluationResponse(TypedDict):
-    task_id: str
+    job_id: str
     org_id: str
-    status: TaskStatus
+    status: JobStatus
     result: ClassificationMetrics | RegressionMetrics | None
     created_at: str
     updated_at: str
+    task_id: str
 
 
 class EvaluationResponseClassificationMetrics(TypedDict):
-    task_id: str
+    job_id: str
     org_id: str
-    status: TaskStatus
+    status: JobStatus
     result: ClassificationMetrics | None
     created_at: str
     updated_at: str
+    task_id: str
 
 
 class EvaluationResponseRegressionMetrics(TypedDict):
-    task_id: str
+    job_id: str
     org_id: str
-    status: TaskStatus
+    status: JobStatus
     result: RegressionMetrics | None
     created_at: str
     updated_at: str
+    task_id: str
 
 
 class FinetuneEmbeddingModelRequest(TypedDict):
@@ -1307,7 +1374,8 @@ class FinetuneEmbeddingModelRequest(TypedDict):
     train_memoryset_name_or_id: NotRequired[str | None]
     train_datasource_name_or_id: NotRequired[str | None]
     eval_datasource_name_or_id: NotRequired[str | None]
-    label_column: NotRequired[str]
+    label_column: NotRequired[str | None]
+    score_column: NotRequired[str | None]
     value_column: NotRequired[str]
     training_method: NotRequired[EmbeddingFinetuningMethod]
     training_args: NotRequired[dict[str, str | int | float | bool]]
@@ -1324,8 +1392,9 @@ class FinetunedEmbeddingModelMetadata(TypedDict):
     created_at: str
     updated_at: str
     base_model: PretrainedEmbeddingModelName
+    finetuning_job_id: str
+    finetuning_status: JobStatus
     finetuning_task_id: str
-    finetuning_status: TaskStatus
 
 
 class HTTPValidationError(TypedDict):
@@ -1337,10 +1406,28 @@ class InvalidInputErrorResponse(TypedDict):
     validation_issues: list[FieldValidationError]
 
 
+class Job(TypedDict):
+    status: JobStatus
+    steps_total: int | None
+    steps_completed: int | None
+    exception: str | None
+    updated_at: str
+    created_at: str
+    id: str
+    org_id: str
+    worker_id: str | None
+    type: str
+    payload: BaseModel
+    result: BaseModel | None
+    depends_on: NotRequired[list[str]]
+    lease_token: str | None
+
+
 class LabelPredictionMemoryLookup(TypedDict):
     value: str | bytes
     embedding: list[float]
     source_id: str | None
+    partition_id: str | None
     metadata: dict[str, str | int | float | bool | None]
     memory_id: str
     memory_version: int
@@ -1382,6 +1469,7 @@ class LabeledMemory(TypedDict):
     value: str | bytes
     embedding: list[float]
     source_id: str | None
+    partition_id: str | None
     metadata: dict[str, str | int | float | bool | None]
     memory_id: str
     memory_version: int
@@ -1397,6 +1485,7 @@ class LabeledMemoryLookup(TypedDict):
     value: str | bytes
     embedding: list[float]
     source_id: str | None
+    partition_id: str | None
     metadata: dict[str, str | int | float | bool | None]
     memory_id: str
     memory_version: int
@@ -1414,6 +1503,7 @@ class LabeledMemoryUpdate(TypedDict):
     value: NotRequired[str | bytes]
     metadata: NotRequired[dict[str, str | int | float | bool | None] | None]
     source_id: NotRequired[str | None]
+    partition_id: NotRequired[str | None]
     metrics: NotRequired[MemoryMetrics | None]
     label: NotRequired[int | None]
 
@@ -1422,6 +1512,7 @@ class LabeledMemoryWithFeedbackMetrics(TypedDict):
     value: str | bytes
     embedding: list[float]
     source_id: str | None
+    partition_id: str | None
     metadata: dict[str, str | int | float | bool | None]
     memory_id: str
     memory_version: int
@@ -1441,7 +1532,8 @@ class ListPredictionsRequest(TypedDict):
     prediction_ids: NotRequired[list[str] | None]
     start_timestamp: NotRequired[str | None]
     end_timestamp: NotRequired[str | None]
-    limit: NotRequired[int | None]
+    memory_id: NotRequired[str | None]
+    limit: NotRequired[int]
     offset: NotRequired[int | None]
     sort: NotRequired[PredictionSort]
     expected_label_match: NotRequired[bool | None]
@@ -1462,6 +1554,7 @@ class MemorysetAnalysisRequest(TypedDict):
     batch_size: NotRequired[int]
     clear_metrics: NotRequired[bool]
     configs: MemorysetAnalysisConfigs
+    partition_filter_mode: NotRequired[Literal["ignore_partitions", "include_global", "exclude_global", "only_global"]]
 
 
 class MemorysetConceptMetrics(TypedDict):
@@ -1480,6 +1573,13 @@ class MemorysetMetrics(TypedDict):
     concepts: NotRequired[MemorysetConceptMetrics | None]
 
 
+class PaginatedJob(TypedDict):
+    items: list[Job]
+    total: int
+    offset: int
+    limit: int
+
+
 class PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics(TypedDict):
     items: list[LabeledMemoryWithFeedbackMetrics | ScoredMemoryWithFeedbackMetrics]
     total: int
@@ -1497,23 +1597,6 @@ class PretrainedEmbeddingModelMetadata(TypedDict):
     num_params: int
 
 
-class Task(TypedDict):
-    status: TaskStatus
-    steps_total: int | None
-    steps_completed: int | None
-    exception: str | None
-    updated_at: str
-    created_at: str
-    id: str
-    org_id: str
-    worker_id: str | None
-    type: str
-    payload: BaseModel
-    result: BaseModel | None
-    depends_on: list[str]
-    lease_token: str | None
-
-
 class TelemetryMemoriesRequest(TypedDict):
     memoryset_id: str
     offset: NotRequired[int]
@@ -1545,10 +1628,10 @@ class CascadingEditSuggestion(TypedDict):
 
 
 class MemorysetAnalysisResponse(TypedDict):
-    task_id: str
+    job_id: str
     org_id: str
     memoryset_id: str
-    status: TaskStatus
+    status: JobStatus
     lookup_count: int
     batch_size: int
     clear_metrics: bool
@@ -1556,6 +1639,7 @@ class MemorysetAnalysisResponse(TypedDict):
     results: MemorysetMetrics | None
     created_at: str
     updated_at: str
+    task_id: str
 
 
 class MemorysetMetadata(TypedDict):
@@ -1571,8 +1655,8 @@ class MemorysetMetadata(TypedDict):
     created_at: str
     updated_at: str
     memories_updated_at: str
-    insertion_task_id: str
-    insertion_status: TaskStatus
+    insertion_job_id: str
+    insertion_status: JobStatus
     metrics: MemorysetMetrics
     memory_type: MemoryType
     label_names: list[str] | None
@@ -1582,13 +1666,7 @@ class MemorysetMetadata(TypedDict):
     document_prompt_override: str | None
     query_prompt_override: str | None
     hidden: bool
-
-
-class PaginatedTask(TypedDict):
-    items: list[Task]
-    total: int
-    offset: int
-    limit: int
+    insertion_task_id: str
 
 
 class PaginatedWorkerInfo(TypedDict):
@@ -1606,11 +1684,12 @@ class BootstrapClassificationModelMeta(TypedDict):
 
 
 class BootstrapClassificationModelResponse(TypedDict):
-    task_id: str
+    job_id: str
     org_id: str
-    status: TaskStatus
+    status: JobStatus
     result: BootstrapClassificationModelMeta | None
     input: BootstrapClassificationModelRequest | None
+    task_id: str
 
 
 class OrcaAsyncClient(AsyncClient):
@@ -1889,9 +1968,9 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def GET(
         self,
-        path: Literal["/memoryset/{name_or_id}/analysis/{analysis_task_id}"],
+        path: Literal["/memoryset/{name_or_id}/analysis/{analysis_job_id}"],
         *,
-        params: GetMemorysetByNameOrIdAnalysisByAnalysisTaskIdParams,
+        params: GetMemorysetByNameOrIdAnalysisByAnalysisJobIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1939,9 +2018,9 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def GET(
         self,
-        path: Literal["/finetuned_embedding_model/{name_or_id}/evaluation/{task_id}"],
+        path: Literal["/pretrained_embedding_model"],
         *,
-        params: GetFinetunedEmbeddingModelByNameOrIdEvaluationByTaskIdParams,
+        params: None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1949,16 +2028,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> EmbeddingEvaluationResponse:
-        """Get evaluation results for a finetuned embedding model by task ID."""
+    ) -> list[PretrainedEmbeddingModelMetadata]:
+        """List all available pretrained embedding models."""
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/finetuned_embedding_model/{name_or_id}/evaluations"],
+        path: Literal["/pretrained_embedding_model/{model_name}"],
         *,
-        params: GetFinetunedEmbeddingModelByNameOrIdEvaluationsParams,
+        params: GetPretrainedEmbeddingModelByModelNameParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1966,16 +2045,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[EmbeddingEvaluationResponse]:
-        """List all evaluation results for a finetuned embedding model."""
+    ) -> PretrainedEmbeddingModelMetadata:
+        """Get metadata for a specific pretrained embedding model."""
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/pretrained_embedding_model"],
+        path: Literal["/finetuned_embedding_model/{name_or_id}/evaluation/{job_id}"],
         *,
-        params: None = None,
+        params: GetFinetunedEmbeddingModelByNameOrIdEvaluationByJobIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -1983,16 +2062,16 @@ class OrcaAsyncClient(AsyncClient):
         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."""
+    ) -> EmbeddingEvaluationResponse:
+        """Get evaluation results for a finetuned embedding model by job ID."""
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/pretrained_embedding_model/{model_name}"],
+        path: Literal["/pretrained_embedding_model/{model_name}/evaluation/{job_id}"],
         *,
-        params: GetPretrainedEmbeddingModelByModelNameParams,
+        params: GetPretrainedEmbeddingModelByModelNameEvaluationByJobIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2000,16 +2079,16 @@ class OrcaAsyncClient(AsyncClient):
         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."""
+    ) -> EmbeddingEvaluationResponse:
+        """Get evaluation results for a pretrained embedding model by job ID."""
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/pretrained_embedding_model/{model_name}/evaluation/{task_id}"],
+        path: Literal["/finetuned_embedding_model/{name_or_id}/evaluations"],
         *,
-        params: GetPretrainedEmbeddingModelByModelNameEvaluationByTaskIdParams,
+        params: GetFinetunedEmbeddingModelByNameOrIdEvaluationsParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2017,8 +2096,8 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> EmbeddingEvaluationResponse:
-        """Get evaluation results for a pretrained embedding model by task ID."""
+    ) -> list[EmbeddingEvaluationResponse]:
+        """List all evaluation results for a finetuned embedding model."""
         pass
 
     @overload
@@ -2143,7 +2222,7 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def GET(
         self,
-        path: Literal["/predictive_model"],
+        path: Literal["/classification_model"],
         *,
         params: None = None,
         parse_as: Literal["json"] = "json",
@@ -2153,13 +2232,13 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[ClassificationModelMetadata | RegressionModelMetadata]:
+    ) -> list[ClassificationModelMetadata]:
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/classification_model"],
+        path: Literal["/regression_model"],
         *,
         params: None = None,
         parse_as: Literal["json"] = "json",
@@ -2169,7 +2248,7 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[ClassificationModelMetadata]:
+    ) -> list[RegressionModelMetadata]:
         pass
 
     @overload
@@ -2191,9 +2270,9 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def GET(
         self,
-        path: Literal["/classification_model/{model_name_or_id}/evaluation"],
+        path: Literal["/regression_model/{name_or_id}"],
         *,
-        params: GetClassificationModelByModelNameOrIdEvaluationParams,
+        params: GetRegressionModelByNameOrIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2201,15 +2280,15 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[EvaluationResponseClassificationMetrics]:
+    ) -> RegressionModelMetadata:
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/classification_model/{model_name_or_id}/evaluation/{task_id}"],
+        path: Literal["/predictive_model"],
         *,
-        params: GetClassificationModelByModelNameOrIdEvaluationByTaskIdParams,
+        params: None = None,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2217,15 +2296,15 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> EvaluationResponseClassificationMetrics:
+    ) -> list[ClassificationModelMetadata | RegressionModelMetadata]:
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/regression_model"],
+        path: Literal["/classification_model/{model_name_or_id}/evaluation"],
         *,
-        params: None = None,
+        params: GetClassificationModelByModelNameOrIdEvaluationParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2233,15 +2312,15 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[RegressionModelMetadata]:
+    ) -> list[EvaluationResponseClassificationMetrics]:
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/regression_model/{name_or_id}"],
+        path: Literal["/regression_model/{model_name_or_id}/evaluation"],
         *,
-        params: GetRegressionModelByNameOrIdParams,
+        params: GetRegressionModelByModelNameOrIdEvaluationParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2249,15 +2328,15 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> RegressionModelMetadata:
+    ) -> list[EvaluationResponseRegressionMetrics]:
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/regression_model/{model_name_or_id}/evaluation"],
+        path: Literal["/classification_model/{model_name_or_id}/evaluation/{job_id}"],
         *,
-        params: GetRegressionModelByModelNameOrIdEvaluationParams,
+        params: GetClassificationModelByModelNameOrIdEvaluationByJobIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2265,15 +2344,15 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[EvaluationResponseRegressionMetrics]:
+    ) -> EvaluationResponseClassificationMetrics:
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/regression_model/{model_name_or_id}/evaluation/{task_id}"],
+        path: Literal["/regression_model/{model_name_or_id}/evaluation/{job_id}"],
         *,
-        params: GetRegressionModelByModelNameOrIdEvaluationByTaskIdParams,
+        params: GetRegressionModelByModelNameOrIdEvaluationByJobIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2287,9 +2366,9 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def GET(
         self,
-        path: Literal["/task/{task_id}"],
+        path: Literal["/job/{job_id}"],
         *,
-        params: GetTaskByTaskIdParams,
+        params: GetJobByJobIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2297,15 +2376,15 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Task:
+    ) -> Job:
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/task/{task_id}/status"],
+        path: Literal["/job/{job_id}/status"],
         *,
-        params: GetTaskByTaskIdStatusParams,
+        params: GetJobByJobIdStatusParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2313,15 +2392,15 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> TaskStatusInfo:
+    ) -> JobStatusInfo:
         pass
 
     @overload
     async def GET(
         self,
-        path: Literal["/task"],
+        path: Literal["/job"],
         *,
-        params: GetTaskParams,
+        params: GetJobParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2329,7 +2408,7 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> PaginatedTask:
+    ) -> PaginatedJob:
         pass
 
     @overload
@@ -2480,9 +2559,9 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def GET(
         self,
-        path: Literal["/agents/bootstrap_classification_model/{task_id}"],
+        path: Literal["/agents/bootstrap_classification_model/{job_id}"],
         *,
-        params: GetAgentsBootstrapClassificationModelByTaskIdParams,
+        params: GetAgentsBootstrapClassificationModelByJobIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2491,7 +2570,7 @@ class OrcaAsyncClient(AsyncClient):
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
     ) -> BootstrapClassificationModelResponse:
-        """Get the status of a bootstrap classification model task"""
+        """Get the status of a bootstrap classification model job"""
         pass
 
     async def GET(
@@ -2661,9 +2740,9 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def DELETE(
         self,
-        path: Literal["/classification_model/{model_name_or_id}/evaluation/{task_id}"],
+        path: Literal["/regression_model/{name_or_id}"],
         *,
-        params: DeleteClassificationModelByModelNameOrIdEvaluationByTaskIdParams,
+        params: DeleteRegressionModelByNameOrIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2677,9 +2756,9 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def DELETE(
         self,
-        path: Literal["/regression_model/{name_or_id}"],
+        path: Literal["/classification_model/{model_name_or_id}/evaluation/{job_id}"],
         *,
-        params: DeleteRegressionModelByNameOrIdParams,
+        params: DeleteClassificationModelByModelNameOrIdEvaluationByJobIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2693,9 +2772,9 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def DELETE(
         self,
-        path: Literal["/regression_model/{model_name_or_id}/evaluation/{task_id}"],
+        path: Literal["/regression_model/{model_name_or_id}/evaluation/{job_id}"],
         *,
-        params: DeleteRegressionModelByModelNameOrIdEvaluationByTaskIdParams,
+        params: DeleteRegressionModelByModelNameOrIdEvaluationByJobIdParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2709,9 +2788,9 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def DELETE(
         self,
-        path: Literal["/task/{task_id}/abort"],
+        path: Literal["/job/{job_id}/abort"],
         *,
-        params: DeleteTaskByTaskIdAbortParams,
+        params: DeleteJobByJobIdAbortParams,
         parse_as: Literal["json"] = "json",
         headers: HeaderTypes | None = None,
         cookies: CookieTypes | None = None,
@@ -2872,6 +2951,26 @@ class OrcaAsyncClient(AsyncClient):
     ) -> None:
         pass
 
+    @overload
+    async def POST(
+        self,
+        path: Literal["/gpu/memoryset/{name_or_id}/lookup"],
+        *,
+        params: PostGpuMemorysetByNameOrIdLookupParams,
+        json: LookupRequest,
+        data: None = None,
+        files: None = None,
+        content: 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[list[LabeledMemoryLookup | ScoredMemoryLookup]]:
+        pass
+
     @overload
     async def POST(
         self,
@@ -2932,6 +3031,26 @@ class OrcaAsyncClient(AsyncClient):
     ) -> None:
         pass
 
+    @overload
+    async def POST(
+        self,
+        path: Literal["/gpu/memoryset/{name_or_id}/memory"],
+        *,
+        params: PostGpuMemorysetByNameOrIdMemoryParams,
+        json: PostGpuMemorysetByNameOrIdMemoryRequest,
+        data: None = None,
+        files: None = None,
+        content: 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[str]:
+        pass
+
     @overload
     async def POST(
         self,
@@ -2993,6 +3112,48 @@ class OrcaAsyncClient(AsyncClient):
         """Create a finetuned embedding model."""
         pass
 
+    @overload
+    async def POST(
+        self,
+        path: Literal["/gpu/finetuned_embedding_model/{name_or_id}/embedding"],
+        *,
+        params: PostGpuFinetunedEmbeddingModelByNameOrIdEmbeddingParams,
+        json: EmbedRequest,
+        data: None = None,
+        files: None = None,
+        content: 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[list[float]]:
+        """Embed values using a finetuned embedding model."""
+        pass
+
+    @overload
+    async def POST(
+        self,
+        path: Literal["/gpu/pretrained_embedding_model/{model_name}/embedding"],
+        *,
+        params: PostGpuPretrainedEmbeddingModelByModelNameEmbeddingParams,
+        json: EmbedRequest,
+        data: None = None,
+        files: None = None,
+        content: 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[list[float]]:
+        """Embed values using a pretrained embedding model."""
+        pass
+
     @overload
     async def POST(
         self,
@@ -3092,10 +3253,10 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def POST(
         self,
-        path: Literal["/classification_model"],
+        path: Literal["/datasource/{name_or_id}/rows"],
         *,
-        params: None = None,
-        json: CreateClassificationModelRequest,
+        params: PostDatasourceByNameOrIdRowsParams,
+        json: GetDatasourceRowsRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3106,16 +3267,17 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> ClassificationModelMetadata:
+    ) -> list[dict[str, Any]]:
+        """Get rows from a specific datasource with optional filtering."""
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/classification_model/{model_name_or_id}/evaluation"],
+        path: Literal["/datasource/{name_or_id}/rows/count"],
         *,
-        params: PostClassificationModelByModelNameOrIdEvaluationParams,
-        json: ClassificationEvaluationRequest,
+        params: PostDatasourceByNameOrIdRowsCountParams,
+        json: GetDatasourceRowCountRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3126,16 +3288,17 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> EvaluationResponse:
+    ) -> int:
+        """Get row count from a specific datasource with optional filtering."""
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/regression_model"],
+        path: Literal["/classification_model"],
         *,
         params: None = None,
-        json: CreateRegressionModelRequest,
+        json: CreateClassificationModelRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3146,16 +3309,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> RegressionModelMetadata:
+    ) -> ClassificationModelMetadata:
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/regression_model/{model_name_or_id}/evaluation"],
+        path: Literal["/regression_model"],
         *,
-        params: PostRegressionModelByModelNameOrIdEvaluationParams,
-        json: RegressionEvaluationRequest,
+        params: None = None,
+        json: CreateRegressionModelRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3166,16 +3329,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> EvaluationResponse:
+    ) -> RegressionModelMetadata:
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/telemetry/prediction"],
+        path: Literal["/gpu/classification_model/{name_or_id}/prediction"],
         *,
-        params: None = None,
-        json: ListPredictionsRequest | None = None,
+        params: PostGpuClassificationModelByNameOrIdPredictionParams,
+        json: ClassificationPredictionRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3186,17 +3349,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[LabelPredictionWithMemoriesAndFeedback | ScorePredictionWithMemoriesAndFeedback]:
-        """List predictions with optional filtering and sorting."""
+    ) -> list[BaseLabelPredictionResult]:
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/telemetry/prediction/count"],
+        path: Literal["/classification_model/{name_or_id}/prediction"],
         *,
-        params: None = None,
-        json: CountPredictionsRequest | None = None,
+        params: PostClassificationModelByNameOrIdPredictionParams,
+        json: ClassificationPredictionRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3207,17 +3369,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> int:
-        """Count predictions with optional filtering."""
+    ) -> list[BaseLabelPredictionResult]:
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/telemetry/memories"],
+        path: Literal["/gpu/regression_model/{name_or_id}/prediction"],
         *,
-        params: None = None,
-        json: TelemetryMemoriesRequest,
+        params: PostGpuRegressionModelByNameOrIdPredictionParams,
+        json: RegressionPredictionRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3228,21 +3389,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics:
-        """
-        List memories with feedback metrics.
-        **Note**: This endpoint will ONLY return memories that have been used in a prediction.
-        If you want to query ALL memories WITHOUT feedback metrics, use the query_memoryset endpoint.
-        """
+    ) -> list[BaseScorePredictionResult]:
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/agents/bootstrap_classification_model"],
+        path: Literal["/regression_model/{name_or_id}/prediction"],
         *,
-        params: None = None,
-        json: BootstrapClassificationModelRequest,
+        params: PostRegressionModelByNameOrIdPredictionParams,
+        json: RegressionPredictionRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3253,30 +3409,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> BootstrapClassificationModelResponse:
-        """
-        Bootstrap a classification model by creating a memoryset with generated memories and a classification model.
-
-        This endpoint uses the bootstrap_classification_model agent to generate:
-        1. Memoryset configuration with appropriate settings
-        2. Model configuration with optimal parameters
-        3. High-quality training memories for each label
-
-        The process involves:
-        1. Calling the agent to generate configurations and memories
-        2. Creating a datasource from the generated memories
-        3. Creating a memoryset from the datasource
-        4. Creating a classification model from the memoryset
-        """
+    ) -> list[BaseScorePredictionResult]:
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/gpu/memoryset/{name_or_id}/lookup"],
+        path: Literal["/classification_model/{model_name_or_id}/evaluation"],
         *,
-        params: PostGpuMemorysetByNameOrIdLookupParams,
-        json: LookupRequest,
+        params: PostClassificationModelByModelNameOrIdEvaluationParams,
+        json: ClassificationEvaluationRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3287,16 +3429,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[list[LabeledMemoryLookup | ScoredMemoryLookup]]:
+    ) -> EvaluationResponse:
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/gpu/memoryset/{name_or_id}/memory"],
+        path: Literal["/regression_model/{model_name_or_id}/evaluation"],
         *,
-        params: PostGpuMemorysetByNameOrIdMemoryParams,
-        json: PostGpuMemorysetByNameOrIdMemoryRequest,
+        params: PostRegressionModelByModelNameOrIdEvaluationParams,
+        json: RegressionEvaluationRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3307,16 +3449,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[str]:
+    ) -> EvaluationResponse:
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/gpu/classification_model/{name_or_id}/prediction"],
+        path: Literal["/telemetry/prediction"],
         *,
-        params: PostGpuClassificationModelByNameOrIdPredictionParams,
-        json: ClassificationPredictionRequest,
+        params: None = None,
+        json: ListPredictionsRequest | None = None,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3327,16 +3469,17 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[BaseLabelPredictionResult]:
+    ) -> list[LabelPredictionWithMemoriesAndFeedback | ScorePredictionWithMemoriesAndFeedback]:
+        """List predictions with optional filtering and sorting."""
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/gpu/regression_model/{name_or_id}/prediction"],
+        path: Literal["/telemetry/prediction/count"],
         *,
-        params: PostGpuRegressionModelByNameOrIdPredictionParams,
-        json: RegressionPredictionRequest,
+        params: None = None,
+        json: CountPredictionsRequest | None = None,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3347,16 +3490,17 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[BaseScorePredictionResult]:
+    ) -> int:
+        """Count predictions with optional filtering."""
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/gpu/finetuned_embedding_model/{name_or_id}/embedding"],
+        path: Literal["/telemetry/memories"],
         *,
-        params: PostGpuFinetunedEmbeddingModelByNameOrIdEmbeddingParams,
-        json: EmbedRequest,
+        params: None = None,
+        json: TelemetryMemoriesRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3367,17 +3511,21 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[list[float]]:
-        """Embed values using a finetuned embedding model."""
+    ) -> PaginatedUnionLabeledMemoryWithFeedbackMetricsScoredMemoryWithFeedbackMetrics:
+        """
+        List memories with feedback metrics.
+        **Note**: This endpoint will ONLY return memories that have been used in a prediction.
+        If you want to query ALL memories WITHOUT feedback metrics, use the query_memoryset endpoint.
+        """
         pass
 
     @overload
     async def POST(
         self,
-        path: Literal["/gpu/pretrained_embedding_model/{model_name}/embedding"],
+        path: Literal["/agents/bootstrap_classification_model"],
         *,
-        params: PostGpuPretrainedEmbeddingModelByModelNameEmbeddingParams,
-        json: EmbedRequest,
+        params: None = None,
+        json: BootstrapClassificationModelRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3388,8 +3536,21 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[list[float]]:
-        """Embed values using a pretrained embedding model."""
+    ) -> BootstrapClassificationModelResponse:
+        """
+        Bootstrap a classification model by creating a memoryset with generated memories and a classification model.
+
+        This endpoint uses the bootstrap_classification_model agent to generate:
+        1. Memoryset configuration with appropriate settings
+        2. Model configuration with optimal parameters
+        3. High-quality training memories for each label
+
+        The process involves:
+        1. Calling the agent to generate configurations and memories
+        2. Creating a datasource from the generated memories
+        3. Creating a memoryset from the datasource
+        4. Creating a classification model from the memoryset
+        """
         pass
 
     async def POST(
@@ -3535,10 +3696,10 @@ class OrcaAsyncClient(AsyncClient):
     @overload
     async def PATCH(
         self,
-        path: Literal["/classification_model/{name_or_id}"],
+        path: Literal["/gpu/memoryset/{name_or_id}/memory"],
         *,
-        params: PatchClassificationModelByNameOrIdParams,
-        json: PredictiveModelUpdate,
+        params: PatchGpuMemorysetByNameOrIdMemoryParams,
+        json: PatchGpuMemorysetByNameOrIdMemoryRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3549,16 +3710,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> ClassificationModelMetadata:
+    ) -> LabeledMemory | ScoredMemory:
         pass
 
     @overload
     async def PATCH(
         self,
-        path: Literal["/regression_model/{name_or_id}"],
+        path: Literal["/gpu/memoryset/{name_or_id}/memories"],
         *,
-        params: PatchRegressionModelByNameOrIdParams,
-        json: PredictiveModelUpdate,
+        params: PatchGpuMemorysetByNameOrIdMemoriesParams,
+        json: PatchGpuMemorysetByNameOrIdMemoriesRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3569,16 +3730,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> RegressionModelMetadata:
+    ) -> list[LabeledMemory] | list[ScoredMemory]:
         pass
 
     @overload
     async def PATCH(
         self,
-        path: Literal["/telemetry/prediction/{prediction_id}"],
+        path: Literal["/classification_model/{name_or_id}"],
         *,
-        params: PatchTelemetryPredictionByPredictionIdParams,
-        json: UpdatePredictionRequest,
+        params: PatchClassificationModelByNameOrIdParams,
+        json: PredictiveModelUpdate,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3589,17 +3750,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Any:
-        """Update a prediction with new expected values, tags, or memory ID."""
+    ) -> ClassificationModelMetadata:
         pass
 
     @overload
     async def PATCH(
         self,
-        path: Literal["/gpu/memoryset/{name_or_id}/memory"],
+        path: Literal["/regression_model/{name_or_id}"],
         *,
-        params: PatchGpuMemorysetByNameOrIdMemoryParams,
-        json: PatchGpuMemorysetByNameOrIdMemoryRequest,
+        params: PatchRegressionModelByNameOrIdParams,
+        json: PredictiveModelUpdate,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3610,16 +3770,16 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> LabeledMemory | ScoredMemory:
+    ) -> RegressionModelMetadata:
         pass
 
     @overload
     async def PATCH(
         self,
-        path: Literal["/gpu/memoryset/{name_or_id}/memories"],
+        path: Literal["/telemetry/prediction/{prediction_id}"],
         *,
-        params: PatchGpuMemorysetByNameOrIdMemoriesParams,
-        json: PatchGpuMemorysetByNameOrIdMemoriesRequest,
+        params: PatchTelemetryPredictionByPredictionIdParams,
+        json: UpdatePredictionRequest,
         data: None = None,
         files: None = None,
         content: None = None,
@@ -3630,7 +3790,8 @@ class OrcaAsyncClient(AsyncClient):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[LabeledMemory] | list[ScoredMemory]:
+    ) -> Any:
+        """Update a prediction with new expected values, tags, or memory ID."""
         pass
 
     async def PATCH(