@mcp-graph-workflow/mcp-graph 5.7.0 → 5.8.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/core/context/rag-context.d.ts.map +1 -1
- package/dist/core/context/rag-context.js +9 -5
- package/dist/core/context/rag-context.js.map +1 -1
- package/dist/mcp/tools/rag-context.d.ts +2 -0
- package/dist/mcp/tools/rag-context.d.ts.map +1 -1
- package/dist/mcp/tools/rag-context.js +127 -9
- package/dist/mcp/tools/rag-context.js.map +1 -1
- package/dist/mcp/tools/reindex-knowledge.d.ts.map +1 -1
- package/dist/mcp/tools/reindex-knowledge.js +2 -0
- package/dist/mcp/tools/reindex-knowledge.js.map +1 -1
- package/dist/schemas/siebel.schema.d.ts +2 -2
- package/package.json +1 -1
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rag-context.d.ts","sourceRoot":"","sources":["../../../src/core/context/rag-context.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAG5D,OAAO,EAAoB,KAAK,WAAW,EAAE,MAAM,sBAAsB,CAAC;
|
|
1
|
+
{"version":3,"file":"rag-context.d.ts","sourceRoot":"","sources":["../../../src/core/context/rag-context.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC;AAG5D,OAAO,EAAoB,KAAK,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAK1E,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAEpE,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,EAAE,cAAc,EAAE,CAAC;IAChC,gBAAgB,EAAE,gBAAgB,EAAE,CAAC;IACrC,gBAAgB,EAAE,WAAW,EAAE,CAAC;IAChC,UAAU,EAAE;QACV,MAAM,EAAE,MAAM,CAAC;QACf,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;CACH;AAED,UAAU,cAAc;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;CACf;AAED;;;;;;;GAOG;AACH,wBAAgB,eAAe,CAC7B,KAAK,EAAE,WAAW,EAClB,KAAK,EAAE,MAAM,EACb,WAAW,GAAE,MAA6B,EAC1C,KAAK,CAAC,EAAE,cAAc,GACrB,UAAU,CA4IZ"}
|
|
@@ -6,6 +6,7 @@ import { KnowledgeStore } from "../store/knowledge-store.js";
|
|
|
6
6
|
import { searchNodes } from "../search/fts-search.js";
|
|
7
7
|
import { buildTaskContext } from "./compact-context.js";
|
|
8
8
|
import { estimateTokens } from "./token-estimator.js";
|
|
9
|
+
import { understandQuery } from "../rag/query-understanding.js";
|
|
9
10
|
import { DEFAULT_TOKEN_BUDGET } from "../utils/constants.js";
|
|
10
11
|
import { logger } from "../utils/logger.js";
|
|
11
12
|
/**
|
|
@@ -17,9 +18,12 @@ import { logger } from "../utils/logger.js";
|
|
|
17
18
|
* 4. Manage token budget to stay within limits
|
|
18
19
|
*/
|
|
19
20
|
export function ragBuildContext(store, query, tokenBudget = DEFAULT_TOKEN_BUDGET, phase) {
|
|
20
|
-
|
|
21
|
+
// Query understanding: rewrite query for better FTS matching
|
|
22
|
+
const understanding = understandQuery(query);
|
|
23
|
+
const effectiveQuery = understanding.rewrittenQuery || query;
|
|
24
|
+
logger.info(`RAG context: query="${query}", effective="${effectiveQuery}", budget=${tokenBudget} tokens, phase=${phase ?? "none"}, intent=${understanding.intent}`);
|
|
21
25
|
// Stage 1: Search for relevant nodes with TF-IDF reranking + substring fallback
|
|
22
|
-
let searchResults = searchNodes(store,
|
|
26
|
+
let searchResults = searchNodes(store, effectiveQuery, { limit: 10, rerank: true });
|
|
23
27
|
// Fallback: if FTS returns nothing, try substring match
|
|
24
28
|
if (searchResults.length === 0) {
|
|
25
29
|
logger.debug("RAG FTS returned 0 results, falling back to substring search");
|
|
@@ -62,7 +66,7 @@ export function ragBuildContext(store, query, tokenBudget = DEFAULT_TOKEN_BUDGET
|
|
|
62
66
|
let kResults;
|
|
63
67
|
try {
|
|
64
68
|
if (phase) {
|
|
65
|
-
const phaseResults = knowledgeStore.searchWithPhaseBoost(
|
|
69
|
+
const phaseResults = knowledgeStore.searchWithPhaseBoost(effectiveQuery, phase, 10);
|
|
66
70
|
// Apply quality weighting on top of phase-boosted results
|
|
67
71
|
kResults = phaseResults.map((r) => ({
|
|
68
72
|
...r,
|
|
@@ -70,12 +74,12 @@ export function ragBuildContext(store, query, tokenBudget = DEFAULT_TOKEN_BUDGET
|
|
|
70
74
|
}));
|
|
71
75
|
}
|
|
72
76
|
else {
|
|
73
|
-
kResults = knowledgeStore.searchWithQuality(
|
|
77
|
+
kResults = knowledgeStore.searchWithQuality(effectiveQuery, 10);
|
|
74
78
|
}
|
|
75
79
|
}
|
|
76
80
|
catch {
|
|
77
81
|
// Fall back to basic search if quality columns not yet available
|
|
78
|
-
kResults = knowledgeStore.search(
|
|
82
|
+
kResults = knowledgeStore.search(effectiveQuery, 10);
|
|
79
83
|
}
|
|
80
84
|
knowledgeResults = kResults.slice(0, 5).map((r) => ({
|
|
81
85
|
id: r.id,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rag-context.js","sourceRoot":"","sources":["../../../src/core/context/rag-context.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAoB,MAAM,sBAAsB,CAAC;AAC1E,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAkC5C;;;;;;;GAOG;AACH,MAAM,UAAU,eAAe,CAC7B,KAAkB,EAClB,KAAa,EACb,cAAsB,oBAAoB,EAC1C,KAAsB;IAEtB,MAAM,CAAC,IAAI,CAAC,uBAAuB,KAAK,aAAa,WAAW,kBAAkB,KAAK,IAAI,MAAM,EAAE,CAAC,CAAC;
|
|
1
|
+
{"version":3,"file":"rag-context.js","sourceRoot":"","sources":["../../../src/core/context/rag-context.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAoB,MAAM,sBAAsB,CAAC;AAC1E,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,+BAA+B,CAAC;AAChE,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAkC5C;;;;;;;GAOG;AACH,MAAM,UAAU,eAAe,CAC7B,KAAkB,EAClB,KAAa,EACb,cAAsB,oBAAoB,EAC1C,KAAsB;IAEtB,6DAA6D;IAC7D,MAAM,aAAa,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IAC7C,MAAM,cAAc,GAAG,aAAa,CAAC,cAAc,IAAI,KAAK,CAAC;IAE7D,MAAM,CAAC,IAAI,CAAC,uBAAuB,KAAK,iBAAiB,cAAc,aAAa,WAAW,kBAAkB,KAAK,IAAI,MAAM,YAAY,aAAa,CAAC,MAAM,EAAE,CAAC,CAAC;IAEpK,gFAAgF;IAChF,IAAI,aAAa,GAAG,WAAW,CAAC,KAAK,EAAE,cAAc,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IAEpF,wDAAwD;IACxD,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,8DAA8D,CAAC,CAAC;QAC7E,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;YACrC,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;YACvC,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAClE,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACrB,MAAM,OAAO,GAAG,QAAQ;qBACrB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CACZ,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACf,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;oBACjC,CAAC,CAAC,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAChD,CACF;qBACA,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBAChB,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED,MAAM,aAAa,GAAqB,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC9D,MAAM,OAAO,GAAmB;YAC9B,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE;YACb,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI;YACjB,KAAK,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK;YACnB,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM;YACrB,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ;YACzB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,IAAI;SACzC,CAAC;QACF,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW;YAAE,OAAO,CAAC,WAAW,GAAG,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC;QACjE,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM;YAAE,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QACpD,OAAO,OAAO,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,8EAA8E;IAC9E,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;IACzD,IAAI,gBAAgB,GAAuB,EAAE,CAAC;IAC9C,IAAI,CAAC;QACH,yFAAyF;QACzF,IAAI,QAAkG,CAAC;QACvG,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,YAAY,GAAG,cAAc,CAAC,oBAAoB,CAAC,cAAc,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBACpF,0DAA0D;gBAC1D,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBAClC,GAAG,CAAC;oBACJ,KAAK,EAAE,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,YAAsB,IAAK,CAA6B,CAAC,YAAsB,IAAI,CAAC,CAAC;iBACpH,CAAC,CAAC,CAAC;YACN,CAAC;iBAAM,CAAC;gBACN,QAAQ,GAAG,cAAc,CAAC,iBAAiB,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;YAClE,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,iEAAiE;YACjE,QAAQ,GAAG,cAAc,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;QACvD,CAAC;QACD,gBAAgB,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAClD,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,UAAU,EAAE,CAAC,CAAC,UAAU;YACxB,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO;YAC7E,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,IAAI;SACzC,CAAC,CAAC,CAAC;IACN,CAAC;IAAC,MAAM,CAAC;QACP,mEAAmE;QACnE,MAAM,CAAC,KAAK,CAAC,qDAAqD,CAAC,CAAC;IACtE,CAAC;IAED,mFAAmF;IACnF,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9D,gBAAgB,GAAG,aAAa;aAC7B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;aAC5B,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;aACX,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACd,EAAE,EAAE,YAAY,CAAC,EAAE;YACnB,UAAU,EAAE,cAAc;YAC1B,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,OAAO,EAAE,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC,KAAK;YACjC,KAAK,EAAE,GAAG;SACX,CAAC,CAAC,CAAC;IACR,CAAC;IAED,wDAAwD;IACxD,MAAM,gBAAgB,GAAkB,EAAE,CAAC;IAC3C,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,gBAAgB,EAAE,CAAC,CAAC;IAC/E,IAAI,UAAU,GAAG,cAAc,CAAC,WAAW,CAAC,CAAC;IAE7C,wEAAwE;IACxE,IAAI,UAAU,GAAG,WAAW,EAAE,CAAC;QAC7B,KAAK,MAAM,MAAM,IAAI,aAAa,EAAE,CAAC;YACnC,IAAI,UAAU,IAAI,WAAW;gBAAE,MAAM;YAErC,MAAM,GAAG,GAAG,gBAAgB,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACpD,IAAI,CAAC,GAAG;gBAAE,SAAS;YAEnB,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC;YAC9C,IAAI,UAAU,GAAG,SAAS,GAAG,WAAW,EAAE,CAAC;gBACzC,mEAAmE;gBACnE,IAAI,gBAAgB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAClC,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAC3B,UAAU,IAAI,SAAS,CAAC;gBAC1B,CAAC;gBACD,MAAM;YACR,CAAC;YAED,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC3B,UAAU,IAAI,SAAS,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,sEAAsE;IACtE,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;IAEvD,MAAM,CAAC,IAAI,CACT,sBAAsB,aAAa,CAAC,MAAM,WAAW,gBAAgB,CAAC,MAAM,eAAe,gBAAgB,CAAC,MAAM,cAAc,UAAU,IAAI,WAAW,SAAS,CACnK,CAAC;IAEF,OAAO;QACL,KAAK;QACL,aAAa;QACb,gBAAgB;QAChB,gBAAgB;QAChB,UAAU,EAAE;YACV,MAAM,EAAE,WAAW;YACnB,IAAI,EAAE,YAAY;YAClB,SAAS,EAAE,WAAW,GAAG,YAAY;SACtC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -1,4 +1,6 @@
|
|
|
1
1
|
import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
|
2
2
|
import type { SqliteStore } from "../../core/store/sqlite-store.js";
|
|
3
|
+
/** Invalidate the RAG query cache (e.g., after reindex_knowledge). */
|
|
4
|
+
export declare function invalidateRagCache(): void;
|
|
3
5
|
export declare function registerRagContext(server: McpServer, store: SqliteStore): void;
|
|
4
6
|
//# sourceMappingURL=rag-context.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rag-context.d.ts","sourceRoot":"","sources":["../../../src/mcp/tools/rag-context.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kCAAkC,CAAC;
|
|
1
|
+
{"version":3,"file":"rag-context.d.ts","sourceRoot":"","sources":["../../../src/mcp/tools/rag-context.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kCAAkC,CAAC;AAkBpE,sEAAsE;AACtE,wBAAgB,kBAAkB,IAAI,IAAI,CAGzC;AAED,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,GAAG,IAAI,CAkN9E"}
|
|
@@ -3,10 +3,22 @@ import { ragBuildContext } from "../../core/context/rag-context.js";
|
|
|
3
3
|
import { assembleContext } from "../../core/context/context-assembler.js";
|
|
4
4
|
import { multiStrategySearch } from "../../core/rag/multi-strategy-retrieval.js";
|
|
5
5
|
import { recordUsage } from "../../core/rag/knowledge-quality.js";
|
|
6
|
+
import { understandQuery } from "../../core/rag/query-understanding.js";
|
|
7
|
+
import { postRetrievalPipeline } from "../../core/rag/post-retrieval.js";
|
|
8
|
+
import { buildCitedContext } from "../../core/rag/citation-mapper.js";
|
|
9
|
+
import { QueryCache } from "../../core/rag/query-cache.js";
|
|
10
|
+
import { RagTracer } from "../../core/rag/rag-trace.js";
|
|
6
11
|
import { detectCurrentPhase } from "../../core/planner/lifecycle-phase.js";
|
|
7
12
|
import { DEFAULT_TOKEN_BUDGET } from "../../core/utils/constants.js";
|
|
8
13
|
import { logger } from "../../core/utils/logger.js";
|
|
9
14
|
import { mcpText } from "../response-helpers.js";
|
|
15
|
+
/** Module-level query cache — shared across all rag_context calls. */
|
|
16
|
+
const ragCache = new QueryCache({ ttlMs: 5 * 60 * 1000, maxSize: 100 });
|
|
17
|
+
/** Invalidate the RAG query cache (e.g., after reindex_knowledge). */
|
|
18
|
+
export function invalidateRagCache() {
|
|
19
|
+
ragCache.invalidateAll();
|
|
20
|
+
logger.debug("rag_context:cache_invalidated");
|
|
21
|
+
}
|
|
10
22
|
export function registerRagContext(server, store) {
|
|
11
23
|
server.tool("rag_context", "Build a RAG context from a natural language query. Returns relevant nodes with expanded subgraph context, managed within a token budget. Supports detail levels: summary (~20 tok/node), standard (~150 tok/node), deep (~500+ tok/node with knowledge).", {
|
|
12
24
|
query: z.string().describe("Natural language query to search for"),
|
|
@@ -24,7 +36,7 @@ export function registerRagContext(server, store) {
|
|
|
24
36
|
strategy: z
|
|
25
37
|
.enum(["fts", "multi"])
|
|
26
38
|
.optional()
|
|
27
|
-
.describe("Search strategy: 'fts' (traditional BM25), 'multi' (multi-strategy with
|
|
39
|
+
.describe("Search strategy: 'fts' (traditional BM25), 'multi' (multi-strategy with query understanding, post-retrieval, citations). Default: fts"),
|
|
28
40
|
}, async ({ query, tokenBudget, detail, strategy }) => {
|
|
29
41
|
logger.debug("tool:rag_context", { query, tier: detail });
|
|
30
42
|
const budget = tokenBudget ?? DEFAULT_TOKEN_BUDGET;
|
|
@@ -51,25 +63,121 @@ export function registerRagContext(server, store) {
|
|
|
51
63
|
logger.info("tool:rag_context:ok", { query, tier: detail, phase: currentPhase, strategy });
|
|
52
64
|
return mcpText(ctx);
|
|
53
65
|
}
|
|
54
|
-
// Multi-strategy search mode
|
|
66
|
+
// Multi-strategy search mode — full pipeline
|
|
55
67
|
if (strategy === "multi") {
|
|
56
|
-
const
|
|
57
|
-
|
|
68
|
+
const tracer = new RagTracer(query);
|
|
69
|
+
// Stage 1: Query Understanding
|
|
70
|
+
tracer.startStage("query_understanding");
|
|
71
|
+
const understanding = understandQuery(query);
|
|
72
|
+
const effectiveQuery = understanding.rewrittenQuery || query;
|
|
73
|
+
tracer.endStage("query_understanding", {
|
|
74
|
+
inputCount: 1,
|
|
75
|
+
outputCount: 1,
|
|
76
|
+
details: {
|
|
77
|
+
intent: understanding.intent,
|
|
78
|
+
entities: understanding.entities.length,
|
|
79
|
+
sourceFilters: understanding.sourceTypeFilter,
|
|
80
|
+
expandedTerms: understanding.expandedTerms.length,
|
|
81
|
+
},
|
|
82
|
+
});
|
|
83
|
+
// Stage 2: Cache check
|
|
84
|
+
const cached = ragCache.get(effectiveQuery);
|
|
85
|
+
if (cached) {
|
|
86
|
+
logger.debug("rag_context:cache_hit", { query: effectiveQuery });
|
|
87
|
+
const cachedCited = buildCitedContext(cached);
|
|
88
|
+
tracer.startStage("citation");
|
|
89
|
+
tracer.endStage("citation", { inputCount: cached.length, outputCount: cachedCited.citations.length });
|
|
90
|
+
const trace = tracer.finalize();
|
|
91
|
+
return mcpText({
|
|
92
|
+
query,
|
|
93
|
+
strategy: "multi",
|
|
94
|
+
fromCache: true,
|
|
95
|
+
intent: understanding.intent,
|
|
96
|
+
citedContext: cachedCited.assembledText,
|
|
97
|
+
citations: cachedCited.citations,
|
|
98
|
+
sourceBreakdown: cachedCited.sourceBreakdown,
|
|
99
|
+
results: cached.map((r) => ({
|
|
100
|
+
id: r.id,
|
|
101
|
+
sourceType: r.sourceType,
|
|
102
|
+
title: r.title,
|
|
103
|
+
content: r.content.length > 500 ? r.content.slice(0, 500) + "..." : r.content,
|
|
104
|
+
score: r.score,
|
|
105
|
+
qualityScore: r.qualityScore,
|
|
106
|
+
strategies: r.strategies,
|
|
107
|
+
})),
|
|
108
|
+
tokenUsage: {
|
|
109
|
+
budget,
|
|
110
|
+
used: Math.ceil(cached.reduce((sum, r) => sum + r.content.length / 4, 0)),
|
|
111
|
+
remaining: budget,
|
|
112
|
+
},
|
|
113
|
+
trace: { traceId: trace.traceId, totalLatencyMs: trace.totalLatencyMs, stages: trace.stages.length },
|
|
114
|
+
});
|
|
115
|
+
}
|
|
116
|
+
// Stage 3: Retrieval (request more results to give post-retrieval room to dedup/rerank)
|
|
117
|
+
tracer.startStage("retrieval");
|
|
118
|
+
const rawResults = multiStrategySearch(store.getDb(), effectiveQuery, {
|
|
119
|
+
limit: 20,
|
|
58
120
|
phase: currentPhase,
|
|
59
121
|
});
|
|
60
|
-
|
|
122
|
+
tracer.endStage("retrieval", { inputCount: 1, outputCount: rawResults.length });
|
|
123
|
+
// Stage 4: Post-retrieval (dedup, rerank, stitch)
|
|
124
|
+
tracer.startStage("post_retrieval");
|
|
125
|
+
const postProcessed = postRetrievalPipeline({
|
|
126
|
+
query: effectiveQuery,
|
|
127
|
+
results: rawResults,
|
|
128
|
+
maxResults: 10,
|
|
129
|
+
});
|
|
130
|
+
const multiResults = postProcessed.results;
|
|
131
|
+
tracer.endStage("post_retrieval", {
|
|
132
|
+
inputCount: rawResults.length,
|
|
133
|
+
outputCount: multiResults.length,
|
|
134
|
+
details: {
|
|
135
|
+
deduplicated: postProcessed.deduplicated,
|
|
136
|
+
stitchedChunks: postProcessed.stitchedChunks,
|
|
137
|
+
},
|
|
138
|
+
});
|
|
139
|
+
// Cache the post-processed results
|
|
140
|
+
ragCache.set(effectiveQuery, multiResults);
|
|
141
|
+
// Stage 5: Citation mapping
|
|
142
|
+
tracer.startStage("citation");
|
|
143
|
+
const cited = buildCitedContext(multiResults);
|
|
144
|
+
tracer.endStage("citation", {
|
|
145
|
+
inputCount: multiResults.length,
|
|
146
|
+
outputCount: cited.citations.length,
|
|
147
|
+
});
|
|
148
|
+
// Record source contributions in trace
|
|
149
|
+
for (const [sourceType, count] of Object.entries(cited.sourceBreakdown)) {
|
|
150
|
+
tracer.recordSourceContribution(sourceType, count);
|
|
151
|
+
}
|
|
152
|
+
tracer.setCitationCount(cited.citations.length);
|
|
153
|
+
tracer.setTokensUsed(Math.ceil(multiResults.reduce((sum, r) => sum + r.content.length / 4, 0)));
|
|
154
|
+
// Record usage for retrieved docs (best-effort)
|
|
61
155
|
try {
|
|
62
156
|
for (const result of multiResults.slice(0, 5)) {
|
|
63
|
-
recordUsage(store.getDb(), result.id,
|
|
157
|
+
recordUsage(store.getDb(), result.id, effectiveQuery, "retrieved", { tool: "rag_context", strategy: "multi" });
|
|
64
158
|
}
|
|
65
159
|
}
|
|
66
160
|
catch {
|
|
67
161
|
// Usage recording is best-effort
|
|
68
162
|
}
|
|
69
|
-
|
|
163
|
+
const trace = tracer.finalize();
|
|
164
|
+
logger.info("tool:rag_context:ok", {
|
|
165
|
+
query,
|
|
166
|
+
strategy: "multi",
|
|
167
|
+
phase: currentPhase,
|
|
168
|
+
intent: understanding.intent,
|
|
169
|
+
rawResults: rawResults.length,
|
|
170
|
+
postProcessed: multiResults.length,
|
|
171
|
+
citations: cited.citations.length,
|
|
172
|
+
totalLatencyMs: trace.totalLatencyMs,
|
|
173
|
+
});
|
|
70
174
|
return mcpText({
|
|
71
175
|
query,
|
|
72
176
|
strategy: "multi",
|
|
177
|
+
intent: understanding.intent,
|
|
178
|
+
citedContext: cited.assembledText,
|
|
179
|
+
citations: cited.citations,
|
|
180
|
+
sourceBreakdown: cited.sourceBreakdown,
|
|
73
181
|
results: multiResults.map((r) => ({
|
|
74
182
|
id: r.id,
|
|
75
183
|
sourceType: r.sourceType,
|
|
@@ -81,8 +189,18 @@ export function registerRagContext(server, store) {
|
|
|
81
189
|
})),
|
|
82
190
|
tokenUsage: {
|
|
83
191
|
budget,
|
|
84
|
-
used:
|
|
85
|
-
remaining: budget,
|
|
192
|
+
used: trace.totalTokensUsed,
|
|
193
|
+
remaining: budget - trace.totalTokensUsed,
|
|
194
|
+
},
|
|
195
|
+
trace: {
|
|
196
|
+
traceId: trace.traceId,
|
|
197
|
+
totalLatencyMs: trace.totalLatencyMs,
|
|
198
|
+
stages: trace.stages.map((s) => ({
|
|
199
|
+
stage: s.stage,
|
|
200
|
+
latencyMs: s.latencyMs,
|
|
201
|
+
inputCount: s.inputCount,
|
|
202
|
+
outputCount: s.outputCount,
|
|
203
|
+
})),
|
|
86
204
|
},
|
|
87
205
|
});
|
|
88
206
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"rag-context.js","sourceRoot":"","sources":["../../../src/mcp/tools/rag-context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAC;AAG3B,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,MAAM,4CAA4C,CAAC;AACjF,OAAO,EAAE,WAAW,EAAE,MAAM,qCAAqC,CAAC;AAClE,OAAO,EAAE,kBAAkB,EAAuB,MAAM,uCAAuC,CAAC;AAChG,OAAO,EAAE,oBAAoB,EAAE,MAAM,+BAA+B,CAAC;AACrE,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAEjD,MAAM,UAAU,kBAAkB,CAAC,MAAiB,EAAE,KAAkB;IACtE,MAAM,CAAC,IAAI,CACT,aAAa,EACb,0PAA0P,EAC1P;QACE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,sCAAsC,CAAC;QAClE,WAAW,EAAE,CAAC;aACX,MAAM,EAAE;aACR,GAAG,EAAE;aACL,GAAG,CAAC,GAAG,CAAC;aACR,GAAG,CAAC,KAAK,CAAC;aACV,QAAQ,EAAE;aACV,QAAQ,CAAC,sDAAsD,CAAC;QACnE,MAAM,EAAE,CAAC;aACN,IAAI,CAAC,CAAC,SAAS,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;aACrC,QAAQ,EAAE;aACV,QAAQ,CAAC,kHAAkH,CAAC;QAC/H,QAAQ,EAAE,CAAC;aACR,IAAI,CAAC,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;aACtB,QAAQ,EAAE;aACV,QAAQ,CAAC,
|
|
1
|
+
{"version":3,"file":"rag-context.js","sourceRoot":"","sources":["../../../src/mcp/tools/rag-context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAC;AAG3B,OAAO,EAAE,eAAe,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,EAAE,eAAe,EAAE,MAAM,yCAAyC,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,MAAM,4CAA4C,CAAC;AACjF,OAAO,EAAE,WAAW,EAAE,MAAM,qCAAqC,CAAC;AAClE,OAAO,EAAE,eAAe,EAAE,MAAM,uCAAuC,CAAC;AACxE,OAAO,EAAE,qBAAqB,EAAE,MAAM,kCAAkC,CAAC;AACzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,mCAAmC,CAAC;AACtE,OAAO,EAAE,UAAU,EAAE,MAAM,+BAA+B,CAAC;AAC3D,OAAO,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AACxD,OAAO,EAAE,kBAAkB,EAAuB,MAAM,uCAAuC,CAAC;AAChG,OAAO,EAAE,oBAAoB,EAAE,MAAM,+BAA+B,CAAC;AACrE,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAEjD,sEAAsE;AACtE,MAAM,QAAQ,GAAG,IAAI,UAAU,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;AAExE,sEAAsE;AACtE,MAAM,UAAU,kBAAkB;IAChC,QAAQ,CAAC,aAAa,EAAE,CAAC;IACzB,MAAM,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC;AAChD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,MAAiB,EAAE,KAAkB;IACtE,MAAM,CAAC,IAAI,CACT,aAAa,EACb,0PAA0P,EAC1P;QACE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,sCAAsC,CAAC;QAClE,WAAW,EAAE,CAAC;aACX,MAAM,EAAE;aACR,GAAG,EAAE;aACL,GAAG,CAAC,GAAG,CAAC;aACR,GAAG,CAAC,KAAK,CAAC;aACV,QAAQ,EAAE;aACV,QAAQ,CAAC,sDAAsD,CAAC;QACnE,MAAM,EAAE,CAAC;aACN,IAAI,CAAC,CAAC,SAAS,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;aACrC,QAAQ,EAAE;aACV,QAAQ,CAAC,kHAAkH,CAAC;QAC/H,QAAQ,EAAE,CAAC;aACR,IAAI,CAAC,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;aACtB,QAAQ,EAAE;aACV,QAAQ,CAAC,uIAAuI,CAAC;KACrJ,EACD,KAAK,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE;QACjD,MAAM,CAAC,KAAK,CAAC,kBAAkB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAG,WAAW,IAAI,oBAAoB,CAAC;QAEnD,oEAAoE;QACpE,IAAI,YAAwC,CAAC;QAC7C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,KAAK,CAAC,eAAe,EAAE,CAAC;YACpC,MAAM,aAAa,GAAG,KAAK,CAAC,iBAAiB,CAAC,0BAA0B,CAAC,CAAC;YAC1E,YAAY,GAAG,kBAAkB,CAAC,GAAG,EAAE;gBACrC,aAAa,EAAE,aAAa,CAAC,CAAC,CAAC,aAA+B,CAAC,CAAC,CAAC,IAAI;aACtE,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,wEAAwE;YACxE,MAAM,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC3D,CAAC;QAED,IAAI,MAAM,EAAE,CAAC;YACX,oDAAoD;YACpD,MAAM,GAAG,GAAG,eAAe,CAAC,KAAK,EAAE,KAAK,EAAE;gBACxC,WAAW,EAAE,MAAM;gBACnB,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,YAAY;aACpB,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC3F,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC;QACtB,CAAC;QAED,6CAA6C;QAC7C,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC;YAEpC,+BAA+B;YAC/B,MAAM,CAAC,UAAU,CAAC,qBAAqB,CAAC,CAAC;YACzC,MAAM,aAAa,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;YAC7C,MAAM,cAAc,GAAG,aAAa,CAAC,cAAc,IAAI,KAAK,CAAC;YAC7D,MAAM,CAAC,QAAQ,CAAC,qBAAqB,EAAE;gBACrC,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,CAAC;gBACd,OAAO,EAAE;oBACP,MAAM,EAAE,aAAa,CAAC,MAAM;oBAC5B,QAAQ,EAAE,aAAa,CAAC,QAAQ,CAAC,MAAM;oBACvC,aAAa,EAAE,aAAa,CAAC,gBAAgB;oBAC7C,aAAa,EAAE,aAAa,CAAC,aAAa,CAAC,MAAM;iBAClD;aACF,CAAC,CAAC;YAEH,uBAAuB;YACvB,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YAC5C,IAAI,MAAM,EAAE,CAAC;gBACX,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC;gBACjE,MAAM,WAAW,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAC9C,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;gBAC9B,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE,EAAE,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,EAAE,WAAW,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;gBACtG,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAEhC,OAAO,OAAO,CAAC;oBACb,KAAK;oBACL,QAAQ,EAAE,OAAO;oBACjB,SAAS,EAAE,IAAI;oBACf,MAAM,EAAE,aAAa,CAAC,MAAM;oBAC5B,YAAY,EAAE,WAAW,CAAC,aAAa;oBACvC,SAAS,EAAE,WAAW,CAAC,SAAS;oBAChC,eAAe,EAAE,WAAW,CAAC,eAAe;oBAC5C,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC1B,EAAE,EAAE,CAAC,CAAC,EAAE;wBACR,UAAU,EAAE,CAAC,CAAC,UAAU;wBACxB,KAAK,EAAE,CAAC,CAAC,KAAK;wBACd,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO;wBAC7E,KAAK,EAAE,CAAC,CAAC,KAAK;wBACd,YAAY,EAAE,CAAC,CAAC,YAAY;wBAC5B,UAAU,EAAE,CAAC,CAAC,UAAU;qBACzB,CAAC,CAAC;oBACH,UAAU,EAAE;wBACV,MAAM;wBACN,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;wBACzE,SAAS,EAAE,MAAM;qBAClB;oBACD,KAAK,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE;iBACrG,CAAC,CAAC;YACL,CAAC;YAED,wFAAwF;YACxF,MAAM,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YAC/B,MAAM,UAAU,GAAG,mBAAmB,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,cAAc,EAAE;gBACpE,KAAK,EAAE,EAAE;gBACT,KAAK,EAAE,YAAY;aACpB,CAAC,CAAC;YACH,MAAM,CAAC,QAAQ,CAAC,WAAW,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,WAAW,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;YAEhF,kDAAkD;YAClD,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;YACpC,MAAM,aAAa,GAAG,qBAAqB,CAAC;gBAC1C,KAAK,EAAE,cAAc;gBACrB,OAAO,EAAE,UAAU;gBACnB,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;YACH,MAAM,YAAY,GAAG,aAAa,CAAC,OAAO,CAAC;YAC3C,MAAM,CAAC,QAAQ,CAAC,gBAAgB,EAAE;gBAChC,UAAU,EAAE,UAAU,CAAC,MAAM;gBAC7B,WAAW,EAAE,YAAY,CAAC,MAAM;gBAChC,OAAO,EAAE;oBACP,YAAY,EAAE,aAAa,CAAC,YAAY;oBACxC,cAAc,EAAE,aAAa,CAAC,cAAc;iBAC7C;aACF,CAAC,CAAC;YAEH,mCAAmC;YACnC,QAAQ,CAAC,GAAG,CAAC,cAAc,EAAE,YAAY,CAAC,CAAC;YAE3C,4BAA4B;YAC5B,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YAC9B,MAAM,KAAK,GAAG,iBAAiB,CAAC,YAAY,CAAC,CAAC;YAC9C,MAAM,CAAC,QAAQ,CAAC,UAAU,EAAE;gBAC1B,UAAU,EAAE,YAAY,CAAC,MAAM;gBAC/B,WAAW,EAAE,KAAK,CAAC,SAAS,CAAC,MAAM;aACpC,CAAC,CAAC;YAEH,uCAAuC;YACvC,KAAK,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,eAAe,CAAC,EAAE,CAAC;gBACxE,MAAM,CAAC,wBAAwB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YACrD,CAAC;YACD,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAEhG,gDAAgD;YAChD,IAAI,CAAC;gBACH,KAAK,MAAM,MAAM,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;oBAC9C,WAAW,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,cAAc,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;gBACjH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,iCAAiC;YACnC,CAAC;YAED,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE;gBACjC,KAAK;gBACL,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,YAAY;gBACnB,MAAM,EAAE,aAAa,CAAC,MAAM;gBAC5B,UAAU,EAAE,UAAU,CAAC,MAAM;gBAC7B,aAAa,EAAE,YAAY,CAAC,MAAM;gBAClC,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,MAAM;gBACjC,cAAc,EAAE,KAAK,CAAC,cAAc;aACrC,CAAC,CAAC;YAEH,OAAO,OAAO,CAAC;gBACb,KAAK;gBACL,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE,aAAa,CAAC,MAAM;gBAC5B,YAAY,EAAE,KAAK,CAAC,aAAa;gBACjC,SAAS,EAAE,KAAK,CAAC,SAAS;gBAC1B,eAAe,EAAE,KAAK,CAAC,eAAe;gBACtC,OAAO,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBAChC,EAAE,EAAE,CAAC,CAAC,EAAE;oBACR,UAAU,EAAE,CAAC,CAAC,UAAU;oBACxB,KAAK,EAAE,CAAC,CAAC,KAAK;oBACd,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO;oBAC7E,KAAK,EAAE,CAAC,CAAC,KAAK;oBACd,YAAY,EAAE,CAAC,CAAC,YAAY;oBAC5B,UAAU,EAAE,CAAC,CAAC,UAAU;iBACzB,CAAC,CAAC;gBACH,UAAU,EAAE;oBACV,MAAM;oBACN,IAAI,EAAE,KAAK,CAAC,eAAe;oBAC3B,SAAS,EAAE,MAAM,GAAG,KAAK,CAAC,eAAe;iBAC1C;gBACD,KAAK,EAAE;oBACL,OAAO,EAAE,KAAK,CAAC,OAAO;oBACtB,cAAc,EAAE,KAAK,CAAC,cAAc;oBACpC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC/B,KAAK,EAAE,CAAC,CAAC,KAAK;wBACd,SAAS,EAAE,CAAC,CAAC,SAAS;wBACtB,UAAU,EAAE,CAAC,CAAC,UAAU;wBACxB,WAAW,EAAE,CAAC,CAAC,WAAW;qBAC3B,CAAC,CAAC;iBACJ;aACF,CAAC,CAAC;QACL,CAAC;QAED,iEAAiE;QACjE,MAAM,GAAG,GAAG,eAAe,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;QAEhE,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QACrF,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC;IACtB,CAAC,CACF,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"reindex-knowledge.d.ts","sourceRoot":"","sources":["../../../src/mcp/tools/reindex-knowledge.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kCAAkC,CAAC;
|
|
1
|
+
{"version":3,"file":"reindex-knowledge.d.ts","sourceRoot":"","sources":["../../../src/mcp/tools/reindex-knowledge.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACzE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kCAAkC,CAAC;AAiBpE,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,GAAG,IAAI,CAoEpF"}
|
|
@@ -13,6 +13,7 @@ import { linkBySharedContext } from "../../core/rag/knowledge-linker.js";
|
|
|
13
13
|
import { runSynthesisCycle } from "../../core/rag/knowledge-synthesizer.js";
|
|
14
14
|
import { logger } from "../../core/utils/logger.js";
|
|
15
15
|
import { mcpText } from "../response-helpers.js";
|
|
16
|
+
import { invalidateRagCache } from "./rag-context.js";
|
|
16
17
|
export function registerReindexKnowledge(server, store) {
|
|
17
18
|
server.tool("reindex_knowledge", "Reindex all knowledge sources (memories, cached docs) into the unified knowledge store and rebuild embeddings.", {
|
|
18
19
|
basePath: z
|
|
@@ -25,6 +26,7 @@ export function registerReindexKnowledge(server, store) {
|
|
|
25
26
|
.describe("Which sources to reindex. 'quality' recalculates scores, 'relations' links docs, 'synthesis' generates insights. (default: all)"),
|
|
26
27
|
}, async ({ basePath, sources }) => {
|
|
27
28
|
logger.debug("tool:reindex_knowledge", {});
|
|
29
|
+
invalidateRagCache();
|
|
28
30
|
const projectPath = basePath ?? process.cwd();
|
|
29
31
|
const allSources = !sources || sources.length === 0;
|
|
30
32
|
const knowledgeStore = new KnowledgeStore(store.getDb());
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"reindex-knowledge.js","sourceRoot":"","sources":["../../../src/mcp/tools/reindex-knowledge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAC;AAG3B,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,cAAc,EAAE,MAAM,mCAAmC,CAAC;AACnE,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AACrE,OAAO,EAAE,YAAY,EAAE,MAAM,qCAAqC,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AACpE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;
|
|
1
|
+
{"version":3,"file":"reindex-knowledge.js","sourceRoot":"","sources":["../../../src/mcp/tools/reindex-knowledge.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,QAAQ,CAAC;AAG3B,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EAAE,cAAc,EAAE,MAAM,mCAAmC,CAAC;AACnE,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAC;AACjE,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,OAAO,EAAE,WAAW,EAAE,MAAM,iCAAiC,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AACrE,OAAO,EAAE,YAAY,EAAE,MAAM,qCAAqC,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,gCAAgC,CAAC;AACpE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,OAAO,EAAE,mBAAmB,EAAE,MAAM,oCAAoC,CAAC;AACzE,OAAO,EAAE,iBAAiB,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AACpD,OAAO,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAEtD,MAAM,UAAU,wBAAwB,CAAC,MAAiB,EAAE,KAAkB;IAC5E,MAAM,CAAC,IAAI,CACT,mBAAmB,EACnB,gHAAgH,EAChH;QACE,QAAQ,EAAE,CAAC;aACR,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CAAC,uDAAuD,CAAC;QACpE,OAAO,EAAE,CAAC;aACP,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,YAAY,EAAE,SAAS,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC;aACnH,QAAQ,EAAE;aACV,QAAQ,CAAC,iIAAiI,CAAC;KAC/I,EACD,KAAK,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE;QAC9B,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,EAAE,CAAC,CAAC;QAC3C,kBAAkB,EAAE,CAAC;QACrB,MAAM,WAAW,GAAG,QAAQ,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC;QACpD,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;QAEzD,MAAM,OAAO,GAA4B,EAAE,CAAC;QAE5C,IAAI,UAAU,IAAI,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7E,OAAO,CAAC,QAAQ,GAAG,MAAM,aAAa,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QACtE,CAAC;QAED,IAAI,UAAU,IAAI,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5C,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;YACzD,OAAO,CAAC,IAAI,GAAG,eAAe,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC;QACjE,CAAC;QAED,IAAI,UAAU,IAAI,OAAO,EAAE,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC9C,OAAO,CAAC,MAAM,GAAG,MAAM,WAAW,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,UAAU,IAAI,OAAO,EAAE,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,EAAE,CAAC;YACnC,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,YAAY,GAAG,IAAI,YAAY,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;gBACjE,OAAO,CAAC,OAAO,GAAG,gBAAgB,CAAC,cAAc,EAAE,YAAY,CAAC,CAAC;YACnE,CAAC;QACH,CAAC;QAED,IAAI,UAAU,IAAI,OAAO,EAAE,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC;YAClD,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,KAAK,CAAC,CAAC;YACjD,cAAc,CAAC,KAAK,EAAE,CAAC;YACvB,OAAO,CAAC,UAAU,GAAG,MAAM,kBAAkB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QACvE,CAAC;QAED,IAAI,UAAU,IAAI,OAAO,EAAE,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/C,OAAO,CAAC,OAAO,GAAG,mBAAmB,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,UAAU,IAAI,OAAO,EAAE,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YACjD,OAAO,CAAC,SAAS,GAAG,mBAAmB,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;QAED,IAAI,OAAO,EAAE,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YACnC,OAAO,CAAC,SAAS,GAAG,iBAAiB,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,OAAO,CAAC,cAAc,GAAG,cAAc,CAAC,KAAK,EAAE,CAAC;QAEhD,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,cAAc,EAAE,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC;QACrF,OAAO,OAAO,CAAC,OAAO,CAAC,CAAC;IAC1B,CAAC,CACF,CAAC;AACJ,CAAC"}
|
|
@@ -28,8 +28,8 @@ export declare const SiebelObjectTypeSchema: z.ZodEnum<{
|
|
|
28
28
|
* Siebel environment type.
|
|
29
29
|
*/
|
|
30
30
|
export declare const SiebelEnvironmentTypeSchema: z.ZodEnum<{
|
|
31
|
-
dev: "dev";
|
|
32
31
|
test: "test";
|
|
32
|
+
dev: "dev";
|
|
33
33
|
staging: "staging";
|
|
34
34
|
prod: "prod";
|
|
35
35
|
}>;
|
|
@@ -41,8 +41,8 @@ export declare const SiebelEnvironmentSchema: z.ZodObject<{
|
|
|
41
41
|
url: z.ZodString;
|
|
42
42
|
version: z.ZodString;
|
|
43
43
|
type: z.ZodEnum<{
|
|
44
|
-
dev: "dev";
|
|
45
44
|
test: "test";
|
|
45
|
+
dev: "dev";
|
|
46
46
|
staging: "staging";
|
|
47
47
|
prod: "prod";
|
|
48
48
|
}>;
|