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

orca_sdk/client.py

@@ -133,6 +133,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):
@@ -161,6 +165,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):
@@ -191,6 +196,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]
 
@@ -208,7 +219,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]
@@ -217,7 +228,7 @@ class EmbeddingEvaluationRequest(TypedDict):
     weigh_memories: NotRequired[bool]
 
 
-EmbeddingFinetuningMethod = Literal["classification", "batch_triplet_loss"]
+EmbeddingFinetuningMethod = Literal["classification", "regression", "batch_triplet_loss"]
 
 
 class FeedbackMetrics(TypedDict):
@@ -231,7 +242,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):
@@ -252,6 +275,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]
@@ -274,6 +309,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
 
 
@@ -281,12 +317,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):
@@ -338,8 +378,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]
@@ -368,6 +406,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]
 
 
@@ -437,7 +476,7 @@ class NotFoundErrorResponse(TypedDict):
             "memory",
             "evaluation",
             "analysis",
-            "task",
+            "job",
             "pretrained_embedding_model",
             "finetuned_embedding_model",
             "feedback_category",
@@ -551,6 +590,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):
@@ -593,12 +636,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
@@ -636,6 +683,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
@@ -651,6 +699,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
 
 
@@ -658,6 +707,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
@@ -674,6 +724,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]
 
@@ -682,6 +733,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
@@ -707,18 +759,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
 
 
@@ -791,6 +831,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
@@ -823,20 +867,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):
@@ -852,34 +911,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):
@@ -909,6 +976,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]
@@ -939,36 +1014,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
 
@@ -977,26 +1058,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]
@@ -1004,8 +1095,8 @@ class GetTaskParams(TypedDict):
     end_timestamp: NotRequired[str | None]
 
 
-class DeleteTaskByTaskIdAbortParams(TypedDict):
-    task_id: str
+class DeleteJobByJobIdAbortParams(TypedDict):
+    job_id: str
 
 
 class GetWorkerParams(TypedDict):
@@ -1061,43 +1152,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):
@@ -1169,6 +1225,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):
@@ -1185,6 +1244,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]
 
@@ -1221,6 +1281,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]
@@ -1231,6 +1292,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):
@@ -1255,48 +1318,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):
@@ -1305,7 +1372,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]]
@@ -1322,8 +1390,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):
@@ -1335,10 +1404,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
@@ -1380,6 +1467,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
@@ -1395,6 +1483,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
@@ -1412,6 +1501,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]
 
@@ -1420,6 +1510,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
@@ -1439,7 +1530,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]
@@ -1460,6 +1552,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):
@@ -1478,6 +1571,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
@@ -1495,23 +1595,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]
@@ -1543,10 +1626,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
@@ -1554,6 +1637,7 @@ class MemorysetAnalysisResponse(TypedDict):
     results: MemorysetMetrics | None
     created_at: str
     updated_at: str
+    task_id: str
 
 
 class MemorysetMetadata(TypedDict):
@@ -1569,8 +1653,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
@@ -1580,13 +1664,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):
@@ -1604,11 +1682,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 OrcaClient(Client):
@@ -1887,9 +1966,9 @@ class OrcaClient(Client):
     @overload
     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,
@@ -1937,9 +2016,9 @@ class OrcaClient(Client):
     @overload
     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,
@@ -1947,16 +2026,16 @@ class OrcaClient(Client):
         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
     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,
@@ -1964,16 +2043,16 @@ class OrcaClient(Client):
         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
     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,
@@ -1981,16 +2060,16 @@ 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."""
+    ) -> EmbeddingEvaluationResponse:
+        """Get evaluation results for a finetuned embedding model by job ID."""
         pass
 
     @overload
     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,
@@ -1998,16 +2077,16 @@ 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."""
+    ) -> EmbeddingEvaluationResponse:
+        """Get evaluation results for a pretrained embedding model by job ID."""
         pass
 
     @overload
     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,
@@ -2015,8 +2094,8 @@ class OrcaClient(Client):
         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
@@ -2141,7 +2220,7 @@ class OrcaClient(Client):
     @overload
     def GET(
         self,
-        path: Literal["/predictive_model"],
+        path: Literal["/classification_model"],
         *,
         params: None = None,
         parse_as: Literal["json"] = "json",
@@ -2151,13 +2230,13 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[ClassificationModelMetadata | RegressionModelMetadata]:
+    ) -> list[ClassificationModelMetadata]:
         pass
 
     @overload
     def GET(
         self,
-        path: Literal["/classification_model"],
+        path: Literal["/regression_model"],
         *,
         params: None = None,
         parse_as: Literal["json"] = "json",
@@ -2167,7 +2246,7 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[ClassificationModelMetadata]:
+    ) -> list[RegressionModelMetadata]:
         pass
 
     @overload
@@ -2189,9 +2268,9 @@ class OrcaClient(Client):
     @overload
     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,
@@ -2199,15 +2278,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[EvaluationResponseClassificationMetrics]:
+    ) -> RegressionModelMetadata:
         pass
 
     @overload
     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,
@@ -2215,15 +2294,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> EvaluationResponseClassificationMetrics:
+    ) -> list[ClassificationModelMetadata | RegressionModelMetadata]:
         pass
 
     @overload
     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,
@@ -2231,15 +2310,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[RegressionModelMetadata]:
+    ) -> list[EvaluationResponseClassificationMetrics]:
         pass
 
     @overload
     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,
@@ -2247,15 +2326,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> RegressionModelMetadata:
+    ) -> list[EvaluationResponseRegressionMetrics]:
         pass
 
     @overload
     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,
@@ -2263,15 +2342,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[EvaluationResponseRegressionMetrics]:
+    ) -> EvaluationResponseClassificationMetrics:
         pass
 
     @overload
     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,
@@ -2285,9 +2364,9 @@ class OrcaClient(Client):
     @overload
     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,
@@ -2295,15 +2374,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> Task:
+    ) -> Job:
         pass
 
     @overload
     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,
@@ -2311,15 +2390,15 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> TaskStatusInfo:
+    ) -> JobStatusInfo:
         pass
 
     @overload
     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,
@@ -2327,7 +2406,7 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> PaginatedTask:
+    ) -> PaginatedJob:
         pass
 
     @overload
@@ -2478,9 +2557,9 @@ class OrcaClient(Client):
     @overload
     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,
@@ -2489,7 +2568,7 @@ class OrcaClient(Client):
         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
 
     def GET(
@@ -2659,9 +2738,9 @@ class OrcaClient(Client):
     @overload
     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,
@@ -2675,9 +2754,9 @@ class OrcaClient(Client):
     @overload
     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,
@@ -2691,9 +2770,9 @@ class OrcaClient(Client):
     @overload
     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,
@@ -2707,9 +2786,9 @@ class OrcaClient(Client):
     @overload
     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,
@@ -2870,6 +2949,26 @@ class OrcaClient(Client):
     ) -> None:
         pass
 
+    @overload
+    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
     def POST(
         self,
@@ -2930,6 +3029,26 @@ class OrcaClient(Client):
     ) -> None:
         pass
 
+    @overload
+    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
     def POST(
         self,
@@ -2991,6 +3110,48 @@ class OrcaClient(Client):
         """Create a finetuned embedding model."""
         pass
 
+    @overload
+    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
+    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
     def POST(
         self,
@@ -3090,10 +3251,10 @@ class OrcaClient(Client):
     @overload
     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,
@@ -3104,16 +3265,17 @@ class OrcaClient(Client):
         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
     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,
@@ -3124,16 +3286,17 @@ class OrcaClient(Client):
         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
     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,
@@ -3144,16 +3307,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> RegressionModelMetadata:
+    ) -> ClassificationModelMetadata:
         pass
 
     @overload
     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,
@@ -3164,16 +3327,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> EvaluationResponse:
+    ) -> RegressionModelMetadata:
         pass
 
     @overload
     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,
@@ -3184,17 +3347,16 @@ class OrcaClient(Client):
         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
     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,
@@ -3205,17 +3367,16 @@ class OrcaClient(Client):
         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
     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,
@@ -3226,21 +3387,16 @@ class OrcaClient(Client):
         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
     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,
@@ -3251,30 +3407,16 @@ class OrcaClient(Client):
         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
     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,
@@ -3285,16 +3427,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[list[LabeledMemoryLookup | ScoredMemoryLookup]]:
+    ) -> EvaluationResponse:
         pass
 
     @overload
     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,
@@ -3305,16 +3447,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> list[str]:
+    ) -> EvaluationResponse:
         pass
 
     @overload
     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,
@@ -3325,16 +3467,17 @@ class OrcaClient(Client):
         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
     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,
@@ -3345,16 +3488,17 @@ class OrcaClient(Client):
         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
     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,
@@ -3365,17 +3509,21 @@ class OrcaClient(Client):
         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
     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,
@@ -3386,8 +3534,21 @@ class OrcaClient(Client):
         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
 
     def POST(
@@ -3533,10 +3694,10 @@ class OrcaClient(Client):
     @overload
     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,
@@ -3547,16 +3708,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> ClassificationModelMetadata:
+    ) -> LabeledMemory | ScoredMemory:
         pass
 
     @overload
     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,
@@ -3567,16 +3728,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> RegressionModelMetadata:
+    ) -> list[LabeledMemory] | list[ScoredMemory]:
         pass
 
     @overload
     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,
@@ -3587,17 +3748,16 @@ class OrcaClient(Client):
         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
     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,
@@ -3608,16 +3768,16 @@ class OrcaClient(Client):
         follow_redirects: bool | UseClientDefault = USE_CLIENT_DEFAULT,
         timeout: TimeoutTypes | UseClientDefault = USE_CLIENT_DEFAULT,
         extensions: RequestExtensions | None = None,
-    ) -> LabeledMemory | ScoredMemory:
+    ) -> RegressionModelMetadata:
         pass
 
     @overload
     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,
@@ -3628,7 +3788,8 @@ class OrcaClient(Client):
         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
 
     def PATCH(