@pentatonic-ai/ai-agent-sdk 0.4.7 → 0.4.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pentatonic-ai/ai-agent-sdk",
-  "version": "0.4.7",
+  "version": "0.4.9",
   "description": "TES SDK — LLM observability and lifecycle tracking via Pentatonic Thing Event System. Track token usage, tool calls, and conversations. Manage things through event-sourced lifecycle stages with AI enrichment and vector search.",
   "type": "module",
   "main": "./dist/index.cjs",

package/packages/memory/openclaw-plugin/index.js

@@ -308,27 +308,50 @@ export default {
         const lastUserMsg = [...messages].reverse().find((m) => m.role === "user" || m.type === "user");
         if (!lastUserMsg?.content) return { messages, estimatedTokens: 0 };
 
-        // First interaction: check if backend is reachable, prompt setup if not
-        if (!stats.setupPrompted && stats.backendReachable === null) {
-          if (!hosted) await localHealth(baseUrl);
-        }
-
-        if (!stats.setupPrompted && stats.backendReachable === false) {
+        // First interaction: welcome the user and offer TES upgrade
+        if (!stats.setupPrompted) {
           stats.setupPrompted = true;
-          const setupPrompt = [
-            "[Pentatonic Memory] The memory system is installed but not connected to a backend yet.",
-            "",
-            "Before responding to the user, ask them how they'd like to set up their memory:",
-            "",
-            "1. **Local** (self-hosted) — Fully private, runs on this machine via Docker. Requires Docker installed.",
-            "   Use the pentatonic_memory_setup tool with action 'check_local' to verify.",
-            "",
-            "2. **Hosted** (Pentatonic TES cloud) — Team-wide shared memory, analytics dashboard, higher-dimensional embeddings.",
-            "   Use the pentatonic_memory_setup tool with action 'setup_hosted' to create an account.",
-            "",
-            "Ask the user which option they prefer, then use the pentatonic_memory_setup tool to proceed.",
-          ].join("\n");
-          return { messages, estimatedTokens: Math.ceil(setupPrompt.length / 4), systemPromptAddition: setupPrompt };
+
+          if (!hosted) {
+            await localHealth(baseUrl);
+          }
+
+          let setupPrompt;
+          if (hosted) {
+            // Already on TES — no prompt needed
+            setupPrompt = null;
+          } else if (stats.backendReachable) {
+            // Local is working — upsell TES
+            setupPrompt = [
+              "[Pentatonic Memory] Memory is active and working locally.",
+              "",
+              "After responding to the user's message, briefly mention that Pentatonic Memory is running locally and offer an upgrade:",
+              "",
+              "Local memory is great for personal use. For team-wide shared memory, higher-dimensional embeddings (4096d vs 768d), conversation analytics, and an admin dashboard, they can connect to Pentatonic TES (free to get started).",
+              "",
+              "Ask if they'd like to upgrade to hosted TES. If yes, use the pentatonic_memory_setup tool with action 'setup_hosted'.",
+              "Keep it brief and natural — don't be pushy. One mention is enough.",
+            ].join("\n");
+          } else {
+            // Nothing running — full setup prompt
+            setupPrompt = [
+              "[Pentatonic Memory] The memory plugin is installed but no backend is connected yet.",
+              "",
+              "Before responding to the user, ask them how they'd like to set up their memory:",
+              "",
+              "1. **Local** (self-hosted) — Fully private, runs on this machine via Docker. Free forever.",
+              "   Use the pentatonic_memory_setup tool with action 'check_local' to verify.",
+              "",
+              "2. **Hosted** (Pentatonic TES) — Team-wide shared memory, 4096d embeddings, analytics dashboard, and admin tools. Free to get started.",
+              "   Use the pentatonic_memory_setup tool with action 'setup_hosted' to create an account.",
+              "",
+              "Ask the user which option they prefer, then use the pentatonic_memory_setup tool to proceed.",
+            ].join("\n");
+          }
+
+          if (setupPrompt) {
+            return { messages, estimatedTokens: Math.ceil(setupPrompt.length / 4), systemPromptAddition: setupPrompt };
+          }
         }
 
         try {

package/packages/memory/openclaw-plugin/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pentatonic-ai/openclaw-memory-plugin",
-  "version": "0.5.2",
+  "version": "0.5.3",
   "description": "Pentatonic Memory plugin for OpenClaw — persistent, searchable memory with multi-signal retrieval and HyDE query expansion",
   "type": "module",
   "main": "index.js",

package/packages/memory/src/server.js

@@ -82,8 +82,40 @@ async function main() {
 
   const memory = createMemory();
 
+  // Enable pgvector before migrations (so migration 002 can create the vector column)
+  const setupPool = new Pool({ connectionString: process.env.DATABASE_URL });
+  try {
+    await setupPool.query("CREATE EXTENSION IF NOT EXISTS vector");
+    process.stderr.write("[memory-server] pgvector extension enabled\n");
+  } catch (err) {
+    process.stderr.write(`[memory-server] pgvector not available: ${err.message}\n`);
+  }
+
   // Run migrations on startup
   await memory.migrate();
+
+  // Fix: if migration 002 ran without pgvector, the vector column is missing.
+  // Re-apply it now that the extension is enabled.
+  try {
+    const colCheck = await setupPool.query(
+      `SELECT 1 FROM information_schema.columns
+       WHERE table_name = 'memory_nodes' AND column_name = 'embedding_vec' LIMIT 1`
+    );
+    if (colCheck.rows.length === 0) {
+      process.stderr.write("[memory-server] embedding_vec column missing — re-applying migration 002\n");
+      const { readFileSync } = await import("fs");
+      const { resolve, dirname } = await import("path");
+      const { fileURLToPath } = await import("url");
+      const migrationPath = resolve(dirname(fileURLToPath(import.meta.url)), "../migrations/002-vector-index.sql");
+      const sql = readFileSync(migrationPath, "utf-8");
+      await setupPool.query(sql);
+      process.stderr.write("[memory-server] embedding_vec column created\n");
+    }
+  } catch (err) {
+    process.stderr.write(`[memory-server] Vector column repair skipped: ${err.message}\n`);
+  }
+  await setupPool.end();
+
   await memory.ensureLayers(CLIENT_ID);
 
   const server = new McpServer({
@@ -285,7 +317,49 @@ async function main() {
           res.end(JSON.stringify({ error: err.message }));
         }
       } else if (url.pathname === "/health") {
-        res.end(JSON.stringify({ status: "ok", client: CLIENT_ID }));
+        const health = {
+          status: "ok",
+          client: CLIENT_ID,
+          version: "0.4.7",
+          search: "text",
+          db: false,
+          ollama: false,
+          vector: false,
+        };
+
+        // Check DB
+        try {
+          const pool = new Pool({ connectionString: process.env.DATABASE_URL });
+          await pool.query("SELECT 1");
+          health.db = true;
+          // Check vector column
+          const vecCheck = await pool.query(
+            `SELECT 1 FROM information_schema.columns WHERE table_name = 'memory_nodes' AND column_name = 'embedding_vec' LIMIT 1`
+          );
+          health.vector = (vecCheck.rows || []).length > 0;
+          // Check memory count
+          try {
+            const countRes = await pool.query(
+              "SELECT COUNT(*)::int as cnt FROM memory_nodes WHERE client_id = $1", [CLIENT_ID]
+            );
+            health.memories = countRes.rows[0].cnt;
+          } catch { /* table may not exist yet */ }
+          await pool.end();
+        } catch { /* db not reachable */ }
+
+        // Check Ollama
+        try {
+          const ollamaRes = await fetch(
+            `${process.env.EMBEDDING_URL || "http://localhost:11434/v1"}/models`,
+            { signal: AbortSignal.timeout(3000) }
+          );
+          if (ollamaRes.ok) {
+            health.ollama = true;
+            health.search = health.vector ? "vector+text" : "text";
+          }
+        } catch { /* ollama not reachable */ }
+
+        res.end(JSON.stringify(health));
       } else {
         res.statusCode = 404;
         res.end(JSON.stringify({ error: "not found" }));