@fllf/agent-sdk 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (326) hide show
  1. package/README.md +240 -0
  2. package/dist/agent/Agent.d.ts +34 -0
  3. package/dist/agent/Agent.d.ts.map +1 -0
  4. package/dist/agent/Agent.js +168 -0
  5. package/dist/agent/Agent.js.map +1 -0
  6. package/dist/agent/createAgent.d.ts +4 -0
  7. package/dist/agent/createAgent.d.ts.map +1 -0
  8. package/dist/agent/createAgent.js +8 -0
  9. package/dist/agent/createAgent.js.map +1 -0
  10. package/dist/agent/index.d.ts +4 -0
  11. package/dist/agent/index.d.ts.map +1 -0
  12. package/dist/agent/index.js +8 -0
  13. package/dist/agent/index.js.map +1 -0
  14. package/dist/agent/types.d.ts +23 -0
  15. package/dist/agent/types.d.ts.map +1 -0
  16. package/dist/agent/types.js +3 -0
  17. package/dist/agent/types.js.map +1 -0
  18. package/dist/config/config.d.ts +35 -0
  19. package/dist/config/config.d.ts.map +1 -0
  20. package/dist/config/config.js +123 -0
  21. package/dist/config/config.js.map +1 -0
  22. package/dist/config/index.d.ts +3 -0
  23. package/dist/config/index.d.ts.map +1 -0
  24. package/dist/config/index.js +6 -0
  25. package/dist/config/index.js.map +1 -0
  26. package/dist/executors/base.d.ts +44 -0
  27. package/dist/executors/base.d.ts.map +1 -0
  28. package/dist/executors/base.js +3 -0
  29. package/dist/executors/base.js.map +1 -0
  30. package/dist/executors/index.d.ts +7 -0
  31. package/dist/executors/index.d.ts.map +1 -0
  32. package/dist/executors/index.js +10 -0
  33. package/dist/executors/index.js.map +1 -0
  34. package/dist/executors/rag-executor.d.ts +16 -0
  35. package/dist/executors/rag-executor.d.ts.map +1 -0
  36. package/dist/executors/rag-executor.js +120 -0
  37. package/dist/executors/rag-executor.js.map +1 -0
  38. package/dist/executors/simple-chat-executor.d.ts +5 -0
  39. package/dist/executors/simple-chat-executor.d.ts.map +1 -0
  40. package/dist/executors/simple-chat-executor.js +77 -0
  41. package/dist/executors/simple-chat-executor.js.map +1 -0
  42. package/dist/executors/tool-calling-executor.d.ts +10 -0
  43. package/dist/executors/tool-calling-executor.d.ts.map +1 -0
  44. package/dist/executors/tool-calling-executor.js +151 -0
  45. package/dist/executors/tool-calling-executor.js.map +1 -0
  46. package/dist/history/base.d.ts +16 -0
  47. package/dist/history/base.d.ts.map +1 -0
  48. package/dist/history/base.js +10 -0
  49. package/dist/history/base.js.map +1 -0
  50. package/dist/history/in-memory.d.ts +26 -0
  51. package/dist/history/in-memory.d.ts.map +1 -0
  52. package/dist/history/in-memory.js +88 -0
  53. package/dist/history/in-memory.js.map +1 -0
  54. package/dist/history/index.d.ts +5 -0
  55. package/dist/history/index.d.ts.map +1 -0
  56. package/dist/history/index.js +10 -0
  57. package/dist/history/index.js.map +1 -0
  58. package/dist/index.d.ts +10 -0
  59. package/dist/index.d.ts.map +1 -0
  60. package/dist/index.js +26 -0
  61. package/dist/index.js.map +1 -0
  62. package/dist/llm/LLM.d.ts +23 -0
  63. package/dist/llm/LLM.d.ts.map +1 -0
  64. package/dist/llm/LLM.js +404 -0
  65. package/dist/llm/LLM.js.map +1 -0
  66. package/dist/llm/errors.d.ts +43 -0
  67. package/dist/llm/errors.d.ts.map +1 -0
  68. package/dist/llm/errors.js +128 -0
  69. package/dist/llm/errors.js.map +1 -0
  70. package/dist/llm/factory.d.ts +9 -0
  71. package/dist/llm/factory.d.ts.map +1 -0
  72. package/dist/llm/factory.js +28 -0
  73. package/dist/llm/factory.js.map +1 -0
  74. package/dist/llm/index.d.ts +6 -0
  75. package/dist/llm/index.d.ts.map +1 -0
  76. package/dist/llm/index.js +18 -0
  77. package/dist/llm/index.js.map +1 -0
  78. package/dist/llm/providers/base.d.ts +3 -0
  79. package/dist/llm/providers/base.d.ts.map +1 -0
  80. package/dist/llm/providers/base.js +3 -0
  81. package/dist/llm/providers/base.js.map +1 -0
  82. package/dist/llm/providers/local.d.ts +6 -0
  83. package/dist/llm/providers/local.d.ts.map +1 -0
  84. package/dist/llm/providers/local.js +19 -0
  85. package/dist/llm/providers/local.js.map +1 -0
  86. package/dist/llm/providers/openai-compatible.d.ts +13 -0
  87. package/dist/llm/providers/openai-compatible.d.ts.map +1 -0
  88. package/dist/llm/providers/openai-compatible.js +176 -0
  89. package/dist/llm/providers/openai-compatible.js.map +1 -0
  90. package/dist/llm/providers/openai.d.ts +6 -0
  91. package/dist/llm/providers/openai.d.ts.map +1 -0
  92. package/dist/llm/providers/openai.js +23 -0
  93. package/dist/llm/providers/openai.js.map +1 -0
  94. package/dist/llm/types.d.ts +116 -0
  95. package/dist/llm/types.d.ts.map +1 -0
  96. package/dist/llm/types.js +3 -0
  97. package/dist/llm/types.js.map +1 -0
  98. package/dist/messages/index.d.ts +3 -0
  99. package/dist/messages/index.d.ts.map +1 -0
  100. package/dist/messages/index.js +6 -0
  101. package/dist/messages/index.js.map +1 -0
  102. package/dist/messages/message.d.ts +25 -0
  103. package/dist/messages/message.d.ts.map +1 -0
  104. package/dist/messages/message.js +110 -0
  105. package/dist/messages/message.js.map +1 -0
  106. package/dist/messages/types.d.ts +20 -0
  107. package/dist/messages/types.d.ts.map +1 -0
  108. package/dist/messages/types.js +3 -0
  109. package/dist/messages/types.js.map +1 -0
  110. package/dist/observability/console-observer.d.ts +10 -0
  111. package/dist/observability/console-observer.d.ts.map +1 -0
  112. package/dist/observability/console-observer.js +28 -0
  113. package/dist/observability/console-observer.js.map +1 -0
  114. package/dist/observability/index.d.ts +5 -0
  115. package/dist/observability/index.d.ts.map +1 -0
  116. package/dist/observability/index.js +11 -0
  117. package/dist/observability/index.js.map +1 -0
  118. package/dist/observability/observer.d.ts +8 -0
  119. package/dist/observability/observer.d.ts.map +1 -0
  120. package/dist/observability/observer.js +46 -0
  121. package/dist/observability/observer.js.map +1 -0
  122. package/dist/observability/types.d.ts +34 -0
  123. package/dist/observability/types.d.ts.map +1 -0
  124. package/dist/observability/types.js +3 -0
  125. package/dist/observability/types.js.map +1 -0
  126. package/dist/rag/chunking/auto-chunker.d.ts +25 -0
  127. package/dist/rag/chunking/auto-chunker.d.ts.map +1 -0
  128. package/dist/rag/chunking/auto-chunker.js +101 -0
  129. package/dist/rag/chunking/auto-chunker.js.map +1 -0
  130. package/dist/rag/chunking/chunker.d.ts +44 -0
  131. package/dist/rag/chunking/chunker.d.ts.map +1 -0
  132. package/dist/rag/chunking/chunker.js +162 -0
  133. package/dist/rag/chunking/chunker.js.map +1 -0
  134. package/dist/rag/chunking/index.d.ts +6 -0
  135. package/dist/rag/chunking/index.d.ts.map +1 -0
  136. package/dist/rag/chunking/index.js +22 -0
  137. package/dist/rag/chunking/index.js.map +1 -0
  138. package/dist/rag/chunking/markdown-chunker.d.ts +13 -0
  139. package/dist/rag/chunking/markdown-chunker.d.ts.map +1 -0
  140. package/dist/rag/chunking/markdown-chunker.js +144 -0
  141. package/dist/rag/chunking/markdown-chunker.js.map +1 -0
  142. package/dist/rag/chunking/parent-child.d.ts +16 -0
  143. package/dist/rag/chunking/parent-child.d.ts.map +1 -0
  144. package/dist/rag/chunking/parent-child.js +76 -0
  145. package/dist/rag/chunking/parent-child.js.map +1 -0
  146. package/dist/rag/chunking/recursive-chunker.d.ts +11 -0
  147. package/dist/rag/chunking/recursive-chunker.d.ts.map +1 -0
  148. package/dist/rag/chunking/recursive-chunker.js +27 -0
  149. package/dist/rag/chunking/recursive-chunker.js.map +1 -0
  150. package/dist/rag/embeddings/embedder.d.ts +24 -0
  151. package/dist/rag/embeddings/embedder.d.ts.map +1 -0
  152. package/dist/rag/embeddings/embedder.js +73 -0
  153. package/dist/rag/embeddings/embedder.js.map +1 -0
  154. package/dist/rag/embeddings/embedding-cache.d.ts +40 -0
  155. package/dist/rag/embeddings/embedding-cache.d.ts.map +1 -0
  156. package/dist/rag/embeddings/embedding-cache.js +118 -0
  157. package/dist/rag/embeddings/embedding-cache.js.map +1 -0
  158. package/dist/rag/embeddings/fake-embedder.d.ts +20 -0
  159. package/dist/rag/embeddings/fake-embedder.d.ts.map +1 -0
  160. package/dist/rag/embeddings/fake-embedder.js +55 -0
  161. package/dist/rag/embeddings/fake-embedder.js.map +1 -0
  162. package/dist/rag/embeddings/index.d.ts +5 -0
  163. package/dist/rag/embeddings/index.d.ts.map +1 -0
  164. package/dist/rag/embeddings/index.js +21 -0
  165. package/dist/rag/embeddings/index.js.map +1 -0
  166. package/dist/rag/embeddings/openai-compatible-embedder.d.ts +46 -0
  167. package/dist/rag/embeddings/openai-compatible-embedder.d.ts.map +1 -0
  168. package/dist/rag/embeddings/openai-compatible-embedder.js +145 -0
  169. package/dist/rag/embeddings/openai-compatible-embedder.js.map +1 -0
  170. package/dist/rag/generation/context-builder.d.ts +22 -0
  171. package/dist/rag/generation/context-builder.d.ts.map +1 -0
  172. package/dist/rag/generation/context-builder.js +166 -0
  173. package/dist/rag/generation/context-builder.js.map +1 -0
  174. package/dist/rag/generation/generator.d.ts +25 -0
  175. package/dist/rag/generation/generator.d.ts.map +1 -0
  176. package/dist/rag/generation/generator.js +185 -0
  177. package/dist/rag/generation/generator.js.map +1 -0
  178. package/dist/rag/generation/index.d.ts +4 -0
  179. package/dist/rag/generation/index.d.ts.map +1 -0
  180. package/dist/rag/generation/index.js +20 -0
  181. package/dist/rag/generation/index.js.map +1 -0
  182. package/dist/rag/generation/verifier.d.ts +13 -0
  183. package/dist/rag/generation/verifier.d.ts.map +1 -0
  184. package/dist/rag/generation/verifier.js +43 -0
  185. package/dist/rag/generation/verifier.js.map +1 -0
  186. package/dist/rag/index.d.ts +9 -0
  187. package/dist/rag/index.d.ts.map +1 -0
  188. package/dist/rag/index.js +25 -0
  189. package/dist/rag/index.js.map +1 -0
  190. package/dist/rag/ingestion/index.d.ts +4 -0
  191. package/dist/rag/ingestion/index.d.ts.map +1 -0
  192. package/dist/rag/ingestion/index.js +20 -0
  193. package/dist/rag/ingestion/index.js.map +1 -0
  194. package/dist/rag/ingestion/loaders.d.ts +69 -0
  195. package/dist/rag/ingestion/loaders.d.ts.map +1 -0
  196. package/dist/rag/ingestion/loaders.js +653 -0
  197. package/dist/rag/ingestion/loaders.js.map +1 -0
  198. package/dist/rag/ingestion/metadata.d.ts +31 -0
  199. package/dist/rag/ingestion/metadata.d.ts.map +1 -0
  200. package/dist/rag/ingestion/metadata.js +81 -0
  201. package/dist/rag/ingestion/metadata.js.map +1 -0
  202. package/dist/rag/ingestion/normalizer.d.ts +20 -0
  203. package/dist/rag/ingestion/normalizer.d.ts.map +1 -0
  204. package/dist/rag/ingestion/normalizer.js +161 -0
  205. package/dist/rag/ingestion/normalizer.js.map +1 -0
  206. package/dist/rag/pipeline.d.ts +62 -0
  207. package/dist/rag/pipeline.d.ts.map +1 -0
  208. package/dist/rag/pipeline.js +75 -0
  209. package/dist/rag/pipeline.js.map +1 -0
  210. package/dist/rag/retrieval/dense-retriever.d.ts +17 -0
  211. package/dist/rag/retrieval/dense-retriever.d.ts.map +1 -0
  212. package/dist/rag/retrieval/dense-retriever.js +21 -0
  213. package/dist/rag/retrieval/dense-retriever.js.map +1 -0
  214. package/dist/rag/retrieval/fusion.d.ts +12 -0
  215. package/dist/rag/retrieval/fusion.d.ts.map +1 -0
  216. package/dist/rag/retrieval/fusion.js +54 -0
  217. package/dist/rag/retrieval/fusion.js.map +1 -0
  218. package/dist/rag/retrieval/http-rerank-model-client.d.ts +41 -0
  219. package/dist/rag/retrieval/http-rerank-model-client.d.ts.map +1 -0
  220. package/dist/rag/retrieval/http-rerank-model-client.js +130 -0
  221. package/dist/rag/retrieval/http-rerank-model-client.js.map +1 -0
  222. package/dist/rag/retrieval/hybrid-retriever.d.ts +22 -0
  223. package/dist/rag/retrieval/hybrid-retriever.d.ts.map +1 -0
  224. package/dist/rag/retrieval/hybrid-retriever.js +49 -0
  225. package/dist/rag/retrieval/hybrid-retriever.js.map +1 -0
  226. package/dist/rag/retrieval/index.d.ts +8 -0
  227. package/dist/rag/retrieval/index.d.ts.map +1 -0
  228. package/dist/rag/retrieval/index.js +24 -0
  229. package/dist/rag/retrieval/index.js.map +1 -0
  230. package/dist/rag/retrieval/reranker.d.ts +30 -0
  231. package/dist/rag/retrieval/reranker.d.ts.map +1 -0
  232. package/dist/rag/retrieval/reranker.js +85 -0
  233. package/dist/rag/retrieval/reranker.js.map +1 -0
  234. package/dist/rag/retrieval/sparse-retriever.d.ts +14 -0
  235. package/dist/rag/retrieval/sparse-retriever.d.ts.map +1 -0
  236. package/dist/rag/retrieval/sparse-retriever.js +18 -0
  237. package/dist/rag/retrieval/sparse-retriever.js.map +1 -0
  238. package/dist/rag/retrieval/types.d.ts +11 -0
  239. package/dist/rag/retrieval/types.d.ts.map +1 -0
  240. package/dist/rag/retrieval/types.js +10 -0
  241. package/dist/rag/retrieval/types.js.map +1 -0
  242. package/dist/rag/retrieval/utils.d.ts +8 -0
  243. package/dist/rag/retrieval/utils.d.ts.map +1 -0
  244. package/dist/rag/retrieval/utils.js +114 -0
  245. package/dist/rag/retrieval/utils.js.map +1 -0
  246. package/dist/rag/stores/in-memory-document-store.d.ts +14 -0
  247. package/dist/rag/stores/in-memory-document-store.d.ts.map +1 -0
  248. package/dist/rag/stores/in-memory-document-store.js +64 -0
  249. package/dist/rag/stores/in-memory-document-store.js.map +1 -0
  250. package/dist/rag/stores/in-memory-keyword-store.d.ts +10 -0
  251. package/dist/rag/stores/in-memory-keyword-store.d.ts.map +1 -0
  252. package/dist/rag/stores/in-memory-keyword-store.js +99 -0
  253. package/dist/rag/stores/in-memory-keyword-store.js.map +1 -0
  254. package/dist/rag/stores/in-memory-vector-store.d.ts +14 -0
  255. package/dist/rag/stores/in-memory-vector-store.d.ts.map +1 -0
  256. package/dist/rag/stores/in-memory-vector-store.js +61 -0
  257. package/dist/rag/stores/in-memory-vector-store.js.map +1 -0
  258. package/dist/rag/stores/index.d.ts +6 -0
  259. package/dist/rag/stores/index.d.ts.map +1 -0
  260. package/dist/rag/stores/index.js +22 -0
  261. package/dist/rag/stores/index.js.map +1 -0
  262. package/dist/rag/stores/postgres/index.d.ts +6 -0
  263. package/dist/rag/stores/postgres/index.d.ts.map +1 -0
  264. package/dist/rag/stores/postgres/index.js +22 -0
  265. package/dist/rag/stores/postgres/index.js.map +1 -0
  266. package/dist/rag/stores/postgres/pg-vector-store.d.ts +16 -0
  267. package/dist/rag/stores/postgres/pg-vector-store.d.ts.map +1 -0
  268. package/dist/rag/stores/postgres/pg-vector-store.js +77 -0
  269. package/dist/rag/stores/postgres/pg-vector-store.js.map +1 -0
  270. package/dist/rag/stores/postgres/postgres-document-store.d.ts +15 -0
  271. package/dist/rag/stores/postgres/postgres-document-store.d.ts.map +1 -0
  272. package/dist/rag/stores/postgres/postgres-document-store.js +174 -0
  273. package/dist/rag/stores/postgres/postgres-document-store.js.map +1 -0
  274. package/dist/rag/stores/postgres/postgres-keyword-store.d.ts +15 -0
  275. package/dist/rag/stores/postgres/postgres-keyword-store.d.ts.map +1 -0
  276. package/dist/rag/stores/postgres/postgres-keyword-store.js +163 -0
  277. package/dist/rag/stores/postgres/postgres-keyword-store.js.map +1 -0
  278. package/dist/rag/stores/postgres/schema.d.ts +8 -0
  279. package/dist/rag/stores/postgres/schema.d.ts.map +1 -0
  280. package/dist/rag/stores/postgres/schema.js +91 -0
  281. package/dist/rag/stores/postgres/schema.js.map +1 -0
  282. package/dist/rag/stores/postgres/sql.d.ts +23 -0
  283. package/dist/rag/stores/postgres/sql.d.ts.map +1 -0
  284. package/dist/rag/stores/postgres/sql.js +177 -0
  285. package/dist/rag/stores/postgres/sql.js.map +1 -0
  286. package/dist/rag/stores/postgres/types.d.ts +17 -0
  287. package/dist/rag/stores/postgres/types.d.ts.map +1 -0
  288. package/dist/rag/stores/postgres/types.js +3 -0
  289. package/dist/rag/stores/postgres/types.js.map +1 -0
  290. package/dist/rag/stores/types.d.ts +36 -0
  291. package/dist/rag/stores/types.d.ts.map +1 -0
  292. package/dist/rag/stores/types.js +3 -0
  293. package/dist/rag/stores/types.js.map +1 -0
  294. package/dist/rag/stores/utils.d.ts +11 -0
  295. package/dist/rag/stores/utils.d.ts.map +1 -0
  296. package/dist/rag/stores/utils.js +200 -0
  297. package/dist/rag/stores/utils.js.map +1 -0
  298. package/dist/rag/types.d.ts +174 -0
  299. package/dist/rag/types.d.ts.map +1 -0
  300. package/dist/rag/types.js +3 -0
  301. package/dist/rag/types.js.map +1 -0
  302. package/dist/tools/base.d.ts +28 -0
  303. package/dist/tools/base.d.ts.map +1 -0
  304. package/dist/tools/base.js +68 -0
  305. package/dist/tools/base.js.map +1 -0
  306. package/dist/tools/builtin/advancedSearchTool.d.ts +24 -0
  307. package/dist/tools/builtin/advancedSearchTool.d.ts.map +1 -0
  308. package/dist/tools/builtin/advancedSearchTool.js +134 -0
  309. package/dist/tools/builtin/advancedSearchTool.js.map +1 -0
  310. package/dist/tools/builtin/ragSearchTool.d.ts +29 -0
  311. package/dist/tools/builtin/ragSearchTool.d.ts.map +1 -0
  312. package/dist/tools/builtin/ragSearchTool.js +91 -0
  313. package/dist/tools/builtin/ragSearchTool.js.map +1 -0
  314. package/dist/tools/executor.d.ts +10 -0
  315. package/dist/tools/executor.d.ts.map +1 -0
  316. package/dist/tools/executor.js +86 -0
  317. package/dist/tools/executor.js.map +1 -0
  318. package/dist/tools/index.d.ts +9 -0
  319. package/dist/tools/index.d.ts.map +1 -0
  320. package/dist/tools/index.js +16 -0
  321. package/dist/tools/index.js.map +1 -0
  322. package/dist/tools/registry.d.ts +16 -0
  323. package/dist/tools/registry.d.ts.map +1 -0
  324. package/dist/tools/registry.js +35 -0
  325. package/dist/tools/registry.js.map +1 -0
  326. package/package.json +43 -0
@@ -0,0 +1,174 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.PostgresDocumentStore = void 0;
4
+ const sql_1 = require("./sql");
5
+ class PostgresDocumentStore {
6
+ db;
7
+ tables;
8
+ constructor(options) {
9
+ this.db = options.db;
10
+ this.tables = (0, sql_1.resolvePostgresTableNames)(options.tablePrefix);
11
+ }
12
+ async upsertDocuments(documents) {
13
+ for (const document of documents) {
14
+ await this.db.query(`
15
+ INSERT INTO ${this.tables.documents} (
16
+ id, content, metadata, blocks, source, title, mime_type, tenant_id, knowledge_base_id,
17
+ acl, version, checksum, created_at_meta, updated_at_meta, updated_at
18
+ ) VALUES (
19
+ $1, $2, $3::jsonb, $4::jsonb, $5, $6, $7, $8, $9, $10::text[], $11, $12, $13, $14, now()
20
+ )
21
+ ON CONFLICT (id) DO UPDATE SET
22
+ content = EXCLUDED.content,
23
+ metadata = EXCLUDED.metadata,
24
+ blocks = EXCLUDED.blocks,
25
+ source = EXCLUDED.source,
26
+ title = EXCLUDED.title,
27
+ mime_type = EXCLUDED.mime_type,
28
+ tenant_id = EXCLUDED.tenant_id,
29
+ knowledge_base_id = EXCLUDED.knowledge_base_id,
30
+ acl = EXCLUDED.acl,
31
+ version = EXCLUDED.version,
32
+ checksum = EXCLUDED.checksum,
33
+ created_at_meta = EXCLUDED.created_at_meta,
34
+ updated_at_meta = EXCLUDED.updated_at_meta,
35
+ updated_at = now()
36
+ `.trim(), [
37
+ document.id,
38
+ document.content,
39
+ (0, sql_1.createJson)(document.metadata),
40
+ (0, sql_1.createOptionalJson)(document.blocks),
41
+ document.metadata.source,
42
+ document.metadata.title ?? null,
43
+ document.metadata.mimeType ?? null,
44
+ document.metadata.tenantId ?? null,
45
+ document.metadata.knowledgeBaseId ?? null,
46
+ document.metadata.acl ?? null,
47
+ document.metadata.version ?? null,
48
+ document.metadata.checksum ?? null,
49
+ document.metadata.createdAt ?? null,
50
+ document.metadata.updatedAt ?? null,
51
+ ]);
52
+ }
53
+ }
54
+ async upsertChunks(chunks) {
55
+ for (const chunk of chunks) {
56
+ await this.db.query(`
57
+ INSERT INTO ${this.tables.chunks} (
58
+ id, document_id, parent_id, content, kind, metadata, source, title, tenant_id,
59
+ knowledge_base_id, heading_path, page, start_offset, end_offset, chunk_index, acl, extra,
60
+ updated_at
61
+ ) VALUES (
62
+ $1, $2, $3, $4, $5, $6::jsonb, $7, $8, $9, $10, $11::text[], $12, $13, $14, $15, $16::text[],
63
+ $17::jsonb, now()
64
+ )
65
+ ON CONFLICT (id) DO UPDATE SET
66
+ document_id = EXCLUDED.document_id,
67
+ parent_id = EXCLUDED.parent_id,
68
+ content = EXCLUDED.content,
69
+ kind = EXCLUDED.kind,
70
+ metadata = EXCLUDED.metadata,
71
+ source = EXCLUDED.source,
72
+ title = EXCLUDED.title,
73
+ tenant_id = EXCLUDED.tenant_id,
74
+ knowledge_base_id = EXCLUDED.knowledge_base_id,
75
+ heading_path = EXCLUDED.heading_path,
76
+ page = EXCLUDED.page,
77
+ start_offset = EXCLUDED.start_offset,
78
+ end_offset = EXCLUDED.end_offset,
79
+ chunk_index = EXCLUDED.chunk_index,
80
+ acl = EXCLUDED.acl,
81
+ extra = EXCLUDED.extra,
82
+ updated_at = now()
83
+ `.trim(), [
84
+ chunk.id,
85
+ chunk.documentId,
86
+ chunk.parentId ?? null,
87
+ chunk.content,
88
+ chunk.kind,
89
+ (0, sql_1.createJson)(chunk.metadata),
90
+ chunk.metadata.source,
91
+ chunk.metadata.title ?? null,
92
+ chunk.metadata.tenantId ?? null,
93
+ chunk.metadata.knowledgeBaseId ?? null,
94
+ chunk.metadata.headingPath ?? null,
95
+ chunk.metadata.page ?? null,
96
+ chunk.metadata.startOffset ?? null,
97
+ chunk.metadata.endOffset ?? null,
98
+ chunk.metadata.chunkIndex,
99
+ chunk.metadata.acl ?? null,
100
+ (0, sql_1.createOptionalJson)(chunk.metadata.extra),
101
+ ]);
102
+ }
103
+ }
104
+ async getChunks(ids) {
105
+ if (ids.length === 0) {
106
+ return [];
107
+ }
108
+ const result = await this.db.query(`
109
+ SELECT *
110
+ FROM ${this.tables.chunks}
111
+ WHERE id = ANY($1::text[])
112
+ `.trim(), [ids]);
113
+ const chunksById = new Map(result.rows.map((row) => [String(row.id), rowToChunk(row)]));
114
+ return ids
115
+ .map((id) => chunksById.get(id))
116
+ .filter((chunk) => chunk !== undefined);
117
+ }
118
+ async getChunksByDocument(documentId) {
119
+ const result = await this.db.query(`
120
+ SELECT *
121
+ FROM ${this.tables.chunks}
122
+ WHERE document_id = $1
123
+ ORDER BY chunk_index ASC, id ASC
124
+ `.trim(), [documentId]);
125
+ return result.rows.map(rowToChunk);
126
+ }
127
+ async deleteDocument(documentId) {
128
+ await this.db.query(`DELETE FROM ${this.tables.documents} WHERE id = $1`, [documentId]);
129
+ }
130
+ async getDocument(documentId) {
131
+ const result = await this.db.query(`SELECT * FROM ${this.tables.documents} WHERE id = $1`, [documentId]);
132
+ const row = result.rows[0];
133
+ return row === undefined ? undefined : rowToRawDocument(row);
134
+ }
135
+ }
136
+ exports.PostgresDocumentStore = PostgresDocumentStore;
137
+ function rowToRawDocument(row) {
138
+ const metadata = (0, sql_1.readJsonObject)(row.metadata);
139
+ const documentMetadata = {
140
+ source: (0, sql_1.readOptionalString)(row.source) ?? metadata?.source ?? 'inline',
141
+ };
142
+ (0, sql_1.assignIfDefined)(documentMetadata, 'title', (0, sql_1.readOptionalString)(row.title) ?? metadata?.title);
143
+ (0, sql_1.assignIfDefined)(documentMetadata, 'mimeType', (0, sql_1.readOptionalString)(row.mime_type) ?? metadata?.mimeType);
144
+ (0, sql_1.assignIfDefined)(documentMetadata, 'tenantId', (0, sql_1.readOptionalString)(row.tenant_id) ?? metadata?.tenantId);
145
+ (0, sql_1.assignIfDefined)(documentMetadata, 'knowledgeBaseId', (0, sql_1.readOptionalString)(row.knowledge_base_id) ?? metadata?.knowledgeBaseId);
146
+ (0, sql_1.assignIfDefined)(documentMetadata, 'version', (0, sql_1.readOptionalString)(row.version) ?? metadata?.version);
147
+ (0, sql_1.assignIfDefined)(documentMetadata, 'checksum', (0, sql_1.readOptionalString)(row.checksum) ?? metadata?.checksum);
148
+ (0, sql_1.assignIfDefined)(documentMetadata, 'createdAt', (0, sql_1.readOptionalString)(row.created_at_meta) ?? metadata?.createdAt);
149
+ (0, sql_1.assignIfDefined)(documentMetadata, 'updatedAt', (0, sql_1.readOptionalString)(row.updated_at_meta) ?? metadata?.updatedAt);
150
+ (0, sql_1.assignIfDefined)(documentMetadata, 'acl', (0, sql_1.readOptionalStringArray)(row.acl) ?? metadata?.acl);
151
+ (0, sql_1.assignIfDefined)(documentMetadata, 'extra', metadata?.extra);
152
+ const document = {
153
+ id: String(row.id),
154
+ content: String(row.content ?? ''),
155
+ metadata: documentMetadata,
156
+ };
157
+ const blocks = (0, sql_1.readJsonArray)(row.blocks);
158
+ if (blocks !== undefined) {
159
+ document.blocks = blocks;
160
+ }
161
+ return document;
162
+ }
163
+ function rowToChunk(row) {
164
+ const chunk = {
165
+ id: String(row.id),
166
+ documentId: String(row.document_id),
167
+ content: String(row.content ?? ''),
168
+ kind: String(row.kind),
169
+ metadata: (0, sql_1.rowToChunkMetadata)(row),
170
+ };
171
+ (0, sql_1.assignIfDefined)(chunk, 'parentId', (0, sql_1.readOptionalString)(row.parent_id));
172
+ return chunk;
173
+ }
174
+ //# sourceMappingURL=postgres-document-store.js.map
@@ -0,0 +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"}
@@ -0,0 +1,15 @@
1
+ import type { Chunk } from '../../types';
2
+ import type { KeywordStore, ScoredChunkRef, SearchOptions } from '../types';
3
+ import type { PostgresRagStoreOptions } from './types';
4
+ export declare class PostgresKeywordStore implements KeywordStore {
5
+ private readonly db;
6
+ private readonly tables;
7
+ constructor(options: PostgresRagStoreOptions);
8
+ upsert(chunks: Chunk[]): Promise<void>;
9
+ search(query: string, options?: SearchOptions): Promise<ScoredChunkRef[]>;
10
+ deleteByDocument(documentId: string): Promise<void>;
11
+ private loadMatchingEntries;
12
+ private calculateDocumentFrequency;
13
+ private countEntries;
14
+ }
15
+ //# sourceMappingURL=postgres-keyword-store.d.ts.map
@@ -0,0 +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"}
@@ -0,0 +1,163 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.PostgresKeywordStore = void 0;
4
+ const sql_1 = require("./sql");
5
+ class PostgresKeywordStore {
6
+ db;
7
+ tables;
8
+ constructor(options) {
9
+ this.db = options.db;
10
+ this.tables = (0, sql_1.resolvePostgresTableNames)(options.tablePrefix);
11
+ }
12
+ 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
+ ]);
34
+ }
35
+ }
36
+ async search(query, options = {}) {
37
+ const queryTerms = [...new Set(tokenize(query))];
38
+ const topK = (0, sql_1.normalizeTopK)(options.topK);
39
+ if (queryTerms.length === 0 || topK === 0) {
40
+ return [];
41
+ }
42
+ const entries = await this.loadMatchingEntries(queryTerms, options);
43
+ const documentFrequency = await this.calculateDocumentFrequency(queryTerms);
44
+ const totalDocuments = await this.countEntries();
45
+ const refs = entries
46
+ .map((entry) => ({
47
+ entry,
48
+ score: scoreEntry({
49
+ entry,
50
+ queryTerms,
51
+ documentFrequency,
52
+ totalDocuments,
53
+ }),
54
+ }))
55
+ .filter((item) => item.score > 0)
56
+ .filter((item) => options.minScore === undefined || item.score >= options.minScore)
57
+ .sort((left, right) => (right.score - left.score
58
+ || left.entry.documentId.localeCompare(right.entry.documentId)
59
+ || left.entry.chunkId.localeCompare(right.entry.chunkId)))
60
+ .slice(0, topK);
61
+ return refs.map((item) => ({
62
+ chunkId: item.entry.chunkId,
63
+ documentId: item.entry.documentId,
64
+ score: item.score,
65
+ metadata: {
66
+ source: 'keyword',
67
+ matchedTerms: queryTerms.filter((term) => item.entry.termCounts.has(term)),
68
+ },
69
+ }));
70
+ }
71
+ async deleteByDocument(documentId) {
72
+ await this.db.query(`DELETE FROM ${this.tables.keywordEntries} WHERE document_id = $1`, [documentId]);
73
+ }
74
+ async loadMatchingEntries(queryTerms, options) {
75
+ 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)}` : ''}
87
+ `.trim(), [queryTerms, ...where.values]);
88
+ return result.rows.map((row) => ({
89
+ chunkId: String(row.chunk_id),
90
+ documentId: String(row.document_id),
91
+ termCounts: jsonToTermCounts(row.term_counts),
92
+ length: Number(row.length ?? 0),
93
+ content: String(row.content ?? ''),
94
+ }));
95
+ }
96
+ async calculateDocumentFrequency(queryTerms) {
97
+ const result = await this.db.query(`
98
+ SELECT terms
99
+ FROM ${this.tables.keywordEntries}
100
+ WHERE terms && $1::text[]
101
+ `.trim(), [queryTerms]);
102
+ const frequency = new Map(queryTerms.map((term) => [term, 0]));
103
+ for (const row of result.rows) {
104
+ const terms = readStringArray(row.terms);
105
+ for (const term of queryTerms) {
106
+ if (terms.includes(term)) {
107
+ frequency.set(term, (frequency.get(term) ?? 0) + 1);
108
+ }
109
+ }
110
+ }
111
+ return frequency;
112
+ }
113
+ async countEntries() {
114
+ const result = await this.db.query(`SELECT count(*) AS count FROM ${this.tables.keywordEntries}`);
115
+ const count = Number(result.rows[0]?.count ?? 0);
116
+ return Math.max(count, 1);
117
+ }
118
+ }
119
+ exports.PostgresKeywordStore = PostgresKeywordStore;
120
+ function scoreEntry(input) {
121
+ let score = 0;
122
+ for (const term of input.queryTerms) {
123
+ const termCount = input.entry.termCounts.get(term) ?? 0;
124
+ if (termCount === 0) {
125
+ continue;
126
+ }
127
+ const df = input.documentFrequency.get(term) ?? 0;
128
+ const idf = Math.log(1 + (input.totalDocuments - df + 0.5) / (df + 0.5));
129
+ const normalizedTf = termCount / Math.sqrt(Math.max(input.entry.length, 1));
130
+ score += normalizedTf * idf;
131
+ }
132
+ return score;
133
+ }
134
+ function tokenize(text) {
135
+ return text
136
+ .normalize('NFKC')
137
+ .toLowerCase()
138
+ .match(/[\p{L}\p{N}_-]+/gu) ?? [];
139
+ }
140
+ function countTerms(terms) {
141
+ const counts = new Map();
142
+ for (const term of terms) {
143
+ counts.set(term, (counts.get(term) ?? 0) + 1);
144
+ }
145
+ return counts;
146
+ }
147
+ function jsonToTermCounts(value) {
148
+ const record = (0, sql_1.readJsonObject)(value) ?? {};
149
+ const counts = new Map();
150
+ for (const [term, count] of Object.entries(record)) {
151
+ if (typeof count === 'number') {
152
+ counts.set(term, count);
153
+ }
154
+ }
155
+ return counts;
156
+ }
157
+ function readStringArray(value) {
158
+ if (!Array.isArray(value)) {
159
+ return [];
160
+ }
161
+ return value.filter((item) => typeof item === 'string');
162
+ }
163
+ //# sourceMappingURL=postgres-keyword-store.js.map
@@ -0,0 +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"}
@@ -0,0 +1,8 @@
1
+ export interface PostgresRagSchemaOptions {
2
+ tablePrefix?: string;
3
+ dimensions: number;
4
+ createVectorExtension?: boolean;
5
+ createIndexes?: boolean;
6
+ }
7
+ export declare function createPostgresRagSchemaSql(options: PostgresRagSchemaOptions): string;
8
+ //# sourceMappingURL=schema.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/schema.ts"],"names":[],"mappings":"AAOA,MAAM,WAAW,wBAAwB;IACrC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,aAAa,CAAC,EAAE,OAAO,CAAC;CAC3B;AAED,wBAAgB,0BAA0B,CAAC,OAAO,EAAE,wBAAwB,GAAG,MAAM,CAuFpF"}
@@ -0,0 +1,91 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.createPostgresRagSchemaSql = createPostgresRagSchemaSql;
4
+ const embeddings_1 = require("../../embeddings");
5
+ const sql_1 = require("./sql");
6
+ function createPostgresRagSchemaSql(options) {
7
+ const dimensions = (0, embeddings_1.assertValidEmbeddingDimensions)(options.dimensions);
8
+ const tables = (0, sql_1.resolvePostgresTableNames)(options.tablePrefix);
9
+ const statements = [];
10
+ if (options.createVectorExtension ?? true) {
11
+ statements.push('CREATE EXTENSION IF NOT EXISTS vector;');
12
+ }
13
+ statements.push(`
14
+ CREATE TABLE IF NOT EXISTS ${tables.documents} (
15
+ id text PRIMARY KEY,
16
+ content text NOT NULL,
17
+ metadata jsonb NOT NULL DEFAULT '{}'::jsonb,
18
+ blocks jsonb,
19
+ source text NOT NULL,
20
+ title text,
21
+ mime_type text,
22
+ tenant_id text,
23
+ knowledge_base_id text,
24
+ acl text[],
25
+ version text,
26
+ checksum text,
27
+ created_at_meta text,
28
+ updated_at_meta text,
29
+ created_at timestamptz NOT NULL DEFAULT now(),
30
+ updated_at timestamptz NOT NULL DEFAULT now()
31
+ );`.trim());
32
+ statements.push(`
33
+ CREATE TABLE IF NOT EXISTS ${tables.chunks} (
34
+ id text PRIMARY KEY,
35
+ document_id text NOT NULL REFERENCES ${tables.documents}(id) ON DELETE CASCADE,
36
+ parent_id text,
37
+ content text NOT NULL,
38
+ kind text NOT NULL,
39
+ metadata jsonb NOT NULL DEFAULT '{}'::jsonb,
40
+ source text NOT NULL,
41
+ title text,
42
+ tenant_id text,
43
+ knowledge_base_id text,
44
+ heading_path text[],
45
+ page integer,
46
+ start_offset integer,
47
+ end_offset integer,
48
+ chunk_index integer NOT NULL,
49
+ acl text[],
50
+ extra jsonb,
51
+ created_at timestamptz NOT NULL DEFAULT now(),
52
+ updated_at timestamptz NOT NULL DEFAULT now(),
53
+ search_vector tsvector GENERATED ALWAYS AS (to_tsvector('simple', content)) STORED
54
+ );`.trim());
55
+ statements.push(`
56
+ CREATE TABLE IF NOT EXISTS ${tables.embeddings} (
57
+ chunk_id text PRIMARY KEY REFERENCES ${tables.chunks}(id) ON DELETE CASCADE,
58
+ document_id text NOT NULL,
59
+ embedding vector(${dimensions}) NOT NULL,
60
+ created_at timestamptz NOT NULL DEFAULT now(),
61
+ updated_at timestamptz NOT NULL DEFAULT now()
62
+ );`.trim());
63
+ statements.push(`
64
+ CREATE TABLE IF NOT EXISTS ${tables.keywordEntries} (
65
+ chunk_id text PRIMARY KEY REFERENCES ${tables.chunks}(id) ON DELETE CASCADE,
66
+ document_id text NOT NULL,
67
+ terms text[] NOT NULL,
68
+ term_counts jsonb NOT NULL,
69
+ length integer NOT NULL,
70
+ created_at timestamptz NOT NULL DEFAULT now(),
71
+ updated_at timestamptz NOT NULL DEFAULT now()
72
+ );`.trim());
73
+ if (options.createIndexes ?? true) {
74
+ statements.push(`CREATE INDEX IF NOT EXISTS ${indexName(options.tablePrefix, 'chunks_document_id_idx')} ON ${tables.chunks} (document_id);`);
75
+ statements.push(`CREATE INDEX IF NOT EXISTS ${indexName(options.tablePrefix, 'chunks_tenant_kb_idx')} ON ${tables.chunks} (tenant_id, knowledge_base_id);`);
76
+ statements.push(`CREATE INDEX IF NOT EXISTS ${indexName(options.tablePrefix, 'chunks_source_idx')} ON ${tables.chunks} (source);`);
77
+ statements.push(`CREATE INDEX IF NOT EXISTS ${indexName(options.tablePrefix, 'chunks_acl_idx')} ON ${tables.chunks} USING gin (acl);`);
78
+ statements.push(`CREATE INDEX IF NOT EXISTS ${indexName(options.tablePrefix, 'chunks_extra_idx')} ON ${tables.chunks} USING gin (extra);`);
79
+ statements.push(`CREATE INDEX IF NOT EXISTS ${indexName(options.tablePrefix, 'chunks_search_vector_idx')} ON ${tables.chunks} USING gin (search_vector);`);
80
+ statements.push(`CREATE INDEX IF NOT EXISTS ${indexName(options.tablePrefix, 'embeddings_document_id_idx')} ON ${tables.embeddings} (document_id);`);
81
+ statements.push(`CREATE INDEX IF NOT EXISTS ${indexName(options.tablePrefix, 'embeddings_vector_idx')} ON ${tables.embeddings} USING hnsw (embedding vector_cosine_ops);`);
82
+ statements.push(`CREATE INDEX IF NOT EXISTS ${indexName(options.tablePrefix, 'keyword_document_id_idx')} ON ${tables.keywordEntries} (document_id);`);
83
+ statements.push(`CREATE INDEX IF NOT EXISTS ${indexName(options.tablePrefix, 'keyword_terms_idx')} ON ${tables.keywordEntries} USING gin (terms);`);
84
+ }
85
+ return `${statements.join('\n\n')}\n`;
86
+ }
87
+ function indexName(tablePrefix, suffix) {
88
+ const prefix = tablePrefix ?? 'rag';
89
+ return `"${prefix}_${suffix}"`;
90
+ }
91
+ //# sourceMappingURL=schema.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"schema.js","sourceRoot":"","sources":["../../../../src/rag/stores/postgres/schema.ts"],"names":[],"mappings":";;AAcA,gEAuFC;AArGD,iDAE0B;AAC1B,+BAEe;AASf,SAAgB,0BAA0B,CAAC,OAAiC;IACxE,MAAM,UAAU,GAAG,IAAA,2CAA8B,EAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IACtE,MAAM,MAAM,GAAG,IAAA,+BAAyB,EAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC9D,MAAM,UAAU,GAAa,EAAE,CAAC;IAEhC,IAAI,OAAO,CAAC,qBAAqB,IAAI,IAAI,EAAE,CAAC;QACxC,UAAU,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;IAC9D,CAAC;IAED,UAAU,CAAC,IAAI,CAAC;6BACS,MAAM,CAAC,SAAS;;;;;;;;;;;;;;;;;GAiB1C,CAAC,IAAI,EAAE,CAAC,CAAC;IAER,UAAU,CAAC,IAAI,CAAC;6BACS,MAAM,CAAC,MAAM;;2CAEC,MAAM,CAAC,SAAS;;;;;;;;;;;;;;;;;;;GAmBxD,CAAC,IAAI,EAAE,CAAC,CAAC;IAER,UAAU,CAAC,IAAI,CAAC;6BACS,MAAM,CAAC,UAAU;2CACH,MAAM,CAAC,MAAM;;uBAEjC,UAAU;;;GAG9B,CAAC,IAAI,EAAE,CAAC,CAAC;IAER,UAAU,CAAC,IAAI,CAAC;6BACS,MAAM,CAAC,cAAc;2CACP,MAAM,CAAC,MAAM;;;;;;;GAOrD,CAAC,IAAI,EAAE,CAAC,CAAC;IAER,IAAI,OAAO,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;QAChC,UAAU,CAAC,IAAI,CAAC,8BAA8B,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,wBAAwB,CAAC,OAAO,MAAM,CAAC,MAAM,iBAAiB,CAAC,CAAC;QAC7I,UAAU,CAAC,IAAI,CAAC,8BAA8B,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,sBAAsB,CAAC,OAAO,MAAM,CAAC,MAAM,kCAAkC,CAAC,CAAC;QAC5J,UAAU,CAAC,IAAI,CAAC,8BAA8B,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,mBAAmB,CAAC,OAAO,MAAM,CAAC,MAAM,YAAY,CAAC,CAAC;QACnI,UAAU,CAAC,IAAI,CAAC,8BAA8B,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,gBAAgB,CAAC,OAAO,MAAM,CAAC,MAAM,mBAAmB,CAAC,CAAC;QACvI,UAAU,CAAC,IAAI,CAAC,8BAA8B,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,kBAAkB,CAAC,OAAO,MAAM,CAAC,MAAM,qBAAqB,CAAC,CAAC;QAC3I,UAAU,CAAC,IAAI,CAAC,8BAA8B,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,0BAA0B,CAAC,OAAO,MAAM,CAAC,MAAM,6BAA6B,CAAC,CAAC;QAC3J,UAAU,CAAC,IAAI,CAAC,8BAA8B,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,4BAA4B,CAAC,OAAO,MAAM,CAAC,UAAU,iBAAiB,CAAC,CAAC;QACrJ,UAAU,CAAC,IAAI,CAAC,8BAA8B,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,uBAAuB,CAAC,OAAO,MAAM,CAAC,UAAU,4CAA4C,CAAC,CAAC;QAC3K,UAAU,CAAC,IAAI,CAAC,8BAA8B,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,yBAAyB,CAAC,OAAO,MAAM,CAAC,cAAc,iBAAiB,CAAC,CAAC;QACtJ,UAAU,CAAC,IAAI,CAAC,8BAA8B,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,mBAAmB,CAAC,OAAO,MAAM,CAAC,cAAc,qBAAqB,CAAC,CAAC;IACxJ,CAAC;IAED,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;AAC1C,CAAC;AAED,SAAS,SAAS,CAAC,WAA+B,EAAE,MAAc;IAC9D,MAAM,MAAM,GAAG,WAAW,IAAI,KAAK,CAAC;IAEpC,OAAO,IAAI,MAAM,IAAI,MAAM,GAAG,CAAC;AACnC,CAAC"}
@@ -0,0 +1,23 @@
1
+ import type { ChunkMetadata } from '../../types';
2
+ import type { SearchOptions } from '../types';
3
+ import type { PostgresTableNames } from './types';
4
+ export declare function resolvePostgresTableNames(tablePrefix?: string): PostgresTableNames;
5
+ export declare function quoteIdentifier(identifier: string): string;
6
+ export declare function createJson(value: unknown): string;
7
+ export declare function createOptionalJson(value: unknown): string | null;
8
+ export declare function readJsonObject(value: unknown): Record<string, unknown> | undefined;
9
+ export declare function readJsonArray<T = unknown>(value: unknown): T[] | undefined;
10
+ export declare function readJson(value: unknown): unknown;
11
+ export declare function toVectorLiteral(vector: readonly number[]): string;
12
+ export interface SqlWhere {
13
+ clause: string;
14
+ values: unknown[];
15
+ }
16
+ export declare function buildChunkSearchWhere(options: SearchOptions, startIndex?: number, alias?: string): SqlWhere;
17
+ export declare function assignIfDefined<T extends object, K extends keyof T>(target: T, key: K, value: T[K] | undefined): void;
18
+ export declare function normalizeTopK(topK: number | undefined): number;
19
+ export declare function rowToChunkMetadata(row: Record<string, unknown>): ChunkMetadata;
20
+ export declare function readOptionalString(value: unknown): string | undefined;
21
+ export declare function readOptionalNumber(value: unknown): number | undefined;
22
+ export declare function readOptionalStringArray(value: unknown): string[] | undefined;
23
+ //# sourceMappingURL=sql.d.ts.map
@@ -0,0 +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"}