@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.
- package/README.md +240 -0
- package/dist/agent/Agent.d.ts +34 -0
- package/dist/agent/Agent.d.ts.map +1 -0
- package/dist/agent/Agent.js +168 -0
- package/dist/agent/Agent.js.map +1 -0
- package/dist/agent/createAgent.d.ts +4 -0
- package/dist/agent/createAgent.d.ts.map +1 -0
- package/dist/agent/createAgent.js +8 -0
- package/dist/agent/createAgent.js.map +1 -0
- package/dist/agent/index.d.ts +4 -0
- package/dist/agent/index.d.ts.map +1 -0
- package/dist/agent/index.js +8 -0
- package/dist/agent/index.js.map +1 -0
- package/dist/agent/types.d.ts +23 -0
- package/dist/agent/types.d.ts.map +1 -0
- package/dist/agent/types.js +3 -0
- package/dist/agent/types.js.map +1 -0
- package/dist/config/config.d.ts +35 -0
- package/dist/config/config.d.ts.map +1 -0
- package/dist/config/config.js +123 -0
- package/dist/config/config.js.map +1 -0
- package/dist/config/index.d.ts +3 -0
- package/dist/config/index.d.ts.map +1 -0
- package/dist/config/index.js +6 -0
- package/dist/config/index.js.map +1 -0
- package/dist/executors/base.d.ts +44 -0
- package/dist/executors/base.d.ts.map +1 -0
- package/dist/executors/base.js +3 -0
- package/dist/executors/base.js.map +1 -0
- package/dist/executors/index.d.ts +7 -0
- package/dist/executors/index.d.ts.map +1 -0
- package/dist/executors/index.js +10 -0
- package/dist/executors/index.js.map +1 -0
- package/dist/executors/rag-executor.d.ts +16 -0
- package/dist/executors/rag-executor.d.ts.map +1 -0
- package/dist/executors/rag-executor.js +120 -0
- package/dist/executors/rag-executor.js.map +1 -0
- package/dist/executors/simple-chat-executor.d.ts +5 -0
- package/dist/executors/simple-chat-executor.d.ts.map +1 -0
- package/dist/executors/simple-chat-executor.js +77 -0
- package/dist/executors/simple-chat-executor.js.map +1 -0
- package/dist/executors/tool-calling-executor.d.ts +10 -0
- package/dist/executors/tool-calling-executor.d.ts.map +1 -0
- package/dist/executors/tool-calling-executor.js +151 -0
- package/dist/executors/tool-calling-executor.js.map +1 -0
- package/dist/history/base.d.ts +16 -0
- package/dist/history/base.d.ts.map +1 -0
- package/dist/history/base.js +10 -0
- package/dist/history/base.js.map +1 -0
- package/dist/history/in-memory.d.ts +26 -0
- package/dist/history/in-memory.d.ts.map +1 -0
- package/dist/history/in-memory.js +88 -0
- package/dist/history/in-memory.js.map +1 -0
- package/dist/history/index.d.ts +5 -0
- package/dist/history/index.d.ts.map +1 -0
- package/dist/history/index.js +10 -0
- package/dist/history/index.js.map +1 -0
- package/dist/index.d.ts +10 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +26 -0
- package/dist/index.js.map +1 -0
- package/dist/llm/LLM.d.ts +23 -0
- package/dist/llm/LLM.d.ts.map +1 -0
- package/dist/llm/LLM.js +404 -0
- package/dist/llm/LLM.js.map +1 -0
- package/dist/llm/errors.d.ts +43 -0
- package/dist/llm/errors.d.ts.map +1 -0
- package/dist/llm/errors.js +128 -0
- package/dist/llm/errors.js.map +1 -0
- package/dist/llm/factory.d.ts +9 -0
- package/dist/llm/factory.d.ts.map +1 -0
- package/dist/llm/factory.js +28 -0
- package/dist/llm/factory.js.map +1 -0
- package/dist/llm/index.d.ts +6 -0
- package/dist/llm/index.d.ts.map +1 -0
- package/dist/llm/index.js +18 -0
- package/dist/llm/index.js.map +1 -0
- package/dist/llm/providers/base.d.ts +3 -0
- package/dist/llm/providers/base.d.ts.map +1 -0
- package/dist/llm/providers/base.js +3 -0
- package/dist/llm/providers/base.js.map +1 -0
- package/dist/llm/providers/local.d.ts +6 -0
- package/dist/llm/providers/local.d.ts.map +1 -0
- package/dist/llm/providers/local.js +19 -0
- package/dist/llm/providers/local.js.map +1 -0
- package/dist/llm/providers/openai-compatible.d.ts +13 -0
- package/dist/llm/providers/openai-compatible.d.ts.map +1 -0
- package/dist/llm/providers/openai-compatible.js +176 -0
- package/dist/llm/providers/openai-compatible.js.map +1 -0
- package/dist/llm/providers/openai.d.ts +6 -0
- package/dist/llm/providers/openai.d.ts.map +1 -0
- package/dist/llm/providers/openai.js +23 -0
- package/dist/llm/providers/openai.js.map +1 -0
- package/dist/llm/types.d.ts +116 -0
- package/dist/llm/types.d.ts.map +1 -0
- package/dist/llm/types.js +3 -0
- package/dist/llm/types.js.map +1 -0
- package/dist/messages/index.d.ts +3 -0
- package/dist/messages/index.d.ts.map +1 -0
- package/dist/messages/index.js +6 -0
- package/dist/messages/index.js.map +1 -0
- package/dist/messages/message.d.ts +25 -0
- package/dist/messages/message.d.ts.map +1 -0
- package/dist/messages/message.js +110 -0
- package/dist/messages/message.js.map +1 -0
- package/dist/messages/types.d.ts +20 -0
- package/dist/messages/types.d.ts.map +1 -0
- package/dist/messages/types.js +3 -0
- package/dist/messages/types.js.map +1 -0
- package/dist/observability/console-observer.d.ts +10 -0
- package/dist/observability/console-observer.d.ts.map +1 -0
- package/dist/observability/console-observer.js +28 -0
- package/dist/observability/console-observer.js.map +1 -0
- package/dist/observability/index.d.ts +5 -0
- package/dist/observability/index.d.ts.map +1 -0
- package/dist/observability/index.js +11 -0
- package/dist/observability/index.js.map +1 -0
- package/dist/observability/observer.d.ts +8 -0
- package/dist/observability/observer.d.ts.map +1 -0
- package/dist/observability/observer.js +46 -0
- package/dist/observability/observer.js.map +1 -0
- package/dist/observability/types.d.ts +34 -0
- package/dist/observability/types.d.ts.map +1 -0
- package/dist/observability/types.js +3 -0
- package/dist/observability/types.js.map +1 -0
- package/dist/rag/chunking/auto-chunker.d.ts +25 -0
- package/dist/rag/chunking/auto-chunker.d.ts.map +1 -0
- package/dist/rag/chunking/auto-chunker.js +101 -0
- package/dist/rag/chunking/auto-chunker.js.map +1 -0
- package/dist/rag/chunking/chunker.d.ts +44 -0
- package/dist/rag/chunking/chunker.d.ts.map +1 -0
- package/dist/rag/chunking/chunker.js +162 -0
- package/dist/rag/chunking/chunker.js.map +1 -0
- package/dist/rag/chunking/index.d.ts +6 -0
- package/dist/rag/chunking/index.d.ts.map +1 -0
- package/dist/rag/chunking/index.js +22 -0
- package/dist/rag/chunking/index.js.map +1 -0
- package/dist/rag/chunking/markdown-chunker.d.ts +13 -0
- package/dist/rag/chunking/markdown-chunker.d.ts.map +1 -0
- package/dist/rag/chunking/markdown-chunker.js +144 -0
- package/dist/rag/chunking/markdown-chunker.js.map +1 -0
- package/dist/rag/chunking/parent-child.d.ts +16 -0
- package/dist/rag/chunking/parent-child.d.ts.map +1 -0
- package/dist/rag/chunking/parent-child.js +76 -0
- package/dist/rag/chunking/parent-child.js.map +1 -0
- package/dist/rag/chunking/recursive-chunker.d.ts +11 -0
- package/dist/rag/chunking/recursive-chunker.d.ts.map +1 -0
- package/dist/rag/chunking/recursive-chunker.js +27 -0
- package/dist/rag/chunking/recursive-chunker.js.map +1 -0
- package/dist/rag/embeddings/embedder.d.ts +24 -0
- package/dist/rag/embeddings/embedder.d.ts.map +1 -0
- package/dist/rag/embeddings/embedder.js +73 -0
- package/dist/rag/embeddings/embedder.js.map +1 -0
- package/dist/rag/embeddings/embedding-cache.d.ts +40 -0
- package/dist/rag/embeddings/embedding-cache.d.ts.map +1 -0
- package/dist/rag/embeddings/embedding-cache.js +118 -0
- package/dist/rag/embeddings/embedding-cache.js.map +1 -0
- package/dist/rag/embeddings/fake-embedder.d.ts +20 -0
- package/dist/rag/embeddings/fake-embedder.d.ts.map +1 -0
- package/dist/rag/embeddings/fake-embedder.js +55 -0
- package/dist/rag/embeddings/fake-embedder.js.map +1 -0
- package/dist/rag/embeddings/index.d.ts +5 -0
- package/dist/rag/embeddings/index.d.ts.map +1 -0
- package/dist/rag/embeddings/index.js +21 -0
- package/dist/rag/embeddings/index.js.map +1 -0
- package/dist/rag/embeddings/openai-compatible-embedder.d.ts +46 -0
- package/dist/rag/embeddings/openai-compatible-embedder.d.ts.map +1 -0
- package/dist/rag/embeddings/openai-compatible-embedder.js +145 -0
- package/dist/rag/embeddings/openai-compatible-embedder.js.map +1 -0
- package/dist/rag/generation/context-builder.d.ts +22 -0
- package/dist/rag/generation/context-builder.d.ts.map +1 -0
- package/dist/rag/generation/context-builder.js +166 -0
- package/dist/rag/generation/context-builder.js.map +1 -0
- package/dist/rag/generation/generator.d.ts +25 -0
- package/dist/rag/generation/generator.d.ts.map +1 -0
- package/dist/rag/generation/generator.js +185 -0
- package/dist/rag/generation/generator.js.map +1 -0
- package/dist/rag/generation/index.d.ts +4 -0
- package/dist/rag/generation/index.d.ts.map +1 -0
- package/dist/rag/generation/index.js +20 -0
- package/dist/rag/generation/index.js.map +1 -0
- package/dist/rag/generation/verifier.d.ts +13 -0
- package/dist/rag/generation/verifier.d.ts.map +1 -0
- package/dist/rag/generation/verifier.js +43 -0
- package/dist/rag/generation/verifier.js.map +1 -0
- package/dist/rag/index.d.ts +9 -0
- package/dist/rag/index.d.ts.map +1 -0
- package/dist/rag/index.js +25 -0
- package/dist/rag/index.js.map +1 -0
- package/dist/rag/ingestion/index.d.ts +4 -0
- package/dist/rag/ingestion/index.d.ts.map +1 -0
- package/dist/rag/ingestion/index.js +20 -0
- package/dist/rag/ingestion/index.js.map +1 -0
- package/dist/rag/ingestion/loaders.d.ts +69 -0
- package/dist/rag/ingestion/loaders.d.ts.map +1 -0
- package/dist/rag/ingestion/loaders.js +653 -0
- package/dist/rag/ingestion/loaders.js.map +1 -0
- package/dist/rag/ingestion/metadata.d.ts +31 -0
- package/dist/rag/ingestion/metadata.d.ts.map +1 -0
- package/dist/rag/ingestion/metadata.js +81 -0
- package/dist/rag/ingestion/metadata.js.map +1 -0
- package/dist/rag/ingestion/normalizer.d.ts +20 -0
- package/dist/rag/ingestion/normalizer.d.ts.map +1 -0
- package/dist/rag/ingestion/normalizer.js +161 -0
- package/dist/rag/ingestion/normalizer.js.map +1 -0
- package/dist/rag/pipeline.d.ts +62 -0
- package/dist/rag/pipeline.d.ts.map +1 -0
- package/dist/rag/pipeline.js +75 -0
- package/dist/rag/pipeline.js.map +1 -0
- package/dist/rag/retrieval/dense-retriever.d.ts +17 -0
- package/dist/rag/retrieval/dense-retriever.d.ts.map +1 -0
- package/dist/rag/retrieval/dense-retriever.js +21 -0
- package/dist/rag/retrieval/dense-retriever.js.map +1 -0
- package/dist/rag/retrieval/fusion.d.ts +12 -0
- package/dist/rag/retrieval/fusion.d.ts.map +1 -0
- package/dist/rag/retrieval/fusion.js +54 -0
- package/dist/rag/retrieval/fusion.js.map +1 -0
- package/dist/rag/retrieval/http-rerank-model-client.d.ts +41 -0
- package/dist/rag/retrieval/http-rerank-model-client.d.ts.map +1 -0
- package/dist/rag/retrieval/http-rerank-model-client.js +130 -0
- package/dist/rag/retrieval/http-rerank-model-client.js.map +1 -0
- package/dist/rag/retrieval/hybrid-retriever.d.ts +22 -0
- package/dist/rag/retrieval/hybrid-retriever.d.ts.map +1 -0
- package/dist/rag/retrieval/hybrid-retriever.js +49 -0
- package/dist/rag/retrieval/hybrid-retriever.js.map +1 -0
- package/dist/rag/retrieval/index.d.ts +8 -0
- package/dist/rag/retrieval/index.d.ts.map +1 -0
- package/dist/rag/retrieval/index.js +24 -0
- package/dist/rag/retrieval/index.js.map +1 -0
- package/dist/rag/retrieval/reranker.d.ts +30 -0
- package/dist/rag/retrieval/reranker.d.ts.map +1 -0
- package/dist/rag/retrieval/reranker.js +85 -0
- package/dist/rag/retrieval/reranker.js.map +1 -0
- package/dist/rag/retrieval/sparse-retriever.d.ts +14 -0
- package/dist/rag/retrieval/sparse-retriever.d.ts.map +1 -0
- package/dist/rag/retrieval/sparse-retriever.js +18 -0
- package/dist/rag/retrieval/sparse-retriever.js.map +1 -0
- package/dist/rag/retrieval/types.d.ts +11 -0
- package/dist/rag/retrieval/types.d.ts.map +1 -0
- package/dist/rag/retrieval/types.js +10 -0
- package/dist/rag/retrieval/types.js.map +1 -0
- package/dist/rag/retrieval/utils.d.ts +8 -0
- package/dist/rag/retrieval/utils.d.ts.map +1 -0
- package/dist/rag/retrieval/utils.js +114 -0
- package/dist/rag/retrieval/utils.js.map +1 -0
- package/dist/rag/stores/in-memory-document-store.d.ts +14 -0
- package/dist/rag/stores/in-memory-document-store.d.ts.map +1 -0
- package/dist/rag/stores/in-memory-document-store.js +64 -0
- package/dist/rag/stores/in-memory-document-store.js.map +1 -0
- package/dist/rag/stores/in-memory-keyword-store.d.ts +10 -0
- package/dist/rag/stores/in-memory-keyword-store.d.ts.map +1 -0
- package/dist/rag/stores/in-memory-keyword-store.js +99 -0
- package/dist/rag/stores/in-memory-keyword-store.js.map +1 -0
- package/dist/rag/stores/in-memory-vector-store.d.ts +14 -0
- package/dist/rag/stores/in-memory-vector-store.d.ts.map +1 -0
- package/dist/rag/stores/in-memory-vector-store.js +61 -0
- package/dist/rag/stores/in-memory-vector-store.js.map +1 -0
- package/dist/rag/stores/index.d.ts +6 -0
- package/dist/rag/stores/index.d.ts.map +1 -0
- package/dist/rag/stores/index.js +22 -0
- package/dist/rag/stores/index.js.map +1 -0
- package/dist/rag/stores/postgres/index.d.ts +6 -0
- package/dist/rag/stores/postgres/index.d.ts.map +1 -0
- package/dist/rag/stores/postgres/index.js +22 -0
- package/dist/rag/stores/postgres/index.js.map +1 -0
- package/dist/rag/stores/postgres/pg-vector-store.d.ts +16 -0
- package/dist/rag/stores/postgres/pg-vector-store.d.ts.map +1 -0
- package/dist/rag/stores/postgres/pg-vector-store.js +77 -0
- package/dist/rag/stores/postgres/pg-vector-store.js.map +1 -0
- package/dist/rag/stores/postgres/postgres-document-store.d.ts +15 -0
- package/dist/rag/stores/postgres/postgres-document-store.d.ts.map +1 -0
- package/dist/rag/stores/postgres/postgres-document-store.js +174 -0
- package/dist/rag/stores/postgres/postgres-document-store.js.map +1 -0
- package/dist/rag/stores/postgres/postgres-keyword-store.d.ts +15 -0
- package/dist/rag/stores/postgres/postgres-keyword-store.d.ts.map +1 -0
- package/dist/rag/stores/postgres/postgres-keyword-store.js +163 -0
- package/dist/rag/stores/postgres/postgres-keyword-store.js.map +1 -0
- package/dist/rag/stores/postgres/schema.d.ts +8 -0
- package/dist/rag/stores/postgres/schema.d.ts.map +1 -0
- package/dist/rag/stores/postgres/schema.js +91 -0
- package/dist/rag/stores/postgres/schema.js.map +1 -0
- package/dist/rag/stores/postgres/sql.d.ts +23 -0
- package/dist/rag/stores/postgres/sql.d.ts.map +1 -0
- package/dist/rag/stores/postgres/sql.js +177 -0
- package/dist/rag/stores/postgres/sql.js.map +1 -0
- package/dist/rag/stores/postgres/types.d.ts +17 -0
- package/dist/rag/stores/postgres/types.d.ts.map +1 -0
- package/dist/rag/stores/postgres/types.js +3 -0
- package/dist/rag/stores/postgres/types.js.map +1 -0
- package/dist/rag/stores/types.d.ts +36 -0
- package/dist/rag/stores/types.d.ts.map +1 -0
- package/dist/rag/stores/types.js +3 -0
- package/dist/rag/stores/types.js.map +1 -0
- package/dist/rag/stores/utils.d.ts +11 -0
- package/dist/rag/stores/utils.d.ts.map +1 -0
- package/dist/rag/stores/utils.js +200 -0
- package/dist/rag/stores/utils.js.map +1 -0
- package/dist/rag/types.d.ts +174 -0
- package/dist/rag/types.d.ts.map +1 -0
- package/dist/rag/types.js +3 -0
- package/dist/rag/types.js.map +1 -0
- package/dist/tools/base.d.ts +28 -0
- package/dist/tools/base.d.ts.map +1 -0
- package/dist/tools/base.js +68 -0
- package/dist/tools/base.js.map +1 -0
- package/dist/tools/builtin/advancedSearchTool.d.ts +24 -0
- package/dist/tools/builtin/advancedSearchTool.d.ts.map +1 -0
- package/dist/tools/builtin/advancedSearchTool.js +134 -0
- package/dist/tools/builtin/advancedSearchTool.js.map +1 -0
- package/dist/tools/builtin/ragSearchTool.d.ts +29 -0
- package/dist/tools/builtin/ragSearchTool.d.ts.map +1 -0
- package/dist/tools/builtin/ragSearchTool.js +91 -0
- package/dist/tools/builtin/ragSearchTool.js.map +1 -0
- package/dist/tools/executor.d.ts +10 -0
- package/dist/tools/executor.d.ts.map +1 -0
- package/dist/tools/executor.js +86 -0
- package/dist/tools/executor.js.map +1 -0
- package/dist/tools/index.d.ts +9 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +16 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/registry.d.ts +16 -0
- package/dist/tools/registry.d.ts.map +1 -0
- package/dist/tools/registry.js +35 -0
- package/dist/tools/registry.js.map +1 -0
- 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"}
|