@dudousxd/nestjs-agent-rag 0.13.1 → 0.14.0

package/dist/nestjs/index.cjs

@@ -0,0 +1,174 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/nestjs/index.ts
+var nestjs_exports = {};
+__export(nestjs_exports, {
+  RAG_INDEX_DEPS: () => RAG_INDEX_DEPS,
+  RagModule: () => RagModule,
+  RagRepository: () => RagRepository
+});
+module.exports = __toCommonJS(nestjs_exports);
+var import_nestjs_agent_rag = require("@dudousxd/nestjs-agent-rag");
+var import_common = require("@nestjs/common");
+function _ts_decorate(decorators, target, key, desc) {
+  var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+  if (typeof Reflect === "object" && typeof Reflect.decorate === "function") {
+    r = Reflect.decorate(decorators, target, key, desc);
+  } else {
+    for (var i = decorators.length - 1; i >= 0; i--) {
+      if (d = decorators[i]) {
+        r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+      }
+    }
+  }
+  return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+__name(_ts_decorate, "_ts_decorate");
+function _ts_metadata(metadataKey, metadataValue) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function") {
+    return Reflect.metadata(metadataKey, metadataValue);
+  }
+}
+__name(_ts_metadata, "_ts_metadata");
+var RAG_INDEX_DEPS = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent-rag/index-deps");
+var RagRepository = class {
+  static {
+    __name(this, "RagRepository");
+  }
+  ragIndexDeps;
+  chunk;
+  maxChunks;
+  minScore;
+  source(_row) {
+    return void 0;
+  }
+  metadata(_row) {
+    return {};
+  }
+  core;
+  // Built on first use: `name` and `scope` are subclass fields, and the deps are injected after both.
+  get ragIndex() {
+    this.core ??= new import_nestjs_agent_rag.RagIndex({
+      name: this.name,
+      scope: this.scope,
+      id: /* @__PURE__ */ __name((row) => this.id(row), "id"),
+      text: /* @__PURE__ */ __name((row) => this.text(row), "text"),
+      source: /* @__PURE__ */ __name((row) => this.source(row), "source"),
+      metadata: /* @__PURE__ */ __name((row) => this.metadata(row), "metadata"),
+      ...this.chunk !== void 0 ? {
+        chunk: this.chunk
+      } : {},
+      ...this.maxChunks !== void 0 ? {
+        maxChunks: this.maxChunks
+      } : {},
+      ...this.minScore !== void 0 ? {
+        minScore: this.minScore
+      } : {}
+    }, this.ragIndexDeps);
+    return this.core;
+  }
+  search(query, options) {
+    return this.ragIndex.search(query, options);
+  }
+  scoped(scope) {
+    return this.ragIndex.scoped(scope);
+  }
+  index(row) {
+    return this.ragIndex.index(row);
+  }
+  remove(rowOrId) {
+    return this.ragIndex.remove(rowOrId);
+  }
+  reconcile(liveIds, options) {
+    return this.ragIndex.reconcile(liveIds, options);
+  }
+  tool(options) {
+    return this.ragIndex.tool(options);
+  }
+};
+_ts_decorate([
+  (0, import_common.Inject)(RAG_INDEX_DEPS),
+  _ts_metadata("design:type", typeof RagIndexDeps === "undefined" ? Object : RagIndexDeps)
+], RagRepository.prototype, "ragIndexDeps", void 0);
+var RagFeatureModule = class RagFeatureModule2 {
+  static {
+    __name(this, "RagFeatureModule");
+  }
+};
+RagFeatureModule = _ts_decorate([
+  (0, import_common.Module)({})
+], RagFeatureModule);
+var RagModule = class _RagModule {
+  static {
+    __name(this, "RagModule");
+  }
+  static forRoot(deps) {
+    return {
+      module: _RagModule,
+      global: true,
+      providers: [
+        {
+          provide: RAG_INDEX_DEPS,
+          useValue: deps
+        }
+      ],
+      exports: [
+        RAG_INDEX_DEPS
+      ]
+    };
+  }
+  static forRootAsync(options) {
+    return {
+      module: _RagModule,
+      global: true,
+      ...options.imports !== void 0 ? {
+        imports: options.imports
+      } : {},
+      providers: [
+        {
+          provide: RAG_INDEX_DEPS,
+          inject: options.inject ?? [],
+          useFactory: options.useFactory
+        }
+      ],
+      exports: [
+        RAG_INDEX_DEPS
+      ]
+    };
+  }
+  static forFeature(repositories) {
+    return {
+      module: RagFeatureModule,
+      providers: repositories,
+      exports: repositories
+    };
+  }
+};
+RagModule = _ts_decorate([
+  (0, import_common.Module)({})
+], RagModule);
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  RAG_INDEX_DEPS,
+  RagModule,
+  RagRepository
+});
+//# sourceMappingURL=index.cjs.map

package/dist/nestjs/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/nestjs/index.ts"],"sourcesContent":["import type { Passage, Retriever } from '@dudousxd/nestjs-agent-core';\nimport {\n  type ChunkOptions,\n  RagIndex,\n  type RagIndexDeps,\n  type RagReconcileOptions,\n  type RagReconcileResult,\n  type RagScopeDefinition,\n  type RagScopeOf,\n  type RagScopeValue,\n  type RagSearchOptions,\n  type RagToolOptions,\n  type RetrievalTool,\n} from '@dudousxd/nestjs-agent-rag';\nimport { type DynamicModule, Inject, Module, type Type } from '@nestjs/common';\n\n/** Token of the `store` and `embedder` every {@link RagRepository} uses. */\nexport const RAG_INDEX_DEPS = Symbol.for('@dudousxd/nestjs-agent-rag/index-deps');\n\n/**\n * RAG for one model, as a provider of your own. The entity (or Drizzle table) doesn't change:\n *\n * ```ts\n * @Injectable()\n * export class DocumentsRag extends RagRepository<ProjectDocument> {\n *   protected readonly name = 'project_documents';\n *   readonly scope = { projectId: (d: ProjectDocument) => d.project.id };\n *   protected id(d: ProjectDocument) { return d.id; }\n *   protected text(d: ProjectDocument) { return d.textContent; }\n * }\n *\n * await documentsRag.index(doc);\n * await documentsRag.search('question', { scope: { projectId } });\n * ```\n *\n * `scope` is abstract, so leaving it out doesn't compile, and its shape types `search`. A corpus\n * shared by everyone has to say `scope = {}`.\n */\nexport abstract class RagRepository<Row> {\n  @Inject(RAG_INDEX_DEPS) private readonly ragIndexDeps!: RagIndexDeps;\n\n  /** Namespaces document ids and the metadata that separates models sharing a store. */\n  protected abstract readonly name: string;\n  abstract readonly scope: RagScopeDefinition<Row>;\n  protected readonly chunk?: ChunkOptions;\n  protected readonly maxChunks?: number;\n  protected readonly minScore?: number;\n\n  /** The value `remove` needs once the row is gone. */\n  abstract id(row: Row): RagScopeValue;\n  protected abstract text(row: Row): string | null | undefined;\n  protected source(_row: Row): string | null | undefined {\n    return undefined;\n  }\n  protected metadata(_row: Row): Record<string, unknown> {\n    return {};\n  }\n\n  private core?: RagIndex<Row, RagScopeDefinition<Row>>;\n\n  // Built on first use: `name` and `scope` are subclass fields, and the deps are injected after both.\n  private get ragIndex(): RagIndex<Row, RagScopeDefinition<Row>> {\n    this.core ??= new RagIndex(\n      {\n        name: this.name,\n        scope: this.scope,\n        id: (row) => this.id(row),\n        text: (row) => this.text(row),\n        source: (row) => this.source(row),\n        metadata: (row) => this.metadata(row),\n        ...(this.chunk !== undefined ? { chunk: this.chunk } : {}),\n        ...(this.maxChunks !== undefined ? { maxChunks: this.maxChunks } : {}),\n        ...(this.minScore !== undefined ? { minScore: this.minScore } : {}),\n      },\n      this.ragIndexDeps,\n    );\n    return this.core;\n  }\n\n  search(query: string, options: RagSearchOptions<this['scope']>): Promise<Passage[]> {\n    return this.ragIndex.search(query, options as never);\n  }\n\n  scoped(scope: RagScopeOf<this['scope']>): Retriever {\n    return this.ragIndex.scoped(scope as never);\n  }\n\n  index(row: Row): Promise<number> {\n    return this.ragIndex.index(row);\n  }\n\n  remove(rowOrId: Row | RagScopeValue): Promise<void> {\n    return this.ragIndex.remove(rowOrId);\n  }\n\n  reconcile(\n    liveIds: Iterable<RagScopeValue> | AsyncIterable<RagScopeValue>,\n    options?: RagReconcileOptions<this['scope']>,\n  ): Promise<RagReconcileResult> {\n    return this.ragIndex.reconcile(liveIds, options as never);\n  }\n\n  tool(options: RagToolOptions<this['scope']>): RetrievalTool {\n    return this.ragIndex.tool(options as never);\n  }\n}\n\nexport interface RagModuleAsyncOptions {\n  imports?: DynamicModule['imports'];\n  inject?: unknown[];\n  useFactory: (...args: never[]) => RagIndexDeps | Promise<RagIndexDeps>;\n}\n\n@Module({})\nclass RagFeatureModule {}\n\n/** `forRoot` holds the store and embedder, globally; `forFeature` registers repositories. */\n@Module({})\nexport class RagModule {\n  static forRoot(deps: RagIndexDeps): DynamicModule {\n    return {\n      module: RagModule,\n      global: true,\n      providers: [{ provide: RAG_INDEX_DEPS, useValue: deps }],\n      exports: [RAG_INDEX_DEPS],\n    };\n  }\n\n  static forRootAsync(options: RagModuleAsyncOptions): DynamicModule {\n    return {\n      module: RagModule,\n      global: true,\n      ...(options.imports !== undefined ? { imports: options.imports } : {}),\n      providers: [\n        {\n          provide: RAG_INDEX_DEPS,\n          inject: (options.inject ?? []) as never[],\n          useFactory: options.useFactory as (...args: unknown[]) => RagIndexDeps,\n        },\n      ],\n      exports: [RAG_INDEX_DEPS],\n    };\n  }\n\n  static forFeature(repositories: Type<unknown>[]): DynamicModule {\n    return { module: RagFeatureModule, providers: repositories, exports: repositories };\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AACA,8BAYO;AACP,oBAA8D;;;;;;;;;;;;;;;;;;;;;AAGvD,IAAMA,iBAAiBC,uBAAOC,IAAI,uCAAA;AAqBlC,IAAeC,gBAAf,MAAeA;SAAAA;;;EACqBC;EAKtBC;EACAC;EACAC;EAKTC,OAAOC,MAAsC;AACrD,WAAOC;EACT;EACUC,SAASF,MAAoC;AACrD,WAAO,CAAC;EACV;EAEQG;;EAGR,IAAYC,WAAmD;AAC7D,SAAKD,SAAS,IAAIE,iCAChB;MACEC,MAAM,KAAKA;MACXC,OAAO,KAAKA;MACZC,IAAI,wBAACC,QAAQ,KAAKD,GAAGC,GAAAA,GAAjB;MACJC,MAAM,wBAACD,QAAQ,KAAKC,KAAKD,GAAAA,GAAnB;MACNV,QAAQ,wBAACU,QAAQ,KAAKV,OAAOU,GAAAA,GAArB;MACRP,UAAU,wBAACO,QAAQ,KAAKP,SAASO,GAAAA,GAAvB;MACV,GAAI,KAAKb,UAAUK,SAAY;QAAEL,OAAO,KAAKA;MAAM,IAAI,CAAC;MACxD,GAAI,KAAKC,cAAcI,SAAY;QAAEJ,WAAW,KAAKA;MAAU,IAAI,CAAC;MACpE,GAAI,KAAKC,aAAaG,SAAY;QAAEH,UAAU,KAAKA;MAAS,IAAI,CAAC;IACnE,GACA,KAAKH,YAAY;AAEnB,WAAO,KAAKQ;EACd;EAEAQ,OAAOC,OAAeC,SAA8D;AAClF,WAAO,KAAKT,SAASO,OAAOC,OAAOC,OAAAA;EACrC;EAEAC,OAAOP,OAA6C;AAClD,WAAO,KAAKH,SAASU,OAAOP,KAAAA;EAC9B;EAEAQ,MAAMN,KAA2B;AAC/B,WAAO,KAAKL,SAASW,MAAMN,GAAAA;EAC7B;EAEAO,OAAOC,SAA6C;AAClD,WAAO,KAAKb,SAASY,OAAOC,OAAAA;EAC9B;EAEAC,UACEC,SACAN,SAC6B;AAC7B,WAAO,KAAKT,SAASc,UAAUC,SAASN,OAAAA;EAC1C;EAEAO,KAAKP,SAAuD;AAC1D,WAAO,KAAKT,SAASgB,KAAKP,OAAAA;EAC5B;AACF;;;;;AAQA,IACMQ,mBADN,MACMA,kBAAAA;SAAAA;;;AAAkB;;;;AAIjB,IAAMC,YAAN,MAAMA,WAAAA;SAAAA;;;EACX,OAAOC,QAAQC,MAAmC;AAChD,WAAO;MACLC,QAAQH;MACRI,QAAQ;MACRC,WAAW;QAAC;UAAEC,SAASrC;UAAgBsC,UAAUL;QAAK;;MACtDM,SAAS;QAACvC;;IACZ;EACF;EAEA,OAAOwC,aAAalB,SAA+C;AACjE,WAAO;MACLY,QAAQH;MACRI,QAAQ;MACR,GAAIb,QAAQmB,YAAY/B,SAAY;QAAE+B,SAASnB,QAAQmB;MAAQ,IAAI,CAAC;MACpEL,WAAW;QACT;UACEC,SAASrC;UACT0C,QAASpB,QAAQoB,UAAU,CAAA;UAC3BC,YAAYrB,QAAQqB;QACtB;;MAEFJ,SAAS;QAACvC;;IACZ;EACF;EAEA,OAAO4C,WAAWC,cAA8C;AAC9D,WAAO;MAAEX,QAAQJ;MAAkBM,WAAWS;MAAcN,SAASM;IAAa;EACpF;AACF;;;;","names":["RAG_INDEX_DEPS","Symbol","for","RagRepository","ragIndexDeps","chunk","maxChunks","minScore","source","_row","undefined","metadata","core","ragIndex","RagIndex","name","scope","id","row","text","search","query","options","scoped","index","remove","rowOrId","reconcile","liveIds","tool","RagFeatureModule","RagModule","forRoot","deps","module","global","providers","provide","useValue","exports","forRootAsync","imports","inject","useFactory","forFeature","repositories"]}

package/dist/nestjs/index.d.cts

@@ -0,0 +1,60 @@
+import { Passage, Retriever } from '@dudousxd/nestjs-agent-core';
+import { RagIndexDeps, RagScopeDefinition, ChunkOptions, RagScopeValue, RagSearchOptions, RagScopeOf, RagReconcileOptions, RagReconcileResult, RagToolOptions, RetrievalTool } from '@dudousxd/nestjs-agent-rag';
+import { DynamicModule, Type } from '@nestjs/common';
+
+/** Token of the `store` and `embedder` every {@link RagRepository} uses. */
+declare const RAG_INDEX_DEPS: unique symbol;
+/**
+ * RAG for one model, as a provider of your own. The entity (or Drizzle table) doesn't change:
+ *
+ * ```ts
+ * @Injectable()
+ * export class DocumentsRag extends RagRepository<ProjectDocument> {
+ *   protected readonly name = 'project_documents';
+ *   readonly scope = { projectId: (d: ProjectDocument) => d.project.id };
+ *   protected id(d: ProjectDocument) { return d.id; }
+ *   protected text(d: ProjectDocument) { return d.textContent; }
+ * }
+ *
+ * await documentsRag.index(doc);
+ * await documentsRag.search('question', { scope: { projectId } });
+ * ```
+ *
+ * `scope` is abstract, so leaving it out doesn't compile, and its shape types `search`. A corpus
+ * shared by everyone has to say `scope = {}`.
+ */
+declare abstract class RagRepository<Row> {
+    private readonly ragIndexDeps;
+    /** Namespaces document ids and the metadata that separates models sharing a store. */
+    protected abstract readonly name: string;
+    abstract readonly scope: RagScopeDefinition<Row>;
+    protected readonly chunk?: ChunkOptions;
+    protected readonly maxChunks?: number;
+    protected readonly minScore?: number;
+    /** The value `remove` needs once the row is gone. */
+    abstract id(row: Row): RagScopeValue;
+    protected abstract text(row: Row): string | null | undefined;
+    protected source(_row: Row): string | null | undefined;
+    protected metadata(_row: Row): Record<string, unknown>;
+    private core?;
+    private get ragIndex();
+    search(query: string, options: RagSearchOptions<this['scope']>): Promise<Passage[]>;
+    scoped(scope: RagScopeOf<this['scope']>): Retriever;
+    index(row: Row): Promise<number>;
+    remove(rowOrId: Row | RagScopeValue): Promise<void>;
+    reconcile(liveIds: Iterable<RagScopeValue> | AsyncIterable<RagScopeValue>, options?: RagReconcileOptions<this['scope']>): Promise<RagReconcileResult>;
+    tool(options: RagToolOptions<this['scope']>): RetrievalTool;
+}
+interface RagModuleAsyncOptions {
+    imports?: DynamicModule['imports'];
+    inject?: unknown[];
+    useFactory: (...args: never[]) => RagIndexDeps | Promise<RagIndexDeps>;
+}
+/** `forRoot` holds the store and embedder, globally; `forFeature` registers repositories. */
+declare class RagModule {
+    static forRoot(deps: RagIndexDeps): DynamicModule;
+    static forRootAsync(options: RagModuleAsyncOptions): DynamicModule;
+    static forFeature(repositories: Type<unknown>[]): DynamicModule;
+}
+
+export { RAG_INDEX_DEPS, RagModule, type RagModuleAsyncOptions, RagRepository };

package/dist/nestjs/index.d.ts

@@ -0,0 +1,60 @@
+import { Passage, Retriever } from '@dudousxd/nestjs-agent-core';
+import { RagIndexDeps, RagScopeDefinition, ChunkOptions, RagScopeValue, RagSearchOptions, RagScopeOf, RagReconcileOptions, RagReconcileResult, RagToolOptions, RetrievalTool } from '@dudousxd/nestjs-agent-rag';
+import { DynamicModule, Type } from '@nestjs/common';
+
+/** Token of the `store` and `embedder` every {@link RagRepository} uses. */
+declare const RAG_INDEX_DEPS: unique symbol;
+/**
+ * RAG for one model, as a provider of your own. The entity (or Drizzle table) doesn't change:
+ *
+ * ```ts
+ * @Injectable()
+ * export class DocumentsRag extends RagRepository<ProjectDocument> {
+ *   protected readonly name = 'project_documents';
+ *   readonly scope = { projectId: (d: ProjectDocument) => d.project.id };
+ *   protected id(d: ProjectDocument) { return d.id; }
+ *   protected text(d: ProjectDocument) { return d.textContent; }
+ * }
+ *
+ * await documentsRag.index(doc);
+ * await documentsRag.search('question', { scope: { projectId } });
+ * ```
+ *
+ * `scope` is abstract, so leaving it out doesn't compile, and its shape types `search`. A corpus
+ * shared by everyone has to say `scope = {}`.
+ */
+declare abstract class RagRepository<Row> {
+    private readonly ragIndexDeps;
+    /** Namespaces document ids and the metadata that separates models sharing a store. */
+    protected abstract readonly name: string;
+    abstract readonly scope: RagScopeDefinition<Row>;
+    protected readonly chunk?: ChunkOptions;
+    protected readonly maxChunks?: number;
+    protected readonly minScore?: number;
+    /** The value `remove` needs once the row is gone. */
+    abstract id(row: Row): RagScopeValue;
+    protected abstract text(row: Row): string | null | undefined;
+    protected source(_row: Row): string | null | undefined;
+    protected metadata(_row: Row): Record<string, unknown>;
+    private core?;
+    private get ragIndex();
+    search(query: string, options: RagSearchOptions<this['scope']>): Promise<Passage[]>;
+    scoped(scope: RagScopeOf<this['scope']>): Retriever;
+    index(row: Row): Promise<number>;
+    remove(rowOrId: Row | RagScopeValue): Promise<void>;
+    reconcile(liveIds: Iterable<RagScopeValue> | AsyncIterable<RagScopeValue>, options?: RagReconcileOptions<this['scope']>): Promise<RagReconcileResult>;
+    tool(options: RagToolOptions<this['scope']>): RetrievalTool;
+}
+interface RagModuleAsyncOptions {
+    imports?: DynamicModule['imports'];
+    inject?: unknown[];
+    useFactory: (...args: never[]) => RagIndexDeps | Promise<RagIndexDeps>;
+}
+/** `forRoot` holds the store and embedder, globally; `forFeature` registers repositories. */
+declare class RagModule {
+    static forRoot(deps: RagIndexDeps): DynamicModule;
+    static forRootAsync(options: RagModuleAsyncOptions): DynamicModule;
+    static forFeature(repositories: Type<unknown>[]): DynamicModule;
+}
+
+export { RAG_INDEX_DEPS, RagModule, type RagModuleAsyncOptions, RagRepository };

package/dist/nestjs/index.js

@@ -0,0 +1,149 @@
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+
+// src/nestjs/index.ts
+import { RagIndex } from "@dudousxd/nestjs-agent-rag";
+import { Inject, Module } from "@nestjs/common";
+function _ts_decorate(decorators, target, key, desc) {
+  var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+  if (typeof Reflect === "object" && typeof Reflect.decorate === "function") {
+    r = Reflect.decorate(decorators, target, key, desc);
+  } else {
+    for (var i = decorators.length - 1; i >= 0; i--) {
+      if (d = decorators[i]) {
+        r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+      }
+    }
+  }
+  return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+__name(_ts_decorate, "_ts_decorate");
+function _ts_metadata(metadataKey, metadataValue) {
+  if (typeof Reflect === "object" && typeof Reflect.metadata === "function") {
+    return Reflect.metadata(metadataKey, metadataValue);
+  }
+}
+__name(_ts_metadata, "_ts_metadata");
+var RAG_INDEX_DEPS = /* @__PURE__ */ Symbol.for("@dudousxd/nestjs-agent-rag/index-deps");
+var RagRepository = class {
+  static {
+    __name(this, "RagRepository");
+  }
+  ragIndexDeps;
+  chunk;
+  maxChunks;
+  minScore;
+  source(_row) {
+    return void 0;
+  }
+  metadata(_row) {
+    return {};
+  }
+  core;
+  // Built on first use: `name` and `scope` are subclass fields, and the deps are injected after both.
+  get ragIndex() {
+    this.core ??= new RagIndex({
+      name: this.name,
+      scope: this.scope,
+      id: /* @__PURE__ */ __name((row) => this.id(row), "id"),
+      text: /* @__PURE__ */ __name((row) => this.text(row), "text"),
+      source: /* @__PURE__ */ __name((row) => this.source(row), "source"),
+      metadata: /* @__PURE__ */ __name((row) => this.metadata(row), "metadata"),
+      ...this.chunk !== void 0 ? {
+        chunk: this.chunk
+      } : {},
+      ...this.maxChunks !== void 0 ? {
+        maxChunks: this.maxChunks
+      } : {},
+      ...this.minScore !== void 0 ? {
+        minScore: this.minScore
+      } : {}
+    }, this.ragIndexDeps);
+    return this.core;
+  }
+  search(query, options) {
+    return this.ragIndex.search(query, options);
+  }
+  scoped(scope) {
+    return this.ragIndex.scoped(scope);
+  }
+  index(row) {
+    return this.ragIndex.index(row);
+  }
+  remove(rowOrId) {
+    return this.ragIndex.remove(rowOrId);
+  }
+  reconcile(liveIds, options) {
+    return this.ragIndex.reconcile(liveIds, options);
+  }
+  tool(options) {
+    return this.ragIndex.tool(options);
+  }
+};
+_ts_decorate([
+  Inject(RAG_INDEX_DEPS),
+  _ts_metadata("design:type", typeof RagIndexDeps === "undefined" ? Object : RagIndexDeps)
+], RagRepository.prototype, "ragIndexDeps", void 0);
+var RagFeatureModule = class RagFeatureModule2 {
+  static {
+    __name(this, "RagFeatureModule");
+  }
+};
+RagFeatureModule = _ts_decorate([
+  Module({})
+], RagFeatureModule);
+var RagModule = class _RagModule {
+  static {
+    __name(this, "RagModule");
+  }
+  static forRoot(deps) {
+    return {
+      module: _RagModule,
+      global: true,
+      providers: [
+        {
+          provide: RAG_INDEX_DEPS,
+          useValue: deps
+        }
+      ],
+      exports: [
+        RAG_INDEX_DEPS
+      ]
+    };
+  }
+  static forRootAsync(options) {
+    return {
+      module: _RagModule,
+      global: true,
+      ...options.imports !== void 0 ? {
+        imports: options.imports
+      } : {},
+      providers: [
+        {
+          provide: RAG_INDEX_DEPS,
+          inject: options.inject ?? [],
+          useFactory: options.useFactory
+        }
+      ],
+      exports: [
+        RAG_INDEX_DEPS
+      ]
+    };
+  }
+  static forFeature(repositories) {
+    return {
+      module: RagFeatureModule,
+      providers: repositories,
+      exports: repositories
+    };
+  }
+};
+RagModule = _ts_decorate([
+  Module({})
+], RagModule);
+export {
+  RAG_INDEX_DEPS,
+  RagModule,
+  RagRepository
+};
+//# sourceMappingURL=index.js.map

package/dist/nestjs/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/nestjs/index.ts"],"sourcesContent":["import type { Passage, Retriever } from '@dudousxd/nestjs-agent-core';\nimport {\n  type ChunkOptions,\n  RagIndex,\n  type RagIndexDeps,\n  type RagReconcileOptions,\n  type RagReconcileResult,\n  type RagScopeDefinition,\n  type RagScopeOf,\n  type RagScopeValue,\n  type RagSearchOptions,\n  type RagToolOptions,\n  type RetrievalTool,\n} from '@dudousxd/nestjs-agent-rag';\nimport { type DynamicModule, Inject, Module, type Type } from '@nestjs/common';\n\n/** Token of the `store` and `embedder` every {@link RagRepository} uses. */\nexport const RAG_INDEX_DEPS = Symbol.for('@dudousxd/nestjs-agent-rag/index-deps');\n\n/**\n * RAG for one model, as a provider of your own. The entity (or Drizzle table) doesn't change:\n *\n * ```ts\n * @Injectable()\n * export class DocumentsRag extends RagRepository<ProjectDocument> {\n *   protected readonly name = 'project_documents';\n *   readonly scope = { projectId: (d: ProjectDocument) => d.project.id };\n *   protected id(d: ProjectDocument) { return d.id; }\n *   protected text(d: ProjectDocument) { return d.textContent; }\n * }\n *\n * await documentsRag.index(doc);\n * await documentsRag.search('question', { scope: { projectId } });\n * ```\n *\n * `scope` is abstract, so leaving it out doesn't compile, and its shape types `search`. A corpus\n * shared by everyone has to say `scope = {}`.\n */\nexport abstract class RagRepository<Row> {\n  @Inject(RAG_INDEX_DEPS) private readonly ragIndexDeps!: RagIndexDeps;\n\n  /** Namespaces document ids and the metadata that separates models sharing a store. */\n  protected abstract readonly name: string;\n  abstract readonly scope: RagScopeDefinition<Row>;\n  protected readonly chunk?: ChunkOptions;\n  protected readonly maxChunks?: number;\n  protected readonly minScore?: number;\n\n  /** The value `remove` needs once the row is gone. */\n  abstract id(row: Row): RagScopeValue;\n  protected abstract text(row: Row): string | null | undefined;\n  protected source(_row: Row): string | null | undefined {\n    return undefined;\n  }\n  protected metadata(_row: Row): Record<string, unknown> {\n    return {};\n  }\n\n  private core?: RagIndex<Row, RagScopeDefinition<Row>>;\n\n  // Built on first use: `name` and `scope` are subclass fields, and the deps are injected after both.\n  private get ragIndex(): RagIndex<Row, RagScopeDefinition<Row>> {\n    this.core ??= new RagIndex(\n      {\n        name: this.name,\n        scope: this.scope,\n        id: (row) => this.id(row),\n        text: (row) => this.text(row),\n        source: (row) => this.source(row),\n        metadata: (row) => this.metadata(row),\n        ...(this.chunk !== undefined ? { chunk: this.chunk } : {}),\n        ...(this.maxChunks !== undefined ? { maxChunks: this.maxChunks } : {}),\n        ...(this.minScore !== undefined ? { minScore: this.minScore } : {}),\n      },\n      this.ragIndexDeps,\n    );\n    return this.core;\n  }\n\n  search(query: string, options: RagSearchOptions<this['scope']>): Promise<Passage[]> {\n    return this.ragIndex.search(query, options as never);\n  }\n\n  scoped(scope: RagScopeOf<this['scope']>): Retriever {\n    return this.ragIndex.scoped(scope as never);\n  }\n\n  index(row: Row): Promise<number> {\n    return this.ragIndex.index(row);\n  }\n\n  remove(rowOrId: Row | RagScopeValue): Promise<void> {\n    return this.ragIndex.remove(rowOrId);\n  }\n\n  reconcile(\n    liveIds: Iterable<RagScopeValue> | AsyncIterable<RagScopeValue>,\n    options?: RagReconcileOptions<this['scope']>,\n  ): Promise<RagReconcileResult> {\n    return this.ragIndex.reconcile(liveIds, options as never);\n  }\n\n  tool(options: RagToolOptions<this['scope']>): RetrievalTool {\n    return this.ragIndex.tool(options as never);\n  }\n}\n\nexport interface RagModuleAsyncOptions {\n  imports?: DynamicModule['imports'];\n  inject?: unknown[];\n  useFactory: (...args: never[]) => RagIndexDeps | Promise<RagIndexDeps>;\n}\n\n@Module({})\nclass RagFeatureModule {}\n\n/** `forRoot` holds the store and embedder, globally; `forFeature` registers repositories. */\n@Module({})\nexport class RagModule {\n  static forRoot(deps: RagIndexDeps): DynamicModule {\n    return {\n      module: RagModule,\n      global: true,\n      providers: [{ provide: RAG_INDEX_DEPS, useValue: deps }],\n      exports: [RAG_INDEX_DEPS],\n    };\n  }\n\n  static forRootAsync(options: RagModuleAsyncOptions): DynamicModule {\n    return {\n      module: RagModule,\n      global: true,\n      ...(options.imports !== undefined ? { imports: options.imports } : {}),\n      providers: [\n        {\n          provide: RAG_INDEX_DEPS,\n          inject: (options.inject ?? []) as never[],\n          useFactory: options.useFactory as (...args: unknown[]) => RagIndexDeps,\n        },\n      ],\n      exports: [RAG_INDEX_DEPS],\n    };\n  }\n\n  static forFeature(repositories: Type<unknown>[]): DynamicModule {\n    return { module: RagFeatureModule, providers: repositories, exports: repositories };\n  }\n}\n"],"mappings":";;;;AACA,SAEEA,gBAUK;AACP,SAA6BC,QAAQC,cAAyB;;;;;;;;;;;;;;;;;;;;;AAGvD,IAAMC,iBAAiBC,uBAAOC,IAAI,uCAAA;AAqBlC,IAAeC,gBAAf,MAAeA;SAAAA;;;EACqBC;EAKtBC;EACAC;EACAC;EAKTC,OAAOC,MAAsC;AACrD,WAAOC;EACT;EACUC,SAASF,MAAoC;AACrD,WAAO,CAAC;EACV;EAEQG;;EAGR,IAAYC,WAAmD;AAC7D,SAAKD,SAAS,IAAIE,SAChB;MACEC,MAAM,KAAKA;MACXC,OAAO,KAAKA;MACZC,IAAI,wBAACC,QAAQ,KAAKD,GAAGC,GAAAA,GAAjB;MACJC,MAAM,wBAACD,QAAQ,KAAKC,KAAKD,GAAAA,GAAnB;MACNV,QAAQ,wBAACU,QAAQ,KAAKV,OAAOU,GAAAA,GAArB;MACRP,UAAU,wBAACO,QAAQ,KAAKP,SAASO,GAAAA,GAAvB;MACV,GAAI,KAAKb,UAAUK,SAAY;QAAEL,OAAO,KAAKA;MAAM,IAAI,CAAC;MACxD,GAAI,KAAKC,cAAcI,SAAY;QAAEJ,WAAW,KAAKA;MAAU,IAAI,CAAC;MACpE,GAAI,KAAKC,aAAaG,SAAY;QAAEH,UAAU,KAAKA;MAAS,IAAI,CAAC;IACnE,GACA,KAAKH,YAAY;AAEnB,WAAO,KAAKQ;EACd;EAEAQ,OAAOC,OAAeC,SAA8D;AAClF,WAAO,KAAKT,SAASO,OAAOC,OAAOC,OAAAA;EACrC;EAEAC,OAAOP,OAA6C;AAClD,WAAO,KAAKH,SAASU,OAAOP,KAAAA;EAC9B;EAEAQ,MAAMN,KAA2B;AAC/B,WAAO,KAAKL,SAASW,MAAMN,GAAAA;EAC7B;EAEAO,OAAOC,SAA6C;AAClD,WAAO,KAAKb,SAASY,OAAOC,OAAAA;EAC9B;EAEAC,UACEC,SACAN,SAC6B;AAC7B,WAAO,KAAKT,SAASc,UAAUC,SAASN,OAAAA;EAC1C;EAEAO,KAAKP,SAAuD;AAC1D,WAAO,KAAKT,SAASgB,KAAKP,OAAAA;EAC5B;AACF;;;;;AAQA,IACMQ,mBADN,MACMA,kBAAAA;SAAAA;;;AAAkB;;;;AAIjB,IAAMC,YAAN,MAAMA,WAAAA;SAAAA;;;EACX,OAAOC,QAAQC,MAAmC;AAChD,WAAO;MACLC,QAAQH;MACRI,QAAQ;MACRC,WAAW;QAAC;UAAEC,SAASrC;UAAgBsC,UAAUL;QAAK;;MACtDM,SAAS;QAACvC;;IACZ;EACF;EAEA,OAAOwC,aAAalB,SAA+C;AACjE,WAAO;MACLY,QAAQH;MACRI,QAAQ;MACR,GAAIb,QAAQmB,YAAY/B,SAAY;QAAE+B,SAASnB,QAAQmB;MAAQ,IAAI,CAAC;MACpEL,WAAW;QACT;UACEC,SAASrC;UACT0C,QAASpB,QAAQoB,UAAU,CAAA;UAC3BC,YAAYrB,QAAQqB;QACtB;;MAEFJ,SAAS;QAACvC;;IACZ;EACF;EAEA,OAAO4C,WAAWC,cAA8C;AAC9D,WAAO;MAAEX,QAAQJ;MAAkBM,WAAWS;MAAcN,SAASM;IAAa;EACpF;AACF;;;;","names":["RagIndex","Inject","Module","RAG_INDEX_DEPS","Symbol","for","RagRepository","ragIndexDeps","chunk","maxChunks","minScore","source","_row","undefined","metadata","core","ragIndex","RagIndex","name","scope","id","row","text","search","query","options","scoped","index","remove","rowOrId","reconcile","liveIds","tool","RagFeatureModule","RagModule","forRoot","deps","module","global","providers","provide","useValue","exports","forRootAsync","imports","inject","useFactory","forFeature","repositories"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dudousxd/nestjs-agent-rag",
-  "version": "0.13.1",
+  "version": "0.14.0",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/DavideCarvalho/nestjs-agent.git",
@@ -32,6 +32,16 @@
         "types": "./dist/index.d.cts",
         "default": "./dist/index.cjs"
       }
+    },
+    "./nestjs": {
+      "import": {
+        "types": "./dist/nestjs/index.d.ts",
+        "default": "./dist/nestjs/index.js"
+      },
+      "require": {
+        "types": "./dist/nestjs/index.d.cts",
+        "default": "./dist/nestjs/index.cjs"
+      }
     }
   },
   "files": [
@@ -39,24 +49,34 @@
     "NOTICE"
   ],
   "peerDependencies": {
-    "@dudousxd/nestjs-agent-core": ">=0.10.0 <1.0.0"
+    "@dudousxd/nestjs-agent-core": ">=0.10.0 <1.0.0",
+    "@nestjs/common": "^10.0.0 || ^11.0.0 || ^12.0.0"
   },
   "dependencies": {
     "@dudousxd/nestjs-diagnostics": "0.7.2",
     "@standard-schema/spec": "1.1.0"
   },
   "devDependencies": {
+    "@nestjs/common": "12.0.4",
+    "@nestjs/core": "12.0.4",
+    "@nestjs/testing": "12.0.4",
     "@testcontainers/postgresql": "12.2.0",
     "@types/pg": "8.23.1",
     "drizzle-orm": "0.45.3",
     "pg": "8.23.1",
     "redis": "6.2.1",
+    "reflect-metadata": "0.2.2",
     "testcontainers": "12.2.0",
     "tsup": "8.5.1",
     "typescript": "5.9.3",
     "vitest": "2.1.9",
-    "@dudousxd/nestjs-agent-core": "0.41.0",
-    "@dudousxd/nestjs-agent-testing": "0.26.1"
+    "@dudousxd/nestjs-agent-core": "0.61.0",
+    "@dudousxd/nestjs-agent-testing": "0.34.0"
+  },
+  "peerDependenciesMeta": {
+    "@nestjs/common": {
+      "optional": true
+    }
   },
   "scripts": {
     "build": "tsup",