@shahmilsaari/memory-core 0.2.0 → 0.2.4

package/README.md

@@ -18,11 +18,10 @@ Without memory-core, every AI agent starts fresh. It doesn't know you're using C
 
 With memory-core:
 
-1. You run `init` once and answer a few questions about your project
-2. memory-core generates config files for every AI agent you use
-3. Those agents read the files and follow your rules — automatically
-4. When you commit code, a hook catches violations before they land in the repo
-5. Watch mode catches violations as you type, not just at commit time
+1. You run `init` once — it verifies your PostgreSQL and Ollama connections, picks your model, and generates config files for every AI agent
+2. Those agents read the files and follow your rules — automatically
+3. Watch mode catches violations as you type, not just at commit time
+4. When you commit, Claude Code automatically checks your code before the commit goes through
 
 ---
 
@@ -106,10 +105,10 @@ ollama serve &
 
 **Windows** — [download from ollama.com](https://ollama.com/download)
 
-Then pull the two models:
+Pull the embedding model. The code-checking model is chosen during `init`:
 ```bash
-ollama pull nomic-embed-text   # used for search
-ollama pull llama3.2           # used to check your code
+ollama pull nomic-embed-text   # required for search
+ollama pull llama3.2           # or whichever model you plan to use
 ```
 
 ---
@@ -153,14 +152,11 @@ CREATE INDEX IF NOT EXISTS memories_scope_idx ON memories (scope);
 # 1. Go to your project
 cd my-api
 
-# 2. Initialize — answers a few questions, generates all config files
+# 2. Initialize — verifies connections, picks your model, generates all config files
 npx @shahmilsaari/memory-core init
 
 # 3. Load 281 predefined best-practice rules
 npx @shahmilsaari/memory-core seed
-
-# 4. Install the pre-commit hook (optional but recommended)
-npx @shahmilsaari/memory-core hook install
 ```
 
 That's it. Every AI agent in your project now has your rules.
@@ -175,15 +171,18 @@ That's it. Every AI agent in your project now has your rules.
 npx @shahmilsaari/memory-core init
 ```
 
-Asks you:
-- Project name
-- Project type (Backend / Frontend / Fullstack)
-- Architecture (Clean Architecture, MVC, React, Vue, etc.)
-- Language
+Walks you through:
+- PostgreSQL connection URL — **tested live, retries until connected**
+- Ollama URL — **tested live, retries until reachable**
+- Code-checking model — **verified to be installed in your Ollama**
+- Project name, type, architecture, language
+- Whether to enable the pre-commit hook (asked during setup — no separate step needed)
 - Whether to enable caveman mode (optional token saver)
 
 Generates config files for every supported AI agent and saves your choices to `.memory-core.json`.
 
+At the end, the banner shows live ✓/✗ status for PostgreSQL and Ollama so you know everything is working.
+
 ---
 
 ### `remember` — Save a decision
@@ -253,7 +252,9 @@ npx @shahmilsaari/memory-core seed --force                     # re-seed existin
 npx @shahmilsaari/memory-core hook install
 ```
 
-Installs a git pre-commit hook. Every time you run `git commit`, your staged files are checked against your architecture rules before the commit goes through.
+Installs a git pre-commit hook in the current project. Every time you run `git commit`, your code is checked against your architecture rules before the commit goes through.
+
+> **Note:** If you use Claude Code, the hook is offered automatically during `init` — no separate step needed.
 
 When a violation is found, the commit is blocked and you see exactly what's wrong and how to fix it:
 
@@ -396,7 +397,7 @@ Pick the one that matches how your project is structured.
 | React | Functional components, hooks, React Query, Zustand |
 | Vue 3 | Composition API, Pinia, composables |
 | Angular | Standalone components, signals, OnPush strategy |
-| Svelte | Svelte 5 runes, SvelteKit load functions |
+| Svelte | Svelte 5 runes, SvelteKit load functions, snippets |
 | Nuxt 3 | Fullstack Vue with SSR |
 | React Native | Mobile apps |
 
@@ -421,9 +422,8 @@ Cuts AI response length by 65–75% by removing filler words. Opt in during `ini
 ```bash
 # Starting a new project
 cd my-api
-npx @shahmilsaari/memory-core init
-npx @shahmilsaari/memory-core seed
-npx @shahmilsaari/memory-core hook install
+npx @shahmilsaari/memory-core init   # verifies connections, picks model, installs hook
+npx @shahmilsaari/memory-core seed   # load 281 best-practice rules
 
 # Made an architectural decision? Save it.
 npx @shahmilsaari/memory-core remember "All auth goes through middleware, never in controllers" \
@@ -435,7 +435,7 @@ npx @shahmilsaari/memory-core sync
 # Not sure how something was decided? Search.
 npx @shahmilsaari/memory-core search "caching strategy"
 
-# Commit code → hook checks it automatically
+# Commit code → Claude Code checks it automatically before committing
 git commit -m "add user endpoint"
 ```
 
@@ -449,8 +449,8 @@ memory-core creates `.memory-core.env` automatically during `init`. You can also
 |---|---|---|---|
 | `DATABASE_URL` | Yes | — | PostgreSQL connection string |
 | `OLLAMA_URL` | No | `http://localhost:11434` | Where Ollama is running |
-| `OLLAMA_MODEL` | No | `nomic-embed-text` | Model used for search |
-| `OLLAMA_CHAT_MODEL` | No | `llama3.2` | Model used for code checking |
+| `OLLAMA_MODEL` | No | `nomic-embed-text` | Model used for search (embeddings) |
+| `OLLAMA_CHAT_MODEL` | No | `llama3.2` | Model used for code checking — chosen during `init` |
 
 ---
 
@@ -483,6 +483,10 @@ It won't — the hook only checks source code files: `.ts .tsx .js .jsx .py .php
 | | Feature |
 |---|---|
 | ✓ | Watch mode — real-time violation alerts on save |
+| ✓ | Model picker — choose your Ollama model during init |
+| ✓ | Connection validation — PostgreSQL and Ollama verified during setup |
+| ✓ | Svelte 5 / SvelteKit profile and 37 rules |
+| ✓ | NestJS profile and 39 rules |
 | | CI/CD — fail PRs that violate your rules |
 | | Violation memory — auto-save what went wrong as a new rule |
 | | Rule analytics — see which rules get broken most |
@@ -494,3 +498,5 @@ It won't — the hook only checks source code files: `.ts .tsx .js .jsx .py .php
 ## License
 
 MIT
+
+*Built by [Shahmil Saari](https://github.com/shahmilsaari)*

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}`,
@@ -1309,6 +1400,10 @@ program.command("init").description("Initialize memory-core in the current proje
       ]
     });
   }
+  const enableHook = await confirm({
+    message: "Enable pre-commit hook? (blocks commits that violate your rules)",
+    default: true
+  });
   const config2 = {
     projectName,
     projectType,
@@ -1347,7 +1442,15 @@ program.command("init").description("Initialize memory-core in the current proje
   );
   writeProjectConfig(config2);
   spinner.succeed(`Generated ${written.length} files`);
-  printBanner(config2.projectName, written.length);
+  if (enableHook) {
+    installHook();
+  }
+  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.4",
   "description": "Universal AI memory core — generate AI context files from architecture profiles with RAG support",
   "type": "module",
   "bin": {