@hazeljs/rag 0.2.0-alpha.1
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/LICENSE +192 -0
- package/README.md +504 -0
- package/dist/__tests__/graph/community-detector.test.d.ts +2 -0
- package/dist/__tests__/graph/community-detector.test.d.ts.map +1 -0
- package/dist/__tests__/graph/community-detector.test.js +87 -0
- package/dist/__tests__/graph/community-detector.test.js.map +1 -0
- package/dist/__tests__/graph/community-summarizer.test.d.ts +2 -0
- package/dist/__tests__/graph/community-summarizer.test.d.ts.map +1 -0
- package/dist/__tests__/graph/community-summarizer.test.js +131 -0
- package/dist/__tests__/graph/community-summarizer.test.js.map +1 -0
- package/dist/__tests__/graph/entity-extractor.test.d.ts +2 -0
- package/dist/__tests__/graph/entity-extractor.test.d.ts.map +1 -0
- package/dist/__tests__/graph/entity-extractor.test.js +129 -0
- package/dist/__tests__/graph/entity-extractor.test.js.map +1 -0
- package/dist/__tests__/graph/graph-rag-pipeline.test.d.ts +2 -0
- package/dist/__tests__/graph/graph-rag-pipeline.test.d.ts.map +1 -0
- package/dist/__tests__/graph/graph-rag-pipeline.test.js +158 -0
- package/dist/__tests__/graph/graph-rag-pipeline.test.js.map +1 -0
- package/dist/__tests__/graph/knowledge-graph.test.d.ts +2 -0
- package/dist/__tests__/graph/knowledge-graph.test.d.ts.map +1 -0
- package/dist/__tests__/graph/knowledge-graph.test.js +208 -0
- package/dist/__tests__/graph/knowledge-graph.test.js.map +1 -0
- package/dist/__tests__/loaders/base.loader.test.d.ts +2 -0
- package/dist/__tests__/loaders/base.loader.test.d.ts.map +1 -0
- package/dist/__tests__/loaders/base.loader.test.js +114 -0
- package/dist/__tests__/loaders/base.loader.test.js.map +1 -0
- package/dist/__tests__/loaders/csv-file.loader.test.d.ts +2 -0
- package/dist/__tests__/loaders/csv-file.loader.test.d.ts.map +1 -0
- package/dist/__tests__/loaders/csv-file.loader.test.js +98 -0
- package/dist/__tests__/loaders/csv-file.loader.test.js.map +1 -0
- package/dist/__tests__/loaders/directory.loader.test.d.ts +2 -0
- package/dist/__tests__/loaders/directory.loader.test.d.ts.map +1 -0
- package/dist/__tests__/loaders/directory.loader.test.js +154 -0
- package/dist/__tests__/loaders/directory.loader.test.js.map +1 -0
- package/dist/__tests__/loaders/html-file.loader.test.d.ts +2 -0
- package/dist/__tests__/loaders/html-file.loader.test.d.ts.map +1 -0
- package/dist/__tests__/loaders/html-file.loader.test.js +93 -0
- package/dist/__tests__/loaders/html-file.loader.test.js.map +1 -0
- package/dist/__tests__/loaders/json-file.loader.test.d.ts +2 -0
- package/dist/__tests__/loaders/json-file.loader.test.d.ts.map +1 -0
- package/dist/__tests__/loaders/json-file.loader.test.js +84 -0
- package/dist/__tests__/loaders/json-file.loader.test.js.map +1 -0
- package/dist/__tests__/loaders/markdown-file.loader.test.d.ts +2 -0
- package/dist/__tests__/loaders/markdown-file.loader.test.d.ts.map +1 -0
- package/dist/__tests__/loaders/markdown-file.loader.test.js +83 -0
- package/dist/__tests__/loaders/markdown-file.loader.test.js.map +1 -0
- package/dist/__tests__/loaders/text-file.loader.test.d.ts +2 -0
- package/dist/__tests__/loaders/text-file.loader.test.d.ts.map +1 -0
- package/dist/__tests__/loaders/text-file.loader.test.js +50 -0
- package/dist/__tests__/loaders/text-file.loader.test.js.map +1 -0
- package/dist/__tests__/rag-pipeline.test.d.ts +2 -0
- package/dist/__tests__/rag-pipeline.test.d.ts.map +1 -0
- package/dist/__tests__/rag-pipeline.test.js +210 -0
- package/dist/__tests__/rag-pipeline.test.js.map +1 -0
- package/dist/__tests__/retrieval/bm25.test.d.ts +2 -0
- package/dist/__tests__/retrieval/bm25.test.d.ts.map +1 -0
- package/dist/__tests__/retrieval/bm25.test.js +86 -0
- package/dist/__tests__/retrieval/bm25.test.js.map +1 -0
- package/dist/__tests__/retrieval/hybrid-search.test.d.ts +2 -0
- package/dist/__tests__/retrieval/hybrid-search.test.d.ts.map +1 -0
- package/dist/__tests__/retrieval/hybrid-search.test.js +85 -0
- package/dist/__tests__/retrieval/hybrid-search.test.js.map +1 -0
- package/dist/__tests__/retrieval/multi-query.test.d.ts +2 -0
- package/dist/__tests__/retrieval/multi-query.test.d.ts.map +1 -0
- package/dist/__tests__/retrieval/multi-query.test.js +90 -0
- package/dist/__tests__/retrieval/multi-query.test.js.map +1 -0
- package/dist/__tests__/text-splitters/recursive-text-splitter.test.d.ts +2 -0
- package/dist/__tests__/text-splitters/recursive-text-splitter.test.d.ts.map +1 -0
- package/dist/__tests__/text-splitters/recursive-text-splitter.test.js +97 -0
- package/dist/__tests__/text-splitters/recursive-text-splitter.test.js.map +1 -0
- package/dist/__tests__/utils/similarity.test.d.ts +2 -0
- package/dist/__tests__/utils/similarity.test.d.ts.map +1 -0
- package/dist/__tests__/utils/similarity.test.js +47 -0
- package/dist/__tests__/utils/similarity.test.js.map +1 -0
- package/dist/agentic/agentic-rag.service.d.ts +49 -0
- package/dist/agentic/agentic-rag.service.d.ts.map +1 -0
- package/dist/agentic/agentic-rag.service.js +149 -0
- package/dist/agentic/agentic-rag.service.js.map +1 -0
- package/dist/agentic/decorators/active-learning.decorator.d.ts +19 -0
- package/dist/agentic/decorators/active-learning.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/active-learning.decorator.js +98 -0
- package/dist/agentic/decorators/active-learning.decorator.js.map +1 -0
- package/dist/agentic/decorators/adaptive-retrieval.decorator.d.ts +17 -0
- package/dist/agentic/decorators/adaptive-retrieval.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/adaptive-retrieval.decorator.js +103 -0
- package/dist/agentic/decorators/adaptive-retrieval.decorator.js.map +1 -0
- package/dist/agentic/decorators/cached.decorator.d.ts +18 -0
- package/dist/agentic/decorators/cached.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/cached.decorator.js +93 -0
- package/dist/agentic/decorators/cached.decorator.js.map +1 -0
- package/dist/agentic/decorators/context-aware.decorator.d.ts +16 -0
- package/dist/agentic/decorators/context-aware.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/context-aware.decorator.js +169 -0
- package/dist/agentic/decorators/context-aware.decorator.js.map +1 -0
- package/dist/agentic/decorators/corrective-rag.decorator.d.ts +16 -0
- package/dist/agentic/decorators/corrective-rag.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/corrective-rag.decorator.js +142 -0
- package/dist/agentic/decorators/corrective-rag.decorator.js.map +1 -0
- package/dist/agentic/decorators/hyde.decorator.d.ts +15 -0
- package/dist/agentic/decorators/hyde.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/hyde.decorator.js +91 -0
- package/dist/agentic/decorators/hyde.decorator.js.map +1 -0
- package/dist/agentic/decorators/index.d.ts +16 -0
- package/dist/agentic/decorators/index.d.ts.map +1 -0
- package/dist/agentic/decorators/index.js +32 -0
- package/dist/agentic/decorators/index.js.map +1 -0
- package/dist/agentic/decorators/multi-hop.decorator.d.ts +15 -0
- package/dist/agentic/decorators/multi-hop.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/multi-hop.decorator.js +109 -0
- package/dist/agentic/decorators/multi-hop.decorator.js.map +1 -0
- package/dist/agentic/decorators/query-planner.decorator.d.ts +20 -0
- package/dist/agentic/decorators/query-planner.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/query-planner.decorator.js +213 -0
- package/dist/agentic/decorators/query-planner.decorator.js.map +1 -0
- package/dist/agentic/decorators/query-rewriter.decorator.d.ts +16 -0
- package/dist/agentic/decorators/query-rewriter.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/query-rewriter.decorator.js +143 -0
- package/dist/agentic/decorators/query-rewriter.decorator.js.map +1 -0
- package/dist/agentic/decorators/self-reflective.decorator.d.ts +20 -0
- package/dist/agentic/decorators/self-reflective.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/self-reflective.decorator.js +189 -0
- package/dist/agentic/decorators/self-reflective.decorator.js.map +1 -0
- package/dist/agentic/decorators/source-verification.decorator.d.ts +15 -0
- package/dist/agentic/decorators/source-verification.decorator.d.ts.map +1 -0
- package/dist/agentic/decorators/source-verification.decorator.js +121 -0
- package/dist/agentic/decorators/source-verification.decorator.js.map +1 -0
- package/dist/agentic/index.d.ts +9 -0
- package/dist/agentic/index.d.ts.map +1 -0
- package/dist/agentic/index.js +25 -0
- package/dist/agentic/index.js.map +1 -0
- package/dist/agentic/types.d.ts +210 -0
- package/dist/agentic/types.d.ts.map +1 -0
- package/dist/agentic/types.js +7 -0
- package/dist/agentic/types.js.map +1 -0
- package/dist/decorators/embeddable.decorator.d.ts +31 -0
- package/dist/decorators/embeddable.decorator.d.ts.map +1 -0
- package/dist/decorators/embeddable.decorator.js +44 -0
- package/dist/decorators/embeddable.decorator.js.map +1 -0
- package/dist/decorators/rag.decorator.d.ts +58 -0
- package/dist/decorators/rag.decorator.d.ts.map +1 -0
- package/dist/decorators/rag.decorator.js +78 -0
- package/dist/decorators/rag.decorator.js.map +1 -0
- package/dist/decorators/semantic-search.decorator.d.ts +69 -0
- package/dist/decorators/semantic-search.decorator.d.ts.map +1 -0
- package/dist/decorators/semantic-search.decorator.js +116 -0
- package/dist/decorators/semantic-search.decorator.js.map +1 -0
- package/dist/embeddings/cohere-embeddings.d.ts +33 -0
- package/dist/embeddings/cohere-embeddings.d.ts.map +1 -0
- package/dist/embeddings/cohere-embeddings.js +91 -0
- package/dist/embeddings/cohere-embeddings.js.map +1 -0
- package/dist/embeddings/openai-embeddings.d.ts +21 -0
- package/dist/embeddings/openai-embeddings.d.ts.map +1 -0
- package/dist/embeddings/openai-embeddings.js +53 -0
- package/dist/embeddings/openai-embeddings.js.map +1 -0
- package/dist/graph/community-detector.d.ts +45 -0
- package/dist/graph/community-detector.d.ts.map +1 -0
- package/dist/graph/community-detector.js +153 -0
- package/dist/graph/community-detector.js.map +1 -0
- package/dist/graph/community-summarizer.d.ts +41 -0
- package/dist/graph/community-summarizer.d.ts.map +1 -0
- package/dist/graph/community-summarizer.js +119 -0
- package/dist/graph/community-summarizer.js.map +1 -0
- package/dist/graph/entity-extractor.d.ts +47 -0
- package/dist/graph/entity-extractor.d.ts.map +1 -0
- package/dist/graph/entity-extractor.js +224 -0
- package/dist/graph/entity-extractor.js.map +1 -0
- package/dist/graph/graph-rag-pipeline.d.ts +83 -0
- package/dist/graph/graph-rag-pipeline.d.ts.map +1 -0
- package/dist/graph/graph-rag-pipeline.js +390 -0
- package/dist/graph/graph-rag-pipeline.js.map +1 -0
- package/dist/graph/graph.types.d.ts +186 -0
- package/dist/graph/graph.types.d.ts.map +1 -0
- package/dist/graph/graph.types.js +20 -0
- package/dist/graph/graph.types.js.map +1 -0
- package/dist/graph/index.d.ts +15 -0
- package/dist/graph/index.d.ts.map +1 -0
- package/dist/graph/index.js +31 -0
- package/dist/graph/index.js.map +1 -0
- package/dist/graph/knowledge-graph.d.ts +57 -0
- package/dist/graph/knowledge-graph.d.ts.map +1 -0
- package/dist/graph/knowledge-graph.js +198 -0
- package/dist/graph/knowledge-graph.js.map +1 -0
- package/dist/index.d.ts +29 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +58 -0
- package/dist/index.js.map +1 -0
- package/dist/loaders/base.loader.d.ts +108 -0
- package/dist/loaders/base.loader.d.ts.map +1 -0
- package/dist/loaders/base.loader.js +123 -0
- package/dist/loaders/base.loader.js.map +1 -0
- package/dist/loaders/csv-file.loader.d.ts +61 -0
- package/dist/loaders/csv-file.loader.d.ts.map +1 -0
- package/dist/loaders/csv-file.loader.js +162 -0
- package/dist/loaders/csv-file.loader.js.map +1 -0
- package/dist/loaders/directory.loader.d.ts +67 -0
- package/dist/loaders/directory.loader.d.ts.map +1 -0
- package/dist/loaders/directory.loader.js +163 -0
- package/dist/loaders/directory.loader.js.map +1 -0
- package/dist/loaders/docx.loader.d.ts +52 -0
- package/dist/loaders/docx.loader.d.ts.map +1 -0
- package/dist/loaders/docx.loader.js +110 -0
- package/dist/loaders/docx.loader.js.map +1 -0
- package/dist/loaders/github.loader.d.ts +114 -0
- package/dist/loaders/github.loader.d.ts.map +1 -0
- package/dist/loaders/github.loader.js +217 -0
- package/dist/loaders/github.loader.js.map +1 -0
- package/dist/loaders/html-file.loader.d.ts +55 -0
- package/dist/loaders/html-file.loader.d.ts.map +1 -0
- package/dist/loaders/html-file.loader.js +170 -0
- package/dist/loaders/html-file.loader.js.map +1 -0
- package/dist/loaders/index.d.ts +52 -0
- package/dist/loaders/index.d.ts.map +1 -0
- package/dist/loaders/index.js +61 -0
- package/dist/loaders/index.js.map +1 -0
- package/dist/loaders/json-file.loader.d.ts +51 -0
- package/dist/loaders/json-file.loader.d.ts.map +1 -0
- package/dist/loaders/json-file.loader.js +100 -0
- package/dist/loaders/json-file.loader.js.map +1 -0
- package/dist/loaders/markdown-file.loader.d.ts +61 -0
- package/dist/loaders/markdown-file.loader.d.ts.map +1 -0
- package/dist/loaders/markdown-file.loader.js +148 -0
- package/dist/loaders/markdown-file.loader.js.map +1 -0
- package/dist/loaders/pdf.loader.d.ts +64 -0
- package/dist/loaders/pdf.loader.d.ts.map +1 -0
- package/dist/loaders/pdf.loader.js +163 -0
- package/dist/loaders/pdf.loader.js.map +1 -0
- package/dist/loaders/text-file.loader.d.ts +39 -0
- package/dist/loaders/text-file.loader.d.ts.map +1 -0
- package/dist/loaders/text-file.loader.js +69 -0
- package/dist/loaders/text-file.loader.js.map +1 -0
- package/dist/loaders/web.loader.d.ts +87 -0
- package/dist/loaders/web.loader.d.ts.map +1 -0
- package/dist/loaders/web.loader.js +194 -0
- package/dist/loaders/web.loader.js.map +1 -0
- package/dist/loaders/youtube-transcript.loader.d.ts +92 -0
- package/dist/loaders/youtube-transcript.loader.d.ts.map +1 -0
- package/dist/loaders/youtube-transcript.loader.js +254 -0
- package/dist/loaders/youtube-transcript.loader.js.map +1 -0
- package/dist/memory/index.d.ts +11 -0
- package/dist/memory/index.d.ts.map +1 -0
- package/dist/memory/index.js +31 -0
- package/dist/memory/index.js.map +1 -0
- package/dist/memory/memory-manager.d.ts +96 -0
- package/dist/memory/memory-manager.d.ts.map +1 -0
- package/dist/memory/memory-manager.js +369 -0
- package/dist/memory/memory-manager.js.map +1 -0
- package/dist/memory/memory-store.interface.d.ts +73 -0
- package/dist/memory/memory-store.interface.d.ts.map +1 -0
- package/dist/memory/memory-store.interface.js +6 -0
- package/dist/memory/memory-store.interface.js.map +1 -0
- package/dist/memory/stores/buffer-memory.d.ts +47 -0
- package/dist/memory/stores/buffer-memory.d.ts.map +1 -0
- package/dist/memory/stores/buffer-memory.js +280 -0
- package/dist/memory/stores/buffer-memory.js.map +1 -0
- package/dist/memory/stores/hybrid-memory.d.ts +49 -0
- package/dist/memory/stores/hybrid-memory.d.ts.map +1 -0
- package/dist/memory/stores/hybrid-memory.js +194 -0
- package/dist/memory/stores/hybrid-memory.js.map +1 -0
- package/dist/memory/stores/vector-memory.d.ts +48 -0
- package/dist/memory/stores/vector-memory.d.ts.map +1 -0
- package/dist/memory/stores/vector-memory.js +312 -0
- package/dist/memory/stores/vector-memory.js.map +1 -0
- package/dist/memory/types.d.ts +119 -0
- package/dist/memory/types.d.ts.map +1 -0
- package/dist/memory/types.js +18 -0
- package/dist/memory/types.js.map +1 -0
- package/dist/prompts/agentic/adaptive-retrieval.prompt.d.ts +8 -0
- package/dist/prompts/agentic/adaptive-retrieval.prompt.d.ts.map +1 -0
- package/dist/prompts/agentic/adaptive-retrieval.prompt.js +27 -0
- package/dist/prompts/agentic/adaptive-retrieval.prompt.js.map +1 -0
- package/dist/prompts/agentic/corrective-rag.prompt.d.ts +9 -0
- package/dist/prompts/agentic/corrective-rag.prompt.d.ts.map +1 -0
- package/dist/prompts/agentic/corrective-rag.prompt.js +23 -0
- package/dist/prompts/agentic/corrective-rag.prompt.js.map +1 -0
- package/dist/prompts/agentic/hyde.prompt.d.ts +9 -0
- package/dist/prompts/agentic/hyde.prompt.d.ts.map +1 -0
- package/dist/prompts/agentic/hyde.prompt.js +18 -0
- package/dist/prompts/agentic/hyde.prompt.js.map +1 -0
- package/dist/prompts/agentic/multi-hop.prompt.d.ts +15 -0
- package/dist/prompts/agentic/multi-hop.prompt.d.ts.map +1 -0
- package/dist/prompts/agentic/multi-hop.prompt.js +38 -0
- package/dist/prompts/agentic/multi-hop.prompt.js.map +1 -0
- package/dist/prompts/agentic/query-planner.prompt.d.ts +8 -0
- package/dist/prompts/agentic/query-planner.prompt.d.ts.map +1 -0
- package/dist/prompts/agentic/query-planner.prompt.js +30 -0
- package/dist/prompts/agentic/query-planner.prompt.js.map +1 -0
- package/dist/prompts/agentic/query-rewriter.prompt.d.ts +10 -0
- package/dist/prompts/agentic/query-rewriter.prompt.d.ts.map +1 -0
- package/dist/prompts/agentic/query-rewriter.prompt.js +17 -0
- package/dist/prompts/agentic/query-rewriter.prompt.js.map +1 -0
- package/dist/prompts/agentic/self-reflective-improve.prompt.d.ts +10 -0
- package/dist/prompts/agentic/self-reflective-improve.prompt.d.ts.map +1 -0
- package/dist/prompts/agentic/self-reflective-improve.prompt.js +24 -0
- package/dist/prompts/agentic/self-reflective-improve.prompt.js.map +1 -0
- package/dist/prompts/agentic/self-reflective.prompt.d.ts +9 -0
- package/dist/prompts/agentic/self-reflective.prompt.d.ts.map +1 -0
- package/dist/prompts/agentic/self-reflective.prompt.js +32 -0
- package/dist/prompts/agentic/self-reflective.prompt.js.map +1 -0
- package/dist/prompts/community-summary.prompt.d.ts +9 -0
- package/dist/prompts/community-summary.prompt.d.ts.map +1 -0
- package/dist/prompts/community-summary.prompt.js +30 -0
- package/dist/prompts/community-summary.prompt.js.map +1 -0
- package/dist/prompts/entity-extraction.prompt.d.ts +10 -0
- package/dist/prompts/entity-extraction.prompt.d.ts.map +1 -0
- package/dist/prompts/entity-extraction.prompt.js +39 -0
- package/dist/prompts/entity-extraction.prompt.js.map +1 -0
- package/dist/prompts/graph-search.prompt.d.ts +10 -0
- package/dist/prompts/graph-search.prompt.d.ts.map +1 -0
- package/dist/prompts/graph-search.prompt.js +23 -0
- package/dist/prompts/graph-search.prompt.js.map +1 -0
- package/dist/prompts/index.d.ts +13 -0
- package/dist/prompts/index.d.ts.map +1 -0
- package/dist/prompts/index.js +29 -0
- package/dist/prompts/index.js.map +1 -0
- package/dist/prompts/rag-answer.prompt.d.ts +9 -0
- package/dist/prompts/rag-answer.prompt.d.ts.map +1 -0
- package/dist/prompts/rag-answer.prompt.js +20 -0
- package/dist/prompts/rag-answer.prompt.js.map +1 -0
- package/dist/rag-pipeline-with-memory.d.ts +68 -0
- package/dist/rag-pipeline-with-memory.d.ts.map +1 -0
- package/dist/rag-pipeline-with-memory.js +186 -0
- package/dist/rag-pipeline-with-memory.js.map +1 -0
- package/dist/rag-pipeline.d.ts +59 -0
- package/dist/rag-pipeline.d.ts.map +1 -0
- package/dist/rag-pipeline.js +181 -0
- package/dist/rag-pipeline.js.map +1 -0
- package/dist/rag.module.d.ts +26 -0
- package/dist/rag.module.d.ts.map +1 -0
- package/dist/rag.module.js +40 -0
- package/dist/rag.module.js.map +1 -0
- package/dist/rag.service.d.ts +96 -0
- package/dist/rag.service.d.ts.map +1 -0
- package/dist/rag.service.js +173 -0
- package/dist/rag.service.js.map +1 -0
- package/dist/retrieval/bm25.d.ts +57 -0
- package/dist/retrieval/bm25.d.ts.map +1 -0
- package/dist/retrieval/bm25.js +106 -0
- package/dist/retrieval/bm25.js.map +1 -0
- package/dist/retrieval/hybrid-search.d.ts +48 -0
- package/dist/retrieval/hybrid-search.d.ts.map +1 -0
- package/dist/retrieval/hybrid-search.js +123 -0
- package/dist/retrieval/hybrid-search.js.map +1 -0
- package/dist/retrieval/multi-query.d.ts +38 -0
- package/dist/retrieval/multi-query.d.ts.map +1 -0
- package/dist/retrieval/multi-query.js +135 -0
- package/dist/retrieval/multi-query.js.map +1 -0
- package/dist/text-splitters/recursive-text-splitter.d.ts +21 -0
- package/dist/text-splitters/recursive-text-splitter.d.ts.map +1 -0
- package/dist/text-splitters/recursive-text-splitter.js +95 -0
- package/dist/text-splitters/recursive-text-splitter.js.map +1 -0
- package/dist/types/index.d.ts +144 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +16 -0
- package/dist/types/index.js.map +1 -0
- package/dist/utils/similarity.d.ts +16 -0
- package/dist/utils/similarity.d.ts.map +1 -0
- package/dist/utils/similarity.js +58 -0
- package/dist/utils/similarity.js.map +1 -0
- package/dist/vector-stores/chroma.store.d.ts +42 -0
- package/dist/vector-stores/chroma.store.d.ts.map +1 -0
- package/dist/vector-stores/chroma.store.js +242 -0
- package/dist/vector-stores/chroma.store.js.map +1 -0
- package/dist/vector-stores/memory-vector-store.d.ts +20 -0
- package/dist/vector-stores/memory-vector-store.d.ts.map +1 -0
- package/dist/vector-stores/memory-vector-store.js +94 -0
- package/dist/vector-stores/memory-vector-store.js.map +1 -0
- package/dist/vector-stores/pinecone.store.d.ts +34 -0
- package/dist/vector-stores/pinecone.store.d.ts.map +1 -0
- package/dist/vector-stores/pinecone.store.js +146 -0
- package/dist/vector-stores/pinecone.store.js.map +1 -0
- package/dist/vector-stores/qdrant.store.d.ts +33 -0
- package/dist/vector-stores/qdrant.store.d.ts.map +1 -0
- package/dist/vector-stores/qdrant.store.js +174 -0
- package/dist/vector-stores/qdrant.store.js.map +1 -0
- package/dist/vector-stores/weaviate.store.d.ts +37 -0
- package/dist/vector-stores/weaviate.store.d.ts.map +1 -0
- package/dist/vector-stores/weaviate.store.js +226 -0
- package/dist/vector-stores/weaviate.store.js.map +1 -0
- package/package.json +146 -0
|
@@ -0,0 +1,109 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Multi-Hop Decorator
|
|
4
|
+
* Performs multi-hop reasoning across documents
|
|
5
|
+
*/
|
|
6
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
7
|
+
exports.MultiHop = MultiHop;
|
|
8
|
+
exports.getReasoningChain = getReasoningChain;
|
|
9
|
+
require("reflect-metadata");
|
|
10
|
+
const prompts_1 = require("@hazeljs/prompts");
|
|
11
|
+
require("../../prompts/agentic/multi-hop.prompt");
|
|
12
|
+
const multi_hop_prompt_1 = require("../../prompts/agentic/multi-hop.prompt");
|
|
13
|
+
const MULTIHOP_METADATA_KEY = Symbol('multiHop');
|
|
14
|
+
function MultiHop(config = {}) {
|
|
15
|
+
return function (target, propertyKey, descriptor) {
|
|
16
|
+
const originalMethod = descriptor.value;
|
|
17
|
+
descriptor.value = async function (...args) {
|
|
18
|
+
const query = args[0];
|
|
19
|
+
const maxHops = config.maxHops || 3;
|
|
20
|
+
const chain = {
|
|
21
|
+
query,
|
|
22
|
+
hops: [],
|
|
23
|
+
finalAnswer: '',
|
|
24
|
+
confidence: 0,
|
|
25
|
+
sources: [],
|
|
26
|
+
};
|
|
27
|
+
let currentQuery = query;
|
|
28
|
+
for (let hop = 0; hop < maxHops; hop++) {
|
|
29
|
+
// Retrieve for current query
|
|
30
|
+
const results = await originalMethod.apply(this, [currentQuery, ...args.slice(1)]);
|
|
31
|
+
// Generate reasoning
|
|
32
|
+
const reasoning = await generateReasoning(currentQuery, results, config);
|
|
33
|
+
const hopResult = {
|
|
34
|
+
hopNumber: hop + 1,
|
|
35
|
+
query: currentQuery,
|
|
36
|
+
results: Array.isArray(results) ? results : [results],
|
|
37
|
+
reasoning: reasoning.reasoning,
|
|
38
|
+
nextQuery: reasoning.nextQuery,
|
|
39
|
+
};
|
|
40
|
+
chain.hops.push(hopResult);
|
|
41
|
+
chain.sources.push(...hopResult.results);
|
|
42
|
+
// Check if we should continue
|
|
43
|
+
if (!reasoning.nextQuery || reasoning.shouldStop) {
|
|
44
|
+
break;
|
|
45
|
+
}
|
|
46
|
+
currentQuery = reasoning.nextQuery;
|
|
47
|
+
}
|
|
48
|
+
// Generate final answer
|
|
49
|
+
chain.finalAnswer = await synthesizeFinalAnswer(chain, config);
|
|
50
|
+
chain.confidence = calculateConfidence(chain);
|
|
51
|
+
Reflect.defineMetadata(MULTIHOP_METADATA_KEY, chain, target, propertyKey);
|
|
52
|
+
return chain;
|
|
53
|
+
};
|
|
54
|
+
return descriptor;
|
|
55
|
+
};
|
|
56
|
+
}
|
|
57
|
+
async function generateReasoning(query,
|
|
58
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
59
|
+
results, config) {
|
|
60
|
+
if (!config.llmProvider) {
|
|
61
|
+
return {
|
|
62
|
+
reasoning: 'Retrieved results for: ' + query,
|
|
63
|
+
shouldStop: true,
|
|
64
|
+
};
|
|
65
|
+
}
|
|
66
|
+
const resultsStr = JSON.stringify(results, null, 2).slice(0, 1500);
|
|
67
|
+
const prompt = prompts_1.PromptRegistry.get(multi_hop_prompt_1.MULTI_HOP_KEY).render({
|
|
68
|
+
query,
|
|
69
|
+
resultsStr,
|
|
70
|
+
});
|
|
71
|
+
try {
|
|
72
|
+
return await config.llmProvider.generateStructured(prompt, {});
|
|
73
|
+
}
|
|
74
|
+
catch {
|
|
75
|
+
return {
|
|
76
|
+
reasoning: 'Retrieved results',
|
|
77
|
+
shouldStop: true,
|
|
78
|
+
};
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
async function synthesizeFinalAnswer(chain, config) {
|
|
82
|
+
if (!config.llmProvider) {
|
|
83
|
+
return chain.hops.map((h) => h.reasoning).join('\n\n');
|
|
84
|
+
}
|
|
85
|
+
const hopsStr = chain.hops.map((h) => `Hop ${h.hopNumber}: ${h.reasoning}`).join('\n');
|
|
86
|
+
const prompt = prompts_1.PromptRegistry.get(multi_hop_prompt_1.MULTI_HOP_SYNTHESIZE_KEY).render({ originalQuery: chain.query, hopsStr });
|
|
87
|
+
try {
|
|
88
|
+
return await config.llmProvider.generate(prompt);
|
|
89
|
+
}
|
|
90
|
+
catch {
|
|
91
|
+
return hopsStr;
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
function calculateConfidence(chain) {
|
|
95
|
+
const hopCount = chain.hops.length;
|
|
96
|
+
const sourceCount = chain.sources.length;
|
|
97
|
+
let confidence = 0.5;
|
|
98
|
+
if (sourceCount >= 5)
|
|
99
|
+
confidence += 0.2;
|
|
100
|
+
if (hopCount >= 2)
|
|
101
|
+
confidence += 0.1;
|
|
102
|
+
if (chain.finalAnswer.length > 100)
|
|
103
|
+
confidence += 0.1;
|
|
104
|
+
return Math.min(confidence, 1.0);
|
|
105
|
+
}
|
|
106
|
+
function getReasoningChain(target, propertyKey) {
|
|
107
|
+
return Reflect.getMetadata(MULTIHOP_METADATA_KEY, target, propertyKey);
|
|
108
|
+
}
|
|
109
|
+
//# sourceMappingURL=multi-hop.decorator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"multi-hop.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/multi-hop.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAgBH,4BAuDC;AAsED,8CAKC;AAhJD,4BAA0B;AAE1B,8CAAkD;AAClD,kDAAgD;AAChD,6EAAiG;AAQjG,MAAM,qBAAqB,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;AAEjD,SAAgB,QAAQ,CAAC,SAAyB,EAAE;IAClD,OAAO,UAAU,MAAc,EAAE,WAA4B,EAAE,UAA8B;QAC3F,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,CAAC;QAExC,UAAU,CAAC,KAAK,GAAG,KAAK,WAAW,GAAG,IAAe;YACnD,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;YAChC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC;YAEpC,MAAM,KAAK,GAAmB;gBAC5B,KAAK;gBACL,IAAI,EAAE,EAAE;gBACR,WAAW,EAAE,EAAE;gBACf,UAAU,EAAE,CAAC;gBACb,OAAO,EAAE,EAAE;aACZ,CAAC;YAEF,IAAI,YAAY,GAAG,KAAK,CAAC;YAEzB,KAAK,IAAI,GAAG,GAAG,CAAC,EAAE,GAAG,GAAG,OAAO,EAAE,GAAG,EAAE,EAAE,CAAC;gBACvC,6BAA6B;gBAC7B,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,YAAY,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAEnF,qBAAqB;gBACrB,MAAM,SAAS,GAAG,MAAM,iBAAiB,CAAC,YAAY,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;gBAEzE,MAAM,SAAS,GAAiB;oBAC9B,SAAS,EAAE,GAAG,GAAG,CAAC;oBAClB,KAAK,EAAE,YAAY;oBACnB,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;oBACrD,SAAS,EAAE,SAAS,CAAC,SAAS;oBAC9B,SAAS,EAAE,SAAS,CAAC,SAAS;iBAC/B,CAAC;gBAEF,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC3B,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC;gBAEzC,8BAA8B;gBAC9B,IAAI,CAAC,SAAS,CAAC,SAAS,IAAI,SAAS,CAAC,UAAU,EAAE,CAAC;oBACjD,MAAM;gBACR,CAAC;gBAED,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC;YACrC,CAAC;YAED,wBAAwB;YACxB,KAAK,CAAC,WAAW,GAAG,MAAM,qBAAqB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC/D,KAAK,CAAC,UAAU,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;YAE9C,OAAO,CAAC,cAAc,CAAC,qBAAqB,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAE1E,OAAO,KAAK,CAAC;QACf,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,iBAAiB,CAC9B,KAAa;AACb,8DAA8D;AAC9D,OAAY,EACZ,MAAsB;IAEtB,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;QACxB,OAAO;YACL,SAAS,EAAE,yBAAyB,GAAG,KAAK;YAC5C,UAAU,EAAE,IAAI;SACjB,CAAC;IACJ,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAEnE,MAAM,MAAM,GAAG,wBAAc,CAAC,GAAG,CAAwC,gCAAa,CAAC,CAAC,MAAM,CAAC;QAC7F,KAAK;QACL,UAAU;KACX,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,OAAO,MAAM,MAAM,CAAC,WAAW,CAAC,kBAAkB,CAI/C,MAAM,EAAE,EAAE,CAAC,CAAC;IACjB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO;YACL,SAAS,EAAE,mBAAmB;YAC9B,UAAU,EAAE,IAAI;SACjB,CAAC;IACJ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,KAAqB,EACrB,MAAsB;IAEtB,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;QACxB,OAAO,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACzD,CAAC;IAED,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,SAAS,KAAK,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEvF,MAAM,MAAM,GAAG,wBAAc,CAAC,GAAG,CAC/B,2CAAwB,CACzB,CAAC,MAAM,CAAC,EAAE,aAAa,EAAE,KAAK,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;IAElD,IAAI,CAAC;QACH,OAAO,MAAM,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IACnD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,OAAO,CAAC;IACjB,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAqB;IAChD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;IACnC,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;IAEzC,IAAI,UAAU,GAAG,GAAG,CAAC;IAErB,IAAI,WAAW,IAAI,CAAC;QAAE,UAAU,IAAI,GAAG,CAAC;IACxC,IAAI,QAAQ,IAAI,CAAC;QAAE,UAAU,IAAI,GAAG,CAAC;IACrC,IAAI,KAAK,CAAC,WAAW,CAAC,MAAM,GAAG,GAAG;QAAE,UAAU,IAAI,GAAG,CAAC;IAEtD,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;AACnC,CAAC;AAED,SAAgB,iBAAiB,CAC/B,MAAc,EACd,WAA4B;IAE5B,OAAO,OAAO,CAAC,WAAW,CAAC,qBAAqB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;AACzE,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Query Planner Decorator
|
|
3
|
+
* Automatically decomposes complex queries into sub-queries
|
|
4
|
+
*/
|
|
5
|
+
import 'reflect-metadata';
|
|
6
|
+
import { QueryPlan, AgenticLLMProvider } from '../types';
|
|
7
|
+
import '../../prompts/agentic/query-planner.prompt';
|
|
8
|
+
export interface QueryPlannerConfig {
|
|
9
|
+
decompose?: boolean;
|
|
10
|
+
rewrite?: boolean;
|
|
11
|
+
maxSubQueries?: number;
|
|
12
|
+
parallel?: boolean;
|
|
13
|
+
llmProvider?: AgenticLLMProvider;
|
|
14
|
+
}
|
|
15
|
+
export declare function QueryPlanner(config?: QueryPlannerConfig): MethodDecorator;
|
|
16
|
+
/**
|
|
17
|
+
* Get query plan metadata
|
|
18
|
+
*/
|
|
19
|
+
export declare function getQueryPlan(target: object, propertyKey: string | symbol): QueryPlan | undefined;
|
|
20
|
+
//# sourceMappingURL=query-planner.decorator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"query-planner.decorator.d.ts","sourceRoot":"","sources":["../../../src/agentic/decorators/query-planner.decorator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,kBAAkB,CAAC;AAC1B,OAAO,EAAE,SAAS,EAAY,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAEnE,OAAO,4CAA4C,CAAC;AAGpD,MAAM,WAAW,kBAAkB;IACjC,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,WAAW,CAAC,EAAE,kBAAkB,CAAC;CAClC;AAID,wBAAgB,YAAY,CAAC,MAAM,GAAE,kBAAuB,GAAG,eAAe,CA2B7E;AAuND;;GAEG;AACH,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,GAAG,SAAS,CAEhG"}
|
|
@@ -0,0 +1,213 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Query Planner Decorator
|
|
4
|
+
* Automatically decomposes complex queries into sub-queries
|
|
5
|
+
*/
|
|
6
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
7
|
+
exports.QueryPlanner = QueryPlanner;
|
|
8
|
+
exports.getQueryPlan = getQueryPlan;
|
|
9
|
+
require("reflect-metadata");
|
|
10
|
+
const prompts_1 = require("@hazeljs/prompts");
|
|
11
|
+
require("../../prompts/agentic/query-planner.prompt");
|
|
12
|
+
const query_planner_prompt_1 = require("../../prompts/agentic/query-planner.prompt");
|
|
13
|
+
const QUERY_PLANNER_METADATA_KEY = Symbol('queryPlanner');
|
|
14
|
+
function QueryPlanner(config = {}) {
|
|
15
|
+
return function (target, propertyKey, descriptor) {
|
|
16
|
+
const originalMethod = descriptor.value;
|
|
17
|
+
descriptor.value = async function (...args) {
|
|
18
|
+
const query = args[0];
|
|
19
|
+
if (!config.decompose) {
|
|
20
|
+
return originalMethod.apply(this, args);
|
|
21
|
+
}
|
|
22
|
+
// Decompose query into sub-queries
|
|
23
|
+
const plan = await decomposeQuery(query, config);
|
|
24
|
+
// Store plan in metadata
|
|
25
|
+
Reflect.defineMetadata(QUERY_PLANNER_METADATA_KEY, plan, target, propertyKey);
|
|
26
|
+
// Execute based on strategy
|
|
27
|
+
if (plan.strategy === 'parallel' && config.parallel) {
|
|
28
|
+
return executeParallel.call(this, plan, originalMethod, args);
|
|
29
|
+
}
|
|
30
|
+
else {
|
|
31
|
+
return executeSequential.call(this, plan, originalMethod, args);
|
|
32
|
+
}
|
|
33
|
+
};
|
|
34
|
+
return descriptor;
|
|
35
|
+
};
|
|
36
|
+
}
|
|
37
|
+
/**
|
|
38
|
+
* Decompose query into sub-queries
|
|
39
|
+
*/
|
|
40
|
+
async function decomposeQuery(query, config) {
|
|
41
|
+
// Check if query is complex enough to decompose
|
|
42
|
+
const complexity = estimateComplexity(query);
|
|
43
|
+
if (complexity < 3) {
|
|
44
|
+
return {
|
|
45
|
+
originalQuery: query,
|
|
46
|
+
subQueries: [
|
|
47
|
+
{
|
|
48
|
+
id: '1',
|
|
49
|
+
query,
|
|
50
|
+
type: 'factual',
|
|
51
|
+
dependencies: [],
|
|
52
|
+
priority: 1,
|
|
53
|
+
},
|
|
54
|
+
],
|
|
55
|
+
strategy: 'sequential',
|
|
56
|
+
estimatedComplexity: complexity,
|
|
57
|
+
};
|
|
58
|
+
}
|
|
59
|
+
// Use LLM to decompose if available
|
|
60
|
+
if (config.llmProvider) {
|
|
61
|
+
return decomposeWithLLM(query, config);
|
|
62
|
+
}
|
|
63
|
+
// Fallback to rule-based decomposition
|
|
64
|
+
return decomposeRuleBased(query, config);
|
|
65
|
+
}
|
|
66
|
+
/**
|
|
67
|
+
* LLM-based query decomposition
|
|
68
|
+
*/
|
|
69
|
+
async function decomposeWithLLM(query, config) {
|
|
70
|
+
const prompt = prompts_1.PromptRegistry.get(query_planner_prompt_1.QUERY_PLANNER_KEY).render({ query });
|
|
71
|
+
try {
|
|
72
|
+
const result = await config.llmProvider.generateStructured(prompt, {});
|
|
73
|
+
return {
|
|
74
|
+
originalQuery: query,
|
|
75
|
+
subQueries: result.subQueries.slice(0, config.maxSubQueries || 5),
|
|
76
|
+
strategy: result.strategy,
|
|
77
|
+
estimatedComplexity: result.subQueries.length,
|
|
78
|
+
};
|
|
79
|
+
}
|
|
80
|
+
catch {
|
|
81
|
+
return decomposeRuleBased(query, config);
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
/**
|
|
85
|
+
* Rule-based query decomposition
|
|
86
|
+
*/
|
|
87
|
+
function decomposeRuleBased(query, config) {
|
|
88
|
+
const subQueries = [];
|
|
89
|
+
// Split by conjunctions
|
|
90
|
+
const parts = query.split(/\band\b|\bor\b|\bthen\b/i);
|
|
91
|
+
parts.forEach((part, index) => {
|
|
92
|
+
const trimmed = part.trim();
|
|
93
|
+
if (trimmed.length > 10) {
|
|
94
|
+
subQueries.push({
|
|
95
|
+
id: String(index + 1),
|
|
96
|
+
query: trimmed,
|
|
97
|
+
type: detectQueryType(trimmed),
|
|
98
|
+
dependencies: index > 0 ? [String(index)] : [],
|
|
99
|
+
priority: index + 1,
|
|
100
|
+
});
|
|
101
|
+
}
|
|
102
|
+
});
|
|
103
|
+
if (subQueries.length === 0) {
|
|
104
|
+
subQueries.push({
|
|
105
|
+
id: '1',
|
|
106
|
+
query,
|
|
107
|
+
type: 'factual',
|
|
108
|
+
dependencies: [],
|
|
109
|
+
priority: 1,
|
|
110
|
+
});
|
|
111
|
+
}
|
|
112
|
+
return {
|
|
113
|
+
originalQuery: query,
|
|
114
|
+
subQueries: subQueries.slice(0, config.maxSubQueries || 5),
|
|
115
|
+
strategy: subQueries.length > 1 ? 'parallel' : 'sequential',
|
|
116
|
+
estimatedComplexity: subQueries.length,
|
|
117
|
+
};
|
|
118
|
+
}
|
|
119
|
+
/**
|
|
120
|
+
* Detect query type
|
|
121
|
+
*/
|
|
122
|
+
function detectQueryType(query) {
|
|
123
|
+
const lowerQuery = query.toLowerCase();
|
|
124
|
+
if (lowerQuery.includes('compare') || lowerQuery.includes('difference')) {
|
|
125
|
+
return 'comparative';
|
|
126
|
+
}
|
|
127
|
+
if (lowerQuery.includes('when') || lowerQuery.includes('timeline')) {
|
|
128
|
+
return 'temporal';
|
|
129
|
+
}
|
|
130
|
+
if (lowerQuery.includes('why') || lowerQuery.includes('how') || lowerQuery.includes('analyze')) {
|
|
131
|
+
return 'analytical';
|
|
132
|
+
}
|
|
133
|
+
return 'factual';
|
|
134
|
+
}
|
|
135
|
+
/**
|
|
136
|
+
* Estimate query complexity
|
|
137
|
+
*/
|
|
138
|
+
function estimateComplexity(query) {
|
|
139
|
+
let score = 1;
|
|
140
|
+
// Length factor
|
|
141
|
+
if (query.length > 100)
|
|
142
|
+
score += 1;
|
|
143
|
+
if (query.length > 200)
|
|
144
|
+
score += 1;
|
|
145
|
+
// Multiple questions
|
|
146
|
+
const questionMarks = (query.match(/\?/g) || []).length;
|
|
147
|
+
score += questionMarks;
|
|
148
|
+
// Conjunctions
|
|
149
|
+
const conjunctions = (query.match(/\band\b|\bor\b|\bthen\b/gi) || []).length;
|
|
150
|
+
score += conjunctions;
|
|
151
|
+
// Complex words
|
|
152
|
+
const complexWords = ['compare', 'analyze', 'evaluate', 'synthesize'];
|
|
153
|
+
complexWords.forEach((word) => {
|
|
154
|
+
if (query.toLowerCase().includes(word))
|
|
155
|
+
score += 1;
|
|
156
|
+
});
|
|
157
|
+
return Math.min(score, 10);
|
|
158
|
+
}
|
|
159
|
+
/**
|
|
160
|
+
* Execute sub-queries in parallel
|
|
161
|
+
*/
|
|
162
|
+
async function executeParallel(plan,
|
|
163
|
+
// eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
|
|
164
|
+
originalMethod, args) {
|
|
165
|
+
const results = await Promise.all(plan.subQueries.map((subQuery) => originalMethod.apply(this, [subQuery.query, ...args.slice(1)])));
|
|
166
|
+
// Merge results
|
|
167
|
+
return mergeResults(results, plan);
|
|
168
|
+
}
|
|
169
|
+
/**
|
|
170
|
+
* Execute sub-queries sequentially
|
|
171
|
+
*/
|
|
172
|
+
async function executeSequential(plan,
|
|
173
|
+
// eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
|
|
174
|
+
originalMethod, args) {
|
|
175
|
+
const results = [];
|
|
176
|
+
for (const subQuery of plan.subQueries) {
|
|
177
|
+
const result = await originalMethod.apply(this, [subQuery.query, ...args.slice(1)]);
|
|
178
|
+
results.push(result);
|
|
179
|
+
}
|
|
180
|
+
return mergeResults(results, plan);
|
|
181
|
+
}
|
|
182
|
+
/**
|
|
183
|
+
* Merge results from sub-queries
|
|
184
|
+
*/
|
|
185
|
+
function mergeResults(results, _plan) {
|
|
186
|
+
if (!Array.isArray(results[0])) {
|
|
187
|
+
return results;
|
|
188
|
+
}
|
|
189
|
+
// Flatten and deduplicate search results
|
|
190
|
+
const allResults = results.flat();
|
|
191
|
+
const seen = new Set();
|
|
192
|
+
const merged = [];
|
|
193
|
+
for (const result of allResults) {
|
|
194
|
+
if (typeof result === 'object' &&
|
|
195
|
+
result !== null &&
|
|
196
|
+
'id' in result &&
|
|
197
|
+
typeof result.id === 'string') {
|
|
198
|
+
const id = result.id;
|
|
199
|
+
if (!seen.has(id)) {
|
|
200
|
+
seen.add(id);
|
|
201
|
+
merged.push(result);
|
|
202
|
+
}
|
|
203
|
+
}
|
|
204
|
+
}
|
|
205
|
+
return merged;
|
|
206
|
+
}
|
|
207
|
+
/**
|
|
208
|
+
* Get query plan metadata
|
|
209
|
+
*/
|
|
210
|
+
function getQueryPlan(target, propertyKey) {
|
|
211
|
+
return Reflect.getMetadata(QUERY_PLANNER_METADATA_KEY, target, propertyKey);
|
|
212
|
+
}
|
|
213
|
+
//# sourceMappingURL=query-planner.decorator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"query-planner.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/query-planner.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAkBH,oCA2BC;AA0ND,oCAEC;AAvQD,4BAA0B;AAE1B,8CAAkD;AAClD,sDAAoD;AACpD,qFAA+E;AAU/E,MAAM,0BAA0B,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;AAE1D,SAAgB,YAAY,CAAC,SAA6B,EAAE;IAC1D,OAAO,UAAU,MAAc,EAAE,WAA4B,EAAE,UAA8B;QAC3F,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,CAAC;QAExC,UAAU,CAAC,KAAK,GAAG,KAAK,WAAW,GAAG,IAAe;YACnD,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;YAEhC,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;gBACtB,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC1C,CAAC;YAED,mCAAmC;YACnC,MAAM,IAAI,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAEjD,yBAAyB;YACzB,OAAO,CAAC,cAAc,CAAC,0BAA0B,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAE9E,4BAA4B;YAC5B,IAAI,IAAI,CAAC,QAAQ,KAAK,UAAU,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpD,OAAO,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,CAAC,CAAC;YAChE,CAAC;iBAAM,CAAC;gBACN,OAAO,iBAAiB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,CAAC,CAAC;YAClE,CAAC;QACH,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,cAAc,CAAC,KAAa,EAAE,MAA0B;IACrE,gDAAgD;IAChD,MAAM,UAAU,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;IAE7C,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;QACnB,OAAO;YACL,aAAa,EAAE,KAAK;YACpB,UAAU,EAAE;gBACV;oBACE,EAAE,EAAE,GAAG;oBACP,KAAK;oBACL,IAAI,EAAE,SAAS;oBACf,YAAY,EAAE,EAAE;oBAChB,QAAQ,EAAE,CAAC;iBACZ;aACF;YACD,QAAQ,EAAE,YAAY;YACtB,mBAAmB,EAAE,UAAU;SAChC,CAAC;IACJ,CAAC;IAED,oCAAoC;IACpC,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,OAAO,gBAAgB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IACzC,CAAC;IAED,uCAAuC;IACvC,OAAO,kBAAkB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;AAC3C,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,gBAAgB,CAAC,KAAa,EAAE,MAA0B;IACvE,MAAM,MAAM,GAAG,wBAAc,CAAC,GAAG,CAAoB,wCAAiB,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAE1F,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,WAAY,CAAC,kBAAkB,CAGxD,MAAM,EAAE,EAAE,CAAC,CAAC;QAEf,OAAO;YACL,aAAa,EAAE,KAAK;YACpB,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,aAAa,IAAI,CAAC,CAAC;YACjE,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,mBAAmB,EAAE,MAAM,CAAC,UAAU,CAAC,MAAM;SAC9C,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,kBAAkB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC3C,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,kBAAkB,CAAC,KAAa,EAAE,MAA0B;IACnE,MAAM,UAAU,GAAe,EAAE,CAAC;IAElC,wBAAwB;IACxB,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;IAEtD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAC5B,IAAI,OAAO,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YACxB,UAAU,CAAC,IAAI,CAAC;gBACd,EAAE,EAAE,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC;gBACrB,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,eAAe,CAAC,OAAO,CAAC;gBAC9B,YAAY,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;gBAC9C,QAAQ,EAAE,KAAK,GAAG,CAAC;aACpB,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,UAAU,CAAC,IAAI,CAAC;YACd,EAAE,EAAE,GAAG;YACP,KAAK;YACL,IAAI,EAAE,SAAS;YACf,YAAY,EAAE,EAAE;YAChB,QAAQ,EAAE,CAAC;SACZ,CAAC,CAAC;IACL,CAAC;IAED,OAAO;QACL,aAAa,EAAE,KAAK;QACpB,UAAU,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,aAAa,IAAI,CAAC,CAAC;QAC1D,QAAQ,EAAE,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,YAAY;QAC3D,mBAAmB,EAAE,UAAU,CAAC,MAAM;KACvC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,KAAa;IACpC,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAEvC,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;QACxE,OAAO,aAAa,CAAC;IACvB,CAAC;IACD,IAAI,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;QACnE,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,IAAI,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAC/F,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;GAEG;AACH,SAAS,kBAAkB,CAAC,KAAa;IACvC,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,gBAAgB;IAChB,IAAI,KAAK,CAAC,MAAM,GAAG,GAAG;QAAE,KAAK,IAAI,CAAC,CAAC;IACnC,IAAI,KAAK,CAAC,MAAM,GAAG,GAAG;QAAE,KAAK,IAAI,CAAC,CAAC;IAEnC,qBAAqB;IACrB,MAAM,aAAa,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;IACxD,KAAK,IAAI,aAAa,CAAC;IAEvB,eAAe;IACf,MAAM,YAAY,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,2BAA2B,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;IAC7E,KAAK,IAAI,YAAY,CAAC;IAEtB,gBAAgB;IAChB,MAAM,YAAY,GAAG,CAAC,SAAS,EAAE,SAAS,EAAE,UAAU,EAAE,YAAY,CAAC,CAAC;IACtE,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAC5B,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC;YAAE,KAAK,IAAI,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;AAC7B,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,eAAe,CAE5B,IAAe;AACf,sEAAsE;AACtE,cAAwB,EACxB,IAAe;IAEf,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAC/B,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAC/D,CACF,CAAC;IAEF,gBAAgB;IAChB,OAAO,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;AACrC,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,iBAAiB,CAE9B,IAAe;AACf,sEAAsE;AACtE,cAAwB,EACxB,IAAe;IAEf,MAAM,OAAO,GAAG,EAAE,CAAC;IAEnB,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACpF,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACvB,CAAC;IAED,OAAO,YAAY,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;AACrC,CAAC;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,OAAkB,EAAE,KAAgB;IACxD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/B,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,yCAAyC;IACzC,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;IAClC,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,MAAM,MAAM,GAAc,EAAE,CAAC;IAE7B,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;QAChC,IACE,OAAO,MAAM,KAAK,QAAQ;YAC1B,MAAM,KAAK,IAAI;YACf,IAAI,IAAI,MAAM;YACd,OAAQ,MAA0B,CAAC,EAAE,KAAK,QAAQ,EAClD,CAAC;YACD,MAAM,EAAE,GAAI,MAAyB,CAAC,EAAE,CAAC;YACzC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;gBAClB,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACb,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAgB,YAAY,CAAC,MAAc,EAAE,WAA4B;IACvE,OAAO,OAAO,CAAC,WAAW,CAAC,0BAA0B,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;AAC9E,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Query Rewriter Decorator
|
|
3
|
+
* Rewrites and expands queries for better retrieval
|
|
4
|
+
*/
|
|
5
|
+
import 'reflect-metadata';
|
|
6
|
+
import { AgenticLLMProvider } from '../types';
|
|
7
|
+
import '../../prompts/agentic/query-rewriter.prompt';
|
|
8
|
+
export interface QueryRewriterConfig {
|
|
9
|
+
techniques?: ('expansion' | 'clarification' | 'synonym' | 'decomposition')[];
|
|
10
|
+
llmBased?: boolean;
|
|
11
|
+
llmProvider?: AgenticLLMProvider;
|
|
12
|
+
maxVariations?: number;
|
|
13
|
+
}
|
|
14
|
+
export declare function QueryRewriter(config?: QueryRewriterConfig): MethodDecorator;
|
|
15
|
+
export declare function getQueryVariations(target: object, propertyKey: string | symbol): string[];
|
|
16
|
+
//# sourceMappingURL=query-rewriter.decorator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"query-rewriter.decorator.d.ts","sourceRoot":"","sources":["../../../src/agentic/decorators/query-rewriter.decorator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,kBAAkB,CAAC;AAC1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAE9C,OAAO,6CAA6C,CAAC;AAGrD,MAAM,WAAW,mBAAmB;IAClC,UAAU,CAAC,EAAE,CAAC,WAAW,GAAG,eAAe,GAAG,SAAS,GAAG,eAAe,CAAC,EAAE,CAAC;IAC7E,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,WAAW,CAAC,EAAE,kBAAkB,CAAC;IACjC,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAID,wBAAgB,aAAa,CAAC,MAAM,GAAE,mBAAwB,GAAG,eAAe,CAqC/E;AAiID,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,GAAG,MAAM,EAAE,CAEzF"}
|
|
@@ -0,0 +1,143 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Query Rewriter Decorator
|
|
4
|
+
* Rewrites and expands queries for better retrieval
|
|
5
|
+
*/
|
|
6
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
7
|
+
exports.QueryRewriter = QueryRewriter;
|
|
8
|
+
exports.getQueryVariations = getQueryVariations;
|
|
9
|
+
require("reflect-metadata");
|
|
10
|
+
const prompts_1 = require("@hazeljs/prompts");
|
|
11
|
+
require("../../prompts/agentic/query-rewriter.prompt");
|
|
12
|
+
const query_rewriter_prompt_1 = require("../../prompts/agentic/query-rewriter.prompt");
|
|
13
|
+
const REWRITER_METADATA_KEY = Symbol('queryRewriter');
|
|
14
|
+
function QueryRewriter(config = {}) {
|
|
15
|
+
return function (target, propertyKey, descriptor) {
|
|
16
|
+
const originalMethod = descriptor.value;
|
|
17
|
+
descriptor.value = async function (...args) {
|
|
18
|
+
const query = args[0];
|
|
19
|
+
// Generate query variations
|
|
20
|
+
const variations = await generateQueryVariations(query, config);
|
|
21
|
+
// Execute search with all variations
|
|
22
|
+
const allResults = [];
|
|
23
|
+
const seenIds = new Set();
|
|
24
|
+
for (const variation of variations) {
|
|
25
|
+
const results = await originalMethod.apply(this, [variation, ...args.slice(1)]);
|
|
26
|
+
if (Array.isArray(results)) {
|
|
27
|
+
for (const result of results) {
|
|
28
|
+
if (result.id && !seenIds.has(result.id)) {
|
|
29
|
+
seenIds.add(result.id);
|
|
30
|
+
allResults.push(result);
|
|
31
|
+
}
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
// Re-rank by frequency across variations
|
|
36
|
+
const ranked = rerankByFrequency(allResults, variations.length);
|
|
37
|
+
Reflect.defineMetadata(REWRITER_METADATA_KEY, variations, target, propertyKey);
|
|
38
|
+
return ranked;
|
|
39
|
+
};
|
|
40
|
+
return descriptor;
|
|
41
|
+
};
|
|
42
|
+
}
|
|
43
|
+
async function generateQueryVariations(query, config) {
|
|
44
|
+
const variations = [query]; // Always include original
|
|
45
|
+
const techniques = config.techniques || ['expansion', 'synonym'];
|
|
46
|
+
const maxVariations = config.maxVariations || 3;
|
|
47
|
+
if (config.llmBased && config.llmProvider) {
|
|
48
|
+
return generateWithLLM(query, techniques, config.llmProvider, maxVariations);
|
|
49
|
+
}
|
|
50
|
+
// Rule-based variations
|
|
51
|
+
for (const technique of techniques) {
|
|
52
|
+
switch (technique) {
|
|
53
|
+
case 'expansion':
|
|
54
|
+
variations.push(...expandQuery(query));
|
|
55
|
+
break;
|
|
56
|
+
case 'synonym':
|
|
57
|
+
variations.push(...addSynonyms(query));
|
|
58
|
+
break;
|
|
59
|
+
case 'clarification':
|
|
60
|
+
variations.push(...clarifyQuery(query));
|
|
61
|
+
break;
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
return [...new Set(variations)].slice(0, maxVariations + 1);
|
|
65
|
+
}
|
|
66
|
+
async function generateWithLLM(query, techniques, llmProvider, maxVariations) {
|
|
67
|
+
const prompt = prompts_1.PromptRegistry.get(query_rewriter_prompt_1.QUERY_REWRITER_KEY).render({
|
|
68
|
+
maxVariations: String(maxVariations),
|
|
69
|
+
techniques: techniques.join(', '),
|
|
70
|
+
query,
|
|
71
|
+
});
|
|
72
|
+
try {
|
|
73
|
+
const response = await llmProvider.generate(prompt, { temperature: 0.7 });
|
|
74
|
+
const variations = response
|
|
75
|
+
.split('\n')
|
|
76
|
+
.map((line) => line.replace(/^\d+\.\s*/, '').trim())
|
|
77
|
+
.filter((line) => line.length > 5)
|
|
78
|
+
.slice(0, maxVariations);
|
|
79
|
+
return [query, ...variations];
|
|
80
|
+
}
|
|
81
|
+
catch {
|
|
82
|
+
return [query];
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
function expandQuery(query) {
|
|
86
|
+
const expansions = [];
|
|
87
|
+
// Add question words if not present
|
|
88
|
+
if (!query.match(/^(what|who|where|when|why|how)/i)) {
|
|
89
|
+
expansions.push(`What is ${query}?`);
|
|
90
|
+
expansions.push(`How does ${query} work?`);
|
|
91
|
+
}
|
|
92
|
+
// Add context
|
|
93
|
+
expansions.push(`${query} explanation`);
|
|
94
|
+
expansions.push(`${query} details`);
|
|
95
|
+
return expansions;
|
|
96
|
+
}
|
|
97
|
+
function addSynonyms(query) {
|
|
98
|
+
const synonymMap = {
|
|
99
|
+
find: ['search', 'locate', 'discover'],
|
|
100
|
+
show: ['display', 'present', 'demonstrate'],
|
|
101
|
+
explain: ['describe', 'clarify', 'elaborate'],
|
|
102
|
+
compare: ['contrast', 'differentiate', 'analyze'],
|
|
103
|
+
};
|
|
104
|
+
const variations = [];
|
|
105
|
+
const words = query.split(/\s+/);
|
|
106
|
+
words.forEach((word, index) => {
|
|
107
|
+
const lowerWord = word.toLowerCase();
|
|
108
|
+
if (synonymMap[lowerWord]) {
|
|
109
|
+
synonymMap[lowerWord].forEach((synonym) => {
|
|
110
|
+
const newWords = [...words];
|
|
111
|
+
newWords[index] = synonym;
|
|
112
|
+
variations.push(newWords.join(' '));
|
|
113
|
+
});
|
|
114
|
+
}
|
|
115
|
+
});
|
|
116
|
+
return variations;
|
|
117
|
+
}
|
|
118
|
+
function clarifyQuery(query) {
|
|
119
|
+
const clarifications = [];
|
|
120
|
+
// Add specificity
|
|
121
|
+
if (query.length < 50) {
|
|
122
|
+
clarifications.push(`${query} with examples`);
|
|
123
|
+
clarifications.push(`${query} in detail`);
|
|
124
|
+
}
|
|
125
|
+
return clarifications;
|
|
126
|
+
}
|
|
127
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
128
|
+
function rerankByFrequency(results, variationCount) {
|
|
129
|
+
const frequency = new Map();
|
|
130
|
+
results.forEach((result) => {
|
|
131
|
+
frequency.set(result.id, (frequency.get(result.id) || 0) + 1);
|
|
132
|
+
});
|
|
133
|
+
return results
|
|
134
|
+
.map((result) => ({
|
|
135
|
+
...result,
|
|
136
|
+
score: (result.score || 0) * (1 + (frequency.get(result.id) || 0) / variationCount),
|
|
137
|
+
}))
|
|
138
|
+
.sort((a, b) => (b.score || 0) - (a.score || 0));
|
|
139
|
+
}
|
|
140
|
+
function getQueryVariations(target, propertyKey) {
|
|
141
|
+
return Reflect.getMetadata(REWRITER_METADATA_KEY, target, propertyKey) || [];
|
|
142
|
+
}
|
|
143
|
+
//# sourceMappingURL=query-rewriter.decorator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"query-rewriter.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/query-rewriter.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAiBH,sCAqCC;AAiID,gDAEC;AAvLD,4BAA0B;AAE1B,8CAAkD;AAClD,uDAAqD;AACrD,uFAAiF;AASjF,MAAM,qBAAqB,GAAG,MAAM,CAAC,eAAe,CAAC,CAAC;AAEtD,SAAgB,aAAa,CAAC,SAA8B,EAAE;IAC5D,OAAO,UAAU,MAAc,EAAE,WAA4B,EAAE,UAA8B;QAC3F,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,CAAC;QAExC,UAAU,CAAC,KAAK,GAAG,KAAK,WAAW,GAAG,IAAe;YACnD,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;YAEhC,4BAA4B;YAC5B,MAAM,UAAU,GAAG,MAAM,uBAAuB,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAEhE,qCAAqC;YACrC,MAAM,UAAU,GAAG,EAAE,CAAC;YACtB,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;YAElC,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;gBACnC,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,SAAS,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAEhF,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC3B,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;wBAC7B,IAAI,MAAM,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC;4BACzC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;4BACvB,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;wBAC1B,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,yCAAyC;YACzC,MAAM,MAAM,GAAG,iBAAiB,CAAC,UAAU,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;YAEhE,OAAO,CAAC,cAAc,CAAC,qBAAqB,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAE/E,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,uBAAuB,CACpC,KAAa,EACb,MAA2B;IAE3B,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,0BAA0B;IACtD,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;IACjE,MAAM,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,CAAC,CAAC;IAEhD,IAAI,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QAC1C,OAAO,eAAe,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;IAC/E,CAAC;IAED,wBAAwB;IACxB,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACnC,QAAQ,SAAS,EAAE,CAAC;YAClB,KAAK,WAAW;gBACd,UAAU,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,SAAS;gBACZ,UAAU,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,eAAe;gBAClB,UAAU,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC;gBACxC,MAAM;QACV,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,aAAa,GAAG,CAAC,CAAC,CAAC;AAC9D,CAAC;AAED,KAAK,UAAU,eAAe,CAC5B,KAAa,EACb,UAAoB,EACpB,WAA+B,EAC/B,aAAqB;IAErB,MAAM,MAAM,GAAG,wBAAc,CAAC,GAAG,CAC/B,0CAAkB,CACnB,CAAC,MAAM,CAAC;QACP,aAAa,EAAE,MAAM,CAAC,aAAa,CAAC;QACpC,UAAU,EAAE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;QACjC,KAAK;KACN,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QAC1E,MAAM,UAAU,GAAG,QAAQ;aACxB,KAAK,CAAC,IAAI,CAAC;aACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;aACnD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC;aACjC,KAAK,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC;QAE3B,OAAO,CAAC,KAAK,EAAE,GAAG,UAAU,CAAC,CAAC;IAChC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,KAAK,CAAC,CAAC;IACjB,CAAC;AACH,CAAC;AAED,SAAS,WAAW,CAAC,KAAa;IAChC,MAAM,UAAU,GAAa,EAAE,CAAC;IAEhC,oCAAoC;IACpC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,iCAAiC,CAAC,EAAE,CAAC;QACpD,UAAU,CAAC,IAAI,CAAC,WAAW,KAAK,GAAG,CAAC,CAAC;QACrC,UAAU,CAAC,IAAI,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC;IAC7C,CAAC;IAED,cAAc;IACd,UAAU,CAAC,IAAI,CAAC,GAAG,KAAK,cAAc,CAAC,CAAC;IACxC,UAAU,CAAC,IAAI,CAAC,GAAG,KAAK,UAAU,CAAC,CAAC;IAEpC,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,WAAW,CAAC,KAAa;IAChC,MAAM,UAAU,GAA6B;QAC3C,IAAI,EAAE,CAAC,QAAQ,EAAE,QAAQ,EAAE,UAAU,CAAC;QACtC,IAAI,EAAE,CAAC,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC;QAC3C,OAAO,EAAE,CAAC,UAAU,EAAE,SAAS,EAAE,WAAW,CAAC;QAC7C,OAAO,EAAE,CAAC,UAAU,EAAE,eAAe,EAAE,SAAS,CAAC;KAClD,CAAC;IAEF,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAEjC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QAC5B,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACrC,IAAI,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1B,UAAU,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBACxC,MAAM,QAAQ,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC;gBAC5B,QAAQ,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC;gBAC1B,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,YAAY,CAAC,KAAa;IACjC,MAAM,cAAc,GAAa,EAAE,CAAC;IAEpC,kBAAkB;IAClB,IAAI,KAAK,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;QACtB,cAAc,CAAC,IAAI,CAAC,GAAG,KAAK,gBAAgB,CAAC,CAAC;QAC9C,cAAc,CAAC,IAAI,CAAC,GAAG,KAAK,YAAY,CAAC,CAAC;IAC5C,CAAC;IAED,OAAO,cAAc,CAAC;AACxB,CAAC;AAED,8DAA8D;AAC9D,SAAS,iBAAiB,CAAC,OAAc,EAAE,cAAsB;IAC/D,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;IAE5C,OAAO,CAAC,OAAO,CAAC,CAAC,MAAsB,EAAE,EAAE;QACzC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,OAAO,OAAO;SACX,GAAG,CAAC,CAAC,MAAsC,EAAE,EAAE,CAAC,CAAC;QAChD,GAAG,MAAM;QACT,KAAK,EAAE,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,cAAc,CAAC;KACpF,CAAC,CAAC;SACF,IAAI,CAAC,CAAC,CAAqB,EAAE,CAAqB,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC;AAC7F,CAAC;AAED,SAAgB,kBAAkB,CAAC,MAAc,EAAE,WAA4B;IAC7E,OAAO,OAAO,CAAC,WAAW,CAAC,qBAAqB,EAAE,MAAM,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC;AAC/E,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Self-Reflective Decorator
|
|
3
|
+
* Evaluates and iteratively improves retrieval results
|
|
4
|
+
*/
|
|
5
|
+
import 'reflect-metadata';
|
|
6
|
+
import { ReflectionResult, AgenticLLMProvider } from '../types';
|
|
7
|
+
import '../../prompts/agentic/self-reflective.prompt';
|
|
8
|
+
import '../../prompts/agentic/self-reflective-improve.prompt';
|
|
9
|
+
export interface SelfReflectiveConfig {
|
|
10
|
+
maxIterations?: number;
|
|
11
|
+
qualityThreshold?: number;
|
|
12
|
+
llmProvider?: AgenticLLMProvider;
|
|
13
|
+
enableAutoImprovement?: boolean;
|
|
14
|
+
}
|
|
15
|
+
export declare function SelfReflective(config?: SelfReflectiveConfig): MethodDecorator;
|
|
16
|
+
/**
|
|
17
|
+
* Get reflection metadata
|
|
18
|
+
*/
|
|
19
|
+
export declare function getReflections(target: object, propertyKey: string | symbol): ReflectionResult[];
|
|
20
|
+
//# sourceMappingURL=self-reflective.decorator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"self-reflective.decorator.d.ts","sourceRoot":"","sources":["../../../src/agentic/decorators/self-reflective.decorator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,kBAAkB,CAAC;AAC1B,OAAO,EAAE,gBAAgB,EAAqB,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAEnF,OAAO,8CAA8C,CAAC;AACtD,OAAO,sDAAsD,CAAC;AAI9D,MAAM,WAAW,oBAAoB;IACnC,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,WAAW,CAAC,EAAE,kBAAkB,CAAC;IACjC,qBAAqB,CAAC,EAAE,OAAO,CAAC;CACjC;AAID,wBAAgB,cAAc,CAAC,MAAM,GAAE,oBAAyB,GAAG,eAAe,CA8DjF;AA+JD;;GAEG;AACH,wBAAgB,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,MAAM,GAAG,gBAAgB,EAAE,CAE/F"}
|