@toloka-platform/sdk 0.1.0 → 0.2.0-preview.1

package/src/client.ts

@@ -8,8 +8,10 @@ import type {
   CopyPipelineBody,
   CreateDatasetBody,
   CreateDatasetExportBody,
+  CreateTrainingRunBody,
   CreateProjectModelBody,
   CreateDatasetFieldBodyAnyOf,
+  CreateProjectTrainingBody,
   CreateDatasetUploadBody,
   CreatePipelineCheckBody,
   CreateProjectBody,
@@ -18,9 +20,11 @@ import type {
   DatasetField as GeneratedDatasetField,
   DatasetItem as GeneratedDatasetItem,
   ExportJob as GeneratedExportJob,
+  EstimateTrainingCostBody,
   EstimatePipelineCostBody,
   ImportPipelineBody,
   InferredField as GeneratedInferredField,
+  GetTrainingRunLogsParams,
   ListDatasetFieldsParams,
   ListModelVersionsParams,
   ListDatasetItemsParams,
@@ -30,10 +34,20 @@ import type {
   ListProjectPipelinesParams,
   ListProjectModelsParams,
   ListProjectsParams,
+  ListTrainingRunsParams,
+  ListProjectTrainingsParams,
   Pipeline as GeneratedPipeline,
   Model as GeneratedModel,
   ModelVersion as GeneratedModelVersion,
+  PrepareTrainingCostEstimateInputsBody,
   PatchPipelineNodeBody,
+  Training as GeneratedTraining,
+  TrainingCostEstimate as GeneratedTrainingCostEstimate,
+  TrainingCostEstimateInputSet as GeneratedTrainingCostEstimateInputSet,
+  TrainingRun as GeneratedTrainingRun,
+  TrainingRunLogs as GeneratedTrainingRunLogs,
+  ReplaceTrainingConfigurationBody,
+  UpdateTrainingMetadataBody,
   PipelineGraphSnapshot,
   Project as GeneratedProject,
   ProjectFile as GeneratedProjectFile,
@@ -66,6 +80,9 @@ export type DatasetItemId = Branded<string, "DatasetItemId">;
 export type ExportId = Branded<string, "ExportId">;
 export type ModelId = Branded<string, "ModelId">;
 export type ModelVersionId = Branded<string, "ModelVersionId">;
+export type TrainingId = Branded<string, "TrainingId">;
+export type TrainingCostEstimateInputSetId = Branded<string, "TrainingCostEstimateInputSetId">;
+export type TrainingRunId = Branded<string, "TrainingRunId">;
 export type PipelineId = Branded<string, "PipelineId">;
 export type ProjectFileId = Branded<string, "ProjectFileId">;
 export type ProjectId = Branded<string, "ProjectId">;
@@ -87,6 +104,9 @@ type ResourceIdsByProperty = {
   fileId: ProjectFileId;
   modelId: ModelId;
   modelVersionId: ModelVersionId;
+  trainingId: TrainingId;
+  inputSetId: TrainingCostEstimateInputSetId;
+  trainingRunId: TrainingRunId;
   goldenDatasetId: DatasetId;
   itemId: RunItemId;
   pipelineId: PipelineId;
@@ -126,6 +146,8 @@ type PublicInput<T> = T extends readonly unknown[]
       }
     : T;
 
+type BrandResourceUnion<T> = T extends object ? BrandResourceReferences<T> : T;
+
 export type CheckJob = BrandedResource<GeneratedCheckJob, CheckRunId>;
 export type Dataset = BrandedResource<GeneratedDataset, DatasetId>;
 export type DatasetField = BrandedResource<GeneratedDatasetField, DatasetFieldId>;
@@ -134,6 +156,14 @@ export type ExportJob = BrandedResource<GeneratedExportJob, ExportId>;
 export type InferredField = GeneratedInferredField;
 export type Model = BrandedResource<GeneratedModel, ModelId>;
 export type ModelVersion = BrandedResource<GeneratedModelVersion, ModelVersionId>;
+export type Training = BrandedResource<GeneratedTraining, TrainingId>;
+export type TrainingCostEstimateInputSet = BrandedResource<
+  GeneratedTrainingCostEstimateInputSet,
+  TrainingCostEstimateInputSetId
+>;
+export type TrainingCostEstimate = BrandResourceUnion<GeneratedTrainingCostEstimate>;
+export type TrainingRun = BrandedResource<GeneratedTrainingRun, TrainingRunId>;
+export type TrainingRunLogs = GeneratedTrainingRunLogs;
 export type BaseModel = GeneratedBaseModel;
 export type BaseModelId = GeneratedBaseModelId;
 export type Pipeline = BrandedResource<GeneratedPipeline, PipelineId>;
@@ -229,7 +259,7 @@ const createClientImplementation = (config: ClientConfig) => {
       case "datasetExport":
         return pollUntilTerminal(
           (signal) => wire.getDatasetExport(target.exportId, { ...requestOptions, signal }),
-          (job) => job.status === "succeeded",
+          (job) => job.status === "succeeded" || job.status === "failed",
           options,
         );
       case "pipelineCheck":
@@ -360,6 +390,8 @@ const createClientImplementation = (config: ClientConfig) => {
       wire.createProjectModel(projectId, body, requestOptions),
     getVersion: ({ modelVersionId }: { modelVersionId: ModelVersionId }) =>
       wire.getModelVersion(modelVersionId, requestOptions),
+    getVersionContent: ({ modelVersionId }: { modelVersionId: ModelVersionId }) =>
+      wire.getModelVersionContent(modelVersionId, requestOptions),
     archive: ({ projectId, modelId }: { projectId: ProjectId; modelId: ModelId }) => {
       const headers = new Headers(requestOptions.headers);
       headers.set("Content-Type", "application/json");
@@ -367,6 +399,91 @@ const createClientImplementation = (config: ClientConfig) => {
     },
   };
 
+  const trainings = {
+    listPage: ({ projectId, ...params }: { projectId: ProjectId } & ListProjectTrainingsParams) =>
+      wire.listProjectTrainings(projectId, params, requestOptions),
+    listAll: ({ projectId, ...params }: { projectId: ProjectId } & ListProjectTrainingsParams) =>
+      listAllPages(async (pageOptions) => {
+        const page = await wire.listProjectTrainings(
+          projectId,
+          { ...params, ...pageOptions },
+          requestOptions,
+        );
+        return { items: page.trainings, pagination: page.pagination };
+      }, params),
+    get: ({ trainingId }: { trainingId: TrainingId }) =>
+      wire.getTraining(trainingId, requestOptions),
+    create: ({ projectId, ...body }: { projectId: ProjectId } & CreateProjectTrainingBody) =>
+      wire.createProjectTraining(projectId, body, requestOptions),
+    updateMetadata: ({
+      trainingId,
+      ...body
+    }: { trainingId: TrainingId } & UpdateTrainingMetadataBody) =>
+      wire.updateTrainingMetadata(trainingId, body, requestOptions),
+    replaceConfiguration: ({
+      trainingId,
+      ...body
+    }: { trainingId: TrainingId } & ReplaceTrainingConfigurationBody) =>
+      wire.replaceTrainingConfiguration(trainingId, body, requestOptions),
+    prepareCostEstimateInputs: ({
+      trainingId,
+      idempotencyKey,
+      ...body
+    }: {
+      trainingId: TrainingId;
+      idempotencyKey: string;
+    } & PrepareTrainingCostEstimateInputsBody) => {
+      const headers = new Headers(requestOptions.headers);
+      headers.set("Idempotency-Key", idempotencyKey);
+      return wire.prepareTrainingCostEstimateInputs(trainingId, body, {
+        ...requestOptions,
+        headers: Object.fromEntries(headers.entries()),
+      });
+    },
+    estimateCost: ({
+      trainingId,
+      ...body
+    }: { trainingId: TrainingId } & EstimateTrainingCostBody) =>
+      wire.estimateTrainingCost(trainingId, body, requestOptions),
+    resumeRun: ({ trainingRunId }: { trainingRunId: TrainingRunId }) =>
+      wire.resumeTrainingRun(trainingRunId, requestOptions),
+    cancelRun: ({ trainingRunId }: { trainingRunId: TrainingRunId }) =>
+      wire.cancelTrainingRun(trainingRunId, requestOptions),
+    createRun: ({
+      trainingId,
+      idempotencyKey,
+      ...body
+    }: { trainingId: TrainingId; idempotencyKey: string } & CreateTrainingRunBody) => {
+      const headers = new Headers(requestOptions.headers);
+      headers.set("Idempotency-Key", idempotencyKey);
+      return wire.createTrainingRun(trainingId, body, {
+        ...requestOptions,
+        headers: Object.fromEntries(headers.entries()),
+      });
+    },
+    listRunsPage: ({
+      trainingId,
+      ...params
+    }: { trainingId: TrainingId } & ListTrainingRunsParams) =>
+      wire.listTrainingRuns(trainingId, params, requestOptions),
+    listAllRuns: ({ trainingId, ...params }: { trainingId: TrainingId } & ListTrainingRunsParams) =>
+      listAllPages(async (pageOptions) => {
+        const page = await wire.listTrainingRuns(
+          trainingId,
+          { ...params, ...pageOptions },
+          requestOptions,
+        );
+        return { items: page.trainingRuns, pagination: page.pagination };
+      }, params),
+    getRun: ({ trainingRunId }: { trainingRunId: TrainingRunId }) =>
+      wire.getTrainingRun(trainingRunId, requestOptions),
+    getRunLogs: ({
+      trainingRunId,
+      ...params
+    }: { trainingRunId: TrainingRunId } & GetTrainingRunLogsParams) =>
+      wire.getTrainingRunLogs(trainingRunId, params, requestOptions),
+  };
+
   const datasets = {
     create: ({ projectId, ...body }: { projectId: ProjectId } & CreateDatasetBody) =>
       wire.createDataset(projectId, body, requestOptions),
@@ -514,6 +631,7 @@ const createClientImplementation = (config: ClientConfig) => {
     datasets,
     baseModels,
     models,
+    trainings,
     runs,
     secrets,
     files,
@@ -532,35 +650,51 @@ type PublicOutput<T> = T extends GeneratedCheckJob
         ? UploadJob
         : T extends GeneratedModel
           ? Model
-          : T extends GeneratedModelVersion
-            ? ModelVersion
-            : T extends GeneratedProject
-              ? Project
-              : T extends GeneratedPipeline
-                ? Pipeline
-                : T extends GeneratedDataset
-                  ? Dataset
-                  : T extends GeneratedDatasetField
-                    ? DatasetField
-                    : T extends GeneratedDatasetItem
-                      ? DatasetItem
-                      : T extends GeneratedProjectFile
-                        ? ProjectFile
-                        : T extends GeneratedRunItem
-                          ? RunItem
-                          : T extends GeneratedRunSummary
-                            ? RunSummary
-                            : T extends Blob
-                              ? Blob
-                              : T extends Promise<infer Value>
-                                ? Promise<PublicOutput<Value>>
-                                : T extends AsyncGenerator<infer Value, infer Return, infer Next>
-                                  ? AsyncGenerator<PublicOutput<Value>, PublicOutput<Return>, Next>
-                                  : T extends readonly unknown[]
-                                    ? { [K in keyof T]: PublicOutput<T[K]> }
-                                    : T extends object
-                                      ? { [K in keyof T]: PublicOutput<T[K]> }
-                                      : T;
+          : T extends GeneratedTraining
+            ? Training
+            : T extends GeneratedModelVersion
+              ? ModelVersion
+              : T extends GeneratedTrainingCostEstimateInputSet
+                ? TrainingCostEstimateInputSet
+                : T extends GeneratedTrainingCostEstimate
+                  ? TrainingCostEstimate
+                  : T extends GeneratedTrainingRun
+                    ? TrainingRun
+                    : T extends GeneratedProject
+                      ? Project
+                      : T extends GeneratedPipeline
+                        ? Pipeline
+                        : T extends GeneratedDataset
+                          ? Dataset
+                          : T extends GeneratedDatasetField
+                            ? DatasetField
+                            : T extends GeneratedDatasetItem
+                              ? DatasetItem
+                              : T extends GeneratedProjectFile
+                                ? ProjectFile
+                                : T extends GeneratedRunItem
+                                  ? RunItem
+                                  : T extends GeneratedRunSummary
+                                    ? RunSummary
+                                    : T extends Blob
+                                      ? Blob
+                                      : T extends Promise<infer Value>
+                                        ? Promise<PublicOutput<Value>>
+                                        : T extends AsyncGenerator<
+                                              infer Value,
+                                              infer Return,
+                                              infer Next
+                                            >
+                                          ? AsyncGenerator<
+                                              PublicOutput<Value>,
+                                              PublicOutput<Return>,
+                                              Next
+                                            >
+                                          : T extends readonly unknown[]
+                                            ? { [K in keyof T]: PublicOutput<T[K]> }
+                                            : T extends object
+                                              ? { [K in keyof T]: PublicOutput<T[K]> }
+                                              : T;
 
 type PublicClientMember<T> = T extends (...args: infer Args) => infer Result
   ? (...args: { [K in keyof Args]: PublicInput<Args[K]> }) => PublicOutput<Result>

package/src/index.ts

@@ -30,6 +30,8 @@ export type {
   RunSummary,
   TerminalJob,
   UploadId,
+  Training,
+  TrainingId,
   UploadJob,
   WaitForTarget,
 } from "./client";

package/openapi/gen/models/createDatasetExportBodyOneOf.ts

@@ -1,6 +0,0 @@
-import type { CreateDatasetExportBodyOneOfFormat } from './createDatasetExportBodyOneOfFormat';
-
-export type CreateDatasetExportBodyOneOf = {
-  format?: CreateDatasetExportBodyOneOfFormat;
-  fields: string[];
-};

package/openapi/gen/models/createDatasetExportBodyOneOfFormat.ts

@@ -1,2 +0,0 @@
-
-export type CreateDatasetExportBodyOneOfFormat = 'json-array';

package/openapi/gen/models/createDatasetExportBodyOneOfThree.ts

@@ -1,10 +0,0 @@
-import type { CreateDatasetExportBodyOneOfThreeFormat } from './createDatasetExportBodyOneOfThreeFormat';
-import type { CreateDatasetExportBodyOneOfThreeContract } from './createDatasetExportBodyOneOfThreeContract';
-import type { CreateDatasetExportBodyOneOfThreeItemFilter } from './createDatasetExportBodyOneOfThreeItemFilter';
-
-export type CreateDatasetExportBodyOneOfThree = {
-  format: CreateDatasetExportBodyOneOfThreeFormat;
-  contract: CreateDatasetExportBodyOneOfThreeContract;
-  itemFilter?: CreateDatasetExportBodyOneOfThreeItemFilter;
-  validateOnly?: boolean;
-};

package/openapi/gen/models/createDatasetExportBodyOneOfThreeContract.ts

@@ -1,2 +0,0 @@
-
-export type CreateDatasetExportBodyOneOfThreeContract = 'openai-chat-v1';

package/openapi/gen/models/createDatasetExportBodyOneOfThreeFormat.ts

@@ -1,2 +0,0 @@
-
-export type CreateDatasetExportBodyOneOfThreeFormat = 'training-jsonl';

package/openapi/gen/models/createDatasetExportBodyOneOfThreeItemFilter.ts

@@ -1,9 +0,0 @@
-import type { CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItem } from './createDatasetExportBodyOneOfThreeItemFilterFieldValuesItem';
-import type { CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItem } from './createDatasetExportBodyOneOfThreeItemFilterFieldSourcesItem';
-
-export type CreateDatasetExportBodyOneOfThreeItemFilter = {
-  fieldValues?: CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItem[];
-  fieldSources?: CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItem[];
-  missingRevisionFields?: string[];
-  createdBefore?: string;
-};

package/openapi/gen/models/createDatasetExportBodyOneOfThreeItemFilterFieldSourcesItem.ts

@@ -1,6 +0,0 @@
-import type { CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItem } from './createDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItem';
-
-export type CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItem = {
-  fieldId: string;
-  sources: CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItem[];
-};

package/openapi/gen/models/createDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItem.ts

@@ -1,4 +0,0 @@
-import type { CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItemOneOf } from './createDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItemOneOf';
-import type { CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItemOneOfTwo } from './createDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItemOneOfTwo';
-
-export type CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItem = CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItemOneOf | CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItemOneOfTwo;

package/openapi/gen/models/createDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItemOneOf.ts

@@ -1,5 +0,0 @@
-
-export type CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItemOneOf = {
-  type: 'user';
-  userId: string;
-};

package/openapi/gen/models/createDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItemOneOfTwo.ts

@@ -1,5 +0,0 @@
-
-export type CreateDatasetExportBodyOneOfThreeItemFilterFieldSourcesItemSourcesItemOneOfTwo = {
-  type: 'system';
-  source: string;
-};

package/openapi/gen/models/createDatasetExportBodyOneOfThreeItemFilterFieldValuesItem.ts

@@ -1,12 +0,0 @@
-import type { CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemOperator } from './createDatasetExportBodyOneOfThreeItemFilterFieldValuesItemOperator';
-import type { CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemValue } from './createDatasetExportBodyOneOfThreeItemFilterFieldValuesItemValue';
-import type { CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItem } from './createDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItem';
-
-export type CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItem = {
-  jsonPath: string;
-  operator: CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemOperator;
-  value: CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemValue;
-  fieldId: string;
-  latestOnly?: boolean;
-  sources?: CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItem[];
-};

package/openapi/gen/models/createDatasetExportBodyOneOfThreeItemFilterFieldValuesItemOperator.ts

@@ -1,2 +0,0 @@
-
-export type CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemOperator = 'text_eq' | 'text_neq';

package/openapi/gen/models/createDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItem.ts

@@ -1,4 +0,0 @@
-import type { CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItemOneOf } from './createDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItemOneOf';
-import type { CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItemOneOfTwo } from './createDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItemOneOfTwo';
-
-export type CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItem = CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItemOneOf | CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItemOneOfTwo;

package/openapi/gen/models/createDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItemOneOf.ts

@@ -1,5 +0,0 @@
-
-export type CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItemOneOf = {
-  type: 'user';
-  userId: string;
-};

package/openapi/gen/models/createDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItemOneOfTwo.ts

@@ -1,5 +0,0 @@
-
-export type CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemSourcesItemOneOfTwo = {
-  type: 'system';
-  source: string;
-};

package/openapi/gen/models/createDatasetExportBodyOneOfThreeItemFilterFieldValuesItemValue.ts

@@ -1,2 +0,0 @@
-
-export type CreateDatasetExportBodyOneOfThreeItemFilterFieldValuesItemValue = string | number | boolean | null;