@toloka-platform/sdk 0.11.1 → 0.11.3

package/src/cli/generation-models.ts

@@ -0,0 +1,46 @@
+import type { PotatoClient } from "../client";
+import { formatTable, type TableIo, writeOutput } from "./table";
+
+type GenerationModelApi = Pick<PotatoClient, "generationModels">;
+
+const dollars = (amount: number): string => `$${amount.toFixed(2)}`;
+
+export const listGenerationModels = async (
+  client: GenerationModelApi,
+  input: { json: boolean },
+  io: TableIo,
+): Promise<void> => {
+  const catalog = await client.generationModels.list();
+  const recommended = catalog.generationModels
+    .filter((model) => model.recommended)
+    .map((model) => model.id);
+  writeOutput(
+    io,
+    input.json,
+    catalog,
+    catalog.generationModels.length === 0
+      ? "No generation models available."
+      : [
+          formatTable([
+            ["ID", "NAME", "REASONING EFFORT", "TYPICAL", "MIN", "MAX"],
+            ...catalog.generationModels.flatMap((model) =>
+              model.budgets.map((budget) => [
+                model.id,
+                model.displayName,
+                budget.reasoningEffort === undefined
+                  ? "-"
+                  : budget.reasoningEffort === model.defaultReasoningEffort
+                    ? `${budget.reasoningEffort} (default)`
+                    : budget.reasoningEffort,
+                dollars(budget.typical),
+                dollars(budget.min),
+                dollars(budget.maxEstimate),
+              ]),
+            ),
+          ]),
+          "",
+          `Default model: ${catalog.defaultGenerationModelId}`,
+          ...(recommended.length === 0 ? [] : [`Recommended: ${recommended.join(", ")}`]),
+        ].join("\n"),
+  );
+};

package/src/cli/program.ts

@@ -5,11 +5,13 @@ import type { CliAnalytics, CliAnalyticsCommandOutcome } from "./analytics";
 import type { AutoUpdateHooks } from "./auto-update";
 import { maybeAutoUpdate } from "./auto-update";
 import type { BrowserLoginFlow } from "./browser-login";
+import type { DescribeTaskUiComponents } from "../testing/task-ui-browser";
 import { CHECK_RESULT_FILE_ENV, type checkProject } from "./check";
 import { registerAudienceCommands } from "./commands/audience";
 import { registerAuthCommands } from "./commands/auth";
 import { registerConfigCommands } from "./commands/config";
 import { registerDatasetCommands } from "./commands/dataset";
+import { registerGenerationModelCommands } from "./commands/generation-model";
 import { registerDeliveryOrderCommands } from "./commands/delivery-order";
 import { registerModelCommands } from "./commands/models";
 import { registerPipelineCommands } from "./commands/pipeline";
@@ -20,6 +22,7 @@ import { registerSkillCommands } from "./commands/skill";
 import { registerSourceCommands } from "./commands/source";
 import { registerStatusCommand } from "./commands/status";
 import { registerTrainingCommands } from "./commands/training";
+import { registerUiCommands } from "./commands/ui";
 import { registerVersionCommand } from "./commands/version";
 import type { TolokaEnvironment } from "./config";
 import {
@@ -94,6 +97,7 @@ export function createProgram(context: ProgramContext): CreatedProgram {
   registerProjectCommands(program, host);
   registerDeliveryOrderCommands(program, host);
   registerAudienceCommands(program, host);
+  registerGenerationModelCommands(program, host);
   registerSourceCommands(program, host);
   registerSecretCommands(program, host);
   registerModelCommands(program, host);
@@ -103,6 +107,7 @@ export function createProgram(context: ProgramContext): CreatedProgram {
   registerStatusCommand(program, host);
   registerReportCommand(program, host);
   registerSkillCommands(program, host);
+  registerUiCommands(program, host);
 
   return { program, exitCode };
 }
@@ -198,6 +203,7 @@ export async function runTolokaCli(
   runChecks?: typeof checkProject,
   autoUpdate?: AutoUpdateHooks,
   browserLogin?: BrowserLoginFlow,
+  taskUiComponents?: DescribeTaskUiComponents,
 ): Promise<number> {
   const { environment, warnings } = normalizeEnvironment(rawEnvironment);
   for (const warning of warnings) io.stderr(warning);
@@ -220,6 +226,7 @@ export async function runTolokaCli(
     ...(injectedClient === undefined ? {} : { injectedClient }),
     ...(browserLogin === undefined ? {} : { browserLogin }),
     ...(runChecks === undefined ? {} : { runChecks }),
+    ...(taskUiComponents === undefined ? {} : { taskUiComponents }),
   };
   const { program, exitCode } = createProgram(baseContext);
   const startedAt = Date.now();

package/src/cli/project-context.ts

@@ -50,7 +50,7 @@ export const NO_PROJECT_SELECTED_MESSAGE =
   "No project selected. Run toloka project create for a new project, toloka project add for an existing project, or cd into a checkout.";
 
 const NETWORK_MISMATCH_MESSAGE =
-  "project add writes the machine registry; unset --api-key / --base-url / TOLOKA_API_KEY / TOLOKA_BASE_URL, or run login against that platform first.";
+  "project add writes the machine registry; unset --api-key / --base-url / TOLOKA_SCOPED_TOKEN / TOLOKA_API_KEY / TOLOKA_BASE_URL, or run login against that platform first.";
 
 const ALIAS_PATTERN = /^[^/\s]+$/u;
 
@@ -187,7 +187,10 @@ const assertStoredNetworkMatches = (
   environment: TolokaEnvironment,
   stored: StoredCredentials,
 ) => {
-  const flagOrEnvKey = firstNonEmpty(flags.apiKey) ?? firstNonEmpty(environment.TOLOKA_API_KEY);
+  const flagOrEnvKey =
+    firstNonEmpty(flags.apiKey) ??
+    firstNonEmpty(environment.TOLOKA_SCOPED_TOKEN) ??
+    firstNonEmpty(environment.TOLOKA_API_KEY);
   const flagOrEnvBase =
     firstNonEmpty(flags.baseUrl) ??
     firstNonEmpty(environment.TOLOKA_BASE_URL) ??