@echoes-io/mcp-server 7.1.0 → 8.0.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/README.md +2 -2
- package/cli/program.d.ts.map +1 -1
- package/cli/program.js +5 -5
- package/cli/program.js.map +1 -1
- package/lib/constants.d.ts +0 -3
- package/lib/constants.d.ts.map +1 -1
- package/lib/constants.js +1 -7
- package/lib/constants.js.map +1 -1
- package/lib/prompts/arc-resume.d.ts +2 -0
- package/lib/prompts/arc-resume.d.ts.map +1 -0
- package/lib/{tools → prompts}/arc-resume.js +51 -44
- package/lib/prompts/arc-resume.js.map +1 -0
- package/lib/prompts/index.d.ts +1 -0
- package/lib/prompts/index.d.ts.map +1 -1
- package/lib/prompts/index.js +28 -0
- package/lib/prompts/index.js.map +1 -1
- package/lib/rag/index.d.ts +13 -0
- package/lib/rag/index.d.ts.map +1 -0
- package/lib/rag/index.js +54 -0
- package/lib/rag/index.js.map +1 -0
- package/lib/rag/parser.d.ts +8 -0
- package/lib/rag/parser.d.ts.map +1 -0
- package/lib/rag/parser.js +39 -0
- package/lib/rag/parser.js.map +1 -0
- package/lib/rag/schema.d.ts +6 -0
- package/lib/rag/schema.d.ts.map +1 -0
- package/lib/rag/schema.js +49 -0
- package/lib/rag/schema.js.map +1 -0
- package/lib/server.d.ts.map +1 -1
- package/lib/server.js +0 -12
- package/lib/server.js.map +1 -1
- package/lib/tools/consistency/index.d.ts +1 -1
- package/lib/tools/consistency/rules/entity-duplicate.d.ts.map +1 -1
- package/lib/tools/consistency/rules/entity-duplicate.js +3 -5
- package/lib/tools/consistency/rules/entity-duplicate.js.map +1 -1
- package/lib/tools/consistency/rules/first-time-content.d.ts.map +1 -1
- package/lib/tools/consistency/rules/first-time-content.js +7 -2
- package/lib/tools/consistency/rules/first-time-content.js.map +1 -1
- package/lib/tools/consistency/rules/relation-jump.d.ts.map +1 -1
- package/lib/tools/consistency/rules/relation-jump.js +11 -8
- package/lib/tools/consistency/rules/relation-jump.js.map +1 -1
- package/lib/tools/consistency/types.d.ts.map +1 -1
- package/lib/tools/graph-export.d.ts +1 -2
- package/lib/tools/graph-export.d.ts.map +1 -1
- package/lib/tools/graph-export.js +49 -116
- package/lib/tools/graph-export.js.map +1 -1
- package/lib/tools/history.d.ts +1 -1
- package/lib/tools/history.d.ts.map +1 -1
- package/lib/tools/history.js +43 -55
- package/lib/tools/history.js.map +1 -1
- package/lib/tools/index.d.ts +7 -2
- package/lib/tools/index.d.ts.map +1 -1
- package/lib/tools/index.js +28 -4
- package/lib/tools/index.js.map +1 -1
- package/lib/tools/list.d.ts +1 -1
- package/lib/tools/list.d.ts.map +1 -1
- package/lib/tools/list.js +35 -30
- package/lib/tools/list.js.map +1 -1
- package/lib/tools/review-apply.d.ts.map +1 -1
- package/lib/tools/review-apply.js +49 -207
- package/lib/tools/review-apply.js.map +1 -1
- package/lib/tools/review-generate.d.ts.map +1 -1
- package/lib/tools/review-generate.js +44 -55
- package/lib/tools/review-generate.js.map +1 -1
- package/lib/tools/review-status.d.ts +12 -17
- package/lib/tools/review-status.d.ts.map +1 -1
- package/lib/tools/review-status.js +21 -33
- package/lib/tools/review-status.js.map +1 -1
- package/lib/tools/search.d.ts +1 -1
- package/lib/tools/search.d.ts.map +1 -1
- package/lib/tools/search.js +62 -44
- package/lib/tools/search.js.map +1 -1
- package/lib/tools/stats.d.ts.map +1 -1
- package/lib/tools/stats.js +43 -7
- package/lib/tools/stats.js.map +1 -1
- package/lib/types.d.ts.map +1 -1
- package/package.json +8 -6
- package/lib/database/index.d.ts +0 -36
- package/lib/database/index.d.ts.map +0 -1
- package/lib/database/index.js +0 -287
- package/lib/database/index.js.map +0 -1
- package/lib/database/schemas.d.ts +0 -55
- package/lib/database/schemas.d.ts.map +0 -1
- package/lib/database/schemas.js +0 -78
- package/lib/database/schemas.js.map +0 -1
- package/lib/indexer/embeddings.d.ts +0 -10
- package/lib/indexer/embeddings.d.ts.map +0 -1
- package/lib/indexer/embeddings.js +0 -64
- package/lib/indexer/embeddings.js.map +0 -1
- package/lib/indexer/extractor.d.ts +0 -81
- package/lib/indexer/extractor.d.ts.map +0 -1
- package/lib/indexer/extractor.js +0 -68
- package/lib/indexer/extractor.js.map +0 -1
- package/lib/indexer/scanner.d.ts +0 -8
- package/lib/indexer/scanner.d.ts.map +0 -1
- package/lib/indexer/scanner.js +0 -73
- package/lib/indexer/scanner.js.map +0 -1
- package/lib/indexer/tasks.d.ts +0 -32
- package/lib/indexer/tasks.d.ts.map +0 -1
- package/lib/indexer/tasks.js +0 -235
- package/lib/indexer/tasks.js.map +0 -1
- package/lib/tools/arc-resume.d.ts +0 -26
- package/lib/tools/arc-resume.d.ts.map +0 -1
- package/lib/tools/arc-resume.js.map +0 -1
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
import { defineSchema } from '@flowrag/core';
|
|
2
|
+
export const ENTITY_TYPES = ['CHARACTER', 'LOCATION', 'EVENT', 'OBJECT', 'EMOTION'];
|
|
3
|
+
export const RELATION_TYPES = [
|
|
4
|
+
'LOVES',
|
|
5
|
+
'HATES',
|
|
6
|
+
'KNOWS',
|
|
7
|
+
'RELATED_TO',
|
|
8
|
+
'FRIENDS_WITH',
|
|
9
|
+
'ENEMIES_WITH',
|
|
10
|
+
'LOCATED_IN',
|
|
11
|
+
'LIVES_IN',
|
|
12
|
+
'TRAVELS_TO',
|
|
13
|
+
'HAPPENS_BEFORE',
|
|
14
|
+
'HAPPENS_AFTER',
|
|
15
|
+
'CAUSES',
|
|
16
|
+
'OWNS',
|
|
17
|
+
'USES',
|
|
18
|
+
'SEEKS',
|
|
19
|
+
];
|
|
20
|
+
export const narrativeSchema = defineSchema({
|
|
21
|
+
entityTypes: ENTITY_TYPES,
|
|
22
|
+
relationTypes: RELATION_TYPES,
|
|
23
|
+
documentFields: {
|
|
24
|
+
arc: { type: 'string', filterable: true },
|
|
25
|
+
episode: { type: 'string', filterable: true },
|
|
26
|
+
chapter: { type: 'string', filterable: true },
|
|
27
|
+
pov: { type: 'string', filterable: true },
|
|
28
|
+
title: { type: 'string' },
|
|
29
|
+
location: { type: 'string', filterable: true },
|
|
30
|
+
date: { type: 'string' },
|
|
31
|
+
file_path: { type: 'string' },
|
|
32
|
+
file_hash: { type: 'string' },
|
|
33
|
+
word_count: { type: 'string' },
|
|
34
|
+
},
|
|
35
|
+
entityFields: {
|
|
36
|
+
aliases: { type: 'string' },
|
|
37
|
+
review_status: {
|
|
38
|
+
type: 'enum',
|
|
39
|
+
values: ['pending', 'approved', 'rejected'],
|
|
40
|
+
default: 'approved',
|
|
41
|
+
},
|
|
42
|
+
reviewed_at: { type: 'string' },
|
|
43
|
+
original_extraction: { type: 'string' },
|
|
44
|
+
},
|
|
45
|
+
relationFields: {
|
|
46
|
+
weight: { type: 'string', default: '0.5' },
|
|
47
|
+
},
|
|
48
|
+
});
|
|
49
|
+
//# sourceMappingURL=schema.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"schema.js","sourceRoot":"","sources":["schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAE7C,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,WAAW,EAAE,UAAU,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,CAAU,CAAC;AAG7F,MAAM,CAAC,MAAM,cAAc,GAAG;IAC5B,OAAO;IACP,OAAO;IACP,OAAO;IACP,YAAY;IACZ,cAAc;IACd,cAAc;IACd,YAAY;IACZ,UAAU;IACV,YAAY;IACZ,gBAAgB;IAChB,eAAe;IACf,QAAQ;IACR,MAAM;IACN,MAAM;IACN,OAAO;CACC,CAAC;AAGX,MAAM,CAAC,MAAM,eAAe,GAAG,YAAY,CAAC;IAC1C,WAAW,EAAE,YAAY;IACzB,aAAa,EAAE,cAAc;IAC7B,cAAc,EAAE;QACd,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE;QACzC,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE;QAC7C,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE;QAC7C,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE;QACzC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QACzB,QAAQ,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE;QAC9C,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QACxB,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QAC7B,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QAC7B,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;KAC/B;IACD,YAAY,EAAE;QACZ,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QAC3B,aAAa,EAAE;YACb,IAAI,EAAE,MAAM;YACZ,MAAM,EAAE,CAAC,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC;YAC3C,OAAO,EAAE,UAAU;SACpB;QACD,WAAW,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;QAC/B,mBAAmB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;KACxC;IACD,cAAc,EAAE;QACd,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE;KAC3C;CACF,CAAC,CAAC"}
|
package/lib/server.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;
|
|
1
|
+
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAoBpE,KAAK,UAAU,GAAG;IAAE,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAAC,OAAO,CAAC,EAAE,OAAO,CAAA;CAAE,CAAC;AAMxF,wBAAgB,WAAW,CAAC,GAAG,EAAE,OAAO,GAAG,UAAU,CAGpD;AAED,wBAAgB,YAAY,IAAI,SAAS,CA8KxC;AAGD,wBAAsB,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC,CAIjD"}
|
package/lib/server.js
CHANGED
|
@@ -1,7 +1,6 @@
|
|
|
1
1
|
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
2
2
|
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
|
|
3
3
|
import { getPrompt, PROMPTS } from './prompts/index.js';
|
|
4
|
-
import { arcResume, arcResumeConfig, arcResumeSchema } from './tools/arc-resume.js';
|
|
5
4
|
import { graphExport, graphExportConfig, graphExportSchema } from './tools/graph-export.js';
|
|
6
5
|
import { history, historyConfig, historySchema } from './tools/history.js';
|
|
7
6
|
import { index, indexConfig, indexSchema } from './tools/index.js';
|
|
@@ -135,17 +134,6 @@ export function createServer() {
|
|
|
135
134
|
return formatError(err);
|
|
136
135
|
}
|
|
137
136
|
});
|
|
138
|
-
server.registerTool(arcResumeConfig.name, {
|
|
139
|
-
description: arcResumeConfig.description,
|
|
140
|
-
inputSchema: arcResumeSchema,
|
|
141
|
-
}, (args) => {
|
|
142
|
-
try {
|
|
143
|
-
return success(arcResume(args));
|
|
144
|
-
}
|
|
145
|
-
catch (err) {
|
|
146
|
-
return formatError(err);
|
|
147
|
-
}
|
|
148
|
-
});
|
|
149
137
|
/* v8 ignore stop */
|
|
150
138
|
// Register prompts
|
|
151
139
|
for (const prompt of PROMPTS) {
|
package/lib/server.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"server.js","sourceRoot":"","sources":["server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAEjF,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AAEjF,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5F,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAC3E,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AACnE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5F,OAAO,EACL,cAAc,EACd,oBAAoB,EACpB,oBAAoB,GACrB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAChG,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACvE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AACnE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AACxF,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAI9C,SAAS,OAAO,CAAC,IAAa;IAC5B,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;AAC9E,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,GAAY;IACtC,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IACjE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,OAAO,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AACnF,CAAC;AAED,MAAM,UAAU,YAAY;IAC1B,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,gBAAgB,EAAE,CAAC;IAE1D,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;IAE7D,iBAAiB;IACjB,MAAM,CAAC,YAAY,CACjB,gBAAgB,CAAC,IAAI,EACrB;QACE,WAAW,EAAE,gBAAgB,CAAC,WAAW;QACzC,WAAW,EAAE,gBAAgB;KAC9B,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,OAAO,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,WAAW,CAAC,IAAI,EAChB;QACE,WAAW,EAAE,WAAW,CAAC,WAAW;QACpC,WAAW,EAAE,WAAW;KACzB,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,OAAO,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,WAAW,CAAC,IAAI,EAChB;QACE,WAAW,EAAE,WAAW,CAAC,WAAW;QACpC,WAAW,EAAE,WAAW;KACzB,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,OAAO,OAAO,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,YAAY,CAAC,IAAI,EACjB;QACE,WAAW,EAAE,YAAY,CAAC,WAAW;QACrC,WAAW,EAAE,YAAY;QACzB,qBAAqB;KACtB,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,OAAO,OAAO,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACrC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,UAAU,CAAC,IAAI,EACf;QACE,WAAW,EAAE,UAAU,CAAC,WAAW;QACnC,WAAW,EAAE,UAAU;KACxB,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,OAAO,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,iBAAiB,CAAC,IAAI,EACtB;QACE,WAAW,EAAE,iBAAiB,CAAC,WAAW;QAC1C,WAAW,EAAE,iBAAiB;KAC/B,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,OAAO,OAAO,CAAC,MAAM,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,aAAa,CAAC,IAAI,EAClB;QACE,WAAW,EAAE,aAAa,CAAC,WAAW;QACtC,WAAW,EAAE,aAAa;KAC3B,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,OAAO,OAAO,CAAC,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,oBAAoB,CAAC,IAAI,EACzB;QACE,WAAW,EAAE,oBAAoB,CAAC,WAAW;QAC7C,WAAW,EAAE,oBAAoB;KAClC,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,OAAO,OAAO,CAAC,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,kBAAkB,CAAC,IAAI,EACvB;QACE,WAAW,EAAE,kBAAkB,CAAC,WAAW;QAC3C,WAAW,EAAE,kBAAkB;KAChC,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,OAAO,OAAO,CAAC,MAAM,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CACF,CAAC;IAEF,MAAM,CAAC,YAAY,CACjB,iBAAiB,CAAC,IAAI,EACtB;QACE,WAAW,EAAE,iBAAiB,CAAC,WAAW;QAC1C,WAAW,EAAE,iBAAiB;KAC/B,EACD,KAAK,EAAE,IAAI,EAAE,EAAE;QACb,IAAI,CAAC;YACH,OAAO,OAAO,CAAC,MAAM,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC,CACF,CAAC;IACF,oBAAoB;IAEpB,mBAAmB;IACnB,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;YACnE,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,IAA8B,CAAC,CAAC;gBACpE,OAAO,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC;YAC3E,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACtF,OAAO;oBACL,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,OAAO,EAAE,EAAE,EAAE,CAAC;iBACnF,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,qBAAqB;AACrB,MAAM,CAAC,KAAK,UAAU,WAAW;IAC/B,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;IAC9B,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;AAClC,CAAC;AACD,oBAAoB"}
|
|
@@ -13,8 +13,8 @@ export declare const checkConsistencySchema: z.ZodObject<{
|
|
|
13
13
|
"entity-duplicate": "entity-duplicate";
|
|
14
14
|
}>>>;
|
|
15
15
|
severity: z.ZodOptional<z.ZodEnum<{
|
|
16
|
-
warning: "warning";
|
|
17
16
|
error: "error";
|
|
17
|
+
warning: "warning";
|
|
18
18
|
info: "info";
|
|
19
19
|
}>>;
|
|
20
20
|
dbPath: z.ZodDefault<z.ZodString>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"entity-duplicate.d.ts","sourceRoot":"","sources":["entity-duplicate.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AA4EzC,wBAAsB,oBAAoB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,
|
|
1
|
+
{"version":3,"file":"entity-duplicate.d.ts","sourceRoot":"","sources":["entity-duplicate.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AA4EzC,wBAAsB,oBAAoB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAmBxF"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { createEchoesRAG } from '../../../rag/index.js';
|
|
2
2
|
// Simple Levenshtein distance
|
|
3
3
|
function levenshtein(a, b) {
|
|
4
4
|
const matrix = [];
|
|
@@ -62,10 +62,8 @@ function groupSimilarEntities(names) {
|
|
|
62
62
|
return groups;
|
|
63
63
|
}
|
|
64
64
|
export async function checkEntityDuplicate(dbPath, arc) {
|
|
65
|
-
const
|
|
66
|
-
await
|
|
67
|
-
const entities = await db.getEntities(arc);
|
|
68
|
-
db.close();
|
|
65
|
+
const { storage } = createEchoesRAG({ dbPath, arc });
|
|
66
|
+
const entities = await storage.graph.getEntities();
|
|
69
67
|
if (entities.length === 0)
|
|
70
68
|
return [];
|
|
71
69
|
const names = entities.map((e) => e.name);
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"entity-duplicate.js","sourceRoot":"","sources":["entity-duplicate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"entity-duplicate.js","sourceRoot":"","sources":["entity-duplicate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAGxD,8BAA8B;AAC9B,SAAS,WAAW,CAAC,CAAS,EAAE,CAAS;IACvC,MAAM,MAAM,GAAe,EAAE,CAAC;IAE9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IACD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;IACnB,CAAC;IAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBACxC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YACtC,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CACrB,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EACxB,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EACpB,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CACrB,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;AACpC,CAAC;AAED,yCAAyC;AACzC,SAAS,UAAU,CAAC,KAAa,EAAE,KAAa,EAAE,SAAS,GAAG,CAAC;IAC7D,MAAM,EAAE,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAC/B,MAAM,EAAE,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;IAE/B,iCAAiC;IACjC,IAAI,EAAE,KAAK,EAAE;QAAE,OAAO,KAAK,CAAC,CAAC,mCAAmC;IAEhE,6EAA6E;IAC7E,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC;QAAE,OAAO,IAAI,CAAC;IACrD,IAAI,EAAE,CAAC,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC;QAAE,OAAO,IAAI,CAAC;IAErD,uDAAuD;IACvD,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC;IAChD,IAAI,OAAO,IAAI,CAAC,IAAI,WAAW,CAAC,EAAE,EAAE,EAAE,CAAC,IAAI,SAAS;QAAE,OAAO,IAAI,CAAC;IAElE,OAAO,KAAK,CAAC;AACf,CAAC;AAED,yBAAyB;AACzB,SAAS,oBAAoB,CAAC,KAAe;IAC3C,MAAM,MAAM,GAAe,EAAE,CAAC;IAC9B,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAE/B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE,SAAS;QAE7B,MAAM,KAAK,GAAG,CAAC,IAAI,CAAC,CAAC;QACrB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAEf,KAAK,MAAM,KAAK,IAAI,KAAK,EAAE,CAAC;YAC1B,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC;gBAAE,SAAS;YAC9B,IAAI,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC;gBAC5B,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBAClB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAClB,CAAC;QACH,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,MAAc,EAAE,GAAW;IACpE,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;IACrD,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;IAEnD,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAErC,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;IAE3C,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5B,IAAI,EAAE,kBAA2B;QACjC,QAAQ,EAAE,MAAe;QACzB,OAAO,EAAE,gCAAgC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;QAC3D,OAAO,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE;QACxC,OAAO,EAAE;YACP,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,oCAAoC;SACjD;KACF,CAAC,CAAC,CAAC;AACN,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"first-time-content.d.ts","sourceRoot":"","sources":["first-time-content.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"first-time-content.d.ts","sourceRoot":"","sources":["first-time-content.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EAAc,KAAK,EAAE,MAAM,aAAa,CAAC;AA0FrD,wBAAsB,qBAAqB,CACzC,WAAW,EAAE,MAAM,EACnB,GAAG,EAAE,MAAM,EACX,mBAAmB,SAAO,GACzB,OAAO,CAAC,KAAK,EAAE,CAAC,CA8ClB"}
|
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
import { readdirSync, readFileSync } from 'node:fs';
|
|
2
2
|
import { join } from 'node:path';
|
|
3
|
-
import {
|
|
3
|
+
import { LocalEmbedder } from '@flowrag/provider-local';
|
|
4
|
+
import { DEFAULT_EMBEDDING_MODEL } from '../../../constants.js';
|
|
4
5
|
import { parseMarkdown } from '../../../utils.js';
|
|
5
6
|
// Patterns to find "first time" claims in content
|
|
6
7
|
const FIRST_TIME_PATTERNS = [
|
|
@@ -80,7 +81,11 @@ export async function checkFirstTimeContent(contentPath, arc, similarityThreshol
|
|
|
80
81
|
return [];
|
|
81
82
|
// Generate embeddings for all claims
|
|
82
83
|
const texts = claims.map((c) => c.text);
|
|
83
|
-
const
|
|
84
|
+
const embedder = new LocalEmbedder({
|
|
85
|
+
model: process.env.ECHOES_EMBEDDING_MODEL ?? DEFAULT_EMBEDDING_MODEL,
|
|
86
|
+
dtype: process.env.ECHOES_EMBEDDING_DTYPE ?? 'fp32',
|
|
87
|
+
});
|
|
88
|
+
const embeddings = await embedder.embedBatch(texts);
|
|
84
89
|
const issues = [];
|
|
85
90
|
const reported = new Set();
|
|
86
91
|
// Find similar claims
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"first-time-content.js","sourceRoot":"","sources":["first-time-content.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACpD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"first-time-content.js","sourceRoot":"","sources":["first-time-content.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACpD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAExD,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGlD,kDAAkD;AAClD,MAAM,mBAAmB,GAAG;IAC1B,6BAA6B;IAC7B,6BAA6B;IAC7B,6BAA6B;IAC7B,gDAAgD;IAChD,6BAA6B;IAC7B,gCAAgC;CACjC,CAAC;AAQF,SAAS,sBAAsB,CAAC,OAAe,EAAE,GAAe;IAC9D,MAAM,MAAM,GAAqB,EAAE,CAAC;IAEpC,KAAK,MAAM,OAAO,IAAI,mBAAmB,EAAE,CAAC;QAC1C,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QAC1C,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAC7B,sDAAsD;YACtD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;YACnD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;YAChF,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;YAEtE,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,wBAAwB,CAAC,WAAmB,EAAE,GAAW;IAChE,MAAM,MAAM,GAAqB,EAAE,CAAC;IAEpC,SAAS,OAAO,CAAC,GAAW;QAC1B,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,GAAG,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;YAC9D,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YACvC,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;gBACxB,OAAO,CAAC,QAAQ,CAAC,CAAC;YACpB,CAAC;iBAAM,IAAI,KAAK,CAAC,MAAM,EAAE,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBACxD,IAAI,CAAC;oBACH,MAAM,GAAG,GAAG,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;oBAC5C,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;oBAEjD,IAAI,QAAQ,CAAC,GAAG,KAAK,GAAG;wBAAE,SAAS;oBAEnC,MAAM,GAAG,GAAe;wBACtB,GAAG,EAAE,QAAQ,CAAC,GAAG;wBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,CAAC;wBAC9B,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,CAAC;qBAC/B,CAAC;oBAEF,MAAM,CAAC,IAAI,CAAC,GAAG,sBAAsB,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;gBACvD,CAAC;gBAAC,MAAM,CAAC;oBACP,yBAAyB;gBAC3B,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,CAAC,WAAW,CAAC,CAAC;IAErB,gCAAgC;IAChC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QACnB,IAAI,CAAC,CAAC,GAAG,CAAC,OAAO,KAAK,CAAC,CAAC,GAAG,CAAC,OAAO;YAAE,OAAO,CAAC,CAAC,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC;QAC1E,OAAO,CAAC,CAAC,GAAG,CAAC,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,wCAAwC;AACxC,SAAS,gBAAgB,CAAC,CAAW,EAAE,CAAW;IAChD,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACnB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,CAAC;IACD,OAAO,GAAG,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;AACrD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACzC,WAAmB,EACnB,GAAW,EACX,mBAAmB,GAAG,IAAI;IAE1B,MAAM,MAAM,GAAG,wBAAwB,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;IAE1D,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC;QAAE,OAAO,EAAE,CAAC;IAEjC,qCAAqC;IACrC,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC;QACjC,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,uBAAuB;QACpE,KAAK,EAAG,OAAO,CAAC,GAAG,CAAC,sBAA+C,IAAI,MAAM;KAC9E,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,MAAM,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IAEpD,MAAM,MAAM,GAAY,EAAE,CAAC;IAC3B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;IAEnC,sBAAsB;IACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACvC,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3C,MAAM,UAAU,GAAG,gBAAgB,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;YAElE,IAAI,UAAU,IAAI,mBAAmB,EAAE,CAAC;gBACtC,MAAM,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxB,IAAI,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC;oBAAE,SAAS;gBAChC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBAElB,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;gBAC1B,MAAM,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;gBAE3B,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,sBAAsB;oBAC5B,QAAQ,EAAE,MAAM;oBAChB,OAAO,EAAE,mCAAmC;oBAC5C,OAAO,EAAE,OAAO,CAAC,GAAG;oBACpB,QAAQ,EAAE,QAAQ,CAAC,GAAG;oBACtB,OAAO,EAAE;wBACP,WAAW,EAAE,OAAO,CAAC,IAAI;wBACzB,YAAY,EAAE,QAAQ,CAAC,IAAI;wBAC3B,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,GAAG,CAAC,GAAG,GAAG;qBAC/C;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"relation-jump.d.ts","sourceRoot":"","sources":["relation-jump.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAiCzC,wBAAsB,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,
|
|
1
|
+
{"version":3,"file":"relation-jump.d.ts","sourceRoot":"","sources":["relation-jump.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAiCzC,wBAAsB,iBAAiB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,KAAK,EAAE,CAAC,CAyGrF"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import { createEchoesRAG } from '../../../rag/index.js';
|
|
2
2
|
// Parse chapter ID to get episode and chapter numbers
|
|
3
3
|
function parseChapterId(id) {
|
|
4
4
|
const parts = id.split(':');
|
|
@@ -19,18 +19,21 @@ function isDrasticTypeChange(from, to) {
|
|
|
19
19
|
return (fromPositive && toNegative) || (fromNegative && toPositive);
|
|
20
20
|
}
|
|
21
21
|
export async function checkRelationJump(dbPath, arc) {
|
|
22
|
-
const
|
|
23
|
-
await
|
|
24
|
-
const relations =
|
|
25
|
-
|
|
22
|
+
const { storage } = createEchoesRAG({ dbPath, arc });
|
|
23
|
+
const entities = await storage.graph.getEntities();
|
|
24
|
+
const relations = [];
|
|
25
|
+
for (const entity of entities) {
|
|
26
|
+
const rels = await storage.graph.getRelations(entity.id, 'out');
|
|
27
|
+
relations.push(...rels);
|
|
28
|
+
}
|
|
26
29
|
if (relations.length === 0)
|
|
27
30
|
return [];
|
|
28
31
|
const issues = [];
|
|
29
32
|
// Group relations by source+target pair
|
|
30
33
|
const pairs = new Map();
|
|
31
34
|
for (const rel of relations) {
|
|
32
|
-
const key = `${rel.
|
|
33
|
-
for (const chapterId of rel.
|
|
35
|
+
const key = `${rel.sourceId}:${rel.targetId}`;
|
|
36
|
+
for (const chapterId of rel.sourceChunkIds) {
|
|
34
37
|
const { episode, chapter } = parseChapterId(chapterId);
|
|
35
38
|
if (!pairs.has(key)) {
|
|
36
39
|
pairs.set(key, []);
|
|
@@ -40,7 +43,7 @@ export async function checkRelationJump(dbPath, arc) {
|
|
|
40
43
|
pairArray.push({
|
|
41
44
|
chapter: chapterId,
|
|
42
45
|
type: rel.type,
|
|
43
|
-
weight: rel.weight,
|
|
46
|
+
weight: Number(rel.fields?.weight ?? 0.5),
|
|
44
47
|
episode,
|
|
45
48
|
chapterNum: chapter,
|
|
46
49
|
});
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"relation-jump.js","sourceRoot":"","sources":["relation-jump.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"relation-jump.js","sourceRoot":"","sources":["relation-jump.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAWxD,sDAAsD;AACtD,SAAS,cAAc,CAAC,EAAU;IAChC,MAAM,KAAK,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAC5B,OAAO;QACL,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC;QAC3C,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC;KAC5C,CAAC;AACJ,CAAC;AAED,6CAA6C;AAC7C,SAAS,mBAAmB,CAAC,IAAY,EAAE,EAAU;IACnD,MAAM,QAAQ,GAAG,CAAC,OAAO,EAAE,cAAc,EAAE,OAAO,CAAC,CAAC;IACpD,MAAM,QAAQ,GAAG,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;IAE3C,MAAM,YAAY,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC7C,MAAM,YAAY,GAAG,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC7C,MAAM,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACzC,MAAM,UAAU,GAAG,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEzC,gDAAgD;IAChD,OAAO,CAAC,YAAY,IAAI,UAAU,CAAC,IAAI,CAAC,YAAY,IAAI,UAAU,CAAC,CAAC;AACtE,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,MAAc,EAAE,GAAW;IACjE,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;IAErD,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;IACnD,MAAM,SAAS,GAMT,EAAE,CAAC;IACT,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAChE,SAAS,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAEtC,MAAM,MAAM,GAAY,EAAE,CAAC;IAE3B,wCAAwC;IACxC,MAAM,KAAK,GAAG,IAAI,GAAG,EAA4B,CAAC;IAElD,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;QAC5B,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,QAAQ,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;QAE9C,KAAK,MAAM,SAAS,IAAI,GAAG,CAAC,cAAc,EAAE,CAAC;YAC3C,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC;YAEvD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;gBACpB,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACrB,CAAC;YAED,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACjC,IAAI,SAAS,EAAE,CAAC;gBACd,SAAS,CAAC,IAAI,CAAC;oBACb,OAAO,EAAE,SAAS;oBAClB,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,IAAI,GAAG,CAAC;oBACzC,OAAO;oBACP,UAAU,EAAE,OAAO;iBACpB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,sCAAsC;IACtC,KAAK,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,KAAK,EAAE,CAAC;QACnC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;YAAE,SAAS;QAEjC,gCAAgC;QAChC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACpB,IAAI,CAAC,CAAC,OAAO,KAAK,CAAC,CAAC,OAAO;gBAAE,OAAO,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;YAC1D,OAAO,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,4BAA4B;QAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAC5B,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAExB,gCAAgC;YAChC,IAAI,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9C,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;gBAEvE,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,eAAe;oBACrB,QAAQ,EAAE,SAAS;oBACnB,OAAO,EAAE,YAAY,MAAM,MAAM,MAAM,iBAAiB,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,IAAI,EAAE;oBACnF,OAAO,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;oBACjE,QAAQ,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;oBAClE,OAAO,EAAE;wBACP,MAAM;wBACN,MAAM;wBACN,YAAY,EAAE,IAAI,CAAC,IAAI;wBACvB,WAAW,EAAE,IAAI,CAAC,IAAI;wBACtB,cAAc,EAAE,IAAI,CAAC,MAAM;wBAC3B,aAAa,EAAE,IAAI,CAAC,MAAM;qBAC3B;iBACF,CAAC,CAAC;YACL,CAAC;YAED,uCAAuC;YACvC,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;gBAC/D,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;gBAEvE,MAAM,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,eAAe;oBACrB,QAAQ,EAAE,MAAM;oBAChB,OAAO,EAAE,YAAY,MAAM,MAAM,MAAM,wBAAwB,IAAI,CAAC,MAAM,OAAO,IAAI,CAAC,MAAM,EAAE;oBAC9F,OAAO,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;oBACjE,QAAQ,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE;oBAClE,OAAO,EAAE;wBACP,MAAM;wBACN,MAAM;wBACN,IAAI,EAAE,IAAI,CAAC,IAAI;wBACf,cAAc,EAAE,IAAI,CAAC,MAAM;wBAC3B,aAAa,EAAE,IAAI,CAAC,MAAM;qBAC3B;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["types.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["types.ts"],"names":[],"mappings":"AACA,MAAM,MAAM,aAAa,GAAG,OAAO,GAAG,SAAS,GAAG,MAAM,CAAC;AAEzD,MAAM,MAAM,SAAS,GACjB,sBAAsB,GACtB,sBAAsB,GACtB,sBAAsB,GACtB,eAAe,GACf,kBAAkB,CAAC;AAEvB,MAAM,WAAW,UAAU;IACzB,GAAG,EAAE,MAAM,CAAC;IACZ,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,KAAK;IACpB,IAAI,EAAE,SAAS,CAAC;IAChB,QAAQ,EAAE,aAAa,CAAC;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,UAAU,CAAC;IACpB,QAAQ,CAAC,EAAE,UAAU,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAClC;AAED,MAAM,WAAW,WAAW;IAC1B,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,KAAK,EAAE,CAAC;IAChB,OAAO,EAAE;QACP,MAAM,EAAE,MAAM,CAAC;QACf,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;CACH;AAED,MAAM,MAAM,QAAQ,GAChB,aAAa,GACb,eAAe,GACf,oBAAoB,GACpB,eAAe,GACf,kBAAkB,CAAC;AAEvB,MAAM,MAAM,YAAY,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC"}
|
|
@@ -5,7 +5,6 @@ export declare const graphExportSchema: z.ZodObject<{
|
|
|
5
5
|
arc: z.ZodString;
|
|
6
6
|
format: z.ZodEnum<{
|
|
7
7
|
json: "json";
|
|
8
|
-
mermaid: "mermaid";
|
|
9
8
|
dot: "dot";
|
|
10
9
|
}>;
|
|
11
10
|
entityTypes: z.ZodOptional<z.ZodArray<z.ZodEnum<{
|
|
@@ -37,7 +36,7 @@ export declare const graphExportSchema: z.ZodObject<{
|
|
|
37
36
|
}, z.z.core.$strip>;
|
|
38
37
|
export type GraphExportInput = z.infer<typeof graphExportSchema>;
|
|
39
38
|
export declare function graphExport(input: GraphExportInput): Promise<{
|
|
40
|
-
format: "json" | "
|
|
39
|
+
format: "json" | "dot";
|
|
41
40
|
content: string;
|
|
42
41
|
stats: {
|
|
43
42
|
nodes: number;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"graph-export.d.ts","sourceRoot":"","sources":["graph-export.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,KAAK,CAAC;AAKpB,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAE9C,eAAO,MAAM,iBAAiB,EAAE,UAW/B,CAAC;AAEF,eAAO,MAAM,iBAAiB
|
|
1
|
+
{"version":3,"file":"graph-export.d.ts","sourceRoot":"","sources":["graph-export.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,KAAK,CAAC;AAKpB,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAE9C,eAAO,MAAM,iBAAiB,EAAE,UAW/B,CAAC;AAEF,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;mBAa5B,CAAC;AAEH,MAAM,MAAM,gBAAgB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAmBjE,wBAAsB,WAAW,CAAC,KAAK,EAAE,gBAAgB;;;;;;;GAoCxD"}
|
|
@@ -1,12 +1,13 @@
|
|
|
1
1
|
import z from 'zod';
|
|
2
2
|
import { DEFAULT_DB_PATH } from '../constants.js';
|
|
3
|
-
import {
|
|
3
|
+
import { createEchoesRAG } from '../rag/index.js';
|
|
4
|
+
import { ENTITY_TYPES, RELATION_TYPES } from '../rag/schema.js';
|
|
4
5
|
export const graphExportConfig = {
|
|
5
6
|
name: 'graph-export',
|
|
6
|
-
description: 'Export knowledge graph in
|
|
7
|
+
description: 'Export knowledge graph in JSON or DOT format.',
|
|
7
8
|
arguments: {
|
|
8
9
|
arc: 'Arc name to export.',
|
|
9
|
-
format: 'Output format:
|
|
10
|
+
format: 'Output format: json or dot.',
|
|
10
11
|
entityTypes: 'Filter by entity types (comma-separated).',
|
|
11
12
|
relationTypes: 'Filter by relation types (comma-separated).',
|
|
12
13
|
characters: 'Filter by specific character names (comma-separated).',
|
|
@@ -15,138 +16,70 @@ export const graphExportConfig = {
|
|
|
15
16
|
};
|
|
16
17
|
export const graphExportSchema = z.object({
|
|
17
18
|
arc: z.string().describe(graphExportConfig.arguments.arc),
|
|
18
|
-
format: z.enum(['
|
|
19
|
+
format: z.enum(['json', 'dot']).describe(graphExportConfig.arguments.format),
|
|
19
20
|
entityTypes: z
|
|
20
|
-
.array(z.enum(
|
|
21
|
+
.array(z.enum(ENTITY_TYPES))
|
|
21
22
|
.optional()
|
|
22
23
|
.describe(graphExportConfig.arguments.entityTypes),
|
|
23
24
|
relationTypes: z
|
|
24
|
-
.array(z.enum(
|
|
25
|
-
'LOVES',
|
|
26
|
-
'HATES',
|
|
27
|
-
'KNOWS',
|
|
28
|
-
'RELATED_TO',
|
|
29
|
-
'FRIENDS_WITH',
|
|
30
|
-
'ENEMIES_WITH',
|
|
31
|
-
'LOCATED_IN',
|
|
32
|
-
'LIVES_IN',
|
|
33
|
-
'TRAVELS_TO',
|
|
34
|
-
'HAPPENS_BEFORE',
|
|
35
|
-
'HAPPENS_AFTER',
|
|
36
|
-
'CAUSES',
|
|
37
|
-
'OWNS',
|
|
38
|
-
'USES',
|
|
39
|
-
'SEEKS',
|
|
40
|
-
]))
|
|
25
|
+
.array(z.enum(RELATION_TYPES))
|
|
41
26
|
.optional()
|
|
42
27
|
.describe(graphExportConfig.arguments.relationTypes),
|
|
43
28
|
characters: z.array(z.string()).optional().describe(graphExportConfig.arguments.characters),
|
|
44
29
|
dbPath: z.string().default(DEFAULT_DB_PATH).describe(graphExportConfig.arguments.dbPath),
|
|
45
30
|
});
|
|
46
|
-
async function
|
|
47
|
-
const
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
const
|
|
51
|
-
|
|
52
|
-
const
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
}));
|
|
63
|
-
return { nodes, edges };
|
|
64
|
-
}
|
|
65
|
-
function filterGraph(graph, filters) {
|
|
66
|
-
let { nodes, edges } = graph;
|
|
67
|
-
// Filter nodes by entity types
|
|
68
|
-
if (filters.entityTypes) {
|
|
69
|
-
nodes = nodes.filter((node) => filters.entityTypes?.includes(node.type));
|
|
70
|
-
}
|
|
71
|
-
// Filter nodes by character names
|
|
72
|
-
if (filters.characters) {
|
|
73
|
-
nodes = nodes.filter((node) => filters.characters?.includes(node.name));
|
|
74
|
-
}
|
|
75
|
-
// Filter edges by relation types
|
|
76
|
-
if (filters.relationTypes) {
|
|
77
|
-
edges = edges.filter((edge) => filters.relationTypes?.includes(edge.type));
|
|
78
|
-
}
|
|
79
|
-
// Remove edges that reference filtered-out nodes
|
|
80
|
-
const nodeIds = new Set(nodes.map((node) => node.id));
|
|
81
|
-
edges = edges.filter((edge) => nodeIds.has(edge.source) && nodeIds.has(edge.target));
|
|
82
|
-
return { nodes, edges };
|
|
83
|
-
}
|
|
84
|
-
function toMermaid(graph) {
|
|
85
|
-
const lines = ['graph LR'];
|
|
86
|
-
for (const edge of graph.edges) {
|
|
87
|
-
const source = edge.source.replace(/[^a-zA-Z0-9]/g, '_');
|
|
88
|
-
const target = edge.target.replace(/[^a-zA-Z0-9]/g, '_');
|
|
89
|
-
lines.push(` ${source} -->|${edge.type}| ${target}`);
|
|
31
|
+
export async function graphExport(input) {
|
|
32
|
+
const { arc, format, entityTypes, relationTypes, characters, dbPath } = graphExportSchema.parse(input);
|
|
33
|
+
const { storage } = createEchoesRAG({ dbPath, arc });
|
|
34
|
+
// Build graph from storage
|
|
35
|
+
const entities = await storage.graph.getEntities();
|
|
36
|
+
let nodes = entities.map((e) => ({ id: e.name, name: e.name, type: e.type }));
|
|
37
|
+
const allRelations = [];
|
|
38
|
+
for (const entity of entities) {
|
|
39
|
+
const rels = await storage.graph.getRelations(entity.id, 'out');
|
|
40
|
+
for (const r of rels) {
|
|
41
|
+
const source = entities.find((e) => e.id === r.sourceId);
|
|
42
|
+
const target = entities.find((e) => e.id === r.targetId);
|
|
43
|
+
if (source && target) {
|
|
44
|
+
allRelations.push({ source: source.name, target: target.name, type: r.type });
|
|
45
|
+
}
|
|
46
|
+
}
|
|
90
47
|
}
|
|
91
|
-
|
|
48
|
+
let edges = allRelations;
|
|
49
|
+
// Apply filters
|
|
50
|
+
if (entityTypes)
|
|
51
|
+
nodes = nodes.filter((n) => entityTypes.includes(n.type));
|
|
52
|
+
if (characters)
|
|
53
|
+
nodes = nodes.filter((n) => characters.includes(n.name));
|
|
54
|
+
if (relationTypes)
|
|
55
|
+
edges = edges.filter((e) => relationTypes.includes(e.type));
|
|
56
|
+
const nodeIds = new Set(nodes.map((n) => n.id));
|
|
57
|
+
edges = edges.filter((e) => nodeIds.has(e.source) && nodeIds.has(e.target));
|
|
58
|
+
const graph = { nodes, edges };
|
|
59
|
+
const content = format === 'json' ? toJSON(graph) : toDOT(graph);
|
|
60
|
+
return { format, content, stats: { nodes: graph.nodes.length, edges: graph.edges.length } };
|
|
92
61
|
}
|
|
93
62
|
function toJSON(graph) {
|
|
94
|
-
|
|
95
|
-
nodes: graph.nodes.map((
|
|
96
|
-
id:
|
|
97
|
-
name:
|
|
98
|
-
type:
|
|
99
|
-
group:
|
|
100
|
-
})),
|
|
101
|
-
links: graph.edges.map((edge) => ({
|
|
102
|
-
source: edge.source,
|
|
103
|
-
target: edge.target,
|
|
104
|
-
type: edge.type,
|
|
105
|
-
weight: edge.weight,
|
|
63
|
+
return JSON.stringify({
|
|
64
|
+
nodes: graph.nodes.map((n) => ({
|
|
65
|
+
id: n.id,
|
|
66
|
+
name: n.name,
|
|
67
|
+
type: n.type,
|
|
68
|
+
group: n.type === 'CHARACTER' ? 1 : n.type === 'LOCATION' ? 2 : 3,
|
|
106
69
|
})),
|
|
107
|
-
|
|
108
|
-
|
|
70
|
+
links: graph.edges.map((e) => ({ source: e.source, target: e.target, type: e.type })),
|
|
71
|
+
}, null, 2);
|
|
109
72
|
}
|
|
110
73
|
function toDOT(graph) {
|
|
111
74
|
const lines = ['digraph G {'];
|
|
112
|
-
|
|
113
|
-
|
|
114
|
-
|
|
115
|
-
lines.push(` "${node.name}" [shape=${shape}];`);
|
|
75
|
+
for (const n of graph.nodes) {
|
|
76
|
+
const shape = n.type === 'CHARACTER' ? 'ellipse' : n.type === 'LOCATION' ? 'box' : 'diamond';
|
|
77
|
+
lines.push(` "${n.name}" [shape=${shape}];`);
|
|
116
78
|
}
|
|
117
|
-
|
|
118
|
-
|
|
119
|
-
lines.push(` "${edge.source}" -> "${edge.target}" [label="${edge.type}"];`);
|
|
79
|
+
for (const e of graph.edges) {
|
|
80
|
+
lines.push(` "${e.source}" -> "${e.target}" [label="${e.type}"];`);
|
|
120
81
|
}
|
|
121
82
|
lines.push('}');
|
|
122
83
|
return lines.join('\n');
|
|
123
84
|
}
|
|
124
|
-
export async function graphExport(input) {
|
|
125
|
-
const { arc, format, entityTypes, relationTypes, characters, dbPath } = graphExportSchema.parse(input);
|
|
126
|
-
// Build graph from database
|
|
127
|
-
let graph = await buildGraph(dbPath, arc);
|
|
128
|
-
// Apply filters
|
|
129
|
-
graph = filterGraph(graph, { entityTypes, relationTypes, characters });
|
|
130
|
-
// Generate output based on format
|
|
131
|
-
let content;
|
|
132
|
-
switch (format) {
|
|
133
|
-
case 'mermaid':
|
|
134
|
-
content = toMermaid(graph);
|
|
135
|
-
break;
|
|
136
|
-
case 'json':
|
|
137
|
-
content = toJSON(graph);
|
|
138
|
-
break;
|
|
139
|
-
case 'dot':
|
|
140
|
-
content = toDOT(graph);
|
|
141
|
-
break;
|
|
142
|
-
}
|
|
143
|
-
return {
|
|
144
|
-
format,
|
|
145
|
-
content,
|
|
146
|
-
stats: {
|
|
147
|
-
nodes: graph.nodes.length,
|
|
148
|
-
edges: graph.edges.length,
|
|
149
|
-
},
|
|
150
|
-
};
|
|
151
|
-
}
|
|
152
85
|
//# sourceMappingURL=graph-export.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"graph-export.js","sourceRoot":"","sources":["graph-export.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"graph-export.js","sourceRoot":"","sources":["graph-export.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,KAAK,CAAC;AAEpB,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAGhE,MAAM,CAAC,MAAM,iBAAiB,GAAe;IAC3C,IAAI,EAAE,cAAc;IACpB,WAAW,EAAE,+CAA+C;IAC5D,SAAS,EAAE;QACT,GAAG,EAAE,qBAAqB;QAC1B,MAAM,EAAE,6BAA6B;QACrC,WAAW,EAAE,2CAA2C;QACxD,aAAa,EAAE,6CAA6C;QAC5D,UAAU,EAAE,uDAAuD;QACnE,MAAM,EAAE,2BAA2B,eAAe,IAAI;KACvD;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC,SAAS,CAAC,GAAG,CAAC;IACzD,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,iBAAiB,CAAC,SAAS,CAAC,MAAM,CAAC;IAC5E,WAAW,EAAE,CAAC;SACX,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;SAC3B,QAAQ,EAAE;SACV,QAAQ,CAAC,iBAAiB,CAAC,SAAS,CAAC,WAAW,CAAC;IACpD,aAAa,EAAE,CAAC;SACb,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;SAC7B,QAAQ,EAAE;SACV,QAAQ,CAAC,iBAAiB,CAAC,SAAS,CAAC,aAAa,CAAC;IACtD,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC,SAAS,CAAC,UAAU,CAAC;IAC3F,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,iBAAiB,CAAC,SAAS,CAAC,MAAM,CAAC;CACzF,CAAC,CAAC;AAqBH,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,KAAuB;IACvD,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,EAAE,GACnE,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAEjC,MAAM,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;IAErD,2BAA2B;IAC3B,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;IACnD,IAAI,KAAK,GAAgB,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IAE3F,MAAM,YAAY,GAAgB,EAAE,CAAC;IACrC,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAChE,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YACrB,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,QAAQ,CAAC,CAAC;YACzD,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,QAAQ,CAAC,CAAC;YACzD,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;gBACrB,YAAY,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YAChF,CAAC;QACH,CAAC;IACH,CAAC;IACD,IAAI,KAAK,GAAG,YAAY,CAAC;IAEzB,gBAAgB;IAChB,IAAI,WAAW;QAAE,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,WAAwB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IACzF,IAAI,UAAU;QAAE,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IACzE,IAAI,aAAa;QAAE,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,aAA0B,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAE7F,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAChD,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;IAE5E,MAAM,KAAK,GAAU,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;IAEtC,MAAM,OAAO,GAAG,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAEjE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC;AAC9F,CAAC;AAED,SAAS,MAAM,CAAC,KAAY;IAC1B,OAAO,IAAI,CAAC,SAAS,CACnB;QACE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7B,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,KAAK,EAAE,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;SAClE,CAAC,CAAC;QACH,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;KACtF,EACD,IAAI,EACJ,CAAC,CACF,CAAC;AACJ,CAAC;AAED,SAAS,KAAK,CAAC,KAAY;IACzB,MAAM,KAAK,GAAG,CAAC,aAAa,CAAC,CAAC;IAC9B,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QAC5B,MAAM,KAAK,GAAG,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;QAC7F,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,YAAY,KAAK,IAAI,CAAC,CAAC;IAClD,CAAC;IACD,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QAC5B,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,SAAS,CAAC,CAAC,MAAM,aAAa,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC;IACxE,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAChB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC"}
|
package/lib/tools/history.d.ts
CHANGED
|
@@ -5,10 +5,10 @@ export declare const historySchema: z.ZodObject<{
|
|
|
5
5
|
arc: z.ZodString;
|
|
6
6
|
character: z.ZodOptional<z.ZodString>;
|
|
7
7
|
only: z.ZodOptional<z.ZodEnum<{
|
|
8
|
-
relations: "relations";
|
|
9
8
|
kinks: "kinks";
|
|
10
9
|
outfits: "outfits";
|
|
11
10
|
locations: "locations";
|
|
11
|
+
relations: "relations";
|
|
12
12
|
}>>;
|
|
13
13
|
search: z.ZodOptional<z.ZodString>;
|
|
14
14
|
dbPath: z.ZodDefault<z.ZodString>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"history.d.ts","sourceRoot":"","sources":["history.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,KAAK,CAAC;AAIpB,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAE9C,eAAO,MAAM,aAAa,EAAE,UAU3B,CAAC;AAEF,eAAO,MAAM,aAAa;;;;;;;;;;;mBASxB,CAAC;AAEH,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAEzD,UAAU,SAAS;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,OAAO,CAAC;CAClB;
|
|
1
|
+
{"version":3,"file":"history.d.ts","sourceRoot":"","sources":["history.ts"],"names":[],"mappings":"AAAA,OAAO,CAAC,MAAM,KAAK,CAAC;AAIpB,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAE9C,eAAO,MAAM,aAAa,EAAE,UAU3B,CAAC;AAEF,eAAO,MAAM,aAAa;;;;;;;;;;;mBASxB,CAAC;AAEH,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,aAAa,CAAC,CAAC;AAEzD,UAAU,SAAS;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,OAAO,CAAC;CAClB;AACD,UAAU,WAAW;IACnB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;CAChB;AACD,UAAU,aAAa;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;CAClB;AACD,UAAU,aAAa;IACrB,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;CACd;AAED,UAAU,aAAa;IACrB,GAAG,EAAE,MAAM,CAAC;IACZ,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,OAAO,EAAE,WAAW,EAAE,CAAC;IACvB,SAAS,EAAE,aAAa,EAAE,CAAC;IAC3B,SAAS,EAAE,aAAa,EAAE,CAAC;CAC5B;AAqBD,wBAAsB,OAAO,CAAC,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC,CAuFzE"}
|