@hazeljs/rag 0.2.0-beta.8 → 0.2.0-beta.80

package/dist/__tests__/rag-pipeline.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rag-pipeline.test.js","sourceRoot":"","sources":["../../src/__tests__/rag-pipeline.test.ts"],"names":[],"mappings":";;AAAA,kDAA8C;AAC9C,oCAA6C;AAS7C,SAAS,gBAAgB,CAAC,EAAU,EAAE,OAAe,EAAE,KAAK,GAAG,GAAG;IAChE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;AAC/F,CAAC;AAED,SAAS,eAAe,CAAC,UAA0B,EAAE;IACnD,OAAO;QACL,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAClD,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,KAAK,CAAC,CAAC;QAClD,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC;QAC5C,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC;QACpD,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACvD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACtD,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACnD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;KAC9C,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB;IAC5B,OAAO;QACL,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QACnD,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QAC1D,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;KAC3C,CAAC;AACJ,CAAC;AAED,SAAS,gBAAgB;IACvB,OAAO;QACL,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QACxD,cAAc,EAAE,IAAI;aACjB,EAAE,EAAE;aACJ,kBAAkB,CAAC,CAAC,IAAgB,EAAE,EAAE,CACvC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,GAAG,UAAU,EAAE,CAAC,CAAC,CAC7D;KACJ,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,EAAE,GAAG,eAAe,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC5B,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,EAAE,GAAG,eAAe,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,IAAI,GAAe,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YAC9E,MAAM,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,EAAE,GAAG,eAAe,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,gBAAgB,EAAE,CAAC;YACpC,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;gBAC1C,YAAY,EAAE,QAAQ;aACvB,CAAC,CAAC;YACH,MAAM,IAAI,GAAe,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YACpE,MAAM,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;YAC3D,MAAM,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC1C,MAAM,CAAC,eAAe,CAAC;gBACrB,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,CAAC;aACzE,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,gBAAgB,CAAC,IAAI,EAAE,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC5E,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;YACzD,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,gBAAgB,CAAC,IAAI,EAAE,oBAAoB,CAAC,CAAC,CAAC,CAAC;YAC3E,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,kBAAkB,CAAC,CAAC;YAC5D,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAC9B,EAAE,WAAW,EAAE,EAAE,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,EAAE,EAC/D,GAAG,CACJ,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC7C,SAAS,EAAE,gCAAgC;aAC5C,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC/B,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,gBAAgB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;YAChE,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAC9B,EAAE,WAAW,EAAE,EAAE,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,EAAE,EAC/D,GAAG,CACJ,CAAC;YACF,MAAM,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC9B,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,gBAAgB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;YAChE,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,EAAE,GAAG,eAAe,CAAC,EAAE,CAAC,CAAC;YAC/B,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;gBAC1C,IAAI,EAAE,CAAC;aACR,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC9B,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,EAAE,GAAG,eAAe,CAAC,EAAE,CAAC,CAAC;YAC/B,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,EAAE,yBAAiB,CAAC,UAAU,CAAC,CAAC;YACnE,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,EAAE,GAAG,eAAe,CAAC,EAAE,CAAC,CAAC;YAC/B,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,EAAE,yBAAiB,CAAC,MAAM,CAAC,CAAC;YAC/D,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,OAAO,GAAG;gBACd,gBAAgB,CAAC,IAAI,EAAE,WAAW,CAAC;gBACnC,gBAAgB,CAAC,IAAI,EAAE,WAAW,CAAC;gBACnC,gBAAgB,CAAC,IAAI,EAAE,WAAW,CAAC;aACpC,CAAC;YACF,MAAM,EAAE,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;YACpC,MAAM,QAAQ,GAAG,qBAAqB,EAAE,CAAC;YACzC,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC,EAAE,WAAW,EAAE,EAAE,EAAE,iBAAiB,EAAE,QAAQ,EAAE,CAAC,CAAC;YACnF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,yBAAiB,CAAC,GAAG,CAAC,CAAC;YACtF,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,oBAAoB,CACpC,OAAO,EACP,MAAM,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAC7D,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,EAAE,GAAG,eAAe,CAAC,EAAE,CAAC,CAAC;YAC/B,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,EAAE,yBAAiB,CAAC,GAAG,CAAC,CAAC;YAC7E,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,OAAO,GAAG;gBACd,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,eAAe;aACjF,CAAC;YACF,MAAM,EAAE,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;YACpC,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,yBAAiB,CAAC,GAAG,CAAC,CAAC;YACtF,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,EAAE,GAAG,eAAe,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,eAAe,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,EAAE,GAAG,eAAe,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAC;YACvB,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACtC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,gBAAgB,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC;YACrE,MAAM,QAAQ,GAAG,IAAI,0BAAW,CAAC;gBAC/B,WAAW,EAAE,EAAE;gBACf,iBAAiB,EAAE,qBAAqB,EAAE;aAC3C,CAAC,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC/C,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YAC1C,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/retrieval/bm25.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=bm25.test.d.ts.map

package/dist/__tests__/retrieval/bm25.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bm25.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/retrieval/bm25.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/retrieval/bm25.test.js

@@ -0,0 +1,86 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const bm25_1 = require("../../retrieval/bm25");
+function makeDocs() {
+    return [
+        {
+            id: 'd1',
+            content: 'TypeScript is a typed superset of JavaScript',
+            tokens: ['typescript', 'is', 'a', 'typed', 'superset', 'of', 'javascript'],
+        },
+        {
+            id: 'd2',
+            content: 'JavaScript runs in the browser',
+            tokens: ['javascript', 'runs', 'in', 'the', 'browser'],
+        },
+        {
+            id: 'd3',
+            content: 'Python is a general purpose language',
+            tokens: ['python', 'is', 'a', 'general', 'purpose', 'language'],
+        },
+    ];
+}
+describe('BM25', () => {
+    it('searches and returns ranked results', () => {
+        const bm25 = new bm25_1.BM25();
+        bm25.addDocuments(makeDocs());
+        const results = bm25.search('typescript');
+        expect(results.length).toBeGreaterThan(0);
+        expect(results[0].id).toBe('d1');
+        expect(results[0].score).toBeGreaterThan(0);
+    });
+    it('results are sorted by score descending', () => {
+        const bm25 = new bm25_1.BM25();
+        bm25.addDocuments(makeDocs());
+        const results = bm25.search('javascript');
+        for (let i = 1; i < results.length; i++) {
+            expect(results[i - 1].score).toBeGreaterThanOrEqual(results[i].score);
+        }
+    });
+    it('returns at most topK results', () => {
+        const bm25 = new bm25_1.BM25();
+        bm25.addDocuments(makeDocs());
+        const results = bm25.search('is', 2);
+        expect(results.length).toBeLessThanOrEqual(2);
+    });
+    it('returns zero score for terms not in any document', () => {
+        const bm25 = new bm25_1.BM25();
+        bm25.addDocuments(makeDocs());
+        const results = bm25.search('xylophone');
+        results.forEach((r) => expect(r.score).toBe(0));
+    });
+    it('handles empty query gracefully', () => {
+        const bm25 = new bm25_1.BM25();
+        bm25.addDocuments(makeDocs());
+        const results = bm25.search('');
+        expect(Array.isArray(results)).toBe(true);
+    });
+    it('addDocuments accumulates across calls', () => {
+        const bm25 = new bm25_1.BM25();
+        bm25.addDocuments([makeDocs()[0]]);
+        bm25.addDocuments([makeDocs()[1]]);
+        const results = bm25.search('javascript');
+        expect(results.some((r) => r.id === 'd2')).toBe(true);
+    });
+    it('clear resets all state', () => {
+        const bm25 = new bm25_1.BM25();
+        bm25.addDocuments(makeDocs());
+        bm25.clear();
+        const results = bm25.search('typescript');
+        expect(results).toHaveLength(0);
+    });
+    it('uses custom k1 and b parameters', () => {
+        const bm25 = new bm25_1.BM25({ k1: 2.0, b: 0.5 });
+        bm25.addDocuments(makeDocs());
+        const results = bm25.search('typescript');
+        expect(results.length).toBeGreaterThan(0);
+    });
+    it('tokenizes queries before scoring', () => {
+        const bm25 = new bm25_1.BM25();
+        bm25.addDocuments(makeDocs());
+        // Query with punctuation should still match
+        const results = bm25.search('TypeScript!');
+        expect(results.some((r) => r.id === 'd1')).toBe(true);
+    });
+});
+//# sourceMappingURL=bm25.test.js.map

package/dist/__tests__/retrieval/bm25.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bm25.test.js","sourceRoot":"","sources":["../../../src/__tests__/retrieval/bm25.test.ts"],"names":[],"mappings":";;AAAA,+CAA4C;AAE5C,SAAS,QAAQ;IACf,OAAO;QACL;YACE,EAAE,EAAE,IAAI;YACR,OAAO,EAAE,8CAA8C;YACvD,MAAM,EAAE,CAAC,YAAY,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,YAAY,CAAC;SAC3E;QACD;YACE,EAAE,EAAE,IAAI;YACR,OAAO,EAAE,gCAAgC;YACzC,MAAM,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,CAAC;SACvD;QACD;YACE,EAAE,EAAE,IAAI;YACR,OAAO,EAAE,sCAAsC;YAC/C,MAAM,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,SAAS,EAAE,SAAS,EAAE,UAAU,CAAC;SAChE;KACF,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;IACpB,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,IAAI,GAAG,IAAI,WAAI,EAAE,CAAC;QACxB,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,IAAI,GAAG,IAAI,WAAI,EAAE,CAAC;QACxB,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC1C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QACxE,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,IAAI,GAAG,IAAI,WAAI,EAAE,CAAC;QACxB,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,IAAI,GAAG,IAAI,WAAI,EAAE,CAAC;QACxB,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACzC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,IAAI,GAAG,IAAI,WAAI,EAAE,CAAC;QACxB,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAChC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,IAAI,GAAG,IAAI,WAAI,EAAE,CAAC;QACxB,IAAI,CAAC,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,YAAY,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACnC,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE;QAChC,MAAM,IAAI,GAAG,IAAI,WAAI,EAAE,CAAC;QACxB,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9B,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,IAAI,GAAG,IAAI,WAAI,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,IAAI,GAAG,IAAI,WAAI,EAAE,CAAC;QACxB,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9B,4CAA4C;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAC3C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/retrieval/hybrid-search.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=hybrid-search.test.d.ts.map

package/dist/__tests__/retrieval/hybrid-search.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid-search.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/retrieval/hybrid-search.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/retrieval/hybrid-search.test.js

@@ -0,0 +1,85 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const hybrid_search_1 = require("../../retrieval/hybrid-search");
+function makeVectorStore(results = []) {
+    return {
+        initialize: jest.fn().mockResolvedValue(undefined),
+        addDocuments: jest.fn().mockResolvedValue([]),
+        search: jest.fn().mockResolvedValue(results),
+        searchByVector: jest.fn().mockResolvedValue(results),
+        deleteDocuments: jest.fn().mockResolvedValue(undefined),
+        updateDocument: jest.fn().mockResolvedValue(undefined),
+        getDocument: jest.fn().mockResolvedValue(undefined),
+        clear: jest.fn().mockResolvedValue(undefined),
+    };
+}
+const VECTOR_RESULTS = [
+    { id: 'd1', content: 'TypeScript typed language', score: 0.9, metadata: {} },
+    { id: 'd2', content: 'JavaScript dynamic language', score: 0.7, metadata: {} },
+];
+describe('HybridSearchRetrieval', () => {
+    it('throws when weights do not sum to 1', () => {
+        const vs = makeVectorStore();
+        expect(() => new hybrid_search_1.HybridSearchRetrieval(vs, { vectorWeight: 0.5, keywordWeight: 0.3 })).toThrow('sum to 1.0');
+    });
+    it('throws if search is called before indexDocuments', async () => {
+        const vs = makeVectorStore();
+        const hybrid = new hybrid_search_1.HybridSearchRetrieval(vs);
+        await expect(hybrid.search('query')).rejects.toThrow('must be indexed');
+    });
+    it('performs hybrid search after indexing', async () => {
+        const vs = makeVectorStore(VECTOR_RESULTS);
+        const hybrid = new hybrid_search_1.HybridSearchRetrieval(vs);
+        await hybrid.indexDocuments([
+            { id: 'd1', content: 'TypeScript typed language' },
+            { id: 'd2', content: 'JavaScript dynamic language' },
+        ]);
+        const results = await hybrid.search('typescript');
+        expect(Array.isArray(results)).toBe(true);
+    });
+    it('returns at most topK results', async () => {
+        const vs = makeVectorStore(VECTOR_RESULTS);
+        const hybrid = new hybrid_search_1.HybridSearchRetrieval(vs);
+        await hybrid.indexDocuments([
+            { id: 'd1', content: 'TypeScript typed language' },
+            { id: 'd2', content: 'JavaScript dynamic language' },
+        ]);
+        const results = await hybrid.search('language', { topK: 1 });
+        expect(results.length).toBeLessThanOrEqual(1);
+    });
+    it('handles empty vector results gracefully', async () => {
+        const vs = makeVectorStore([]);
+        const hybrid = new hybrid_search_1.HybridSearchRetrieval(vs);
+        await hybrid.indexDocuments([{ id: 'd1', content: 'Some content' }]);
+        const results = await hybrid.search('typescript');
+        expect(Array.isArray(results)).toBe(true);
+        expect(results).toHaveLength(0);
+    });
+    it('clear resets the index', async () => {
+        const vs = makeVectorStore(VECTOR_RESULTS);
+        const hybrid = new hybrid_search_1.HybridSearchRetrieval(vs);
+        await hybrid.indexDocuments([{ id: 'd1', content: 'content' }]);
+        hybrid.clear();
+        await expect(hybrid.search('query')).rejects.toThrow('must be indexed');
+    });
+    it('uses custom weights', async () => {
+        const vs = makeVectorStore(VECTOR_RESULTS);
+        const hybrid = new hybrid_search_1.HybridSearchRetrieval(vs, { vectorWeight: 0.8, keywordWeight: 0.2 });
+        await hybrid.indexDocuments([{ id: 'd1', content: 'TypeScript typed language' }]);
+        const results = await hybrid.search('typescript');
+        expect(Array.isArray(results)).toBe(true);
+    });
+    it('results have hybridScore applied to score field', async () => {
+        const vs = makeVectorStore(VECTOR_RESULTS);
+        const hybrid = new hybrid_search_1.HybridSearchRetrieval(vs);
+        await hybrid.indexDocuments([
+            { id: 'd1', content: 'TypeScript typed language' },
+            { id: 'd2', content: 'JavaScript dynamic language' },
+        ]);
+        const results = await hybrid.search('typescript language');
+        for (const r of results) {
+            expect(typeof r.score).toBe('number');
+        }
+    });
+});
+//# sourceMappingURL=hybrid-search.test.js.map

package/dist/__tests__/retrieval/hybrid-search.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hybrid-search.test.js","sourceRoot":"","sources":["../../../src/__tests__/retrieval/hybrid-search.test.ts"],"names":[],"mappings":";;AAAA,iEAAsE;AAGtE,SAAS,eAAe,CAAC,UAA0B,EAAE;IACnD,OAAO;QACL,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAClD,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;QAC7C,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC;QAC5C,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC;QACpD,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACvD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACtD,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACnD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;KAC9C,CAAC;AACJ,CAAC;AAED,MAAM,cAAc,GAAmB;IACrC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,2BAA2B,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC5E,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,6BAA6B,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE;CAC/E,CAAC;AAEF,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,EAAE,GAAG,eAAe,EAAE,CAAC;QAC7B,MAAM,CAAC,GAAG,EAAE,CAAC,IAAI,qCAAqB,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,OAAO,CAC5F,YAAY,CACb,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,GAAG,eAAe,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,IAAI,qCAAqB,CAAC,EAAE,CAAC,CAAC;QAC7C,MAAM,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,EAAE,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,qCAAqB,CAAC,EAAE,CAAC,CAAC;QAC7C,MAAM,MAAM,CAAC,cAAc,CAAC;YAC1B,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,2BAA2B,EAAE;YAClD,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,6BAA6B,EAAE;SACrD,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,EAAE,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,qCAAqB,CAAC,EAAE,CAAC,CAAC;QAC7C,MAAM,MAAM,CAAC,cAAc,CAAC;YAC1B,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,2BAA2B,EAAE;YAClD,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,6BAA6B,EAAE;SACrD,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,EAAE,GAAG,eAAe,CAAC,EAAE,CAAC,CAAC;QAC/B,MAAM,MAAM,GAAG,IAAI,qCAAqB,CAAC,EAAE,CAAC,CAAC;QAC7C,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;QACrE,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,EAAE,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,qCAAqB,CAAC,EAAE,CAAC,CAAC;QAC7C,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QAChE,MAAM,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACnC,MAAM,EAAE,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,qCAAqB,CAAC,EAAE,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;QACxF,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,2BAA2B,EAAE,CAAC,CAAC,CAAC;QAClF,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,EAAE,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,qCAAqB,CAAC,EAAE,CAAC,CAAC;QAC7C,MAAM,MAAM,CAAC,cAAc,CAAC;YAC1B,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,2BAA2B,EAAE;YAClD,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,6BAA6B,EAAE;SACrD,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC;QAC3D,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACxB,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/retrieval/multi-query.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=multi-query.test.d.ts.map

package/dist/__tests__/retrieval/multi-query.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"multi-query.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/retrieval/multi-query.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/retrieval/multi-query.test.js

@@ -0,0 +1,90 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const multi_query_1 = require("../../retrieval/multi-query");
+function makeVectorStore(results = []) {
+    return {
+        initialize: jest.fn().mockResolvedValue(undefined),
+        addDocuments: jest.fn().mockResolvedValue([]),
+        search: jest.fn().mockResolvedValue(results),
+        searchByVector: jest.fn().mockResolvedValue(results),
+        deleteDocuments: jest.fn().mockResolvedValue(undefined),
+        updateDocument: jest.fn().mockResolvedValue(undefined),
+        getDocument: jest.fn().mockResolvedValue(undefined),
+        clear: jest.fn().mockResolvedValue(undefined),
+    };
+}
+const SEARCH_RESULTS = [
+    { id: 'd1', content: 'TypeScript docs', score: 0.9, metadata: {} },
+    { id: 'd2', content: 'JavaScript docs', score: 0.7, metadata: {} },
+];
+describe('MultiQueryRetrieval', () => {
+    it('retrieves using a custom generator', async () => {
+        const vs = makeVectorStore(SEARCH_RESULTS);
+        const customGenerator = jest.fn().mockResolvedValue(['query 1', 'query 2', 'query 3']);
+        const retrieval = new multi_query_1.MultiQueryRetrieval(vs, { numQueries: 3, customGenerator });
+        const results = await retrieval.retrieve('TypeScript');
+        expect(customGenerator).toHaveBeenCalledWith('TypeScript', 3);
+        expect(Array.isArray(results)).toBe(true);
+    });
+    it('deduplicates results from multiple queries', async () => {
+        const searchMock = jest.fn().mockResolvedValue(SEARCH_RESULTS);
+        const vs = { ...makeVectorStore(), search: searchMock };
+        const customGenerator = jest.fn().mockResolvedValue(['q1', 'q2', 'q3']);
+        const retrieval = new multi_query_1.MultiQueryRetrieval(vs, { numQueries: 3, customGenerator });
+        const results = await retrieval.retrieve('test');
+        // d1 and d2 appear in every query result but should only appear once
+        const ids = results.map((r) => r.id);
+        expect(new Set(ids).size).toBe(ids.length);
+    });
+    it('respects topK limit', async () => {
+        const vs = makeVectorStore(SEARCH_RESULTS);
+        const customGenerator = jest.fn().mockResolvedValue(['q1', 'q2']);
+        const retrieval = new multi_query_1.MultiQueryRetrieval(vs, { customGenerator });
+        const results = await retrieval.retrieve('test', { topK: 1 });
+        expect(results.length).toBeLessThanOrEqual(1);
+    });
+    it('falls back to simple variations when no customGenerator and no LLM key', async () => {
+        const vs = makeVectorStore(SEARCH_RESULTS);
+        // Default provider is openai, but no apiKey — will fallback to simple variations
+        const retrieval = new multi_query_1.MultiQueryRetrieval(vs, { llmProvider: 'openai' });
+        // Won't actually call OpenAI; the require('openai') will fail and fall back
+        const results = await retrieval.retrieve('TypeScript');
+        expect(Array.isArray(results)).toBe(true);
+    });
+    it('ranks results by score and frequency bonus', async () => {
+        const highScoreResult = {
+            id: 'high',
+            content: 'high score doc',
+            score: 0.99,
+            metadata: {},
+        };
+        const lowScoreResult = {
+            id: 'low',
+            content: 'low score doc',
+            score: 0.1,
+            metadata: {},
+        };
+        const vs = makeVectorStore([highScoreResult, lowScoreResult]);
+        const customGenerator = jest.fn().mockResolvedValue(['q1', 'q2', 'q3']);
+        const retrieval = new multi_query_1.MultiQueryRetrieval(vs, { numQueries: 3, customGenerator });
+        const results = await retrieval.retrieve('test');
+        // Highest score doc should rank first
+        if (results.length >= 2) {
+            expect(results[0].score).toBeGreaterThanOrEqual(results[1].score);
+        }
+    });
+    it('handles empty search results gracefully', async () => {
+        const vs = makeVectorStore([]);
+        const customGenerator = jest.fn().mockResolvedValue(['q1']);
+        const retrieval = new multi_query_1.MultiQueryRetrieval(vs, { customGenerator });
+        const results = await retrieval.retrieve('nothing');
+        expect(results).toHaveLength(0);
+    });
+    it('uses simple variations as fallback for anthropic provider', async () => {
+        const vs = makeVectorStore(SEARCH_RESULTS);
+        const retrieval = new multi_query_1.MultiQueryRetrieval(vs, { llmProvider: 'anthropic' });
+        const results = await retrieval.retrieve('TypeScript');
+        expect(Array.isArray(results)).toBe(true);
+    });
+});
+//# sourceMappingURL=multi-query.test.js.map

package/dist/__tests__/retrieval/multi-query.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"multi-query.test.js","sourceRoot":"","sources":["../../../src/__tests__/retrieval/multi-query.test.ts"],"names":[],"mappings":";;AAAA,6DAAkE;AAGlE,SAAS,eAAe,CAAC,UAA0B,EAAE;IACnD,OAAO;QACL,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QAClD,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC;QAC7C,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC;QAC5C,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,OAAO,CAAC;QACpD,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACvD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACtD,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;QACnD,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;KAC9C,CAAC;AACJ,CAAC;AAED,MAAM,cAAc,GAAmB;IACrC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE;IAClE,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE;CACnE,CAAC;AAEF,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,EAAE,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;QAC3C,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;QACvF,MAAM,SAAS,GAAG,IAAI,iCAAmB,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC;QAClF,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QACvD,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QAC/D,MAAM,EAAE,GAAG,EAAE,GAAG,eAAe,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;QACxD,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QACxE,MAAM,SAAS,GAAG,IAAI,iCAAmB,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC;QAClF,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACjD,qEAAqE;QACrE,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACnC,MAAM,EAAE,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;QAC3C,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QAClE,MAAM,SAAS,GAAG,IAAI,iCAAmB,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,CAAC,CAAC;QACnE,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9D,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,EAAE,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;QAC3C,iFAAiF;QACjF,MAAM,SAAS,GAAG,IAAI,iCAAmB,CAAC,EAAE,EAAE,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAC;QACzE,4EAA4E;QAC5E,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QACvD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,eAAe,GAAiB;YACpC,EAAE,EAAE,MAAM;YACV,OAAO,EAAE,gBAAgB;YACzB,KAAK,EAAE,IAAI;YACX,QAAQ,EAAE,EAAE;SACb,CAAC;QACF,MAAM,cAAc,GAAiB;YACnC,EAAE,EAAE,KAAK;YACT,OAAO,EAAE,eAAe;YACxB,KAAK,EAAE,GAAG;YACV,QAAQ,EAAE,EAAE;SACb,CAAC;QACF,MAAM,EAAE,GAAG,eAAe,CAAC,CAAC,eAAe,EAAE,cAAc,CAAC,CAAC,CAAC;QAC9D,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QACxE,MAAM,SAAS,GAAG,IAAI,iCAAmB,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC;QAClF,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACjD,sCAAsC;QACtC,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACxB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;QACpE,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;QACvD,MAAM,EAAE,GAAG,eAAe,CAAC,EAAE,CAAC,CAAC;QAC/B,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5D,MAAM,SAAS,GAAG,IAAI,iCAAmB,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,CAAC,CAAC;QACnE,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,EAAE,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;QAC3C,MAAM,SAAS,GAAG,IAAI,iCAAmB,CAAC,EAAE,EAAE,EAAE,WAAW,EAAE,WAAoB,EAAE,CAAC,CAAC;QACrF,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;QACvD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/text-splitters/recursive-text-splitter.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=recursive-text-splitter.test.d.ts.map

package/dist/__tests__/text-splitters/recursive-text-splitter.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"recursive-text-splitter.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/text-splitters/recursive-text-splitter.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/text-splitters/recursive-text-splitter.test.js

@@ -0,0 +1,97 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const recursive_text_splitter_1 = require("../../text-splitters/recursive-text-splitter");
+describe('RecursiveTextSplitter', () => {
+    describe('split', () => {
+        it('returns the whole text when it fits in one chunk', () => {
+            const splitter = new recursive_text_splitter_1.RecursiveTextSplitter({ chunkSize: 1000, chunkOverlap: 0 });
+            const result = splitter.split('hello world');
+            expect(result).toEqual(['hello world']);
+        });
+        it('splits on double newlines first', () => {
+            const splitter = new recursive_text_splitter_1.RecursiveTextSplitter({ chunkSize: 20, chunkOverlap: 0 });
+            const text = 'paragraph one here\n\nparagraph two here';
+            const result = splitter.split(text);
+            expect(result.length).toBeGreaterThan(1);
+        });
+        it('splits on single newlines when paragraphs are too big', () => {
+            const splitter = new recursive_text_splitter_1.RecursiveTextSplitter({ chunkSize: 15, chunkOverlap: 0 });
+            const text = 'line one\nline two\nline three';
+            const result = splitter.split(text);
+            expect(result.length).toBeGreaterThanOrEqual(2);
+        });
+        it('splits long text into multiple chunks', () => {
+            const splitter = new recursive_text_splitter_1.RecursiveTextSplitter({ chunkSize: 10, chunkOverlap: 0 });
+            const text = 'word1 word2 word3 word4 word5 word6';
+            const result = splitter.split(text);
+            expect(result.length).toBeGreaterThan(1);
+        });
+        it('respects custom separators', () => {
+            const splitter = new recursive_text_splitter_1.RecursiveTextSplitter({
+                chunkSize: 20,
+                chunkOverlap: 0,
+                separators: ['|'],
+            });
+            const text = 'part1|part2|part3|part4|part5';
+            const result = splitter.split(text);
+            expect(result.length).toBeGreaterThan(1);
+        });
+        it('adds overlap between chunks', () => {
+            const splitter = new recursive_text_splitter_1.RecursiveTextSplitter({ chunkSize: 20, chunkOverlap: 5 });
+            const text = 'first second third fourth fifth sixth seventh eighth';
+            const result = splitter.split(text);
+            if (result.length > 1) {
+                // second chunk should start with tail of first chunk
+                const firstEnd = result[0].slice(-5);
+                expect(result[1].startsWith(firstEnd)).toBe(true);
+            }
+        });
+        it('no overlap when chunkOverlap is 0', () => {
+            const splitter = new recursive_text_splitter_1.RecursiveTextSplitter({ chunkSize: 10, chunkOverlap: 0 });
+            const text = 'word1 word2 word3 word4 word5';
+            const result = splitter.split(text);
+            // With 0 overlap, splitting result should still be valid
+            expect(result.every((c) => c.length > 0)).toBe(true);
+        });
+    });
+    describe('splitDocuments', () => {
+        it('splits each document and preserves metadata', () => {
+            const splitter = new recursive_text_splitter_1.RecursiveTextSplitter({ chunkSize: 20, chunkOverlap: 0 });
+            const docs = [
+                {
+                    content: 'first paragraph here\n\nsecond paragraph here',
+                    metadata: { source: 'test.txt' },
+                },
+            ];
+            const result = splitter.splitDocuments(docs);
+            expect(result.length).toBeGreaterThan(1);
+            expect(result[0].metadata?.source).toBe('test.txt');
+            expect(result[0].metadata?.chunkIndex).toBe(0);
+            expect(result[0].metadata?.totalChunks).toBeGreaterThan(0);
+        });
+        it('handles multiple documents', () => {
+            const splitter = new recursive_text_splitter_1.RecursiveTextSplitter({ chunkSize: 100, chunkOverlap: 0 });
+            const docs = [
+                { id: 'doc1', content: 'short text', metadata: {} },
+                { id: 'doc2', content: 'another short text', metadata: {} },
+            ];
+            const result = splitter.splitDocuments(docs);
+            expect(result.length).toBe(2);
+            expect(result[0].metadata?.sourceDocId).toBe('doc1');
+            expect(result[1].metadata?.sourceDocId).toBe('doc2');
+        });
+        it('returns empty array for empty input', () => {
+            const splitter = new recursive_text_splitter_1.RecursiveTextSplitter();
+            expect(splitter.splitDocuments([])).toEqual([]);
+        });
+    });
+    describe('default config', () => {
+        it('uses default chunkSize of 1000', () => {
+            const splitter = new recursive_text_splitter_1.RecursiveTextSplitter();
+            const text = 'a'.repeat(500);
+            const result = splitter.split(text);
+            expect(result).toHaveLength(1);
+        });
+    });
+});
+//# sourceMappingURL=recursive-text-splitter.test.js.map

package/dist/__tests__/text-splitters/recursive-text-splitter.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"recursive-text-splitter.test.js","sourceRoot":"","sources":["../../../src/__tests__/text-splitters/recursive-text-splitter.test.ts"],"names":[],"mappings":";;AAAA,0FAAqF;AAErF,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,QAAQ,GAAG,IAAI,+CAAqB,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;YACjF,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,QAAQ,GAAG,IAAI,+CAAqB,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;YAC/E,MAAM,IAAI,GAAG,0CAA0C,CAAC;YACxD,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,QAAQ,GAAG,IAAI,+CAAqB,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;YAC/E,MAAM,IAAI,GAAG,gCAAgC,CAAC;YAC9C,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,QAAQ,GAAG,IAAI,+CAAqB,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;YAC/E,MAAM,IAAI,GAAG,qCAAqC,CAAC;YACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,QAAQ,GAAG,IAAI,+CAAqB,CAAC;gBACzC,SAAS,EAAE,EAAE;gBACb,YAAY,EAAE,CAAC;gBACf,UAAU,EAAE,CAAC,GAAG,CAAC;aAClB,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,+BAA+B,CAAC;YAC7C,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,QAAQ,GAAG,IAAI,+CAAqB,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;YAC/E,MAAM,IAAI,GAAG,sDAAsD,CAAC;YACpE,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtB,qDAAqD;gBACrD,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;gBACrC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,QAAQ,GAAG,IAAI,+CAAqB,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;YAC/E,MAAM,IAAI,GAAG,+BAA+B,CAAC;YAC7C,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpC,yDAAyD;YACzD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,QAAQ,GAAG,IAAI,+CAAqB,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;YAC/E,MAAM,IAAI,GAAG;gBACX;oBACE,OAAO,EAAE,+CAA+C;oBACxD,QAAQ,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE;iBACjC;aACF,CAAC;YACF,MAAM,MAAM,GAAG,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,QAAQ,GAAG,IAAI,+CAAqB,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;YAChF,MAAM,IAAI,GAAG;gBACX,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,QAAQ,EAAE,EAAE,EAAE;gBACnD,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,QAAQ,EAAE,EAAE,EAAE;aAC5D,CAAC;YACF,MAAM,MAAM,GAAG,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,QAAQ,GAAG,IAAI,+CAAqB,EAAE,CAAC;YAC7C,MAAM,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,QAAQ,GAAG,IAAI,+CAAqB,EAAE,CAAC;YAC7C,MAAM,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/__tests__/utils/similarity.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=similarity.test.d.ts.map

package/dist/__tests__/utils/similarity.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"similarity.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/utils/similarity.test.ts"],"names":[],"mappings":""}

package/dist/__tests__/utils/similarity.test.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const similarity_1 = require("../../utils/similarity");
+describe('cosineSimilarity', () => {
+    it('returns 1 for identical vectors', () => {
+        expect((0, similarity_1.cosineSimilarity)([1, 2, 3], [1, 2, 3])).toBeCloseTo(1);
+    });
+    it('returns 0 for orthogonal vectors', () => {
+        expect((0, similarity_1.cosineSimilarity)([1, 0], [0, 1])).toBeCloseTo(0);
+    });
+    it('returns -1 for opposite vectors', () => {
+        expect((0, similarity_1.cosineSimilarity)([1, 0], [-1, 0])).toBeCloseTo(-1);
+    });
+    it('returns 0 for a zero vector', () => {
+        expect((0, similarity_1.cosineSimilarity)([0, 0, 0], [1, 2, 3])).toBe(0);
+    });
+    it('throws when vectors have different lengths', () => {
+        expect(() => (0, similarity_1.cosineSimilarity)([1, 2], [1, 2, 3])).toThrow('same length');
+    });
+    it('computes partial similarity correctly', () => {
+        const result = (0, similarity_1.cosineSimilarity)([1, 1], [1, 0]);
+        expect(result).toBeCloseTo(Math.SQRT1_2);
+    });
+});
+describe('euclideanDistance', () => {
+    it('returns 0 for identical vectors', () => {
+        expect((0, similarity_1.euclideanDistance)([1, 2, 3], [1, 2, 3])).toBe(0);
+    });
+    it('returns correct distance for simple vectors', () => {
+        expect((0, similarity_1.euclideanDistance)([0, 0], [3, 4])).toBeCloseTo(5);
+    });
+    it('throws when vectors have different lengths', () => {
+        expect(() => (0, similarity_1.euclideanDistance)([1, 2], [1])).toThrow('same length');
+    });
+});
+describe('dotProduct', () => {
+    it('returns correct dot product', () => {
+        expect((0, similarity_1.dotProduct)([1, 2, 3], [4, 5, 6])).toBe(32);
+    });
+    it('returns 0 for orthogonal vectors', () => {
+        expect((0, similarity_1.dotProduct)([1, 0], [0, 1])).toBe(0);
+    });
+    it('throws when vectors have different lengths', () => {
+        expect(() => (0, similarity_1.dotProduct)([1, 2], [1])).toThrow('same length');
+    });
+});
+//# sourceMappingURL=similarity.test.js.map

package/dist/__tests__/utils/similarity.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"similarity.test.js","sourceRoot":"","sources":["../../../src/__tests__/utils/similarity.test.ts"],"names":[],"mappings":";;AAAA,uDAAyF;AAEzF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,CAAC,IAAA,6BAAgB,EAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,CAAC,IAAA,6BAAgB,EAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,CAAC,IAAA,6BAAgB,EAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,CAAC,IAAA,6BAAgB,EAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,6BAAgB,EAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,MAAM,GAAG,IAAA,6BAAgB,EAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,CAAC,IAAA,8BAAiB,EAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,CAAC,IAAA,8BAAiB,EAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,8BAAiB,EAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,CAAC,IAAA,uBAAU,EAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,CAAC,IAAA,uBAAU,EAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,CAAC,GAAG,EAAE,CAAC,IAAA,uBAAU,EAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/agentic/decorators/adaptive-retrieval.decorator.d.ts

@@ -5,6 +5,7 @@
 import 'reflect-metadata';
 import { AdaptiveStrategyResult, AgenticLLMProvider } from '../types';
 import { RetrievalStrategy } from '../../types';
+import '../../prompts/agentic/adaptive-retrieval.prompt';
 export interface AdaptiveRetrievalConfig {
     strategies?: RetrievalStrategy[];
     autoSelect?: boolean;

package/dist/agentic/decorators/adaptive-retrieval.decorator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"adaptive-retrieval.decorator.d.ts","sourceRoot":"","sources":["../../../src/agentic/decorators/adaptive-retrieval.decorator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,kBAAkB,CAAC;AAC1B,OAAO,EAAE,sBAAsB,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AACtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAEhD,MAAM,WAAW,uBAAuB;IACtC,UAAU,CAAC,EAAE,iBAAiB,EAAE,CAAC;IACjC,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,WAAW,CAAC,EAAE,kBAAkB,CAAC;CAClC;AAID,wBAAgB,iBAAiB,CAAC,MAAM,GAAE,uBAA4B,GAAG,eAAe,CA4BvF;AAwGD,wBAAgB,mBAAmB,CACjC,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,MAAM,GAAG,MAAM,GAC3B,sBAAsB,GAAG,SAAS,CAEpC"}
+{"version":3,"file":"adaptive-retrieval.decorator.d.ts","sourceRoot":"","sources":["../../../src/agentic/decorators/adaptive-retrieval.decorator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,kBAAkB,CAAC;AAC1B,OAAO,EAAE,sBAAsB,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AACtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAEhD,OAAO,iDAAiD,CAAC;AAGzD,MAAM,WAAW,uBAAuB;IACtC,UAAU,CAAC,EAAE,iBAAiB,EAAE,CAAC;IACjC,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,YAAY,CAAC,EAAE,OAAO,CAAC;IACvB,WAAW,CAAC,EAAE,kBAAkB,CAAC;CAClC;AAID,wBAAgB,iBAAiB,CAAC,MAAM,GAAE,uBAA4B,GAAG,eAAe,CA4BvF;AA0FD,wBAAgB,mBAAmB,CACjC,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,MAAM,GAAG,MAAM,GAC3B,sBAAsB,GAAG,SAAS,CAEpC"}

package/dist/agentic/decorators/adaptive-retrieval.decorator.js

@@ -8,6 +8,9 @@ exports.AdaptiveRetrieval = AdaptiveRetrieval;
 exports.getAdaptiveStrategy = getAdaptiveStrategy;
 require("reflect-metadata");
 const types_1 = require("../../types");
+const prompts_1 = require("@hazeljs/prompts");
+require("../../prompts/agentic/adaptive-retrieval.prompt");
+const adaptive_retrieval_prompt_1 = require("../../prompts/agentic/adaptive-retrieval.prompt");
 const ADAPTIVE_METADATA_KEY = Symbol('adaptive');
 function AdaptiveRetrieval(config = {}) {
     return function (target, propertyKey, descriptor) {
@@ -43,21 +46,7 @@ async function selectStrategy(query, config) {
     return selectStrategyHeuristic(query, strategies);
 }
 async function selectStrategyWithLLM(query, strategies, config) {
-    const prompt = `Select the best retrieval strategy for this query:
-
-Query: ${query}
-
-Available strategies:
-- similarity: Best for semantic similarity
-- hybrid: Combines keyword and semantic search
-- mmr: Maximizes diversity while maintaining relevance
-
-Respond in JSON:
-{
-  "selectedStrategy": "similarity|hybrid|mmr",
-  "reason": "explanation",
-  "confidence": 0.0-1.0
-}`;
+    const prompt = prompts_1.PromptRegistry.get(adaptive_retrieval_prompt_1.ADAPTIVE_RETRIEVAL_KEY).render({ query });
     try {
         const result = await config.llmProvider.generateStructured(prompt, {});
         return {

package/dist/agentic/decorators/adaptive-retrieval.decorator.js.map

@@ -1 +1 @@
-{"version":3,"file":"adaptive-retrieval.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/adaptive-retrieval.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAeH,8CA4BC;AAwGD,kDAKC;AAtJD,4BAA0B;AAE1B,uCAAgD;AAShD,MAAM,qBAAqB,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;AAEjD,SAAgB,iBAAiB,CAAC,SAAkC,EAAE;IACpE,OAAO,UAAU,MAAc,EAAE,WAA4B,EAAE,UAA8B;QAC3F,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,CAAC;QAExC,UAAU,CAAC,KAAK,GAAG,KAAK,WAAW,GAAG,IAAe;YACnD,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;YAEhC,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;gBACvB,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC1C,CAAC;YAED,uBAAuB;YACvB,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAE3D,iBAAiB;YACjB,OAAO,CAAC,cAAc,CAAC,qBAAqB,EAAE,cAAc,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAEnF,iCAAiC;YACjC,MAAM,YAAY,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;YAC/B,IAAI,YAAY,CAAC,CAAC,CAAC,IAAI,OAAO,YAAY,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;gBAC3D,YAAY,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,YAAY,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,gBAAgB,EAAE,CAAC;YACtF,CAAC;YAED,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;QAClD,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,cAAc,CAC3B,KAAa,EACb,MAA+B;IAE/B,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI;QACtC,yBAAiB,CAAC,UAAU;QAC5B,yBAAiB,CAAC,MAAM;QACxB,yBAAiB,CAAC,GAAG;KACtB,CAAC;IAEF,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,OAAO,qBAAqB,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAC1D,CAAC;IAED,OAAO,uBAAuB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;AACpD,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,KAAa,EACb,UAA+B,EAC/B,MAA+B;IAE/B,MAAM,MAAM,GAAG;;SAER,KAAK;;;;;;;;;;;;EAYZ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,WAAY,CAAC,kBAAkB,CAIxD,MAAM,EAAE,EAAE,CAAC,CAAC;QAEf,OAAO;YACL,GAAG,MAAM;YACT,YAAY,EAAE,UAAU;iBACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,MAAM,CAAC,gBAAgB,CAAC;iBAC5C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;SAC7C,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,uBAAuB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IACpD,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAC9B,KAAa,EACb,UAA+B;IAE/B,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAEvC,4BAA4B;IAC5B,IACE,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC;QAChC,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC9B,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,EAC9B,CAAC;QACD,OAAO;YACL,gBAAgB,EAAE,yBAAiB,CAAC,GAAG;YACvC,MAAM,EAAE,gCAAgC;YACxC,UAAU,EAAE,GAAG;YACf,YAAY,EAAE,UAAU;iBACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,yBAAiB,CAAC,GAAG,CAAC;iBAC1C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;SAC7C,CAAC;IACJ,CAAC;IAED,kCAAkC;IAClC,MAAM,gBAAgB,GAAG,oCAAoC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1E,IAAI,gBAAgB,EAAE,CAAC;QACrB,OAAO;YACL,gBAAgB,EAAE,yBAAiB,CAAC,MAAM;YAC1C,MAAM,EAAE,0DAA0D;YAClE,UAAU,EAAE,IAAI;YAChB,YAAY,EAAE,UAAU;iBACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,yBAAiB,CAAC,MAAM,CAAC;iBAC7C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;SAC7C,CAAC;IACJ,CAAC;IAED,wBAAwB;IACxB,OAAO;QACL,gBAAgB,EAAE,yBAAiB,CAAC,UAAU;QAC9C,MAAM,EAAE,yBAAyB;QACjC,UAAU,EAAE,GAAG;QACf,YAAY,EAAE,UAAU;aACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,yBAAiB,CAAC,UAAU,CAAC;aACjD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;KAC7C,CAAC;AACJ,CAAC;AAED,SAAgB,mBAAmB,CACjC,MAAc,EACd,WAA4B;IAE5B,OAAO,OAAO,CAAC,WAAW,CAAC,qBAAqB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;AACzE,CAAC"}
+{"version":3,"file":"adaptive-retrieval.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/adaptive-retrieval.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAkBH,8CA4BC;AA0FD,kDAKC;AA3ID,4BAA0B;AAE1B,uCAAgD;AAChD,8CAAkD;AAClD,2DAAyD;AACzD,+FAAyF;AASzF,MAAM,qBAAqB,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;AAEjD,SAAgB,iBAAiB,CAAC,SAAkC,EAAE;IACpE,OAAO,UAAU,MAAc,EAAE,WAA4B,EAAE,UAA8B;QAC3F,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,CAAC;QAExC,UAAU,CAAC,KAAK,GAAG,KAAK,WAAW,GAAG,IAAe;YACnD,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;YAEhC,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;gBACvB,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC1C,CAAC;YAED,uBAAuB;YACvB,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAE3D,iBAAiB;YACjB,OAAO,CAAC,cAAc,CAAC,qBAAqB,EAAE,cAAc,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAEnF,iCAAiC;YACjC,MAAM,YAAY,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;YAC/B,IAAI,YAAY,CAAC,CAAC,CAAC,IAAI,OAAO,YAAY,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;gBAC3D,YAAY,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,YAAY,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,gBAAgB,EAAE,CAAC;YACtF,CAAC;YAED,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;QAClD,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,cAAc,CAC3B,KAAa,EACb,MAA+B;IAE/B,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI;QACtC,yBAAiB,CAAC,UAAU;QAC5B,yBAAiB,CAAC,MAAM;QACxB,yBAAiB,CAAC,GAAG;KACtB,CAAC;IAEF,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,OAAO,qBAAqB,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAC1D,CAAC;IAED,OAAO,uBAAuB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;AACpD,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,KAAa,EACb,UAA+B,EAC/B,MAA+B;IAE/B,MAAM,MAAM,GAAG,wBAAc,CAAC,GAAG,CAAoB,kDAAsB,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAE/F,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,WAAY,CAAC,kBAAkB,CAIxD,MAAM,EAAE,EAAE,CAAC,CAAC;QAEf,OAAO;YACL,GAAG,MAAM;YACT,YAAY,EAAE,UAAU;iBACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,MAAM,CAAC,gBAAgB,CAAC;iBAC5C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;SAC7C,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,uBAAuB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IACpD,CAAC;AACH,CAAC;AAED,SAAS,uBAAuB,CAC9B,KAAa,EACb,UAA+B;IAE/B,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAEvC,4BAA4B;IAC5B,IACE,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC;QAChC,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC9B,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,EAC9B,CAAC;QACD,OAAO;YACL,gBAAgB,EAAE,yBAAiB,CAAC,GAAG;YACvC,MAAM,EAAE,gCAAgC;YACxC,UAAU,EAAE,GAAG;YACf,YAAY,EAAE,UAAU;iBACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,yBAAiB,CAAC,GAAG,CAAC;iBAC1C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;SAC7C,CAAC;IACJ,CAAC;IAED,kCAAkC;IAClC,MAAM,gBAAgB,GAAG,oCAAoC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1E,IAAI,gBAAgB,EAAE,CAAC;QACrB,OAAO;YACL,gBAAgB,EAAE,yBAAiB,CAAC,MAAM;YAC1C,MAAM,EAAE,0DAA0D;YAClE,UAAU,EAAE,IAAI;YAChB,YAAY,EAAE,UAAU;iBACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,yBAAiB,CAAC,MAAM,CAAC;iBAC7C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;SAC7C,CAAC;IACJ,CAAC;IAED,wBAAwB;IACxB,OAAO;QACL,gBAAgB,EAAE,yBAAiB,CAAC,UAAU;QAC9C,MAAM,EAAE,yBAAyB;QACjC,UAAU,EAAE,GAAG;QACf,YAAY,EAAE,UAAU;aACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,yBAAiB,CAAC,UAAU,CAAC;aACjD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;KAC7C,CAAC;AACJ,CAAC;AAED,SAAgB,mBAAmB,CACjC,MAAc,EACd,WAA4B;IAE5B,OAAO,OAAO,CAAC,WAAW,CAAC,qBAAqB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;AACzE,CAAC"}

package/dist/agentic/decorators/corrective-rag.decorator.d.ts

@@ -4,6 +4,7 @@
  */
 import 'reflect-metadata';
 import { CorrectiveRAGResult, AgenticLLMProvider } from '../types';
+import '../../prompts/agentic/corrective-rag.prompt';
 export interface CorrectiveRAGConfig {
     relevanceThreshold?: number;
     fallbackToWeb?: boolean;

package/dist/agentic/decorators/corrective-rag.decorator.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"corrective-rag.decorator.d.ts","sourceRoot":"","sources":["../../../src/agentic/decorators/corrective-rag.decorator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,kBAAkB,CAAC;AAC1B,OAAO,EAAE,mBAAmB,EAAc,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAE/E,MAAM,WAAW,mBAAmB;IAClC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,CAAC,EAAE,kBAAkB,CAAC;CAClC;AAID,wBAAgB,aAAa,CAAC,MAAM,GAAE,mBAAwB,GAAG,eAAe,CAgD/E;AAuHD,wBAAgB,sBAAsB,CACpC,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,MAAM,GAAG,MAAM,GAC3B,mBAAmB,GAAG,SAAS,CAEjC"}
+{"version":3,"file":"corrective-rag.decorator.d.ts","sourceRoot":"","sources":["../../../src/agentic/decorators/corrective-rag.decorator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,kBAAkB,CAAC;AAC1B,OAAO,EAAE,mBAAmB,EAAc,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAE/E,OAAO,6CAA6C,CAAC;AAGrD,MAAM,WAAW,mBAAmB;IAClC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,CAAC,EAAE,kBAAkB,CAAC;CAClC;AAID,wBAAgB,aAAa,CAAC,MAAM,GAAE,mBAAwB,GAAG,eAAe,CAgD/E;AAgHD,wBAAgB,sBAAsB,CACpC,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,MAAM,GAAG,MAAM,GAC3B,mBAAmB,GAAG,SAAS,CAEjC"}