@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,169 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Context-Aware Decorator
|
|
4
|
+
* Maintains conversation context for improved retrieval
|
|
5
|
+
*/
|
|
6
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
7
|
+
exports.ContextAware = ContextAware;
|
|
8
|
+
exports.getContext = getContext;
|
|
9
|
+
exports.clearContext = clearContext;
|
|
10
|
+
require("reflect-metadata");
|
|
11
|
+
const CONTEXT_METADATA_KEY = Symbol('contextAware');
|
|
12
|
+
const contextStore = new Map();
|
|
13
|
+
function ContextAware(config = {}) {
|
|
14
|
+
return function (target, propertyKey, descriptor) {
|
|
15
|
+
const originalMethod = descriptor.value;
|
|
16
|
+
descriptor.value = async function (...args) {
|
|
17
|
+
const query = args[0];
|
|
18
|
+
const sessionId = (typeof args[1] === 'object' && args[1] !== null && 'sessionId' in args[1]
|
|
19
|
+
? args[1].sessionId
|
|
20
|
+
: undefined) ||
|
|
21
|
+
config.sessionId ||
|
|
22
|
+
'default';
|
|
23
|
+
// Get or create context
|
|
24
|
+
let context = contextStore.get(sessionId);
|
|
25
|
+
if (!context) {
|
|
26
|
+
context = {
|
|
27
|
+
sessionId,
|
|
28
|
+
conversationHistory: [],
|
|
29
|
+
entities: [],
|
|
30
|
+
topics: [],
|
|
31
|
+
metadata: {},
|
|
32
|
+
};
|
|
33
|
+
contextStore.set(sessionId, context);
|
|
34
|
+
}
|
|
35
|
+
// Update context with current query
|
|
36
|
+
context.conversationHistory.push({
|
|
37
|
+
role: 'user',
|
|
38
|
+
content: query,
|
|
39
|
+
timestamp: new Date(),
|
|
40
|
+
});
|
|
41
|
+
// Maintain window size
|
|
42
|
+
const windowSize = config.windowSize || 5;
|
|
43
|
+
if (context.conversationHistory.length > windowSize * 2) {
|
|
44
|
+
context.conversationHistory = context.conversationHistory.slice(-windowSize * 2);
|
|
45
|
+
}
|
|
46
|
+
// Extract entities if enabled
|
|
47
|
+
if (config.entityTracking) {
|
|
48
|
+
const entities = extractEntities(query);
|
|
49
|
+
updateEntities(context, entities);
|
|
50
|
+
}
|
|
51
|
+
// Extract topics if enabled
|
|
52
|
+
if (config.topicModeling) {
|
|
53
|
+
const topics = extractTopics(query, context);
|
|
54
|
+
updateTopics(context, topics);
|
|
55
|
+
}
|
|
56
|
+
// Enhance query with context
|
|
57
|
+
const enhancedQuery = enhanceQueryWithContext(query, context);
|
|
58
|
+
// Execute with enhanced query
|
|
59
|
+
const modifiedArgs = [enhancedQuery, ...args.slice(1)];
|
|
60
|
+
const results = await originalMethod.apply(this, modifiedArgs);
|
|
61
|
+
// Store assistant response in context
|
|
62
|
+
if (results) {
|
|
63
|
+
context.conversationHistory.push({
|
|
64
|
+
role: 'assistant',
|
|
65
|
+
content: JSON.stringify(results).slice(0, 500),
|
|
66
|
+
timestamp: new Date(),
|
|
67
|
+
});
|
|
68
|
+
}
|
|
69
|
+
Reflect.defineMetadata(CONTEXT_METADATA_KEY, context, target, propertyKey);
|
|
70
|
+
return results;
|
|
71
|
+
};
|
|
72
|
+
return descriptor;
|
|
73
|
+
};
|
|
74
|
+
}
|
|
75
|
+
function extractEntities(text) {
|
|
76
|
+
const entities = [];
|
|
77
|
+
// Simple named entity extraction (capitalized words)
|
|
78
|
+
const words = text.split(/\s+/);
|
|
79
|
+
const capitalizedWords = words.filter((w) => /^[A-Z][a-z]+/.test(w));
|
|
80
|
+
const entityCounts = new Map();
|
|
81
|
+
capitalizedWords.forEach((word) => {
|
|
82
|
+
entityCounts.set(word, (entityCounts.get(word) || 0) + 1);
|
|
83
|
+
});
|
|
84
|
+
entityCounts.forEach((count, name) => {
|
|
85
|
+
entities.push({
|
|
86
|
+
name,
|
|
87
|
+
type: 'UNKNOWN',
|
|
88
|
+
mentions: count,
|
|
89
|
+
lastSeen: new Date(),
|
|
90
|
+
attributes: {},
|
|
91
|
+
});
|
|
92
|
+
});
|
|
93
|
+
return entities;
|
|
94
|
+
}
|
|
95
|
+
function updateEntities(context, newEntities) {
|
|
96
|
+
newEntities.forEach((newEntity) => {
|
|
97
|
+
const existing = context.entities.find((e) => e.name === newEntity.name);
|
|
98
|
+
if (existing) {
|
|
99
|
+
existing.mentions += newEntity.mentions;
|
|
100
|
+
existing.lastSeen = new Date();
|
|
101
|
+
}
|
|
102
|
+
else {
|
|
103
|
+
context.entities.push(newEntity);
|
|
104
|
+
}
|
|
105
|
+
});
|
|
106
|
+
// Keep only recent entities
|
|
107
|
+
context.entities = context.entities
|
|
108
|
+
.sort((a, b) => b.lastSeen.getTime() - a.lastSeen.getTime())
|
|
109
|
+
.slice(0, 20);
|
|
110
|
+
}
|
|
111
|
+
function extractTopics(query, _context) {
|
|
112
|
+
const topics = [];
|
|
113
|
+
// Extract keywords
|
|
114
|
+
const words = query.toLowerCase().split(/\s+/);
|
|
115
|
+
const stopWords = new Set(['the', 'a', 'an', 'and', 'or', 'but', 'in', 'on', 'at', 'to', 'for']);
|
|
116
|
+
const keywords = words.filter((w) => w.length > 3 && !stopWords.has(w));
|
|
117
|
+
// Group into topics
|
|
118
|
+
const topicMap = new Map();
|
|
119
|
+
keywords.forEach((keyword) => {
|
|
120
|
+
const topic = keyword.slice(0, 5); // Simple topic grouping
|
|
121
|
+
if (!topicMap.has(topic)) {
|
|
122
|
+
topicMap.set(topic, []);
|
|
123
|
+
}
|
|
124
|
+
topicMap.get(topic).push(keyword);
|
|
125
|
+
});
|
|
126
|
+
topicMap.forEach((keywords, name) => {
|
|
127
|
+
topics.push({
|
|
128
|
+
name,
|
|
129
|
+
relevance: keywords.length / words.length,
|
|
130
|
+
keywords,
|
|
131
|
+
});
|
|
132
|
+
});
|
|
133
|
+
return topics;
|
|
134
|
+
}
|
|
135
|
+
function updateTopics(context, newTopics) {
|
|
136
|
+
newTopics.forEach((newTopic) => {
|
|
137
|
+
const existing = context.topics.find((t) => t.name === newTopic.name);
|
|
138
|
+
if (existing) {
|
|
139
|
+
existing.relevance = (existing.relevance + newTopic.relevance) / 2;
|
|
140
|
+
existing.keywords = [...new Set([...existing.keywords, ...newTopic.keywords])];
|
|
141
|
+
}
|
|
142
|
+
else {
|
|
143
|
+
context.topics.push(newTopic);
|
|
144
|
+
}
|
|
145
|
+
});
|
|
146
|
+
// Keep top topics
|
|
147
|
+
context.topics = context.topics.sort((a, b) => b.relevance - a.relevance).slice(0, 10);
|
|
148
|
+
}
|
|
149
|
+
function enhanceQueryWithContext(query, context) {
|
|
150
|
+
const enhancements = [query];
|
|
151
|
+
// Add recent entities
|
|
152
|
+
const topEntities = context.entities.slice(0, 3).map((e) => e.name);
|
|
153
|
+
if (topEntities.length > 0) {
|
|
154
|
+
enhancements.push(`Context entities: ${topEntities.join(', ')}`);
|
|
155
|
+
}
|
|
156
|
+
// Add recent topics
|
|
157
|
+
const topTopics = context.topics.slice(0, 2).map((t) => t.keywords[0]);
|
|
158
|
+
if (topTopics.length > 0) {
|
|
159
|
+
enhancements.push(`Related topics: ${topTopics.join(', ')}`);
|
|
160
|
+
}
|
|
161
|
+
return enhancements.join(' | ');
|
|
162
|
+
}
|
|
163
|
+
function getContext(target, propertyKey) {
|
|
164
|
+
return Reflect.getMetadata(CONTEXT_METADATA_KEY, target, propertyKey);
|
|
165
|
+
}
|
|
166
|
+
function clearContext(sessionId) {
|
|
167
|
+
contextStore.delete(sessionId);
|
|
168
|
+
}
|
|
169
|
+
//# sourceMappingURL=context-aware.decorator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"context-aware.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/context-aware.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAeH,oCA0EC;AA0GD,gCAEC;AAED,oCAEC;AAvMD,4BAA0B;AAU1B,MAAM,oBAAoB,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;AACpD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAmB,CAAC;AAEhD,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;YAChC,MAAM,SAAS,GACb,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC;gBACxE,CAAC,CAAE,IAAI,CAAC,CAAC,CAA4B,CAAC,SAAS;gBAC/C,CAAC,CAAC,SAAS,CAAC;gBACd,MAAM,CAAC,SAAS;gBAChB,SAAS,CAAC;YAEZ,wBAAwB;YACxB,IAAI,OAAO,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC1C,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,GAAG;oBACR,SAAS;oBACT,mBAAmB,EAAE,EAAE;oBACvB,QAAQ,EAAE,EAAE;oBACZ,MAAM,EAAE,EAAE;oBACV,QAAQ,EAAE,EAAE;iBACb,CAAC;gBACF,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;YACvC,CAAC;YAED,oCAAoC;YACpC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC;gBAC/B,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,KAAK;gBACd,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC,CAAC;YAEH,uBAAuB;YACvB,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,CAAC,CAAC;YAC1C,IAAI,OAAO,CAAC,mBAAmB,CAAC,MAAM,GAAG,UAAU,GAAG,CAAC,EAAE,CAAC;gBACxD,OAAO,CAAC,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;YACnF,CAAC;YAED,8BAA8B;YAC9B,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;gBAC1B,MAAM,QAAQ,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;gBACxC,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YACpC,CAAC;YAED,4BAA4B;YAC5B,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;gBAC7C,YAAY,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;YAChC,CAAC;YAED,6BAA6B;YAC7B,MAAM,aAAa,GAAG,uBAAuB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAE9D,8BAA8B;YAC9B,MAAM,YAAY,GAAG,CAAC,aAAa,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;YACvD,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;YAE/D,sCAAsC;YACtC,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC;oBAC/B,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;oBAC9C,SAAS,EAAE,IAAI,IAAI,EAAE;iBACtB,CAAC,CAAC;YACL,CAAC;YAED,OAAO,CAAC,cAAc,CAAC,oBAAoB,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAE3E,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,IAAY;IACnC,MAAM,QAAQ,GAAa,EAAE,CAAC;IAE9B,qDAAqD;IACrD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAChC,MAAM,gBAAgB,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAErE,MAAM,YAAY,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC/C,gBAAgB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAChC,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,YAAY,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACnC,QAAQ,CAAC,IAAI,CAAC;YACZ,IAAI;YACJ,IAAI,EAAE,SAAS;YACf,QAAQ,EAAE,KAAK;YACf,QAAQ,EAAE,IAAI,IAAI,EAAE;YACpB,UAAU,EAAE,EAAE;SACf,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,cAAc,CAAC,OAAgB,EAAE,WAAqB;IAC7D,WAAW,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;QAChC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,IAAI,CAAC,CAAC;QACzE,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,QAAQ,IAAI,SAAS,CAAC,QAAQ,CAAC;YACxC,QAAQ,CAAC,QAAQ,GAAG,IAAI,IAAI,EAAE,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,4BAA4B;IAC5B,OAAO,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ;SAChC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;SAC3D,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AAClB,CAAC;AAED,SAAS,aAAa,CAAC,KAAa,EAAE,QAAiB;IACrD,MAAM,MAAM,GAAY,EAAE,CAAC;IAE3B,mBAAmB;IACnB,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC/C,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;IACjG,MAAM,QAAQ,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAExE,oBAAoB;IACpB,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAoB,CAAC;IAC7C,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC3B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,wBAAwB;QAC3D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;YACzB,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC1B,CAAC;QACD,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,IAAI,EAAE,EAAE;QAClC,MAAM,CAAC,IAAI,CAAC;YACV,IAAI;YACJ,SAAS,EAAE,QAAQ,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM;YACzC,QAAQ;SACT,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,YAAY,CAAC,OAAgB,EAAE,SAAkB;IACxD,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;QAC7B,MAAM,QAAQ,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,CAAC,CAAC;QACtE,IAAI,QAAQ,EAAE,CAAC;YACb,QAAQ,CAAC,SAAS,GAAG,CAAC,QAAQ,CAAC,SAAS,GAAG,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACnE,QAAQ,CAAC,QAAQ,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,QAAQ,CAAC,QAAQ,EAAE,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QACjF,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,kBAAkB;IAClB,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;AACzF,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAa,EAAE,OAAgB;IAC9D,MAAM,YAAY,GAAa,CAAC,KAAK,CAAC,CAAC;IAEvC,sBAAsB;IACtB,MAAM,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACpE,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,YAAY,CAAC,IAAI,CAAC,qBAAqB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACnE,CAAC;IAED,oBAAoB;IACpB,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;IACvE,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzB,YAAY,CAAC,IAAI,CAAC,mBAAmB,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAClC,CAAC;AAED,SAAgB,UAAU,CAAC,MAAc,EAAE,WAA4B;IACrE,OAAO,OAAO,CAAC,WAAW,CAAC,oBAAoB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;AACxE,CAAC;AAED,SAAgB,YAAY,CAAC,SAAiB;IAC5C,YAAY,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;AACjC,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Corrective RAG (CRAG) Decorator
|
|
3
|
+
* Self-corrects retrieval errors with fallback mechanisms
|
|
4
|
+
*/
|
|
5
|
+
import 'reflect-metadata';
|
|
6
|
+
import { CorrectiveRAGResult, AgenticLLMProvider } from '../types';
|
|
7
|
+
import '../../prompts/agentic/corrective-rag.prompt';
|
|
8
|
+
export interface CorrectiveRAGConfig {
|
|
9
|
+
relevanceThreshold?: number;
|
|
10
|
+
fallbackToWeb?: boolean;
|
|
11
|
+
maxCorrections?: number;
|
|
12
|
+
llmProvider?: AgenticLLMProvider;
|
|
13
|
+
}
|
|
14
|
+
export declare function CorrectiveRAG(config?: CorrectiveRAGConfig): MethodDecorator;
|
|
15
|
+
export declare function getCorrectiveRAGResult(target: object, propertyKey: string | symbol): CorrectiveRAGResult | undefined;
|
|
16
|
+
//# sourceMappingURL=corrective-rag.decorator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"corrective-rag.decorator.d.ts","sourceRoot":"","sources":["../../../src/agentic/decorators/corrective-rag.decorator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,kBAAkB,CAAC;AAC1B,OAAO,EAAE,mBAAmB,EAAc,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAE/E,OAAO,6CAA6C,CAAC;AAGrD,MAAM,WAAW,mBAAmB;IAClC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,CAAC,EAAE,kBAAkB,CAAC;CAClC;AAID,wBAAgB,aAAa,CAAC,MAAM,GAAE,mBAAwB,GAAG,eAAe,CAgD/E;AAgHD,wBAAgB,sBAAsB,CACpC,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,MAAM,GAAG,MAAM,GAC3B,mBAAmB,GAAG,SAAS,CAEjC"}
|
|
@@ -0,0 +1,142 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Corrective RAG (CRAG) Decorator
|
|
4
|
+
* Self-corrects retrieval errors with fallback mechanisms
|
|
5
|
+
*/
|
|
6
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
7
|
+
exports.CorrectiveRAG = CorrectiveRAG;
|
|
8
|
+
exports.getCorrectiveRAGResult = getCorrectiveRAGResult;
|
|
9
|
+
require("reflect-metadata");
|
|
10
|
+
const prompts_1 = require("@hazeljs/prompts");
|
|
11
|
+
require("../../prompts/agentic/corrective-rag.prompt");
|
|
12
|
+
const corrective_rag_prompt_1 = require("../../prompts/agentic/corrective-rag.prompt");
|
|
13
|
+
const CRAG_METADATA_KEY = Symbol('correctiveRAG');
|
|
14
|
+
function CorrectiveRAG(config = {}) {
|
|
15
|
+
return function (target, propertyKey, descriptor) {
|
|
16
|
+
const originalMethod = descriptor.value;
|
|
17
|
+
descriptor.value = async function (...args) {
|
|
18
|
+
let results = await originalMethod.apply(this, args);
|
|
19
|
+
const query = args[0];
|
|
20
|
+
// Evaluate results
|
|
21
|
+
const evaluation = await evaluateResults(results, query, config);
|
|
22
|
+
const corrections = [];
|
|
23
|
+
let fallbackUsed = false;
|
|
24
|
+
// Apply corrections if needed
|
|
25
|
+
if (evaluation.needsCorrection) {
|
|
26
|
+
for (const issue of evaluation.issues) {
|
|
27
|
+
corrections.push({
|
|
28
|
+
type: issue.type,
|
|
29
|
+
description: issue.description,
|
|
30
|
+
action: issue.action,
|
|
31
|
+
});
|
|
32
|
+
}
|
|
33
|
+
// Attempt to correct
|
|
34
|
+
const corrected = await correctResults(results, query, evaluation, config);
|
|
35
|
+
if (corrected) {
|
|
36
|
+
results = corrected;
|
|
37
|
+
}
|
|
38
|
+
else if (config.fallbackToWeb) {
|
|
39
|
+
// Fallback mechanism (placeholder)
|
|
40
|
+
fallbackUsed = true;
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
const cragResult = {
|
|
44
|
+
results: Array.isArray(results) ? results : [results],
|
|
45
|
+
corrections,
|
|
46
|
+
fallbackUsed,
|
|
47
|
+
confidence: evaluation.confidence,
|
|
48
|
+
};
|
|
49
|
+
Reflect.defineMetadata(CRAG_METADATA_KEY, cragResult, target, propertyKey);
|
|
50
|
+
return results;
|
|
51
|
+
};
|
|
52
|
+
return descriptor;
|
|
53
|
+
};
|
|
54
|
+
}
|
|
55
|
+
async function evaluateResults(
|
|
56
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
57
|
+
results, query, config) {
|
|
58
|
+
const threshold = config.relevanceThreshold || 0.7;
|
|
59
|
+
const issues = [];
|
|
60
|
+
if (!results || (Array.isArray(results) && results.length === 0)) {
|
|
61
|
+
issues.push({
|
|
62
|
+
type: 'missing_info',
|
|
63
|
+
description: 'No results returned',
|
|
64
|
+
action: 'Expand query or use fallback',
|
|
65
|
+
});
|
|
66
|
+
return { needsCorrection: true, confidence: 0, issues };
|
|
67
|
+
}
|
|
68
|
+
const resultArray = Array.isArray(results) ? results : [results];
|
|
69
|
+
// Check relevance scores
|
|
70
|
+
const avgScore = resultArray.reduce((sum, r) => sum + (r.score || 0), 0) / resultArray.length;
|
|
71
|
+
if (avgScore < threshold) {
|
|
72
|
+
issues.push({
|
|
73
|
+
type: 'low_relevance',
|
|
74
|
+
description: `Average relevance score ${avgScore.toFixed(2)} below threshold ${threshold}`,
|
|
75
|
+
action: 'Rewrite query or adjust retrieval strategy',
|
|
76
|
+
});
|
|
77
|
+
}
|
|
78
|
+
// Check for contradictions
|
|
79
|
+
if (resultArray.length > 1 && config.llmProvider) {
|
|
80
|
+
const hasContradictions = await checkContradictions(resultArray, config.llmProvider);
|
|
81
|
+
if (hasContradictions) {
|
|
82
|
+
issues.push({
|
|
83
|
+
type: 'contradictory',
|
|
84
|
+
description: 'Results contain contradictory information',
|
|
85
|
+
action: 'Filter or re-rank results',
|
|
86
|
+
});
|
|
87
|
+
}
|
|
88
|
+
}
|
|
89
|
+
return {
|
|
90
|
+
needsCorrection: issues.length > 0,
|
|
91
|
+
confidence: Math.max(avgScore, 0.5),
|
|
92
|
+
issues,
|
|
93
|
+
};
|
|
94
|
+
}
|
|
95
|
+
async function correctResults(
|
|
96
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
97
|
+
results, query,
|
|
98
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
99
|
+
evaluation, config
|
|
100
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
101
|
+
) {
|
|
102
|
+
if (!config.llmProvider) {
|
|
103
|
+
return null;
|
|
104
|
+
}
|
|
105
|
+
const prompt = prompts_1.PromptRegistry.get(corrective_rag_prompt_1.CORRECTIVE_RAG_KEY).render({
|
|
106
|
+
query,
|
|
107
|
+
issues: evaluation.issues.map((i) => i.description).join(', '),
|
|
108
|
+
});
|
|
109
|
+
try {
|
|
110
|
+
await config.llmProvider.generateStructured(prompt, {});
|
|
111
|
+
// Return null to signal re-query needed
|
|
112
|
+
return null;
|
|
113
|
+
}
|
|
114
|
+
catch {
|
|
115
|
+
return null;
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
async function checkContradictions(
|
|
119
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
120
|
+
results, llmProvider) {
|
|
121
|
+
if (results.length < 2)
|
|
122
|
+
return false;
|
|
123
|
+
const contents = results
|
|
124
|
+
.slice(0, 3)
|
|
125
|
+
.map((r) => r.content)
|
|
126
|
+
.join('\n\n');
|
|
127
|
+
const prompt = `Do these texts contain contradictory information? Answer yes or no.
|
|
128
|
+
|
|
129
|
+
Texts:
|
|
130
|
+
${contents.slice(0, 1000)}`;
|
|
131
|
+
try {
|
|
132
|
+
const response = await llmProvider.generate(prompt, { maxTokens: 10 });
|
|
133
|
+
return response.toLowerCase().includes('yes');
|
|
134
|
+
}
|
|
135
|
+
catch {
|
|
136
|
+
return false;
|
|
137
|
+
}
|
|
138
|
+
}
|
|
139
|
+
function getCorrectiveRAGResult(target, propertyKey) {
|
|
140
|
+
return Reflect.getMetadata(CRAG_METADATA_KEY, target, propertyKey);
|
|
141
|
+
}
|
|
142
|
+
//# sourceMappingURL=corrective-rag.decorator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"corrective-rag.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/corrective-rag.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAiBH,sCAgDC;AAgHD,wDAKC;AApLD,4BAA0B;AAE1B,8CAAkD;AAClD,uDAAqD;AACrD,uFAAiF;AASjF,MAAM,iBAAiB,GAAG,MAAM,CAAC,eAAe,CAAC,CAAC;AAElD,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,IAAI,OAAO,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACrD,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;YAEhC,mBAAmB;YACnB,MAAM,UAAU,GAAG,MAAM,eAAe,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;YAEjE,MAAM,WAAW,GAAiB,EAAE,CAAC;YACrC,IAAI,YAAY,GAAG,KAAK,CAAC;YAEzB,8BAA8B;YAC9B,IAAI,UAAU,CAAC,eAAe,EAAE,CAAC;gBAC/B,KAAK,MAAM,KAAK,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;oBACtC,WAAW,CAAC,IAAI,CAAC;wBACf,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,WAAW,EAAE,KAAK,CAAC,WAAW;wBAC9B,MAAM,EAAE,KAAK,CAAC,MAAM;qBACrB,CAAC,CAAC;gBACL,CAAC;gBAED,qBAAqB;gBACrB,MAAM,SAAS,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;gBAC3E,IAAI,SAAS,EAAE,CAAC;oBACd,OAAO,GAAG,SAAS,CAAC;gBACtB,CAAC;qBAAM,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;oBAChC,mCAAmC;oBACnC,YAAY,GAAG,IAAI,CAAC;gBACtB,CAAC;YACH,CAAC;YAED,MAAM,UAAU,GAAwB;gBACtC,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBACrD,WAAW;gBACX,YAAY;gBACZ,UAAU,EAAE,UAAU,CAAC,UAAU;aAClC,CAAC;YAEF,OAAO,CAAC,cAAc,CAAC,iBAAiB,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAE3E,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,eAAe;AAC5B,8DAA8D;AAC9D,OAAY,EACZ,KAAa,EACb,MAA2B;IAM3B,MAAM,SAAS,GAAG,MAAM,CAAC,kBAAkB,IAAI,GAAG,CAAC;IACnD,MAAM,MAAM,GAA6E,EAAE,CAAC;IAE5F,IAAI,CAAC,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC,EAAE,CAAC;QACjE,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,cAAc;YACpB,WAAW,EAAE,qBAAqB;YAClC,MAAM,EAAE,8BAA8B;SACvC,CAAC,CAAC;QACH,OAAO,EAAE,eAAe,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1D,CAAC;IAED,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAEjE,yBAAyB;IACzB,MAAM,QAAQ,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,WAAW,CAAC,MAAM,CAAC;IAE9F,IAAI,QAAQ,GAAG,SAAS,EAAE,CAAC;QACzB,MAAM,CAAC,IAAI,CAAC;YACV,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,2BAA2B,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,oBAAoB,SAAS,EAAE;YAC1F,MAAM,EAAE,4CAA4C;SACrD,CAAC,CAAC;IACL,CAAC;IAED,2BAA2B;IAC3B,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QACjD,MAAM,iBAAiB,GAAG,MAAM,mBAAmB,CAAC,WAAW,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;QACrF,IAAI,iBAAiB,EAAE,CAAC;YACtB,MAAM,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,eAAe;gBACrB,WAAW,EAAE,2CAA2C;gBACxD,MAAM,EAAE,2BAA2B;aACpC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO;QACL,eAAe,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC;QAClC,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC;QACnC,MAAM;KACP,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,cAAc;AAC3B,8DAA8D;AAC9D,OAAY,EACZ,KAAa;AACb,8DAA8D;AAC9D,UAAe,EACf,MAA2B;AAC3B,8DAA8D;;IAE9D,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;QACxB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAG,wBAAc,CAAC,GAAG,CAAoC,0CAAkB,CAAC,CAAC,MAAM,CAAC;QAC9F,KAAK;QACL,MAAM,EAAE,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAA0B,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;KACxF,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,WAAW,CAAC,kBAAkB,CAIxC,MAAM,EAAE,EAAE,CAAC,CAAC;QAEf,wCAAwC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,KAAK,UAAU,mBAAmB;AAChC,8DAA8D;AAC9D,OAAc,EACd,WAA+B;IAE/B,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,KAAK,CAAC;IAErC,MAAM,QAAQ,GAAG,OAAO;SACrB,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;SACX,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;SACrB,IAAI,CAAC,MAAM,CAAC,CAAC;IAChB,MAAM,MAAM,GAAG;;;EAGf,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC;IAE1B,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;QACvE,OAAO,QAAQ,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAChD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED,SAAgB,sBAAsB,CACpC,MAAc,EACd,WAA4B;IAE5B,OAAO,OAAO,CAAC,WAAW,CAAC,iBAAiB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;AACrE,CAAC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* HyDE (Hypothetical Document Embeddings) Decorator
|
|
3
|
+
* Generates hypothetical documents to improve retrieval
|
|
4
|
+
*/
|
|
5
|
+
import 'reflect-metadata';
|
|
6
|
+
import { HyDEResult, AgenticLLMProvider } from '../types';
|
|
7
|
+
import '../../prompts/agentic/hyde.prompt';
|
|
8
|
+
export interface HyDEConfig {
|
|
9
|
+
generateHypothesis?: boolean;
|
|
10
|
+
numHypotheses?: number;
|
|
11
|
+
llmProvider?: AgenticLLMProvider;
|
|
12
|
+
}
|
|
13
|
+
export declare function HyDE(config?: HyDEConfig): MethodDecorator;
|
|
14
|
+
export declare function getHyDEResult(target: object, propertyKey: string | symbol): HyDEResult | undefined;
|
|
15
|
+
//# sourceMappingURL=hyde.decorator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hyde.decorator.d.ts","sourceRoot":"","sources":["../../../src/agentic/decorators/hyde.decorator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,kBAAkB,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAE1D,OAAO,mCAAmC,CAAC;AAG3C,MAAM,WAAW,UAAU;IACzB,kBAAkB,CAAC,EAAE,OAAO,CAAC;IAC7B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,WAAW,CAAC,EAAE,kBAAkB,CAAC;CAClC;AAID,wBAAgB,IAAI,CAAC,MAAM,GAAE,UAAe,GAAG,eAAe,CAmD7D;AAgDD,wBAAgB,aAAa,CAC3B,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,MAAM,GAAG,MAAM,GAC3B,UAAU,GAAG,SAAS,CAExB"}
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* HyDE (Hypothetical Document Embeddings) Decorator
|
|
4
|
+
* Generates hypothetical documents to improve retrieval
|
|
5
|
+
*/
|
|
6
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
7
|
+
exports.HyDE = HyDE;
|
|
8
|
+
exports.getHyDEResult = getHyDEResult;
|
|
9
|
+
require("reflect-metadata");
|
|
10
|
+
const prompts_1 = require("@hazeljs/prompts");
|
|
11
|
+
require("../../prompts/agentic/hyde.prompt");
|
|
12
|
+
const hyde_prompt_1 = require("../../prompts/agentic/hyde.prompt");
|
|
13
|
+
const HYDE_METADATA_KEY = Symbol('hyde');
|
|
14
|
+
function HyDE(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.generateHypothesis || !config.llmProvider) {
|
|
20
|
+
return originalMethod.apply(this, args);
|
|
21
|
+
}
|
|
22
|
+
// Generate hypothetical documents
|
|
23
|
+
const hypotheses = await generateHypotheticalDocuments(query, config.numHypotheses || 3, config.llmProvider);
|
|
24
|
+
// Retrieve using each hypothesis
|
|
25
|
+
const allResults = [];
|
|
26
|
+
const seenIds = new Set();
|
|
27
|
+
for (const hypothesis of hypotheses) {
|
|
28
|
+
const results = await originalMethod.apply(this, [hypothesis, ...args.slice(1)]);
|
|
29
|
+
if (Array.isArray(results)) {
|
|
30
|
+
for (const result of results) {
|
|
31
|
+
if (result.id && !seenIds.has(result.id)) {
|
|
32
|
+
seenIds.add(result.id);
|
|
33
|
+
allResults.push(result);
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
// Re-rank by aggregated scores
|
|
39
|
+
const reranked = rerankResults(allResults, hypotheses.length);
|
|
40
|
+
const hydeResult = {
|
|
41
|
+
hypotheticalDocuments: hypotheses,
|
|
42
|
+
retrievedResults: reranked,
|
|
43
|
+
aggregatedScore: calculateAggregatedScore(reranked),
|
|
44
|
+
};
|
|
45
|
+
Reflect.defineMetadata(HYDE_METADATA_KEY, hydeResult, target, propertyKey);
|
|
46
|
+
return reranked;
|
|
47
|
+
};
|
|
48
|
+
return descriptor;
|
|
49
|
+
};
|
|
50
|
+
}
|
|
51
|
+
async function generateHypotheticalDocuments(query, count, llmProvider) {
|
|
52
|
+
const prompt = prompts_1.PromptRegistry.get(hyde_prompt_1.HYDE_KEY).render({
|
|
53
|
+
count: String(count),
|
|
54
|
+
query,
|
|
55
|
+
});
|
|
56
|
+
try {
|
|
57
|
+
const response = await llmProvider.generate(prompt, {
|
|
58
|
+
temperature: 0.7,
|
|
59
|
+
maxTokens: 1000,
|
|
60
|
+
});
|
|
61
|
+
const documents = response
|
|
62
|
+
.split('\n')
|
|
63
|
+
.map((line) => line.replace(/^\d+\.\s*/, '').trim())
|
|
64
|
+
.filter((line) => line.length > 50)
|
|
65
|
+
.slice(0, count);
|
|
66
|
+
return documents.length > 0 ? documents : [query];
|
|
67
|
+
}
|
|
68
|
+
catch {
|
|
69
|
+
return [query];
|
|
70
|
+
}
|
|
71
|
+
}
|
|
72
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
73
|
+
function rerankResults(results, hypothesisCount) {
|
|
74
|
+
return results
|
|
75
|
+
.map((result) => ({
|
|
76
|
+
...result,
|
|
77
|
+
score: (result.score || 0) * (1 + 0.1 * hypothesisCount),
|
|
78
|
+
}))
|
|
79
|
+
.sort((a, b) => (b.score || 0) - (a.score || 0));
|
|
80
|
+
}
|
|
81
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
82
|
+
function calculateAggregatedScore(results) {
|
|
83
|
+
if (results.length === 0)
|
|
84
|
+
return 0;
|
|
85
|
+
const totalScore = results.reduce((sum, r) => sum + (r.score || 0), 0);
|
|
86
|
+
return totalScore / results.length;
|
|
87
|
+
}
|
|
88
|
+
function getHyDEResult(target, propertyKey) {
|
|
89
|
+
return Reflect.getMetadata(HYDE_METADATA_KEY, target, propertyKey);
|
|
90
|
+
}
|
|
91
|
+
//# sourceMappingURL=hyde.decorator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hyde.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/hyde.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAgBH,oBAmDC;AAgDD,sCAKC;AAtHD,4BAA0B;AAE1B,8CAAkD;AAClD,6CAA2C;AAC3C,mEAA6D;AAQ7D,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;AAEzC,SAAgB,IAAI,CAAC,SAAqB,EAAE;IAC1C,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,kBAAkB,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;gBACtD,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC1C,CAAC;YAED,kCAAkC;YAClC,MAAM,UAAU,GAAG,MAAM,6BAA6B,CACpD,KAAK,EACL,MAAM,CAAC,aAAa,IAAI,CAAC,EACzB,MAAM,CAAC,WAAW,CACnB,CAAC;YAEF,iCAAiC;YACjC,MAAM,UAAU,GAAG,EAAE,CAAC;YACtB,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;YAElC,KAAK,MAAM,UAAU,IAAI,UAAU,EAAE,CAAC;gBACpC,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,UAAU,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAEjF,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,+BAA+B;YAC/B,MAAM,QAAQ,GAAG,aAAa,CAAC,UAAU,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;YAE9D,MAAM,UAAU,GAAe;gBAC7B,qBAAqB,EAAE,UAAU;gBACjC,gBAAgB,EAAE,QAAQ;gBAC1B,eAAe,EAAE,wBAAwB,CAAC,QAAQ,CAAC;aACpD,CAAC;YAEF,OAAO,CAAC,cAAc,CAAC,iBAAiB,EAAE,UAAU,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAE3E,OAAO,QAAQ,CAAC;QAClB,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,6BAA6B,CAC1C,KAAa,EACb,KAAa,EACb,WAA+B;IAE/B,MAAM,MAAM,GAAG,wBAAc,CAAC,GAAG,CAAmC,sBAAQ,CAAC,CAAC,MAAM,CAAC;QACnF,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;QACpB,KAAK;KACN,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,QAAQ,CAAC,MAAM,EAAE;YAClD,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,QAAQ;aACvB,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,EAAE,CAAC;aAClC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QAEnB,OAAO,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IACpD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,KAAK,CAAC,CAAC;IACjB,CAAC;AACH,CAAC;AAED,8DAA8D;AAC9D,SAAS,aAAa,CAAC,OAAc,EAAE,eAAuB;IAC5D,OAAO,OAAO;SACX,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QAChB,GAAG,MAAM;QACT,KAAK,EAAE,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,GAAG,GAAG,eAAe,CAAC;KACzD,CAAC,CAAC;SACF,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC;AACrD,CAAC;AAED,8DAA8D;AAC9D,SAAS,wBAAwB,CAAC,OAAc;IAC9C,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAC;IAEnC,MAAM,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACvE,OAAO,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC;AACrC,CAAC;AAED,SAAgB,aAAa,CAC3B,MAAc,EACd,WAA4B;IAE5B,OAAO,OAAO,CAAC,WAAW,CAAC,iBAAiB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;AACrE,CAAC"}
|
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Agentic RAG Decorators
|
|
3
|
+
* Export all decorators
|
|
4
|
+
*/
|
|
5
|
+
export * from './query-planner.decorator';
|
|
6
|
+
export * from './self-reflective.decorator';
|
|
7
|
+
export * from './adaptive-retrieval.decorator';
|
|
8
|
+
export * from './multi-hop.decorator';
|
|
9
|
+
export * from './hyde.decorator';
|
|
10
|
+
export * from './corrective-rag.decorator';
|
|
11
|
+
export * from './context-aware.decorator';
|
|
12
|
+
export * from './query-rewriter.decorator';
|
|
13
|
+
export * from './source-verification.decorator';
|
|
14
|
+
export * from './active-learning.decorator';
|
|
15
|
+
export * from './cached.decorator';
|
|
16
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/agentic/decorators/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,cAAc,2BAA2B,CAAC;AAC1C,cAAc,6BAA6B,CAAC;AAC5C,cAAc,gCAAgC,CAAC;AAC/C,cAAc,uBAAuB,CAAC;AACtC,cAAc,kBAAkB,CAAC;AACjC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,2BAA2B,CAAC;AAC1C,cAAc,4BAA4B,CAAC;AAC3C,cAAc,iCAAiC,CAAC;AAChD,cAAc,6BAA6B,CAAC;AAC5C,cAAc,oBAAoB,CAAC"}
|
|
@@ -0,0 +1,32 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Agentic RAG Decorators
|
|
4
|
+
* Export all decorators
|
|
5
|
+
*/
|
|
6
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
7
|
+
if (k2 === undefined) k2 = k;
|
|
8
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
9
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
10
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
11
|
+
}
|
|
12
|
+
Object.defineProperty(o, k2, desc);
|
|
13
|
+
}) : (function(o, m, k, k2) {
|
|
14
|
+
if (k2 === undefined) k2 = k;
|
|
15
|
+
o[k2] = m[k];
|
|
16
|
+
}));
|
|
17
|
+
var __exportStar = (this && this.__exportStar) || function(m, exports) {
|
|
18
|
+
for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
|
|
19
|
+
};
|
|
20
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
21
|
+
__exportStar(require("./query-planner.decorator"), exports);
|
|
22
|
+
__exportStar(require("./self-reflective.decorator"), exports);
|
|
23
|
+
__exportStar(require("./adaptive-retrieval.decorator"), exports);
|
|
24
|
+
__exportStar(require("./multi-hop.decorator"), exports);
|
|
25
|
+
__exportStar(require("./hyde.decorator"), exports);
|
|
26
|
+
__exportStar(require("./corrective-rag.decorator"), exports);
|
|
27
|
+
__exportStar(require("./context-aware.decorator"), exports);
|
|
28
|
+
__exportStar(require("./query-rewriter.decorator"), exports);
|
|
29
|
+
__exportStar(require("./source-verification.decorator"), exports);
|
|
30
|
+
__exportStar(require("./active-learning.decorator"), exports);
|
|
31
|
+
__exportStar(require("./cached.decorator"), exports);
|
|
32
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agentic/decorators/index.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;;;;;;;;;;;;;;AAEH,4DAA0C;AAC1C,8DAA4C;AAC5C,iEAA+C;AAC/C,wDAAsC;AACtC,mDAAiC;AACjC,6DAA2C;AAC3C,4DAA0C;AAC1C,6DAA2C;AAC3C,kEAAgD;AAChD,8DAA4C;AAC5C,qDAAmC"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Multi-Hop Decorator
|
|
3
|
+
* Performs multi-hop reasoning across documents
|
|
4
|
+
*/
|
|
5
|
+
import 'reflect-metadata';
|
|
6
|
+
import { ReasoningChain, AgenticLLMProvider } from '../types';
|
|
7
|
+
import '../../prompts/agentic/multi-hop.prompt';
|
|
8
|
+
export interface MultiHopConfig {
|
|
9
|
+
maxHops?: number;
|
|
10
|
+
strategy?: 'breadth-first' | 'depth-first' | 'adaptive';
|
|
11
|
+
llmProvider?: AgenticLLMProvider;
|
|
12
|
+
}
|
|
13
|
+
export declare function MultiHop(config?: MultiHopConfig): MethodDecorator;
|
|
14
|
+
export declare function getReasoningChain(target: object, propertyKey: string | symbol): ReasoningChain | undefined;
|
|
15
|
+
//# sourceMappingURL=multi-hop.decorator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"multi-hop.decorator.d.ts","sourceRoot":"","sources":["../../../src/agentic/decorators/multi-hop.decorator.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,kBAAkB,CAAC;AAC1B,OAAO,EAAE,cAAc,EAAgB,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAE5E,OAAO,wCAAwC,CAAC;AAGhD,MAAM,WAAW,cAAc;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,eAAe,GAAG,aAAa,GAAG,UAAU,CAAC;IACxD,WAAW,CAAC,EAAE,kBAAkB,CAAC;CAClC;AAID,wBAAgB,QAAQ,CAAC,MAAM,GAAE,cAAmB,GAAG,eAAe,CAuDrE;AAsED,wBAAgB,iBAAiB,CAC/B,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,MAAM,GAAG,MAAM,GAC3B,cAAc,GAAG,SAAS,CAE5B"}
|