@hasna/brains 0.0.13 → 0.0.14

package/dist/cli/index.js

@@ -13434,7 +13434,7 @@ __export(exports_sessions2, {
   gatherFromSessions: () => gatherFromSessions
 });
 import { readdir, readFile, stat } from "fs/promises";
-import { existsSync as existsSync9 } from "fs";
+import { existsSync as existsSync10 } from "fs";
 import { join as join12 } from "path";
 import { homedir as homedir11 } from "os";
 function extractText(content) {
@@ -13447,7 +13447,7 @@ async function gatherFromSessions(options = {}) {
   const { limit: limit2 = 1000 } = options;
   const examples = [];
   const claudeDir = join12(homedir11(), ".claude", "projects");
-  if (!existsSync9(claudeDir)) {
+  if (!existsSync10(claudeDir)) {
     return { source: "sessions", examples: [], count: 0 };
   }
   const projectDirs = await readdir(claudeDir).catch(() => []);
@@ -14976,10 +14976,8 @@ function mapRelationalRow(tablesConfig, tableConfig, row, buildQueryResultSelect
 }
 
 // src/cli/index.ts
-import { randomUUID } from "crypto";
-import { readFileSync as readFileSync6, existsSync as existsSync10, mkdirSync as mkdirSync6, writeFileSync as writeFileSync5 } from "fs";
-import { join as join13 } from "path";
-import { homedir as homedir12 } from "os";
+import { readFileSync as readFileSync7 } from "fs";
+import { join as join14 } from "path";
 
 // node_modules/drizzle-orm/bun-sqlite/driver.js
 import { Database } from "bun:sqlite";
@@ -17402,6 +17400,57 @@ function getRawDb(dbPath) {
   return dbPath ? new SqliteAdapter(dbPath) : createDatabase({ service: "brains" });
 }
 
+// src/cli/ui.ts
+import chalk from "chalk";
+function printTable(headers, rows) {
+  if (rows.length === 0) {
+    console.log(chalk.dim("  (no records)"));
+    return;
+  }
+  const colWidths = headers.map((h, i) => Math.max(h.length, ...rows.map((r) => (r[i] ?? "").length)));
+  const separator = colWidths.map((w) => "\u2500".repeat(w + 2)).join("\u253C");
+  const headerLine = headers.map((h, i) => ` ${chalk.bold(h.padEnd(colWidths[i] ?? 0))} `).join("\u2502");
+  const topBorder = colWidths.map((w) => "\u2500".repeat(w + 2)).join("\u252C");
+  const midBorder = separator;
+  const bottomBorder = colWidths.map((w) => "\u2500".repeat(w + 2)).join("\u2534");
+  console.log("\u250C" + topBorder + "\u2510");
+  console.log("\u2502" + headerLine + "\u2502");
+  console.log("\u251C" + midBorder + "\u2524");
+  for (const row of rows) {
+    const rowLine = row.map((cell, i) => ` ${(cell ?? "").padEnd(colWidths[i] ?? 0)} `).join("\u2502");
+    console.log("\u2502" + rowLine + "\u2502");
+  }
+  console.log("\u2514" + bottomBorder + "\u2518");
+}
+var STATUS_COLORS = {
+  succeeded: chalk.green,
+  running: chalk.cyan,
+  pending: chalk.yellow,
+  failed: chalk.red,
+  cancelled: chalk.dim,
+  queued: chalk.yellow,
+  validating_files: chalk.blue
+};
+function printStatus(status) {
+  const colorFn = STATUS_COLORS[status] ?? chalk.white;
+  return colorFn(`\u25CF ${status}`);
+}
+function printJson(obj) {
+  console.log(JSON.stringify(obj, null, 2));
+}
+function printError(message) {
+  console.error(chalk.red("\u2717 Error: ") + message);
+}
+function printSuccess(message) {
+  console.log(chalk.green("\u2713 ") + message);
+}
+function printInfo(message) {
+  console.log(chalk.dim("  " + message));
+}
+
+// src/cli/commands/models.ts
+import { randomUUID } from "crypto";
+
 // node_modules/openai/internal/qs/formats.mjs
 var default_format = "RFC3986";
 var formatters = {
@@ -23149,345 +23198,137 @@ class ThinkerLabsProvider {
   }
 }
 
-// src/cli/ui.ts
-import chalk from "chalk";
-function printTable(headers, rows) {
-  if (rows.length === 0) {
-    console.log(chalk.dim("  (no records)"));
-    return;
-  }
-  const colWidths = headers.map((h, i) => Math.max(h.length, ...rows.map((r) => (r[i] ?? "").length)));
-  const separator = colWidths.map((w) => "\u2500".repeat(w + 2)).join("\u253C");
-  const headerLine = headers.map((h, i) => ` ${chalk.bold(h.padEnd(colWidths[i] ?? 0))} `).join("\u2502");
-  const topBorder = colWidths.map((w) => "\u2500".repeat(w + 2)).join("\u252C");
-  const midBorder = separator;
-  const bottomBorder = colWidths.map((w) => "\u2500".repeat(w + 2)).join("\u2534");
-  console.log("\u250C" + topBorder + "\u2510");
-  console.log("\u2502" + headerLine + "\u2502");
-  console.log("\u251C" + midBorder + "\u2524");
-  for (const row of rows) {
-    const rowLine = row.map((cell, i) => ` ${(cell ?? "").padEnd(colWidths[i] ?? 0)} `).join("\u2502");
-    console.log("\u2502" + rowLine + "\u2502");
-  }
-  console.log("\u2514" + bottomBorder + "\u2518");
-}
-var STATUS_COLORS = {
-  succeeded: chalk.green,
-  running: chalk.cyan,
-  pending: chalk.yellow,
-  failed: chalk.red,
-  cancelled: chalk.dim,
-  queued: chalk.yellow,
-  validating_files: chalk.blue
-};
-function printStatus(status) {
-  const colorFn = STATUS_COLORS[status] ?? chalk.white;
-  return colorFn(`\u25CF ${status}`);
-}
-function printJson(obj) {
-  console.log(JSON.stringify(obj, null, 2));
-}
-function printError(message) {
-  console.error(chalk.red("\u2717 Error: ") + message);
-}
-function printSuccess(message) {
-  console.log(chalk.green("\u2713 ") + message);
-}
-function printInfo(message) {
-  console.log(chalk.dim("  " + message));
-}
-
-// src/cli/index.ts
-var program2 = new Command;
-program2.name("brains").description("Fine-tuned model tracker and trainer").version("0.0.1");
-var modelsCmd = program2.command("models").description("Manage tracked fine-tuned models");
-modelsCmd.command("list").description("List all tracked fine-tuned models").option("--json", "Output as JSON").action(async (opts) => {
-  try {
-    const db = getDb();
-    const models = await db.select().from(fineTunedModels);
-    if (opts.json) {
-      printJson(models);
-      return;
-    }
-    if (models.length === 0) {
-      printInfo("No models tracked yet. Use 'brains finetune start' to train one.");
-      return;
-    }
-    printTable(["ID", "Display Name", "Provider", "Status", "Collection", "Base Model"], models.map((m) => [
-      m.id,
-      m.displayName ?? m.name,
-      m.provider,
-      printStatus(m.status),
-      m.collection ?? "",
-      m.baseModel
-    ]));
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-modelsCmd.command("show <id>").description("Show details of a specific model").option("--json", "Output as JSON").action(async (id, opts) => {
-  try {
-    const db = getDb();
-    const [model] = await db.select().from(fineTunedModels).where(eq(fineTunedModels.id, id));
-    if (!model) {
+// src/cli/commands/models.ts
+function registerModelsCommands(program2) {
+  const modelsCmd = program2.command("models").description("Manage tracked fine-tuned models");
+  modelsCmd.command("list").description("List all tracked fine-tuned models").option("--json", "Output as JSON").action(async (opts) => {
+    try {
+      const db = getDb();
+      const models = await db.select().from(fineTunedModels);
       if (opts.json) {
-        printJson({ error: `Model not found: ${id}` });
-      } else {
-        printError(`Model not found: ${id}`);
+        printJson(models);
+        return;
+      }
+      if (models.length === 0) {
+        printInfo("No models tracked yet. Use 'brains finetune start' to train one.");
+        return;
       }
+      printTable(["ID", "Display Name", "Provider", "Status", "Collection", "Base Model"], models.map((m) => [
+        m.id,
+        m.displayName ?? m.name,
+        m.provider,
+        printStatus(m.status),
+        m.collection ?? "",
+        m.baseModel
+      ]));
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
       process.exit(1);
     }
-    if (opts.json) {
-      printJson(model);
-      return;
-    }
-    console.log();
-    const tagsList = model.tags ? JSON.parse(model.tags).join(", ") : "(none)";
-    console.log(`  ID:            ${model.id}`);
-    console.log(`  Name:          ${model.name}`);
-    console.log(`  Display Name:  ${model.displayName ?? "(none)"}`);
-    console.log(`  Description:   ${model.description ?? "(none)"}`);
-    console.log(`  Collection:    ${model.collection ?? "(none)"}`);
-    console.log(`  Tags:          ${tagsList}`);
-    console.log(`  Provider:      ${model.provider}`);
-    console.log(`  Status:        ${printStatus(model.status)}`);
-    console.log(`  Base Model:    ${model.baseModel}`);
-    console.log(`  Job ID:        ${model.fineTuneJobId ?? "(none)"}`);
-    console.log(`  Created:       ${new Date(model.createdAt).toISOString()}`);
-    console.log(`  Updated:       ${new Date(model.updatedAt).toISOString()}`);
-    console.log();
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-modelsCmd.command("rename <id> <displayName>").description("Set the display name of a model").action(async (id, displayName) => {
-  try {
-    const db = getDb();
-    await db.update(fineTunedModels).set({ displayName, updatedAt: Date.now() }).where(eq(fineTunedModels.id, id));
-    printSuccess(`Display name set to "${displayName}"`);
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-modelsCmd.command("describe <id> <description>").description("Set the description of a model").action(async (id, description) => {
-  try {
-    const db = getDb();
-    await db.update(fineTunedModels).set({ description, updatedAt: Date.now() }).where(eq(fineTunedModels.id, id));
-    printSuccess(`Description updated.`);
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-modelsCmd.command("tag <id> <tag>").description("Add a tag to a model").action(async (id, tag) => {
-  try {
-    const db = getDb();
-    const [model] = await db.select().from(fineTunedModels).where(eq(fineTunedModels.id, id));
-    if (!model) {
-      printError(`Model not found: ${id}`);
+  });
+  modelsCmd.command("show <id>").description("Show details of a specific model").option("--json", "Output as JSON").action(async (id, opts) => {
+    try {
+      const db = getDb();
+      const [model] = await db.select().from(fineTunedModels).where(eq(fineTunedModels.id, id));
+      if (!model) {
+        if (opts.json) {
+          printJson({ error: `Model not found: ${id}` });
+        } else {
+          printError(`Model not found: ${id}`);
+        }
+        process.exit(1);
+      }
+      if (opts.json) {
+        printJson(model);
+        return;
+      }
+      console.log();
+      const tagsList = model.tags ? JSON.parse(model.tags).join(", ") : "(none)";
+      console.log(`  ID:            ${model.id}`);
+      console.log(`  Name:          ${model.name}`);
+      console.log(`  Display Name:  ${model.displayName ?? "(none)"}`);
+      console.log(`  Description:   ${model.description ?? "(none)"}`);
+      console.log(`  Collection:    ${model.collection ?? "(none)"}`);
+      console.log(`  Tags:          ${tagsList}`);
+      console.log(`  Provider:      ${model.provider}`);
+      console.log(`  Status:        ${printStatus(model.status)}`);
+      console.log(`  Base Model:    ${model.baseModel}`);
+      console.log(`  Job ID:        ${model.fineTuneJobId ?? "(none)"}`);
+      console.log(`  Created:       ${new Date(model.createdAt).toISOString()}`);
+      console.log(`  Updated:       ${new Date(model.updatedAt).toISOString()}`);
+      console.log();
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
       process.exit(1);
     }
-    const existing = model.tags ? JSON.parse(model.tags) : [];
-    if (!existing.includes(tag)) {
-      existing.push(tag);
-    }
-    await db.update(fineTunedModels).set({ tags: JSON.stringify(existing), updatedAt: Date.now() }).where(eq(fineTunedModels.id, id));
-    printSuccess(`Tag "${tag}" added. Tags: ${existing.join(", ")}`);
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-modelsCmd.command("untag <id> <tag>").description("Remove a tag from a model").action(async (id, tag) => {
-  try {
-    const db = getDb();
-    const [model] = await db.select().from(fineTunedModels).where(eq(fineTunedModels.id, id));
-    if (!model) {
-      printError(`Model not found: ${id}`);
+  });
+  modelsCmd.command("rename <id> <displayName>").description("Set the display name of a model").action(async (id, displayName) => {
+    try {
+      const db = getDb();
+      await db.update(fineTunedModels).set({ displayName, updatedAt: Date.now() }).where(eq(fineTunedModels.id, id));
+      printSuccess(`Display name set to "${displayName}"`);
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
       process.exit(1);
     }
-    const existing = model.tags ? JSON.parse(model.tags) : [];
-    const updated = existing.filter((t) => t !== tag);
-    await db.update(fineTunedModels).set({ tags: JSON.stringify(updated), updatedAt: Date.now() }).where(eq(fineTunedModels.id, id));
-    printSuccess(`Tag "${tag}" removed. Tags: ${updated.join(", ") || "(none)"}`);
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-modelsCmd.command("collection <id> <collectionName>").description("Set the collection of a model").action(async (id, collectionName) => {
-  try {
-    const db = getDb();
-    await db.update(fineTunedModels).set({ collection: collectionName, updatedAt: Date.now() }).where(eq(fineTunedModels.id, id));
-    printSuccess(`Collection set to "${collectionName}"`);
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-modelsCmd.command("import <job-id>").description("Import an externally created fine-tuned model into local tracking").option("--provider <provider>", "Provider (openai|thinker-labs)", "openai").option("--name <name>", "Display name for the model").action(async (jobId, opts) => {
-  try {
-    let result;
-    if (opts.provider === "openai") {
-      result = await getFineTuneStatus(jobId);
-    } else {
-      const tl = new ThinkerLabsProvider;
-      result = await tl.getFineTuneStatus(jobId);
-    }
-    const db = getDb();
-    const [existing] = await db.select().from(fineTunedModels).where(eq(fineTunedModels.fineTuneJobId, jobId));
-    if (existing) {
-      printInfo(`Model already tracked as: ${existing.id}`);
-      return;
-    }
-    const modelId = randomUUID();
-    const now = Date.now();
-    const name = opts.name ?? result.fineTunedModel ?? `imported-${jobId}`;
-    await db.insert(fineTunedModels).values({
-      id: modelId,
-      name,
-      provider: opts.provider,
-      baseModel: result.baseModel ?? "unknown",
-      status: result.status,
-      fineTuneJobId: jobId,
-      createdAt: now,
-      updatedAt: now
-    });
-    await db.insert(trainingJobs).values({
-      id: randomUUID(),
-      modelId,
-      provider: opts.provider,
-      status: result.status,
-      startedAt: now
-    });
-    printSuccess(`Model imported successfully.`);
-    console.log();
-    console.log(`  Local ID:  ${modelId}`);
-    console.log(`  Job ID:    ${jobId}`);
-    console.log(`  Name:      ${name}`);
-    console.log(`  Status:    ${printStatus(result.status)}`);
-    if (result.fineTunedModel)
-      console.log(`  Model:     ${result.fineTunedModel}`);
-    console.log();
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-var finetuneCmd = program2.command("finetune").description("Manage fine-tuning jobs");
-finetuneCmd.command("start").description("Start a fine-tuning job").requiredOption("--provider <provider>", "Provider to use (openai|thinker-labs)").requiredOption("--base-model <model>", "Base model to fine-tune (e.g. gpt-4o-mini-2024-07-18)").option("--dataset <path>", "Path to the JSONL training dataset (auto-detects latest if omitted)").requiredOption("--name <name>", "Human-readable name for this fine-tuned model").action(async (opts) => {
-  try {
-    if (opts.provider !== "openai" && opts.provider !== "thinker-labs") {
-      printError(`Unknown provider: ${opts.provider}. Use 'openai' or 'thinker-labs'.`);
+  });
+  modelsCmd.command("describe <id> <description>").description("Set the description of a model").action(async (id, description) => {
+    try {
+      const db = getDb();
+      await db.update(fineTunedModels).set({ description, updatedAt: Date.now() }).where(eq(fineTunedModels.id, id));
+      printSuccess(`Description updated.`);
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
       process.exit(1);
     }
-    let datasetPath = opts.dataset;
-    if (!datasetPath) {
-      const db2 = getDb();
-      const [latest] = await db2.select().from(trainingDatasets).orderBy(desc(trainingDatasets.createdAt)).limit(1);
-      if (!latest?.filePath) {
-        printError("No datasets found. Run 'brains data gather' first.");
+  });
+  modelsCmd.command("tag <id> <tag>").description("Add a tag to a model").action(async (id, tag) => {
+    try {
+      const db = getDb();
+      const [model] = await db.select().from(fineTunedModels).where(eq(fineTunedModels.id, id));
+      if (!model) {
+        printError(`Model not found: ${id}`);
         process.exit(1);
       }
-      datasetPath = latest.filePath;
-      printInfo(`Using latest dataset: ${datasetPath} (${latest.exampleCount} examples)`);
-    }
-    if (!existsSync10(datasetPath)) {
-      printError(`Dataset file not found: ${datasetPath}`);
+      const existing = model.tags ? JSON.parse(model.tags) : [];
+      if (!existing.includes(tag)) {
+        existing.push(tag);
+      }
+      await db.update(fineTunedModels).set({ tags: JSON.stringify(existing), updatedAt: Date.now() }).where(eq(fineTunedModels.id, id));
+      printSuccess(`Tag "${tag}" added. Tags: ${existing.join(", ")}`);
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
       process.exit(1);
     }
-    printInfo(`Uploading training file: ${datasetPath} \u2026`);
-    let fileId;
-    let jobId;
-    let jobStatus;
-    if (opts.provider === "openai") {
-      ({ fileId } = await uploadTrainingFile(datasetPath));
-      printSuccess(`File uploaded. fileId = ${fileId}`);
-      printInfo(`Creating fine-tune job on OpenAI \u2026`);
-      ({ jobId, status: jobStatus } = await createFineTuneJob(fileId, opts.baseModel, opts.name));
-    } else {
-      const tl = new ThinkerLabsProvider;
-      ({ fileId } = await tl.uploadTrainingFile(datasetPath));
-      printSuccess(`File uploaded. fileId = ${fileId}`);
-      printInfo(`Creating fine-tune job on Thinker Labs \u2026`);
-      ({ jobId, status: jobStatus } = await tl.createFineTuneJob(fileId, opts.baseModel, opts.name));
-    }
-    const db = getDb();
-    const modelId = randomUUID();
-    const now = Date.now();
-    await db.insert(fineTunedModels).values({
-      id: modelId,
-      name: opts.name,
-      provider: opts.provider,
-      baseModel: opts.baseModel,
-      status: "running",
-      fineTuneJobId: jobId,
-      createdAt: now,
-      updatedAt: now
-    });
-    const trainingJobId = randomUUID();
-    await db.insert(trainingJobs).values({
-      id: trainingJobId,
-      modelId,
-      provider: opts.provider,
-      status: jobStatus,
-      startedAt: now
-    });
-    printSuccess(`Fine-tune job started!`);
-    console.log();
-    console.log(`  Model ID:   ${modelId}`);
-    console.log(`  Job ID:     ${jobId}`);
-    console.log(`  Status:     ${printStatus(jobStatus)}`);
-    console.log();
-    printInfo(`Use 'brains finetune status ${jobId}' to check progress.`);
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-finetuneCmd.command("status <job-id>").description("Get the status of a fine-tuning job").option("--provider <provider>", "Provider (openai|thinker-labs)", "openai").option("--json", "Output as JSON").action(async (jobId, opts) => {
-  try {
-    let result;
-    if (opts.provider === "openai") {
-      result = await getFineTuneStatus(jobId);
-    } else {
-      const tl = new ThinkerLabsProvider;
-      result = await tl.getFineTuneStatus(jobId);
-    }
-    if (opts.json) {
-      printJson(result);
-    } else {
-      console.log();
-      console.log(`  Job ID:           ${result.jobId}`);
-      console.log(`  Status:           ${printStatus(result.status)}`);
-      if (result.fineTunedModel) {
-        console.log(`  Fine-tuned model: ${result.fineTunedModel}`);
-      }
-      if (result.error) {
-        console.log(`  Error:            ${result.error}`);
+  });
+  modelsCmd.command("untag <id> <tag>").description("Remove a tag from a model").action(async (id, tag) => {
+    try {
+      const db = getDb();
+      const [model] = await db.select().from(fineTunedModels).where(eq(fineTunedModels.id, id));
+      if (!model) {
+        printError(`Model not found: ${id}`);
+        process.exit(1);
       }
-      console.log();
-    }
-    const db = getDb();
-    const [model] = await db.select().from(fineTunedModels).where(eq(fineTunedModels.fineTuneJobId, jobId));
-    if (model) {
-      const status = result.status;
-      await db.update(fineTunedModels).set({ status, updatedAt: Date.now() }).where(eq(fineTunedModels.fineTuneJobId, jobId));
+      const existing = model.tags ? JSON.parse(model.tags) : [];
+      const updated = existing.filter((t) => t !== tag);
+      await db.update(fineTunedModels).set({ tags: JSON.stringify(updated), updatedAt: Date.now() }).where(eq(fineTunedModels.id, id));
+      printSuccess(`Tag "${tag}" removed. Tags: ${updated.join(", ") || "(none)"}`);
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
+      process.exit(1);
     }
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-finetuneCmd.command("watch <job-id>").description("Poll a fine-tuning job until it completes or fails").option("--provider <provider>", "Provider (openai|thinker-labs)", "openai").option("--interval <seconds>", "Poll interval in seconds", "30").action(async (jobId, opts) => {
-  const intervalMs = Math.max(5, parseInt(opts.interval, 10) || 30) * 1000;
-  const terminalStates = new Set(["succeeded", "failed", "cancelled"]);
-  printInfo(`Watching job ${jobId} (polling every ${intervalMs / 1000}s) \u2026`);
-  console.log();
-  const poll = async () => {
+  });
+  modelsCmd.command("collection <id> <collectionName>").description("Set the collection of a model").action(async (id, collectionName) => {
+    try {
+      const db = getDb();
+      await db.update(fineTunedModels).set({ collection: collectionName, updatedAt: Date.now() }).where(eq(fineTunedModels.id, id));
+      printSuccess(`Collection set to "${collectionName}"`);
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
+      process.exit(1);
+    }
+  });
+  modelsCmd.command("import <job-id>").description("Import an externally created fine-tuned model into local tracking").option("--provider <provider>", "Provider (openai|thinker-labs)", "openai").option("--name <name>", "Display name for the model").action(async (jobId, opts) => {
     try {
       let result;
       if (opts.provider === "openai") {
@@ -23496,255 +23337,425 @@ finetuneCmd.command("watch <job-id>").description("Poll a fine-tuning job until
         const tl = new ThinkerLabsProvider;
         result = await tl.getFineTuneStatus(jobId);
       }
-      const ts = new Date().toISOString().replace("T", " ").slice(0, 19);
-      process.stdout.write(`  [${ts}]  ${printStatus(result.status)}`);
+      const db = getDb();
+      const [existing] = await db.select().from(fineTunedModels).where(eq(fineTunedModels.fineTuneJobId, jobId));
+      if (existing) {
+        printInfo(`Model already tracked as: ${existing.id}`);
+        return;
+      }
+      const modelId = randomUUID();
+      const now = Date.now();
+      const name = opts.name ?? result.fineTunedModel ?? `imported-${jobId}`;
+      await db.insert(fineTunedModels).values({
+        id: modelId,
+        name,
+        provider: opts.provider,
+        baseModel: result.baseModel ?? "unknown",
+        status: result.status,
+        fineTuneJobId: jobId,
+        createdAt: now,
+        updatedAt: now
+      });
+      await db.insert(trainingJobs).values({
+        id: randomUUID(),
+        modelId,
+        provider: opts.provider,
+        status: result.status,
+        startedAt: now
+      });
+      printSuccess(`Model imported successfully.`);
+      console.log();
+      console.log(`  Local ID:  ${modelId}`);
+      console.log(`  Job ID:    ${jobId}`);
+      console.log(`  Name:      ${name}`);
+      console.log(`  Status:    ${printStatus(result.status)}`);
       if (result.fineTunedModel)
-        process.stdout.write(`  model: ${result.fineTunedModel}`);
-      process.stdout.write(`
-`);
+        console.log(`  Model:     ${result.fineTunedModel}`);
+      console.log();
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
+      process.exit(1);
+    }
+  });
+}
+
+// src/cli/commands/finetune.ts
+import { randomUUID as randomUUID2 } from "crypto";
+import { existsSync as existsSync9 } from "fs";
+function registerFinetuneCommands(program2) {
+  const finetuneCmd = program2.command("finetune").description("Manage fine-tuning jobs");
+  finetuneCmd.command("start").description("Start a fine-tuning job").requiredOption("--provider <provider>", "Provider to use (openai|thinker-labs)").requiredOption("--base-model <model>", "Base model to fine-tune (e.g. gpt-4o-mini-2024-07-18)").option("--dataset <path>", "Path to the JSONL training dataset (auto-detects latest if omitted)").requiredOption("--name <name>", "Human-readable name for this fine-tuned model").action(async (opts) => {
+    try {
+      if (opts.provider !== "openai" && opts.provider !== "thinker-labs") {
+        printError(`Unknown provider: ${opts.provider}. Use 'openai' or 'thinker-labs'.`);
+        process.exit(1);
+      }
+      let datasetPath = opts.dataset;
+      if (!datasetPath) {
+        const db2 = getDb();
+        const [latest] = await db2.select().from(trainingDatasets).orderBy(desc(trainingDatasets.createdAt)).limit(1);
+        if (!latest?.filePath) {
+          printError("No datasets found. Run 'brains data gather' first.");
+          process.exit(1);
+        }
+        datasetPath = latest.filePath;
+        printInfo(`Using latest dataset: ${datasetPath} (${latest.exampleCount} examples)`);
+      }
+      if (!existsSync9(datasetPath)) {
+        printError(`Dataset file not found: ${datasetPath}`);
+        process.exit(1);
+      }
+      printInfo(`Uploading training file: ${datasetPath} \u2026`);
+      let fileId;
+      let jobId;
+      let jobStatus;
+      if (opts.provider === "openai") {
+        ({ fileId } = await uploadTrainingFile(datasetPath));
+        printSuccess(`File uploaded. fileId = ${fileId}`);
+        printInfo(`Creating fine-tune job on OpenAI \u2026`);
+        ({ jobId, status: jobStatus } = await createFineTuneJob(fileId, opts.baseModel, opts.name));
+      } else {
+        const tl = new ThinkerLabsProvider;
+        ({ fileId } = await tl.uploadTrainingFile(datasetPath));
+        printSuccess(`File uploaded. fileId = ${fileId}`);
+        printInfo(`Creating fine-tune job on Thinker Labs \u2026`);
+        ({ jobId, status: jobStatus } = await tl.createFineTuneJob(fileId, opts.baseModel, opts.name));
+      }
+      const db = getDb();
+      const modelId = randomUUID2();
+      const now = Date.now();
+      await db.insert(fineTunedModels).values({
+        id: modelId,
+        name: opts.name,
+        provider: opts.provider,
+        baseModel: opts.baseModel,
+        status: "running",
+        fineTuneJobId: jobId,
+        createdAt: now,
+        updatedAt: now
+      });
+      const trainingJobId = randomUUID2();
+      await db.insert(trainingJobs).values({
+        id: trainingJobId,
+        modelId,
+        provider: opts.provider,
+        status: jobStatus,
+        startedAt: now
+      });
+      printSuccess(`Fine-tune job started!`);
+      console.log();
+      console.log(`  Model ID:   ${modelId}`);
+      console.log(`  Job ID:     ${jobId}`);
+      console.log(`  Status:     ${printStatus(jobStatus)}`);
+      console.log();
+      printInfo(`Use 'brains finetune status ${jobId}' to check progress.`);
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
+      process.exit(1);
+    }
+  });
+  finetuneCmd.command("status <job-id>").description("Get the status of a fine-tuning job").option("--provider <provider>", "Provider (openai|thinker-labs)", "openai").option("--json", "Output as JSON").action(async (jobId, opts) => {
+    try {
+      let result;
+      if (opts.provider === "openai") {
+        result = await getFineTuneStatus(jobId);
+      } else {
+        const tl = new ThinkerLabsProvider;
+        result = await tl.getFineTuneStatus(jobId);
+      }
+      if (opts.json) {
+        printJson(result);
+      } else {
+        console.log();
+        console.log(`  Job ID:           ${result.jobId}`);
+        console.log(`  Status:           ${printStatus(result.status)}`);
+        if (result.fineTunedModel)
+          console.log(`  Fine-tuned model: ${result.fineTunedModel}`);
+        if (result.error)
+          console.log(`  Error:            ${result.error}`);
+        console.log();
+      }
       const db = getDb();
       const [model] = await db.select().from(fineTunedModels).where(eq(fineTunedModels.fineTuneJobId, jobId));
       if (model) {
-        await db.update(fineTunedModels).set({ status: result.status, updatedAt: Date.now() }).where(eq(fineTunedModels.fineTuneJobId, jobId));
+        const status = result.status;
+        await db.update(fineTunedModels).set({ status, updatedAt: Date.now() }).where(eq(fineTunedModels.fineTuneJobId, jobId));
       }
-      if (terminalStates.has(result.status)) {
-        console.log();
-        if (result.status === "succeeded") {
-          printSuccess(`Job completed successfully.`);
-          if (result.fineTunedModel)
-            printSuccess(`Fine-tuned model: ${result.fineTunedModel}`);
-        } else if (result.status === "failed") {
-          printError(`Job failed.${result.error ? " Error: " + result.error : ""}`);
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
+      process.exit(1);
+    }
+  });
+  finetuneCmd.command("watch <job-id>").description("Poll a fine-tuning job until it completes or fails").option("--provider <provider>", "Provider (openai|thinker-labs)", "openai").option("--interval <seconds>", "Poll interval in seconds", "30").action(async (jobId, opts) => {
+    const intervalMs = Math.max(5, parseInt(opts.interval, 10) || 30) * 1000;
+    const terminalStates = new Set(["succeeded", "failed", "cancelled"]);
+    printInfo(`Watching job ${jobId} (polling every ${intervalMs / 1000}s) \u2026`);
+    console.log();
+    const poll = async () => {
+      try {
+        let result;
+        if (opts.provider === "openai") {
+          result = await getFineTuneStatus(jobId);
         } else {
-          printInfo(`Job ${result.status}.`);
+          const tl = new ThinkerLabsProvider;
+          result = await tl.getFineTuneStatus(jobId);
+        }
+        const ts = new Date().toISOString().replace("T", " ").slice(0, 19);
+        process.stdout.write(`  [${ts}]  ${printStatus(result.status)}`);
+        if (result.fineTunedModel)
+          process.stdout.write(`  model: ${result.fineTunedModel}`);
+        process.stdout.write(`
+`);
+        const db = getDb();
+        const [model] = await db.select().from(fineTunedModels).where(eq(fineTunedModels.fineTuneJobId, jobId));
+        if (model) {
+          await db.update(fineTunedModels).set({ status: result.status, updatedAt: Date.now() }).where(eq(fineTunedModels.fineTuneJobId, jobId));
+        }
+        if (terminalStates.has(result.status)) {
+          console.log();
+          if (result.status === "succeeded") {
+            printSuccess(`Job completed successfully.`);
+            if (result.fineTunedModel)
+              printSuccess(`Fine-tuned model: ${result.fineTunedModel}`);
+          } else if (result.status === "failed") {
+            printError(`Job failed.${result.error ? " Error: " + result.error : ""}`);
+          } else {
+            printInfo(`Job ${result.status}.`);
+          }
+          return true;
         }
-        return true;
+        return false;
+      } catch (err) {
+        printError(err instanceof Error ? err.message : String(err));
+        return false;
       }
-      return false;
+    };
+    const done = await poll();
+    if (!done) {
+      await new Promise((resolve2) => {
+        const timer = setInterval(async () => {
+          const finished = await poll();
+          if (finished) {
+            clearInterval(timer);
+            resolve2();
+          }
+        }, intervalMs);
+      });
+    }
+  });
+  finetuneCmd.command("list").description("List all fine-tuning jobs").option("--provider <provider>", "Provider to query (openai|thinker-labs)", "openai").option("--json", "Output as JSON").action(async (opts) => {
+    try {
+      let jobs;
+      if (opts.provider === "openai") {
+        jobs = await listFineTunedModels();
+      } else {
+        const tl = new ThinkerLabsProvider;
+        jobs = await tl.listFineTunedModels();
+      }
+      if (opts.json) {
+        printJson(jobs);
+        return;
+      }
+      if (jobs.length === 0) {
+        printInfo("No fine-tuning jobs found.");
+        return;
+      }
+      printTable(["Job ID", "Model", "Status", "Created"], jobs.map((j) => [
+        j.id,
+        j.model,
+        printStatus(j.status),
+        new Date(j.created * 1000).toISOString().split("T")[0] ?? ""
+      ]));
     } catch (err) {
       printError(err instanceof Error ? err.message : String(err));
-      return false;
-    }
-  };
-  const done = await poll();
-  if (!done) {
-    await new Promise((resolve2) => {
-      const timer = setInterval(async () => {
-        const finished = await poll();
-        if (finished) {
-          clearInterval(timer);
-          resolve2();
-        }
-      }, intervalMs);
-    });
-  }
-});
-finetuneCmd.command("list").description("List all fine-tuning jobs").option("--provider <provider>", "Provider to query (openai|thinker-labs)", "openai").option("--json", "Output as JSON").action(async (opts) => {
-  try {
-    let jobs;
-    if (opts.provider === "openai") {
-      jobs = await listFineTunedModels();
-    } else {
-      const tl = new ThinkerLabsProvider;
-      jobs = await tl.listFineTunedModels();
+      process.exit(1);
     }
-    if (opts.json) {
-      printJson(jobs);
-      return;
+  });
+}
+
+// src/cli/commands/data.ts
+import { randomUUID as randomUUID3 } from "crypto";
+import { readFileSync as readFileSync6, existsSync as existsSync11, mkdirSync as mkdirSync6, writeFileSync as writeFileSync5 } from "fs";
+import { join as join13 } from "path";
+import { homedir as homedir12 } from "os";
+var DEFAULT_DATASETS_DIR = join13(homedir12(), ".hasna", "brains", "datasets");
+function registerDataCommands(program2) {
+  const dataCmd = program2.command("data").description("Manage training datasets");
+  dataCmd.command("gather").description("Gather training data from agent memory sources").option("--source <source>", "Data source: todos|mementos|conversations|sessions|all", "all").option("--output <dir>", "Output directory", DEFAULT_DATASETS_DIR).option("--limit <n>", "Maximum number of examples to gather", "500").action(async (opts) => {
+    const validSources = ["todos", "mementos", "conversations", "sessions", "all"];
+    if (!validSources.includes(opts.source)) {
+      printError(`Invalid source: ${opts.source}. Choose one of: ${validSources.join(", ")}`);
+      process.exit(1);
     }
-    if (jobs.length === 0) {
-      printInfo("No fine-tuning jobs found.");
-      return;
+    const limit2 = parseInt(opts.limit, 10);
+    if (isNaN(limit2) || limit2 <= 0) {
+      printError(`Invalid --limit value: ${opts.limit}`);
+      process.exit(1);
     }
-    printTable(["Job ID", "Model", "Status", "Created"], jobs.map((j) => [
-      j.id,
-      j.model,
-      printStatus(j.status),
-      new Date(j.created * 1000).toISOString().split("T")[0] ?? ""
-    ]));
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-var DEFAULT_DATASETS_DIR = join13(homedir12(), ".hasna", "brains", "datasets");
-var dataCmd = program2.command("data").description("Manage training datasets");
-dataCmd.command("gather").description("Gather training data from agent memory sources").option("--source <source>", "Data source: todos|mementos|conversations|sessions|all", "all").option("--output <dir>", "Output directory", DEFAULT_DATASETS_DIR).option("--limit <n>", "Maximum number of examples to gather", "500").action(async (opts) => {
-  const validSources = ["todos", "mementos", "conversations", "sessions", "all"];
-  if (!validSources.includes(opts.source)) {
-    printError(`Invalid source: ${opts.source}. Choose one of: ${validSources.join(", ")}`);
-    process.exit(1);
-  }
-  const limit2 = parseInt(opts.limit, 10);
-  if (isNaN(limit2) || limit2 <= 0) {
-    printError(`Invalid --limit value: ${opts.limit}`);
-    process.exit(1);
-  }
-  try {
-    mkdirSync6(opts.output, { recursive: true });
-    const sources = opts.source === "all" ? ["todos", "mementos", "conversations", "sessions"] : [opts.source];
-    const now = Date.now();
-    const db = getDb();
-    const gathererMap = {
-      todos: (o) => Promise.resolve().then(() => (init_todos(), exports_todos)).then((m) => m.gatherFromTodos(o)),
-      mementos: (o) => Promise.resolve().then(() => (init_mementos(), exports_mementos)).then((m) => m.gatherFromMementos(o)),
-      conversations: (o) => Promise.resolve().then(() => (init_conversations(), exports_conversations)).then((m) => m.gatherFromConversations(o)),
-      sessions: (o) => Promise.resolve().then(() => (init_sessions(), exports_sessions2)).then((m) => m.gatherFromSessions(o))
-    };
-    let totalExamples = 0;
-    let successfulSources = 0;
-    for (const source of sources) {
-      printInfo(`Gathering from ${source} \u2026`);
-      try {
-        const gatherer = gathererMap[source];
-        if (!gatherer) {
-          printError(`  Unknown source: ${source}`);
-          continue;
-        }
-        const { examples, count } = await gatherer({ limit: limit2 });
-        if (count === 0) {
-          printInfo(`  No examples found in ${source}.`);
-          continue;
-        }
-        const fileName = `${source}-${now}.jsonl`;
-        const filePath = join13(opts.output, fileName);
-        writeFileSync5(filePath, examples.map((e) => JSON.stringify(e)).join(`
+    try {
+      mkdirSync6(opts.output, { recursive: true });
+      const sources = opts.source === "all" ? ["todos", "mementos", "conversations", "sessions"] : [opts.source];
+      const now = Date.now();
+      const db = getDb();
+      const gathererMap = {
+        todos: (o) => Promise.resolve().then(() => (init_todos(), exports_todos)).then((m) => m.gatherFromTodos(o)),
+        mementos: (o) => Promise.resolve().then(() => (init_mementos(), exports_mementos)).then((m) => m.gatherFromMementos(o)),
+        conversations: (o) => Promise.resolve().then(() => (init_conversations(), exports_conversations)).then((m) => m.gatherFromConversations(o)),
+        sessions: (o) => Promise.resolve().then(() => (init_sessions(), exports_sessions2)).then((m) => m.gatherFromSessions(o))
+      };
+      let totalExamples = 0;
+      let successfulSources = 0;
+      for (const source of sources) {
+        printInfo(`Gathering from ${source} \u2026`);
+        try {
+          const gatherer = gathererMap[source];
+          if (!gatherer) {
+            printError(`  Unknown source: ${source}`);
+            continue;
+          }
+          const { examples, count } = await gatherer({ limit: limit2 });
+          if (count === 0) {
+            printInfo(`  No examples found in ${source}.`);
+            continue;
+          }
+          const fileName = `${source}-${now}.jsonl`;
+          const filePath = join13(opts.output, fileName);
+          writeFileSync5(filePath, examples.map((e) => JSON.stringify(e)).join(`
 `) + `
 `, "utf8");
-        await db.insert(trainingDatasets).values({
-          id: randomUUID(),
-          source,
-          filePath,
-          exampleCount: count,
-          createdAt: now
-        });
-        printSuccess(`  \u2713 ${count} examples \u2192 ${filePath}`);
-        totalExamples += count;
-        successfulSources++;
-      } catch (sourceErr) {
-        printError(`  \u2717 ${source}: ${sourceErr instanceof Error ? sourceErr.message : String(sourceErr)}`);
+          await db.insert(trainingDatasets).values({
+            id: randomUUID3(),
+            source,
+            filePath,
+            exampleCount: count,
+            createdAt: now
+          });
+          printSuccess(`  \u2713 ${count} examples \u2192 ${filePath}`);
+          totalExamples += count;
+          successfulSources++;
+        } catch (sourceErr) {
+          printError(`  \u2717 ${source}: ${sourceErr instanceof Error ? sourceErr.message : String(sourceErr)}`);
+        }
       }
+      console.log();
+      printSuccess(`Total: ${totalExamples} examples from ${successfulSources} source(s)`);
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
+      process.exit(1);
     }
-    console.log();
-    printSuccess(`Total: ${totalExamples} examples from ${successfulSources} source(s)`);
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-dataCmd.command("preview <file>").description("Preview a JSONL training file").option("-n, --count <n>", "Number of examples to show", "5").action((file, opts) => {
-  if (!existsSync10(file)) {
-    printError(`File not found: ${file}`);
-    process.exit(1);
-  }
-  const n = parseInt(opts.count, 10);
-  if (isNaN(n) || n <= 0) {
-    printError(`Invalid --count value: ${opts.count}`);
-    process.exit(1);
-  }
-  try {
-    const content = readFileSync6(file, "utf8");
-    const lines = content.trim().split(`
-`).filter(Boolean);
-    const total = lines.length;
-    const preview = lines.slice(0, n);
-    console.log();
-    printInfo(`File: ${file}`);
-    printInfo(`Total examples: ${total}`);
-    printInfo(`Showing first ${Math.min(n, total)}:`);
-    console.log();
-    preview.forEach((line, idx) => {
-      try {
-        const parsed = JSON.parse(line);
-        console.log(`\u2500\u2500\u2500 Example ${idx + 1} \u2500\u2500\u2500`);
-        printJson(parsed);
-        console.log();
-      } catch {
-        printError(`  Line ${idx + 1} is not valid JSON: ${line.slice(0, 80)}\u2026`);
-      }
-    });
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-dataCmd.command("merge <files...>").description("Merge multiple JSONL datasets into one").option("--output <path>", "Output file path", join13(DEFAULT_DATASETS_DIR, `merged-${Date.now()}.jsonl`)).option("--no-dedupe", "Skip deduplication").action(async (files, opts) => {
-  try {
-    for (const f of files) {
-      if (!existsSync10(f)) {
-        printError(`File not found: ${f}`);
-        process.exit(1);
-      }
+  });
+  dataCmd.command("preview <file>").description("Preview a JSONL training file").option("-n, --count <n>", "Number of examples to show", "5").action((file, opts) => {
+    if (!existsSync11(file)) {
+      printError(`File not found: ${file}`);
+      process.exit(1);
     }
-    mkdirSync6(join13(opts.output, "..").replace(/\/\.\.$/, "") || DEFAULT_DATASETS_DIR, { recursive: true });
-    const allExamples = [];
-    for (const f of files) {
-      const lines = readFileSync6(f, "utf8").split(`
-`).map((l) => l.trim()).filter(Boolean);
-      allExamples.push(...lines);
-      printInfo(`  Read ${lines.length} examples from ${f}`);
-    }
-    let finalLines = allExamples;
-    let dupeCount = 0;
-    if (opts.dedupe) {
-      const seen = new Set;
-      finalLines = allExamples.filter((line) => {
-        if (seen.has(line)) {
-          dupeCount++;
-          return false;
+    const n = parseInt(opts.count, 10);
+    if (isNaN(n) || n <= 0) {
+      printError(`Invalid --count value: ${opts.count}`);
+      process.exit(1);
+    }
+    try {
+      const content = readFileSync6(file, "utf8");
+      const lines = content.trim().split(`
+`).filter(Boolean);
+      const total = lines.length;
+      const preview = lines.slice(0, n);
+      console.log();
+      printInfo(`File: ${file}`);
+      printInfo(`Total examples: ${total}`);
+      printInfo(`Showing first ${Math.min(n, total)}:`);
+      console.log();
+      preview.forEach((line, idx) => {
+        try {
+          const parsed = JSON.parse(line);
+          console.log(`\u2500\u2500\u2500 Example ${idx + 1} \u2500\u2500\u2500`);
+          printJson(parsed);
+          console.log();
+        } catch {
+          printError(`  Line ${idx + 1} is not valid JSON: ${line.slice(0, 80)}\u2026`);
         }
-        seen.add(line);
-        return true;
       });
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
+      process.exit(1);
     }
-    writeFileSync5(opts.output, finalLines.join(`
+  });
+  dataCmd.command("merge <files...>").description("Merge multiple JSONL datasets into one").option("--output <path>", "Output file path", join13(DEFAULT_DATASETS_DIR, `merged-${Date.now()}.jsonl`)).option("--no-dedupe", "Skip deduplication").action(async (files, opts) => {
+    try {
+      for (const f of files) {
+        if (!existsSync11(f)) {
+          printError(`File not found: ${f}`);
+          process.exit(1);
+        }
+      }
+      mkdirSync6(join13(opts.output, "..").replace(/\/\.\.$/, "") || DEFAULT_DATASETS_DIR, { recursive: true });
+      const allExamples = [];
+      for (const f of files) {
+        const lines = readFileSync6(f, "utf8").split(`
+`).map((l) => l.trim()).filter(Boolean);
+        allExamples.push(...lines);
+        printInfo(`  Read ${lines.length} examples from ${f}`);
+      }
+      let finalLines = allExamples;
+      let dupeCount = 0;
+      if (opts.dedupe) {
+        const seen = new Set;
+        finalLines = allExamples.filter((line) => {
+          if (seen.has(line)) {
+            dupeCount++;
+            return false;
+          }
+          seen.add(line);
+          return true;
+        });
+      }
+      writeFileSync5(opts.output, finalLines.join(`
 `) + `
 `, "utf8");
-    const db = getDb();
-    await db.insert(trainingDatasets).values({
-      id: randomUUID(),
-      source: "mixed",
-      filePath: opts.output,
-      exampleCount: finalLines.length,
-      createdAt: Date.now()
-    });
-    console.log();
-    printSuccess(`Merged ${files.length} files \u2014 ${finalLines.length} examples \u2192 ${opts.output}`);
-    if (opts.dedupe && dupeCount > 0)
-      printInfo(`  Removed ${dupeCount} duplicate(s)`);
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-dataCmd.command("list").description("List all gathered datasets").option("--json", "Output as JSON").action(async (opts) => {
-  try {
-    const db = getDb();
-    const datasets = await db.select().from(trainingDatasets);
-    if (opts.json) {
-      printJson(datasets);
-      return;
+      const db = getDb();
+      await db.insert(trainingDatasets).values({
+        id: randomUUID3(),
+        source: "mixed",
+        filePath: opts.output,
+        exampleCount: finalLines.length,
+        createdAt: Date.now()
+      });
+      console.log();
+      printSuccess(`Merged ${files.length} files \u2014 ${finalLines.length} examples \u2192 ${opts.output}`);
+      if (opts.dedupe && dupeCount > 0)
+        printInfo(`  Removed ${dupeCount} duplicate(s)`);
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
+      process.exit(1);
     }
-    if (datasets.length === 0) {
-      printInfo("No datasets found. Use 'brains data gather' to create one.");
-      return;
+  });
+  dataCmd.command("list").description("List all gathered datasets").option("--json", "Output as JSON").action(async (opts) => {
+    try {
+      const db = getDb();
+      const datasets = await db.select().from(trainingDatasets);
+      if (opts.json) {
+        printJson(datasets);
+        return;
+      }
+      if (datasets.length === 0) {
+        printInfo("No datasets found. Use 'brains data gather' to create one.");
+        return;
+      }
+      printTable(["ID", "Source", "Examples", "File", "Created"], datasets.map((d) => [
+        d.id,
+        d.source,
+        String(d.exampleCount),
+        d.filePath ?? "",
+        new Date(d.createdAt).toISOString().split("T")[0] ?? ""
+      ]));
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
+      process.exit(1);
     }
-    printTable(["ID", "Source", "Examples", "File", "Created"], datasets.map((d) => [
-      d.id,
-      d.source,
-      String(d.exampleCount),
-      d.filePath ?? "",
-      new Date(d.createdAt).toISOString().split("T")[0] ?? ""
-    ]));
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-var collectionsCmd = program2.command("collections").description("Manage model collections");
-collectionsCmd.option("--json", "Output as JSON").action(async (opts) => {
-  await listCollections(opts.json);
-});
+  });
+}
+
+// src/cli/commands/collections.ts
 async function listCollections(json = false) {
   try {
     const db = getDb();
@@ -23761,45 +23772,224 @@ async function listCollections(json = false) {
       printInfo("No collections found. Set a collection with 'brains models set-collection'.");
       return;
     }
-    printTable(["Collection", "Model Count", "Models"], rows.map((r) => [
-      r.collection ?? "(none)",
-      String(r.count),
-      r.names ?? ""
-    ]));
+    printTable(["Collection", "Model Count", "Models"], rows.map((r) => [r.collection ?? "(none)", String(r.count), r.names ?? ""]));
   } catch (err) {
     printError(err instanceof Error ? err.message : String(err));
     process.exit(1);
   }
 }
-collectionsCmd.command("list").description("List all collections with model counts").option("--json", "Output as JSON").action(async (opts) => {
-  await listCollections(opts.json);
-});
-collectionsCmd.command("show <name>").description("List all models in a collection").action(async (name) => {
-  try {
-    const db = getDb();
-    const models = await db.select().from(fineTunedModels).where(eq(fineTunedModels.collection, name));
-    if (models.length === 0) {
-      printInfo(`No models found in collection '${name}'.`);
-      return;
+function registerCollectionsCommands(program2) {
+  const collectionsCmd = program2.command("collections").description("Manage model collections");
+  collectionsCmd.option("--json", "Output as JSON").action(async (opts) => {
+    await listCollections(opts.json);
+  });
+  collectionsCmd.command("list").description("List all collections with model counts").option("--json", "Output as JSON").action(async (opts) => {
+    await listCollections(opts.json);
+  });
+  collectionsCmd.command("show <name>").description("List all models in a collection").action(async (name) => {
+    try {
+      const db = getDb();
+      const models = await db.select().from(fineTunedModels).where(eq(fineTunedModels.collection, name));
+      if (models.length === 0) {
+        printInfo(`No models found in collection '${name}'.`);
+        return;
+      }
+      printTable(["ID", "Name", "Provider", "Status", "Base Model"], models.map((m) => [m.id, m.name, m.provider, printStatus(m.status), m.baseModel]));
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
+      process.exit(1);
     }
-    printTable(["ID", "Name", "Provider", "Status", "Base Model"], models.map((m) => [m.id, m.name, m.provider, printStatus(m.status), m.baseModel]));
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
-collectionsCmd.command("rename <oldName> <newName>").description("Rename a collection across all models").action(async (oldName, newName) => {
-  try {
-    const db = getDb();
-    const affected = await db.select({ id: fineTunedModels.id }).from(fineTunedModels).where(eq(fineTunedModels.collection, oldName));
-    const count = affected.length;
-    await db.update(fineTunedModels).set({ collection: newName, updatedAt: Date.now() }).where(eq(fineTunedModels.collection, oldName));
-    printSuccess(`Renamed collection '${oldName}' \u2192 '${newName}' (${count} models updated)`);
-  } catch (err) {
-    printError(err instanceof Error ? err.message : String(err));
-    process.exit(1);
-  }
-});
+  });
+  collectionsCmd.command("rename <oldName> <newName>").description("Rename a collection across all models").action(async (oldName, newName) => {
+    try {
+      const db = getDb();
+      const affected = await db.select({ id: fineTunedModels.id }).from(fineTunedModels).where(eq(fineTunedModels.collection, oldName));
+      await db.update(fineTunedModels).set({ collection: newName, updatedAt: Date.now() }).where(eq(fineTunedModels.collection, oldName));
+      printSuccess(`Renamed collection '${oldName}' \u2192 '${newName}' (${affected.length} models updated)`);
+    } catch (err) {
+      printError(err instanceof Error ? err.message : String(err));
+      process.exit(1);
+    }
+  });
+}
+
+// src/cli/commands/cloud.ts
+function registerCloudCommands2(program2) {
+  const cloudCmd = program2.command("cloud").description("Cloud sync commands");
+  cloudCmd.command("status").description("Show cloud config and connection health").option("--json", "Output as JSON").action(async (opts) => {
+    try {
+      const { getCloudConfig: getCloudConfig2, getConnectionString: getConnectionString2, PgAdapterAsync: PgAdapterAsync2, SqliteAdapter: SqliteAdapter2, getDbPath: getDbPath2, listSqliteTables: listSqliteTables2, ensureConflictsTable: ensureConflictsTable2, listConflicts: listConflicts2 } = await Promise.resolve().then(() => (init_dist(), exports_dist));
+      const config = getCloudConfig2();
+      const info = {
+        mode: config.mode,
+        service: "brains",
+        rds_host: config.rds?.host || "(not configured)"
+      };
+      if (config.rds?.host && config.rds?.username) {
+        try {
+          const pg = new PgAdapterAsync2(getConnectionString2("postgres"));
+          await pg.get("SELECT 1 as ok");
+          info.postgresql = "connected";
+          await pg.close();
+        } catch (err) {
+          info.postgresql = `failed \u2014 ${err instanceof Error ? err.message : String(err)}`;
+        }
+      }
+      const local = new SqliteAdapter2(getDbPath2("brains"));
+      const tables = listSqliteTables2(local).filter((t) => !t.startsWith("_"));
+      const syncHealth = [];
+      for (const table of tables) {
+        try {
+          const totalRow = local.get(`SELECT COUNT(*) as c FROM "${table}"`);
+          const unsyncedRow = local.get(`SELECT COUNT(*) as c FROM "${table}" WHERE synced_at IS NULL`);
+          syncHealth.push({ table, total: totalRow?.c ?? 0, unsynced: unsyncedRow?.c ?? 0 });
+        } catch {}
+      }
+      info.sync_health = syncHealth.filter((s) => s.total > 0);
+      try {
+        ensureConflictsTable2(local);
+        const unresolved = listConflicts2(local, { resolved: false });
+        info.conflicts_unresolved = unresolved.length;
+      } catch {}
+      local.close();
+      if (opts.json) {
+        console.log(JSON.stringify(info, null, 2));
+        return;
+      }
+      printInfo(`Mode: ${info.mode}`);
+      printInfo(`RDS Host: ${info.rds_host}`);
+      if (info.postgresql)
+        printInfo(`PostgreSQL: ${info.postgresql}`);
+      for (const s of info.sync_health) {
+        const pct = s.total > 0 ? Math.round((s.total - s.unsynced) / s.total * 100) : 100;
+        printInfo(`  ${s.table}: ${pct}% synced (${s.unsynced} unsynced / ${s.total} total)`);
+      }
+      if (info.conflicts_unresolved)
+        printInfo(`Conflicts: ${info.conflicts_unresolved} unresolved`);
+    } catch (e) {
+      printError(e instanceof Error ? e.message : String(e));
+      process.exit(1);
+    }
+  });
+  cloudCmd.command("push").description("Push local data to cloud PostgreSQL").option("--tables <tables>", "Comma-separated table names (default: all)").option("--json", "Output as JSON").action(async (opts) => {
+    try {
+      const { getCloudConfig: getCloudConfig2, getConnectionString: getConnectionString2, syncPush: syncPush2, listSqliteTables: listSqliteTables2, SqliteAdapter: SqliteAdapter2, PgAdapterAsync: PgAdapterAsync2, getDbPath: getDbPath2 } = await Promise.resolve().then(() => (init_dist(), exports_dist));
+      const config = getCloudConfig2();
+      if (config.mode === "local") {
+        printError("Cloud mode not configured.");
+        process.exit(1);
+      }
+      const local = new SqliteAdapter2(getDbPath2("brains"));
+      const cloud = new PgAdapterAsync2(getConnectionString2("brains"));
+      const tableList = opts.tables ? opts.tables.split(",").map((t) => t.trim()) : listSqliteTables2(local).filter((t) => !t.startsWith("_"));
+      const results = await syncPush2(local, cloud, {
+        tables: tableList,
+        onProgress: (p) => {
+          if (!opts.json && p.phase === "done")
+            printInfo(`  ${p.table}: ${p.rowsWritten} rows pushed`);
+        }
+      });
+      local.close();
+      await cloud.close();
+      const total = results.reduce((s, r) => s + r.rowsWritten, 0);
+      if (opts.json) {
+        console.log(JSON.stringify({ total, tables: results }));
+        return;
+      }
+      printSuccess(`Done. ${total} rows pushed.`);
+    } catch (e) {
+      printError(e instanceof Error ? e.message : String(e));
+      process.exit(1);
+    }
+  });
+  cloudCmd.command("pull").description("Pull cloud data to local \u2014 merges by primary key").option("--tables <tables>", "Comma-separated table names (default: all)").option("--json", "Output as JSON").action(async (opts) => {
+    try {
+      const { getCloudConfig: getCloudConfig2, getConnectionString: getConnectionString2, syncPull: syncPull2, listPgTables: listPgTables2, SqliteAdapter: SqliteAdapter2, PgAdapterAsync: PgAdapterAsync2, getDbPath: getDbPath2 } = await Promise.resolve().then(() => (init_dist(), exports_dist));
+      const config = getCloudConfig2();
+      if (config.mode === "local") {
+        printError("Cloud mode not configured.");
+        process.exit(1);
+      }
+      const local = new SqliteAdapter2(getDbPath2("brains"));
+      const cloud = new PgAdapterAsync2(getConnectionString2("brains"));
+      const tableList = opts.tables ? opts.tables.split(",").map((t) => t.trim()) : (await listPgTables2(cloud)).filter((t) => !t.startsWith("_"));
+      const results = await syncPull2(cloud, local, {
+        tables: tableList,
+        onProgress: (p) => {
+          if (!opts.json && p.phase === "done")
+            printInfo(`  ${p.table}: ${p.rowsWritten} rows pulled`);
+        }
+      });
+      local.close();
+      await cloud.close();
+      const total = results.reduce((s, r) => s + r.rowsWritten, 0);
+      if (opts.json) {
+        console.log(JSON.stringify({ total, tables: results }));
+        return;
+      }
+      printSuccess(`Done. ${total} rows pulled.`);
+    } catch (e) {
+      printError(e instanceof Error ? e.message : String(e));
+      process.exit(1);
+    }
+  });
+  cloudCmd.command("sync").description("Bidirectional sync \u2014 pull then push").option("--tables <tables>", "Comma-separated table names (default: all)").option("--json", "Output as JSON").action(async (opts) => {
+    try {
+      const { getCloudConfig: getCloudConfig2, getConnectionString: getConnectionString2, syncPush: syncPush2, syncPull: syncPull2, listSqliteTables: listSqliteTables2, listPgTables: listPgTables2, SqliteAdapter: SqliteAdapter2, PgAdapterAsync: PgAdapterAsync2, getDbPath: getDbPath2 } = await Promise.resolve().then(() => (init_dist(), exports_dist));
+      const config = getCloudConfig2();
+      if (config.mode === "local") {
+        printError("Cloud mode not configured.");
+        process.exit(1);
+      }
+      const local = new SqliteAdapter2(getDbPath2("brains"));
+      const cloud = new PgAdapterAsync2(getConnectionString2("brains"));
+      const localTables = listSqliteTables2(local).filter((t) => !t.startsWith("_"));
+      const remoteTables = (await listPgTables2(cloud)).filter((t) => !t.startsWith("_"));
+      const tableList = opts.tables ? opts.tables.split(",").map((t) => t.trim()) : [...new Set([...localTables, ...remoteTables])];
+      const pullResults = await syncPull2(cloud, local, { tables: tableList.filter((t) => remoteTables.includes(t)) });
+      const pushResults = await syncPush2(local, cloud, { tables: tableList.filter((t) => localTables.includes(t)) });
+      local.close();
+      await cloud.close();
+      const pulled = pullResults.reduce((s, r) => s + r.rowsWritten, 0);
+      const pushed = pushResults.reduce((s, r) => s + r.rowsWritten, 0);
+      if (opts.json) {
+        console.log(JSON.stringify({ pulled, pushed }));
+        return;
+      }
+      printSuccess(`Sync done. Pulled ${pulled} rows, pushed ${pushed} rows.`);
+    } catch (e) {
+      printError(e instanceof Error ? e.message : String(e));
+      process.exit(1);
+    }
+  });
+  cloudCmd.command("migrate-pg").description("Apply PostgreSQL migrations to the cloud database").requiredOption("--connection-string <connStr>", "PostgreSQL connection string").option("--json", "Output as JSON").action(async (opts) => {
+    try {
+      const { applyPgMigrations: applyPgMigrations3 } = await Promise.resolve().then(() => (init_pg_migrate(), exports_pg_migrate));
+      const result = await applyPgMigrations3(opts.connectionString);
+      if (opts.json) {
+        console.log(JSON.stringify(result));
+        return;
+      }
+      printSuccess(`Applied ${result.applied.length} migration(s), skipped ${result.alreadyApplied.length}.`);
+      if (result.errors.length > 0) {
+        printError(result.errors.join(`
+`));
+        process.exit(1);
+      }
+    } catch (e) {
+      printError(e instanceof Error ? e.message : String(e));
+      process.exit(1);
+    }
+  });
+}
+
+// src/cli/index.ts
+var program2 = new Command;
+program2.name("brains").description("Fine-tuned model tracker and trainer").version("0.0.1");
+registerModelsCommands(program2);
+registerFinetuneCommands(program2);
+registerDataCommands(program2);
+registerCollectionsCommands(program2);
 program2.command("remove <id>").alias("rm").alias("uninstall").description("Remove a fine-tuned model or training job by ID").option("--type <type>", "Type: model | job (default: auto-detect)").action(async (id, opts) => {
   const db = getDb();
   try {
@@ -23874,7 +24064,7 @@ var feedbackCmd = program2.command("feedback").description("Feedback commands");
 feedbackCmd.command("send <message>").description("Send feedback about brains").option("--email <email>", "Contact email").action(async (message, opts) => {
   const { sendFeedback: sendFeedback2 } = await Promise.resolve().then(() => (init_dist(), exports_dist));
   const rawDb = getRawDb();
-  const pkg = JSON.parse(readFileSync6(join13(import.meta.dir, "../../package.json"), "utf8"));
+  const pkg = JSON.parse(readFileSync7(join14(import.meta.dir, "../../package.json"), "utf8"));
   const result = await sendFeedback2({ service: "brains", message, email: opts.email, version: pkg.version }, rawDb);
   rawDb.close();
   if (result.sent) {
@@ -23904,170 +24094,5 @@ feedbackCmd.command("list").description("List locally saved feedback").option("-
     e.created_at ?? ""
   ]));
 });
-var cloudCmd = program2.command("cloud").description("Cloud sync commands");
-cloudCmd.command("status").description("Show cloud config and connection health").option("--json", "Output as JSON").action(async (opts) => {
-  try {
-    const { getCloudConfig: getCloudConfig2, getConnectionString: getConnectionString2, PgAdapterAsync: PgAdapterAsync2, SqliteAdapter: SqliteAdapter2, getDbPath: getDbPath2, listSqliteTables: listSqliteTables2, ensureConflictsTable: ensureConflictsTable2, listConflicts: listConflicts2 } = await Promise.resolve().then(() => (init_dist(), exports_dist));
-    const config = getCloudConfig2();
-    const info = {
-      mode: config.mode,
-      service: "brains",
-      rds_host: config.rds?.host || "(not configured)"
-    };
-    if (config.rds?.host && config.rds?.username) {
-      try {
-        const pg = new PgAdapterAsync2(getConnectionString2("postgres"));
-        await pg.get("SELECT 1 as ok");
-        info.postgresql = "connected";
-        await pg.close();
-      } catch (err) {
-        info.postgresql = `failed \u2014 ${err instanceof Error ? err.message : String(err)}`;
-      }
-    }
-    const local = new SqliteAdapter2(getDbPath2("brains"));
-    const tables = listSqliteTables2(local).filter((t) => !t.startsWith("_"));
-    const syncHealth = [];
-    for (const table of tables) {
-      try {
-        const totalRow = local.get(`SELECT COUNT(*) as c FROM "${table}"`);
-        const unsyncedRow = local.get(`SELECT COUNT(*) as c FROM "${table}" WHERE synced_at IS NULL`);
-        syncHealth.push({ table, total: totalRow?.c ?? 0, unsynced: unsyncedRow?.c ?? 0 });
-      } catch {}
-    }
-    info.sync_health = syncHealth.filter((s) => s.total > 0);
-    try {
-      ensureConflictsTable2(local);
-      const unresolved = listConflicts2(local, { resolved: false });
-      info.conflicts_unresolved = unresolved.length;
-    } catch {}
-    local.close();
-    if (opts.json) {
-      console.log(JSON.stringify(info, null, 2));
-      return;
-    }
-    printInfo(`Mode: ${info.mode}`);
-    printInfo(`RDS Host: ${info.rds_host}`);
-    if (info.postgresql)
-      printInfo(`PostgreSQL: ${info.postgresql}`);
-    for (const s of info.sync_health) {
-      const pct = s.total > 0 ? Math.round((s.total - s.unsynced) / s.total * 100) : 100;
-      printInfo(`  ${s.table}: ${pct}% synced (${s.unsynced} unsynced / ${s.total} total)`);
-    }
-    if (info.conflicts_unresolved)
-      printInfo(`Conflicts: ${info.conflicts_unresolved} unresolved`);
-  } catch (e) {
-    printError(e instanceof Error ? e.message : String(e));
-    process.exit(1);
-  }
-});
-cloudCmd.command("push").description("Push local data to cloud PostgreSQL").option("--tables <tables>", "Comma-separated table names (default: all)").option("--json", "Output as JSON").action(async (opts) => {
-  try {
-    const { getCloudConfig: getCloudConfig2, getConnectionString: getConnectionString2, syncPush: syncPush2, listSqliteTables: listSqliteTables2, SqliteAdapter: SqliteAdapter2, PgAdapterAsync: PgAdapterAsync2, getDbPath: getDbPath2 } = await Promise.resolve().then(() => (init_dist(), exports_dist));
-    const config = getCloudConfig2();
-    if (config.mode === "local") {
-      printError("Cloud mode not configured.");
-      process.exit(1);
-    }
-    const local = new SqliteAdapter2(getDbPath2("brains"));
-    const cloud = new PgAdapterAsync2(getConnectionString2("brains"));
-    const tableList = opts.tables ? opts.tables.split(",").map((t) => t.trim()) : listSqliteTables2(local).filter((t) => !t.startsWith("_"));
-    const results = await syncPush2(local, cloud, {
-      tables: tableList,
-      onProgress: (p) => {
-        if (!opts.json && p.phase === "done")
-          printInfo(`  ${p.table}: ${p.rowsWritten} rows pushed`);
-      }
-    });
-    local.close();
-    await cloud.close();
-    const total = results.reduce((s, r) => s + r.rowsWritten, 0);
-    if (opts.json) {
-      console.log(JSON.stringify({ total, tables: results }));
-      return;
-    }
-    printSuccess(`Done. ${total} rows pushed.`);
-  } catch (e) {
-    printError(e instanceof Error ? e.message : String(e));
-    process.exit(1);
-  }
-});
-cloudCmd.command("pull").description("Pull cloud data to local \u2014 merges by primary key").option("--tables <tables>", "Comma-separated table names (default: all)").option("--json", "Output as JSON").action(async (opts) => {
-  try {
-    const { getCloudConfig: getCloudConfig2, getConnectionString: getConnectionString2, syncPull: syncPull2, listPgTables: listPgTables2, SqliteAdapter: SqliteAdapter2, PgAdapterAsync: PgAdapterAsync2, getDbPath: getDbPath2 } = await Promise.resolve().then(() => (init_dist(), exports_dist));
-    const config = getCloudConfig2();
-    if (config.mode === "local") {
-      printError("Cloud mode not configured.");
-      process.exit(1);
-    }
-    const local = new SqliteAdapter2(getDbPath2("brains"));
-    const cloud = new PgAdapterAsync2(getConnectionString2("brains"));
-    const tableList = opts.tables ? opts.tables.split(",").map((t) => t.trim()) : (await listPgTables2(cloud)).filter((t) => !t.startsWith("_"));
-    const results = await syncPull2(cloud, local, {
-      tables: tableList,
-      onProgress: (p) => {
-        if (!opts.json && p.phase === "done")
-          printInfo(`  ${p.table}: ${p.rowsWritten} rows pulled`);
-      }
-    });
-    local.close();
-    await cloud.close();
-    const total = results.reduce((s, r) => s + r.rowsWritten, 0);
-    if (opts.json) {
-      console.log(JSON.stringify({ total, tables: results }));
-      return;
-    }
-    printSuccess(`Done. ${total} rows pulled.`);
-  } catch (e) {
-    printError(e instanceof Error ? e.message : String(e));
-    process.exit(1);
-  }
-});
-cloudCmd.command("sync").description("Bidirectional sync \u2014 pull then push").option("--tables <tables>", "Comma-separated table names (default: all)").option("--json", "Output as JSON").action(async (opts) => {
-  try {
-    const { getCloudConfig: getCloudConfig2, getConnectionString: getConnectionString2, syncPush: syncPush2, syncPull: syncPull2, listSqliteTables: listSqliteTables2, listPgTables: listPgTables2, SqliteAdapter: SqliteAdapter2, PgAdapterAsync: PgAdapterAsync2, getDbPath: getDbPath2 } = await Promise.resolve().then(() => (init_dist(), exports_dist));
-    const config = getCloudConfig2();
-    if (config.mode === "local") {
-      printError("Cloud mode not configured.");
-      process.exit(1);
-    }
-    const local = new SqliteAdapter2(getDbPath2("brains"));
-    const cloud = new PgAdapterAsync2(getConnectionString2("brains"));
-    const localTables = listSqliteTables2(local).filter((t) => !t.startsWith("_"));
-    const remoteTables = (await listPgTables2(cloud)).filter((t) => !t.startsWith("_"));
-    const tableList = opts.tables ? opts.tables.split(",").map((t) => t.trim()) : [...new Set([...localTables, ...remoteTables])];
-    const pullResults = await syncPull2(cloud, local, { tables: tableList.filter((t) => remoteTables.includes(t)) });
-    const pushResults = await syncPush2(local, cloud, { tables: tableList.filter((t) => localTables.includes(t)) });
-    local.close();
-    await cloud.close();
-    const pulled = pullResults.reduce((s, r) => s + r.rowsWritten, 0);
-    const pushed = pushResults.reduce((s, r) => s + r.rowsWritten, 0);
-    if (opts.json) {
-      console.log(JSON.stringify({ pulled, pushed }));
-      return;
-    }
-    printSuccess(`Sync done. Pulled ${pulled} rows, pushed ${pushed} rows.`);
-  } catch (e) {
-    printError(e instanceof Error ? e.message : String(e));
-    process.exit(1);
-  }
-});
-cloudCmd.command("migrate-pg").description("Apply PostgreSQL migrations to the cloud database").requiredOption("--connection-string <connStr>", "PostgreSQL connection string").option("--json", "Output as JSON").action(async (opts) => {
-  try {
-    const { applyPgMigrations: applyPgMigrations3 } = await Promise.resolve().then(() => (init_pg_migrate(), exports_pg_migrate));
-    const result = await applyPgMigrations3(opts.connectionString);
-    if (opts.json) {
-      console.log(JSON.stringify(result));
-      return;
-    }
-    printSuccess(`Applied ${result.applied.length} migration(s), skipped ${result.alreadyApplied.length}.`);
-    if (result.errors.length > 0) {
-      printError(result.errors.join(`
-`));
-      process.exit(1);
-    }
-  } catch (e) {
-    printError(e instanceof Error ? e.message : String(e));
-    process.exit(1);
-  }
-});
+registerCloudCommands2(program2);
 program2.parse();