@toloka-platform/sdk 0.12.0 → 0.12.2

package/src/client.ts

@@ -8,6 +8,7 @@ import type {
   CopyPipelineBody,
   CreateDatasetBody,
   CreateDatasetExportBody,
+  CreateDatasetFieldsFromUploadedRowsBody,
   CreateDatasetSliceBody,
   CreateTrainingRunBody,
   CreateProjectModelBody,
@@ -242,13 +243,20 @@ export type DatasetSplitWaitTarget = {
   readonly column: string;
 };
 
+export type DatasetFieldsWaitTarget = {
+  readonly type: "datasetFields";
+  readonly datasetId: ResourceIdInput<DatasetId>;
+  readonly names?: readonly string[];
+};
+
 export type WaitForTarget =
   | DatasetUploadWaitTarget
   | DatasetExportWaitTarget
   | PipelineCheckWaitTarget
   | PipelineRunWaitTarget
   | DatasetSliceWaitTarget
-  | DatasetSplitWaitTarget;
+  | DatasetSplitWaitTarget
+  | DatasetFieldsWaitTarget;
 
 const createClientImplementation = (config: ClientConfig) => {
   const requestOptions = createRequestOptions(config.baseUrl, config, config.fetch);
@@ -271,6 +279,7 @@ const createClientImplementation = (config: ClientConfig) => {
   ): Promise<TerminalJob<RunJob>>;
   function waitFor(target: DatasetSliceWaitTarget, options?: WaitForOptions): Promise<DatasetSlice>;
   function waitFor(target: DatasetSplitWaitTarget, options?: WaitForOptions): Promise<DatasetField>;
+  function waitFor(target: DatasetFieldsWaitTarget, options?: WaitForOptions): Promise<DatasetField[]>;
   async function waitFor(
     target: WaitForTarget,
     options?: WaitForOptions,
@@ -281,6 +290,7 @@ const createClientImplementation = (config: ClientConfig) => {
     | GeneratedRunJob
     | GeneratedDatasetSlice
     | GeneratedDatasetField
+    | GeneratedDatasetField[]
   > {
     switch (target.type) {
       case "datasetUpload":
@@ -328,6 +338,28 @@ const createClientImplementation = (config: ClientConfig) => {
           (field) => field.split?.status !== "filling",
           options,
         );
+      case "datasetFields":
+        return pollUntilTerminal(
+          async (signal) => {
+            const fields: GeneratedDatasetField[] = [];
+            const pages = listAllPages(async (pageOptions) => {
+              const page = await wire.listDatasetFields(target.datasetId, pageOptions, {
+                ...requestOptions,
+                signal,
+              });
+              return { items: page.fields, pagination: page.pagination };
+            });
+            for await (const field of pages) fields.push(field);
+            return fields;
+          },
+          (fields) =>
+            !fields.some(
+              (field) =>
+                field.operations.some(({ blocking }) => blocking) &&
+                (target.names === undefined || target.names.includes(field.name)),
+            ),
+          options,
+        );
     }
   }
 
@@ -602,6 +634,11 @@ const createClientImplementation = (config: ClientConfig) => {
       wire.listDatasetFields(datasetId, params, requestOptions),
     getInferredFields: ({ datasetId }: { datasetId: DatasetId }) =>
       wire.getDatasetInferredFields(datasetId, requestOptions),
+    createFieldsFromUploadedRows: async ({
+      datasetId,
+      ...body
+    }: { datasetId: DatasetId } & CreateDatasetFieldsFromUploadedRowsBody) =>
+      (await wire.createDatasetFieldsFromUploadedRows(datasetId, body, requestOptions)).fields,
     listAllFields: ({ datasetId, ...params }: { datasetId: DatasetId } & ListDatasetFieldsParams) =>
       listAllPages(async (pageOptions) => {
         const page = await wire.listDatasetFields(

package/src/index.ts

@@ -5,6 +5,7 @@ export type {
   CheckRunId,
   ClientConfig,
   Dataset,
+  DatasetFieldsWaitTarget,
   DatasetExportWaitTarget,
   DatasetField,
   DatasetFieldId,