@eucoder/rag 0.2.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (103) hide show
  1. package/README.md +384 -0
  2. package/dist/ab-testing.d.ts +52 -0
  3. package/dist/ab-testing.d.ts.map +1 -0
  4. package/dist/ab-testing.js +144 -0
  5. package/dist/ab-testing.js.map +1 -0
  6. package/dist/ab-testing.test.d.ts +2 -0
  7. package/dist/ab-testing.test.d.ts.map +1 -0
  8. package/dist/ab-testing.test.js +147 -0
  9. package/dist/ab-testing.test.js.map +1 -0
  10. package/dist/agentic-rag.d.ts +23 -0
  11. package/dist/agentic-rag.d.ts.map +1 -0
  12. package/dist/agentic-rag.js +170 -0
  13. package/dist/agentic-rag.js.map +1 -0
  14. package/dist/agentic-rag.test.d.ts +2 -0
  15. package/dist/agentic-rag.test.d.ts.map +1 -0
  16. package/dist/agentic-rag.test.js +174 -0
  17. package/dist/agentic-rag.test.js.map +1 -0
  18. package/dist/corrective-rag.d.ts +16 -0
  19. package/dist/corrective-rag.d.ts.map +1 -0
  20. package/dist/corrective-rag.js +85 -0
  21. package/dist/corrective-rag.js.map +1 -0
  22. package/dist/corrective-rag.test.d.ts +2 -0
  23. package/dist/corrective-rag.test.d.ts.map +1 -0
  24. package/dist/corrective-rag.test.js +140 -0
  25. package/dist/corrective-rag.test.js.map +1 -0
  26. package/dist/feedback.d.ts +77 -0
  27. package/dist/feedback.d.ts.map +1 -0
  28. package/dist/feedback.js +44 -0
  29. package/dist/feedback.js.map +1 -0
  30. package/dist/feedback.test.d.ts +2 -0
  31. package/dist/feedback.test.d.ts.map +1 -0
  32. package/dist/feedback.test.js +202 -0
  33. package/dist/feedback.test.js.map +1 -0
  34. package/dist/hybrid-search.d.ts +14 -0
  35. package/dist/hybrid-search.d.ts.map +1 -0
  36. package/dist/hybrid-search.js +70 -0
  37. package/dist/hybrid-search.js.map +1 -0
  38. package/dist/hybrid-search.test.d.ts +2 -0
  39. package/dist/hybrid-search.test.d.ts.map +1 -0
  40. package/dist/hybrid-search.test.js +93 -0
  41. package/dist/hybrid-search.test.js.map +1 -0
  42. package/dist/index.d.ts +17 -0
  43. package/dist/index.d.ts.map +1 -0
  44. package/dist/index.js +12 -0
  45. package/dist/index.js.map +1 -0
  46. package/dist/knowledge-graph.d.ts +24 -0
  47. package/dist/knowledge-graph.d.ts.map +1 -0
  48. package/dist/knowledge-graph.js +131 -0
  49. package/dist/knowledge-graph.js.map +1 -0
  50. package/dist/knowledge-graph.test.d.ts +2 -0
  51. package/dist/knowledge-graph.test.d.ts.map +1 -0
  52. package/dist/knowledge-graph.test.js +140 -0
  53. package/dist/knowledge-graph.test.js.map +1 -0
  54. package/dist/llm-grader.d.ts +19 -0
  55. package/dist/llm-grader.d.ts.map +1 -0
  56. package/dist/llm-grader.js +63 -0
  57. package/dist/llm-grader.js.map +1 -0
  58. package/dist/metrics.d.ts +26 -0
  59. package/dist/metrics.d.ts.map +1 -0
  60. package/dist/metrics.js +100 -0
  61. package/dist/metrics.js.map +1 -0
  62. package/dist/optimizer.d.ts +52 -0
  63. package/dist/optimizer.d.ts.map +1 -0
  64. package/dist/optimizer.js +228 -0
  65. package/dist/optimizer.js.map +1 -0
  66. package/dist/optimizer.test.d.ts +2 -0
  67. package/dist/optimizer.test.d.ts.map +1 -0
  68. package/dist/optimizer.test.js +201 -0
  69. package/dist/optimizer.test.js.map +1 -0
  70. package/dist/self-improving.d.ts +85 -0
  71. package/dist/self-improving.d.ts.map +1 -0
  72. package/dist/self-improving.js +163 -0
  73. package/dist/self-improving.js.map +1 -0
  74. package/dist/self-improving.test.d.ts +2 -0
  75. package/dist/self-improving.test.d.ts.map +1 -0
  76. package/dist/self-improving.test.js +234 -0
  77. package/dist/self-improving.test.js.map +1 -0
  78. package/dist/types.d.ts +117 -0
  79. package/dist/types.d.ts.map +1 -0
  80. package/dist/types.js +2 -0
  81. package/dist/types.js.map +1 -0
  82. package/package.json +42 -0
  83. package/src/ab-testing.test.ts +239 -0
  84. package/src/ab-testing.ts +214 -0
  85. package/src/agentic-rag.test.ts +201 -0
  86. package/src/agentic-rag.ts +220 -0
  87. package/src/corrective-rag.test.ts +166 -0
  88. package/src/corrective-rag.ts +115 -0
  89. package/src/feedback.test.ts +227 -0
  90. package/src/feedback.ts +118 -0
  91. package/src/hybrid-search.test.ts +107 -0
  92. package/src/hybrid-search.ts +86 -0
  93. package/src/index.ts +57 -0
  94. package/src/knowledge-graph.test.ts +170 -0
  95. package/src/knowledge-graph.ts +182 -0
  96. package/src/llm-grader.ts +69 -0
  97. package/src/metrics.ts +121 -0
  98. package/src/optimizer.test.ts +232 -0
  99. package/src/optimizer.ts +307 -0
  100. package/src/self-improving.test.ts +341 -0
  101. package/src/self-improving.ts +239 -0
  102. package/src/types.ts +139 -0
  103. package/tsconfig.json +9 -0
@@ -0,0 +1 @@
1
+ {"version":3,"file":"optimizer.test.js","sourceRoot":"","sources":["../src/optimizer.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAGxD,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,SAA6B,CAAC;IAClC,IAAI,OAAgC,CAAC;IAErC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,uBAAuB,EAAE,CAAC;QACxC,SAAS,GAAG,IAAI,kBAAkB,CAAC,OAAO,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,MAAM,GAAsB;YAChC,QAAQ,EAAE,GAAG;YACb,IAAI,EAAE,CAAC;SACR,CAAC;QAEF,mCAAmC;QACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,MAAM,OAAO,CAAC,YAAY,CAAC;gBACzB,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,KAAK,EAAE,SAAS,CAAC,EAAE;gBACnB,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE;gBACtE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;gBACrB,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,iBAAiB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAE5E,MAAM,CAAC,WAAW,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC9C,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,MAAM,GAAsB;YAChC,QAAQ,EAAE,GAAG;YACb,IAAI,EAAE,EAAE;SACT,CAAC;QAEF,qCAAqC;QACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,YAAY,CAAC;gBACzB,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,KAAK,EAAE,SAAS,CAAC,EAAE;gBACnB,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE;gBACtE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;gBACrB,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,iBAAiB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAE5E,MAAM,kBAAkB,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC;QAC1E,MAAM,CAAC,kBAAkB,CAAC,CAAC,WAAW,EAAE,CAAC;QACzC,MAAM,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,QAAS,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,MAAM,GAAsB;YAChC,oBAAoB,EAAE,KAAK;YAC3B,aAAa,EAAE,CAAC;SACjB,CAAC;QAEF,wCAAwC;QACxC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,YAAY,CAAC;gBACzB,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,KAAK,EAAE,SAAS,CAAC,EAAE;gBACnB,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE;gBACtE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;gBACrB,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,iBAAiB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAE5E,MAAM,wBAAwB,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,sBAAsB,CAAC,CAAC;QAC5F,MAAM,CAAC,wBAAwB,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/C,MAAM,CAAC,wBAAwB,EAAE,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,MAAM,GAAsB;YAChC,cAAc,EAAE,GAAG;YACnB,aAAa,EAAE,GAAG;SACnB,CAAC;QAEF,6CAA6C;QAC7C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,YAAY,CAAC;gBACzB,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,KAAK,EAAE,SAAS,CAAC,EAAE;gBACnB,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE;gBACtE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;gBACrB,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,iBAAiB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAE5E,MAAM,wBAAwB,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,gBAAgB,CAAC,CAAC;QACtF,MAAM,CAAC,wBAAwB,CAAC,CAAC,WAAW,EAAE,CAAC;QAC/C,MAAM,CAAC,wBAAwB,EAAE,SAAS,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,cAAe,CAAC,CAAC;IACtF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,aAAa,GAAsB;YACvC,QAAQ,EAAE,GAAG;YACb,IAAI,EAAE,CAAC;YACP,oBAAoB,EAAE,KAAK;SAC5B,CAAC;QAEF,MAAM,WAAW,GAAG;YAClB;gBACE,IAAI,EAAE,WAAoB;gBAC1B,MAAM,EAAE,UAAU;gBAClB,OAAO,EAAE,GAAG;gBACZ,SAAS,EAAE,GAAG;gBACd,mBAAmB,EAAE,IAAI;gBACzB,UAAU,EAAE,GAAG;gBACf,SAAS,EAAE,iBAAiB;aAC7B;YACD;gBACE,IAAI,EAAE,WAAoB;gBAC1B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,CAAC;gBACV,SAAS,EAAE,EAAE;gBACb,mBAAmB,EAAE,GAAG;gBACxB,UAAU,EAAE,GAAG,EAAE,kBAAkB;gBACnC,SAAS,EAAE,iBAAiB;aAC7B;SACF,CAAC;QAEF,MAAM,SAAS,GAAG,SAAS,CAAC,gBAAgB,CAAC,aAAa,EAAE,WAAW,EAAE,GAAG,CAAC,CAAC;QAE9E,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,oCAAoC;QACpE,MAAM,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,MAAM,GAAsB;YAChC,QAAQ,EAAE,GAAG;YACb,IAAI,EAAE,EAAE;SACT,CAAC;QAEF,kBAAkB;QAClB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,YAAY,CAAC;gBACzB,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,KAAK,EAAE,SAAS,CAAC,EAAE;gBACnB,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE;gBACtE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;gBACrB,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,cAAc,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAEpE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mCAAmC,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,MAAM,GAAsB;YAChC,QAAQ,EAAE,GAAG;SACd,CAAC;QAEF,yCAAyC;QACzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,YAAY,CAAC;gBACzB,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,KAAK,EAAE,SAAS,CAAC,EAAE;gBACnB,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE;gBACtE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;gBACrB,SAAS,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC;gBACzC,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC;QAED,yCAAyC;QACzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,OAAO,CAAC,YAAY,CAAC;gBACzB,EAAE,EAAE,OAAO,CAAC,EAAE;gBACd,KAAK,EAAE,SAAS,CAAC,EAAE;gBACnB,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE;gBACtE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;gBACrB,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,SAAS,CAAC,iBAAiB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAE5E,MAAM,eAAe,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,kBAAkB,CAAC,CAAC;QAC/E,MAAM,CAAC,eAAe,CAAC,CAAC,WAAW,EAAE,CAAC;QACtC,MAAM,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,85 @@
1
+ import type { RagAnswer } from "./types.js";
2
+ import type { FeedbackStorage, OptimizationSuggestion } from "./feedback.js";
3
+ import { type RagStrategyConfig } from "./ab-testing.js";
4
+ import { type OptimizableParams } from "./optimizer.js";
5
+ /**
6
+ * Configurazione per il sistema self-improving
7
+ */
8
+ export interface SelfImprovingConfig {
9
+ minFeedbackForOptimization: number;
10
+ minConfidenceForSuggestion: number;
11
+ minSamplesForABTest: number;
12
+ autoApplySuggestions: boolean;
13
+ onOptimization?: (suggestions: OptimizationSuggestion[]) => void;
14
+ onABTestComplete?: (result: any) => void;
15
+ }
16
+ /**
17
+ * Sistema self-improving per RAG
18
+ * Integra feedback collection, A/B testing e ottimizzazione automatica
19
+ */
20
+ export declare class SelfImprovingRag {
21
+ private storage;
22
+ private abFramework;
23
+ private optimizer;
24
+ private config;
25
+ private currentParams;
26
+ constructor(config?: Partial<SelfImprovingConfig>, storage?: FeedbackStorage);
27
+ /**
28
+ * Registra una strategia RAG
29
+ */
30
+ registerStrategy(config: RagStrategyConfig, params: OptimizableParams): void;
31
+ /**
32
+ * Seleziona la migliore strategia basata su metriche storiche
33
+ */
34
+ selectBestStrategy(): Promise<{
35
+ strategy: string;
36
+ params: OptimizableParams;
37
+ } | null>;
38
+ /**
39
+ * Registra feedback per una risposta
40
+ */
41
+ recordFeedback(query: string, answer: RagAnswer, feedback: {
42
+ rating: number;
43
+ relevance: number;
44
+ completeness: number;
45
+ citationsQuality: number;
46
+ comments?: string;
47
+ }, strategy: string): Promise<void>;
48
+ /**
49
+ * Controlla se è tempo di ottimizzare e applica suggerimenti
50
+ */
51
+ private checkAndOptimize;
52
+ /**
53
+ * Esegue A/B test tra due strategie
54
+ */
55
+ runABTest(strategyA: string, strategyB: string): Promise<any>;
56
+ /**
57
+ * Ottiene parametri correnti per una strategia
58
+ */
59
+ getParams(strategy: string): OptimizableParams | undefined;
60
+ /**
61
+ * Aggiorna parametri per una strategia
62
+ */
63
+ updateParams(strategy: string, params: OptimizableParams): void;
64
+ /**
65
+ * Ottiene metriche per una strategia
66
+ */
67
+ getMetrics(strategy: string): Promise<import("./feedback.js").RagMetrics | null>;
68
+ /**
69
+ * Ottiene tutte le metriche
70
+ */
71
+ getAllMetrics(): Promise<import("./feedback.js").RagMetrics[]>;
72
+ /**
73
+ * Genera report completo
74
+ */
75
+ generateReport(strategy: string): Promise<string>;
76
+ /**
77
+ * Ottiene suggerimenti di ottimizzazione
78
+ */
79
+ getSuggestions(strategy: string): Promise<OptimizationSuggestion[]>;
80
+ /**
81
+ * Resetta tutti i feedback (per testing)
82
+ */
83
+ reset(strategy?: string): Promise<void>;
84
+ }
85
+ //# sourceMappingURL=self-improving.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"self-improving.d.ts","sourceRoot":"","sources":["../src/self-improving.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,KAAK,EAAE,eAAe,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAE7E,OAAO,EAAmB,KAAK,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAC1E,OAAO,EAAsB,KAAK,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAE5E;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAElC,0BAA0B,EAAE,MAAM,CAAC;IACnC,0BAA0B,EAAE,MAAM,CAAC;IAGnC,mBAAmB,EAAE,MAAM,CAAC;IAG5B,oBAAoB,EAAE,OAAO,CAAC;IAG9B,cAAc,CAAC,EAAE,CAAC,WAAW,EAAE,sBAAsB,EAAE,KAAK,IAAI,CAAC;IACjE,gBAAgB,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,KAAK,IAAI,CAAC;CAC1C;AAED;;;GAGG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,SAAS,CAAqB;IACtC,OAAO,CAAC,MAAM,CAAgC;IAC9C,OAAO,CAAC,aAAa,CAA6C;gBAEtD,MAAM,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,EAAE,OAAO,CAAC,EAAE,eAAe;IAe5E;;OAEG;IACH,gBAAgB,CAAC,MAAM,EAAE,iBAAiB,EAAE,MAAM,EAAE,iBAAiB,GAAG,IAAI;IAK5E;;OAEG;IACG,kBAAkB,IAAI,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,iBAAiB,CAAA;KAAE,GAAG,IAAI,CAAC;IAiC3F;;OAEG;IACG,cAAc,CAClB,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,SAAS,EACjB,QAAQ,EAAE;QACR,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;QACzB,QAAQ,CAAC,EAAE,MAAM,CAAC;KACnB,EACD,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAgBhB;;OAEG;YACW,gBAAgB;IA4B9B;;OAEG;IACG,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC;IA6BnE;;OAEG;IACH,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,iBAAiB,GAAG,SAAS;IAI1D;;OAEG;IACH,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,iBAAiB,GAAG,IAAI;IAI/D;;OAEG;IACG,UAAU,CAAC,QAAQ,EAAE,MAAM;IAIjC;;OAEG;IACG,aAAa;IAInB;;OAEG;IACG,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAKvD;;OAEG;IACG,cAAc,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC;IAKzE;;OAEG;IACG,KAAK,CAAC,QAAQ,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAG9C"}
@@ -0,0 +1,163 @@
1
+ import { InMemoryFeedbackStorage } from "./feedback.js";
2
+ import { ABTestFramework } from "./ab-testing.js";
3
+ import { ParameterOptimizer } from "./optimizer.js";
4
+ /**
5
+ * Sistema self-improving per RAG
6
+ * Integra feedback collection, A/B testing e ottimizzazione automatica
7
+ */
8
+ export class SelfImprovingRag {
9
+ storage;
10
+ abFramework;
11
+ optimizer;
12
+ config;
13
+ currentParams = new Map();
14
+ constructor(config, storage) {
15
+ this.storage = storage || new InMemoryFeedbackStorage();
16
+ this.abFramework = new ABTestFramework(this.storage);
17
+ this.optimizer = new ParameterOptimizer(this.storage);
18
+ this.config = {
19
+ minFeedbackForOptimization: config?.minFeedbackForOptimization ?? 20,
20
+ minConfidenceForSuggestion: config?.minConfidenceForSuggestion ?? 0.7,
21
+ minSamplesForABTest: config?.minSamplesForABTest ?? 30,
22
+ autoApplySuggestions: config?.autoApplySuggestions ?? false,
23
+ onOptimization: config?.onOptimization ?? (() => { }),
24
+ onABTestComplete: config?.onABTestComplete ?? (() => { }),
25
+ };
26
+ }
27
+ /**
28
+ * Registra una strategia RAG
29
+ */
30
+ registerStrategy(config, params) {
31
+ this.abFramework.registerStrategy(config);
32
+ this.currentParams.set(config.name, params);
33
+ }
34
+ /**
35
+ * Seleziona la migliore strategia basata su metriche storiche
36
+ */
37
+ async selectBestStrategy() {
38
+ const allMetrics = await this.abFramework.getAllMetrics();
39
+ if (allMetrics.length === 0) {
40
+ // Nessuna metrica disponibile, usa prima strategia registrata
41
+ const strategies = Array.from(this.currentParams.keys());
42
+ if (strategies.length === 0)
43
+ return null;
44
+ const strategy = strategies[0];
45
+ return {
46
+ strategy,
47
+ params: this.currentParams.get(strategy),
48
+ };
49
+ }
50
+ // Seleziona strategia con score più alto e confidence sufficiente
51
+ const validMetrics = allMetrics.filter(m => m.confidence >= 0.5);
52
+ if (validMetrics.length === 0) {
53
+ // Usa quella con più feedback
54
+ const best = allMetrics[0];
55
+ return {
56
+ strategy: best.strategy,
57
+ params: this.currentParams.get(best.strategy) || {},
58
+ };
59
+ }
60
+ const best = validMetrics[0];
61
+ return {
62
+ strategy: best.strategy,
63
+ params: this.currentParams.get(best.strategy) || {},
64
+ };
65
+ }
66
+ /**
67
+ * Registra feedback per una risposta
68
+ */
69
+ async recordFeedback(query, answer, feedback, strategy) {
70
+ await this.abFramework.recordFeedback(query, answer, feedback.rating, feedback.relevance, feedback.completeness, feedback.citationsQuality, strategy, feedback.comments);
71
+ // Controlla se è tempo di ottimizzare
72
+ await this.checkAndOptimize(strategy);
73
+ }
74
+ /**
75
+ * Controlla se è tempo di ottimizzare e applica suggerimenti
76
+ */
77
+ async checkAndOptimize(strategy) {
78
+ const feedbacks = await this.storage.getFeedback(strategy);
79
+ if (feedbacks.length < this.config.minFeedbackForOptimization) {
80
+ return; // Non abbastanza feedback
81
+ }
82
+ const currentParams = this.currentParams.get(strategy);
83
+ if (!currentParams)
84
+ return;
85
+ const suggestions = await this.optimizer.analyzeAndSuggest(strategy, currentParams);
86
+ if (suggestions.length > 0) {
87
+ // Notifica suggerimenti
88
+ this.config.onOptimization(suggestions);
89
+ // Auto-applica se configurato
90
+ if (this.config.autoApplySuggestions) {
91
+ const optimizedParams = this.optimizer.applySuggestions(currentParams, suggestions, this.config.minConfidenceForSuggestion);
92
+ this.currentParams.set(strategy, optimizedParams);
93
+ }
94
+ }
95
+ }
96
+ /**
97
+ * Esegue A/B test tra due strategie
98
+ */
99
+ async runABTest(strategyA, strategyB) {
100
+ try {
101
+ const result = await this.abFramework.runABTest(strategyA, strategyB, this.config.minSamplesForABTest);
102
+ this.config.onABTestComplete(result);
103
+ // Se c'è un vincitore chiaro, aggiorna parametri
104
+ if (result.winner !== "tie" && result.statisticalSignificance > 0.8) {
105
+ const winnerStrategy = result.winner === "A" ? strategyA : strategyB;
106
+ const winnerMetrics = result.winner === "A" ? result.metricsA : result.metricsB;
107
+ // Log vittoria
108
+ console.log(`A/B Test: ${winnerStrategy} vince con score ${winnerMetrics.overallScore.toFixed(3)} ` +
109
+ `(miglioramento: ${(result.improvement * 100).toFixed(1)}%)`);
110
+ }
111
+ return result;
112
+ }
113
+ catch (error) {
114
+ console.warn("A/B test non può essere eseguito:", error.message);
115
+ return null;
116
+ }
117
+ }
118
+ /**
119
+ * Ottiene parametri correnti per una strategia
120
+ */
121
+ getParams(strategy) {
122
+ return this.currentParams.get(strategy);
123
+ }
124
+ /**
125
+ * Aggiorna parametri per una strategia
126
+ */
127
+ updateParams(strategy, params) {
128
+ this.currentParams.set(strategy, params);
129
+ }
130
+ /**
131
+ * Ottiene metriche per una strategia
132
+ */
133
+ async getMetrics(strategy) {
134
+ return this.abFramework.getStrategyMetrics(strategy);
135
+ }
136
+ /**
137
+ * Ottiene tutte le metriche
138
+ */
139
+ async getAllMetrics() {
140
+ return this.abFramework.getAllMetrics();
141
+ }
142
+ /**
143
+ * Genera report completo
144
+ */
145
+ async generateReport(strategy) {
146
+ const currentParams = this.currentParams.get(strategy) || {};
147
+ return this.optimizer.generateReport(strategy, currentParams);
148
+ }
149
+ /**
150
+ * Ottiene suggerimenti di ottimizzazione
151
+ */
152
+ async getSuggestions(strategy) {
153
+ const currentParams = this.currentParams.get(strategy) || {};
154
+ return this.optimizer.analyzeAndSuggest(strategy, currentParams);
155
+ }
156
+ /**
157
+ * Resetta tutti i feedback (per testing)
158
+ */
159
+ async reset(strategy) {
160
+ await this.storage.clearFeedback(strategy);
161
+ }
162
+ }
163
+ //# sourceMappingURL=self-improving.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"self-improving.js","sourceRoot":"","sources":["../src/self-improving.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,EAAE,eAAe,EAA0B,MAAM,iBAAiB,CAAC;AAC1E,OAAO,EAAE,kBAAkB,EAA0B,MAAM,gBAAgB,CAAC;AAqB5E;;;GAGG;AACH,MAAM,OAAO,gBAAgB;IACnB,OAAO,CAAkB;IACzB,WAAW,CAAkB;IAC7B,SAAS,CAAqB;IAC9B,MAAM,CAAgC;IACtC,aAAa,GAAmC,IAAI,GAAG,EAAE,CAAC;IAElE,YAAY,MAAqC,EAAE,OAAyB;QAC1E,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,IAAI,uBAAuB,EAAE,CAAC;QACxD,IAAI,CAAC,WAAW,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,IAAI,CAAC,SAAS,GAAG,IAAI,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAEtD,IAAI,CAAC,MAAM,GAAG;YACZ,0BAA0B,EAAE,MAAM,EAAE,0BAA0B,IAAI,EAAE;YACpE,0BAA0B,EAAE,MAAM,EAAE,0BAA0B,IAAI,GAAG;YACrE,mBAAmB,EAAE,MAAM,EAAE,mBAAmB,IAAI,EAAE;YACtD,oBAAoB,EAAE,MAAM,EAAE,oBAAoB,IAAI,KAAK;YAC3D,cAAc,EAAE,MAAM,EAAE,cAAc,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;YACpD,gBAAgB,EAAE,MAAM,EAAE,gBAAgB,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;SACzD,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,MAAyB,EAAE,MAAyB;QACnE,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,kBAAkB;QACtB,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,CAAC;QAE1D,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,8DAA8D;YAC9D,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC;YACzD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC;YAEzC,MAAM,QAAQ,GAAG,UAAU,CAAC,CAAC,CAAE,CAAC;YAChC,OAAO;gBACL,QAAQ;gBACR,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAE;aAC1C,CAAC;QACJ,CAAC;QAED,kEAAkE;QAClE,MAAM,YAAY,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,IAAI,GAAG,CAAC,CAAC;QACjE,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC9B,8BAA8B;YAC9B,MAAM,IAAI,GAAG,UAAU,CAAC,CAAC,CAAE,CAAC;YAC5B,OAAO;gBACL,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE;aACpD,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,GAAG,YAAY,CAAC,CAAC,CAAE,CAAC;QAC9B,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE;SACpD,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAClB,KAAa,EACb,MAAiB,EACjB,QAMC,EACD,QAAgB;QAEhB,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CACnC,KAAK,EACL,MAAM,EACN,QAAQ,CAAC,MAAM,EACf,QAAQ,CAAC,SAAS,EAClB,QAAQ,CAAC,YAAY,EACrB,QAAQ,CAAC,gBAAgB,EACzB,QAAQ,EACR,QAAQ,CAAC,QAAQ,CAClB,CAAC;QAEF,sCAAsC;QACtC,MAAM,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IACxC,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,gBAAgB,CAAC,QAAgB;QAC7C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAE3D,IAAI,SAAS,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,0BAA0B,EAAE,CAAC;YAC9D,OAAO,CAAC,0BAA0B;QACpC,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvD,IAAI,CAAC,aAAa;YAAE,OAAO;QAE3B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;QAEpF,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,wBAAwB;YACxB,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;YAExC,8BAA8B;YAC9B,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;gBACrC,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB,CACrD,aAAa,EACb,WAAW,EACX,IAAI,CAAC,MAAM,CAAC,0BAA0B,CACvC,CAAC;gBACF,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;YACpD,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,SAAiB,EAAE,SAAiB;QAClD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,CAC7C,SAAS,EACT,SAAS,EACT,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAChC,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAErC,iDAAiD;YACjD,IAAI,MAAM,CAAC,MAAM,KAAK,KAAK,IAAI,MAAM,CAAC,uBAAuB,GAAG,GAAG,EAAE,CAAC;gBACpE,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,KAAK,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;gBACrE,MAAM,aAAa,GAAG,MAAM,CAAC,MAAM,KAAK,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC;gBAEhF,eAAe;gBACf,OAAO,CAAC,GAAG,CACT,aAAa,cAAc,oBAAoB,aAAa,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;oBACvF,mBAAmB,CAAC,MAAM,CAAC,WAAW,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAC7D,CAAC;YACJ,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CAAC,mCAAmC,EAAG,KAAe,CAAC,OAAO,CAAC,CAAC;YAC5E,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,QAAgB;QACxB,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,QAAgB,EAAE,MAAyB;QACtD,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,QAAgB;QAC/B,OAAO,IAAI,CAAC,WAAW,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IACvD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa;QACjB,OAAO,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,CAAC;IAC1C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,QAAgB;QACnC,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC7D,OAAO,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IAChE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,QAAgB;QACnC,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC7D,OAAO,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IACnE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,QAAiB;QAC3B,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC7C,CAAC;CACF"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=self-improving.test.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"self-improving.test.d.ts","sourceRoot":"","sources":["../src/self-improving.test.ts"],"names":[],"mappings":""}
@@ -0,0 +1,234 @@
1
+ import { describe, it, expect, beforeEach } from "vitest";
2
+ import { SelfImprovingRag } from "./self-improving.js";
3
+ import { InMemoryFeedbackStorage } from "./feedback.js";
4
+ describe("SelfImprovingRag", () => {
5
+ let selfImproving;
6
+ let storage;
7
+ beforeEach(() => {
8
+ storage = new InMemoryFeedbackStorage();
9
+ selfImproving = new SelfImprovingRag({
10
+ minFeedbackForOptimization: 10,
11
+ minConfidenceForSuggestion: 0.7,
12
+ minSamplesForABTest: 15,
13
+ autoApplySuggestions: false,
14
+ }, storage);
15
+ });
16
+ it("should register strategies with parameters", () => {
17
+ const config = {
18
+ name: "strategy-a",
19
+ type: "corrective",
20
+ params: {},
21
+ };
22
+ const params = {
23
+ minScore: 0.5,
24
+ topK: 10,
25
+ };
26
+ selfImproving.registerStrategy(config, params);
27
+ const retrievedParams = selfImproving.getParams("strategy-a");
28
+ expect(retrievedParams).toEqual(params);
29
+ });
30
+ it("should select best strategy based on metrics", async () => {
31
+ const configA = {
32
+ name: "strategy-a",
33
+ type: "corrective",
34
+ params: {},
35
+ };
36
+ const configB = {
37
+ name: "strategy-b",
38
+ type: "hybrid",
39
+ params: {},
40
+ };
41
+ selfImproving.registerStrategy(configA, { minScore: 0.5 });
42
+ selfImproving.registerStrategy(configB, { semanticWeight: 0.6 });
43
+ // Add feedback for strategy A (lower quality)
44
+ for (let i = 0; i < 20; i++) {
45
+ await selfImproving.recordFeedback(`query ${i}`, { text: `answer ${i}`, citations: [], steps: [], rewrites: 0 }, {
46
+ rating: 3,
47
+ relevance: 0.6,
48
+ completeness: 0.5,
49
+ citationsQuality: 0.7,
50
+ }, "strategy-a");
51
+ }
52
+ // Add feedback for strategy B (higher quality)
53
+ for (let i = 0; i < 20; i++) {
54
+ await selfImproving.recordFeedback(`query ${i}`, { text: `answer ${i}`, citations: [], steps: [], rewrites: 0 }, {
55
+ rating: 5,
56
+ relevance: 0.9,
57
+ completeness: 0.85,
58
+ citationsQuality: 0.95,
59
+ }, "strategy-b");
60
+ }
61
+ const best = await selfImproving.selectBestStrategy();
62
+ expect(best).not.toBeNull();
63
+ expect(best?.strategy).toBe("strategy-b");
64
+ });
65
+ it("should record feedback and trigger optimization check", async () => {
66
+ const config = {
67
+ name: "strategy-a",
68
+ type: "corrective",
69
+ params: {},
70
+ };
71
+ selfImproving.registerStrategy(config, { minScore: 0.3 });
72
+ let optimizationTriggered = false;
73
+ selfImproving = new SelfImprovingRag({
74
+ minFeedbackForOptimization: 10,
75
+ onOptimization: () => {
76
+ optimizationTriggered = true;
77
+ },
78
+ }, storage);
79
+ selfImproving.registerStrategy(config, { minScore: 0.3 });
80
+ // Add 15 feedback with low relevance to trigger optimization
81
+ for (let i = 0; i < 15; i++) {
82
+ await selfImproving.recordFeedback(`query ${i}`, { text: `answer ${i}`, citations: [], steps: [], rewrites: 0 }, {
83
+ rating: 3,
84
+ relevance: 0.5,
85
+ completeness: 0.6,
86
+ citationsQuality: 0.7,
87
+ }, "strategy-a");
88
+ }
89
+ expect(optimizationTriggered).toBe(true);
90
+ });
91
+ it("should auto-apply suggestions when configured", async () => {
92
+ const config = {
93
+ name: "strategy-a",
94
+ type: "corrective",
95
+ params: {},
96
+ };
97
+ selfImproving = new SelfImprovingRag({
98
+ minFeedbackForOptimization: 10,
99
+ autoApplySuggestions: true,
100
+ minConfidenceForSuggestion: 0.7,
101
+ }, storage);
102
+ selfImproving.registerStrategy(config, { minScore: 0.3 });
103
+ // Add 15 feedback with low relevance
104
+ for (let i = 0; i < 15; i++) {
105
+ await selfImproving.recordFeedback(`query ${i}`, { text: `answer ${i}`, citations: [], steps: [], rewrites: 0 }, {
106
+ rating: 3,
107
+ relevance: 0.5,
108
+ completeness: 0.6,
109
+ citationsQuality: 0.7,
110
+ }, "strategy-a");
111
+ }
112
+ const params = selfImproving.getParams("strategy-a");
113
+ // Parameters should be updated if suggestions were applied
114
+ expect(params).toBeDefined();
115
+ });
116
+ it("should run A/B test", async () => {
117
+ const configA = {
118
+ name: "strategy-a",
119
+ type: "corrective",
120
+ params: {},
121
+ };
122
+ const configB = {
123
+ name: "strategy-b",
124
+ type: "hybrid",
125
+ params: {},
126
+ };
127
+ selfImproving.registerStrategy(configA, {});
128
+ selfImproving.registerStrategy(configB, {});
129
+ // Add sufficient samples
130
+ for (let i = 0; i < 20; i++) {
131
+ await selfImproving.recordFeedback(`query ${i}`, { text: `answer ${i}`, citations: [], steps: [], rewrites: 0 }, {
132
+ rating: 3,
133
+ relevance: 0.6,
134
+ completeness: 0.5,
135
+ citationsQuality: 0.7,
136
+ }, "strategy-a");
137
+ await selfImproving.recordFeedback(`query ${i}`, { text: `answer ${i}`, citations: [], steps: [], rewrites: 0 }, {
138
+ rating: 5,
139
+ relevance: 0.9,
140
+ completeness: 0.85,
141
+ citationsQuality: 0.95,
142
+ }, "strategy-b");
143
+ }
144
+ const result = await selfImproving.runABTest("strategy-a", "strategy-b");
145
+ expect(result).not.toBeNull();
146
+ expect(result.winner).toBe("B");
147
+ });
148
+ it("should update parameters manually", () => {
149
+ const config = {
150
+ name: "strategy-a",
151
+ type: "corrective",
152
+ params: {},
153
+ };
154
+ selfImproving.registerStrategy(config, { minScore: 0.5 });
155
+ selfImproving.updateParams("strategy-a", { minScore: 0.7, topK: 15 });
156
+ const params = selfImproving.getParams("strategy-a");
157
+ expect(params?.minScore).toBe(0.7);
158
+ expect(params?.topK).toBe(15);
159
+ });
160
+ it("should get metrics for strategy", async () => {
161
+ const config = {
162
+ name: "strategy-a",
163
+ type: "corrective",
164
+ params: {},
165
+ };
166
+ selfImproving.registerStrategy(config, {});
167
+ for (let i = 0; i < 5; i++) {
168
+ await selfImproving.recordFeedback(`query ${i}`, { text: `answer ${i}`, citations: [], steps: [], rewrites: 0 }, {
169
+ rating: 4,
170
+ relevance: 0.8,
171
+ completeness: 0.7,
172
+ citationsQuality: 0.9,
173
+ }, "strategy-a");
174
+ }
175
+ const metrics = await selfImproving.getMetrics("strategy-a");
176
+ expect(metrics).not.toBeNull();
177
+ expect(metrics?.totalQueries).toBe(5);
178
+ });
179
+ it("should generate report", async () => {
180
+ const config = {
181
+ name: "strategy-a",
182
+ type: "corrective",
183
+ params: {},
184
+ };
185
+ selfImproving.registerStrategy(config, { minScore: 0.5 });
186
+ for (let i = 0; i < 10; i++) {
187
+ await selfImproving.recordFeedback(`query ${i}`, { text: `answer ${i}`, citations: [], steps: [], rewrites: 0 }, {
188
+ rating: 4,
189
+ relevance: 0.7,
190
+ completeness: 0.6,
191
+ citationsQuality: 0.8,
192
+ }, "strategy-a");
193
+ }
194
+ const report = await selfImproving.generateReport("strategy-a");
195
+ expect(report).toContain("Report Ottimizzazione: strategy-a");
196
+ expect(report).toContain("Metriche Attuali");
197
+ });
198
+ it("should get suggestions", async () => {
199
+ const config = {
200
+ name: "strategy-a",
201
+ type: "corrective",
202
+ params: {},
203
+ };
204
+ selfImproving.registerStrategy(config, { minScore: 0.3 });
205
+ for (let i = 0; i < 15; i++) {
206
+ await selfImproving.recordFeedback(`query ${i}`, { text: `answer ${i}`, citations: [], steps: [], rewrites: 0 }, {
207
+ rating: 3,
208
+ relevance: 0.5,
209
+ completeness: 0.6,
210
+ citationsQuality: 0.7,
211
+ }, "strategy-a");
212
+ }
213
+ const suggestions = await selfImproving.getSuggestions("strategy-a");
214
+ expect(suggestions.length).toBeGreaterThan(0);
215
+ });
216
+ it("should reset feedback", async () => {
217
+ const config = {
218
+ name: "strategy-a",
219
+ type: "corrective",
220
+ params: {},
221
+ };
222
+ selfImproving.registerStrategy(config, {});
223
+ await selfImproving.recordFeedback("query", { text: "answer", citations: [], steps: [], rewrites: 0 }, {
224
+ rating: 4,
225
+ relevance: 0.8,
226
+ completeness: 0.7,
227
+ citationsQuality: 0.9,
228
+ }, "strategy-a");
229
+ await selfImproving.reset("strategy-a");
230
+ const metrics = await selfImproving.getMetrics("strategy-a");
231
+ expect(metrics).toBeNull();
232
+ });
233
+ });
234
+ //# sourceMappingURL=self-improving.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"self-improving.test.js","sourceRoot":"","sources":["../src/self-improving.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,uBAAuB,EAAE,MAAM,eAAe,CAAC;AAGxD,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,aAA+B,CAAC;IACpC,IAAI,OAAgC,CAAC;IAErC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,GAAG,IAAI,uBAAuB,EAAE,CAAC;QACxC,aAAa,GAAG,IAAI,gBAAgB,CAClC;YACE,0BAA0B,EAAE,EAAE;YAC9B,0BAA0B,EAAE,GAAG;YAC/B,mBAAmB,EAAE,EAAE;YACvB,oBAAoB,EAAE,KAAK;SAC5B,EACD,OAAO,CACR,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,MAAM,GAAsB;YAChC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,MAAM,MAAM,GAAG;YACb,QAAQ,EAAE,GAAG;YACb,IAAI,EAAE,EAAE;SACT,CAAC;QAEF,aAAa,CAAC,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAE/C,MAAM,eAAe,GAAG,aAAa,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAC9D,MAAM,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,OAAO,GAAsB;YACjC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,MAAM,OAAO,GAAsB;YACjC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,aAAa,CAAC,gBAAgB,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3D,aAAa,CAAC,gBAAgB,CAAC,OAAO,EAAE,EAAE,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC;QAEjE,8CAA8C;QAC9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,aAAa,CAAC,cAAc,CAChC,SAAS,CAAC,EAAE,EACZ,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAC9D;gBACE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;aACtB,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAED,+CAA+C;QAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,aAAa,CAAC,cAAc,CAChC,SAAS,CAAC,EAAE,EACZ,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAC9D;gBACE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,IAAI;gBAClB,gBAAgB,EAAE,IAAI;aACvB,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,aAAa,CAAC,kBAAkB,EAAE,CAAC;QACtD,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC5B,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,MAAM,GAAsB;YAChC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,aAAa,CAAC,gBAAgB,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;QAE1D,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAClC,aAAa,GAAG,IAAI,gBAAgB,CAClC;YACE,0BAA0B,EAAE,EAAE;YAC9B,cAAc,EAAE,GAAG,EAAE;gBACnB,qBAAqB,GAAG,IAAI,CAAC;YAC/B,CAAC;SACF,EACD,OAAO,CACR,CAAC;QACF,aAAa,CAAC,gBAAgB,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;QAE1D,6DAA6D;QAC7D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,aAAa,CAAC,cAAc,CAChC,SAAS,CAAC,EAAE,EACZ,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAC9D;gBACE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;aACtB,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAED,MAAM,CAAC,qBAAqB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,MAAM,GAAsB;YAChC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,aAAa,GAAG,IAAI,gBAAgB,CAClC;YACE,0BAA0B,EAAE,EAAE;YAC9B,oBAAoB,EAAE,IAAI;YAC1B,0BAA0B,EAAE,GAAG;SAChC,EACD,OAAO,CACR,CAAC;QACF,aAAa,CAAC,gBAAgB,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;QAE1D,qCAAqC;QACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,aAAa,CAAC,cAAc,CAChC,SAAS,CAAC,EAAE,EACZ,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAC9D;gBACE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;aACtB,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,aAAa,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACrD,2DAA2D;QAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACnC,MAAM,OAAO,GAAsB;YACjC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,MAAM,OAAO,GAAsB;YACjC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,aAAa,CAAC,gBAAgB,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC5C,aAAa,CAAC,gBAAgB,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAE5C,yBAAyB;QACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,aAAa,CAAC,cAAc,CAChC,SAAS,CAAC,EAAE,EACZ,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAC9D;gBACE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;aACtB,EACD,YAAY,CACb,CAAC;YAEF,MAAM,aAAa,CAAC,cAAc,CAChC,SAAS,CAAC,EAAE,EACZ,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAC9D;gBACE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,IAAI;gBAClB,gBAAgB,EAAE,IAAI;aACvB,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QACzE,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,MAAM,GAAsB;YAChC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,aAAa,CAAC,gBAAgB,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;QAE1D,aAAa,CAAC,YAAY,CAAC,YAAY,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAEtE,MAAM,MAAM,GAAG,aAAa,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,MAAM,GAAsB;YAChC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,aAAa,CAAC,gBAAgB,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAE3C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,MAAM,aAAa,CAAC,cAAc,CAChC,SAAS,CAAC,EAAE,EACZ,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAC9D;gBACE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;aACtB,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QAC7D,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC/B,MAAM,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,MAAM,GAAsB;YAChC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,aAAa,CAAC,gBAAgB,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;QAE1D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,aAAa,CAAC,cAAc,CAChC,SAAS,CAAC,EAAE,EACZ,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAC9D;gBACE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;aACtB,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mCAAmC,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,MAAM,GAAsB;YAChC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,aAAa,CAAC,gBAAgB,CAAC,MAAM,EAAE,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC;QAE1D,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,aAAa,CAAC,cAAc,CAChC,SAAS,CAAC,EAAE,EACZ,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAC9D;gBACE,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,GAAG;gBACd,YAAY,EAAE,GAAG;gBACjB,gBAAgB,EAAE,GAAG;aACtB,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAED,MAAM,WAAW,GAAG,MAAM,aAAa,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QACrE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,KAAK,IAAI,EAAE;QACrC,MAAM,MAAM,GAAsB;YAChC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,EAAE;SACX,CAAC;QAEF,aAAa,CAAC,gBAAgB,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAE3C,MAAM,aAAa,CAAC,cAAc,CAChC,OAAO,EACP,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EACzD;YACE,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,GAAG;YACd,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,GAAG;SACtB,EACD,YAAY,CACb,CAAC;QAEF,MAAM,aAAa,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAExC,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;QAC7D,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC7B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,117 @@
1
+ import type { SearchHit } from "@eucode/indexer";
2
+ export interface Retriever {
3
+ search(query: string, topK: number): Promise<SearchHit[]>;
4
+ }
5
+ export interface KeywordRetriever {
6
+ search(query: string, topK: number): Promise<SearchHit[]>;
7
+ }
8
+ export interface GraderResult {
9
+ relevant: boolean;
10
+ score: number;
11
+ }
12
+ export interface Grader {
13
+ grade(query: string, hit: SearchHit): Promise<GraderResult>;
14
+ }
15
+ export interface QueryRewriter {
16
+ rewrite(query: string, context: string): Promise<string>;
17
+ }
18
+ export interface CorrectiveRagOptions {
19
+ retriever: Retriever;
20
+ grader: Grader;
21
+ rewriter: QueryRewriter;
22
+ maxRewrites?: number;
23
+ minGoodHits?: number;
24
+ minScore?: number;
25
+ webFallback?: (query: string) => Promise<SearchHit[]>;
26
+ }
27
+ export interface RagCitation {
28
+ path: string;
29
+ startLine: number;
30
+ endLine: number;
31
+ text: string;
32
+ }
33
+ export interface RagAnswer {
34
+ text: string;
35
+ citations: RagCitation[];
36
+ steps: string[];
37
+ rewrites: number;
38
+ }
39
+ export interface HybridSearchOptions {
40
+ semanticRetriever: Retriever;
41
+ keywordRetriever: KeywordRetriever;
42
+ semanticWeight?: number;
43
+ keywordWeight?: number;
44
+ rrfK?: number;
45
+ topK?: number;
46
+ }
47
+ export interface HybridSearchResult {
48
+ hits: SearchHit[];
49
+ fusionMethod: "rrf" | "weighted";
50
+ semanticCount: number;
51
+ keywordCount: number;
52
+ }
53
+ export interface Entity {
54
+ id: string;
55
+ name: string;
56
+ type: string;
57
+ properties: Record<string, unknown>;
58
+ }
59
+ export interface Relation {
60
+ id: string;
61
+ source: string;
62
+ target: string;
63
+ type: string;
64
+ properties: Record<string, unknown>;
65
+ }
66
+ export interface KnowledgeGraph {
67
+ entities: Map<string, Entity>;
68
+ relations: Relation[];
69
+ addEntity(entity: Entity): void;
70
+ addRelation(relation: Relation): void;
71
+ getEntity(id: string): Entity | undefined;
72
+ getRelations(entityId: string): Relation[];
73
+ getConnectedEntities(entityId: string, depth?: number): Entity[];
74
+ }
75
+ export interface EntityExtractor {
76
+ extract(text: string): Promise<Entity[]>;
77
+ }
78
+ export interface RelationExtractor {
79
+ extract(text: string, entities: Entity[]): Promise<Relation[]>;
80
+ }
81
+ export interface KnowledgeGraphRagOptions {
82
+ retriever: Retriever;
83
+ entityExtractor: EntityExtractor;
84
+ relationExtractor: RelationExtractor;
85
+ graph: KnowledgeGraph;
86
+ topK?: number;
87
+ maxDepth?: number;
88
+ }
89
+ export interface KnowledgeGraphRagResult {
90
+ answer: string;
91
+ citations: RagCitation[];
92
+ entities: Entity[];
93
+ relations: Relation[];
94
+ graphContext: string;
95
+ }
96
+ export interface ReasoningStep {
97
+ thought: string;
98
+ action: string;
99
+ observation: string;
100
+ }
101
+ export interface AgenticRagOptions {
102
+ retriever: Retriever;
103
+ grader: Grader;
104
+ rewriter: QueryRewriter;
105
+ maxIterations?: number;
106
+ enableSelfReflection?: boolean;
107
+ enableChainOfThought?: boolean;
108
+ }
109
+ export interface AgenticRagResult {
110
+ answer: string;
111
+ citations: RagCitation[];
112
+ reasoningSteps: ReasoningStep[];
113
+ selfReflection?: string;
114
+ iterations: number;
115
+ confidence: number;
116
+ }
117
+ //# sourceMappingURL=types.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAEjD,MAAM,WAAW,SAAS;IACxB,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;CAC3D;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;CAC3D;AAED,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,OAAO,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,MAAM;IACrB,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;CAC7D;AAED,MAAM,WAAW,aAAa;IAC5B,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;CAC1D;AAED,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,SAAS,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,aAAa,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,WAAW,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;CACvD;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAGD,MAAM,WAAW,mBAAmB;IAClC,iBAAiB,EAAE,SAAS,CAAC;IAC7B,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,SAAS,EAAE,CAAC;IAClB,YAAY,EAAE,KAAK,GAAG,UAAU,CAAC;IACjC,aAAa,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;CACtB;AAGD,MAAM,WAAW,MAAM;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC9B,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,WAAW,CAAC,QAAQ,EAAE,QAAQ,GAAG,IAAI,CAAC;IACtC,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;IAC1C,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,QAAQ,EAAE,CAAC;IAC3C,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;CAClE;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;CAC1C;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;CAChE;AAED,MAAM,WAAW,wBAAwB;IACvC,SAAS,EAAE,SAAS,CAAC;IACrB,eAAe,EAAE,eAAe,CAAC;IACjC,iBAAiB,EAAE,iBAAiB,CAAC;IACrC,KAAK,EAAE,cAAc,CAAC;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,uBAAuB;IACtC,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,YAAY,EAAE,MAAM,CAAC;CACtB;AAGD,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE,SAAS,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,aAAa,CAAC;IACxB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oBAAoB,CAAC,EAAE,OAAO,CAAC;IAC/B,oBAAoB,CAAC,EAAE,OAAO,CAAC;CAChC;AAED,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,cAAc,EAAE,aAAa,EAAE,CAAC;IAChC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACpB"}