@pentatonic-ai/ai-agent-sdk 0.4.0-beta.2 → 0.4.0

package/bin/cli.js

@@ -190,13 +190,27 @@ async function setupLocalMemory() {
     llmSpinner.fail(`Failed to pull ${llmModel}. Run manually: docker compose exec ollama ollama pull ${llmModel}`);
   }
 
-  // Write local config
+  // Write local config (warn if hosted config exists)
   const configDir = join(homedir(), ".claude-pentatonic");
   if (!existsSync(configDir)) {
     mkdirSync(configDir, { recursive: true });
   }
 
   const configPath = join(configDir, "tes-memory.local.md");
+  if (existsSync(configPath)) {
+    const existing = readFileSync(configPath, "utf-8");
+    if (existing.includes("tes_endpoint") && !existing.includes("mode: local")) {
+      console.log("\n  ⚠ Hosted TES config detected. Switching to local mode will");
+      console.log("  disable hosted memory. To restore, run: npx @pentatonic-ai/ai-agent-sdk init\n");
+      const confirm = await ask("  Switch to local mode? (y/n): ");
+      if (confirm.toLowerCase() !== "y") {
+        console.log("  Cancelled. Hosted config unchanged.\n");
+        rl.close();
+        return;
+      }
+    }
+  }
+
   writeFileSync(
     configPath,
     `---

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pentatonic-ai/ai-agent-sdk",
-  "version": "0.4.0-beta.2",
+  "version": "0.4.0",
   "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",
@@ -29,7 +29,7 @@
   "scripts": {
     "prepare": "node build.js",
     "build": "node build.js",
-    "test": "NODE_OPTIONS='--experimental-vm-modules' jest --config jest.config.cjs",
+    "test": "NODE_OPTIONS='--experimental-vm-modules' jest --config jest.config.cjs --testPathIgnorePatterns='/e2e/'",
     "test:watch": "NODE_OPTIONS='--experimental-vm-modules' jest --config jest.config.cjs --watch",
     "prepublishOnly": "npm run build && npm test"
   },
@@ -63,6 +63,7 @@
   },
   "devDependencies": {
     "@jest/globals": "^29.7.0",
-    "jest": "^29.7.0"
+    "jest": "^29.7.0",
+    "pg": "^8.20.0"
   }
 }

package/packages/memory/src/migrate.js

@@ -5,22 +5,26 @@
  * in a schema_migrations table.
  */
 
-import { readFileSync, readdirSync } from "fs";
-import { resolve, dirname } from "path";
-import { fileURLToPath } from "url";
-
-const __dirname = dirname(fileURLToPath(import.meta.url));
-const MIGRATIONS_DIR = resolve(__dirname, "../migrations");
-
 /**
  * Apply all pending database migrations.
  *
+ * Note: This function uses Node.js fs/path APIs and will not work in
+ * Cloudflare Workers. TES uses its own migration system instead.
+ *
  * @param {Function} db - Database query function: (sql, params) => {rows}
  * @param {object} [opts]
  * @param {Function} [opts.logger] - Optional logger
+ * @param {string} [opts.migrationsDir] - Override migrations directory
  * @returns {Promise<{applied: string[], total: number}>}
  */
 export async function migrate(db, opts = {}) {
+  const { readFileSync, readdirSync } = await import("fs");
+  const { resolve, dirname } = await import("path");
+  const { fileURLToPath } = await import("url");
+
+  const migrationsDir =
+    opts.migrationsDir ||
+    resolve(dirname(fileURLToPath(import.meta.url)), "../migrations");
   const log = opts.logger || (() => {});
 
   // Ensure migrations tracking table exists
@@ -40,7 +44,7 @@ export async function migrate(db, opts = {}) {
   const appliedSet = new Set((applied.rows || []).map((r) => r.name));
 
   // Read migration files
-  const files = readdirSync(MIGRATIONS_DIR)
+  const files = readdirSync(migrationsDir)
     .filter((f) => f.endsWith(".sql"))
     .sort();
 
@@ -49,7 +53,7 @@ export async function migrate(db, opts = {}) {
   for (const file of files) {
     if (appliedSet.has(file)) continue;
 
-    const sql = readFileSync(resolve(MIGRATIONS_DIR, file), "utf-8");
+    const sql = readFileSync(resolve(migrationsDir, file), "utf-8");
     log(`Applying migration: ${file}`);
 
     await db(sql, []);

package/packages/memory/src/search.js

@@ -174,6 +174,6 @@ function mapRow(row) {
       : null,
     created_at: row.created_at ? new Date(row.created_at).toISOString() : null,
     updated_at: row.updated_at ? new Date(row.updated_at).toISOString() : null,
-    similarity: row.final_score ? parseFloat(row.final_score) : undefined,
+    similarity: parseFloat(row.final_score || row.confidence || 0),
   };
 }