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