shieldcortex 2.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/LICENSE +21 -0
- package/README.md +282 -0
- package/dashboard/components.json +22 -0
- package/dashboard/eslint.config.mjs +42 -0
- package/dashboard/next.config.ts +7 -0
- package/dashboard/package-lock.json +8053 -0
- package/dashboard/package.json +44 -0
- package/dashboard/postcss.config.mjs +7 -0
- package/dashboard/public/file.svg +1 -0
- package/dashboard/public/globe.svg +1 -0
- package/dashboard/public/next.svg +1 -0
- package/dashboard/public/vercel.svg +1 -0
- package/dashboard/public/window.svg +1 -0
- package/dashboard/scripts/ensure-api.mjs +76 -0
- package/dashboard/src/app/error.tsx +49 -0
- package/dashboard/src/app/favicon.ico +0 -0
- package/dashboard/src/app/globals.css +130 -0
- package/dashboard/src/app/layout.tsx +35 -0
- package/dashboard/src/app/page.tsx +364 -0
- package/dashboard/src/components/Providers.tsx +27 -0
- package/dashboard/src/components/brain/ActivityPulseSystem.tsx +229 -0
- package/dashboard/src/components/brain/BrainMesh.tsx +133 -0
- package/dashboard/src/components/brain/BrainRegions.tsx +254 -0
- package/dashboard/src/components/brain/BrainScene.tsx +255 -0
- package/dashboard/src/components/brain/CategoryLabels.tsx +103 -0
- package/dashboard/src/components/brain/CoreSphere.tsx +215 -0
- package/dashboard/src/components/brain/DataFlowParticles.tsx +123 -0
- package/dashboard/src/components/brain/DataStreamRings.tsx +161 -0
- package/dashboard/src/components/brain/ElectronFlow.tsx +323 -0
- package/dashboard/src/components/brain/HolographicGrid.tsx +235 -0
- package/dashboard/src/components/brain/MemoryLinks.tsx +271 -0
- package/dashboard/src/components/brain/MemoryNode.tsx +245 -0
- package/dashboard/src/components/brain/NeuralPathways.tsx +441 -0
- package/dashboard/src/components/brain/SynapseNodes.tsx +312 -0
- package/dashboard/src/components/brain/TimelineControls.tsx +205 -0
- package/dashboard/src/components/chip/ChipScene.tsx +497 -0
- package/dashboard/src/components/chip/ChipSubstrate.tsx +238 -0
- package/dashboard/src/components/chip/CortexCore.tsx +210 -0
- package/dashboard/src/components/chip/DataBus.tsx +416 -0
- package/dashboard/src/components/chip/MemoryCell.tsx +225 -0
- package/dashboard/src/components/chip/MemoryGrid.tsx +328 -0
- package/dashboard/src/components/chip/QuantumCell.tsx +316 -0
- package/dashboard/src/components/chip/SectionLabel.tsx +113 -0
- package/dashboard/src/components/chip/index.ts +14 -0
- package/dashboard/src/components/controls/ControlPanel.tsx +106 -0
- package/dashboard/src/components/controls/VersionPanel.tsx +185 -0
- package/dashboard/src/components/dashboard/StatsPanel.tsx +164 -0
- package/dashboard/src/components/debug/ActivityLog.tsx +250 -0
- package/dashboard/src/components/debug/DebugPanel.tsx +101 -0
- package/dashboard/src/components/debug/QueryTester.tsx +192 -0
- package/dashboard/src/components/debug/RelationshipGraph.tsx +403 -0
- package/dashboard/src/components/debug/SqlConsole.tsx +319 -0
- package/dashboard/src/components/graph/KnowledgeGraph.tsx +230 -0
- package/dashboard/src/components/graph/OntologyGraph.tsx +631 -0
- package/dashboard/src/components/insights/ActivityHeatmap.tsx +131 -0
- package/dashboard/src/components/insights/InsightsView.tsx +46 -0
- package/dashboard/src/components/insights/KnowledgeMapPanel.tsx +80 -0
- package/dashboard/src/components/insights/QualityPanel.tsx +116 -0
- package/dashboard/src/components/memories/MemoriesView.tsx +150 -0
- package/dashboard/src/components/memories/MemoryCard.tsx +103 -0
- package/dashboard/src/components/memory/MemoryDetail.tsx +325 -0
- package/dashboard/src/components/nav/NavRail.tsx +54 -0
- package/dashboard/src/components/ui/button.tsx +62 -0
- package/dashboard/src/components/ui/card.tsx +92 -0
- package/dashboard/src/components/ui/input.tsx +21 -0
- package/dashboard/src/hooks/useDebouncedValue.ts +24 -0
- package/dashboard/src/hooks/useMemories.ts +458 -0
- package/dashboard/src/hooks/useSuggestions.ts +46 -0
- package/dashboard/src/lib/category-colors.ts +84 -0
- package/dashboard/src/lib/position-algorithm.ts +177 -0
- package/dashboard/src/lib/simplex-noise.ts +217 -0
- package/dashboard/src/lib/store.ts +88 -0
- package/dashboard/src/lib/utils.ts +6 -0
- package/dashboard/src/lib/websocket.ts +249 -0
- package/dashboard/src/types/memory.ts +73 -0
- package/dashboard/tsconfig.json +34 -0
- package/dist/__tests__/consolidation-merge.test.d.ts +9 -0
- package/dist/__tests__/consolidation-merge.test.d.ts.map +1 -0
- package/dist/__tests__/consolidation-merge.test.js +137 -0
- package/dist/__tests__/consolidation-merge.test.js.map +1 -0
- package/dist/__tests__/contradictions.test.d.ts +8 -0
- package/dist/__tests__/contradictions.test.d.ts.map +1 -0
- package/dist/__tests__/contradictions.test.js +78 -0
- package/dist/__tests__/contradictions.test.js.map +1 -0
- package/dist/__tests__/salience-evolution.test.d.ts +7 -0
- package/dist/__tests__/salience-evolution.test.d.ts.map +1 -0
- package/dist/__tests__/salience-evolution.test.js +151 -0
- package/dist/__tests__/salience-evolution.test.js.map +1 -0
- package/dist/__tests__/store.test.d.ts +7 -0
- package/dist/__tests__/store.test.d.ts.map +1 -0
- package/dist/__tests__/store.test.js +582 -0
- package/dist/__tests__/store.test.js.map +1 -0
- package/dist/api/control.d.ts +27 -0
- package/dist/api/control.d.ts.map +1 -0
- package/dist/api/control.js +60 -0
- package/dist/api/control.js.map +1 -0
- package/dist/api/events.d.ts +159 -0
- package/dist/api/events.d.ts.map +1 -0
- package/dist/api/events.js +155 -0
- package/dist/api/events.js.map +1 -0
- package/dist/api/version.d.ts +36 -0
- package/dist/api/version.d.ts.map +1 -0
- package/dist/api/version.js +146 -0
- package/dist/api/version.js.map +1 -0
- package/dist/api/visualization-server.d.ts +11 -0
- package/dist/api/visualization-server.d.ts.map +1 -0
- package/dist/api/visualization-server.js +1186 -0
- package/dist/api/visualization-server.js.map +1 -0
- package/dist/context/project-context.d.ts +57 -0
- package/dist/context/project-context.d.ts.map +1 -0
- package/dist/context/project-context.js +135 -0
- package/dist/context/project-context.js.map +1 -0
- package/dist/database/init.d.ts +49 -0
- package/dist/database/init.d.ts.map +1 -0
- package/dist/database/init.js +567 -0
- package/dist/database/init.js.map +1 -0
- package/dist/defence/__tests__/firewall.test.d.ts +8 -0
- package/dist/defence/__tests__/firewall.test.d.ts.map +1 -0
- package/dist/defence/__tests__/firewall.test.js +123 -0
- package/dist/defence/__tests__/firewall.test.js.map +1 -0
- package/dist/defence/__tests__/fragmentation.test.d.ts +7 -0
- package/dist/defence/__tests__/fragmentation.test.d.ts.map +1 -0
- package/dist/defence/__tests__/fragmentation.test.js +51 -0
- package/dist/defence/__tests__/fragmentation.test.js.map +1 -0
- package/dist/defence/__tests__/pipeline.test.d.ts +8 -0
- package/dist/defence/__tests__/pipeline.test.d.ts.map +1 -0
- package/dist/defence/__tests__/pipeline.test.js +61 -0
- package/dist/defence/__tests__/pipeline.test.js.map +1 -0
- package/dist/defence/__tests__/sensitivity.test.d.ts +7 -0
- package/dist/defence/__tests__/sensitivity.test.d.ts.map +1 -0
- package/dist/defence/__tests__/sensitivity.test.js +61 -0
- package/dist/defence/__tests__/sensitivity.test.js.map +1 -0
- package/dist/defence/__tests__/trust.test.d.ts +7 -0
- package/dist/defence/__tests__/trust.test.d.ts.map +1 -0
- package/dist/defence/__tests__/trust.test.js +49 -0
- package/dist/defence/__tests__/trust.test.js.map +1 -0
- package/dist/defence/audit/index.d.ts +4 -0
- package/dist/defence/audit/index.d.ts.map +1 -0
- package/dist/defence/audit/index.js +3 -0
- package/dist/defence/audit/index.js.map +1 -0
- package/dist/defence/audit/logger.d.ts +14 -0
- package/dist/defence/audit/logger.d.ts.map +1 -0
- package/dist/defence/audit/logger.js +54 -0
- package/dist/defence/audit/logger.js.map +1 -0
- package/dist/defence/audit/queries.d.ts +33 -0
- package/dist/defence/audit/queries.d.ts.map +1 -0
- package/dist/defence/audit/queries.js +103 -0
- package/dist/defence/audit/queries.js.map +1 -0
- package/dist/defence/firewall/anomaly-scorer.d.ts +8 -0
- package/dist/defence/firewall/anomaly-scorer.d.ts.map +1 -0
- package/dist/defence/firewall/anomaly-scorer.js +58 -0
- package/dist/defence/firewall/anomaly-scorer.js.map +1 -0
- package/dist/defence/firewall/encoding-detector.d.ts +13 -0
- package/dist/defence/firewall/encoding-detector.d.ts.map +1 -0
- package/dist/defence/firewall/encoding-detector.js +120 -0
- package/dist/defence/firewall/encoding-detector.js.map +1 -0
- package/dist/defence/firewall/index.d.ts +21 -0
- package/dist/defence/firewall/index.d.ts.map +1 -0
- package/dist/defence/firewall/index.js +133 -0
- package/dist/defence/firewall/index.js.map +1 -0
- package/dist/defence/firewall/instruction-detector.d.ts +12 -0
- package/dist/defence/firewall/instruction-detector.d.ts.map +1 -0
- package/dist/defence/firewall/instruction-detector.js +99 -0
- package/dist/defence/firewall/instruction-detector.js.map +1 -0
- package/dist/defence/firewall/privilege-detector.d.ts +13 -0
- package/dist/defence/firewall/privilege-detector.d.ts.map +1 -0
- package/dist/defence/firewall/privilege-detector.js +89 -0
- package/dist/defence/firewall/privilege-detector.js.map +1 -0
- package/dist/defence/fragmentation/assembly-detector.d.ts +18 -0
- package/dist/defence/fragmentation/assembly-detector.d.ts.map +1 -0
- package/dist/defence/fragmentation/assembly-detector.js +72 -0
- package/dist/defence/fragmentation/assembly-detector.js.map +1 -0
- package/dist/defence/fragmentation/entity-extractor.d.ts +19 -0
- package/dist/defence/fragmentation/entity-extractor.d.ts.map +1 -0
- package/dist/defence/fragmentation/entity-extractor.js +86 -0
- package/dist/defence/fragmentation/entity-extractor.js.map +1 -0
- package/dist/defence/fragmentation/index.d.ts +23 -0
- package/dist/defence/fragmentation/index.d.ts.map +1 -0
- package/dist/defence/fragmentation/index.js +49 -0
- package/dist/defence/fragmentation/index.js.map +1 -0
- package/dist/defence/fragmentation/temporal-analyzer.d.ts +28 -0
- package/dist/defence/fragmentation/temporal-analyzer.d.ts.map +1 -0
- package/dist/defence/fragmentation/temporal-analyzer.js +41 -0
- package/dist/defence/fragmentation/temporal-analyzer.js.map +1 -0
- package/dist/defence/index.d.ts +12 -0
- package/dist/defence/index.d.ts.map +1 -0
- package/dist/defence/index.js +18 -0
- package/dist/defence/index.js.map +1 -0
- package/dist/defence/pipeline.d.ts +9 -0
- package/dist/defence/pipeline.d.ts.map +1 -0
- package/dist/defence/pipeline.js +115 -0
- package/dist/defence/pipeline.js.map +1 -0
- package/dist/defence/scanner/index.d.ts +5 -0
- package/dist/defence/scanner/index.d.ts.map +1 -0
- package/dist/defence/scanner/index.js +5 -0
- package/dist/defence/scanner/index.js.map +1 -0
- package/dist/defence/scanner/scan-existing.d.ts +34 -0
- package/dist/defence/scanner/scan-existing.d.ts.map +1 -0
- package/dist/defence/scanner/scan-existing.js +136 -0
- package/dist/defence/scanner/scan-existing.js.map +1 -0
- package/dist/defence/sensitivity/classifier.d.ts +6 -0
- package/dist/defence/sensitivity/classifier.d.ts.map +1 -0
- package/dist/defence/sensitivity/classifier.js +50 -0
- package/dist/defence/sensitivity/classifier.js.map +1 -0
- package/dist/defence/sensitivity/index.d.ts +11 -0
- package/dist/defence/sensitivity/index.d.ts.map +1 -0
- package/dist/defence/sensitivity/index.js +13 -0
- package/dist/defence/sensitivity/index.js.map +1 -0
- package/dist/defence/sensitivity/patterns.d.ts +14 -0
- package/dist/defence/sensitivity/patterns.d.ts.map +1 -0
- package/dist/defence/sensitivity/patterns.js +67 -0
- package/dist/defence/sensitivity/patterns.js.map +1 -0
- package/dist/defence/sensitivity/redaction.d.ts +17 -0
- package/dist/defence/sensitivity/redaction.d.ts.map +1 -0
- package/dist/defence/sensitivity/redaction.js +47 -0
- package/dist/defence/sensitivity/redaction.js.map +1 -0
- package/dist/defence/trust/index.d.ts +3 -0
- package/dist/defence/trust/index.d.ts.map +1 -0
- package/dist/defence/trust/index.js +3 -0
- package/dist/defence/trust/index.js.map +1 -0
- package/dist/defence/trust/recall-filter.d.ts +10 -0
- package/dist/defence/trust/recall-filter.d.ts.map +1 -0
- package/dist/defence/trust/recall-filter.js +38 -0
- package/dist/defence/trust/recall-filter.js.map +1 -0
- package/dist/defence/trust/source-scorer.d.ts +6 -0
- package/dist/defence/trust/source-scorer.d.ts.map +1 -0
- package/dist/defence/trust/source-scorer.js +34 -0
- package/dist/defence/trust/source-scorer.js.map +1 -0
- package/dist/defence/types.d.ts +88 -0
- package/dist/defence/types.d.ts.map +1 -0
- package/dist/defence/types.js +15 -0
- package/dist/defence/types.js.map +1 -0
- package/dist/embeddings/generator.d.ts +20 -0
- package/dist/embeddings/generator.d.ts.map +1 -0
- package/dist/embeddings/generator.js +83 -0
- package/dist/embeddings/generator.js.map +1 -0
- package/dist/embeddings/index.d.ts +2 -0
- package/dist/embeddings/index.d.ts.map +1 -0
- package/dist/embeddings/index.js +2 -0
- package/dist/embeddings/index.js.map +1 -0
- package/dist/errors.d.ts +74 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +131 -0
- package/dist/errors.js.map +1 -0
- package/dist/graph/backfill.d.ts +6 -0
- package/dist/graph/backfill.d.ts.map +1 -0
- package/dist/graph/backfill.js +33 -0
- package/dist/graph/backfill.js.map +1 -0
- package/dist/graph/extract.d.ts +21 -0
- package/dist/graph/extract.d.ts.map +1 -0
- package/dist/graph/extract.js +231 -0
- package/dist/graph/extract.js.map +1 -0
- package/dist/graph/resolve.d.ts +6 -0
- package/dist/graph/resolve.d.ts.map +1 -0
- package/dist/graph/resolve.js +126 -0
- package/dist/graph/resolve.js.map +1 -0
- package/dist/index.d.ts +31 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +248 -0
- package/dist/index.js.map +1 -0
- package/dist/memory/activation.d.ts +69 -0
- package/dist/memory/activation.d.ts.map +1 -0
- package/dist/memory/activation.js +168 -0
- package/dist/memory/activation.js.map +1 -0
- package/dist/memory/consolidate.d.ts +98 -0
- package/dist/memory/consolidate.d.ts.map +1 -0
- package/dist/memory/consolidate.js +511 -0
- package/dist/memory/consolidate.js.map +1 -0
- package/dist/memory/contradiction.d.ts +69 -0
- package/dist/memory/contradiction.d.ts.map +1 -0
- package/dist/memory/contradiction.js +286 -0
- package/dist/memory/contradiction.js.map +1 -0
- package/dist/memory/decay.d.ts +62 -0
- package/dist/memory/decay.d.ts.map +1 -0
- package/dist/memory/decay.js +184 -0
- package/dist/memory/decay.js.map +1 -0
- package/dist/memory/salience.d.ts +36 -0
- package/dist/memory/salience.d.ts.map +1 -0
- package/dist/memory/salience.js +216 -0
- package/dist/memory/salience.js.map +1 -0
- package/dist/memory/similarity.d.ts +57 -0
- package/dist/memory/similarity.d.ts.map +1 -0
- package/dist/memory/similarity.js +114 -0
- package/dist/memory/similarity.js.map +1 -0
- package/dist/memory/store.d.ts +179 -0
- package/dist/memory/store.d.ts.map +1 -0
- package/dist/memory/store.js +1184 -0
- package/dist/memory/store.js.map +1 -0
- package/dist/memory/types.d.ts +97 -0
- package/dist/memory/types.d.ts.map +1 -0
- package/dist/memory/types.js +30 -0
- package/dist/memory/types.js.map +1 -0
- package/dist/server.d.ts +12 -0
- package/dist/server.d.ts.map +1 -0
- package/dist/server.js +568 -0
- package/dist/server.js.map +1 -0
- package/dist/service/install.d.ts +15 -0
- package/dist/service/install.d.ts.map +1 -0
- package/dist/service/install.js +178 -0
- package/dist/service/install.js.map +1 -0
- package/dist/service/templates.d.ts +13 -0
- package/dist/service/templates.d.ts.map +1 -0
- package/dist/service/templates.js +58 -0
- package/dist/service/templates.js.map +1 -0
- package/dist/setup/claude-md.d.ts +12 -0
- package/dist/setup/claude-md.d.ts.map +1 -0
- package/dist/setup/claude-md.js +68 -0
- package/dist/setup/claude-md.js.map +1 -0
- package/dist/setup/clawdbot.d.ts +15 -0
- package/dist/setup/clawdbot.d.ts.map +1 -0
- package/dist/setup/clawdbot.js +118 -0
- package/dist/setup/clawdbot.js.map +1 -0
- package/dist/setup/doctor.d.ts +5 -0
- package/dist/setup/doctor.d.ts.map +1 -0
- package/dist/setup/doctor.js +141 -0
- package/dist/setup/doctor.js.map +1 -0
- package/dist/setup/hooks.d.ts +6 -0
- package/dist/setup/hooks.d.ts.map +1 -0
- package/dist/setup/hooks.js +36 -0
- package/dist/setup/hooks.js.map +1 -0
- package/dist/setup/migrate.d.ts +16 -0
- package/dist/setup/migrate.d.ts.map +1 -0
- package/dist/setup/migrate.js +164 -0
- package/dist/setup/migrate.js.map +1 -0
- package/dist/setup/settings-hooks.d.ts +7 -0
- package/dist/setup/settings-hooks.d.ts.map +1 -0
- package/dist/setup/settings-hooks.js +83 -0
- package/dist/setup/settings-hooks.js.map +1 -0
- package/dist/setup/uninstall.d.ts +12 -0
- package/dist/setup/uninstall.d.ts.map +1 -0
- package/dist/setup/uninstall.js +125 -0
- package/dist/setup/uninstall.js.map +1 -0
- package/dist/tools/context.d.ts +135 -0
- package/dist/tools/context.d.ts.map +1 -0
- package/dist/tools/context.js +273 -0
- package/dist/tools/context.js.map +1 -0
- package/dist/tools/forget.d.ts +53 -0
- package/dist/tools/forget.d.ts.map +1 -0
- package/dist/tools/forget.js +179 -0
- package/dist/tools/forget.js.map +1 -0
- package/dist/tools/graph.d.ts +46 -0
- package/dist/tools/graph.d.ts.map +1 -0
- package/dist/tools/graph.js +206 -0
- package/dist/tools/graph.js.map +1 -0
- package/dist/tools/recall.d.ts +79 -0
- package/dist/tools/recall.d.ts.map +1 -0
- package/dist/tools/recall.js +156 -0
- package/dist/tools/recall.js.map +1 -0
- package/dist/tools/remember.d.ts +83 -0
- package/dist/tools/remember.d.ts.map +1 -0
- package/dist/tools/remember.js +151 -0
- package/dist/tools/remember.js.map +1 -0
- package/dist/worker/brain-worker.d.ts +100 -0
- package/dist/worker/brain-worker.d.ts.map +1 -0
- package/dist/worker/brain-worker.js +283 -0
- package/dist/worker/brain-worker.js.map +1 -0
- package/dist/worker/link-discovery.d.ts +47 -0
- package/dist/worker/link-discovery.d.ts.map +1 -0
- package/dist/worker/link-discovery.js +103 -0
- package/dist/worker/link-discovery.js.map +1 -0
- package/dist/worker/predictive-consolidation.d.ts +46 -0
- package/dist/worker/predictive-consolidation.d.ts.map +1 -0
- package/dist/worker/predictive-consolidation.js +110 -0
- package/dist/worker/predictive-consolidation.js.map +1 -0
- package/dist/worker/types.d.ts +91 -0
- package/dist/worker/types.d.ts.map +1 -0
- package/dist/worker/types.js +22 -0
- package/dist/worker/types.js.map +1 -0
- package/hooks/clawdbot/cortex-memory/HOOK.md +71 -0
- package/hooks/clawdbot/cortex-memory/handler.js +279 -0
- package/package.json +73 -0
- package/scripts/pre-compact-hook.mjs +716 -0
- package/scripts/session-end-hook.mjs +548 -0
- package/scripts/session-start-hook.mjs +221 -0
- package/scripts/start-dashboard.sh +41 -0
- package/scripts/stop-dashboard.sh +21 -0
- package/scripts/stop-hook.mjs +163 -0
|
@@ -0,0 +1,206 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Knowledge Graph query tools
|
|
3
|
+
*
|
|
4
|
+
* Provides BFS traversal, entity listing, and path-finding
|
|
5
|
+
* over the entities/triples tables.
|
|
6
|
+
*/
|
|
7
|
+
import { getDatabase } from '../database/init.js';
|
|
8
|
+
function mcpText(data) {
|
|
9
|
+
return { content: [{ type: 'text', text: JSON.stringify(data, null, 2) }] };
|
|
10
|
+
}
|
|
11
|
+
/**
|
|
12
|
+
* Traverse the knowledge graph from a starting entity using BFS.
|
|
13
|
+
*/
|
|
14
|
+
export function handleGraphQuery(args) {
|
|
15
|
+
const db = getDatabase();
|
|
16
|
+
const maxDepth = args.depth ?? 2;
|
|
17
|
+
const predicateFilter = args.predicates ?? null;
|
|
18
|
+
// Find entity
|
|
19
|
+
const entityRow = db.prepare('SELECT * FROM entities WHERE LOWER(name) = LOWER(?)').get(args.entity);
|
|
20
|
+
if (!entityRow) {
|
|
21
|
+
return mcpText({ error: `Entity "${args.entity}" not found in knowledge graph.` });
|
|
22
|
+
}
|
|
23
|
+
const rootEntity = {
|
|
24
|
+
id: entityRow.id,
|
|
25
|
+
name: entityRow.name,
|
|
26
|
+
type: entityRow.type,
|
|
27
|
+
memoryCount: entityRow.memory_count ?? 0,
|
|
28
|
+
};
|
|
29
|
+
// BFS
|
|
30
|
+
const connections = [];
|
|
31
|
+
const visited = new Set([entityRow.id]);
|
|
32
|
+
let frontier = [entityRow.id];
|
|
33
|
+
for (let d = 1; d <= maxDepth; d++) {
|
|
34
|
+
const nextFrontier = [];
|
|
35
|
+
for (const nodeId of frontier) {
|
|
36
|
+
// Outgoing
|
|
37
|
+
const outgoing = db.prepare('SELECT t.*, e.name, e.type FROM triples t JOIN entities e ON e.id = t.object_id WHERE t.subject_id = ?').all(nodeId);
|
|
38
|
+
for (const row of outgoing) {
|
|
39
|
+
if (predicateFilter && !predicateFilter.includes(row.predicate))
|
|
40
|
+
continue;
|
|
41
|
+
connections.push({
|
|
42
|
+
predicate: row.predicate,
|
|
43
|
+
direction: 'outgoing',
|
|
44
|
+
entity: { id: row.object_id, name: row.name, type: row.type },
|
|
45
|
+
depth: d,
|
|
46
|
+
});
|
|
47
|
+
if (!visited.has(row.object_id)) {
|
|
48
|
+
visited.add(row.object_id);
|
|
49
|
+
nextFrontier.push(row.object_id);
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
// Incoming
|
|
53
|
+
const incoming = db.prepare('SELECT t.*, e.name, e.type FROM triples t JOIN entities e ON e.id = t.subject_id WHERE t.object_id = ?').all(nodeId);
|
|
54
|
+
for (const row of incoming) {
|
|
55
|
+
if (predicateFilter && !predicateFilter.includes(row.predicate))
|
|
56
|
+
continue;
|
|
57
|
+
connections.push({
|
|
58
|
+
predicate: row.predicate,
|
|
59
|
+
direction: 'incoming',
|
|
60
|
+
entity: { id: row.subject_id, name: row.name, type: row.type },
|
|
61
|
+
depth: d,
|
|
62
|
+
});
|
|
63
|
+
if (!visited.has(row.subject_id)) {
|
|
64
|
+
visited.add(row.subject_id);
|
|
65
|
+
nextFrontier.push(row.subject_id);
|
|
66
|
+
}
|
|
67
|
+
}
|
|
68
|
+
}
|
|
69
|
+
frontier = nextFrontier;
|
|
70
|
+
if (frontier.length === 0)
|
|
71
|
+
break;
|
|
72
|
+
}
|
|
73
|
+
return mcpText({ entity: rootEntity, connections });
|
|
74
|
+
}
|
|
75
|
+
/**
|
|
76
|
+
* List entities in the knowledge graph.
|
|
77
|
+
*/
|
|
78
|
+
export function handleGraphEntities(args) {
|
|
79
|
+
const db = getDatabase();
|
|
80
|
+
const minMentions = args.minMentions ?? 1;
|
|
81
|
+
const limit = args.limit ?? 50;
|
|
82
|
+
let query = 'SELECT * FROM entities WHERE 1=1';
|
|
83
|
+
const params = [];
|
|
84
|
+
if (args.type) {
|
|
85
|
+
query += ' AND type = ?';
|
|
86
|
+
params.push(args.type);
|
|
87
|
+
}
|
|
88
|
+
query += ' AND memory_count >= ?';
|
|
89
|
+
params.push(minMentions);
|
|
90
|
+
query += ' ORDER BY memory_count DESC LIMIT ?';
|
|
91
|
+
params.push(limit);
|
|
92
|
+
const rows = db.prepare(query).all(...params);
|
|
93
|
+
const entities = rows.map((r) => {
|
|
94
|
+
let aliases = [];
|
|
95
|
+
try {
|
|
96
|
+
aliases = JSON.parse(r.aliases || '[]');
|
|
97
|
+
}
|
|
98
|
+
catch {
|
|
99
|
+
aliases = [];
|
|
100
|
+
}
|
|
101
|
+
return {
|
|
102
|
+
id: r.id,
|
|
103
|
+
name: r.name,
|
|
104
|
+
type: r.type,
|
|
105
|
+
memoryCount: r.memory_count ?? 0,
|
|
106
|
+
aliases,
|
|
107
|
+
};
|
|
108
|
+
});
|
|
109
|
+
return mcpText({ entities });
|
|
110
|
+
}
|
|
111
|
+
/**
|
|
112
|
+
* Find paths between two entities via BFS.
|
|
113
|
+
*/
|
|
114
|
+
export function handleGraphExplain(args) {
|
|
115
|
+
const db = getDatabase();
|
|
116
|
+
const maxDepth = args.maxDepth ?? 4;
|
|
117
|
+
// Resolve entities
|
|
118
|
+
const fromRow = db.prepare('SELECT * FROM entities WHERE LOWER(name) = LOWER(?)').get(args.from);
|
|
119
|
+
if (!fromRow) {
|
|
120
|
+
return mcpText({ error: `Source entity "${args.from}" not found.` });
|
|
121
|
+
}
|
|
122
|
+
const toRow = db.prepare('SELECT * FROM entities WHERE LOWER(name) = LOWER(?)').get(args.to);
|
|
123
|
+
if (!toRow) {
|
|
124
|
+
return mcpText({ error: `Target entity "${args.to}" not found.` });
|
|
125
|
+
}
|
|
126
|
+
if (fromRow.id === toRow.id) {
|
|
127
|
+
return mcpText({ paths: [{ hops: [{ entity: fromRow.name, predicate: '(self)' }] }], sourceMemories: [] });
|
|
128
|
+
}
|
|
129
|
+
const visited = new Map();
|
|
130
|
+
visited.set(fromRow.id, { id: fromRow.id, name: fromRow.name, parentId: null, predicate: '', sourceMemoryId: null });
|
|
131
|
+
let frontier = [fromRow.id];
|
|
132
|
+
let found = false;
|
|
133
|
+
for (let d = 0; d < maxDepth && !found; d++) {
|
|
134
|
+
const nextFrontier = [];
|
|
135
|
+
for (const nodeId of frontier) {
|
|
136
|
+
// Outgoing
|
|
137
|
+
const outgoing = db.prepare('SELECT t.object_id as next_id, t.predicate, t.source_memory_id, e.name FROM triples t JOIN entities e ON e.id = t.object_id WHERE t.subject_id = ?').all(nodeId);
|
|
138
|
+
for (const row of outgoing) {
|
|
139
|
+
if (!visited.has(row.next_id)) {
|
|
140
|
+
visited.set(row.next_id, {
|
|
141
|
+
id: row.next_id,
|
|
142
|
+
name: row.name,
|
|
143
|
+
parentId: nodeId,
|
|
144
|
+
predicate: row.predicate,
|
|
145
|
+
sourceMemoryId: row.source_memory_id,
|
|
146
|
+
});
|
|
147
|
+
nextFrontier.push(row.next_id);
|
|
148
|
+
if (row.next_id === toRow.id) {
|
|
149
|
+
found = true;
|
|
150
|
+
break;
|
|
151
|
+
}
|
|
152
|
+
}
|
|
153
|
+
}
|
|
154
|
+
if (found)
|
|
155
|
+
break;
|
|
156
|
+
// Incoming
|
|
157
|
+
const incoming = db.prepare('SELECT t.subject_id as next_id, t.predicate, t.source_memory_id, e.name FROM triples t JOIN entities e ON e.id = t.subject_id WHERE t.object_id = ?').all(nodeId);
|
|
158
|
+
for (const row of incoming) {
|
|
159
|
+
if (!visited.has(row.next_id)) {
|
|
160
|
+
visited.set(row.next_id, {
|
|
161
|
+
id: row.next_id,
|
|
162
|
+
name: row.name,
|
|
163
|
+
parentId: nodeId,
|
|
164
|
+
predicate: `~${row.predicate}`,
|
|
165
|
+
sourceMemoryId: row.source_memory_id,
|
|
166
|
+
});
|
|
167
|
+
nextFrontier.push(row.next_id);
|
|
168
|
+
if (row.next_id === toRow.id) {
|
|
169
|
+
found = true;
|
|
170
|
+
break;
|
|
171
|
+
}
|
|
172
|
+
}
|
|
173
|
+
}
|
|
174
|
+
if (found)
|
|
175
|
+
break;
|
|
176
|
+
}
|
|
177
|
+
frontier = nextFrontier;
|
|
178
|
+
if (frontier.length === 0)
|
|
179
|
+
break;
|
|
180
|
+
}
|
|
181
|
+
if (!found) {
|
|
182
|
+
return mcpText({ paths: [], sourceMemories: [], message: `No path found between "${args.from}" and "${args.to}" within ${maxDepth} hops.` });
|
|
183
|
+
}
|
|
184
|
+
// Reconstruct path
|
|
185
|
+
const hops = [];
|
|
186
|
+
const memoryIds = new Set();
|
|
187
|
+
let current = toRow.id;
|
|
188
|
+
while (current !== null) {
|
|
189
|
+
const node = visited.get(current);
|
|
190
|
+
hops.unshift({ entity: node.name, predicate: node.predicate });
|
|
191
|
+
if (node.sourceMemoryId)
|
|
192
|
+
memoryIds.add(node.sourceMemoryId);
|
|
193
|
+
current = node.parentId;
|
|
194
|
+
if (current === null || node.parentId === null)
|
|
195
|
+
break;
|
|
196
|
+
}
|
|
197
|
+
// Fetch source memories
|
|
198
|
+
let sourceMemories = [];
|
|
199
|
+
if (memoryIds.size > 0) {
|
|
200
|
+
const ids = Array.from(memoryIds);
|
|
201
|
+
const placeholders = ids.map(() => '?').join(',');
|
|
202
|
+
sourceMemories = db.prepare(`SELECT id, title FROM memories WHERE id IN (${placeholders})`).all(...ids);
|
|
203
|
+
}
|
|
204
|
+
return mcpText({ paths: [{ hops }], sourceMemories });
|
|
205
|
+
}
|
|
206
|
+
//# sourceMappingURL=graph.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"graph.js","sourceRoot":"","sources":["../../src/tools/graph.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAiBlD,SAAS,OAAO,CAAC,IAAa;IAC5B,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;AACvF,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,IAIhC;IACC,MAAM,EAAE,GAAG,WAAW,EAAE,CAAC;IACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC;IACjC,MAAM,eAAe,GAAG,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC;IAEhD,cAAc;IACd,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,CAC1B,qDAAqD,CACtD,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAQ,CAAC;IAE1B,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,OAAO,CAAC,EAAE,KAAK,EAAE,WAAW,IAAI,CAAC,MAAM,iCAAiC,EAAE,CAAC,CAAC;IACrF,CAAC;IAED,MAAM,UAAU,GAAe;QAC7B,EAAE,EAAE,SAAS,CAAC,EAAE;QAChB,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,IAAI,EAAE,SAAS,CAAC,IAAI;QACpB,WAAW,EAAE,SAAS,CAAC,YAAY,IAAI,CAAC;KACzC,CAAC;IAEF,MAAM;IACN,MAAM,WAAW,GAAiB,EAAE,CAAC;IACrC,MAAM,OAAO,GAAG,IAAI,GAAG,CAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC;IAChD,IAAI,QAAQ,GAAa,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IAExC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,MAAM,YAAY,GAAa,EAAE,CAAC;QAElC,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC9B,WAAW;YACX,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CACzB,wGAAwG,CACzG,CAAC,GAAG,CAAC,MAAM,CAAU,CAAC;YAEvB,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,IAAI,eAAe,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC;oBAAE,SAAS;gBAC1E,WAAW,CAAC,IAAI,CAAC;oBACf,SAAS,EAAE,GAAG,CAAC,SAAS;oBACxB,SAAS,EAAE,UAAU;oBACrB,MAAM,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,SAAS,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE;oBAC7D,KAAK,EAAE,CAAC;iBACT,CAAC,CAAC;gBACH,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;oBAChC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAC3B,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;YAED,WAAW;YACX,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CACzB,wGAAwG,CACzG,CAAC,GAAG,CAAC,MAAM,CAAU,CAAC;YAEvB,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,IAAI,eAAe,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC;oBAAE,SAAS;gBAC1E,WAAW,CAAC,IAAI,CAAC;oBACf,SAAS,EAAE,GAAG,CAAC,SAAS;oBACxB,SAAS,EAAE,UAAU;oBACrB,MAAM,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE;oBAC9D,KAAK,EAAE,CAAC;iBACT,CAAC,CAAC;gBACH,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;oBACjC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC5B,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBACpC,CAAC;YACH,CAAC;QACH,CAAC;QAED,QAAQ,GAAG,YAAY,CAAC;QACxB,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,MAAM;IACnC,CAAC;IAED,OAAO,OAAO,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,CAAC,CAAC;AACtD,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,IAInC;IACC,MAAM,EAAE,GAAG,WAAW,EAAE,CAAC;IACzB,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,IAAI,CAAC,CAAC;IAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;IAE/B,IAAI,KAAK,GAAG,kCAAkC,CAAC;IAC/C,MAAM,MAAM,GAAU,EAAE,CAAC;IAEzB,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;QACd,KAAK,IAAI,eAAe,CAAC;QACzB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzB,CAAC;IAED,KAAK,IAAI,wBAAwB,CAAC;IAClC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAEzB,KAAK,IAAI,qCAAqC,CAAC;IAC/C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAEnB,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,CAAU,CAAC;IAEvD,MAAM,QAAQ,GAAiB,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QAC5C,IAAI,OAAO,GAAa,EAAE,CAAC;QAC3B,IAAI,CAAC;YACH,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,IAAI,IAAI,CAAC,CAAC;QAC1C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,GAAG,EAAE,CAAC;QACf,CAAC;QACD,OAAO;YACL,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,YAAY,IAAI,CAAC;YAChC,OAAO;SACR,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;AAC/B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,IAIlC;IACC,MAAM,EAAE,GAAG,WAAW,EAAE,CAAC;IACzB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,CAAC,CAAC;IAEpC,mBAAmB;IACnB,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,CACxB,qDAAqD,CACtD,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAQ,CAAC;IAExB,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,OAAO,CAAC,EAAE,KAAK,EAAE,kBAAkB,IAAI,CAAC,IAAI,cAAc,EAAE,CAAC,CAAC;IACvE,CAAC;IAED,MAAM,KAAK,GAAG,EAAE,CAAC,OAAO,CACtB,qDAAqD,CACtD,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAQ,CAAC;IAEtB,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,OAAO,CAAC,EAAE,KAAK,EAAE,kBAAkB,IAAI,CAAC,EAAE,cAAc,EAAE,CAAC,CAAC;IACrE,CAAC;IAED,IAAI,OAAO,CAAC,EAAE,KAAK,KAAK,CAAC,EAAE,EAAE,CAAC;QAC5B,OAAO,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,CAAC;IAC7G,CAAC;IAWD,MAAM,OAAO,GAAG,IAAI,GAAG,EAAmB,CAAC;IAC3C,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;IAErH,IAAI,QAAQ,GAAa,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACtC,IAAI,KAAK,GAAG,KAAK,CAAC;IAElB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,IAAI,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5C,MAAM,YAAY,GAAa,EAAE,CAAC;QAElC,KAAK,MAAM,MAAM,IAAI,QAAQ,EAAE,CAAC;YAC9B,WAAW;YACX,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CACzB,oJAAoJ,CACrJ,CAAC,GAAG,CAAC,MAAM,CAAU,CAAC;YAEvB,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC9B,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE;wBACvB,EAAE,EAAE,GAAG,CAAC,OAAO;wBACf,IAAI,EAAE,GAAG,CAAC,IAAI;wBACd,QAAQ,EAAE,MAAM;wBAChB,SAAS,EAAE,GAAG,CAAC,SAAS;wBACxB,cAAc,EAAE,GAAG,CAAC,gBAAgB;qBACrC,CAAC,CAAC;oBACH,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBAC/B,IAAI,GAAG,CAAC,OAAO,KAAK,KAAK,CAAC,EAAE,EAAE,CAAC;wBAAC,KAAK,GAAG,IAAI,CAAC;wBAAC,MAAM;oBAAC,CAAC;gBACxD,CAAC;YACH,CAAC;YACD,IAAI,KAAK;gBAAE,MAAM;YAEjB,WAAW;YACX,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CACzB,qJAAqJ,CACtJ,CAAC,GAAG,CAAC,MAAM,CAAU,CAAC;YAEvB,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBAC3B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC9B,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE;wBACvB,EAAE,EAAE,GAAG,CAAC,OAAO;wBACf,IAAI,EAAE,GAAG,CAAC,IAAI;wBACd,QAAQ,EAAE,MAAM;wBAChB,SAAS,EAAE,IAAI,GAAG,CAAC,SAAS,EAAE;wBAC9B,cAAc,EAAE,GAAG,CAAC,gBAAgB;qBACrC,CAAC,CAAC;oBACH,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBAC/B,IAAI,GAAG,CAAC,OAAO,KAAK,KAAK,CAAC,EAAE,EAAE,CAAC;wBAAC,KAAK,GAAG,IAAI,CAAC;wBAAC,MAAM;oBAAC,CAAC;gBACxD,CAAC;YACH,CAAC;YACD,IAAI,KAAK;gBAAE,MAAM;QACnB,CAAC;QAED,QAAQ,GAAG,YAAY,CAAC;QACxB,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,MAAM;IACnC,CAAC;IAED,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,cAAc,EAAE,EAAE,EAAE,OAAO,EAAE,0BAA0B,IAAI,CAAC,IAAI,UAAU,IAAI,CAAC,EAAE,YAAY,QAAQ,QAAQ,EAAE,CAAC,CAAC;IAC/I,CAAC;IAED,mBAAmB;IACnB,MAAM,IAAI,GAA4C,EAAE,CAAC;IACzD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,IAAI,OAAO,GAAG,KAAK,CAAC,EAAE,CAAC;IAEvB,OAAO,OAAO,KAAK,IAAI,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC;QACnC,IAAI,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QAC/D,IAAI,IAAI,CAAC,cAAc;YAAE,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC5D,OAAO,GAAG,IAAI,CAAC,QAAS,CAAC;QACzB,IAAI,OAAO,KAAK,IAAI,IAAI,IAAI,CAAC,QAAQ,KAAK,IAAI;YAAE,MAAM;IACxD,CAAC;IAED,wBAAwB;IACxB,IAAI,cAAc,GAAoC,EAAE,CAAC;IACzD,IAAI,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;QACvB,MAAM,GAAG,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAClC,MAAM,YAAY,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClD,cAAc,GAAG,EAAE,CAAC,OAAO,CACzB,+CAA+C,YAAY,GAAG,CAC/D,CAAC,GAAG,CAAC,GAAG,GAAG,CAAU,CAAC;IACzB,CAAC;IAED,OAAO,OAAO,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,CAAC;AACxD,CAAC"}
|
|
@@ -0,0 +1,79 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Recall Tool
|
|
3
|
+
*
|
|
4
|
+
* Search and retrieve memories using semantic search and filters.
|
|
5
|
+
*/
|
|
6
|
+
import { z } from 'zod';
|
|
7
|
+
import { Memory } from '../memory/types.js';
|
|
8
|
+
export declare const recallSchema: z.ZodObject<{
|
|
9
|
+
query: z.ZodOptional<z.ZodString>;
|
|
10
|
+
category: z.ZodOptional<z.ZodEnum<["architecture", "pattern", "preference", "error", "context", "learning", "todo", "note", "relationship", "custom"]>>;
|
|
11
|
+
type: z.ZodOptional<z.ZodEnum<["short_term", "long_term", "episodic"]>>;
|
|
12
|
+
project: z.ZodOptional<z.ZodString>;
|
|
13
|
+
tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
|
|
14
|
+
limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
|
|
15
|
+
includeDecayed: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
|
|
16
|
+
includeGlobal: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
|
|
17
|
+
mode: z.ZodDefault<z.ZodOptional<z.ZodEnum<["search", "recent", "important"]>>>;
|
|
18
|
+
}, "strip", z.ZodTypeAny, {
|
|
19
|
+
mode: "search" | "important" | "recent";
|
|
20
|
+
limit: number;
|
|
21
|
+
includeDecayed: boolean;
|
|
22
|
+
includeGlobal: boolean;
|
|
23
|
+
project?: string | undefined;
|
|
24
|
+
type?: "short_term" | "long_term" | "episodic" | undefined;
|
|
25
|
+
category?: "architecture" | "pattern" | "preference" | "error" | "context" | "learning" | "todo" | "note" | "relationship" | "custom" | undefined;
|
|
26
|
+
tags?: string[] | undefined;
|
|
27
|
+
query?: string | undefined;
|
|
28
|
+
}, {
|
|
29
|
+
project?: string | undefined;
|
|
30
|
+
type?: "short_term" | "long_term" | "episodic" | undefined;
|
|
31
|
+
mode?: "search" | "important" | "recent" | undefined;
|
|
32
|
+
limit?: number | undefined;
|
|
33
|
+
category?: "architecture" | "pattern" | "preference" | "error" | "context" | "learning" | "todo" | "note" | "relationship" | "custom" | undefined;
|
|
34
|
+
tags?: string[] | undefined;
|
|
35
|
+
query?: string | undefined;
|
|
36
|
+
includeDecayed?: boolean | undefined;
|
|
37
|
+
includeGlobal?: boolean | undefined;
|
|
38
|
+
}>;
|
|
39
|
+
export type RecallInput = z.infer<typeof recallSchema>;
|
|
40
|
+
/**
|
|
41
|
+
* Execute the recall tool
|
|
42
|
+
*/
|
|
43
|
+
export declare function executeRecall(input: RecallInput): Promise<{
|
|
44
|
+
success: boolean;
|
|
45
|
+
memories?: Memory[];
|
|
46
|
+
contradictions?: Map<number, {
|
|
47
|
+
memoryId: number;
|
|
48
|
+
title: string;
|
|
49
|
+
score: number;
|
|
50
|
+
}[]>;
|
|
51
|
+
count?: number;
|
|
52
|
+
error?: string;
|
|
53
|
+
}>;
|
|
54
|
+
/**
|
|
55
|
+
* Format a single memory for display
|
|
56
|
+
*/
|
|
57
|
+
export declare function formatMemory(memory: Memory, verbose?: boolean): string;
|
|
58
|
+
/**
|
|
59
|
+
* Format the recall result for MCP response
|
|
60
|
+
*/
|
|
61
|
+
export declare function formatRecallResult(result: Awaited<ReturnType<typeof executeRecall>>, verbose?: boolean): string;
|
|
62
|
+
/**
|
|
63
|
+
* Get a single memory by ID
|
|
64
|
+
*/
|
|
65
|
+
export declare const getMemorySchema: z.ZodObject<{
|
|
66
|
+
id: z.ZodNumber;
|
|
67
|
+
}, "strip", z.ZodTypeAny, {
|
|
68
|
+
id: number;
|
|
69
|
+
}, {
|
|
70
|
+
id: number;
|
|
71
|
+
}>;
|
|
72
|
+
export declare function executeGetMemory(input: {
|
|
73
|
+
id: number;
|
|
74
|
+
}): {
|
|
75
|
+
success: boolean;
|
|
76
|
+
memory?: Memory;
|
|
77
|
+
error?: string;
|
|
78
|
+
};
|
|
79
|
+
//# sourceMappingURL=recall.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"recall.d.ts","sourceRoot":"","sources":["../../src/tools/recall.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,MAAM,EAAgB,MAAM,oBAAoB,CAAC;AAK1D,eAAO,MAAM,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAkBvB,CAAC;AAEH,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,YAAY,CAAC,CAAC;AAEvD;;GAEG;AACH,wBAAsB,aAAa,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC;IAC/D,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;IACpB,cAAc,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC,CAAC;IACnF,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC,CAwDD;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,OAAe,GAAG,MAAM,CAqB7E;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,aAAa,CAAC,CAAC,EACjD,OAAO,GAAE,OAAe,GACvB,MAAM,CAoBR;AAED;;GAEG;AACH,eAAO,MAAM,eAAe;;;;;;EAE1B,CAAC;AAEH,wBAAgB,gBAAgB,CAAC,KAAK,EAAE;IAAE,EAAE,EAAE,MAAM,CAAA;CAAE,GAAG;IACvD,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAiBA"}
|
|
@@ -0,0 +1,156 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Recall Tool
|
|
3
|
+
*
|
|
4
|
+
* Search and retrieve memories using semantic search and filters.
|
|
5
|
+
*/
|
|
6
|
+
import { z } from 'zod';
|
|
7
|
+
import { searchMemories, accessMemory, getRecentMemories, getHighPriorityMemories } from '../memory/store.js';
|
|
8
|
+
import { formatTimeSinceAccess } from '../memory/decay.js';
|
|
9
|
+
import { MemoryNotFoundError, formatErrorForMcp } from '../errors.js';
|
|
10
|
+
import { resolveProject } from '../context/project-context.js';
|
|
11
|
+
// Input schema for the recall tool
|
|
12
|
+
export const recallSchema = z.object({
|
|
13
|
+
query: z.string().optional().describe('Search query (semantic search)'),
|
|
14
|
+
category: z.enum([
|
|
15
|
+
'architecture', 'pattern', 'preference', 'error',
|
|
16
|
+
'context', 'learning', 'todo', 'note', 'relationship', 'custom'
|
|
17
|
+
]).optional().describe('Filter by category'),
|
|
18
|
+
type: z.enum(['short_term', 'long_term', 'episodic']).optional()
|
|
19
|
+
.describe('Filter by memory type'),
|
|
20
|
+
project: z.string().optional().describe('Filter by project'),
|
|
21
|
+
tags: z.array(z.string()).optional().describe('Filter by tags'),
|
|
22
|
+
limit: z.number().min(1).max(50).optional().default(10)
|
|
23
|
+
.describe('Maximum number of results'),
|
|
24
|
+
includeDecayed: z.boolean().optional().default(false)
|
|
25
|
+
.describe('Include memories that have decayed below threshold'),
|
|
26
|
+
includeGlobal: z.boolean().optional().default(true)
|
|
27
|
+
.describe('Include global memories in search results (default: true)'),
|
|
28
|
+
mode: z.enum(['search', 'recent', 'important']).optional().default('search')
|
|
29
|
+
.describe('Recall mode: search (query-based), recent (by time), important (by salience)'),
|
|
30
|
+
});
|
|
31
|
+
/**
|
|
32
|
+
* Execute the recall tool
|
|
33
|
+
*/
|
|
34
|
+
export async function executeRecall(input) {
|
|
35
|
+
try {
|
|
36
|
+
// Resolve project (auto-detect if not provided)
|
|
37
|
+
const resolvedProject = resolveProject(input.project);
|
|
38
|
+
const projectFilter = resolvedProject ?? undefined;
|
|
39
|
+
let memories = [];
|
|
40
|
+
let contradictions;
|
|
41
|
+
switch (input.mode) {
|
|
42
|
+
case 'recent':
|
|
43
|
+
memories = getRecentMemories(input.limit, projectFilter);
|
|
44
|
+
break;
|
|
45
|
+
case 'important':
|
|
46
|
+
memories = getHighPriorityMemories(input.limit, projectFilter);
|
|
47
|
+
break;
|
|
48
|
+
case 'search':
|
|
49
|
+
default:
|
|
50
|
+
const results = await searchMemories({
|
|
51
|
+
query: input.query || '',
|
|
52
|
+
category: input.category,
|
|
53
|
+
type: input.type,
|
|
54
|
+
project: projectFilter,
|
|
55
|
+
tags: input.tags,
|
|
56
|
+
limit: input.limit,
|
|
57
|
+
includeDecayed: input.includeDecayed,
|
|
58
|
+
includeGlobal: input.includeGlobal,
|
|
59
|
+
});
|
|
60
|
+
memories = results.map(r => r.memory);
|
|
61
|
+
// Extract contradictions from search results
|
|
62
|
+
const contradictionEntries = results
|
|
63
|
+
.filter(r => r.contradictions && r.contradictions.length > 0)
|
|
64
|
+
.map(r => [r.memory.id, r.contradictions]);
|
|
65
|
+
if (contradictionEntries.length > 0) {
|
|
66
|
+
contradictions = new Map(contradictionEntries);
|
|
67
|
+
}
|
|
68
|
+
break;
|
|
69
|
+
}
|
|
70
|
+
// Access each memory to reinforce it
|
|
71
|
+
memories = memories.map(m => accessMemory(m.id) || m);
|
|
72
|
+
return {
|
|
73
|
+
success: true,
|
|
74
|
+
memories,
|
|
75
|
+
contradictions,
|
|
76
|
+
count: memories.length,
|
|
77
|
+
};
|
|
78
|
+
}
|
|
79
|
+
catch (error) {
|
|
80
|
+
return {
|
|
81
|
+
success: false,
|
|
82
|
+
error: formatErrorForMcp(error),
|
|
83
|
+
};
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
/**
|
|
87
|
+
* Format a single memory for display
|
|
88
|
+
*/
|
|
89
|
+
export function formatMemory(memory, verbose = false) {
|
|
90
|
+
const lines = [
|
|
91
|
+
`[${memory.id}] **${memory.title}**`,
|
|
92
|
+
` ${memory.content.slice(0, 200)}${memory.content.length > 200 ? '...' : ''}`,
|
|
93
|
+
];
|
|
94
|
+
if (verbose) {
|
|
95
|
+
lines.push(` Type: ${memory.type} | Category: ${memory.category}`);
|
|
96
|
+
lines.push(` Salience: ${(memory.salience * 100).toFixed(0)}% | Accessed: ${memory.accessCount}x`);
|
|
97
|
+
lines.push(` Last access: ${formatTimeSinceAccess(memory)}`);
|
|
98
|
+
if (memory.tags.length > 0) {
|
|
99
|
+
lines.push(` Tags: ${memory.tags.join(', ')}`);
|
|
100
|
+
}
|
|
101
|
+
if (memory.project) {
|
|
102
|
+
lines.push(` Project: ${memory.project}`);
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
else {
|
|
106
|
+
lines.push(` (${memory.type}, ${memory.category}, ${formatTimeSinceAccess(memory)})`);
|
|
107
|
+
}
|
|
108
|
+
return lines.join('\n');
|
|
109
|
+
}
|
|
110
|
+
/**
|
|
111
|
+
* Format the recall result for MCP response
|
|
112
|
+
*/
|
|
113
|
+
export function formatRecallResult(result, verbose = false) {
|
|
114
|
+
if (!result.success) {
|
|
115
|
+
return `Failed to recall: ${result.error}`;
|
|
116
|
+
}
|
|
117
|
+
if (!result.memories || result.memories.length === 0) {
|
|
118
|
+
return 'No memories found matching your query.';
|
|
119
|
+
}
|
|
120
|
+
const header = `Found ${result.count} ${result.count === 1 ? 'memory' : 'memories'}:\n`;
|
|
121
|
+
const formattedMemories = result.memories.map(m => {
|
|
122
|
+
let output = formatMemory(m, verbose);
|
|
123
|
+
const contradictions = result.contradictions?.get(m.id);
|
|
124
|
+
if (contradictions && contradictions.length > 0) {
|
|
125
|
+
output += `\n ⚠️ CONTRADICTS: ${contradictions.map(c => `"${c.title}" (ID ${c.memoryId})`).join(', ')}`;
|
|
126
|
+
}
|
|
127
|
+
return output;
|
|
128
|
+
}).join('\n\n');
|
|
129
|
+
return header + formattedMemories;
|
|
130
|
+
}
|
|
131
|
+
/**
|
|
132
|
+
* Get a single memory by ID
|
|
133
|
+
*/
|
|
134
|
+
export const getMemorySchema = z.object({
|
|
135
|
+
id: z.number().describe('Memory ID to retrieve'),
|
|
136
|
+
});
|
|
137
|
+
export function executeGetMemory(input) {
|
|
138
|
+
try {
|
|
139
|
+
const memory = accessMemory(input.id);
|
|
140
|
+
if (!memory) {
|
|
141
|
+
const error = new MemoryNotFoundError(input.id);
|
|
142
|
+
return {
|
|
143
|
+
success: false,
|
|
144
|
+
error: error.toUserMessage(),
|
|
145
|
+
};
|
|
146
|
+
}
|
|
147
|
+
return { success: true, memory };
|
|
148
|
+
}
|
|
149
|
+
catch (error) {
|
|
150
|
+
return {
|
|
151
|
+
success: false,
|
|
152
|
+
error: formatErrorForMcp(error),
|
|
153
|
+
};
|
|
154
|
+
}
|
|
155
|
+
}
|
|
156
|
+
//# sourceMappingURL=recall.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"recall.js","sourceRoot":"","sources":["../../src/tools/recall.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,iBAAiB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAC9G,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAE3D,OAAO,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACtE,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAE/D,mCAAmC;AACnC,MAAM,CAAC,MAAM,YAAY,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;IACvE,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC;QACf,cAAc,EAAE,SAAS,EAAE,YAAY,EAAE,OAAO;QAChD,SAAS,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,EAAE,QAAQ;KAChE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC;IAC5C,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,YAAY,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC,CAAC,QAAQ,EAAE;SAC7D,QAAQ,CAAC,uBAAuB,CAAC;IACpC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;IAC5D,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC;IAC/D,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;SACpD,QAAQ,CAAC,2BAA2B,CAAC;IACxC,cAAc,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC;SAClD,QAAQ,CAAC,oDAAoD,CAAC;IACjE,aAAa,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;SAChD,QAAQ,CAAC,2DAA2D,CAAC;IACxE,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC;SACzE,QAAQ,CAAC,8EAA8E,CAAC;CAC5F,CAAC,CAAC;AAIH;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,KAAkB;IAOpD,IAAI,CAAC;QACH,gDAAgD;QAChD,MAAM,eAAe,GAAG,cAAc,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACtD,MAAM,aAAa,GAAG,eAAe,IAAI,SAAS,CAAC;QAEnD,IAAI,QAAQ,GAAa,EAAE,CAAC;QAC5B,IAAI,cAA6F,CAAC;QAElG,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACnB,KAAK,QAAQ;gBACX,QAAQ,GAAG,iBAAiB,CAAC,KAAK,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;gBACzD,MAAM;YAER,KAAK,WAAW;gBACd,QAAQ,GAAG,uBAAuB,CAAC,KAAK,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;gBAC/D,MAAM;YAER,KAAK,QAAQ,CAAC;YACd;gBACE,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC;oBACnC,KAAK,EAAE,KAAK,CAAC,KAAK,IAAI,EAAE;oBACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ;oBACxB,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,OAAO,EAAE,aAAa;oBACtB,IAAI,EAAE,KAAK,CAAC,IAAI;oBAChB,KAAK,EAAE,KAAK,CAAC,KAAK;oBAClB,cAAc,EAAE,KAAK,CAAC,cAAc;oBACpC,aAAa,EAAE,KAAK,CAAC,aAAa;iBACnC,CAAC,CAAC;gBACH,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;gBACtC,6CAA6C;gBAC7C,MAAM,oBAAoB,GAAG,OAAO;qBACjC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,cAAc,IAAI,CAAC,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC;qBAC5D,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,cAAe,CAAU,CAAC,CAAC;gBACvD,IAAI,oBAAoB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACpC,cAAc,GAAG,IAAI,GAAG,CAAC,oBAAoB,CAAC,CAAC;gBACjD,CAAC;gBACD,MAAM;QACV,CAAC;QAED,qCAAqC;QACrC,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;QAEtD,OAAO;YACL,OAAO,EAAE,IAAI;YACb,QAAQ;YACR,cAAc;YACd,KAAK,EAAE,QAAQ,CAAC,MAAM;SACvB,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,iBAAiB,CAAC,KAAK,CAAC;SAChC,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,MAAc,EAAE,UAAmB,KAAK;IACnE,MAAM,KAAK,GAAG;QACZ,IAAI,MAAM,CAAC,EAAE,OAAO,MAAM,CAAC,KAAK,IAAI;QACpC,OAAO,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;KACjF,CAAC;IAEF,IAAI,OAAO,EAAE,CAAC;QACZ,KAAK,CAAC,IAAI,CAAC,aAAa,MAAM,CAAC,IAAI,gBAAgB,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;QACtE,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,QAAQ,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB,MAAM,CAAC,WAAW,GAAG,CAAC,CAAC;QACtG,KAAK,CAAC,IAAI,CAAC,oBAAoB,qBAAqB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAChE,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,aAAa,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,KAAK,CAAC,IAAI,CAAC,gBAAgB,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;SAAM,CAAC;QACN,KAAK,CAAC,IAAI,CAAC,QAAQ,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,QAAQ,KAAK,qBAAqB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAC3F,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAChC,MAAiD,EACjD,UAAmB,KAAK;IAExB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO,qBAAqB,MAAM,CAAC,KAAK,EAAE,CAAC;IAC7C,CAAC;IAED,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrD,OAAO,wCAAwC,CAAC;IAClD,CAAC;IAED,MAAM,MAAM,GAAG,SAAS,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,KAAK,CAAC;IACxF,MAAM,iBAAiB,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;QAChD,IAAI,MAAM,GAAG,YAAY,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACtC,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACxD,IAAI,cAAc,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChD,MAAM,IAAI,uBAAuB,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3G,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAEhB,OAAO,MAAM,GAAG,iBAAiB,CAAC;AACpC,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IACtC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;CACjD,CAAC,CAAC;AAEH,MAAM,UAAU,gBAAgB,CAAC,KAAqB;IAKpD,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,KAAK,GAAG,IAAI,mBAAmB,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAChD,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,CAAC,aAAa,EAAE;aAC7B,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IACnC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,iBAAiB,CAAC,KAAK,CAAC;SAChC,CAAC;IACJ,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,83 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Remember Tool
|
|
3
|
+
*
|
|
4
|
+
* Store memories with automatic salience detection and categorization.
|
|
5
|
+
*/
|
|
6
|
+
import { z } from 'zod';
|
|
7
|
+
import { MemoryCategory, MemoryType } from '../memory/types.js';
|
|
8
|
+
export declare const rememberSchema: z.ZodObject<{
|
|
9
|
+
title: z.ZodString;
|
|
10
|
+
content: z.ZodString;
|
|
11
|
+
category: z.ZodOptional<z.ZodEnum<["architecture", "pattern", "preference", "error", "context", "learning", "todo", "note", "relationship", "custom"]>>;
|
|
12
|
+
type: z.ZodOptional<z.ZodEnum<["short_term", "long_term", "episodic"]>>;
|
|
13
|
+
project: z.ZodOptional<z.ZodString>;
|
|
14
|
+
tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
|
|
15
|
+
importance: z.ZodOptional<z.ZodEnum<["low", "normal", "high", "critical"]>>;
|
|
16
|
+
scope: z.ZodOptional<z.ZodEnum<["project", "global"]>>;
|
|
17
|
+
transferable: z.ZodOptional<z.ZodBoolean>;
|
|
18
|
+
source: z.ZodOptional<z.ZodObject<{
|
|
19
|
+
type: z.ZodEnum<["user", "email", "web", "agent", "file", "api"]>;
|
|
20
|
+
identifier: z.ZodString;
|
|
21
|
+
}, "strip", z.ZodTypeAny, {
|
|
22
|
+
type: "api" | "file" | "user" | "email" | "web" | "agent";
|
|
23
|
+
identifier: string;
|
|
24
|
+
}, {
|
|
25
|
+
type: "api" | "file" | "user" | "email" | "web" | "agent";
|
|
26
|
+
identifier: string;
|
|
27
|
+
}>>;
|
|
28
|
+
}, "strip", z.ZodTypeAny, {
|
|
29
|
+
content: string;
|
|
30
|
+
title: string;
|
|
31
|
+
scope?: "project" | "global" | undefined;
|
|
32
|
+
transferable?: boolean | undefined;
|
|
33
|
+
source?: {
|
|
34
|
+
type: "api" | "file" | "user" | "email" | "web" | "agent";
|
|
35
|
+
identifier: string;
|
|
36
|
+
} | undefined;
|
|
37
|
+
project?: string | undefined;
|
|
38
|
+
type?: "short_term" | "long_term" | "episodic" | undefined;
|
|
39
|
+
category?: "architecture" | "pattern" | "preference" | "error" | "context" | "learning" | "todo" | "note" | "relationship" | "custom" | undefined;
|
|
40
|
+
tags?: string[] | undefined;
|
|
41
|
+
importance?: "critical" | "low" | "high" | "normal" | undefined;
|
|
42
|
+
}, {
|
|
43
|
+
content: string;
|
|
44
|
+
title: string;
|
|
45
|
+
scope?: "project" | "global" | undefined;
|
|
46
|
+
transferable?: boolean | undefined;
|
|
47
|
+
source?: {
|
|
48
|
+
type: "api" | "file" | "user" | "email" | "web" | "agent";
|
|
49
|
+
identifier: string;
|
|
50
|
+
} | undefined;
|
|
51
|
+
project?: string | undefined;
|
|
52
|
+
type?: "short_term" | "long_term" | "episodic" | undefined;
|
|
53
|
+
category?: "architecture" | "pattern" | "preference" | "error" | "context" | "learning" | "todo" | "note" | "relationship" | "custom" | undefined;
|
|
54
|
+
tags?: string[] | undefined;
|
|
55
|
+
importance?: "critical" | "low" | "high" | "normal" | undefined;
|
|
56
|
+
}>;
|
|
57
|
+
export type RememberInput = z.infer<typeof rememberSchema>;
|
|
58
|
+
/**
|
|
59
|
+
* Execute the remember tool
|
|
60
|
+
*/
|
|
61
|
+
export declare function executeRemember(input: RememberInput): Promise<{
|
|
62
|
+
success: boolean;
|
|
63
|
+
memory?: {
|
|
64
|
+
id: number;
|
|
65
|
+
title: string;
|
|
66
|
+
salience: number;
|
|
67
|
+
type: MemoryType;
|
|
68
|
+
category: MemoryCategory;
|
|
69
|
+
reason: string;
|
|
70
|
+
linksCreated?: number;
|
|
71
|
+
truncated?: {
|
|
72
|
+
wasTruncated: boolean;
|
|
73
|
+
originalLength: number;
|
|
74
|
+
truncatedLength: number;
|
|
75
|
+
};
|
|
76
|
+
};
|
|
77
|
+
error?: string;
|
|
78
|
+
}>;
|
|
79
|
+
/**
|
|
80
|
+
* Format the remember result for MCP response
|
|
81
|
+
*/
|
|
82
|
+
export declare function formatRememberResult(result: Awaited<ReturnType<typeof executeRemember>>): string;
|
|
83
|
+
//# sourceMappingURL=remember.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"remember.d.ts","sourceRoot":"","sources":["../../src/tools/remember.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAKhE,eAAO,MAAM,cAAc;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAqBzB,CAAC;AAEH,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC;AAE3D;;GAEG;AACH,wBAAsB,eAAe,CAAC,KAAK,EAAE,aAAa,GAAG,OAAO,CAAC;IACnE,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE;QACP,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,EAAE,MAAM,CAAC;QACd,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,UAAU,CAAC;QACjB,QAAQ,EAAE,cAAc,CAAC;QACzB,MAAM,EAAE,MAAM,CAAC;QACf,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,SAAS,CAAC,EAAE;YACV,YAAY,EAAE,OAAO,CAAC;YACtB,cAAc,EAAE,MAAM,CAAC;YACvB,eAAe,EAAE,MAAM,CAAC;SACzB,CAAC;KACH,CAAC;IACF,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB,CAAC,CA2FD;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,eAAe,CAAC,CAAC,GAAG,MAAM,CAwBhG"}
|