@hazeljs/rag 0.2.0-alpha.1

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/agentic-rag.service.d.ts

@@ -0,0 +1,49 @@
+/**
+ * Agentic RAG Service
+ * Main service class demonstrating all agentic RAG capabilities
+ */
+import { VectorStore, SearchResult, QueryOptions } from '../types';
+import { AgenticLLMProvider } from './types';
+export interface AgenticRAGServiceConfig {
+    vectorStore: VectorStore;
+    llmProvider?: AgenticLLMProvider;
+    enableAllFeatures?: boolean;
+}
+export declare class AgenticRAGService {
+    private vectorStore;
+    private llmProvider?;
+    constructor(config: AgenticRAGServiceConfig);
+    /**
+     * Basic retrieval with all agentic features
+     */
+    retrieve(query: string, options?: QueryOptions): Promise<SearchResult[]>;
+    /**
+     * Advanced retrieval with HyDE
+     */
+    hydeRetrieve(query: string, options?: QueryOptions): Promise<SearchResult[]>;
+    /**
+     * Multi-hop reasoning retrieval
+     */
+    deepRetrieve(query: string, options?: QueryOptions): Promise<SearchResult[]>;
+    /**
+     * Context-aware conversational retrieval
+     */
+    conversationalRetrieve(query: string, sessionId: string, options?: QueryOptions): Promise<SearchResult[]>;
+    /**
+     * Verified retrieval with source checking
+     */
+    verifiedRetrieve(query: string, options?: QueryOptions): Promise<SearchResult[]>;
+    /**
+     * Learning-enabled retrieval
+     */
+    learningRetrieve(query: string, options?: QueryOptions): Promise<SearchResult[]>;
+    /**
+     * Provide feedback for active learning
+     */
+    provideFeedback(_resultId: string, _rating: number, _relevant: boolean): Promise<void>;
+    /**
+     * Combined: All features enabled
+     */
+    superRetrieve(query: string, sessionId: string, options?: QueryOptions): Promise<SearchResult[]>;
+}
+//# sourceMappingURL=agentic-rag.service.d.ts.map

package/dist/agentic/agentic-rag.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agentic-rag.service.d.ts","sourceRoot":"","sources":["../../src/agentic/agentic-rag.service.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAgB7C,MAAM,WAAW,uBAAuB;IACtC,WAAW,EAAE,WAAW,CAAC;IACzB,WAAW,CAAC,EAAE,kBAAkB,CAAC;IACjC,iBAAiB,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,WAAW,CAAC,CAAqB;gBAE7B,MAAM,EAAE,uBAAuB;IAK3C;;OAEG;IAKG,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAI9E;;OAEG;IAIG,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAIlF;;OAEG;IAEG,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAIlF;;OAEG;IAIG,sBAAsB,CAC1B,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,MAAM,EACjB,OAAO,CAAC,EAAE,YAAY,GACrB,OAAO,CAAC,YAAY,EAAE,CAAC;IAO1B;;OAEG;IAOG,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAItF;;OAEG;IAGG,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAItF;;OAEG;IAEG,eAAe,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IAI5F;;OAEG;IAWG,aAAa,CACjB,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,MAAM,EACjB,OAAO,CAAC,EAAE,YAAY,GACrB,OAAO,CAAC,YAAY,EAAE,CAAC;CAM3B"}