@rudderjs/ai 1.4.0 → 1.6.0

package/dist/vector-stores/index.js

@@ -0,0 +1,153 @@
+/**
+ * `VectorStores` — manage hosted vector stores on the registered AI
+ * provider (#B8 Phase 1).
+ *
+ * Today only OpenAI implements `createVectorStores()` on its provider
+ * factory; calls against other providers throw a helpful error pointing
+ * users at `similaritySearch()` over a local pgvector model.
+ *
+ * @example
+ * ```ts
+ * import { VectorStores } from '@rudderjs/ai'
+ *
+ * const store = await VectorStores.create('Knowledge Base')
+ * await store.add({ filePath: './report.pdf', attributes: { author: 'Alice' } })
+ *
+ * const all = await VectorStores.list()
+ * await VectorStores.delete(store.id)
+ * ```
+ *
+ * Provider override per-call via `opts.provider`. The default falls
+ * back to the registered AI default (first entry in the registry, or
+ * `AiRegistry.getDefault()`'s provider name).
+ *
+ * Phase 2 of B8 adds the `fileSearch({ stores })` agent tool that
+ * consumes these stores. Phase 3 wires the local pgvector fallback.
+ */
+import { AiRegistry } from '../registry.js';
+/**
+ * Wrapper around a single hosted vector store. Holds the provider name
+ * + store id so per-store operations don't need to repeat them.
+ */
+export class VectorStore {
+    id;
+    name;
+    createdAt;
+    fileCount;
+    bytesUsed;
+    metadata;
+    /** Provider that owns this store (e.g. `'openai'`). */
+    provider;
+    constructor(info, provider) {
+        this.id = info.id;
+        this.name = info.name;
+        this.createdAt = info.createdAt;
+        this.fileCount = info.fileCount;
+        this.bytesUsed = info.bytesUsed;
+        this.metadata = info.metadata;
+        this.provider = provider;
+    }
+    /**
+     * Attach a file to this store. Either pass an existing provider
+     * `fileId` or a local source (`filePath` or `fileBuffer`) — the
+     * adapter uploads via the Files API first and reuses the returned id.
+     *
+     * Defaults to waiting until the file is fully indexed
+     * (`status === 'completed'`). Pass `wait: false` for fire-and-forget.
+     */
+    async add(opts) {
+        const adapter = AiRegistry.resolveVectorStores(this.provider);
+        return adapter.addFile(this.id, opts);
+    }
+    /** Remove a file from this store. The file remains in the provider's
+     *  Files API (use `AI.files(provider).delete(id)` to fully delete). */
+    async remove(fileId) {
+        const adapter = AiRegistry.resolveVectorStores(this.provider);
+        await adapter.removeFile(this.id, fileId);
+    }
+    /** List all files attached to this store. */
+    async files(opts) {
+        const adapter = AiRegistry.resolveVectorStores(this.provider);
+        const result = await adapter.listFiles(this.id, opts);
+        return result.files;
+    }
+    /** Delete this store. Files attached to it stay in the provider's
+     *  Files API; call `AI.files(provider).delete(id)` to clean those up
+     *  separately. */
+    async delete() {
+        const adapter = AiRegistry.resolveVectorStores(this.provider);
+        await adapter.delete(this.id);
+    }
+}
+/**
+ * Static facade for hosted-vector-store CRUD across all providers
+ * implementing `createVectorStores()`.
+ */
+export class VectorStores {
+    /**
+     * Create a new vector store on the registered (or specified)
+     * provider.
+     */
+    static async create(name, opts = {}) {
+        const provider = resolveProvider(opts.provider);
+        const adapter = AiRegistry.resolveVectorStores(provider);
+        const createOpts = { name };
+        if (opts.metadata)
+            createOpts.metadata = opts.metadata;
+        if (opts.expiresAfter)
+            createOpts.expiresAfter = opts.expiresAfter;
+        const info = await adapter.create(createOpts);
+        return new VectorStore(info, provider);
+    }
+    /**
+     * List vector stores on the registered (or specified) provider.
+     *
+     * `opts.limit` / `opts.after` / `opts.before` map directly to the
+     * provider's pagination cursor — apps wanting a complete list need to
+     * iterate manually.
+     */
+    static async list(opts = {}) {
+        const provider = resolveProvider(opts.provider);
+        const adapter = AiRegistry.resolveVectorStores(provider);
+        const listOpts = {};
+        if (opts.limit !== undefined)
+            listOpts.limit = opts.limit;
+        if (opts.after !== undefined)
+            listOpts.after = opts.after;
+        if (opts.before !== undefined)
+            listOpts.before = opts.before;
+        const result = await adapter.list(listOpts);
+        return result.stores.map(info => new VectorStore(info, provider));
+    }
+    /**
+     * Fetch a vector store by id. Throws when not found (provider-specific
+     * error surface — typically a 404).
+     */
+    static async get(id, opts = {}) {
+        const provider = resolveProvider(opts.provider);
+        const adapter = AiRegistry.resolveVectorStores(provider);
+        const info = await adapter.get(id);
+        return new VectorStore(info, provider);
+    }
+    /**
+     * Delete a vector store by id. Files attached to it stay in the
+     * provider's Files API.
+     */
+    static async delete(id, opts = {}) {
+        const provider = resolveProvider(opts.provider);
+        const adapter = AiRegistry.resolveVectorStores(provider);
+        await adapter.delete(id);
+    }
+}
+/**
+ * Resolve the provider name to dispatch to. Explicit `opts.provider`
+ * wins; otherwise we read the registered AI default and extract the
+ * provider segment (`'openai/text-embedding-3-small'` → `'openai'`).
+ */
+function resolveProvider(explicit) {
+    if (explicit)
+        return explicit;
+    const def = AiRegistry.getDefault();
+    return AiRegistry.parseModelString(def)[0];
+}
+//# sourceMappingURL=index.js.map

package/dist/vector-stores/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/vector-stores/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAA;AAS3C;;;GAGG;AACH,MAAM,OAAO,WAAW;IACb,EAAE,CAAe;IACjB,IAAI,CAAa;IACjB,SAAS,CAAQ;IACjB,SAAS,CAAQ;IACjB,SAAS,CAAoB;IAC7B,QAAQ,CAAqC;IACtD,uDAAuD;IAC9C,QAAQ,CAAS;IAE1B,YAAY,IAAqB,EAAE,QAAgB;QACjD,IAAI,CAAC,EAAE,GAAU,IAAI,CAAC,EAAE,CAAA;QACxB,IAAI,CAAC,IAAI,GAAQ,IAAI,CAAC,IAAI,CAAA;QAC1B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAA;QAC/B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAA;QAC/B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAA;QAC/B,IAAI,CAAC,QAAQ,GAAI,IAAI,CAAC,QAAQ,CAAA;QAC9B,IAAI,CAAC,QAAQ,GAAI,QAAQ,CAAA;IAC3B,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,GAAG,CAAC,IAA2B;QACnC,MAAM,OAAO,GAAG,UAAU,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC7D,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;IACvC,CAAC;IAED;2EACuE;IACvE,KAAK,CAAC,MAAM,CAAC,MAAc;QACzB,MAAM,OAAO,GAAG,UAAU,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC7D,MAAM,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,CAAC,CAAA;IAC3C,CAAC;IAED,6CAA6C;IAC7C,KAAK,CAAC,KAAK,CAAC,IAA6B;QACvC,MAAM,OAAO,GAAG,UAAU,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC7D,MAAM,MAAM,GAAI,MAAM,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAA;QACtD,OAAO,MAAM,CAAC,KAAK,CAAA;IACrB,CAAC;IAED;;sBAEkB;IAClB,KAAK,CAAC,MAAM;QACV,MAAM,OAAO,GAAG,UAAU,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC7D,MAAM,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IAC/B,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,YAAY;IACvB;;;OAGG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAY,EAAE,OAA+C,EAAE;QACjF,MAAM,QAAQ,GAAG,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC/C,MAAM,OAAO,GAAI,UAAU,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAA;QACzD,MAAM,UAAU,GAA6B,EAAE,IAAI,EAAE,CAAA;QACrD,IAAI,IAAI,CAAC,QAAQ;YAAM,UAAU,CAAC,QAAQ,GAAO,IAAI,CAAC,QAAQ,CAAA;QAC9D,IAAI,IAAI,CAAC,YAAY;YAAE,UAAU,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAA;QAClE,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAA;QAC7C,OAAO,IAAI,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAA;IACxC,CAAC;IAED;;;;;;OAMG;IACH,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,OAAuD,EAAE;QACzE,MAAM,QAAQ,GAAG,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC/C,MAAM,OAAO,GAAI,UAAU,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAA;QACzD,MAAM,QAAQ,GAA2B,EAAE,CAAA;QAC3C,IAAI,IAAI,CAAC,KAAK,KAAM,SAAS;YAAE,QAAQ,CAAC,KAAK,GAAI,IAAI,CAAC,KAAK,CAAA;QAC3D,IAAI,IAAI,CAAC,KAAK,KAAM,SAAS;YAAE,QAAQ,CAAC,KAAK,GAAI,IAAI,CAAC,KAAK,CAAA;QAC3D,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;YAAE,QAAQ,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;QAC5D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC3C,OAAO,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAA;IACnE,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAU,EAAE,OAA8B,EAAE;QAC3D,MAAM,QAAQ,GAAG,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC/C,MAAM,OAAO,GAAI,UAAU,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAA;QACzD,MAAM,IAAI,GAAO,MAAM,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QACtC,OAAO,IAAI,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAA;IACxC,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,EAAU,EAAE,OAA8B,EAAE;QAC9D,MAAM,QAAQ,GAAG,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;QAC/C,MAAM,OAAO,GAAI,UAAU,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAA;QACzD,MAAM,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;IAC1B,CAAC;CACF;AAED;;;;GAIG;AACH,SAAS,eAAe,CAAC,QAA4B;IACnD,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAA;IAC7B,MAAM,GAAG,GAAG,UAAU,CAAC,UAAU,EAAE,CAAA;IACnC,OAAO,UAAU,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;AAC5C,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rudderjs/ai",
-  "version": "1.4.0",
+  "version": "1.6.0",
   "description": "AI engine — providers, agents, tools, streaming, middleware",
   "rudderjs": {
     "provider": "AiProvider",
@@ -37,33 +37,72 @@
       "import": "./dist/commands/make-agent.js",
       "types": "./dist/commands/make-agent.d.ts"
     },
+    "./commands/ai-eval": {
+      "import": "./dist/commands/ai-eval.js",
+      "types": "./dist/commands/ai-eval.d.ts"
+    },
     "./observers": {
       "import": "./dist/observers.js",
       "types": "./dist/observers.d.ts"
+    },
+    "./mcp": {
+      "import": "./dist/mcp/index.js",
+      "types": "./dist/mcp/index.d.ts"
+    },
+    "./memory-orm": {
+      "import": "./dist/memory-orm/index.js",
+      "types": "./dist/memory-orm/index.d.ts"
+    },
+    "./budget-orm": {
+      "import": "./dist/budget-orm/index.js",
+      "types": "./dist/budget-orm/index.d.ts"
+    },
+    "./memory-embedding": {
+      "import": "./dist/memory-embedding/index.js",
+      "types": "./dist/memory-embedding/index.d.ts"
+    },
+    "./eval": {
+      "import": "./dist/eval/index.js",
+      "types": "./dist/eval/index.d.ts"
+    },
+    "./computer-use": {
+      "import": "./dist/computer-use/index.js",
+      "types": "./dist/computer-use/index.d.ts"
     }
   },
   "dependencies": {
     "zod": "^4.0.0"
   },
   "peerDependencies": {
-    "@rudderjs/core": "^1.1.3"
+    "@modelcontextprotocol/sdk": "^1.13.0",
+    "@rudderjs/core": "^1.1.3",
+    "@rudderjs/orm": "^1.9.0"
   },
   "peerDependenciesMeta": {
     "@rudderjs/core": {
       "optional": true
+    },
+    "@rudderjs/orm": {
+      "optional": true
+    },
+    "@modelcontextprotocol/sdk": {
+      "optional": true
     }
   },
   "optionalDependencies": {
     "@anthropic-ai/sdk": ">=0.30.0",
     "openai": ">=4.70.0",
     "@google/genai": ">=1.0.0",
-    "cohere-ai": ">=7.0.0"
+    "cohere-ai": ">=7.0.0",
+    "@aws-sdk/client-bedrock-runtime": ">=3.600.0"
   },
   "devDependencies": {
+    "@modelcontextprotocol/sdk": "^1.13.0",
     "@types/node": "^20.0.0",
     "typescript": "^5.4.0",
     "@rudderjs/core": "^1.1.3",
-    "@rudderjs/console": "^1.0.1"
+    "@rudderjs/console": "^1.0.1",
+    "@rudderjs/orm": "^1.9.0"
   },
   "author": "Suleiman Shahbari",
   "scripts": {