@shahmilsaari/memory-core 0.2.0 → 0.2.3

package/dist/cli.js

@@ -530,9 +530,18 @@ import { join as join2 } from "path";
 var localEnv = join2(process.cwd(), ".memory-core.env");
 config({ path: existsSync2(localEnv) ? localEnv : join2(process.cwd(), ".env") });
 var Config = {
-  databaseUrl: process.env.DATABASE_URL ?? "",
-  ollamaUrl: process.env.OLLAMA_URL ?? "http://localhost:11434",
-  ollamaModel: process.env.OLLAMA_MODEL ?? "nomic-embed-text"
+  get databaseUrl() {
+    return process.env.DATABASE_URL ?? "";
+  },
+  get ollamaUrl() {
+    return process.env.OLLAMA_URL ?? "http://localhost:11434";
+  },
+  get ollamaModel() {
+    return process.env.OLLAMA_MODEL ?? "nomic-embed-text";
+  },
+  get chatModel() {
+    return process.env.OLLAMA_CHAT_MODEL ?? "llama3.2";
+  }
 };
 
 // src/embedding.ts
@@ -1157,7 +1166,9 @@ function startWatch(options = {}) {
 }
 
 // src/cli.ts
-function printBanner(projectName, agentCount) {
+function printBanner(projectName, agentCount, status) {
+  const pg2 = status ? status.postgresOk ? chalk3.green("  \u2713 PostgreSQL ") + chalk3.bold("connected") : chalk3.red("  \u2717 PostgreSQL ") + chalk3.bold("not connected \u2014 check DATABASE_URL") : chalk3.green("  \u2713 Memory    ") + chalk3.bold("PostgreSQL + pgvector ready");
+  const ol = status ? status.ollamaOk ? chalk3.green("  \u2713 Ollama    ") + chalk3.bold(`connected  (model: ${status.chatModel})`) : chalk3.red("  \u2717 Ollama    ") + chalk3.bold("not running \u2014 start with: ollama serve") : null;
   const lines = [
     "",
     chalk3.cyan("   \u2588\u2588\u2588\u2588\u2588\u2557 \u2588\u2588\u2588\u2588\u2588\u2588\u2557  \u2588\u2588\u2588\u2588\u2588\u2588\u2557\u2588\u2588\u2557  \u2588\u2588\u2557\u2588\u2588\u2588\u2557   \u2588\u2588\u2588\u2557\u2588\u2588\u2557\u2588\u2588\u2588\u2557   \u2588\u2588\u2557\u2588\u2588\u2588\u2588\u2588\u2588\u2557 "),
@@ -1171,7 +1182,8 @@ function printBanner(projectName, agentCount) {
     "",
     chalk3.green(`  \u2713 Project   `) + chalk3.bold(projectName),
     chalk3.green(`  \u2713 Agents    `) + chalk3.bold(`${agentCount} AI agents configured`),
-    chalk3.green(`  \u2713 Memory    `) + chalk3.bold("PostgreSQL + pgvector ready"),
+    pg2,
+    ...ol ? [ol] : [],
     "",
     chalk3.dim("  \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500"),
     chalk3.dim("  Built by ") + chalk3.bold.white("Shahmil Saari"),
@@ -1186,6 +1198,35 @@ function printBanner(projectName, agentCount) {
   ];
   lines.forEach((l) => console.log(l));
 }
+async function checkConnections(dbUrl, ollamaUrl, chatModel) {
+  const spinner = ora("Checking connections\u2026").start();
+  let postgresOk = false;
+  let ollamaOk = false;
+  try {
+    const { Pool: Pool2 } = (await import("pg")).default;
+    const testPool = new Pool2({ connectionString: dbUrl, connectionTimeoutMillis: 5e3 });
+    await testPool.query("SELECT 1");
+    await testPool.end();
+    postgresOk = true;
+  } catch {
+    postgresOk = false;
+  }
+  try {
+    const res = await fetch(`${ollamaUrl}/api/tags`, { signal: AbortSignal.timeout(5e3) });
+    ollamaOk = res.ok;
+  } catch {
+    ollamaOk = false;
+  }
+  spinner.stop();
+  console.log(
+    postgresOk ? chalk3.green("  \u2713 PostgreSQL") + chalk3.dim(" \u2014 connected") : chalk3.red("  \u2717 PostgreSQL") + chalk3.dim(" \u2014 cannot connect. Check DATABASE_URL and that PostgreSQL is running.")
+  );
+  console.log(
+    ollamaOk ? chalk3.green("  \u2713 Ollama    ") + chalk3.dim(` \u2014 connected (${chatModel})`) : chalk3.red("  \u2717 Ollama    ") + chalk3.dim(" \u2014 not reachable. Run: ollama serve")
+  );
+  console.log();
+  return { postgresOk, ollamaOk, chatModel };
+}
 var { version } = JSON.parse(readFileSync5(new URL("../package.json", import.meta.url), "utf-8"));
 var CONFIG_FILE = ".memory-core.json";
 function readProjectConfig() {
@@ -1208,27 +1249,77 @@ program.command("init").description("Initialize memory-core in the current proje
   const envPath = join6(process.cwd(), ".memory-core.env");
   const hasEnv = existsSync6(envPath) || existsSync6(join6(process.cwd(), ".env")) || !!process.env.DATABASE_URL;
   if (!hasEnv) {
-    console.log(chalk3.dim("  No .memory-core.env found \u2014 let's set up your database connection.\n"));
+    console.log(chalk3.dim("  No .memory-core.env found \u2014 let's set up your connection.\n"));
     const dbUser = process.env.USER ?? process.env.USERNAME ?? "postgres";
-    const dbUrl = await input({
-      message: "PostgreSQL connection URL?",
-      default: `postgresql://${dbUser}@localhost:5432/memory_core`
-    });
-    const ollamaUrl = await input({
-      message: "Ollama URL?",
-      default: "http://localhost:11434"
-    });
-    const chatModelChoice = await select({
-      message: "Which Ollama model for code checking?",
-      choices: [
-        { name: "llama3.2        (fast, 2GB, recommended for most machines)", value: "llama3.2" },
-        { name: "qwen2.5-coder  (better code understanding, 4.7GB)", value: "qwen2.5-coder" },
-        { name: "mistral        (balanced, 4.1GB)", value: "mistral" },
-        { name: "codellama      (code-focused, 3.8GB)", value: "codellama" },
-        { name: "Other (enter manually)", value: "__custom__" }
-      ]
-    });
-    const chatModel = chatModelChoice === "__custom__" ? await input({ message: "Model name (must be available in your Ollama)?", default: "llama3.2" }) : chatModelChoice;
+    let dbUrl = "";
+    while (true) {
+      dbUrl = await input({
+        message: "PostgreSQL connection URL?",
+        default: dbUrl || `postgresql://${dbUser}@localhost:5432/memory_core`
+      });
+      const pgSpinner = ora("  Testing PostgreSQL connection\u2026").start();
+      try {
+        const { Pool: Pool2 } = (await import("pg")).default;
+        const testPool = new Pool2({ connectionString: dbUrl, connectionTimeoutMillis: 5e3 });
+        await testPool.query("SELECT 1");
+        await testPool.end();
+        pgSpinner.succeed(chalk3.green("PostgreSQL connected"));
+        break;
+      } catch (err) {
+        pgSpinner.fail(chalk3.red(`Cannot connect: ${err.message}`));
+        console.log(chalk3.yellow("  Please check that PostgreSQL is running and the URL is correct.\n"));
+      }
+    }
+    let ollamaUrl = "";
+    while (true) {
+      ollamaUrl = await input({
+        message: "Ollama URL?",
+        default: ollamaUrl || "http://localhost:11434"
+      });
+      const ollamaSpinner = ora("  Testing Ollama connection\u2026").start();
+      try {
+        const res = await fetch(`${ollamaUrl}/api/tags`, { signal: AbortSignal.timeout(5e3) });
+        if (!res.ok) throw new Error(`HTTP ${res.status}`);
+        ollamaSpinner.succeed(chalk3.green("Ollama connected"));
+        break;
+      } catch (err) {
+        ollamaSpinner.fail(chalk3.red(`Cannot reach Ollama: ${err.message}`));
+        console.log(chalk3.yellow("  Make sure Ollama is running: ollama serve\n"));
+      }
+    }
+    let chatModel = "";
+    while (true) {
+      const chatModelChoice = await select({
+        message: "Which Ollama model for code checking?",
+        choices: [
+          { name: "llama3.2        (fast, 2GB, recommended for most machines)", value: "llama3.2" },
+          { name: "qwen2.5-coder  (better code understanding, 4.7GB)", value: "qwen2.5-coder" },
+          { name: "mistral        (balanced, 4.1GB)", value: "mistral" },
+          { name: "codellama      (code-focused, 3.8GB)", value: "codellama" },
+          { name: "Other (enter manually)", value: "__custom__" }
+        ]
+      });
+      chatModel = chatModelChoice === "__custom__" ? await input({ message: "Model name?", default: "llama3.2" }) : chatModelChoice;
+      const modelSpinner = ora(`  Checking if ${chatModel} is installed\u2026`).start();
+      try {
+        const res = await fetch(`${ollamaUrl}/api/tags`, { signal: AbortSignal.timeout(5e3) });
+        const data = await res.json();
+        const installed = (data.models ?? []).some(
+          (m) => m.name === chatModel || m.name.startsWith(`${chatModel}:`)
+        );
+        if (installed) {
+          modelSpinner.succeed(chalk3.green(`${chatModel} is installed and ready`));
+          break;
+        } else {
+          modelSpinner.fail(chalk3.red(`${chatModel} is not installed in your Ollama`));
+          console.log(chalk3.yellow(`  Run: ollama pull ${chatModel}    \u2014 or pick a different model.
+`));
+        }
+      } catch {
+        modelSpinner.warn(chalk3.yellow("Could not verify model \u2014 continuing anyway"));
+        break;
+      }
+    }
     const envContent = [
       `DATABASE_URL=${dbUrl}`,
       `OLLAMA_URL=${ollamaUrl}`,
@@ -1347,7 +1438,12 @@ program.command("init").description("Initialize memory-core in the current proje
   );
   writeProjectConfig(config2);
   spinner.succeed(`Generated ${written.length} files`);
-  printBanner(config2.projectName, written.length);
+  const status = await checkConnections(
+    process.env.DATABASE_URL ?? "",
+    process.env.OLLAMA_URL ?? "http://localhost:11434",
+    process.env.OLLAMA_CHAT_MODEL ?? "llama3.2"
+  );
+  printBanner(config2.projectName, written.length, status);
   await closePool();
 });
 program.command("sync").description("Re-pull memories and regenerate all AI agent files").action(async () => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shahmilsaari/memory-core",
-  "version": "0.2.0",
+  "version": "0.2.3",
   "description": "Universal AI memory core — generate AI context files from architecture profiles with RAG support",
   "type": "module",
   "bin": {