@fllf/agent-sdk 0.1.2 → 0.1.3

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 (266) hide show
  1. package/CHANGELOG.md +169 -0
  2. package/README.md +824 -198
  3. package/dist/agent/Agent.d.ts +16 -5
  4. package/dist/agent/Agent.d.ts.map +1 -1
  5. package/dist/agent/Agent.js +283 -2
  6. package/dist/agent/Agent.js.map +1 -1
  7. package/dist/agent/index.d.ts +1 -1
  8. package/dist/agent/index.d.ts.map +1 -1
  9. package/dist/agent/types.d.ts +27 -3
  10. package/dist/agent/types.d.ts.map +1 -1
  11. package/dist/config/config.d.ts +22 -3
  12. package/dist/config/config.d.ts.map +1 -1
  13. package/dist/config/config.js +194 -47
  14. package/dist/config/config.js.map +1 -1
  15. package/dist/config/index.d.ts +1 -1
  16. package/dist/config/index.d.ts.map +1 -1
  17. package/dist/errors.d.ts +74 -0
  18. package/dist/errors.d.ts.map +1 -0
  19. package/dist/errors.js +186 -0
  20. package/dist/errors.js.map +1 -0
  21. package/dist/executors/base-executor.d.ts +14 -0
  22. package/dist/executors/base-executor.d.ts.map +1 -0
  23. package/dist/executors/base-executor.js +31 -0
  24. package/dist/executors/base-executor.js.map +1 -0
  25. package/dist/executors/base.d.ts +36 -5
  26. package/dist/executors/base.d.ts.map +1 -1
  27. package/dist/executors/chat-request-builder.d.ts +10 -0
  28. package/dist/executors/chat-request-builder.d.ts.map +1 -0
  29. package/dist/executors/chat-request-builder.js +96 -0
  30. package/dist/executors/chat-request-builder.js.map +1 -0
  31. package/dist/executors/index.d.ts +4 -1
  32. package/dist/executors/index.d.ts.map +1 -1
  33. package/dist/executors/index.js +6 -1
  34. package/dist/executors/index.js.map +1 -1
  35. package/dist/executors/rag-executor.js +1 -1
  36. package/dist/executors/rag-executor.js.map +1 -1
  37. package/dist/executors/simple-chat-executor.d.ts +4 -2
  38. package/dist/executors/simple-chat-executor.d.ts.map +1 -1
  39. package/dist/executors/simple-chat-executor.js +59 -57
  40. package/dist/executors/simple-chat-executor.js.map +1 -1
  41. package/dist/executors/tool-calling-executor.d.ts +20 -2
  42. package/dist/executors/tool-calling-executor.d.ts.map +1 -1
  43. package/dist/executors/tool-calling-executor.js +189 -91
  44. package/dist/executors/tool-calling-executor.js.map +1 -1
  45. package/dist/history/base.d.ts +55 -2
  46. package/dist/history/base.d.ts.map +1 -1
  47. package/dist/history/base.js +49 -0
  48. package/dist/history/base.js.map +1 -1
  49. package/dist/history/compression.d.ts +49 -0
  50. package/dist/history/compression.d.ts.map +1 -0
  51. package/dist/history/compression.js +53 -0
  52. package/dist/history/compression.js.map +1 -0
  53. package/dist/history/context-window.d.ts +33 -0
  54. package/dist/history/context-window.d.ts.map +1 -0
  55. package/dist/history/context-window.js +68 -0
  56. package/dist/history/context-window.js.map +1 -0
  57. package/dist/history/in-memory.d.ts +6 -4
  58. package/dist/history/in-memory.d.ts.map +1 -1
  59. package/dist/history/in-memory.js +25 -39
  60. package/dist/history/in-memory.js.map +1 -1
  61. package/dist/history/index.d.ts +9 -2
  62. package/dist/history/index.d.ts.map +1 -1
  63. package/dist/history/index.js +18 -1
  64. package/dist/history/index.js.map +1 -1
  65. package/dist/history/postgres/index.d.ts +4 -0
  66. package/dist/history/postgres/index.d.ts.map +1 -0
  67. package/dist/history/postgres/index.js +20 -0
  68. package/dist/history/postgres/index.js.map +1 -0
  69. package/dist/history/postgres/postgres-message-history-store.d.ts +35 -0
  70. package/dist/history/postgres/postgres-message-history-store.d.ts.map +1 -0
  71. package/dist/history/postgres/postgres-message-history-store.js +195 -0
  72. package/dist/history/postgres/postgres-message-history-store.js.map +1 -0
  73. package/dist/history/postgres/schema.d.ts +18 -0
  74. package/dist/history/postgres/schema.d.ts.map +1 -0
  75. package/dist/history/postgres/schema.js +46 -0
  76. package/dist/history/postgres/schema.js.map +1 -0
  77. package/dist/history/postgres/sql.d.ts +29 -0
  78. package/dist/history/postgres/sql.d.ts.map +1 -0
  79. package/dist/history/postgres/sql.js +102 -0
  80. package/dist/history/postgres/sql.js.map +1 -0
  81. package/dist/history/postgres/types.d.ts +20 -0
  82. package/dist/history/postgres/types.d.ts.map +1 -0
  83. package/dist/history/postgres/types.js +3 -0
  84. package/dist/history/postgres/types.js.map +1 -0
  85. package/dist/history/tool-pairing.d.ts +11 -0
  86. package/dist/history/tool-pairing.d.ts.map +1 -0
  87. package/dist/history/tool-pairing.js +52 -0
  88. package/dist/history/tool-pairing.js.map +1 -0
  89. package/dist/index.d.ts +2 -0
  90. package/dist/index.d.ts.map +1 -1
  91. package/dist/index.js +2 -0
  92. package/dist/index.js.map +1 -1
  93. package/dist/llm/LLM.d.ts +1 -0
  94. package/dist/llm/LLM.d.ts.map +1 -1
  95. package/dist/llm/LLM.js +136 -26
  96. package/dist/llm/LLM.js.map +1 -1
  97. package/dist/llm/errors.d.ts +11 -1
  98. package/dist/llm/errors.d.ts.map +1 -1
  99. package/dist/llm/errors.js +50 -1
  100. package/dist/llm/errors.js.map +1 -1
  101. package/dist/llm/factory.d.ts +14 -2
  102. package/dist/llm/factory.d.ts.map +1 -1
  103. package/dist/llm/factory.js +48 -9
  104. package/dist/llm/factory.js.map +1 -1
  105. package/dist/llm/index.d.ts +8 -3
  106. package/dist/llm/index.d.ts.map +1 -1
  107. package/dist/llm/index.js +10 -1
  108. package/dist/llm/index.js.map +1 -1
  109. package/dist/llm/providers/anthropic.d.ts +14 -0
  110. package/dist/llm/providers/anthropic.d.ts.map +1 -0
  111. package/dist/llm/providers/anthropic.js +336 -0
  112. package/dist/llm/providers/anthropic.js.map +1 -0
  113. package/dist/llm/providers/openai-compatible.d.ts +7 -0
  114. package/dist/llm/providers/openai-compatible.d.ts.map +1 -1
  115. package/dist/llm/providers/openai-compatible.js +151 -47
  116. package/dist/llm/providers/openai-compatible.js.map +1 -1
  117. package/dist/llm/stream-accumulator.d.ts +17 -0
  118. package/dist/llm/stream-accumulator.d.ts.map +1 -0
  119. package/dist/llm/stream-accumulator.js +83 -0
  120. package/dist/llm/stream-accumulator.js.map +1 -0
  121. package/dist/llm/tool-arguments.d.ts +6 -0
  122. package/dist/llm/tool-arguments.d.ts.map +1 -0
  123. package/dist/llm/tool-arguments.js +20 -0
  124. package/dist/llm/tool-arguments.js.map +1 -0
  125. package/dist/llm/types.d.ts +76 -4
  126. package/dist/llm/types.d.ts.map +1 -1
  127. package/dist/messages/content.d.ts +36 -0
  128. package/dist/messages/content.d.ts.map +1 -0
  129. package/dist/messages/content.js +70 -0
  130. package/dist/messages/content.js.map +1 -0
  131. package/dist/messages/index.d.ts +4 -2
  132. package/dist/messages/index.d.ts.map +1 -1
  133. package/dist/messages/index.js +10 -1
  134. package/dist/messages/index.js.map +1 -1
  135. package/dist/messages/message.d.ts +11 -9
  136. package/dist/messages/message.d.ts.map +1 -1
  137. package/dist/messages/message.js +69 -4
  138. package/dist/messages/message.js.map +1 -1
  139. package/dist/messages/types.d.ts +12 -0
  140. package/dist/messages/types.d.ts.map +1 -1
  141. package/dist/observability/collecting-observer.d.ts +8 -0
  142. package/dist/observability/collecting-observer.d.ts.map +1 -0
  143. package/dist/observability/collecting-observer.js +15 -0
  144. package/dist/observability/collecting-observer.js.map +1 -0
  145. package/dist/observability/index.d.ts +1 -0
  146. package/dist/observability/index.d.ts.map +1 -1
  147. package/dist/observability/index.js +3 -1
  148. package/dist/observability/index.js.map +1 -1
  149. package/dist/observability/observer.d.ts.map +1 -1
  150. package/dist/observability/observer.js +14 -1
  151. package/dist/observability/observer.js.map +1 -1
  152. package/dist/observability/types.d.ts +16 -1
  153. package/dist/observability/types.d.ts.map +1 -1
  154. package/dist/rag/chunking/auto-chunker.d.ts +7 -0
  155. package/dist/rag/chunking/auto-chunker.d.ts.map +1 -1
  156. package/dist/rag/chunking/auto-chunker.js +13 -0
  157. package/dist/rag/chunking/auto-chunker.js.map +1 -1
  158. package/dist/rag/chunking/chunker.d.ts.map +1 -1
  159. package/dist/rag/chunking/chunker.js +29 -5
  160. package/dist/rag/chunking/chunker.js.map +1 -1
  161. package/dist/rag/chunking/index.d.ts +2 -0
  162. package/dist/rag/chunking/index.d.ts.map +1 -1
  163. package/dist/rag/chunking/index.js +2 -0
  164. package/dist/rag/chunking/index.js.map +1 -1
  165. package/dist/rag/chunking/markdown-chunker.d.ts.map +1 -1
  166. package/dist/rag/chunking/markdown-chunker.js.map +1 -1
  167. package/dist/rag/chunking/qa-pair-chunker.d.ts +23 -0
  168. package/dist/rag/chunking/qa-pair-chunker.d.ts.map +1 -0
  169. package/dist/rag/chunking/qa-pair-chunker.js +162 -0
  170. package/dist/rag/chunking/qa-pair-chunker.js.map +1 -0
  171. package/dist/rag/chunking/semantic-chunker.d.ts +19 -0
  172. package/dist/rag/chunking/semantic-chunker.d.ts.map +1 -0
  173. package/dist/rag/chunking/semantic-chunker.js +291 -0
  174. package/dist/rag/chunking/semantic-chunker.js.map +1 -0
  175. package/dist/rag/embeddings/embedder.d.ts.map +1 -1
  176. package/dist/rag/embeddings/embedder.js +6 -0
  177. package/dist/rag/embeddings/embedder.js.map +1 -1
  178. package/dist/rag/generation/context-builder.d.ts +7 -0
  179. package/dist/rag/generation/context-builder.d.ts.map +1 -1
  180. package/dist/rag/generation/context-builder.js +4 -1
  181. package/dist/rag/generation/context-builder.js.map +1 -1
  182. package/dist/rag/ingestion/metadata.d.ts +6 -1
  183. package/dist/rag/ingestion/metadata.d.ts.map +1 -1
  184. package/dist/rag/ingestion/metadata.js +6 -2
  185. package/dist/rag/ingestion/metadata.js.map +1 -1
  186. package/dist/rag/pipeline.d.ts.map +1 -1
  187. package/dist/rag/pipeline.js +34 -11
  188. package/dist/rag/pipeline.js.map +1 -1
  189. package/dist/rag/retrieval/index.d.ts +1 -0
  190. package/dist/rag/retrieval/index.d.ts.map +1 -1
  191. package/dist/rag/retrieval/index.js +1 -0
  192. package/dist/rag/retrieval/index.js.map +1 -1
  193. package/dist/rag/retrieval/parent-child-expanding-retriever.d.ts +31 -0
  194. package/dist/rag/retrieval/parent-child-expanding-retriever.d.ts.map +1 -0
  195. package/dist/rag/retrieval/parent-child-expanding-retriever.js +194 -0
  196. package/dist/rag/retrieval/parent-child-expanding-retriever.js.map +1 -0
  197. package/dist/rag/stores/in-memory-keyword-store.d.ts.map +1 -1
  198. package/dist/rag/stores/in-memory-keyword-store.js +3 -8
  199. package/dist/rag/stores/in-memory-keyword-store.js.map +1 -1
  200. package/dist/rag/stores/keyword-tokenizer.d.ts +4 -0
  201. package/dist/rag/stores/keyword-tokenizer.d.ts.map +1 -0
  202. package/dist/rag/stores/keyword-tokenizer.js +113 -0
  203. package/dist/rag/stores/keyword-tokenizer.js.map +1 -0
  204. package/dist/rag/stores/postgres/pg-vector-store.d.ts.map +1 -1
  205. package/dist/rag/stores/postgres/pg-vector-store.js +21 -8
  206. package/dist/rag/stores/postgres/pg-vector-store.js.map +1 -1
  207. package/dist/rag/stores/postgres/postgres-document-store.d.ts.map +1 -1
  208. package/dist/rag/stores/postgres/postgres-document-store.js +26 -13
  209. package/dist/rag/stores/postgres/postgres-document-store.js.map +1 -1
  210. package/dist/rag/stores/postgres/postgres-keyword-store.d.ts.map +1 -1
  211. package/dist/rag/stores/postgres/postgres-keyword-store.js +43 -43
  212. package/dist/rag/stores/postgres/postgres-keyword-store.js.map +1 -1
  213. package/dist/rag/stores/postgres/sql.d.ts +22 -0
  214. package/dist/rag/stores/postgres/sql.d.ts.map +1 -1
  215. package/dist/rag/stores/postgres/sql.js +42 -0
  216. package/dist/rag/stores/postgres/sql.js.map +1 -1
  217. package/dist/rag/stores/types.d.ts +5 -0
  218. package/dist/rag/stores/types.d.ts.map +1 -1
  219. package/dist/rag/types.d.ts +6 -0
  220. package/dist/rag/types.d.ts.map +1 -1
  221. package/dist/testing/agent.d.ts +11 -0
  222. package/dist/testing/agent.d.ts.map +1 -0
  223. package/dist/testing/agent.js +45 -0
  224. package/dist/testing/agent.js.map +1 -0
  225. package/dist/testing/history-contract.d.ts +12 -0
  226. package/dist/testing/history-contract.d.ts.map +1 -0
  227. package/dist/testing/history-contract.js +111 -0
  228. package/dist/testing/history-contract.js.map +1 -0
  229. package/dist/testing/index.d.ts +11 -0
  230. package/dist/testing/index.d.ts.map +1 -0
  231. package/dist/testing/index.js +16 -0
  232. package/dist/testing/index.js.map +1 -0
  233. package/dist/testing/models.d.ts +34 -0
  234. package/dist/testing/models.d.ts.map +1 -0
  235. package/dist/testing/models.js +74 -0
  236. package/dist/testing/models.js.map +1 -0
  237. package/dist/testing/tool-schema.d.ts +6 -0
  238. package/dist/testing/tool-schema.d.ts.map +1 -0
  239. package/dist/testing/tool-schema.js +29 -0
  240. package/dist/testing/tool-schema.js.map +1 -0
  241. package/dist/testing/tools.d.ts +21 -0
  242. package/dist/testing/tools.d.ts.map +1 -0
  243. package/dist/testing/tools.js +43 -0
  244. package/dist/testing/tools.js.map +1 -0
  245. package/dist/tools/base.d.ts +38 -7
  246. package/dist/tools/base.d.ts.map +1 -1
  247. package/dist/tools/base.js +238 -11
  248. package/dist/tools/base.js.map +1 -1
  249. package/dist/tools/builtin/advancedSearchTool.d.ts.map +1 -1
  250. package/dist/tools/builtin/advancedSearchTool.js +30 -4
  251. package/dist/tools/builtin/advancedSearchTool.js.map +1 -1
  252. package/dist/tools/builtin/ragSearchTool.d.ts +6 -3
  253. package/dist/tools/builtin/ragSearchTool.d.ts.map +1 -1
  254. package/dist/tools/builtin/ragSearchTool.js +8 -6
  255. package/dist/tools/builtin/ragSearchTool.js.map +1 -1
  256. package/dist/tools/executor.d.ts +30 -2
  257. package/dist/tools/executor.d.ts.map +1 -1
  258. package/dist/tools/executor.js +83 -15
  259. package/dist/tools/executor.js.map +1 -1
  260. package/dist/tools/index.d.ts +3 -1
  261. package/dist/tools/index.d.ts.map +1 -1
  262. package/dist/tools/registry.d.ts +15 -3
  263. package/dist/tools/registry.d.ts.map +1 -1
  264. package/dist/tools/registry.js +21 -2
  265. package/dist/tools/registry.js.map +1 -1
  266. package/package.json +58 -4
@@ -1 +1 @@
1
- {"version":3,"file":"postgres-document-store.js","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/postgres-document-store.ts"],"names":[],"mappings":";;;AAUA,+BAUe;AAOf,MAAa,qBAAqB;IACb,EAAE,CAAwB;IAC1B,MAAM,CAAqB;IAE5C,YAAY,OAAgC;QACxC,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAA,+BAAyB,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,SAAwB;QAC1C,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAC/B,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CACf;cACF,IAAI,CAAC,MAAM,CAAC,SAAS;;;;;;;;;;;;;;;;;;;;;CAqBlC,CAAC,IAAI,EAAE,EACQ;gBACI,QAAQ,CAAC,EAAE;gBACX,QAAQ,CAAC,OAAO;gBAChB,IAAA,gBAAU,EAAC,QAAQ,CAAC,QAAQ,CAAC;gBAC7B,IAAA,wBAAkB,EAAC,QAAQ,CAAC,MAAM,CAAC;gBACnC,QAAQ,CAAC,QAAQ,CAAC,MAAM;gBACxB,QAAQ,CAAC,QAAQ,CAAC,KAAK,IAAI,IAAI;gBAC/B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,IAAI,IAAI;gBAClC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,IAAI,IAAI;gBAClC,QAAQ,CAAC,QAAQ,CAAC,eAAe,IAAI,IAAI;gBACzC,QAAQ,CAAC,QAAQ,CAAC,GAAG,IAAI,IAAI;gBAC7B,QAAQ,CAAC,QAAQ,CAAC,OAAO,IAAI,IAAI;gBACjC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,IAAI,IAAI;gBAClC,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI;gBACnC,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI;aACtC,CACJ,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAe;QAC9B,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CACf;cACF,IAAI,CAAC,MAAM,CAAC,MAAM;;;;;;;;;;;;;;;;;;;;;;;;;;CA0B/B,CAAC,IAAI,EAAE,EACQ;gBACI,KAAK,CAAC,EAAE;gBACR,KAAK,CAAC,UAAU;gBAChB,KAAK,CAAC,QAAQ,IAAI,IAAI;gBACtB,KAAK,CAAC,OAAO;gBACb,KAAK,CAAC,IAAI;gBACV,IAAA,gBAAU,EAAC,KAAK,CAAC,QAAQ,CAAC;gBAC1B,KAAK,CAAC,QAAQ,CAAC,MAAM;gBACrB,KAAK,CAAC,QAAQ,CAAC,KAAK,IAAI,IAAI;gBAC5B,KAAK,CAAC,QAAQ,CAAC,QAAQ,IAAI,IAAI;gBAC/B,KAAK,CAAC,QAAQ,CAAC,eAAe,IAAI,IAAI;gBACtC,KAAK,CAAC,QAAQ,CAAC,WAAW,IAAI,IAAI;gBAClC,KAAK,CAAC,QAAQ,CAAC,IAAI,IAAI,IAAI;gBAC3B,KAAK,CAAC,QAAQ,CAAC,WAAW,IAAI,IAAI;gBAClC,KAAK,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI;gBAChC,KAAK,CAAC,QAAQ,CAAC,UAAU;gBACzB,KAAK,CAAC,QAAQ,CAAC,GAAG,IAAI,IAAI;gBAC1B,IAAA,wBAAkB,EAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC;aAC3C,CACJ,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,GAAa;QACzB,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnB,OAAO,EAAE,CAAC;QACd,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B;;OAEL,IAAI,CAAC,MAAM,CAAC,MAAM;;CAExB,CAAC,IAAI,EAAE,EACI,CAAC,GAAG,CAAC,CACR,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAExF,OAAO,GAAG;aACL,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;aAC/B,MAAM,CAAC,CAAC,KAAK,EAAkB,EAAE,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,UAAkB;QACxC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B;;OAEL,IAAI,CAAC,MAAM,CAAC,MAAM;;;CAGxB,CAAC,IAAI,EAAE,EACI,CAAC,UAAU,CAAC,CACf,CAAC;QAEF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,UAAkB;QACnC,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CACf,eAAe,IAAI,CAAC,MAAM,CAAC,SAAS,gBAAgB,EACpD,CAAC,UAAU,CAAC,CACf,CAAC;IACN,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,UAAkB;QAChC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B,iBAAiB,IAAI,CAAC,MAAM,CAAC,SAAS,gBAAgB,EACtD,CAAC,UAAU,CAAC,CACf,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE3B,OAAO,GAAG,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;IACjE,CAAC;CACJ;AA/JD,sDA+JC;AAED,SAAS,gBAAgB,CAAC,GAA4B;IAClD,MAAM,QAAQ,GAAG,IAAA,oBAAc,EAAC,GAAG,CAAC,QAAQ,CAA0C,CAAC;IACvF,MAAM,gBAAgB,GAAqB;QACvC,MAAM,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,MAAM,CAAC,IAAI,QAAQ,EAAE,MAAM,IAAI,QAAQ;KACzE,CAAC;IAEF,IAAA,qBAAe,EAAC,gBAAgB,EAAE,OAAO,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,KAAK,CAAC,IAAI,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC7F,IAAA,qBAAe,EAAC,gBAAgB,EAAE,UAAU,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,SAAS,CAAC,IAAI,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACvG,IAAA,qBAAe,EAAC,gBAAgB,EAAE,UAAU,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,SAAS,CAAC,IAAI,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACvG,IAAA,qBAAe,EACX,gBAAgB,EAChB,iBAAiB,EACjB,IAAA,wBAAkB,EAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,QAAQ,EAAE,eAAe,CACzE,CAAC;IACF,IAAA,qBAAe,EAAC,gBAAgB,EAAE,SAAS,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,OAAO,CAAC,IAAI,QAAQ,EAAE,OAAO,CAAC,CAAC;IACnG,IAAA,qBAAe,EAAC,gBAAgB,EAAE,UAAU,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACtG,IAAA,qBAAe,EACX,gBAAgB,EAChB,WAAW,EACX,IAAA,wBAAkB,EAAC,GAAG,CAAC,eAAe,CAAC,IAAI,QAAQ,EAAE,SAAS,CACjE,CAAC;IACF,IAAA,qBAAe,EACX,gBAAgB,EAChB,WAAW,EACX,IAAA,wBAAkB,EAAC,GAAG,CAAC,eAAe,CAAC,IAAI,QAAQ,EAAE,SAAS,CACjE,CAAC;IACF,IAAA,qBAAe,EAAC,gBAAgB,EAAE,KAAK,EAAE,IAAA,6BAAuB,EAAC,GAAG,CAAC,GAAG,CAAC,IAAI,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC5F,IAAA,qBAAe,EAAC,gBAAgB,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;IAE5D,MAAM,QAAQ,GAAgB;QAC1B,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;QAClC,QAAQ,EAAE,gBAAgB;KAC7B,CAAC;IACF,MAAM,MAAM,GAAG,IAAA,mBAAa,EAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACzC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACvB,QAAQ,CAAC,MAAM,GAAG,MAAyB,CAAC;IAChD,CAAC;IAED,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,UAAU,CAAC,GAA4B;IAC5C,MAAM,KAAK,GAAU;QACjB,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC;QACnC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;QAClC,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAc;QACnC,QAAQ,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC;KACpC,CAAC;IAEF,IAAA,qBAAe,EAAC,KAAK,EAAE,UAAU,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IAEtE,OAAO,KAAK,CAAC;AACjB,CAAC"}
1
+ {"version":3,"file":"postgres-document-store.js","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/postgres-document-store.ts"],"names":[],"mappings":";;;AAUA,+BAae;AAOf,gCAAgC;AAChC,qDAAqD;AACrD,MAAM,qBAAqB,GAAG;IAC1B,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE;CACtE,CAAC;AACX,MAAM,kBAAkB,GAAG;IACvB,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS;CAC1F,CAAC;AAEX,MAAa,qBAAqB;IACb,EAAE,CAAwB;IAC1B,MAAM,CAAqB;IAE5C,YAAY,OAAgC;QACxC,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAA,+BAAyB,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,SAAwB;QAC1C,KAAK,MAAM,KAAK,IAAI,IAAA,kBAAY,EAAC,IAAA,iBAAW,EAAC,SAAS,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YAClF,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CACf;cACF,IAAI,CAAC,MAAM,CAAC,SAAS;;;;EAIjC,IAAA,6BAAuB,EAAC;gBACtB,QAAQ,EAAE,KAAK,CAAC,MAAM;gBACtB,KAAK,EAAE,qBAAqB;gBAC5B,SAAS,EAAE,SAAS;aACvB,CAAC;;;;;;;;;;;;;;;;CAgBD,CAAC,IAAI,EAAE,EACQ,KAAK,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC;gBACxB,QAAQ,CAAC,EAAE;gBACX,QAAQ,CAAC,OAAO;gBAChB,IAAA,gBAAU,EAAC,QAAQ,CAAC,QAAQ,CAAC;gBAC7B,IAAA,wBAAkB,EAAC,QAAQ,CAAC,MAAM,CAAC;gBACnC,QAAQ,CAAC,QAAQ,CAAC,MAAM;gBACxB,QAAQ,CAAC,QAAQ,CAAC,KAAK,IAAI,IAAI;gBAC/B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,IAAI,IAAI;gBAClC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,IAAI,IAAI;gBAClC,QAAQ,CAAC,QAAQ,CAAC,eAAe,IAAI,IAAI;gBACzC,QAAQ,CAAC,QAAQ,CAAC,GAAG,IAAI,IAAI;gBAC7B,QAAQ,CAAC,QAAQ,CAAC,OAAO,IAAI,IAAI;gBACjC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,IAAI,IAAI;gBAClC,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI;gBACnC,QAAQ,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI;aACtC,CAAC,CACL,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAAe;QAC9B,KAAK,MAAM,KAAK,IAAI,IAAA,kBAAY,EAAC,IAAA,iBAAW,EAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACzE,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CACf;cACF,IAAI,CAAC,MAAM,CAAC,MAAM;;;;;EAK9B,IAAA,6BAAuB,EAAC;gBACtB,QAAQ,EAAE,KAAK,CAAC,MAAM;gBACtB,KAAK,EAAE,kBAAkB;gBACzB,SAAS,EAAE,SAAS;aACvB,CAAC;;;;;;;;;;;;;;;;;;;CAmBD,CAAC,IAAI,EAAE,EACQ,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;gBACrB,KAAK,CAAC,EAAE;gBACR,KAAK,CAAC,UAAU;gBAChB,KAAK,CAAC,QAAQ,IAAI,IAAI;gBACtB,KAAK,CAAC,OAAO;gBACb,KAAK,CAAC,IAAI;gBACV,IAAA,gBAAU,EAAC,KAAK,CAAC,QAAQ,CAAC;gBAC1B,KAAK,CAAC,QAAQ,CAAC,MAAM;gBACrB,KAAK,CAAC,QAAQ,CAAC,KAAK,IAAI,IAAI;gBAC5B,KAAK,CAAC,QAAQ,CAAC,QAAQ,IAAI,IAAI;gBAC/B,KAAK,CAAC,QAAQ,CAAC,eAAe,IAAI,IAAI;gBACtC,KAAK,CAAC,QAAQ,CAAC,WAAW,IAAI,IAAI;gBAClC,KAAK,CAAC,QAAQ,CAAC,IAAI,IAAI,IAAI;gBAC3B,KAAK,CAAC,QAAQ,CAAC,WAAW,IAAI,IAAI;gBAClC,KAAK,CAAC,QAAQ,CAAC,SAAS,IAAI,IAAI;gBAChC,KAAK,CAAC,QAAQ,CAAC,UAAU;gBACzB,KAAK,CAAC,QAAQ,CAAC,GAAG,IAAI,IAAI;gBAC1B,IAAA,wBAAkB,EAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC;aAC3C,CAAC,CACL,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,GAAa;QACzB,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnB,OAAO,EAAE,CAAC;QACd,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B;;OAEL,IAAI,CAAC,MAAM,CAAC,MAAM;;CAExB,CAAC,IAAI,EAAE,EACI,CAAC,GAAG,CAAC,CACR,CAAC;QACF,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAExF,OAAO,GAAG;aACL,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;aAC/B,MAAM,CAAC,CAAC,KAAK,EAAkB,EAAE,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,UAAkB;QACxC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B;;OAEL,IAAI,CAAC,MAAM,CAAC,MAAM;;;CAGxB,CAAC,IAAI,EAAE,EACI,CAAC,UAAU,CAAC,CACf,CAAC;QAEF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,UAAkB;QACnC,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CACf,eAAe,IAAI,CAAC,MAAM,CAAC,SAAS,gBAAgB,EACpD,CAAC,UAAU,CAAC,CACf,CAAC;IACN,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,UAAkB;QAChC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B,iBAAiB,IAAI,CAAC,MAAM,CAAC,SAAS,gBAAgB,EACtD,CAAC,UAAU,CAAC,CACf,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE3B,OAAO,GAAG,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;IACjE,CAAC;CACJ;AApKD,sDAoKC;AAED,SAAS,gBAAgB,CAAC,GAA4B;IAClD,MAAM,QAAQ,GAAG,IAAA,oBAAc,EAAC,GAAG,CAAC,QAAQ,CAA0C,CAAC;IACvF,MAAM,gBAAgB,GAAqB;QACvC,MAAM,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,MAAM,CAAC,IAAI,QAAQ,EAAE,MAAM,IAAI,QAAQ;KACzE,CAAC;IAEF,IAAA,qBAAe,EAAC,gBAAgB,EAAE,OAAO,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,KAAK,CAAC,IAAI,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC7F,IAAA,qBAAe,EAAC,gBAAgB,EAAE,UAAU,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,SAAS,CAAC,IAAI,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACvG,IAAA,qBAAe,EAAC,gBAAgB,EAAE,UAAU,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,SAAS,CAAC,IAAI,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACvG,IAAA,qBAAe,EACX,gBAAgB,EAChB,iBAAiB,EACjB,IAAA,wBAAkB,EAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,QAAQ,EAAE,eAAe,CACzE,CAAC;IACF,IAAA,qBAAe,EAAC,gBAAgB,EAAE,SAAS,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,OAAO,CAAC,IAAI,QAAQ,EAAE,OAAO,CAAC,CAAC;IACnG,IAAA,qBAAe,EAAC,gBAAgB,EAAE,UAAU,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACtG,IAAA,qBAAe,EACX,gBAAgB,EAChB,WAAW,EACX,IAAA,wBAAkB,EAAC,GAAG,CAAC,eAAe,CAAC,IAAI,QAAQ,EAAE,SAAS,CACjE,CAAC;IACF,IAAA,qBAAe,EACX,gBAAgB,EAChB,WAAW,EACX,IAAA,wBAAkB,EAAC,GAAG,CAAC,eAAe,CAAC,IAAI,QAAQ,EAAE,SAAS,CACjE,CAAC;IACF,IAAA,qBAAe,EAAC,gBAAgB,EAAE,KAAK,EAAE,IAAA,6BAAuB,EAAC,GAAG,CAAC,GAAG,CAAC,IAAI,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC5F,IAAA,qBAAe,EAAC,gBAAgB,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;IAE5D,MAAM,QAAQ,GAAgB;QAC1B,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;QAClC,QAAQ,EAAE,gBAAgB;KAC7B,CAAC;IACF,MAAM,MAAM,GAAG,IAAA,mBAAa,EAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACzC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACvB,QAAQ,CAAC,MAAM,GAAG,MAAyB,CAAC;IAChD,CAAC;IAED,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,UAAU,CAAC,GAA4B;IAC5C,MAAM,KAAK,GAAU;QACjB,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;QAClB,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC;QACnC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;QAClC,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAc;QACnC,QAAQ,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC;KACpC,CAAC;IAEF,IAAA,qBAAe,EAAC,KAAK,EAAE,UAAU,EAAE,IAAA,wBAAkB,EAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IAEtE,OAAO,KAAK,CAAC;AACjB,CAAC"}
@@ -1 +1 @@
1
- {"version":3,"file":"postgres-keyword-store.d.ts","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/postgres-keyword-store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,KAAK,EACR,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EACR,YAAY,EACZ,cAAc,EACd,aAAa,EAChB,MAAM,UAAU,CAAC;AAQlB,OAAO,KAAK,EAER,uBAAuB,EAE1B,MAAM,SAAS,CAAC;AAUjB,qBAAa,oBAAqB,YAAW,YAAY;IACrD,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAwB;IAC3C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAqB;gBAEhC,OAAO,EAAE,uBAAuB;IAKtC,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA4BtC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE,aAAkB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAwC7E,gBAAgB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;YAO3C,mBAAmB;YA8BnB,0BAA0B;YAuB1B,YAAY;CAQ7B"}
1
+ {"version":3,"file":"postgres-keyword-store.d.ts","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/postgres-keyword-store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,KAAK,EACR,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EACR,YAAY,EACZ,cAAc,EACd,aAAa,EAChB,MAAM,UAAU,CAAC;AAWlB,OAAO,KAAK,EAER,uBAAuB,EAE1B,MAAM,SAAS,CAAC;AAcjB,qBAAa,oBAAqB,YAAW,YAAY;IACrD,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAwB;IAC3C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAqB;gBAEhC,OAAO,EAAE,uBAAuB;IAKtC,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAkCtC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE,aAAkB,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAwC7E,gBAAgB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;YAO3C,mBAAmB;YA8BnB,0BAA0B;YAuB1B,YAAY;CAQ7B"}
@@ -2,6 +2,7 @@
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.PostgresKeywordStore = void 0;
4
4
  const sql_1 = require("./sql");
5
+ const keyword_tokenizer_1 = require("../keyword-tokenizer");
5
6
  class PostgresKeywordStore {
6
7
  db;
7
8
  tables;
@@ -10,31 +11,36 @@ class PostgresKeywordStore {
10
11
  this.tables = (0, sql_1.resolvePostgresTableNames)(options.tablePrefix);
11
12
  }
12
13
  async upsert(chunks) {
13
- for (const chunk of chunks) {
14
- const terms = tokenize(chunk.content);
15
- await this.db.query(`
16
- INSERT INTO ${this.tables.keywordEntries} (
17
- chunk_id, document_id, terms, term_counts, length, updated_at
18
- ) VALUES (
19
- $1, $2, $3::text[], $4::jsonb, $5, now()
20
- )
21
- ON CONFLICT (chunk_id) DO UPDATE SET
22
- document_id = EXCLUDED.document_id,
23
- terms = EXCLUDED.terms,
24
- term_counts = EXCLUDED.term_counts,
25
- length = EXCLUDED.length,
26
- updated_at = now()
27
- `.trim(), [
28
- chunk.id,
29
- chunk.documentId,
30
- [...new Set(terms)],
31
- (0, sql_1.createJson)(Object.fromEntries(countTerms(terms))),
32
- terms.length,
33
- ]);
14
+ for (const batch of (0, sql_1.chunkBatches)((0, sql_1.dedupeByKey)(chunks, (chunk) => chunk.id))) {
15
+ await this.db.query(`
16
+ INSERT INTO ${this.tables.keywordEntries} (
17
+ chunk_id, document_id, terms, term_counts, length, updated_at
18
+ ) VALUES
19
+ ${(0, sql_1.createMultiRowValuesSql)({
20
+ rowCount: batch.length,
21
+ casts: ['', '', '::text[]', '::jsonb', ''],
22
+ rowSuffix: ', now()',
23
+ })}
24
+ ON CONFLICT (chunk_id) DO UPDATE SET
25
+ document_id = EXCLUDED.document_id,
26
+ terms = EXCLUDED.terms,
27
+ term_counts = EXCLUDED.term_counts,
28
+ length = EXCLUDED.length,
29
+ updated_at = now()
30
+ `.trim(), batch.flatMap((chunk) => {
31
+ const terms = (0, keyword_tokenizer_1.tokenizeKeywordText)((0, keyword_tokenizer_1.createKeywordIndexText)(chunk));
32
+ return [
33
+ chunk.id,
34
+ chunk.documentId,
35
+ [...new Set(terms)],
36
+ (0, sql_1.createJson)(Object.fromEntries(countTerms(terms))),
37
+ terms.length,
38
+ ];
39
+ }));
34
40
  }
35
41
  }
36
42
  async search(query, options = {}) {
37
- const queryTerms = [...new Set(tokenize(query))];
43
+ const queryTerms = [...new Set((0, keyword_tokenizer_1.tokenizeKeywordText)(query))];
38
44
  const topK = (0, sql_1.normalizeTopK)(options.topK);
39
45
  if (queryTerms.length === 0 || topK === 0) {
40
46
  return [];
@@ -73,17 +79,17 @@ ON CONFLICT (chunk_id) DO UPDATE SET
73
79
  }
74
80
  async loadMatchingEntries(queryTerms, options) {
75
81
  const where = (0, sql_1.buildChunkSearchWhere)(options, 2, 'c');
76
- const result = await this.db.query(`
77
- SELECT
78
- k.chunk_id,
79
- k.document_id,
80
- k.term_counts,
81
- k.length,
82
- c.content
83
- FROM ${this.tables.keywordEntries} k
84
- JOIN ${this.tables.chunks} c ON c.id = k.chunk_id
85
- WHERE k.terms && $1::text[]
86
- ${where.clause.length > 0 ? `AND ${where.clause.slice('WHERE '.length)}` : ''}
82
+ const result = await this.db.query(`
83
+ SELECT
84
+ k.chunk_id,
85
+ k.document_id,
86
+ k.term_counts,
87
+ k.length,
88
+ c.content
89
+ FROM ${this.tables.keywordEntries} k
90
+ JOIN ${this.tables.chunks} c ON c.id = k.chunk_id
91
+ WHERE k.terms && $1::text[]
92
+ ${where.clause.length > 0 ? `AND ${where.clause.slice('WHERE '.length)}` : ''}
87
93
  `.trim(), [queryTerms, ...where.values]);
88
94
  return result.rows.map((row) => ({
89
95
  chunkId: String(row.chunk_id),
@@ -94,10 +100,10 @@ ${where.clause.length > 0 ? `AND ${where.clause.slice('WHERE '.length)}` : ''}
94
100
  }));
95
101
  }
96
102
  async calculateDocumentFrequency(queryTerms) {
97
- const result = await this.db.query(`
98
- SELECT terms
99
- FROM ${this.tables.keywordEntries}
100
- WHERE terms && $1::text[]
103
+ const result = await this.db.query(`
104
+ SELECT terms
105
+ FROM ${this.tables.keywordEntries}
106
+ WHERE terms && $1::text[]
101
107
  `.trim(), [queryTerms]);
102
108
  const frequency = new Map(queryTerms.map((term) => [term, 0]));
103
109
  for (const row of result.rows) {
@@ -131,12 +137,6 @@ function scoreEntry(input) {
131
137
  }
132
138
  return score;
133
139
  }
134
- function tokenize(text) {
135
- return text
136
- .normalize('NFKC')
137
- .toLowerCase()
138
- .match(/[\p{L}\p{N}_-]+/gu) ?? [];
139
- }
140
140
  function countTerms(terms) {
141
141
  const counts = new Map();
142
142
  for (const term of terms) {
@@ -1 +1 @@
1
- {"version":3,"file":"postgres-keyword-store.js","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/postgres-keyword-store.ts"],"names":[],"mappings":";;;AAQA,+BAMe;AAef,MAAa,oBAAoB;IACZ,EAAE,CAAwB;IAC1B,MAAM,CAAqB;IAE5C,YAAY,OAAgC;QACxC,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAA,+BAAyB,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAe;QACxB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACzB,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACtC,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CACf;cACF,IAAI,CAAC,MAAM,CAAC,cAAc;;;;;;;;;;;CAWvC,CAAC,IAAI,EAAE,EACQ;gBACI,KAAK,CAAC,EAAE;gBACR,KAAK,CAAC,UAAU;gBAChB,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;gBACnB,IAAA,gBAAU,EAAC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC;gBACjD,KAAK,CAAC,MAAM;aACf,CACJ,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,UAAyB,EAAE;QACnD,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACjD,MAAM,IAAI,GAAG,IAAA,mBAAa,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;YACxC,OAAO,EAAE,CAAC;QACd,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACpE,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,0BAA0B,CAAC,UAAU,CAAC,CAAC;QAC5E,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QACjD,MAAM,IAAI,GAAG,OAAO;aACf,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YACb,KAAK;YACL,KAAK,EAAE,UAAU,CAAC;gBACd,KAAK;gBACL,UAAU;gBACV,iBAAiB;gBACjB,cAAc;aACjB,CAAC;SACL,CAAC,CAAC;aACF,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;aAChC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,IAAI,OAAO,CAAC,QAAQ,CAAC;aAClF,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CACnB,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK;eACrB,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC;eAC3D,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAC3D,CAAC;aACD,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QAEpB,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACvB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO;YAC3B,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU;YACjC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ,EAAE;gBACN,MAAM,EAAE,SAAS;gBACjB,YAAY,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;aAC7E;SACJ,CAAC,CAAC,CAAC;IACR,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,UAAkB;QACrC,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CACf,eAAe,IAAI,CAAC,MAAM,CAAC,cAAc,yBAAyB,EAClE,CAAC,UAAU,CAAC,CACf,CAAC;IACN,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC7B,UAAoB,EACpB,OAAsB;QAEtB,MAAM,KAAK,GAAG,IAAA,2BAAqB,EAAC,OAAO,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B;;;;;;;OAOL,IAAI,CAAC,MAAM,CAAC,cAAc;OAC1B,IAAI,CAAC,MAAM,CAAC,MAAM;;EAEvB,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE;CAC5E,CAAC,IAAI,EAAE,EACI,CAAC,UAAU,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,CAChC,CAAC;QAEF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAC7B,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;YAC7B,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC;YACnC,UAAU,EAAE,gBAAgB,CAAC,GAAG,CAAC,WAAW,CAAC;YAC7C,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,CAAC;YAC/B,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;SACrC,CAAC,CAAC,CAAC;IACR,CAAC;IAEO,KAAK,CAAC,0BAA0B,CAAC,UAAoB;QACzD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B;;OAEL,IAAI,CAAC,MAAM,CAAC,cAAc;;CAEhC,CAAC,IAAI,EAAE,EACI,CAAC,UAAU,CAAC,CACf,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAE/D,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5B,MAAM,KAAK,GAAG,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACzC,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC5B,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;oBACvB,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;gBACxD,CAAC;YACL,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,YAAY;QACtB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B,iCAAiC,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAChE,CAAC;QACF,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;QAEjD,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAC9B,CAAC;CACJ;AAjJD,oDAiJC;AAED,SAAS,UAAU,CAAC,KAKnB;IACG,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QAClC,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxD,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;YAClB,SAAS;QACb,CAAC;QAED,MAAM,EAAE,GAAG,KAAK,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,GAAG,EAAE,GAAG,GAAG,CAAC,GAAG,CAAC,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC;QACzE,MAAM,YAAY,GAAG,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC;QAC5E,KAAK,IAAI,YAAY,GAAG,GAAG,CAAC;IAChC,CAAC;IAED,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC1B,OAAO,IAAI;SACN,SAAS,CAAC,MAAM,CAAC;SACjB,WAAW,EAAE;SACb,KAAK,CAAC,mBAAmB,CAAC,IAAI,EAAE,CAAC;AAC1C,CAAC;AAED,SAAS,UAAU,CAAC,KAAe;IAC/B,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;IACzC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAClD,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACpC,MAAM,MAAM,GAAG,IAAA,oBAAc,EAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IAC3C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;IACzC,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACjD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC5B,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC5B,CAAC;IACL,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACnC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,EAAE,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC;AAC5E,CAAC"}
1
+ {"version":3,"file":"postgres-keyword-store.js","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/postgres-keyword-store.ts"],"names":[],"mappings":";;;AAQA,+BASe;AAMf,4DAG8B;AAU9B,MAAa,oBAAoB;IACZ,EAAE,CAAwB;IAC1B,MAAM,CAAqB;IAE5C,YAAY,OAAgC;QACxC,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAA,+BAAyB,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAe;QACxB,KAAK,MAAM,KAAK,IAAI,IAAA,kBAAY,EAAC,IAAA,iBAAW,EAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACzE,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CACf;cACF,IAAI,CAAC,MAAM,CAAC,cAAc;;;EAGtC,IAAA,6BAAuB,EAAC;gBACtB,QAAQ,EAAE,KAAK,CAAC,MAAM;gBACtB,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,CAAC;gBAC1C,SAAS,EAAE,SAAS;aACvB,CAAC;;;;;;;CAOD,CAAC,IAAI,EAAE,EACQ,KAAK,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBACpB,MAAM,KAAK,GAAG,IAAA,uCAAmB,EAAC,IAAA,0CAAsB,EAAC,KAAK,CAAC,CAAC,CAAC;gBAEjE,OAAO;oBACH,KAAK,CAAC,EAAE;oBACR,KAAK,CAAC,UAAU;oBAChB,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC;oBACnB,IAAA,gBAAU,EAAC,MAAM,CAAC,WAAW,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC;oBACjD,KAAK,CAAC,MAAM;iBACf,CAAC;YACN,CAAC,CAAC,CACL,CAAC;QACN,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,UAAyB,EAAE;QACnD,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,IAAA,uCAAmB,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5D,MAAM,IAAI,GAAG,IAAA,mBAAa,EAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;YACxC,OAAO,EAAE,CAAC;QACd,CAAC;QAED,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACpE,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,0BAA0B,CAAC,UAAU,CAAC,CAAC;QAC5E,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QACjD,MAAM,IAAI,GAAG,OAAO;aACf,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YACb,KAAK;YACL,KAAK,EAAE,UAAU,CAAC;gBACd,KAAK;gBACL,UAAU;gBACV,iBAAiB;gBACjB,cAAc;aACjB,CAAC;SACL,CAAC,CAAC;aACF,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,GAAG,CAAC,CAAC;aAChC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,KAAK,SAAS,IAAI,IAAI,CAAC,KAAK,IAAI,OAAO,CAAC,QAAQ,CAAC;aAClF,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CACnB,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK;eACrB,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC;eAC3D,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAC3D,CAAC;aACD,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QAEpB,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACvB,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO;YAC3B,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU;YACjC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,QAAQ,EAAE;gBACN,MAAM,EAAE,SAAS;gBACjB,YAAY,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;aAC7E;SACJ,CAAC,CAAC,CAAC;IACR,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,UAAkB;QACrC,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CACf,eAAe,IAAI,CAAC,MAAM,CAAC,cAAc,yBAAyB,EAClE,CAAC,UAAU,CAAC,CACf,CAAC;IACN,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAC7B,UAAoB,EACpB,OAAsB;QAEtB,MAAM,KAAK,GAAG,IAAA,2BAAqB,EAAC,OAAO,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B;;;;;;;OAOL,IAAI,CAAC,MAAM,CAAC,cAAc;OAC1B,IAAI,CAAC,MAAM,CAAC,MAAM;;EAEvB,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE;CAC5E,CAAC,IAAI,EAAE,EACI,CAAC,UAAU,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,CAChC,CAAC;QAEF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAC7B,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;YAC7B,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC;YACnC,UAAU,EAAE,gBAAgB,CAAC,GAAG,CAAC,WAAW,CAAC;YAC7C,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,CAAC;YAC/B,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,OAAO,IAAI,EAAE,CAAC;SACrC,CAAC,CAAC,CAAC;IACR,CAAC;IAEO,KAAK,CAAC,0BAA0B,CAAC,UAAoB;QACzD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B;;OAEL,IAAI,CAAC,MAAM,CAAC,cAAc;;CAEhC,CAAC,IAAI,EAAE,EACI,CAAC,UAAU,CAAC,CACf,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAE/D,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5B,MAAM,KAAK,GAAG,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACzC,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC5B,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;oBACvB,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;gBACxD,CAAC;YACL,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,YAAY;QACtB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,KAAK,CAC9B,iCAAiC,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAChE,CAAC;QACF,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC;QAEjD,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAC9B,CAAC;CACJ;AAvJD,oDAuJC;AAED,SAAS,UAAU,CAAC,KAKnB;IACG,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QAClC,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxD,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;YAClB,SAAS;QACb,CAAC;QAED,MAAM,EAAE,GAAG,KAAK,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,cAAc,GAAG,EAAE,GAAG,GAAG,CAAC,GAAG,CAAC,EAAE,GAAG,GAAG,CAAC,CAAC,CAAC;QACzE,MAAM,YAAY,GAAG,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,CAAC;QAC5E,KAAK,IAAI,YAAY,GAAG,GAAG,CAAC;IAChC,CAAC;IAED,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAS,UAAU,CAAC,KAAe;IAC/B,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;IACzC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAClD,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACpC,MAAM,MAAM,GAAG,IAAA,oBAAc,EAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IAC3C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;IACzC,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACjD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC5B,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC5B,CAAC;IACL,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACnC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,EAAE,CAAC;IACd,CAAC;IAED,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC;AAC5E,CAAC"}
@@ -16,6 +16,28 @@ export interface SqlWhere {
16
16
  export declare function buildChunkSearchWhere(options: SearchOptions, startIndex?: number, alias?: string): SqlWhere;
17
17
  export declare function assignIfDefined<T extends object, K extends keyof T>(target: T, key: K, value: T[K] | undefined): void;
18
18
  export declare function normalizeTopK(topK: number | undefined): number;
19
+ /**
20
+ * 批量 upsert 的单批行数上限。500 行是往返次数与单语句参数量之间的平衡:
21
+ * 最宽的表(chunks,17 列)一批也只有 8500 个参数,远低于 Postgres 65535 的协议上限。
22
+ */
23
+ export declare const UPSERT_BATCH_SIZE = 500;
24
+ export declare function chunkBatches<T>(items: readonly T[], batchSize?: number): T[][];
25
+ /**
26
+ * 同一条 multi-row INSERT 里出现重复冲突键时,Postgres 会报
27
+ * "ON CONFLICT DO UPDATE command cannot affect row a second time"。
28
+ * 逐行 INSERT 的旧语义是"后写覆盖先写",这里按 key 去重保留最后一条以保持语义一致。
29
+ */
30
+ export declare function dedupeByKey<T>(items: readonly T[], keyOf: (item: T) => string): T[];
31
+ /**
32
+ * 生成 multi-row VALUES 的占位符片段,例如:
33
+ * ($1, $2::jsonb, now()),\n($3, $4::jsonb, now())。
34
+ * casts 按列声明类型转换后缀('' 表示无转换),rowSuffix 用于追加 now() 这类字面量列。
35
+ */
36
+ export declare function createMultiRowValuesSql(input: {
37
+ rowCount: number;
38
+ casts: readonly string[];
39
+ rowSuffix?: string;
40
+ }): string;
19
41
  export declare function rowToChunkMetadata(row: Record<string, unknown>): ChunkMetadata;
20
42
  export declare function readOptionalString(value: unknown): string | undefined;
21
43
  export declare function readOptionalNumber(value: unknown): number | undefined;
@@ -1 +1 @@
1
- {"version":3,"file":"sql.d.ts","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/sql.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,aAAa,EAChB,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EACR,aAAa,EAChB,MAAM,UAAU,CAAC;AAClB,OAAO,KAAK,EACR,kBAAkB,EACrB,MAAM,SAAS,CAAC;AAKjB,wBAAgB,yBAAyB,CAAC,WAAW,SAAuB,GAAG,kBAAkB,CAShG;AAED,wBAAgB,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAI1D;AAED,wBAAgB,UAAU,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAEjD;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,IAAI,CAEhE;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAIlF;AAED,wBAAgB,aAAa,CAAC,CAAC,GAAG,OAAO,EAAE,KAAK,EAAE,OAAO,GAAG,CAAC,EAAE,GAAG,SAAS,CAI1E;AAED,wBAAgB,QAAQ,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAMhD;AAED,wBAAgB,eAAe,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,GAAG,MAAM,CAEjE;AAED,MAAM,WAAW,QAAQ;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,OAAO,EAAE,CAAC;CACrB;AAED,wBAAgB,qBAAqB,CACjC,OAAO,EAAE,aAAa,EACtB,UAAU,SAAI,EACd,KAAK,SAAM,GACZ,QAAQ,CAsCV;AA4CD,wBAAgB,eAAe,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,SAAS,MAAM,CAAC,EAC/D,MAAM,EAAE,CAAC,EACT,GAAG,EAAE,CAAC,EACN,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS,GACxB,IAAI,CAIN;AAED,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,GAAG,MAAM,CAS9D;AAED,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,aAAa,CAiB9E;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAErE;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAWrE;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,EAAE,GAAG,SAAS,CAM5E"}
1
+ {"version":3,"file":"sql.d.ts","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/sql.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,aAAa,EAChB,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EACR,aAAa,EAChB,MAAM,UAAU,CAAC;AAClB,OAAO,KAAK,EACR,kBAAkB,EACrB,MAAM,SAAS,CAAC;AAKjB,wBAAgB,yBAAyB,CAAC,WAAW,SAAuB,GAAG,kBAAkB,CAShG;AAED,wBAAgB,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAI1D;AAED,wBAAgB,UAAU,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAEjD;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,IAAI,CAEhE;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAIlF;AAED,wBAAgB,aAAa,CAAC,CAAC,GAAG,OAAO,EAAE,KAAK,EAAE,OAAO,GAAG,CAAC,EAAE,GAAG,SAAS,CAI1E;AAED,wBAAgB,QAAQ,CAAC,KAAK,EAAE,OAAO,GAAG,OAAO,CAMhD;AAED,wBAAgB,eAAe,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,GAAG,MAAM,CAEjE;AAED,MAAM,WAAW,QAAQ;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,OAAO,EAAE,CAAC;CACrB;AAED,wBAAgB,qBAAqB,CACjC,OAAO,EAAE,aAAa,EACtB,UAAU,SAAI,EACd,KAAK,SAAM,GACZ,QAAQ,CAsCV;AA4CD,wBAAgB,eAAe,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,SAAS,MAAM,CAAC,EAC/D,MAAM,EAAE,CAAC,EACT,GAAG,EAAE,CAAC,EACN,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,SAAS,GACxB,IAAI,CAIN;AAED,wBAAgB,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS,GAAG,MAAM,CAS9D;AAED;;;GAGG;AACH,eAAO,MAAM,iBAAiB,MAAM,CAAC;AAErC,wBAAgB,YAAY,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,CAAC,EAAE,EAAE,SAAS,SAAoB,GAAG,CAAC,EAAE,EAAE,CAOzF;AAED;;;;GAIG;AACH,wBAAgB,WAAW,CAAC,CAAC,EAAE,KAAK,EAAE,SAAS,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,MAAM,GAAG,CAAC,EAAE,CAOnF;AAED;;;;GAIG;AACH,wBAAgB,uBAAuB,CAAC,KAAK,EAAE;IAC3C,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,SAAS,MAAM,EAAE,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB,GAAG,MAAM,CAWT;AAED,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,aAAa,CAiB9E;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAErE;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAWrE;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,EAAE,GAAG,SAAS,CAM5E"}
@@ -1,5 +1,6 @@
1
1
  "use strict";
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.UPSERT_BATCH_SIZE = void 0;
3
4
  exports.resolvePostgresTableNames = resolvePostgresTableNames;
4
5
  exports.quoteIdentifier = quoteIdentifier;
5
6
  exports.createJson = createJson;
@@ -11,6 +12,9 @@ exports.toVectorLiteral = toVectorLiteral;
11
12
  exports.buildChunkSearchWhere = buildChunkSearchWhere;
12
13
  exports.assignIfDefined = assignIfDefined;
13
14
  exports.normalizeTopK = normalizeTopK;
15
+ exports.chunkBatches = chunkBatches;
16
+ exports.dedupeByKey = dedupeByKey;
17
+ exports.createMultiRowValuesSql = createMultiRowValuesSql;
14
18
  exports.rowToChunkMetadata = rowToChunkMetadata;
15
19
  exports.readOptionalString = readOptionalString;
16
20
  exports.readOptionalNumber = readOptionalNumber;
@@ -131,6 +135,44 @@ function normalizeTopK(topK) {
131
135
  }
132
136
  return Math.floor(topK);
133
137
  }
138
+ /**
139
+ * 批量 upsert 的单批行数上限。500 行是往返次数与单语句参数量之间的平衡:
140
+ * 最宽的表(chunks,17 列)一批也只有 8500 个参数,远低于 Postgres 65535 的协议上限。
141
+ */
142
+ exports.UPSERT_BATCH_SIZE = 500;
143
+ function chunkBatches(items, batchSize = exports.UPSERT_BATCH_SIZE) {
144
+ const batches = [];
145
+ for (let start = 0; start < items.length; start += batchSize) {
146
+ batches.push(items.slice(start, start + batchSize));
147
+ }
148
+ return batches;
149
+ }
150
+ /**
151
+ * 同一条 multi-row INSERT 里出现重复冲突键时,Postgres 会报
152
+ * "ON CONFLICT DO UPDATE command cannot affect row a second time"。
153
+ * 逐行 INSERT 的旧语义是"后写覆盖先写",这里按 key 去重保留最后一条以保持语义一致。
154
+ */
155
+ function dedupeByKey(items, keyOf) {
156
+ const byKey = new Map();
157
+ for (const item of items) {
158
+ byKey.set(keyOf(item), item);
159
+ }
160
+ return [...byKey.values()];
161
+ }
162
+ /**
163
+ * 生成 multi-row VALUES 的占位符片段,例如:
164
+ * ($1, $2::jsonb, now()),\n($3, $4::jsonb, now())。
165
+ * casts 按列声明类型转换后缀('' 表示无转换),rowSuffix 用于追加 now() 这类字面量列。
166
+ */
167
+ function createMultiRowValuesSql(input) {
168
+ const width = input.casts.length;
169
+ const rows = [];
170
+ for (let rowIndex = 0; rowIndex < input.rowCount; rowIndex += 1) {
171
+ const placeholders = input.casts.map((cast, columnIndex) => `$${rowIndex * width + columnIndex + 1}${cast}`);
172
+ rows.push(`(${placeholders.join(', ')}${input.rowSuffix ?? ''})`);
173
+ }
174
+ return rows.join(',\n');
175
+ }
134
176
  function rowToChunkMetadata(row) {
135
177
  const metadata = {
136
178
  source: String(row.source ?? 'inline'),
@@ -1 +1 @@
1
- {"version":3,"file":"sql.js","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/sql.ts"],"names":[],"mappings":";;AAaA,8DASC;AAED,0CAIC;AAED,gCAEC;AAED,gDAEC;AAED,wCAIC;AAED,sCAIC;AAED,4BAMC;AAED,0CAEC;AAOD,sDA0CC;AA4CD,0CAQC;AAED,sCASC;AAED,gDAiBC;AAED,gDAEC;AAED,gDAWC;AAED,0DAMC;AA9MD,MAAM,oBAAoB,GAAG,KAAK,CAAC;AACnC,MAAM,kBAAkB,GAAG,0BAA0B,CAAC;AAEtD,SAAgB,yBAAyB,CAAC,WAAW,GAAG,oBAAoB;IACxE,oBAAoB,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;IAEjD,OAAO;QACH,SAAS,EAAE,eAAe,CAAC,GAAG,WAAW,YAAY,CAAC;QACtD,MAAM,EAAE,eAAe,CAAC,GAAG,WAAW,SAAS,CAAC;QAChD,UAAU,EAAE,eAAe,CAAC,GAAG,WAAW,aAAa,CAAC;QACxD,cAAc,EAAE,eAAe,CAAC,GAAG,WAAW,kBAAkB,CAAC;KACpE,CAAC;AACN,CAAC;AAED,SAAgB,eAAe,CAAC,UAAkB;IAC9C,oBAAoB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;IAE/C,OAAO,IAAI,UAAU,GAAG,CAAC;AAC7B,CAAC;AAED,SAAgB,UAAU,CAAC,KAAc;IACrC,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC,CAAC;AACzC,CAAC;AAED,SAAgB,kBAAkB,CAAC,KAAc;IAC7C,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AAC9D,CAAC;AAED,SAAgB,cAAc,CAAC,KAAc;IACzC,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAE/B,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AACjD,CAAC;AAED,SAAgB,aAAa,CAAc,KAAc;IACrD,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAE/B,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAa,CAAC,CAAC,CAAC,SAAS,CAAC;AAC7D,CAAC;AAED,SAAgB,QAAQ,CAAC,KAAc;IACnC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC5B,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAY,CAAC;IACxC,CAAC;IAED,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAgB,eAAe,CAAC,MAAyB;IACrD,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;AACjE,CAAC;AAOD,SAAgB,qBAAqB,CACjC,OAAsB,EACtB,UAAU,GAAG,CAAC,EACd,KAAK,GAAG,GAAG;IAEX,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,MAAM,GAAc,EAAE,CAAC;IAC7B,MAAM,MAAM,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,GAAG,KAAK,IAAI,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;IACrE,MAAM,eAAe,GAAG,CAAC,KAAc,EAAU,EAAE;QAC/C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEnB,OAAO,IAAI,UAAU,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;IAChD,CAAC,CAAC;IAEF,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;QACjC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAClF,CAAC;IACD,IAAI,OAAO,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;QACxC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,mBAAmB,CAAC,MAAM,eAAe,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;IACjG,CAAC;IACD,IAAI,OAAO,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;QACpC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,aAAa,CAAC,UAAU,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;IACpG,CAAC;IACD,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QAChC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC3F,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,KAAK,SAAS,EAAE,CAAC;QAC5B,OAAO,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,2BAA2B,MAAM,CAAC,KAAK,CAAC,YAAY,MAAM,CAAC,KAAK,CAAC,OAAO,eAAe,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACnJ,CAAC;IACD,KAAK,MAAM,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC,EAAE,CAAC;QACzE,OAAO,CAAC,IAAI,CAAC,yBAAyB,CAAC;YACnC,GAAG;YACH,QAAQ;YACR,aAAa,EAAE,KAAK;YACpB,eAAe;SAClB,CAAC,CAAC,CAAC;IACR,CAAC;IAED,OAAO;QACH,MAAM,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE;QAClE,MAAM;KACT,CAAC;AACN,CAAC;AAED,SAAS,yBAAyB,CAAC,KAKlC;IACG,MAAM,UAAU,GAAG,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC/C,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC3B,MAAM,MAAM,GAAG,GAAG,KAAK,CAAC,aAAa,IAAI,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC;QACvE,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChC,OAAO,GAAG,MAAM,UAAU,KAAK,CAAC,eAAe,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC;QAC/E,CAAC;QAED,OAAO,GAAG,MAAM,MAAM,KAAK,CAAC,eAAe,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;IAClE,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAC1C,GAAG,KAAK,CAAC,aAAa,IAAI,eAAe,CAAC,OAAO,CAAC,OAAO,KAAK,CAAC,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,SAAS,CAC9H,CAAC,CAAC;QAEH,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;IACvC,CAAC;IAED,OAAO,GAAG,KAAK,CAAC,aAAa,IAAI,eAAe,CAAC,OAAO,CAAC,OAAO,KAAK,CAAC,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,SAAS,CAAC;AAChJ,CAAC;AAED,SAAS,gBAAgB,CAAC,GAAW;IACjC,MAAM,OAAO,GAA2B;QACpC,MAAM,EAAE,QAAQ;QAChB,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,WAAW;QACrB,eAAe,EAAE,mBAAmB;QACpC,IAAI,EAAE,MAAM;QACZ,WAAW,EAAE,cAAc;QAC3B,SAAS,EAAE,YAAY;QACvB,UAAU,EAAE,aAAa;KAC5B,CAAC;IAEF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC;AACxB,CAAC;AAED,SAAgB,eAAe,CAC3B,MAAS,EACT,GAAM,EACN,KAAuB;IAEvB,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACtB,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IACxB,CAAC;AACL,CAAC;AAED,SAAgB,aAAa,CAAC,IAAwB;IAClD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACrB,OAAO,EAAE,CAAC;IACd,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC;QACtC,OAAO,CAAC,CAAC;IACb,CAAC;IAED,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;AAC5B,CAAC;AAED,SAAgB,kBAAkB,CAAC,GAA4B;IAC3D,MAAM,QAAQ,GAAkB;QAC5B,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,IAAI,QAAQ,CAAC;QACtC,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC;KAC3C,CAAC;IAEF,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IAClE,eAAe,CAAC,QAAQ,EAAE,UAAU,EAAE,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IACzE,eAAe,CAAC,QAAQ,EAAE,iBAAiB,EAAE,kBAAkB,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC;IACxF,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IAChE,eAAe,CAAC,QAAQ,EAAE,aAAa,EAAE,kBAAkB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC;IAC/E,eAAe,CAAC,QAAQ,EAAE,WAAW,EAAE,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;IAC3E,eAAe,CAAC,QAAQ,EAAE,aAAa,EAAE,uBAAuB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC;IACpF,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,uBAAuB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IACnE,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IAE9D,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAgB,kBAAkB,CAAC,KAAc;IAC7C,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AAC7E,CAAC;AAED,SAAgB,kBAAkB,CAAC,KAAc;IAC7C,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC5B,OAAO,KAAK,CAAC;IACjB,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChD,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAE7B,OAAO,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;IACxD,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAgB,uBAAuB,CAAC,KAAc;IAClD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC;AAC5E,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAa,EAAE,KAAa;IACtD,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,sBAAsB,KAAK,KAAK,KAAK,EAAE,CAAC,CAAC;IAC7D,CAAC;AACL,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC5B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAChF,CAAC"}
1
+ {"version":3,"file":"sql.js","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/sql.ts"],"names":[],"mappings":";;;AAaA,8DASC;AAED,0CAIC;AAED,gCAEC;AAED,gDAEC;AAED,wCAIC;AAED,sCAIC;AAED,4BAMC;AAED,0CAEC;AAOD,sDA0CC;AA4CD,0CAQC;AAED,sCASC;AAQD,oCAOC;AAOD,kCAOC;AAOD,0DAeC;AAED,gDAiBC;AAED,gDAEC;AAED,gDAWC;AAED,0DAMC;AAjQD,MAAM,oBAAoB,GAAG,KAAK,CAAC;AACnC,MAAM,kBAAkB,GAAG,0BAA0B,CAAC;AAEtD,SAAgB,yBAAyB,CAAC,WAAW,GAAG,oBAAoB;IACxE,oBAAoB,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;IAEjD,OAAO;QACH,SAAS,EAAE,eAAe,CAAC,GAAG,WAAW,YAAY,CAAC;QACtD,MAAM,EAAE,eAAe,CAAC,GAAG,WAAW,SAAS,CAAC;QAChD,UAAU,EAAE,eAAe,CAAC,GAAG,WAAW,aAAa,CAAC;QACxD,cAAc,EAAE,eAAe,CAAC,GAAG,WAAW,kBAAkB,CAAC;KACpE,CAAC;AACN,CAAC;AAED,SAAgB,eAAe,CAAC,UAAkB;IAC9C,oBAAoB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;IAE/C,OAAO,IAAI,UAAU,GAAG,CAAC;AAC7B,CAAC;AAED,SAAgB,UAAU,CAAC,KAAc;IACrC,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC,CAAC;AACzC,CAAC;AAED,SAAgB,kBAAkB,CAAC,KAAc;IAC7C,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AAC9D,CAAC;AAED,SAAgB,cAAc,CAAC,KAAc;IACzC,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAE/B,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AACjD,CAAC;AAED,SAAgB,aAAa,CAAc,KAAc;IACrD,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAE/B,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAa,CAAC,CAAC,CAAC,SAAS,CAAC;AAC7D,CAAC;AAED,SAAgB,QAAQ,CAAC,KAAc;IACnC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC5B,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAY,CAAC;IACxC,CAAC;IAED,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAgB,eAAe,CAAC,MAAyB;IACrD,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;AACjE,CAAC;AAOD,SAAgB,qBAAqB,CACjC,OAAsB,EACtB,UAAU,GAAG,CAAC,EACd,KAAK,GAAG,GAAG;IAEX,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,MAAM,GAAc,EAAE,CAAC;IAC7B,MAAM,MAAM,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,GAAG,KAAK,IAAI,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;IACrE,MAAM,eAAe,GAAG,CAAC,KAAc,EAAU,EAAE;QAC/C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEnB,OAAO,IAAI,UAAU,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;IAChD,CAAC,CAAC;IAEF,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;QACjC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,WAAW,CAAC,MAAM,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAClF,CAAC;IACD,IAAI,OAAO,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;QACxC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,mBAAmB,CAAC,MAAM,eAAe,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;IACjG,CAAC;IACD,IAAI,OAAO,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;QACpC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,aAAa,CAAC,UAAU,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;IACpG,CAAC;IACD,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QAChC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC3F,CAAC;IACD,IAAI,OAAO,CAAC,GAAG,KAAK,SAAS,EAAE,CAAC;QAC5B,OAAO,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,2BAA2B,MAAM,CAAC,KAAK,CAAC,YAAY,MAAM,CAAC,KAAK,CAAC,OAAO,eAAe,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACnJ,CAAC;IACD,KAAK,MAAM,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC,EAAE,CAAC;QACzE,OAAO,CAAC,IAAI,CAAC,yBAAyB,CAAC;YACnC,GAAG;YACH,QAAQ;YACR,aAAa,EAAE,KAAK;YACpB,eAAe;SAClB,CAAC,CAAC,CAAC;IACR,CAAC;IAED,OAAO;QACH,MAAM,EAAE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE;QAClE,MAAM;KACT,CAAC;AACN,CAAC;AAED,SAAS,yBAAyB,CAAC,KAKlC;IACG,MAAM,UAAU,GAAG,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC/C,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;QAC3B,MAAM,MAAM,GAAG,GAAG,KAAK,CAAC,aAAa,IAAI,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC;QACvE,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChC,OAAO,GAAG,MAAM,UAAU,KAAK,CAAC,eAAe,CAAC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC;QAC/E,CAAC;QAED,OAAO,GAAG,MAAM,MAAM,KAAK,CAAC,eAAe,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;IAClE,CAAC;IAED,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,EAAE,CAAC;QAChC,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAC1C,GAAG,KAAK,CAAC,aAAa,IAAI,eAAe,CAAC,OAAO,CAAC,OAAO,KAAK,CAAC,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,SAAS,CAC9H,CAAC,CAAC;QAEH,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;IACvC,CAAC;IAED,OAAO,GAAG,KAAK,CAAC,aAAa,IAAI,eAAe,CAAC,OAAO,CAAC,OAAO,KAAK,CAAC,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,SAAS,CAAC;AAChJ,CAAC;AAED,SAAS,gBAAgB,CAAC,GAAW;IACjC,MAAM,OAAO,GAA2B;QACpC,MAAM,EAAE,QAAQ;QAChB,KAAK,EAAE,OAAO;QACd,QAAQ,EAAE,WAAW;QACrB,eAAe,EAAE,mBAAmB;QACpC,IAAI,EAAE,MAAM;QACZ,WAAW,EAAE,cAAc;QAC3B,SAAS,EAAE,YAAY;QACvB,UAAU,EAAE,aAAa;KAC5B,CAAC;IAEF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC;AACxB,CAAC;AAED,SAAgB,eAAe,CAC3B,MAAS,EACT,GAAM,EACN,KAAuB;IAEvB,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QACtB,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IACxB,CAAC;AACL,CAAC;AAED,SAAgB,aAAa,CAAC,IAAwB;IAClD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;QACrB,OAAO,EAAE,CAAC;IACd,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,EAAE,CAAC;QACtC,OAAO,CAAC,CAAC;IACb,CAAC;IAED,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;AAC5B,CAAC;AAED;;;GAGG;AACU,QAAA,iBAAiB,GAAG,GAAG,CAAC;AAErC,SAAgB,YAAY,CAAI,KAAmB,EAAE,SAAS,GAAG,yBAAiB;IAC9E,MAAM,OAAO,GAAU,EAAE,CAAC;IAC1B,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,IAAI,SAAS,EAAE,CAAC;QAC3D,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC;IACxD,CAAC;IAED,OAAO,OAAO,CAAC;AACnB,CAAC;AAED;;;;GAIG;AACH,SAAgB,WAAW,CAAI,KAAmB,EAAE,KAA0B;IAC1E,MAAM,KAAK,GAAG,IAAI,GAAG,EAAa,CAAC;IACnC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACvB,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,OAAO,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;AAC/B,CAAC;AAED;;;;GAIG;AACH,SAAgB,uBAAuB,CAAC,KAIvC;IACG,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;IACjC,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,KAAK,IAAI,QAAQ,GAAG,CAAC,EAAE,QAAQ,GAAG,KAAK,CAAC,QAAQ,EAAE,QAAQ,IAAI,CAAC,EAAE,CAAC;QAC9D,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAChC,CAAC,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC,IAAI,QAAQ,GAAG,KAAK,GAAG,WAAW,GAAG,CAAC,GAAG,IAAI,EAAE,CACzE,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,SAAS,IAAI,EAAE,GAAG,CAAC,CAAC;IACtE,CAAC;IAED,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC5B,CAAC;AAED,SAAgB,kBAAkB,CAAC,GAA4B;IAC3D,MAAM,QAAQ,GAAkB;QAC5B,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,IAAI,QAAQ,CAAC;QACtC,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC;KAC3C,CAAC;IAEF,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IAClE,eAAe,CAAC,QAAQ,EAAE,UAAU,EAAE,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC;IACzE,eAAe,CAAC,QAAQ,EAAE,iBAAiB,EAAE,kBAAkB,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC;IACxF,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,kBAAkB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IAChE,eAAe,CAAC,QAAQ,EAAE,aAAa,EAAE,kBAAkB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC;IAC/E,eAAe,CAAC,QAAQ,EAAE,WAAW,EAAE,kBAAkB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC;IAC3E,eAAe,CAAC,QAAQ,EAAE,aAAa,EAAE,uBAAuB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC;IACpF,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,uBAAuB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IACnE,eAAe,CAAC,QAAQ,EAAE,OAAO,EAAE,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;IAE9D,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAgB,kBAAkB,CAAC,KAAc;IAC7C,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AAC7E,CAAC;AAED,SAAgB,kBAAkB,CAAC,KAAc;IAC7C,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC5B,OAAO,KAAK,CAAC;IACjB,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChD,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;QAE7B,OAAO,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;IACxD,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAgB,uBAAuB,CAAC,KAAc;IAClD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAkB,EAAE,CAAC,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC;AAC5E,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAa,EAAE,KAAa;IACtD,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,sBAAsB,KAAK,KAAK,KAAK,EAAE,CAAC,CAAC;IAC7D,CAAC;AACL,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC5B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAChF,CAAC"}
@@ -20,6 +20,11 @@ export interface DocumentStore {
20
20
  upsertChunks(chunks: Chunk[]): Promise<void>;
21
21
  getChunks(ids: string[]): Promise<Chunk[]>;
22
22
  getChunksByDocument(documentId: string): Promise<Chunk[]>;
23
+ /**
24
+ * 按稳定 documentId 读取已入库文档。
25
+ * pipeline 依赖它对比新旧 checksum,决定 replaceExisting 时是跳过还是清理重建。
26
+ */
27
+ getDocument(documentId: string): Promise<RawDocument | undefined>;
23
28
  deleteDocument(documentId: string): Promise<void>;
24
29
  }
25
30
  export interface VectorStore {
@@ -1 +1 @@
1
- {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/rag/stores/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,KAAK,EACL,aAAa,EACb,WAAW,EACd,MAAM,UAAU,CAAC;AAElB,MAAM,WAAW,aAAa;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC5C;AAED,MAAM,WAAW,cAAc;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,aAAa;IAC1B,eAAe,CAAC,SAAS,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzD,YAAY,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7C,SAAS,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;IAC3C,mBAAmB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;IAC1D,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACrD;AAED,MAAM,WAAW,WAAW;IACxB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,MAAM,CAAC,KAAK,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9C,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;IAC7E,gBAAgB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACvD;AAED,MAAM,WAAW,YAAY;IACzB,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACvC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;IAC1E,gBAAgB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACvD"}
1
+ {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/rag/stores/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,KAAK,EACL,aAAa,EACb,WAAW,EACd,MAAM,UAAU,CAAC;AAElB,MAAM,WAAW,aAAa;IAC1B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC5C;AAED,MAAM,WAAW,cAAc;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,aAAa;IAC1B,eAAe,CAAC,SAAS,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzD,YAAY,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7C,SAAS,CAAC,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;IAC3C,mBAAmB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;IAC1D;;;OAGG;IACH,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,SAAS,CAAC,CAAC;IAClE,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACrD;AAED,MAAM,WAAW,WAAW;IACxB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,MAAM,CAAC,KAAK,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9C,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;IAC7E,gBAAgB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACvD;AAED,MAAM,WAAW,YAAY;IACzB,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACvC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;IAC1E,gBAAgB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CACvD"}
@@ -167,6 +167,12 @@ export interface AnswerOptions extends RetrieveOptions {
167
167
  */
168
168
  llm?: ChatModel;
169
169
  maxContextTokens?: number;
170
+ /**
171
+ * 作用于 ContextBuilder 的过滤阈值,工作在"融合后分数"尺度(如 RRF,量级 ~0.016)。
172
+ * 与 RetrieveOptions.minScore(retriever 原始分数尺度,如 cosine / TF-IDF)语义不同,
173
+ * 两者不能混用;默认 undefined 表示不过滤。
174
+ */
175
+ minFusedScore?: number;
170
176
  requireCitations?: boolean;
171
177
  systemPrompt?: string;
172
178
  metadata?: Record<string, unknown>;
@@ -1 +1 @@
1
- {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/rag/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAErD,MAAM,MAAM,SAAS,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,YAAY,CAAC;AACjE,MAAM,MAAM,eAAe,GAAG,OAAO,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,CAAC;AACvE,MAAM,MAAM,aAAa,GAAG,MAAM,GAAG,QAAQ,GAAG,KAAK,CAAC;AAEtD;;;GAGG;AACH,MAAM,WAAW,WAAW;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,MAAM,CAAC,EAAE,aAAa,EAAE,CAAC;CAC5B;AAED,MAAM,WAAW,gBAAgB;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED;;;GAGG;AACH,MAAM,MAAM,aAAa,GACnB;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAChD;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACnC;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAA;CAAE,GACrD;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACjD;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,GAAG,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAErD;;;GAGG;AACH,MAAM,WAAW,KAAK;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,SAAS,CAAC;IAChB,QAAQ,EAAE,aAAa,CAAC;CAC3B;AAED,MAAM,WAAW,aAAa;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,aAAa;IAC1B,KAAK,EAAE,KAAK,CAAC;IACb,SAAS,EAAE,MAAM,EAAE,CAAC;CACvB;AAED;;;GAGG;AACH,MAAM,WAAW,cAAc;IAC3B,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,eAAe,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,QAAQ;IACrB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,cAAc;IAC3B,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,KAAK,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,eAAe,CAAC;IACxB,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,UAAU;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;CAC3B;AAED,MAAM,WAAW,SAAS;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,UAAU,EAAE,aAAa,CAAC;IAC1B,SAAS,EAAE,cAAc,EAAE,CAAC;IAC5B,KAAK,CAAC,EAAE,UAAU,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED;;;GAGG;AACH,MAAM,WAAW,WAAW;IACxB,SAAS,CAAC,EAAE,WAAW,EAAE,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,UAAU,GAAG,WAAW,CAAC;IAChC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACrC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,eAAe,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED,MAAM,WAAW,YAAY;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,eAAe;IAC5B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzC,KAAK,CAAC,EAAE,YAAY,CAAC;CACxB;AAED,MAAM,WAAW,aAAc,SAAQ,eAAe;IAClD;;OAEG;IACH,GAAG,CAAC,EAAE,SAAS,CAAC;IAChB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC"}
1
+ {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/rag/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AACpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAErD,MAAM,MAAM,SAAS,GAAG,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,YAAY,CAAC;AACjE,MAAM,MAAM,eAAe,GAAG,OAAO,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,CAAC;AACvE,MAAM,MAAM,aAAa,GAAG,MAAM,GAAG,QAAQ,GAAG,KAAK,CAAC;AAEtD;;;GAGG;AACH,MAAM,WAAW,WAAW;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,MAAM,CAAC,EAAE,aAAa,EAAE,CAAC;CAC5B;AAED,MAAM,WAAW,gBAAgB;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED;;;GAGG;AACH,MAAM,MAAM,aAAa,GACnB;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAChD;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACnC;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,OAAO,EAAE,CAAA;CAAE,GACrD;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GACjD;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,GAAG,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAErD;;;GAGG;AACH,MAAM,WAAW,KAAK;IAClB,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,SAAS,CAAC;IAChB,QAAQ,EAAE,aAAa,CAAC;CAC3B;AAED,MAAM,WAAW,aAAa;IAC1B,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,aAAa;IAC1B,KAAK,EAAE,KAAK,CAAC;IACb,SAAS,EAAE,MAAM,EAAE,CAAC;CACvB;AAED;;;GAGG;AACH,MAAM,WAAW,cAAc;IAC3B,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,eAAe,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,QAAQ;IACrB;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,cAAc;IAC3B,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,KAAK,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,eAAe,CAAC;IACxB,aAAa,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,UAAU;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,eAAe,EAAE,MAAM,CAAC;CAC3B;AAED,MAAM,WAAW,SAAS;IACtB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,QAAQ,EAAE,CAAC;IACtB,UAAU,EAAE,aAAa,CAAC;IAC1B,SAAS,EAAE,cAAc,EAAE,CAAC;IAC5B,KAAK,CAAC,EAAE,UAAU,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED;;;GAGG;AACH,MAAM,WAAW,WAAW;IACxB,SAAS,CAAC,EAAE,WAAW,EAAE,CAAC;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,UAAU,GAAG,WAAW,CAAC;IAChC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;IACrC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,eAAe,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED,MAAM,WAAW,YAAY;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,EAAE,CAAC;IACtB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC;AAED,MAAM,WAAW,eAAe;IAC5B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzC,KAAK,CAAC,EAAE,YAAY,CAAC;CACxB;AAED,MAAM,WAAW,aAAc,SAAQ,eAAe;IAClD;;OAEG;IACH,GAAG,CAAC,EAAE,SAAS,CAAC;IAChB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B;;;;OAIG;IACH,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACtC"}
@@ -0,0 +1,11 @@
1
+ import { Agent } from '../agent';
2
+ import type { AgentComponents } from '../agent';
3
+ import type { ChatModel } from '../llm';
4
+ import type { ScriptedChatResponse } from './models';
5
+ export interface CreateTestAgentOptions extends Omit<Partial<AgentComponents>, 'name' | 'llm'> {
6
+ name?: string;
7
+ llm?: ChatModel;
8
+ responses?: ScriptedChatResponse[];
9
+ }
10
+ export declare function createTestAgent(options?: CreateTestAgentOptions): Agent;
11
+ //# sourceMappingURL=agent.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../src/testing/agent.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,KAAK,EACR,MAAM,UAAU,CAAC;AAClB,OAAO,KAAK,EACR,eAAe,EAClB,MAAM,UAAU,CAAC;AAClB,OAAO,KAAK,EACR,SAAS,EACZ,MAAM,QAAQ,CAAC;AAIhB,OAAO,KAAK,EACR,oBAAoB,EACvB,MAAM,UAAU,CAAC;AAElB,MAAM,WAAW,sBAAuB,SAAQ,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC;IAC1F,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,CAAC,EAAE,SAAS,CAAC;IAChB,SAAS,CAAC,EAAE,oBAAoB,EAAE,CAAC;CACtC;AAED,wBAAgB,eAAe,CAAC,OAAO,GAAE,sBAA2B,GAAG,KAAK,CAwC3E"}
@@ -0,0 +1,45 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.createTestAgent = createTestAgent;
4
+ const agent_1 = require("../agent");
5
+ const models_1 = require("./models");
6
+ function createTestAgent(options = {}) {
7
+ const components = {
8
+ name: options.name ?? 'test-agent',
9
+ llm: options.llm ?? new models_1.ScriptedChatModel({
10
+ responses: options.responses ?? [
11
+ {
12
+ role: 'assistant',
13
+ content: 'ok',
14
+ toolCalls: [],
15
+ },
16
+ ],
17
+ }),
18
+ };
19
+ if (options.llmConfig !== undefined) {
20
+ components.llmConfig = options.llmConfig;
21
+ }
22
+ if (options.historyStore !== undefined) {
23
+ components.historyStore = options.historyStore;
24
+ }
25
+ if (options.requireSessionId !== undefined) {
26
+ components.requireSessionId = options.requireSessionId;
27
+ }
28
+ if (options.tools !== undefined) {
29
+ components.tools = options.tools;
30
+ }
31
+ if (options.systemPrompt !== undefined) {
32
+ components.systemPrompt = options.systemPrompt;
33
+ }
34
+ if (options.config !== undefined) {
35
+ components.config = options.config;
36
+ }
37
+ if (options.executor !== undefined) {
38
+ components.executor = options.executor;
39
+ }
40
+ if (options.observer !== undefined) {
41
+ components.observer = options.observer;
42
+ }
43
+ return new agent_1.Agent(components);
44
+ }
45
+ //# sourceMappingURL=agent.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/testing/agent.ts"],"names":[],"mappings":";;AAsBA,0CAwCC;AA9DD,oCAEkB;AAOlB,qCAEkB;AAWlB,SAAgB,eAAe,CAAC,UAAkC,EAAE;IAChE,MAAM,UAAU,GAAoB;QAChC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,YAAY;QAClC,GAAG,EAAE,OAAO,CAAC,GAAG,IAAI,IAAI,0BAAiB,CAAC;YACtC,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI;gBAC5B;oBACI,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,IAAI;oBACb,SAAS,EAAE,EAAE;iBAChB;aACJ;SACJ,CAAC;KACL,CAAC;IAEF,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;QAClC,UAAU,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;IAC7C,CAAC;IACD,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;QACrC,UAAU,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IACnD,CAAC;IACD,IAAI,OAAO,CAAC,gBAAgB,KAAK,SAAS,EAAE,CAAC;QACzC,UAAU,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAC3D,CAAC;IACD,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAC9B,UAAU,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;IACrC,CAAC;IACD,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;QACrC,UAAU,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;IACnD,CAAC;IACD,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAC/B,UAAU,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IACvC,CAAC;IACD,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;QACjC,UAAU,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;IAC3C,CAAC;IACD,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;QACjC,UAAU,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;IAC3C,CAAC;IAED,OAAO,IAAI,aAAK,CAAC,UAAU,CAAC,CAAC;AACjC,CAAC"}
@@ -0,0 +1,12 @@
1
+ import type { MessageHistoryStore } from '../history';
2
+ export type MessageHistoryStoreFactory = () => MessageHistoryStore | Promise<MessageHistoryStore>;
3
+ export interface MessageHistoryStoreContractOptions {
4
+ sessionA?: string;
5
+ sessionB?: string;
6
+ /**
7
+ * 并发语义断言使用的独立会话(仅当 store 支持 getVersion 时才会跑)。
8
+ */
9
+ sessionVersioned?: string;
10
+ }
11
+ export declare function assertMessageHistoryStoreContract(createStore: MessageHistoryStoreFactory, options?: MessageHistoryStoreContractOptions): Promise<void>;
12
+ //# sourceMappingURL=history-contract.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"history-contract.d.ts","sourceRoot":"","sources":["../../src/testing/history-contract.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACR,mBAAmB,EACtB,MAAM,YAAY,CAAC;AAKpB,MAAM,MAAM,0BAA0B,GAClC,MAAM,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAAC;AAE7D,MAAM,WAAW,kCAAkC;IAC/C,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB;;OAEG;IACH,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED,wBAAsB,iCAAiC,CACnD,WAAW,EAAE,0BAA0B,EACvC,OAAO,GAAE,kCAAuC,GACjD,OAAO,CAAC,IAAI,CAAC,CAyFf"}
@@ -0,0 +1,111 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.assertMessageHistoryStoreContract = assertMessageHistoryStoreContract;
4
+ const errors_1 = require("../errors");
5
+ const messages_1 = require("../messages");
6
+ async function assertMessageHistoryStoreContract(createStore, options = {}) {
7
+ const sessionA = options.sessionA ?? 'contract-session-a';
8
+ const sessionB = options.sessionB ?? 'contract-session-b';
9
+ const store = await createStore();
10
+ const historyA = store.get(sessionA);
11
+ const historyB = store.get(sessionB);
12
+ await historyA.clear();
13
+ await historyB.clear();
14
+ const first = new messages_1.Message('first', 'user', {
15
+ timestamp: new Date('2026-01-01T00:00:01.000Z'),
16
+ });
17
+ const second = new messages_1.Message('second', 'assistant', {
18
+ timestamp: new Date('2026-01-01T00:00:02.000Z'),
19
+ });
20
+ const third = new messages_1.Message('third', 'user', {
21
+ timestamp: new Date('2026-01-01T00:00:03.000Z'),
22
+ });
23
+ await historyA.add(first);
24
+ await historyA.addMany([second, third]);
25
+ await historyB.add(new messages_1.Message('other session', 'user'));
26
+ assertEqual((await historyA.get()).map((message) => message.content).join('|'), 'first|second|third', 'MessageHistory must preserve add/addMany order.');
27
+ assertEqual((await historyB.get()).map((message) => message.content).join('|'), 'other session', 'MessageHistoryStore must isolate sessions.');
28
+ const snapshot = await historyA.get();
29
+ snapshot.push(new messages_1.Message('external mutation', 'user'));
30
+ assertEqual((await historyA.get()).map((message) => message.content).join('|'), 'first|second|third', 'MessageHistory.get() must return a defensive array copy.');
31
+ assertEqual((await historyA.get({ limit: 2 })).map((message) => message.content).join('|'), 'second|third', 'MessageHistory.get({ limit }) must return the most recent messages.');
32
+ assertEqual((await historyA.get({
33
+ before: '2026-01-01T00:00:03.000Z',
34
+ limit: 1,
35
+ })).map((message) => message.content).join('|'), 'second', 'MessageHistory.get({ before, limit }) must apply the timestamp bound before limiting.');
36
+ await store.clear(sessionA);
37
+ assertEqual((await historyA.get()).length, 0, 'MessageHistoryStore.clear(sessionId) must clear the selected session.');
38
+ if (store.deleteSession) {
39
+ await historyA.add(new messages_1.Message('after clear', 'user'));
40
+ await store.deleteSession(sessionA);
41
+ assertEqual((await store.get(sessionA).get()).length, 0, 'MessageHistoryStore.deleteSession(sessionId) must remove the selected session.');
42
+ }
43
+ if (store.clearAll) {
44
+ await store.get(sessionA).add(new messages_1.Message('a', 'user'));
45
+ await store.get(sessionB).add(new messages_1.Message('b', 'user'));
46
+ await store.clearAll();
47
+ assertEqual((await store.get(sessionA).get()).length + (await store.get(sessionB).get()).length, 0, 'MessageHistoryStore.clearAll() must remove all sessions.');
48
+ }
49
+ await assertVersionedHistorySemantics(store, options.sessionVersioned ?? 'contract-session-versions');
50
+ }
51
+ /**
52
+ * 可选并发语义:仅当实现声明了 getVersion 时校验。
53
+ * 断言全部使用相对版本号(baseVersion + n),这样对共享真实数据库、
54
+ * 多次重跑契约的场景同样成立。
55
+ */
56
+ async function assertVersionedHistorySemantics(store, sessionId) {
57
+ const history = store.get(sessionId);
58
+ if (typeof history.getVersion !== 'function') {
59
+ return;
60
+ }
61
+ await history.clear();
62
+ const baseVersion = await history.getVersion();
63
+ if (!Number.isInteger(baseVersion) || baseVersion < 0) {
64
+ throw new errors_1.HistoryError('MessageHistory.getVersion() must return a non-negative integer.', {
65
+ code: 'history_contract_failed',
66
+ metadata: {
67
+ actual: baseVersion,
68
+ },
69
+ });
70
+ }
71
+ await history.addMany([new messages_1.Message('versioned first', 'user')], { expectedVersion: baseVersion });
72
+ assertEqual(await history.getVersion(), baseVersion + 1, 'A successful addMany must increment the version by exactly 1.');
73
+ await history.add(new messages_1.Message('versioned second', 'assistant'));
74
+ assertEqual(await history.getVersion(), baseVersion + 2, 'A successful add must increment the version by exactly 1.');
75
+ let conflict;
76
+ try {
77
+ await history.addMany([new messages_1.Message('stale write', 'user')], { expectedVersion: baseVersion });
78
+ }
79
+ catch (error) {
80
+ conflict = error;
81
+ }
82
+ if (!(conflict instanceof errors_1.HistoryVersionConflictError)) {
83
+ throw new errors_1.HistoryError('addMany with a stale expectedVersion must throw HistoryVersionConflictError.', {
84
+ code: 'history_contract_failed',
85
+ metadata: {
86
+ actual: conflict === undefined ? 'no error thrown' : String(conflict),
87
+ },
88
+ });
89
+ }
90
+ assertEqual(conflict.code, 'history_version_conflict', 'HistoryVersionConflictError must use the history_version_conflict code.');
91
+ assertEqual(conflict.expectedVersion, baseVersion, 'HistoryVersionConflictError.expectedVersion must echo the stale version.');
92
+ assertEqual(conflict.actualVersion, baseVersion + 2, 'HistoryVersionConflictError.actualVersion must report the current version.');
93
+ assertEqual(conflict.metadata?.expectedVersion, baseVersion, 'HistoryVersionConflictError metadata must include expectedVersion.');
94
+ assertEqual(conflict.metadata?.actualVersion, baseVersion + 2, 'HistoryVersionConflictError metadata must include actualVersion.');
95
+ assertEqual((await history.get()).map((message) => message.content).join('|'), 'versioned first|versioned second', 'A conflicting addMany must not write any messages.');
96
+ assertEqual(await history.getVersion(), baseVersion + 2, 'A conflicting addMany must not change the version.');
97
+ await history.addMany([new messages_1.Message('versioned third', 'user')], { expectedVersion: baseVersion + 2 });
98
+ assertEqual(await history.getVersion(), baseVersion + 3, 'addMany with a matching expectedVersion must succeed.');
99
+ }
100
+ function assertEqual(actual, expected, message) {
101
+ if (actual !== expected) {
102
+ throw new errors_1.HistoryError(message, {
103
+ code: 'history_contract_failed',
104
+ metadata: {
105
+ actual,
106
+ expected,
107
+ },
108
+ });
109
+ }
110
+ }
111
+ //# sourceMappingURL=history-contract.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"history-contract.js","sourceRoot":"","sources":["../../src/testing/history-contract.ts"],"names":[],"mappings":";;AAoBA,8EA4FC;AAhHD,sCAAsE;AAItE,0CAEqB;AAcd,KAAK,UAAU,iCAAiC,CACnD,WAAuC,EACvC,UAA8C,EAAE;IAEhD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,oBAAoB,CAAC;IAC1D,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,oBAAoB,CAAC;IAC1D,MAAM,KAAK,GAAG,MAAM,WAAW,EAAE,CAAC;IAClC,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACrC,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAErC,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAC;IACvB,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAC;IAEvB,MAAM,KAAK,GAAG,IAAI,kBAAO,CAAC,OAAO,EAAE,MAAM,EAAE;QACvC,SAAS,EAAE,IAAI,IAAI,CAAC,0BAA0B,CAAC;KAClD,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,IAAI,kBAAO,CAAC,QAAQ,EAAE,WAAW,EAAE;QAC9C,SAAS,EAAE,IAAI,IAAI,CAAC,0BAA0B,CAAC;KAClD,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,kBAAO,CAAC,OAAO,EAAE,MAAM,EAAE;QACvC,SAAS,EAAE,IAAI,IAAI,CAAC,0BAA0B,CAAC;KAClD,CAAC,CAAC;IAEH,MAAM,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC1B,MAAM,QAAQ,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;IACxC,MAAM,QAAQ,CAAC,GAAG,CAAC,IAAI,kBAAO,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC,CAAC;IAEzD,WAAW,CACP,CAAC,MAAM,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAClE,oBAAoB,EACpB,iDAAiD,CACpD,CAAC;IACF,WAAW,CACP,CAAC,MAAM,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAClE,eAAe,EACf,4CAA4C,CAC/C,CAAC;IAEF,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,GAAG,EAAE,CAAC;IACtC,QAAQ,CAAC,IAAI,CAAC,IAAI,kBAAO,CAAC,mBAAmB,EAAE,MAAM,CAAC,CAAC,CAAC;IACxD,WAAW,CACP,CAAC,MAAM,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAClE,oBAAoB,EACpB,0DAA0D,CAC7D,CAAC;IAEF,WAAW,CACP,CAAC,MAAM,QAAQ,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAC9E,cAAc,EACd,qEAAqE,CACxE,CAAC;IACF,WAAW,CACP,CAAC,MAAM,QAAQ,CAAC,GAAG,CAAC;QAChB,MAAM,EAAE,0BAA0B;QAClC,KAAK,EAAE,CAAC;KACX,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAC/C,QAAQ,EACR,uFAAuF,CAC1F,CAAC;IAEF,MAAM,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAC5B,WAAW,CACP,CAAC,MAAM,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,EAC7B,CAAC,EACD,uEAAuE,CAC1E,CAAC;IAEF,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;QACtB,MAAM,QAAQ,CAAC,GAAG,CAAC,IAAI,kBAAO,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC,CAAC;QACvD,MAAM,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QACpC,WAAW,CACP,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,EACxC,CAAC,EACD,gFAAgF,CACnF,CAAC;IACN,CAAC;IAED,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QACjB,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,IAAI,kBAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC;QACxD,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,IAAI,kBAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC;QACxD,MAAM,KAAK,CAAC,QAAQ,EAAE,CAAC;QACvB,WAAW,CACP,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,EACnF,CAAC,EACD,0DAA0D,CAC7D,CAAC;IACN,CAAC;IAED,MAAM,+BAA+B,CACjC,KAAK,EACL,OAAO,CAAC,gBAAgB,IAAI,2BAA2B,CAC1D,CAAC;AACN,CAAC;AAED;;;;GAIG;AACH,KAAK,UAAU,+BAA+B,CAC1C,KAA0B,EAC1B,SAAiB;IAEjB,MAAM,OAAO,GAAG,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAErC,IAAI,OAAO,OAAO,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;QAC3C,OAAO;IACX,CAAC;IAED,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;IAEtB,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;IAC/C,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;QACpD,MAAM,IAAI,qBAAY,CAAC,iEAAiE,EAAE;YACtF,IAAI,EAAE,yBAAyB;YAC/B,QAAQ,EAAE;gBACN,MAAM,EAAE,WAAW;aACtB;SACJ,CAAC,CAAC;IACP,CAAC;IAED,MAAM,OAAO,CAAC,OAAO,CACjB,CAAC,IAAI,kBAAO,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC,EACxC,EAAE,eAAe,EAAE,WAAW,EAAE,CACnC,CAAC;IACF,WAAW,CACP,MAAM,OAAO,CAAC,UAAU,EAAE,EAC1B,WAAW,GAAG,CAAC,EACf,+DAA+D,CAClE,CAAC;IAEF,MAAM,OAAO,CAAC,GAAG,CAAC,IAAI,kBAAO,CAAC,kBAAkB,EAAE,WAAW,CAAC,CAAC,CAAC;IAChE,WAAW,CACP,MAAM,OAAO,CAAC,UAAU,EAAE,EAC1B,WAAW,GAAG,CAAC,EACf,2DAA2D,CAC9D,CAAC;IAEF,IAAI,QAAiB,CAAC;IACtB,IAAI,CAAC;QACD,MAAM,OAAO,CAAC,OAAO,CACjB,CAAC,IAAI,kBAAO,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC,EACpC,EAAE,eAAe,EAAE,WAAW,EAAE,CACnC,CAAC;IACN,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,QAAQ,GAAG,KAAK,CAAC;IACrB,CAAC;IAED,IAAI,CAAC,CAAC,QAAQ,YAAY,oCAA2B,CAAC,EAAE,CAAC;QACrD,MAAM,IAAI,qBAAY,CAClB,8EAA8E,EAC9E;YACI,IAAI,EAAE,yBAAyB;YAC/B,QAAQ,EAAE;gBACN,MAAM,EAAE,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC;aACxE;SACJ,CACJ,CAAC;IACN,CAAC;IAED,WAAW,CACP,QAAQ,CAAC,IAAI,EACb,0BAA0B,EAC1B,yEAAyE,CAC5E,CAAC;IACF,WAAW,CACP,QAAQ,CAAC,eAAe,EACxB,WAAW,EACX,0EAA0E,CAC7E,CAAC;IACF,WAAW,CACP,QAAQ,CAAC,aAAa,EACtB,WAAW,GAAG,CAAC,EACf,4EAA4E,CAC/E,CAAC;IACF,WAAW,CACP,QAAQ,CAAC,QAAQ,EAAE,eAAe,EAClC,WAAW,EACX,oEAAoE,CACvE,CAAC;IACF,WAAW,CACP,QAAQ,CAAC,QAAQ,EAAE,aAAa,EAChC,WAAW,GAAG,CAAC,EACf,kEAAkE,CACrE,CAAC;IAEF,WAAW,CACP,CAAC,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EACjE,kCAAkC,EAClC,oDAAoD,CACvD,CAAC;IACF,WAAW,CACP,MAAM,OAAO,CAAC,UAAU,EAAE,EAC1B,WAAW,GAAG,CAAC,EACf,oDAAoD,CACvD,CAAC;IAEF,MAAM,OAAO,CAAC,OAAO,CACjB,CAAC,IAAI,kBAAO,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC,EACxC,EAAE,eAAe,EAAE,WAAW,GAAG,CAAC,EAAE,CACvC,CAAC;IACF,WAAW,CACP,MAAM,OAAO,CAAC,UAAU,EAAE,EAC1B,WAAW,GAAG,CAAC,EACf,uDAAuD,CAC1D,CAAC;AACN,CAAC;AAED,SAAS,WAAW,CAAC,MAAe,EAAE,QAAiB,EAAE,OAAe;IACpE,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;QACtB,MAAM,IAAI,qBAAY,CAAC,OAAO,EAAE;YAC5B,IAAI,EAAE,yBAAyB;YAC/B,QAAQ,EAAE;gBACN,MAAM;gBACN,QAAQ;aACX;SACJ,CAAC,CAAC;IACP,CAAC;AACL,CAAC"}
@@ -0,0 +1,11 @@
1
+ export { FakeChatModel, ScriptedChatModel, } from './models';
2
+ export type { FakeChatModelOptions, ScriptedChatModelOptions, ScriptedChatResponse, } from './models';
3
+ export { FakeTool, } from './tools';
4
+ export type { FakeToolCall, FakeToolOptions, } from './tools';
5
+ export { createTestAgent, } from './agent';
6
+ export type { CreateTestAgentOptions, } from './agent';
7
+ export { assertMessageHistoryStoreContract, } from './history-contract';
8
+ export type { MessageHistoryStoreContractOptions, MessageHistoryStoreFactory, } from './history-contract';
9
+ export { assertToolSchemaSnapshot, createToolSchemaSnapshot, } from './tool-schema';
10
+ export type { ToolSchemaSnapshot, } from './tool-schema';
11
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/testing/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,aAAa,EACb,iBAAiB,GACpB,MAAM,UAAU,CAAC;AAClB,YAAY,EACR,oBAAoB,EACpB,wBAAwB,EACxB,oBAAoB,GACvB,MAAM,UAAU,CAAC;AAClB,OAAO,EACH,QAAQ,GACX,MAAM,SAAS,CAAC;AACjB,YAAY,EACR,YAAY,EACZ,eAAe,GAClB,MAAM,SAAS,CAAC;AACjB,OAAO,EACH,eAAe,GAClB,MAAM,SAAS,CAAC;AACjB,YAAY,EACR,sBAAsB,GACzB,MAAM,SAAS,CAAC;AACjB,OAAO,EACH,iCAAiC,GACpC,MAAM,oBAAoB,CAAC;AAC5B,YAAY,EACR,kCAAkC,EAClC,0BAA0B,GAC7B,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EACH,wBAAwB,EACxB,wBAAwB,GAC3B,MAAM,eAAe,CAAC;AACvB,YAAY,EACR,kBAAkB,GACrB,MAAM,eAAe,CAAC"}