offgrid-ai 0.3.14 → 0.3.16

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "offgrid-ai",
-  "version": "0.3.14",
+  "version": "0.3.16",
   "description": "Privacy-first CLI for running local LLMs — discover, configure, run, benchmark",
   "author": "Eeshan Srivastava (https://eeshans.com)",
   "type": "module",

package/src/cli.mjs

@@ -52,6 +52,8 @@ export async function run(argv) {
 
   if (command === "help" || command === "--help" || command === "-h") return printHelp();
   if (command === "version" || command === "--version" || command === "-v") return printVersion();
+  if (command === "models") return modelsCommand(argv.slice(1));
+  if (command === "run") return runCommand(argv.slice(1));
   if (command === "status") return statusCommand();
   if (command === "stop") return stopCommand(argv.slice(1));
   if (command === "uninstall" || command === "--uninstall") return uninstallCommand(argv.slice(1));
@@ -151,156 +153,236 @@ export async function mainFlow() {
     return;
   }
 
-  // 6. Interactive: pick an action
+  // 6. Interactive: one command center after onboarding.
   startInteractive("offgrid-ai");
-  const prompt = createPrompt();
-  try {
-    // Show what we found
-    const profiledPaths = new Set(profiles.map((p) => p.modelPath).filter(Boolean));
-    const newModels = ggufModels.filter((m) => !profiledPaths.has(m.path));
-
-    // Managed backend models
-    const managedItems = [];
-    for (const { backendId, models } of managedModels) {
-      const profiledAliases = new Set(
-        profiles.filter((p) => p.backend === backendId).map((p) => backendId === "ollama" ? `ollama:${p.ollamaModel ?? p.modelAlias}` : `omlx:${p.omlxModel ?? p.modelAlias}`)
-      );
-      for (const model of models) {
-        if (!profiledAliases.has(`${backendId}:${model.id}`)) {
-          managedItems.push({ model, backendId });
-        }
-      }
-    }
+  return await modelCommandCenter({ profiles, ggufModels, managedModels });
+}
 
-    // Show what we found
-    if (profiles.length > 0) {
-      console.log(pc.bold("\nSaved profiles"));
-      for (const profile of profiles) {
-        const backend = backendFor(profile.backend);
-        const colorMap = { "llama-cpp": pc.yellow, "llama-cpp-mtp": pc.blue, "ollama": pc.magenta, "omlx": pc.cyan };
-        const running = await isProfileRunning(profile);
-        const c = colorMap[profile.backend] ?? pc.magenta;
-        console.log(`  ${running ? pc.green("●") : pc.dim("○")} ${pc.bold(profile.label)} ${c(`[${backend.label}]`)} · ${pc.cyan(profile.modelAlias)}`);
-      }
-    }
-    if (newModels.length > 0) {
-      console.log(pc.bold("\nNew models"));
-      for (const model of newModels.slice(0, 10)) {
-        console.log(`  ${pc.cyan(model.label)} ${pc.dim(model.quant ?? "")} · ${pc.dim(formatBytes(model.sizeBytes))}`);
-      }
-      if (newModels.length > 10) console.log(pc.dim(`  ... and ${newModels.length - 10} more`));
-    }
-    for (const { backendId, models } of managedModels) {
-      if (models.length > 0) {
-        const be = BACKENDS[backendId];
-        console.log(pc.bold(`\n${be.label} models`));
-        for (const model of models.slice(0, 5)) {
-          console.log(`  ${pc.cyan(model.label)}`);
-        }
-        if (models.length > 5) console.log(pc.dim(`  ... and ${models.length - 5} more`));
-      }
-    }
+// ── Model command center ────────────────────────────────────────────────────
 
-    // Pick what to do
-    const action = await prompt.choice("What next?", [
-      { value: "run", label: "Run a model", hint: "Start server and launch Pi" },
-      ...(profiles.length > 0 ? [{ value: "manage", label: "Manage profiles", hint: "Sync, remove, or inspect" }] : []),
-      { value: "benchmark", label: "Benchmark", hint: "Run a benchmark prompt" },
-    ], "run");
+async function modelsCommand(argv) {
+  await ensureDirs();
+  const catalog = await loadModelCatalog();
+
+  if (argv[0]) {
+    const profile = await readProfile(argv[0]);
+    await printProfileDetails(profile);
+    return;
+  }
+
+  if (process.stdin.isTTY) startInteractive("offgrid-ai");
+  return await modelCommandCenter(catalog);
+}
 
-    if (action === "run") return await pickAndRun(prompt, profiles, newModels, managedItems);
-    if (action === "manage") return await manageProfiles(prompt, profiles);
-    if (action === "benchmark") return await benchmarkFlow(prompt, profiles);
+async function modelCommandCenter(catalog) {
+  const normalized = normalizeCatalog(catalog);
+  await printModelCatalog(normalized);
+  if (!process.stdin.isTTY) return;
+
+  const items = modelCatalogItems(normalized);
+  if (items.length === 0) return;
+
+  const prompt = createPrompt();
+  try {
+    const action = await prompt.choice("What do you want to do?", [
+      { value: "inspect", label: "Inspect", hint: "View details" },
+      { value: "setup", label: "Set up / sync", hint: "Create profile or sync Pi" },
+      { value: "run", label: "Run", hint: "Start server and launch Pi" },
+      { value: "benchmark", label: "Benchmark", hint: "Coming soon: local benchmark project" },
+      { value: "remove", label: "Remove", hint: "Delete a saved profile" },
+    ], "run");
+    if (action === "benchmark") return await benchmarkFlow();
+    const item = await chooseCatalogItem(prompt, items, action);
+    if (!item) return;
+    return await handleCatalogAction(prompt, action, item);
   } finally {
     prompt.close();
   }
 }
 
-// ── Pick and run ────────────────────────────────────────────────────────────
+async function runCommand(argv) {
+  await ensureDirs();
+  const { positional } = parseOptions(argv);
+  if (!positional[0]) return await mainFlow();
+  const profile = await readProfile(positional[0]);
+  return await runProfile(profile);
+}
 
-async function pickAndRun(prompt, profiles, newModels, managedItems) {
-  // If there's exactly one profile and it's already running, offer to connect or start fresh
-  const choices = [];
+async function loadModelCatalog() {
+  const [profiles, ggufModels, managedModels] = await Promise.all([
+    loadProfiles(),
+    scanGgufModels(),
+    scanManagedModels(),
+  ]);
+  return normalizeCatalog({ profiles, ggufModels, managedModels });
+}
 
-  // Existing profiles
-  for (const profile of profiles) {
-    const running = await isProfileRunning(profile);
-    const backend = backendFor(profile.backend);
-    const colorMap = { "llama-cpp": pc.yellow, "llama-cpp-mtp": pc.blue, "ollama": pc.magenta, "omlx": pc.cyan };
-    const c = colorMap[profile.backend] ?? pc.magenta;
-    choices.push({
-      value: `profile:${profile.id}`,
-      label: `${running ? pc.green("● ") : ""}${profile.label}`,
-      hint: `${c(backend.label)} · ${profile.modelAlias} · ${profile.baseUrl}`,
-    });
-  }
-
-  // New GGUF models
-  for (const model of newModels.slice(0, 20)) {
-    choices.push({
-      value: `new:${model.path}`,
-      label: model.label,
-      hint: `${model.quant ?? "GGUF"} · ${formatBytes(model.sizeBytes)}`,
-    });
-  }
-
-  // Managed models
-  for (const { model, backendId } of managedItems) {
+function normalizeCatalog(catalog) {
+  if (catalog.newModels && catalog.managedItems) return catalog;
+  const { profiles, ggufModels, managedModels } = catalog;
+  const profiledPaths = new Set(profiles.map((p) => p.modelPath).filter(Boolean));
+  const newModels = ggufModels.filter((m) => !profiledPaths.has(m.path));
+  const managedItems = [];
+  for (const { backendId, models } of managedModels) {
+    const profiledAliases = new Set(
+      profiles.filter((p) => p.backend === backendId).map((p) => backendId === "ollama" ? `ollama:${p.ollamaModel ?? p.modelAlias}` : `omlx:${p.omlxModel ?? p.modelAlias}`)
+    );
+    for (const model of models) {
+      if (!profiledAliases.has(`${backendId}:${model.id}`)) managedItems.push({ model, backendId });
+    }
+  }
+  return { profiles, ggufModels, managedModels, newModels, managedItems };
+}
+
+async function printModelCatalog({ profiles, newModels, managedModels }) {
+  if (profiles.length > 0) {
+    console.log(pc.bold("\nSaved profiles"));
+    for (const profile of profiles) {
+      const backend = backendFor(profile.backend);
+      const colorMap = { "llama-cpp": pc.yellow, "llama-cpp-mtp": pc.blue, "ollama": pc.magenta, "omlx": pc.cyan };
+      const running = await isProfileRunning(profile);
+      const piConfigured = await hasPiModel(profile);
+      const c = colorMap[profile.backend] ?? pc.magenta;
+      console.log(`  ${running ? pc.green("●") : pc.dim("○")} ${pc.bold(profile.label)} ${c(`[${backend.label}]`)} · ${pc.cyan(profile.modelAlias)} ${piConfigured ? pc.green("· Pi synced") : pc.yellow("· Pi not synced")}`);
+    }
+  } else {
+    console.log(pc.bold("\nSaved profiles"));
+    console.log(pc.dim("  None yet."));
+  }
+
+  if (newModels.length > 0) {
+    console.log(pc.bold("\nNew GGUF models"));
+    for (const model of newModels.slice(0, 20)) {
+      console.log(`  ${pc.cyan(model.label)} ${pc.dim(model.quant ?? "")} · ${pc.dim(formatBytes(model.sizeBytes))}`);
+    }
+    if (newModels.length > 20) console.log(pc.dim(`  ... and ${newModels.length - 20} more`));
+  }
+
+  for (const { backendId, models } of managedModels) {
+    if (models.length === 0) continue;
     const be = BACKENDS[backendId];
-    choices.push({
-      value: `managed:${backendId}:${model.id}`,
-      label: model.label,
-      hint: `${be.label}`,
-    });
+    console.log(pc.bold(`\n${be.label} models`));
+    for (const model of models.slice(0, 10)) console.log(`  ${pc.cyan(model.label)}`);
+    if (models.length > 10) console.log(pc.dim(`  ... and ${models.length - 10} more`));
   }
+}
 
-  if (choices.length === 0) {
-    console.log(pc.yellow("No models available."));
-    return;
+function modelCatalogItems({ profiles, newModels, managedItems }) {
+  return [
+    ...profiles.map((profile) => ({ type: "profile", profile, label: profile.label, hint: `${profile.modelAlias} · ${profile.baseUrl}` })),
+    ...newModels.map((model) => ({ type: "new", model, label: model.label, hint: `${model.quant ?? "GGUF"} · ${formatBytes(model.sizeBytes)}` })),
+    ...managedItems.map(({ model, backendId }) => ({ type: "managed", model, backendId, label: model.label, hint: BACKENDS[backendId].label })),
+  ];
+}
+
+async function chooseCatalogItem(prompt, items, action) {
+  const allowed = action === "remove" ? items.filter((item) => item.type === "profile") : items;
+  if (allowed.length === 0) {
+    console.log(pc.yellow(action === "remove" ? "No saved profiles to remove." : "No models available."));
+    return null;
   }
+  const selected = await prompt.choice("Select", allowed.map((item, index) => ({
+    value: String(index),
+    label: item.label,
+    hint: item.hint,
+  })), "0");
+  return allowed[Number(selected)];
+}
 
-  const selected = await prompt.choice("Pick a model", choices, choices[0].value);
+async function handleCatalogAction(prompt, action, item) {
+  if (action === "inspect") {
+    if (item.type === "profile") return await printProfileDetails(await readProfile(item.profile.id));
+    if (item.type === "managed") return printManagedModelDetails(item.model, BACKENDS[item.backendId]);
+    return printGgufModelDetails(item.model);
+  }
 
-  if (selected.startsWith("profile:")) {
-    const id = selected.slice("profile:".length);
-    const profile = await readProfile(id);
-    return await runProfile(profile);
+  if (action === "setup") {
+    if (item.type === "profile") return await syncPiConfig(await readProfile(item.profile.id));
+    if (item.type === "managed") {
+      const profile = createManagedProfile(item.model, item.backendId);
+      await saveProfile(profile);
+      return await syncPiConfig(profile);
+    }
+    const profile = await createProfileFromModel(item.model);
+    const configured = await configureLocalProfile(prompt, profile);
+    if (!configured) return;
+    await saveProfile(configured);
+    return await syncPiConfig(configured);
   }
 
-  if (selected.startsWith("new:")) {
-    const modelPath = selected.slice("new:".length);
-    const model = newModels.find((m) => m.path === modelPath);
-    if (!model) throw new Error("Model not found.");
-    const profile = await createProfileFromModel(model);
+  if (action === "run") {
+    if (item.type === "profile") return await runProfile(await readProfile(item.profile.id));
+    if (item.type === "managed") {
+      const profile = createManagedProfile(item.model, item.backendId);
+      await saveProfile(profile);
+      await syncPiConfig(profile);
+      return await runProfile(profile);
+    }
+    const profile = await createProfileFromModel(item.model);
     const configured = await configureLocalProfile(prompt, profile);
     if (!configured) return;
     await saveProfile(configured);
-    console.log(pc.green(`Saved profile: ${configured.label}`));
     await syncPiConfig(configured);
     return await runProfile(configured);
   }
 
-  if (selected.startsWith("managed:")) {
-    const managedSelection = selected.slice("managed:".length);
-    const separator = managedSelection.indexOf(":");
-    const backendId = separator === -1 ? managedSelection : managedSelection.slice(0, separator);
-    const modelId = separator === -1 ? "" : managedSelection.slice(separator + 1);
-    const model = managedItems.find((m) => m.model.id === modelId && m.backendId === backendId)?.model;
-    if (!model) throw new Error("Model not found.");
-    const profile = normalizeProfile({
-      id: model.id.replace(/[^a-z0-9._-]+/gi, "-").toLowerCase(),
-      label: model.label,
-      backend: backendId,
-      modelAlias: model.aliasSuggestion,
-      ...(backendId === "ollama" ? { ollamaModel: model.id } : {}),
-      ...(backendId === "omlx" ? { omlxModel: model.id } : {}),
-    });
-    await saveProfile(profile);
-    await syncPiConfig(profile);
-    return await runProfile(profile);
+  if (action === "remove" && item.type === "profile") return await removeProfileInteractive(item.profile.id);
+}
+
+async function printProfileDetails(profile) {
+  const backend = backendFor(profile.backend);
+  const isManaged = backend.type === "managed-server";
+  const piConfigured = await hasPiModel(profile);
+  console.log("\n" + renderSection("Profile", renderRows([
+    ["ID", pc.cyan(profile.id)],
+    ["Label", pc.bold(profile.label)],
+    ["Backend", backend.label],
+    ["Endpoint", pc.green(profile.baseUrl)],
+    ...(!isManaged ? [
+      ["Model", profile.modelPath ?? "unknown"],
+      ["MMProj", profile.mmprojPath ?? "none"],
+      ["Memory", profile.modelPath && existsSync(profile.modelPath) ? formatBytes(statSync(profile.modelPath).size) : "unknown"],
+    ] : []),
+    ["Alias", pc.cyan(profile.modelAlias)],
+    ["Pi", piConfigured ? pc.green("configured") : pc.yellow("not synced")],
+  ])));
+
+  if (!isManaged && profile.commandArgv) {
+    console.log("\n" + pc.bold("llama-server command"));
+    console.log(pc.dim(buildPrettyCommand(profile)));
   }
 }
 
+function printGgufModelDetails(model) {
+  console.log("\n" + renderSection("GGUF model", renderRows([
+    ["Label", pc.bold(model.label)],
+    ["Model", model.path],
+    ["MMProj", model.mmprojPath ?? "none"],
+    ["Quant", model.quant ?? "unknown"],
+    ["Size", formatBytes(model.sizeBytes)],
+  ])));
+}
+
+function printManagedModelDetails(model, backend) {
+  console.log("\n" + renderSection(`${backend.label} model`, renderRows([
+    ["Label", pc.bold(model.label)],
+    ["ID", pc.cyan(model.id)],
+    ["Quant", model.quant ?? "unknown"],
+    ["Family", model.family ?? "unknown"],
+  ])));
+}
+
+function createManagedProfile(model, backendId) {
+  return normalizeProfile({
+    id: model.id.replace(/[^a-z0-9._-]+/gi, "-").toLowerCase(),
+    label: model.label,
+    backend: backendId,
+    modelAlias: model.aliasSuggestion,
+    ...(backendId === "ollama" ? { ollamaModel: model.id } : {}),
+    ...(backendId === "omlx" ? { omlxModel: model.id } : {}),
+  });
+}
+
 async function runProfile(profile, options = {}) {
   const backend = backendFor(profile.backend);
   const withHarness = options.with ?? "pi";
@@ -382,56 +464,6 @@ async function runProfile(profile, options = {}) {
   }
 }
 
-// ── Manage profiles ─────────────────────────────────────────────────────────
-
-async function manageProfiles(prompt, profiles) {
-  const choices = profiles.map((p) => ({
-    value: p.id,
-    label: p.label,
-    hint: `${p.modelAlias} · ${p.baseUrl}`,
-  }));
-
-  const selected = await prompt.choice("Which profile?", choices, choices[0].value);
-  const profile = await readProfile(selected);
-  const backend = backendFor(profile.backend);
-  const isManaged = backend.type === "managed-server";
-  const piConfigured = await hasPiModel(profile);
-
-  // Show profile details
-  console.log("");
-  console.log(renderSection("Profile", renderRows([
-    ["ID", pc.cyan(profile.id)],
-    ["Label", pc.bold(profile.label)],
-    ["Backend", backend.label],
-    ["Endpoint", pc.green(profile.baseUrl)],
-    ...(!isManaged ? [
-      ["Model", profile.modelPath ?? "unknown"],
-      ["MMProj", profile.mmprojPath ?? "none"],
-      ["Memory", existsSync(profile.modelPath) ? formatBytes(statSync(profile.modelPath).size) : "unknown"],
-    ] : []),
-    ["Alias", pc.cyan(profile.modelAlias)],
-    ["Pi", piConfigured ? pc.green("configured") : pc.yellow("not synced")],
-  ])));
-
-  if (!isManaged && profile.commandArgv) {
-    console.log("");
-    console.log(pc.bold("llama-server command"));
-    console.log(pc.dim(buildPrettyCommand(profile)));
-  }
-
-  const action = await prompt.choice("Action", [
-    { value: "sync", label: piConfigured ? `${pc.green("✓")} Pi config synced` : "Sync Pi config", hint: piConfigured ? "Already in ~/.pi/agent/models.json" : "Update ~/.pi/agent/models.json" },
-    { value: "run", label: "Run", hint: "Start server + Pi" },
-    ...(isManaged ? [] : [{ value: "server", label: "Server only", hint: "Start server, no harness" }]),
-    { value: "remove", label: "Remove", hint: "Delete profile + Pi config" },
-  ], "sync");
-
-  if (action === "sync") return await syncPiConfig(profile);
-  if (action === "run") return await runProfile(profile);
-  if (action === "server") return await runProfile(profile, { with: "server" });
-  if (action === "remove") return await removeProfileInteractive(profile.id);
-}
-
 async function removeProfileInteractive(id) {
   const profile = await readProfile(id);
   if (!process.stdin.isTTY) {
@@ -981,7 +1013,7 @@ function printHelp() {
   console.log(`${pc.bold("offgrid-ai")} — privacy-first local LLM runner
 
 Usage:
-  offgrid-ai            Pick a model and run it
+  offgrid-ai            Command center: inspect, set up, run, benchmark, or remove models
   offgrid-ai status     Show running local models
   offgrid-ai stop       Stop a running server (or: offgrid-ai stop <id>)
   offgrid-ai uninstall  Remove offgrid-ai, clean up PATH, optionally keep profiles