@pentatonic-ai/ai-agent-sdk 0.4.6 → 0.4.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pentatonic-ai/ai-agent-sdk",
-  "version": "0.4.6",
+  "version": "0.4.8",
   "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

@@ -25,6 +25,14 @@ import { createMemorySystem } from "./index.js";
 
 const { Pool } = pg;
 
+// Prevent unhandled rejections from killing the process
+process.on("uncaughtException", (err) => {
+  process.stderr.write(`[memory-server] Uncaught: ${err.message}\n`);
+});
+process.on("unhandledRejection", (err) => {
+  process.stderr.write(`[memory-server] Unhandled rejection: ${err?.message || err}\n`);
+});
+
 const CLIENT_ID = process.env.CLIENT_ID || "default";
 
 function createMemory() {
@@ -250,25 +258,19 @@ async function main() {
 
       if (url.pathname === "/search" && req.method === "POST") {
         try {
-          const results = await memory.search(body.query || "", {
+          // Use text search by default (fast, no external dependencies).
+          // Vector search available via ?mode=vector if embeddings are working.
+          const useVector = url.searchParams.get("mode") === "vector";
+          const searchFn = useVector ? memory.search : memory.textSearch;
+          const results = await searchFn(body.query || "", {
             clientId: CLIENT_ID,
             limit: body.limit || 5,
             minScore: body.min_score || 0.3,
           });
           res.end(JSON.stringify({ results }));
         } catch (err) {
-          process.stderr.write(`[memory-server] Search error: ${err.message}\n`);
-          // Fall back to text search
-          try {
-            const results = await memory.textSearch(body.query || "", {
-              clientId: CLIENT_ID,
-              limit: body.limit || 5,
-            });
-            res.end(JSON.stringify({ results }));
-          } catch (err2) {
-            process.stderr.write(`[memory-server] Text search also failed: ${err2.message}\n`);
-            res.end(JSON.stringify({ results: [], error: err2.message }));
-          }
+          process.stderr.write(`[memory-server] Search error: ${err.message}\n${err.stack}\n`);
+          res.end(JSON.stringify({ results: [], error: err.message }));
         }
       } else if (url.pathname === "/store" && req.method === "POST") {
         try {
@@ -283,7 +285,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" }));