@hazeljs/rag 0.2.0-beta.8 → 0.2.0-beta.81
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 -21
- package/README.md +348 -223
- 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/decorators/adaptive-retrieval.decorator.d.ts +1 -0
- package/dist/agentic/decorators/adaptive-retrieval.decorator.d.ts.map +1 -1
- package/dist/agentic/decorators/adaptive-retrieval.decorator.js +4 -15
- package/dist/agentic/decorators/adaptive-retrieval.decorator.js.map +1 -1
- package/dist/agentic/decorators/corrective-rag.decorator.d.ts +1 -0
- package/dist/agentic/decorators/corrective-rag.decorator.d.ts.map +1 -1
- package/dist/agentic/decorators/corrective-rag.decorator.js +7 -11
- package/dist/agentic/decorators/corrective-rag.decorator.js.map +1 -1
- package/dist/agentic/decorators/hyde.decorator.d.ts +1 -0
- package/dist/agentic/decorators/hyde.decorator.d.ts.map +1 -1
- package/dist/agentic/decorators/hyde.decorator.js +7 -6
- package/dist/agentic/decorators/hyde.decorator.js.map +1 -1
- package/dist/agentic/decorators/multi-hop.decorator.d.ts +1 -0
- package/dist/agentic/decorators/multi-hop.decorator.d.ts.map +1 -1
- package/dist/agentic/decorators/multi-hop.decorator.js +8 -19
- package/dist/agentic/decorators/multi-hop.decorator.js.map +1 -1
- package/dist/agentic/decorators/query-planner.decorator.d.ts +1 -0
- package/dist/agentic/decorators/query-planner.decorator.d.ts.map +1 -1
- package/dist/agentic/decorators/query-planner.decorator.js +4 -18
- package/dist/agentic/decorators/query-planner.decorator.js.map +1 -1
- package/dist/agentic/decorators/query-rewriter.decorator.d.ts +1 -0
- package/dist/agentic/decorators/query-rewriter.decorator.d.ts.map +1 -1
- package/dist/agentic/decorators/query-rewriter.decorator.js +8 -5
- package/dist/agentic/decorators/query-rewriter.decorator.js.map +1 -1
- package/dist/agentic/decorators/self-reflective.decorator.d.ts +2 -0
- package/dist/agentic/decorators/self-reflective.decorator.d.ts.map +1 -1
- package/dist/agentic/decorators/self-reflective.decorator.js +11 -32
- package/dist/agentic/decorators/self-reflective.decorator.js.map +1 -1
- package/dist/agentic/index.d.ts +1 -1
- package/dist/agentic/index.d.ts.map +1 -1
- package/dist/agentic/types.d.ts +3 -2
- package/dist/agentic/types.d.ts.map +1 -1
- 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 +2 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +4 -0
- package/dist/index.js.map +1 -1
- 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/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.service.d.ts +1 -0
- package/dist/rag.service.d.ts.map +1 -1
- package/dist/rag.service.js +8 -9
- package/dist/rag.service.js.map +1 -1
- package/dist/vector-stores/qdrant.store.d.ts +2 -0
- package/dist/vector-stores/qdrant.store.d.ts.map +1 -1
- package/dist/vector-stores/qdrant.store.js +1 -0
- package/dist/vector-stores/qdrant.store.js.map +1 -1
- package/package.json +64 -6
|
@@ -7,6 +7,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
7
7
|
exports.CorrectiveRAG = CorrectiveRAG;
|
|
8
8
|
exports.getCorrectiveRAGResult = getCorrectiveRAGResult;
|
|
9
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");
|
|
10
13
|
const CRAG_METADATA_KEY = Symbol('correctiveRAG');
|
|
11
14
|
function CorrectiveRAG(config = {}) {
|
|
12
15
|
return function (target, propertyKey, descriptor) {
|
|
@@ -99,17 +102,10 @@ evaluation, config
|
|
|
99
102
|
if (!config.llmProvider) {
|
|
100
103
|
return null;
|
|
101
104
|
}
|
|
102
|
-
const prompt =
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
Provide corrected query or strategy in JSON:
|
|
108
|
-
{
|
|
109
|
-
"correctedQuery": "improved query",
|
|
110
|
-
"strategy": "new strategy",
|
|
111
|
-
"reasoning": "why this helps"
|
|
112
|
-
}`;
|
|
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
|
+
});
|
|
113
109
|
try {
|
|
114
110
|
await config.llmProvider.generateStructured(prompt, {});
|
|
115
111
|
// Return null to signal re-query needed
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"corrective-rag.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/corrective-rag.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;
|
|
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"}
|
|
@@ -1 +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,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;
|
|
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"}
|
|
@@ -7,6 +7,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
7
7
|
exports.HyDE = HyDE;
|
|
8
8
|
exports.getHyDEResult = getHyDEResult;
|
|
9
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");
|
|
10
13
|
const HYDE_METADATA_KEY = Symbol('hyde');
|
|
11
14
|
function HyDE(config = {}) {
|
|
12
15
|
return function (target, propertyKey, descriptor) {
|
|
@@ -46,12 +49,10 @@ function HyDE(config = {}) {
|
|
|
46
49
|
};
|
|
47
50
|
}
|
|
48
51
|
async function generateHypotheticalDocuments(query, count, llmProvider) {
|
|
49
|
-
const prompt =
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
Generate ${count} hypothetical documents (one per line):`;
|
|
52
|
+
const prompt = prompts_1.PromptRegistry.get(hyde_prompt_1.HYDE_KEY).render({
|
|
53
|
+
count: String(count),
|
|
54
|
+
query,
|
|
55
|
+
});
|
|
55
56
|
try {
|
|
56
57
|
const response = await llmProvider.generate(prompt, {
|
|
57
58
|
temperature: 0.7,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"hyde.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/hyde.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;
|
|
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"}
|
|
@@ -1 +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,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;
|
|
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"}
|
|
@@ -7,6 +7,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
7
7
|
exports.MultiHop = MultiHop;
|
|
8
8
|
exports.getReasoningChain = getReasoningChain;
|
|
9
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");
|
|
10
13
|
const MULTIHOP_METADATA_KEY = Symbol('multiHop');
|
|
11
14
|
function MultiHop(config = {}) {
|
|
12
15
|
return function (target, propertyKey, descriptor) {
|
|
@@ -61,17 +64,10 @@ results, config) {
|
|
|
61
64
|
};
|
|
62
65
|
}
|
|
63
66
|
const resultsStr = JSON.stringify(results, null, 2).slice(0, 1500);
|
|
64
|
-
const prompt =
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
Provide analysis in JSON:
|
|
70
|
-
{
|
|
71
|
-
"reasoning": "what we learned from these results",
|
|
72
|
-
"nextQuery": "follow-up query if needed, or null",
|
|
73
|
-
"shouldStop": true/false
|
|
74
|
-
}`;
|
|
67
|
+
const prompt = prompts_1.PromptRegistry.get(multi_hop_prompt_1.MULTI_HOP_KEY).render({
|
|
68
|
+
query,
|
|
69
|
+
resultsStr,
|
|
70
|
+
});
|
|
75
71
|
try {
|
|
76
72
|
return await config.llmProvider.generateStructured(prompt, {});
|
|
77
73
|
}
|
|
@@ -87,14 +83,7 @@ async function synthesizeFinalAnswer(chain, config) {
|
|
|
87
83
|
return chain.hops.map((h) => h.reasoning).join('\n\n');
|
|
88
84
|
}
|
|
89
85
|
const hopsStr = chain.hops.map((h) => `Hop ${h.hopNumber}: ${h.reasoning}`).join('\n');
|
|
90
|
-
const prompt =
|
|
91
|
-
|
|
92
|
-
Original Query: ${chain.query}
|
|
93
|
-
|
|
94
|
-
Reasoning Chain:
|
|
95
|
-
${hopsStr}
|
|
96
|
-
|
|
97
|
-
Provide a comprehensive final answer:`;
|
|
86
|
+
const prompt = prompts_1.PromptRegistry.get(multi_hop_prompt_1.MULTI_HOP_SYNTHESIZE_KEY).render({ originalQuery: chain.query, hopsStr });
|
|
98
87
|
try {
|
|
99
88
|
return await config.llmProvider.generate(prompt);
|
|
100
89
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"multi-hop.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/multi-hop.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;
|
|
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"}
|
|
@@ -1 +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,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;
|
|
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"}
|
|
@@ -7,6 +7,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
7
7
|
exports.QueryPlanner = QueryPlanner;
|
|
8
8
|
exports.getQueryPlan = getQueryPlan;
|
|
9
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");
|
|
10
13
|
const QUERY_PLANNER_METADATA_KEY = Symbol('queryPlanner');
|
|
11
14
|
function QueryPlanner(config = {}) {
|
|
12
15
|
return function (target, propertyKey, descriptor) {
|
|
@@ -64,24 +67,7 @@ async function decomposeQuery(query, config) {
|
|
|
64
67
|
* LLM-based query decomposition
|
|
65
68
|
*/
|
|
66
69
|
async function decomposeWithLLM(query, config) {
|
|
67
|
-
const prompt =
|
|
68
|
-
Each sub-query should be independent and focused on a specific aspect.
|
|
69
|
-
|
|
70
|
-
Query: ${query}
|
|
71
|
-
|
|
72
|
-
Provide the decomposition in the following JSON format:
|
|
73
|
-
{
|
|
74
|
-
"subQueries": [
|
|
75
|
-
{
|
|
76
|
-
"id": "1",
|
|
77
|
-
"query": "sub-query text",
|
|
78
|
-
"type": "factual|analytical|comparative|temporal",
|
|
79
|
-
"dependencies": [],
|
|
80
|
-
"priority": 1
|
|
81
|
-
}
|
|
82
|
-
],
|
|
83
|
-
"strategy": "sequential|parallel"
|
|
84
|
-
}`;
|
|
70
|
+
const prompt = prompts_1.PromptRegistry.get(query_planner_prompt_1.QUERY_PLANNER_KEY).render({ query });
|
|
85
71
|
try {
|
|
86
72
|
const result = await config.llmProvider.generateStructured(prompt, {});
|
|
87
73
|
return {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"query-planner.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/query-planner.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;
|
|
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"}
|
|
@@ -4,6 +4,7 @@
|
|
|
4
4
|
*/
|
|
5
5
|
import 'reflect-metadata';
|
|
6
6
|
import { AgenticLLMProvider } from '../types';
|
|
7
|
+
import '../../prompts/agentic/query-rewriter.prompt';
|
|
7
8
|
export interface QueryRewriterConfig {
|
|
8
9
|
techniques?: ('expansion' | 'clarification' | 'synonym' | 'decomposition')[];
|
|
9
10
|
llmBased?: boolean;
|
|
@@ -1 +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,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;
|
|
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"}
|
|
@@ -7,6 +7,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
7
7
|
exports.QueryRewriter = QueryRewriter;
|
|
8
8
|
exports.getQueryVariations = getQueryVariations;
|
|
9
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");
|
|
10
13
|
const REWRITER_METADATA_KEY = Symbol('queryRewriter');
|
|
11
14
|
function QueryRewriter(config = {}) {
|
|
12
15
|
return function (target, propertyKey, descriptor) {
|
|
@@ -61,11 +64,11 @@ async function generateQueryVariations(query, config) {
|
|
|
61
64
|
return [...new Set(variations)].slice(0, maxVariations + 1);
|
|
62
65
|
}
|
|
63
66
|
async function generateWithLLM(query, techniques, llmProvider, maxVariations) {
|
|
64
|
-
const prompt =
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
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
|
+
});
|
|
69
72
|
try {
|
|
70
73
|
const response = await llmProvider.generate(prompt, { temperature: 0.7 });
|
|
71
74
|
const variations = response
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"query-rewriter.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/query-rewriter.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;
|
|
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"}
|
|
@@ -4,6 +4,8 @@
|
|
|
4
4
|
*/
|
|
5
5
|
import 'reflect-metadata';
|
|
6
6
|
import { ReflectionResult, AgenticLLMProvider } from '../types';
|
|
7
|
+
import '../../prompts/agentic/self-reflective.prompt';
|
|
8
|
+
import '../../prompts/agentic/self-reflective-improve.prompt';
|
|
7
9
|
export interface SelfReflectiveConfig {
|
|
8
10
|
maxIterations?: number;
|
|
9
11
|
qualityThreshold?: number;
|
|
@@ -1 +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,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;
|
|
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"}
|
|
@@ -7,6 +7,11 @@ Object.defineProperty(exports, "__esModule", { value: true });
|
|
|
7
7
|
exports.SelfReflective = SelfReflective;
|
|
8
8
|
exports.getReflections = getReflections;
|
|
9
9
|
require("reflect-metadata");
|
|
10
|
+
const prompts_1 = require("@hazeljs/prompts");
|
|
11
|
+
require("../../prompts/agentic/self-reflective.prompt");
|
|
12
|
+
require("../../prompts/agentic/self-reflective-improve.prompt");
|
|
13
|
+
const self_reflective_prompt_1 = require("../../prompts/agentic/self-reflective.prompt");
|
|
14
|
+
const self_reflective_improve_prompt_1 = require("../../prompts/agentic/self-reflective-improve.prompt");
|
|
10
15
|
const REFLECTION_METADATA_KEY = Symbol('reflection');
|
|
11
16
|
function SelfReflective(config = {}) {
|
|
12
17
|
return function (target, propertyKey, descriptor) {
|
|
@@ -78,26 +83,7 @@ async function assessQualityWithLLM(
|
|
|
78
83
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
79
84
|
results, query, config) {
|
|
80
85
|
const resultsStr = JSON.stringify(results, null, 2);
|
|
81
|
-
const prompt =
|
|
82
|
-
|
|
83
|
-
Query: ${query}
|
|
84
|
-
|
|
85
|
-
Results: ${resultsStr.slice(0, 2000)}
|
|
86
|
-
|
|
87
|
-
Assess the following aspects (0-1 scale):
|
|
88
|
-
1. Relevance: How relevant are the results to the query?
|
|
89
|
-
2. Completeness: Do the results fully answer the query?
|
|
90
|
-
3. Accuracy: Are the results factually accurate?
|
|
91
|
-
4. Clarity: Are the results clear and well-structured?
|
|
92
|
-
|
|
93
|
-
Provide assessment in JSON format:
|
|
94
|
-
{
|
|
95
|
-
"relevance": 0.0-1.0,
|
|
96
|
-
"completeness": 0.0-1.0,
|
|
97
|
-
"accuracy": 0.0-1.0,
|
|
98
|
-
"clarity": 0.0-1.0,
|
|
99
|
-
"issues": ["issue1", "issue2"]
|
|
100
|
-
}`;
|
|
86
|
+
const prompt = prompts_1.PromptRegistry.get(self_reflective_prompt_1.SELF_REFLECTIVE_KEY).render({ query, resultsStr: resultsStr.slice(0, 2000) });
|
|
101
87
|
try {
|
|
102
88
|
const assessment = await config.llmProvider.generateStructured(prompt, {});
|
|
103
89
|
const score = (assessment.relevance + assessment.completeness + assessment.accuracy + assessment.clarity) /
|
|
@@ -179,18 +165,11 @@ results, quality, query, config
|
|
|
179
165
|
if (!config.llmProvider) {
|
|
180
166
|
return null;
|
|
181
167
|
}
|
|
182
|
-
const prompt =
|
|
183
|
-
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
187
|
-
|
|
188
|
-
Suggest improvements in JSON format:
|
|
189
|
-
{
|
|
190
|
-
"improvedQuery": "better query",
|
|
191
|
-
"strategy": "similarity|hybrid|mmr",
|
|
192
|
-
"reasoning": "why this would be better"
|
|
193
|
-
}`;
|
|
168
|
+
const prompt = prompts_1.PromptRegistry.get(self_reflective_improve_prompt_1.SELF_REFLECTIVE_IMPROVE_KEY).render({
|
|
169
|
+
query,
|
|
170
|
+
issues: quality.issues.join(', '),
|
|
171
|
+
resultsStr: JSON.stringify(results, null, 2).slice(0, 1000),
|
|
172
|
+
});
|
|
194
173
|
try {
|
|
195
174
|
await config.llmProvider.generateStructured(prompt, {});
|
|
196
175
|
// Return null to indicate we should re-query with improvements
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"self-reflective.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/self-reflective.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;
|
|
1
|
+
{"version":3,"file":"self-reflective.decorator.js","sourceRoot":"","sources":["../../../src/agentic/decorators/self-reflective.decorator.ts"],"names":[],"mappings":";AAAA;;;GAGG;;AAmBH,wCA8DC;AAkKD,wCAEC;AAnPD,4BAA0B;AAE1B,8CAAkD;AAClD,wDAAsD;AACtD,gEAA8D;AAC9D,yFAAmF;AACnF,yGAAmG;AASnG,MAAM,uBAAuB,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;AAErD,SAAgB,cAAc,CAAC,SAA+B,EAAE;IAC9D,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,aAAa,GAAG,MAAM,CAAC,aAAa,IAAI,CAAC,CAAC;YAChD,MAAM,SAAS,GAAG,MAAM,CAAC,gBAAgB,IAAI,GAAG,CAAC;YAEjD,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,IAAI,aAAa,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC3D,IAAI,UAAU,GAAG,aAAa,CAAC;YAC/B,IAAI,WAAW,GAAG,CAAC,CAAC;YAEpB,MAAM,WAAW,GAAuB,EAAE,CAAC;YAE3C,OAAO,SAAS,GAAG,aAAa,EAAE,CAAC;gBACjC,SAAS,EAAE,CAAC;gBAEZ,iBAAiB;gBACjB,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;gBAChC,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,aAAa,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;gBAElE,WAAW,CAAC,IAAI,CAAC;oBACf,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;oBAC/C,OAAO;oBACP,YAAY,EAAE,OAAO,CAAC,MAAM;oBAC5B,UAAU,EAAE,SAAS;iBACtB,CAAC,CAAC;gBAEH,oBAAoB;gBACpB,IAAI,OAAO,CAAC,KAAK,GAAG,WAAW,EAAE,CAAC;oBAChC,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC;oBAC5B,UAAU,GAAG,aAAa,CAAC;gBAC7B,CAAC;gBAED,iCAAiC;gBACjC,IAAI,OAAO,CAAC,KAAK,IAAI,SAAS,EAAE,CAAC;oBAC/B,MAAM;gBACR,CAAC;gBAED,sBAAsB;gBACtB,IAAI,MAAM,CAAC,qBAAqB,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;oBACvD,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,aAAa,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;oBAE7E,IAAI,QAAQ,EAAE,CAAC;wBACb,aAAa,GAAG,QAAQ,CAAC;oBAC3B,CAAC;yBAAM,CAAC;wBACN,MAAM;oBACR,CAAC;gBACH,CAAC;qBAAM,CAAC;oBACN,MAAM;gBACR,CAAC;YACH,CAAC;YAED,4BAA4B;YAC5B,OAAO,CAAC,cAAc,CAAC,uBAAuB,EAAE,WAAW,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;YAElF,OAAO,UAAU,CAAC;QACpB,CAAC,CAAC;QAEF,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,aAAa;AAC1B,8DAA8D;AAC9D,OAAY,EACZ,KAAa,EACb,MAA4B;IAE5B,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;QACvB,OAAO,oBAAoB,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;IACtD,CAAC;IAED,OAAO,sBAAsB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;AAChD,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,oBAAoB;AACjC,8DAA8D;AAC9D,OAAY,EACZ,KAAa,EACb,MAA4B;IAE5B,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IAEpD,MAAM,MAAM,GAAG,wBAAc,CAAC,GAAG,CAC/B,4CAAmB,CACpB,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IAE3D,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,WAAY,CAAC,kBAAkB,CAM5D,MAAM,EAAE,EAAE,CAAC,CAAC;QAEf,MAAM,KAAK,GACT,CAAC,UAAU,CAAC,SAAS,GAAG,UAAU,CAAC,YAAY,GAAG,UAAU,CAAC,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC;YAC3F,CAAC,CAAC;QAEJ,OAAO;YACL,KAAK;YACL,GAAG,UAAU;SACd,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,sBAAsB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC;AACH,CAAC;AAED;;GAEG;AACH,8DAA8D;AAC9D,SAAS,sBAAsB,CAAC,OAAY,EAAE,KAAa;IACzD,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,IAAI,SAAS,GAAG,GAAG,CAAC;IACpB,IAAI,YAAY,GAAG,GAAG,CAAC;IACvB,MAAM,QAAQ,GAAG,GAAG,CAAC;IACrB,IAAI,OAAO,GAAG,GAAG,CAAC;IAElB,yBAAyB;IACzB,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,qBAAqB,CAAC,CAAC;QACnC,SAAS,GAAG,CAAC,CAAC;QACd,YAAY,GAAG,CAAC,CAAC;QACjB,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,YAAY,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;IAC1E,CAAC;IAED,qBAAqB;IACrB,MAAM,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAChE,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;QACpB,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QACpC,YAAY,GAAG,GAAG,CAAC;IACrB,CAAC;SAAM,IAAI,WAAW,IAAI,CAAC,EAAE,CAAC;QAC5B,YAAY,GAAG,GAAG,CAAC;IACrB,CAAC;IAED,mCAAmC;IACnC,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACpD,IAAI,UAAU,GAAG,CAAC,CAAC;IAEnB,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,8DAA8D;QAC7D,OAAiB,CAAC,OAAO,CAAC,CAAC,MAA4B,EAAE,EAAE;YAC1D,MAAM,OAAO,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;YACrD,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC1B,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;oBAAE,UAAU,EAAE,CAAC;YAC3C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,GAAG,CAAC,UAAU,CAAC,MAAM,GAAG,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1E,CAAC;IAED,IAAI,SAAS,GAAG,GAAG,EAAE,CAAC;QACpB,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IACxC,CAAC;IAED,sBAAsB;IACtB,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjD,8DAA8D;QAC9D,MAAM,aAAa,GAAG,IAAI,GAAG;QAC3B,8DAA8D;QAC7D,OAAiB,CAAC,GAAG,CAAC,CAAC,CAAuB,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAC9E,CAAC;QACF,IAAI,aAAa,CAAC,IAAI,GAAG,OAAO,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACxC,OAAO,GAAG,GAAG,CAAC;QAChB,CAAC;IACH,CAAC;IAED,MAAM,KAAK,GAAG,CAAC,SAAS,GAAG,YAAY,GAAG,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;IAElE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,YAAY,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;AACvE,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,cAAc;AAC3B,8DAA8D;AAC9D,OAAY,EACZ,OAA0B,EAC1B,KAAa,EACb,MAA4B;AAC5B,8DAA8D;;IAE9D,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;QACxB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,MAAM,GAAG,wBAAc,CAAC,GAAG,CAC/B,4DAA2B,CAC5B,CAAC,MAAM,CAAC;QACP,KAAK;QACL,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC;QACjC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;KAC5D,CAAC,CAAC;IAEH,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,WAAW,CAAC,kBAAkB,CAIxC,MAAM,EAAE,EAAE,CAAC,CAAC;QAEf,+DAA+D;QAC/D,6CAA6C;QAC7C,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAgB,cAAc,CAAC,MAAc,EAAE,WAA4B;IACzE,OAAO,OAAO,CAAC,WAAW,CAAC,uBAAuB,EAAE,MAAM,EAAE,WAAW,CAAC,IAAI,EAAE,CAAC;AACjF,CAAC"}
|
package/dist/agentic/index.d.ts
CHANGED
|
@@ -2,7 +2,7 @@
|
|
|
2
2
|
* Agentic RAG - Advanced Retrieval-Augmented Generation
|
|
3
3
|
* Export all agentic RAG functionality
|
|
4
4
|
*/
|
|
5
|
-
export type { QueryPlan, SubQuery, ReasoningChain, ReasoningHop, ReflectionResult, QualityAssessment, VerifiedResponse, VerifiedSource, Citation, Context, FeedbackData,
|
|
5
|
+
export type { QueryPlan, SubQuery, ReasoningChain, ReasoningHop, ReflectionResult, QualityAssessment, VerifiedResponse, VerifiedSource, Citation, Context, FeedbackData, AgenticGraphSearchResult, GraphNode, GraphEdge, AdaptiveStrategyResult, HyDEResult, CorrectiveRAGResult, Correction, AgenticLLMProvider, LLMOptions, CacheEntry, AgenticRAGConfig, } from './types';
|
|
6
6
|
export type { Entity as AgenticEntity, Message as AgenticMessage } from './types';
|
|
7
7
|
export * from './decorators';
|
|
8
8
|
export * from './agentic-rag.service';
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/agentic/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,YAAY,EACV,SAAS,EACT,QAAQ,EACR,cAAc,EACd,YAAY,EACZ,gBAAgB,EAChB,iBAAiB,EACjB,gBAAgB,EAChB,cAAc,EACd,QAAQ,EACR,OAAO,EACP,YAAY,EACZ,
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/agentic/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,YAAY,EACV,SAAS,EACT,QAAQ,EACR,cAAc,EACd,YAAY,EACZ,gBAAgB,EAChB,iBAAiB,EACjB,gBAAgB,EAChB,cAAc,EACd,QAAQ,EACR,OAAO,EACP,YAAY,EACZ,wBAAwB,EACxB,SAAS,EACT,SAAS,EACT,sBAAsB,EACtB,UAAU,EACV,mBAAmB,EACnB,UAAU,EACV,kBAAkB,EAClB,UAAU,EACV,UAAU,EACV,gBAAgB,GACjB,MAAM,SAAS,CAAC;AAGjB,YAAY,EAAE,MAAM,IAAI,aAAa,EAAE,OAAO,IAAI,cAAc,EAAE,MAAM,SAAS,CAAC;AAGlF,cAAc,cAAc,CAAC;AAG7B,cAAc,uBAAuB,CAAC"}
|
package/dist/agentic/types.d.ts
CHANGED
|
@@ -119,9 +119,10 @@ export interface FeedbackData {
|
|
|
119
119
|
userId?: string;
|
|
120
120
|
}
|
|
121
121
|
/**
|
|
122
|
-
* Graph Search Result - Knowledge graph retrieval result
|
|
122
|
+
* Agentic Graph Search Result - Knowledge graph retrieval result (agentic module)
|
|
123
|
+
* @deprecated Use GraphSearchResult from @hazeljs/rag graph module instead.
|
|
123
124
|
*/
|
|
124
|
-
export interface
|
|
125
|
+
export interface AgenticGraphSearchResult extends SearchResult {
|
|
125
126
|
nodes: GraphNode[];
|
|
126
127
|
edges: GraphEdge[];
|
|
127
128
|
path: string[];
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/agentic/types.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAExC;;GAEG;AACH,MAAM,WAAW,SAAS;IACxB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,QAAQ,EAAE,CAAC;IACvB,QAAQ,EAAE,YAAY,GAAG,UAAU,GAAG,UAAU,CAAC;IACjD,mBAAmB,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,SAAS,GAAG,YAAY,GAAG,aAAa,GAAG,UAAU,CAAC;IAC5D,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,YAAY,EAAE,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,YAAY,EAAE,CAAC;CACzB;AAED,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,iBAAiB,CAAC;IAC3B,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,cAAc,EAAE,CAAC;IAC1B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,EAAE,QAAQ,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,cAAe,SAAQ,YAAY;IAClD,YAAY,EAAE,kBAAkB,CAAC;CAClC;AAED,MAAM,WAAW,kBAAkB;IACjC,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;IACvB,QAAQ,EAAE,OAAO,CAAC;IAClB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,OAAO;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,mBAAmB,EAAE,OAAO,EAAE,CAAC;IAC/B,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,MAAM,EAAE,KAAK,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;CACjB;AAED,MAAM,WAAW,MAAM;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,IAAI,CAAC;IACf,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,KAAK;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,OAAO,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,IAAI,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/agentic/types.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC;AAExC;;GAEG;AACH,MAAM,WAAW,SAAS;IACxB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,QAAQ,EAAE,CAAC;IACvB,QAAQ,EAAE,YAAY,GAAG,UAAU,GAAG,UAAU,CAAC;IACjD,mBAAmB,EAAE,MAAM,CAAC;CAC7B;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,SAAS,GAAG,YAAY,GAAG,aAAa,GAAG,UAAU,CAAC;IAC5D,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,YAAY,EAAE,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,YAAY,EAAE,CAAC;CACzB;AAED,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,gBAAgB,EAAE,MAAM,CAAC;IACzB,OAAO,EAAE,iBAAiB,CAAC;IAC3B,YAAY,EAAE,MAAM,EAAE,CAAC;IACvB,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,cAAc,EAAE,CAAC;IAC1B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,EAAE,QAAQ,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,cAAe,SAAQ,YAAY;IAClD,YAAY,EAAE,kBAAkB,CAAC;CAClC;AAED,MAAM,WAAW,kBAAkB;IACjC,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;IACvB,cAAc,EAAE,MAAM,CAAC;IACvB,QAAQ,EAAE,OAAO,CAAC;IAClB,MAAM,EAAE,MAAM,EAAE,CAAC;CAClB;AAED,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,OAAO;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,mBAAmB,EAAE,OAAO,EAAE,CAAC;IAC/B,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,MAAM,EAAE,KAAK,EAAE,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,CAAC;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;CACjB;AAED,MAAM,WAAW,MAAM;IACrB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,IAAI,CAAC;IACf,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,KAAK;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,OAAO,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,IAAI,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;;GAGG;AACH,MAAM,WAAW,wBAAyB,SAAQ,YAAY;IAC5D,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,IAAI,EAAE,MAAM,EAAE,CAAC;CAChB;AAED,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,SAAS;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACrC;AAED;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,gBAAgB,EAAE,MAAM,CAAC;IACzB,MAAM,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,KAAK,CAAC;QAClB,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,CAAC;KACf,CAAC,CAAC;CACJ;AAED;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,qBAAqB,EAAE,MAAM,EAAE,CAAC;IAChC,gBAAgB,EAAE,YAAY,EAAE,CAAC;IACjC,eAAe,EAAE,MAAM,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,WAAW,EAAE,UAAU,EAAE,CAAC;IAC1B,YAAY,EAAE,OAAO,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,eAAe,GAAG,cAAc,GAAG,eAAe,CAAC;IACzD,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAChE,kBAAkB,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IACnE,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;CACxC;AAED,MAAM,WAAW,UAAU;IACzB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,WAAW,UAAU,CAAC,CAAC;IAC3B,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,CAAC,CAAC;IACT,SAAS,EAAE,MAAM,CAAC;IAClB,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,MAAM,CAAC;CACd;AAED;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,kBAAkB,CAAC;IACjC,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB"}
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* CommunityDetector
|
|
3
|
+
*
|
|
4
|
+
* Detects communities (clusters of closely-related entities) in the knowledge
|
|
5
|
+
* graph using the **Label Propagation Algorithm (LPA)**.
|
|
6
|
+
*
|
|
7
|
+
* Why LPA?
|
|
8
|
+
* - O(V + E) per iteration → scales well to thousands of entities
|
|
9
|
+
* - No parameters to tune (unlike Louvain's resolution parameter)
|
|
10
|
+
* - Converges in ~5–10 iterations for most real-world graphs
|
|
11
|
+
*
|
|
12
|
+
* Algorithm:
|
|
13
|
+
* 1. Assign each entity its own unique community label.
|
|
14
|
+
* 2. Shuffle entity iteration order (breaks symmetry for better convergence).
|
|
15
|
+
* 3. Each entity adopts the most frequent label among its neighbours,
|
|
16
|
+
* weighted by relationship strength.
|
|
17
|
+
* 4. Repeat until labels stabilise or max iterations reached.
|
|
18
|
+
* 5. Groups entities that share the same label into communities.
|
|
19
|
+
*
|
|
20
|
+
* Large-community splitting:
|
|
21
|
+
* If a community exceeds `maxCommunitySize`, it is recursively split by
|
|
22
|
+
* running LPA on the sub-graph of its members only.
|
|
23
|
+
*/
|
|
24
|
+
import type { GraphCommunity } from './graph.types';
|
|
25
|
+
import type { GraphStore } from './knowledge-graph';
|
|
26
|
+
export interface CommunityDetectorConfig {
|
|
27
|
+
maxIterations?: number;
|
|
28
|
+
maxCommunitySize?: number;
|
|
29
|
+
}
|
|
30
|
+
export declare class CommunityDetector {
|
|
31
|
+
private readonly maxIterations;
|
|
32
|
+
private readonly maxCommunitySize;
|
|
33
|
+
constructor(config?: CommunityDetectorConfig);
|
|
34
|
+
/**
|
|
35
|
+
* Run community detection on the full graph.
|
|
36
|
+
* Returns a flat list of GraphCommunity objects (level 0 = leaf communities).
|
|
37
|
+
*/
|
|
38
|
+
detect(graph: GraphStore): GraphCommunity[];
|
|
39
|
+
private buildWeightedAdjacency;
|
|
40
|
+
private runLPA;
|
|
41
|
+
private groupByLabel;
|
|
42
|
+
private splitOversized;
|
|
43
|
+
private shuffle;
|
|
44
|
+
}
|
|
45
|
+
//# sourceMappingURL=community-detector.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"community-detector.d.ts","sourceRoot":"","sources":["../../src/graph/community-detector.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;GAsBG;AAEH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAEpD,MAAM,WAAW,uBAAuB;IACtC,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;gBAE9B,MAAM,GAAE,uBAA4B;IAKhD;;;OAGG;IACH,MAAM,CAAC,KAAK,EAAE,UAAU,GAAG,cAAc,EAAE;IAsB3C,OAAO,CAAC,sBAAsB;IAqB9B,OAAO,CAAC,MAAM;IA8Cd,OAAO,CAAC,YAAY;IAUpB,OAAO,CAAC,cAAc;IAkCtB,OAAO,CAAC,OAAO;CAOhB"}
|