@rudderjs/ai 1.18.1 → 1.18.3

package/dist/memory-orm/index.js

@@ -1,5 +1,187 @@
-// @rudderjs/ai is deprecated. The AI engine now lives in @gemstack/ai-sdk.
-// This module re-exports it for backwards compatibility; import from
-// '@gemstack/ai-sdk/memory-orm' directly in new code.
-export * from '@gemstack/ai-sdk/memory-orm';
+/**
+ * `@rudderjs/ai/memory-orm` — ORM-backed {@link UserMemory} for #A4 Phase 4.
+ *
+ * Stores per-user facts in a `UserMemory` table via the registered
+ * `@rudderjs/orm` adapter (Prisma today; Drizzle as well once the user's
+ * tables are wired). Drop-in alongside Phase 1's in-process
+ * `MemoryUserMemory`.
+ *
+ * Wire it from your AI config:
+ *
+ * ```ts
+ * // config/ai.ts
+ * import type { AiConfig } from '@gemstack/ai-sdk'
+ * import { OrmUserMemory } from '@rudderjs/ai/memory-orm'
+ *
+ * export default {
+ *   default: 'anthropic/claude-sonnet-4-5',
+ *   providers: { ... },
+ *   memory: new OrmUserMemory(),
+ * } satisfies AiConfig
+ * ```
+ *
+ * The schema lives at `@rudderjs/ai/memory-orm`'s {@link userMemoryPrismaSchema}
+ * — copy it into your Prisma schema. The optional `embedding Bytes?`
+ * column is shipped here in Phase 4 (intentionally nullable) so Phase 5's
+ * `EmbeddingUserMemory` can populate it without forcing an additive
+ * migration.
+ */
+import { Model } from '@rudderjs/orm';
+// ─── ORM Model ────────────────────────────────────────────
+/**
+ * The Model row backing {@link OrmUserMemory}. Exposed so apps that
+ * want their own queries (admin views, audit dumps) can use the
+ * familiar `UserMemoryRecord.where(...).get()` instead of routing
+ * everything through the {@link UserMemory} interface.
+ *
+ * Tags persist as a JSON-encoded string in the `tags` column — both
+ * Prisma's portable `String?` and Drizzle's `text` work without
+ * needing native array columns. The {@link UserMemory.recall} path
+ * filters tags in JavaScript for the same reason.
+ *
+ * The `embedding Bytes?` column is in the schema as of Phase 4
+ * (nullable) so `@rudderjs/ai/memory-embedding`'s `EmbeddingUserMemory`
+ * (Phase 5) writes the Float32-packed vector here on `remember()` and
+ * reads it for cosine recall. `OrmUserMemory` ignores it — the
+ * column stays `null` for any row stored without the embedding
+ * composer.
+ */
+export class UserMemoryRecord extends Model {
+    static table = 'userMemory';
+    static fillable = ['userId', 'fact', 'tags', 'score', 'embedding'];
+    /** Parsed tags array; empty when nothing was stored. */
+    getTags() {
+        if (this.tags == null || this.tags === '')
+            return [];
+        try {
+            const parsed = JSON.parse(this.tags);
+            return Array.isArray(parsed) ? parsed.filter(t => typeof t === 'string') : [];
+        }
+        catch {
+            return [];
+        }
+    }
+}
+// ─── UserMemory adapter ───────────────────────────────────
+/**
+ * `UserMemory` implementation that persists rows to the registered
+ * ORM adapter. Designed for production use — the in-process
+ * `MemoryUserMemory` is for tests and dev.
+ *
+ * Adapter coverage:
+ * - Prisma — works out of the box; copy {@link userMemoryPrismaSchema}
+ *   into your schema.
+ * - Drizzle — works once you define a table matching the schema's
+ *   columns and register it via `tables: { userMemory: <table> }` on
+ *   the `drizzle()` config.
+ *
+ * Recall semantics: case-insensitive **token-OR-LIKE** matching against
+ * the `fact` column. The query is tokenized on non-alphanumeric
+ * boundaries (≥3-char tokens) and any row whose `fact` matches at
+ * least one token via `LIKE %tok%` is returned. Mirrors Phase 1's
+ * `MemoryUserMemory.recall()` behavior so the two backends are
+ * swap-compatible. Tag scope is applied JS-side after fetch — pushing
+ * tag-array filtering into the WHERE is adapter-specific and lands in a
+ * follow-up.
+ */
+export class OrmUserMemory {
+    async remember(userId, fact, opts) {
+        const data = { userId, fact };
+        if (opts?.tags !== undefined)
+            data['tags'] = JSON.stringify(opts.tags);
+        if (opts?.score !== undefined)
+            data['score'] = opts.score;
+        const created = await UserMemoryRecord.create(data);
+        return rowToEntry(created);
+    }
+    async recall(userId, query, opts) {
+        const tokens = tokenize(query);
+        let q = UserMemoryRecord.where('userId', userId);
+        if (tokens.size > 0) {
+            const tokenList = [...tokens];
+            q = q.whereGroup(g => {
+                for (const tok of tokenList)
+                    g.orWhere('fact', 'LIKE', `%${tok}%`);
+            });
+        }
+        const rows = await q.orderBy('createdAt', 'ASC').get();
+        const entries = rows.map(rowToEntry).filter(e => matchesTags(e, opts?.tags));
+        return capLimit(entries, opts?.limit);
+    }
+    async forget(userId, factId) {
+        const row = await UserMemoryRecord.where('id', factId).where('userId', userId).first();
+        if (row)
+            await row.delete();
+    }
+    async list(userId, opts) {
+        const rows = await UserMemoryRecord.where('userId', userId).orderBy('createdAt', 'ASC').get();
+        const entries = rows.map(rowToEntry).filter(e => matchesTags(e, opts?.tags));
+        return capLimit(entries, opts?.limit);
+    }
+    async forgetAll(userId) {
+        await UserMemoryRecord.where('userId', userId).deleteAll();
+    }
+}
+// ─── Schema reference ─────────────────────────────────────
+/**
+ * Reference Prisma schema for `OrmUserMemory`. Copy into your
+ * `prisma/schema/<file>.prisma` (or paste alongside an existing
+ * model). The `embedding Bytes?` column is intentionally nullable so
+ * Phase 5's `EmbeddingUserMemory` becomes additive — no schema
+ * migration when you upgrade.
+ *
+ * SQLite stores `Bytes` as `BLOB`; Postgres stores it as `bytea`.
+ * Both work for the dot-product implementation Phase 5 will use.
+ */
+export const userMemoryPrismaSchema = `model UserMemory {
+  id        String   @id @default(cuid())
+  userId    String
+  fact      String
+  /// JSON-encoded \`string[]\` of tags, or null
+  tags      String?
+  /// Confidence score in [0, 1] — extract sets this from the model's self-rating
+  score     Float?
+  /// Phase 5 — vector embedding for cosine recall (nullable so Phase 4 ignores it)
+  embedding Bytes?
+  createdAt DateTime @default(now())
+  updatedAt DateTime @updatedAt
+
+  @@index([userId])
+}
+`;
+// ─── Helpers ──────────────────────────────────────────────
+function rowToEntry(row) {
+    const tags = row.getTags();
+    const out = {
+        id: row.id,
+        userId: row.userId,
+        fact: row.fact,
+        createdAt: row.createdAt,
+    };
+    if (tags.length > 0)
+        out.tags = tags;
+    if (row.score != null)
+        out.score = row.score;
+    if (row.updatedAt != null)
+        out.updatedAt = row.updatedAt;
+    return out;
+}
+function tokenize(s) {
+    const out = new Set();
+    for (const tok of s.toLowerCase().split(/[^a-z0-9]+/)) {
+        if (tok.length >= 3)
+            out.add(tok);
+    }
+    return out;
+}
+function matchesTags(entry, wanted) {
+    if (!wanted || wanted.length === 0)
+        return true;
+    if (!entry.tags || entry.tags.length === 0)
+        return false;
+    return wanted.every(t => entry.tags.includes(t));
+}
+function capLimit(items, limit) {
+    return limit !== undefined && limit > 0 ? items.slice(0, limit) : items;
+}
 //# sourceMappingURL=index.js.map

package/dist/memory-orm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/memory-orm/index.ts"],"names":[],"mappings":"AAAA,2EAA2E;AAC3E,qEAAqE;AACrE,sDAAsD;AACtD,cAAc,6BAA6B,CAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/memory-orm/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AAEH,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAA;AAMrC,6DAA6D;AAE7D;;;;;;;;;;;;;;;;;GAiBG;AACH,MAAM,OAAO,gBAAiB,SAAQ,KAAK;IACzC,MAAM,CAAU,KAAK,GAAG,YAAY,CAAA;IAEpC,MAAM,CAAU,QAAQ,GAAG,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,CAAC,CAAA;IAkB3E,wDAAwD;IACxD,OAAO;QACL,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,KAAK,EAAE;YAAE,OAAO,EAAE,CAAA;QACpD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAY,CAAA;YAC/C,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAA;QAC/E,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAA;QACX,CAAC;IACH,CAAC;;AAGH,6DAA6D;AAE7D;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,OAAO,aAAa;IACxB,KAAK,CAAC,QAAQ,CACZ,MAAc,EACd,IAAc,EACd,IAA2C;QAE3C,MAAM,IAAI,GAA4B,EAAE,MAAM,EAAE,IAAI,EAAE,CAAA;QACtD,IAAI,IAAI,EAAE,IAAI,KAAM,SAAS;YAAE,IAAI,CAAC,MAAM,CAAC,GAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACxE,IAAI,IAAI,EAAE,KAAK,KAAK,SAAS;YAAE,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAA;QAEzD,MAAM,OAAO,GAAG,MAAM,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAgC,CAAA;QAClF,OAAO,UAAU,CAAC,OAAO,CAAC,CAAA;IAC5B,CAAC;IAED,KAAK,CAAC,MAAM,CACV,MAAc,EACd,KAAc,EACd,IAA2C;QAE3C,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAA;QAE9B,IAAI,CAAC,GAAG,gBAAgB,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;QAChD,IAAI,MAAM,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACpB,MAAM,SAAS,GAAG,CAAC,GAAG,MAAM,CAAC,CAAA;YAC7B,CAAC,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE;gBACnB,KAAK,MAAM,GAAG,IAAI,SAAS;oBAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,GAAG,GAAG,CAAC,CAAA;YACpE,CAAC,CAAC,CAAA;QACJ,CAAC;QAED,MAAM,IAAI,GAAM,MAAM,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC,GAAG,EAAmC,CAAA;QAC1F,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAA;QAC5E,OAAO,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,CAAA;IACvC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAc,EAAE,MAAc;QACzC,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,KAAK,EAAwC,CAAA;QAC5H,IAAI,GAAG;YAAE,MAAM,GAAG,CAAC,MAAM,EAAE,CAAA;IAC7B,CAAC;IAED,KAAK,CAAC,IAAI,CACR,MAAc,EACd,IAA2C;QAE3C,MAAM,IAAI,GAAM,MAAM,gBAAgB,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC,GAAG,EAAmC,CAAA;QACjI,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAA;QAC5E,OAAO,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,CAAA;IACvC,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAc;QAC5B,MAAM,gBAAgB,CAAC,KAAK,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,SAAS,EAAE,CAAA;IAC5D,CAAC;CACF;AAED,6DAA6D;AAE7D;;;;;;;;;GASG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG;;;;;;;;;;;;;;;CAerC,CAAA;AAED,6DAA6D;AAE7D,SAAS,UAAU,CAAC,GAAqB;IACvC,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,EAAE,CAAA;IAC1B,MAAM,GAAG,GAAgB;QACvB,EAAE,EAAS,GAAG,CAAC,EAAE;QACjB,MAAM,EAAK,GAAG,CAAC,MAAM;QACrB,IAAI,EAAO,GAAG,CAAC,IAAI;QACnB,SAAS,EAAE,GAAG,CAAC,SAAS;KACzB,CAAA;IACD,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC;QAAS,GAAG,CAAC,IAAI,GAAQ,IAAI,CAAA;IAChD,IAAI,GAAG,CAAC,KAAK,IAAI,IAAI;QAAO,GAAG,CAAC,KAAK,GAAO,GAAG,CAAC,KAAK,CAAA;IACrD,IAAI,GAAG,CAAC,SAAS,IAAI,IAAI;QAAG,GAAG,CAAC,SAAS,GAAG,GAAG,CAAC,SAAS,CAAA;IACzD,OAAO,GAAG,CAAA;AACZ,CAAC;AAED,SAAS,QAAQ,CAAC,CAAS;IACzB,MAAM,GAAG,GAAG,IAAI,GAAG,EAAU,CAAA;IAC7B,KAAK,MAAM,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC;QACtD,IAAI,GAAG,CAAC,MAAM,IAAI,CAAC;YAAE,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;IACnC,CAAC;IACD,OAAO,GAAG,CAAA;AACZ,CAAC;AAED,SAAS,WAAW,CAAC,KAAkB,EAAE,MAA4B;IACnE,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAA;IAC/C,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,KAAK,CAAA;IACxD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAA;AACnD,CAAC;AAED,SAAS,QAAQ,CAAI,KAAU,EAAE,KAAyB;IACxD,OAAO,KAAK,KAAK,SAAS,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAA;AACzE,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rudderjs/ai",
-  "version": "1.18.1",
+  "version": "1.18.3",
   "description": "[Deprecated] Moved to @gemstack/ai-sdk. This package now re-exports it for backwards compatibility.",
   "rudderjs": {
     "provider": "AiProvider",
@@ -93,15 +93,20 @@
     }
   },
   "dependencies": {
-    "@gemstack/ai-sdk": "^0.2.0"
+    "@gemstack/ai-mcp": "^0.1.0",
+    "@gemstack/ai-sdk": "^0.3.0"
   },
   "peerDependencies": {
     "@modelcontextprotocol/sdk": "^1.29.0",
     "react": ">=19.2.0",
-    "@rudderjs/core": "^1.13.3",
-    "@rudderjs/orm": "^1.22.0"
+    "@rudderjs/console": "^1.4.3",
+    "@rudderjs/orm": "^1.22.0",
+    "@rudderjs/core": "^1.13.3"
   },
   "peerDependenciesMeta": {
+    "@rudderjs/console": {
+      "optional": true
+    },
     "@rudderjs/core": {
       "optional": true
     },
@@ -123,12 +128,14 @@
     "openai": ">=4.70.0"
   },
   "devDependencies": {
-    "@gemstack/ai-sdk": "^0.2.0",
+    "@gemstack/ai-mcp": "^0.1.0",
+    "@gemstack/ai-sdk": "^0.3.0",
     "@modelcontextprotocol/sdk": "^1.29.0",
     "@types/node": "^20.0.0",
     "@types/react": "^19.2.0",
     "react": "^19.2.0",
     "typescript": "^5.4.0",
+    "@rudderjs/console": "^1.4.3",
     "@rudderjs/core": "^1.13.3",
     "@rudderjs/orm": "^1.22.0"
   },
@@ -137,6 +144,7 @@
     "build": "tsc -p tsconfig.build.json",
     "dev": "tsc -p tsconfig.build.json --watch",
     "typecheck": "tsc --noEmit",
-    "clean": "rm -rf dist"
+    "test": "tsc -p tsconfig.test.json && node --test \"dist-test/**/*.test.js\"",
+    "clean": "rm -rf dist dist-test"
   }
 }