godspeed-agent 0.1.0 → 0.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "godspeed-agent",
-  "version": "0.1.0",
+  "version": "0.1.3",
   "license": "MIT",
   "type": "module",
   "bin": {
@@ -13,6 +13,14 @@
     "LICENSE",
     "package.json"
   ],
+  "homepage": "https://github.com/bhupeshv29/tui-starter#readme",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/bhupeshv29/tui-starter.git"
+  },
+  "bugs": {
+    "url": "https://github.com/bhupeshv29/tui-starter/issues"
+  },
   "scripts": {
     "dev": "bun run src/cli.ts"
   },

package/src/commands/models.ts

@@ -2,6 +2,7 @@ import { Command } from "commander";
 import chalk from "chalk";
 import { loadConfig, saveConfig } from "../lib/config.js";
 import { SUPPORTED_PROVIDERS } from "../lib/providers.js";
+import { getOllamaModels } from "../providers/ollamaModels.js";
 
 export const modelsCommand = new Command("models");
 
@@ -17,16 +18,14 @@ modelsCommand
     }
 
     const provider = SUPPORTED_PROVIDERS.find(
-      p => p.name === config.activeProvider
+      (p) => p.name === config.activeProvider,
     );
 
     if (!provider) return;
 
     for (const model of provider.models) {
       const active =
-        config.activeModel === model
-          ? chalk.green(" (active)")
-          : "";
+        config.activeModel === model ? chalk.green(" (active)") : "";
 
       console.log(`- ${model}${active}`);
     }
@@ -43,7 +42,69 @@ modelsCommand
 
     saveConfig(config);
 
-    console.log(
-      chalk.green(`Active model set to ${model}`)
-    );
-  });
+    console.log(chalk.green(`Active model set to ${model}`));
+  });
+
+modelsCommand
+  .command("sync")
+  .option("--save", "Save discovered models")
+  .description("Sync/list models from active provider")
+  .action(async (options) => {
+    const config = loadConfig();
+
+    if (!config.activeProvider) {
+      console.log(chalk.red("No active provider selected"));
+      return;
+    }
+
+    if (config.activeProvider !== "ollama") {
+      console.log(
+        chalk.yellow(
+          `models sync is currently only supported for ollama. Active provider: ${config.activeProvider}`,
+        ),
+      );
+      return;
+    }
+
+    try {
+      const models = await getOllamaModels();
+
+      if (models.length === 0) {
+        console.log(chalk.yellow("No Ollama models found."));
+        return;
+      }
+
+      console.log(chalk.green("Installed Ollama models:"));
+
+      for (const model of models) {
+        const sizeInGB = (model.size / 1024 / 1024 / 1024).toFixed(2);
+
+        console.log(`- ${model.name} (${sizeInGB} GB)`);
+      }
+      if (options.save) {
+        config.providers.ollama ??= {};
+
+        config.providers.ollama.models = models.map((model) => model.name);
+
+        saveConfig(config);
+
+        console.log(chalk.green(`Saved ${models.length} models to config.`));
+      }
+    } catch (err) {
+      console.log(chalk.red(err instanceof Error ? err.message : String(err)));
+    }
+  });
+
+modelsCommand
+  .command("use")
+  .argument("<model>")
+  .description("Set active model")
+  .action((model) => {
+    const config = loadConfig();
+
+    config.activeModel = model;
+
+    saveConfig(config);
+
+    console.log(chalk.green(`Active model set to ${model}`));
+  });

package/src/lib/config.ts

@@ -10,7 +10,13 @@ export type ProviderConfig = {
 export type AppConfig = {
   activeProvider?: string;
   activeModel?: string;
-  providers: Record<string, ProviderConfig>;
+  providers: Record<
+    string,
+    {
+      apiKey?: string;
+      models?: string[];
+    }
+  >;
 };
 
 const CONFIG_DIR = path.join(os.homedir(), ".tuistarter");

package/src/providers/ollamaModels.ts

@@ -0,0 +1,17 @@
+export async function getOllamaModels() {
+  const res = await fetch("http://localhost:11434/api/tags");
+
+  if (!res.ok) {
+    throw new Error("Could not connect to Ollama");
+  }
+
+  const data: any = await res.json();
+
+  return (
+    data.models?.map((model: any) => ({
+      name: model.name,
+      size: model.size,
+      modifiedAt: model.modified_at,
+    })) ?? []
+  );
+}