context-gatekeeper 0.3.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 +774 -0
- package/README.zh.md +765 -0
- package/bin/context-gatekeeper-cli.js +60 -0
- package/dist/api/gdpr.d.ts +104 -0
- package/dist/api/gdpr.d.ts.map +1 -0
- package/dist/api/gdpr.js +229 -0
- package/dist/api/gdpr.js.map +1 -0
- package/dist/api/health-check.d.ts +13 -0
- package/dist/api/health-check.d.ts.map +1 -0
- package/dist/api/health-check.js +2 -0
- package/dist/api/health-check.js.map +1 -0
- package/dist/api/index.d.ts +7 -0
- package/dist/api/index.d.ts.map +1 -0
- package/dist/api/index.js +8 -0
- package/dist/api/index.js.map +1 -0
- package/dist/api/observability.d.ts +39 -0
- package/dist/api/observability.d.ts.map +1 -0
- package/dist/api/observability.js +132 -0
- package/dist/api/observability.js.map +1 -0
- package/dist/api/session-manager.d.ts +41 -0
- package/dist/api/session-manager.d.ts.map +1 -0
- package/dist/api/session-manager.js +129 -0
- package/dist/api/session-manager.js.map +1 -0
- package/dist/index.d.ts +2 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +7 -0
- package/dist/index.js.map +1 -0
- package/dist/mcp/server.d.ts +8 -0
- package/dist/mcp/server.d.ts.map +1 -0
- package/dist/mcp/server.js +613 -0
- package/dist/mcp/server.js.map +1 -0
- package/dist/mcp/tools/configure-llm.d.ts +26 -0
- package/dist/mcp/tools/configure-llm.d.ts.map +1 -0
- package/dist/mcp/tools/configure-llm.js +32 -0
- package/dist/mcp/tools/configure-llm.js.map +1 -0
- package/dist/mcp/tools/context-compress.d.ts +15 -0
- package/dist/mcp/tools/context-compress.d.ts.map +1 -0
- package/dist/mcp/tools/context-compress.js +15 -0
- package/dist/mcp/tools/context-compress.js.map +1 -0
- package/dist/mcp/tools/dual-mode-execute.d.ts +78 -0
- package/dist/mcp/tools/dual-mode-execute.d.ts.map +1 -0
- package/dist/mcp/tools/dual-mode-execute.js +299 -0
- package/dist/mcp/tools/dual-mode-execute.js.map +1 -0
- package/dist/mcp/tools/index.d.ts +19 -0
- package/dist/mcp/tools/index.d.ts.map +1 -0
- package/dist/mcp/tools/index.js +20 -0
- package/dist/mcp/tools/index.js.map +1 -0
- package/dist/mcp/tools/intelligent-recall.d.ts +67 -0
- package/dist/mcp/tools/intelligent-recall.d.ts.map +1 -0
- package/dist/mcp/tools/intelligent-recall.js +208 -0
- package/dist/mcp/tools/intelligent-recall.js.map +1 -0
- package/dist/mcp/tools/memory-anchor.d.ts +13 -0
- package/dist/mcp/tools/memory-anchor.d.ts.map +1 -0
- package/dist/mcp/tools/memory-anchor.js +16 -0
- package/dist/mcp/tools/memory-anchor.js.map +1 -0
- package/dist/mcp/tools/memory-delete-batch.d.ts +16 -0
- package/dist/mcp/tools/memory-delete-batch.d.ts.map +1 -0
- package/dist/mcp/tools/memory-delete-batch.js +26 -0
- package/dist/mcp/tools/memory-delete-batch.js.map +1 -0
- package/dist/mcp/tools/memory-extract.d.ts +68 -0
- package/dist/mcp/tools/memory-extract.d.ts.map +1 -0
- package/dist/mcp/tools/memory-extract.js +280 -0
- package/dist/mcp/tools/memory-extract.js.map +1 -0
- package/dist/mcp/tools/memory-recall.d.ts +42 -0
- package/dist/mcp/tools/memory-recall.d.ts.map +1 -0
- package/dist/mcp/tools/memory-recall.js +37 -0
- package/dist/mcp/tools/memory-recall.js.map +1 -0
- package/dist/mcp/tools/memory-report-usage.d.ts +17 -0
- package/dist/mcp/tools/memory-report-usage.d.ts.map +1 -0
- package/dist/mcp/tools/memory-report-usage.js +15 -0
- package/dist/mcp/tools/memory-report-usage.js.map +1 -0
- package/dist/mcp/tools/memory-search.d.ts +43 -0
- package/dist/mcp/tools/memory-search.d.ts.map +1 -0
- package/dist/mcp/tools/memory-search.js +38 -0
- package/dist/mcp/tools/memory-search.js.map +1 -0
- package/dist/mcp/tools/memory-session.d.ts +118 -0
- package/dist/mcp/tools/memory-session.d.ts.map +1 -0
- package/dist/mcp/tools/memory-session.js +113 -0
- package/dist/mcp/tools/memory-session.js.map +1 -0
- package/dist/mcp/tools/memory-stats.d.ts +10 -0
- package/dist/mcp/tools/memory-stats.d.ts.map +1 -0
- package/dist/mcp/tools/memory-stats.js +35 -0
- package/dist/mcp/tools/memory-stats.js.map +1 -0
- package/dist/mcp/tools/memory-store-batch.d.ts +49 -0
- package/dist/mcp/tools/memory-store-batch.d.ts.map +1 -0
- package/dist/mcp/tools/memory-store-batch.js +48 -0
- package/dist/mcp/tools/memory-store-batch.js.map +1 -0
- package/dist/mcp/tools/memory-store.d.ts +37 -0
- package/dist/mcp/tools/memory-store.d.ts.map +1 -0
- package/dist/mcp/tools/memory-store.js +34 -0
- package/dist/mcp/tools/memory-store.js.map +1 -0
- package/dist/mcp/tools/project-create.d.ts +16 -0
- package/dist/mcp/tools/project-create.d.ts.map +1 -0
- package/dist/mcp/tools/project-create.js +14 -0
- package/dist/mcp/tools/project-create.js.map +1 -0
- package/dist/models/types.d.ts +88 -0
- package/dist/models/types.d.ts.map +1 -0
- package/dist/models/types.js +19 -0
- package/dist/models/types.js.map +1 -0
- package/dist/schema/compression.d.ts +7 -0
- package/dist/schema/compression.d.ts.map +1 -0
- package/dist/schema/compression.js +66 -0
- package/dist/schema/compression.js.map +1 -0
- package/dist/schema/fulltext-search.d.ts +10 -0
- package/dist/schema/fulltext-search.d.ts.map +1 -0
- package/dist/schema/fulltext-search.js +73 -0
- package/dist/schema/fulltext-search.js.map +1 -0
- package/dist/schema/index.d.ts +9 -0
- package/dist/schema/index.d.ts.map +1 -0
- package/dist/schema/index.js +9 -0
- package/dist/schema/index.js.map +1 -0
- package/dist/schema/knowledge-graph.d.ts +108 -0
- package/dist/schema/knowledge-graph.d.ts.map +1 -0
- package/dist/schema/knowledge-graph.js +372 -0
- package/dist/schema/knowledge-graph.js.map +1 -0
- package/dist/schema/memory-session.d.ts +62 -0
- package/dist/schema/memory-session.d.ts.map +1 -0
- package/dist/schema/memory-session.js +258 -0
- package/dist/schema/memory-session.js.map +1 -0
- package/dist/schema/memory.d.ts +84 -0
- package/dist/schema/memory.d.ts.map +1 -0
- package/dist/schema/memory.js +622 -0
- package/dist/schema/memory.js.map +1 -0
- package/dist/schema/project.d.ts +8 -0
- package/dist/schema/project.d.ts.map +1 -0
- package/dist/schema/project.js +68 -0
- package/dist/schema/project.js.map +1 -0
- package/dist/schema/schema-init.d.ts +2 -0
- package/dist/schema/schema-init.d.ts.map +1 -0
- package/dist/schema/schema-init.js +199 -0
- package/dist/schema/schema-init.js.map +1 -0
- package/dist/schema/vector-index.d.ts +28 -0
- package/dist/schema/vector-index.d.ts.map +1 -0
- package/dist/schema/vector-index.js +179 -0
- package/dist/schema/vector-index.js.map +1 -0
- package/dist/scripts/agents/base.d.ts +89 -0
- package/dist/scripts/agents/base.d.ts.map +1 -0
- package/dist/scripts/agents/base.js +148 -0
- package/dist/scripts/agents/base.js.map +1 -0
- package/dist/scripts/agents/base.ts +193 -0
- package/dist/scripts/agents/claude-code.d.ts +21 -0
- package/dist/scripts/agents/claude-code.d.ts.map +1 -0
- package/dist/scripts/agents/claude-code.js +33 -0
- package/dist/scripts/agents/claude-code.js.map +1 -0
- package/dist/scripts/agents/claude-code.ts +36 -0
- package/dist/scripts/agents/claude-desktop.d.ts +25 -0
- package/dist/scripts/agents/claude-desktop.d.ts.map +1 -0
- package/dist/scripts/agents/claude-desktop.js +36 -0
- package/dist/scripts/agents/claude-desktop.js.map +1 -0
- package/dist/scripts/agents/claude-desktop.ts +39 -0
- package/dist/scripts/agents/cline.d.ts +22 -0
- package/dist/scripts/agents/cline.d.ts.map +1 -0
- package/dist/scripts/agents/cline.js +35 -0
- package/dist/scripts/agents/cline.js.map +1 -0
- package/dist/scripts/agents/cline.ts +38 -0
- package/dist/scripts/agents/continue.d.ts +20 -0
- package/dist/scripts/agents/continue.d.ts.map +1 -0
- package/dist/scripts/agents/continue.js +35 -0
- package/dist/scripts/agents/continue.js.map +1 -0
- package/dist/scripts/agents/continue.ts +38 -0
- package/dist/scripts/agents/cursor.d.ts +27 -0
- package/dist/scripts/agents/cursor.d.ts.map +1 -0
- package/dist/scripts/agents/cursor.js +38 -0
- package/dist/scripts/agents/cursor.js.map +1 -0
- package/dist/scripts/agents/cursor.ts +41 -0
- package/dist/scripts/cli/config-gen.d.ts +59 -0
- package/dist/scripts/cli/config-gen.d.ts.map +1 -0
- package/dist/scripts/cli/config-gen.js +156 -0
- package/dist/scripts/cli/config-gen.js.map +1 -0
- package/dist/scripts/cli/config-gen.ts +164 -0
- package/dist/scripts/cli/detect.d.ts +42 -0
- package/dist/scripts/cli/detect.d.ts.map +1 -0
- package/dist/scripts/cli/detect.js +131 -0
- package/dist/scripts/cli/detect.js.map +1 -0
- package/dist/scripts/cli/detect.ts +162 -0
- package/dist/scripts/cli/install.d.ts +31 -0
- package/dist/scripts/cli/install.d.ts.map +1 -0
- package/dist/scripts/cli/install.js +125 -0
- package/dist/scripts/cli/install.js.map +1 -0
- package/dist/scripts/cli/install.ts +157 -0
- package/dist/scripts/cli/status.d.ts +8 -0
- package/dist/scripts/cli/status.d.ts.map +1 -0
- package/dist/scripts/cli/status.js +39 -0
- package/dist/scripts/cli/status.js.map +1 -0
- package/dist/scripts/cli/status.ts +48 -0
- package/dist/scripts/cli/uninstall.d.ts +22 -0
- package/dist/scripts/cli/uninstall.d.ts.map +1 -0
- package/dist/scripts/cli/uninstall.js +141 -0
- package/dist/scripts/cli/uninstall.js.map +1 -0
- package/dist/scripts/cli/uninstall.ts +157 -0
- package/dist/scripts/cli.d.ts +23 -0
- package/dist/scripts/cli.d.ts.map +1 -0
- package/dist/scripts/cli.js +166 -0
- package/dist/scripts/cli.js.map +1 -0
- package/dist/scripts/cli.ts +173 -0
- package/dist/services/classifier/index.d.ts +36 -0
- package/dist/services/classifier/index.d.ts.map +1 -0
- package/dist/services/classifier/index.js +104 -0
- package/dist/services/classifier/index.js.map +1 -0
- package/dist/services/classifier/llm.d.ts +37 -0
- package/dist/services/classifier/llm.d.ts.map +1 -0
- package/dist/services/classifier/llm.js +119 -0
- package/dist/services/classifier/llm.js.map +1 -0
- package/dist/services/classifier/rules.d.ts +22 -0
- package/dist/services/classifier/rules.d.ts.map +1 -0
- package/dist/services/classifier/rules.js +98 -0
- package/dist/services/classifier/rules.js.map +1 -0
- package/dist/services/compressor/index.d.ts +3 -0
- package/dist/services/compressor/index.d.ts.map +1 -0
- package/dist/services/compressor/index.js +3 -0
- package/dist/services/compressor/index.js.map +1 -0
- package/dist/services/compressor/threshold.d.ts +35 -0
- package/dist/services/compressor/threshold.d.ts.map +1 -0
- package/dist/services/compressor/threshold.js +60 -0
- package/dist/services/compressor/threshold.js.map +1 -0
- package/dist/services/compressor/trigger.d.ts +24 -0
- package/dist/services/compressor/trigger.d.ts.map +1 -0
- package/dist/services/compressor/trigger.js +91 -0
- package/dist/services/compressor/trigger.js.map +1 -0
- package/dist/services/constraint-extractor.d.ts +25 -0
- package/dist/services/constraint-extractor.d.ts.map +1 -0
- package/dist/services/constraint-extractor.js +97 -0
- package/dist/services/constraint-extractor.js.map +1 -0
- package/dist/services/database-health.d.ts +22 -0
- package/dist/services/database-health.d.ts.map +1 -0
- package/dist/services/database-health.js +122 -0
- package/dist/services/database-health.js.map +1 -0
- package/dist/services/embedding-fixed.d.ts +9 -0
- package/dist/services/embedding-fixed.d.ts.map +1 -0
- package/dist/services/embedding-fixed.js +70 -0
- package/dist/services/embedding-fixed.js.map +1 -0
- package/dist/services/embedding-provider.d.ts +79 -0
- package/dist/services/embedding-provider.d.ts.map +1 -0
- package/dist/services/embedding-provider.js +229 -0
- package/dist/services/embedding-provider.js.map +1 -0
- package/dist/services/embedding.d.ts +17 -0
- package/dist/services/embedding.d.ts.map +1 -0
- package/dist/services/embedding.js +99 -0
- package/dist/services/embedding.js.map +1 -0
- package/dist/services/hnsw-index.d.ts +76 -0
- package/dist/services/hnsw-index.d.ts.map +1 -0
- package/dist/services/hnsw-index.js +301 -0
- package/dist/services/hnsw-index.js.map +1 -0
- package/dist/services/llm.d.ts +39 -0
- package/dist/services/llm.d.ts.map +1 -0
- package/dist/services/llm.js +207 -0
- package/dist/services/llm.js.map +1 -0
- package/dist/services/memory-tiers.d.ts +75 -0
- package/dist/services/memory-tiers.d.ts.map +1 -0
- package/dist/services/memory-tiers.js +275 -0
- package/dist/services/memory-tiers.js.map +1 -0
- package/dist/services/memory.d.ts +33 -0
- package/dist/services/memory.d.ts.map +1 -0
- package/dist/services/memory.js +209 -0
- package/dist/services/memory.js.map +1 -0
- package/dist/services/multi-agent-sharing.d.ts +83 -0
- package/dist/services/multi-agent-sharing.d.ts.map +1 -0
- package/dist/services/multi-agent-sharing.js +278 -0
- package/dist/services/multi-agent-sharing.js.map +1 -0
- package/dist/services/reranker.d.ts +88 -0
- package/dist/services/reranker.d.ts.map +1 -0
- package/dist/services/reranker.js +234 -0
- package/dist/services/reranker.js.map +1 -0
- package/dist/services/triple-extractor.d.ts +35 -0
- package/dist/services/triple-extractor.d.ts.map +1 -0
- package/dist/services/triple-extractor.js +293 -0
- package/dist/services/triple-extractor.js.map +1 -0
- package/dist/services/vector-provider.d.ts +40 -0
- package/dist/services/vector-provider.d.ts.map +1 -0
- package/dist/services/vector-provider.js +225 -0
- package/dist/services/vector-provider.js.map +1 -0
- package/dist/utils/after-chain-executor.d.ts +26 -0
- package/dist/utils/after-chain-executor.d.ts.map +1 -0
- package/dist/utils/after-chain-executor.js +135 -0
- package/dist/utils/after-chain-executor.js.map +1 -0
- package/dist/utils/after-chain.d.ts +94 -0
- package/dist/utils/after-chain.d.ts.map +1 -0
- package/dist/utils/after-chain.js +155 -0
- package/dist/utils/after-chain.js.map +1 -0
- package/dist/utils/db.d.ts +29 -0
- package/dist/utils/db.d.ts.map +1 -0
- package/dist/utils/db.js +201 -0
- package/dist/utils/db.js.map +1 -0
- package/dist/utils/encryption.d.ts +87 -0
- package/dist/utils/encryption.d.ts.map +1 -0
- package/dist/utils/encryption.js +175 -0
- package/dist/utils/encryption.js.map +1 -0
- package/dist/utils/errors.d.ts +35 -0
- package/dist/utils/errors.d.ts.map +1 -0
- package/dist/utils/errors.js +56 -0
- package/dist/utils/errors.js.map +1 -0
- package/dist/utils/logger.d.ts +27 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +177 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/utils/priority.d.ts +26 -0
- package/dist/utils/priority.d.ts.map +1 -0
- package/dist/utils/priority.js +43 -0
- package/dist/utils/priority.js.map +1 -0
- package/dist/utils/watchdog.d.ts +57 -0
- package/dist/utils/watchdog.d.ts.map +1 -0
- package/dist/utils/watchdog.js +164 -0
- package/dist/utils/watchdog.js.map +1 -0
- package/package.json +78 -0
|
@@ -0,0 +1,280 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
/**
|
|
3
|
+
* Constraint extraction input schema
|
|
4
|
+
* Follows AutoSkill's P_ext principles:
|
|
5
|
+
* - Only use user queries, not assistant responses
|
|
6
|
+
* - Extract "durable constraints" not "one-shot requests"
|
|
7
|
+
* - Identify reusable patterns
|
|
8
|
+
*/
|
|
9
|
+
export const MemoryExtractInput = z.object({
|
|
10
|
+
conversation_turns: z.array(z.object({
|
|
11
|
+
role: z.enum(['user', 'assistant']),
|
|
12
|
+
content: z.string()
|
|
13
|
+
})).min(1).describe('Conversation turns to analyze'),
|
|
14
|
+
project_tags: z.array(z.string()).optional().describe('Project tags for extracted constraints'),
|
|
15
|
+
extract_mode: z.enum(['all', 'constraints_only', 'preferences_only']).optional()
|
|
16
|
+
.default('all').describe('Extraction mode'),
|
|
17
|
+
min_confidence: z.number().min(0).max(1).optional().default(0.5)
|
|
18
|
+
.describe('Minimum confidence threshold'),
|
|
19
|
+
use_llm: z.boolean().optional().default(true)
|
|
20
|
+
.describe('Use LLM extraction when available (fallback to keywords if unavailable)')
|
|
21
|
+
});
|
|
22
|
+
/**
|
|
23
|
+
* Keywords that indicate durable constraints vs one-shot requests
|
|
24
|
+
*/
|
|
25
|
+
const CONSTRAINT_INDICATORS = [
|
|
26
|
+
"always", "never", "must", "should", "prefer", "avoid",
|
|
27
|
+
"only", "must not", "required", "mandatory", "forbidden",
|
|
28
|
+
"consistently", "habitually", "typically", "usually", "normally",
|
|
29
|
+
"I want", "I need", "I prefer", "I hate", "I like",
|
|
30
|
+
"follow", "adhere to", "maintain", "keep", "preserve",
|
|
31
|
+
"instead of", "rather than", "instead", "rather"
|
|
32
|
+
];
|
|
33
|
+
const PREFERENCE_INDICATORS = [
|
|
34
|
+
'I like', 'I prefer', 'I enjoy', 'I hate', 'I dislike',
|
|
35
|
+
'better', 'worse', 'more', 'less', 'rather', 'instead',
|
|
36
|
+
'favorite', 'least favorite', 'ideal', 'perfect'
|
|
37
|
+
];
|
|
38
|
+
const WORKFLOW_INDICATORS = [
|
|
39
|
+
'step', 'first', 'then', 'next', 'finally', 'lastly',
|
|
40
|
+
'process', 'workflow', 'procedure', 'sequence', 'order',
|
|
41
|
+
'begin with', 'start by', 'end with', 'finish with',
|
|
42
|
+
'before', 'after', 'during', 'through'
|
|
43
|
+
];
|
|
44
|
+
const RULE_INDICATORS = [
|
|
45
|
+
'rule', 'policy', 'convention', 'standard', ' guideline',
|
|
46
|
+
'pattern', 'format', 'style', 'convention',
|
|
47
|
+
'must follow', 'should follow', 'need to follow'
|
|
48
|
+
];
|
|
49
|
+
/**
|
|
50
|
+
* One-shot request indicators (should be filtered out)
|
|
51
|
+
*/
|
|
52
|
+
const ONESHOT_INDICATORS = [
|
|
53
|
+
'can you', 'could you', 'would you', 'please',
|
|
54
|
+
'help me', 'show me', 'tell me', 'explain',
|
|
55
|
+
'how do i', 'how to', 'what is', 'what are',
|
|
56
|
+
'write a', 'create a', 'make a', 'build a',
|
|
57
|
+
'just', 'only', 'this time'
|
|
58
|
+
];
|
|
59
|
+
/**
|
|
60
|
+
* Check if a sentence contains constraint indicators
|
|
61
|
+
*/
|
|
62
|
+
function containsConstraintIndicators(text) {
|
|
63
|
+
const lowerText = text.toLowerCase();
|
|
64
|
+
return CONSTRAINT_INDICATORS.some(indicator => lowerText.includes(indicator));
|
|
65
|
+
}
|
|
66
|
+
/**
|
|
67
|
+
* Check if a sentence is likely a one-shot request
|
|
68
|
+
*/
|
|
69
|
+
function isOneShotRequest(text) {
|
|
70
|
+
const lowerText = text.toLowerCase();
|
|
71
|
+
// Check if it's a direct question or request
|
|
72
|
+
if (ONESHOT_INDICATORS.some(ind => lowerText.includes(ind))) {
|
|
73
|
+
return true;
|
|
74
|
+
}
|
|
75
|
+
// Check if it's a very short message
|
|
76
|
+
if (text.split(/\s+/).length < 5) {
|
|
77
|
+
return true;
|
|
78
|
+
}
|
|
79
|
+
return false;
|
|
80
|
+
}
|
|
81
|
+
/**
|
|
82
|
+
* Classify the type of extracted constraint
|
|
83
|
+
*/
|
|
84
|
+
function classifyConstraint(text) {
|
|
85
|
+
const lowerText = text.toLowerCase();
|
|
86
|
+
if (WORKFLOW_INDICATORS.some(ind => lowerText.includes(ind))) {
|
|
87
|
+
return 'workflow';
|
|
88
|
+
}
|
|
89
|
+
if (RULE_INDICATORS.some(ind => lowerText.includes(ind))) {
|
|
90
|
+
return 'rule';
|
|
91
|
+
}
|
|
92
|
+
if (PREFERENCE_INDICATORS.some(ind => lowerText.includes(ind))) {
|
|
93
|
+
return 'preference';
|
|
94
|
+
}
|
|
95
|
+
return 'constraint';
|
|
96
|
+
}
|
|
97
|
+
/**
|
|
98
|
+
* Extract triggers from text
|
|
99
|
+
*/
|
|
100
|
+
function extractTriggers(text) {
|
|
101
|
+
const triggers = [];
|
|
102
|
+
const lowerText = text.toLowerCase();
|
|
103
|
+
// Extract quoted phrases
|
|
104
|
+
const quotedMatches = text.match(/"([^"]+)"/g);
|
|
105
|
+
if (quotedMatches) {
|
|
106
|
+
triggers.push(...quotedMatches.map(m => m.slice(1, -1)));
|
|
107
|
+
}
|
|
108
|
+
// Extract technical terms
|
|
109
|
+
const techTerms = text.match(/\b[\w]+[\w-]*\.(ts|js|tsx|jsx|py|rs|go|java|cpp)\b/g);
|
|
110
|
+
if (techTerms) {
|
|
111
|
+
triggers.push(...techTerms.map(t => t.split('.')[1]));
|
|
112
|
+
}
|
|
113
|
+
// Extract patterns after "when" or "if"
|
|
114
|
+
const conditionalMatches = lowerText.match(/(?:when|if|whenever|before|after)\s+([^,.]+)/g);
|
|
115
|
+
if (conditionalMatches) {
|
|
116
|
+
triggers.push(...conditionalMatches);
|
|
117
|
+
}
|
|
118
|
+
return [...new Set(triggers)].slice(0, 5);
|
|
119
|
+
}
|
|
120
|
+
/**
|
|
121
|
+
* Calculate confidence score for extracted constraint
|
|
122
|
+
*/
|
|
123
|
+
function calculateConfidence(text, _turnIndex, totalTurns) {
|
|
124
|
+
let confidence = 0.5;
|
|
125
|
+
// More turns mentioning = higher confidence
|
|
126
|
+
if (totalTurns > 1) {
|
|
127
|
+
confidence += 0.1;
|
|
128
|
+
}
|
|
129
|
+
// Explicit constraint language = higher confidence
|
|
130
|
+
if (containsConstraintIndicators(text)) {
|
|
131
|
+
confidence += 0.2;
|
|
132
|
+
}
|
|
133
|
+
// Not a one-shot request = higher confidence
|
|
134
|
+
if (!isOneShotRequest(text)) {
|
|
135
|
+
confidence += 0.2;
|
|
136
|
+
}
|
|
137
|
+
// Specific and detailed = higher confidence
|
|
138
|
+
if (text.split(/\s+/).length > 10) {
|
|
139
|
+
confidence += 0.1;
|
|
140
|
+
}
|
|
141
|
+
return Math.min(confidence, 1.0);
|
|
142
|
+
}
|
|
143
|
+
/**
|
|
144
|
+
* Extract constraints from user turns only
|
|
145
|
+
*/
|
|
146
|
+
export async function memoryExtractTool(input) {
|
|
147
|
+
const { conversation_turns, project_tags: _project_tags, extract_mode, min_confidence, use_llm = true } = input;
|
|
148
|
+
// 尝试使用 LLM 提取(如果配置了且启用)
|
|
149
|
+
if (use_llm) {
|
|
150
|
+
try {
|
|
151
|
+
const { extractConstraintsWithLLM } = await import('../../services/constraint-extractor.js');
|
|
152
|
+
const llmResult = await extractConstraintsWithLLM(conversation_turns, extract_mode || 'all', min_confidence);
|
|
153
|
+
if (llmResult.used_llm && llmResult.constraints.length > 0) {
|
|
154
|
+
// 合并 LLM 提取结果与关键词提取结果
|
|
155
|
+
const keywordConstraints = extractWithKeywords(conversation_turns, extract_mode || 'all', min_confidence);
|
|
156
|
+
const mergedConstraints = [...llmResult.constraints.map(c => ({
|
|
157
|
+
content: c.content,
|
|
158
|
+
type: c.type,
|
|
159
|
+
confidence: c.confidence,
|
|
160
|
+
triggers: c.triggers,
|
|
161
|
+
source_turns: [],
|
|
162
|
+
reasoning: c.reasoning,
|
|
163
|
+
})), ...keywordConstraints];
|
|
164
|
+
const deduplicated = deduplicateConstraints(mergedConstraints);
|
|
165
|
+
return {
|
|
166
|
+
constraints: deduplicated,
|
|
167
|
+
summary: `${llmResult.summary}; keyword fallback found ${keywordConstraints.length} additional constraint(s)`,
|
|
168
|
+
extraction_mode: extract_mode || 'all',
|
|
169
|
+
total_turns_analyzed: conversation_turns.length,
|
|
170
|
+
};
|
|
171
|
+
}
|
|
172
|
+
}
|
|
173
|
+
catch {
|
|
174
|
+
// LLM 提取失败,降级到关键词提取
|
|
175
|
+
}
|
|
176
|
+
}
|
|
177
|
+
// 关键词提取(降级路径)
|
|
178
|
+
const constraints = extractWithKeywords(conversation_turns, extract_mode || 'all', min_confidence);
|
|
179
|
+
const deduplicated = deduplicateConstraints(constraints);
|
|
180
|
+
const summary = generateSummary(deduplicated);
|
|
181
|
+
return {
|
|
182
|
+
constraints: deduplicated,
|
|
183
|
+
summary,
|
|
184
|
+
extraction_mode: extract_mode || 'all',
|
|
185
|
+
total_turns_analyzed: conversation_turns.length,
|
|
186
|
+
};
|
|
187
|
+
}
|
|
188
|
+
/**
|
|
189
|
+
* 关键词提取(降级方案)
|
|
190
|
+
*/
|
|
191
|
+
function extractWithKeywords(conversation_turns, extract_mode, min_confidence) {
|
|
192
|
+
// Filter to user turns only (following AutoSkill principle)
|
|
193
|
+
const userTurns = conversation_turns
|
|
194
|
+
.filter(turn => turn.role === 'user')
|
|
195
|
+
.map((turn, index) => ({ ...turn, originalIndex: index }));
|
|
196
|
+
const extractedConstraints = [];
|
|
197
|
+
for (const turn of userTurns) {
|
|
198
|
+
const content = turn.content;
|
|
199
|
+
// Skip one-shot requests
|
|
200
|
+
if (isOneShotRequest(content)) {
|
|
201
|
+
continue;
|
|
202
|
+
}
|
|
203
|
+
// Analyze sentence by sentence
|
|
204
|
+
const sentences = content.split(/[.!?]+/).filter(s => s.trim().length > 10);
|
|
205
|
+
for (const sentence of sentences) {
|
|
206
|
+
const trimmed = sentence.trim();
|
|
207
|
+
if (trimmed.length < 15)
|
|
208
|
+
continue;
|
|
209
|
+
const type = classifyConstraint(trimmed);
|
|
210
|
+
const confidence = calculateConfidence(trimmed, turn.originalIndex ?? 0, userTurns.length);
|
|
211
|
+
// Filter by mode and confidence
|
|
212
|
+
if (confidence < min_confidence)
|
|
213
|
+
continue;
|
|
214
|
+
if (extract_mode === 'constraints_only' && type !== 'constraint' && type !== 'rule')
|
|
215
|
+
continue;
|
|
216
|
+
if (extract_mode === 'preferences_only' && type !== 'preference')
|
|
217
|
+
continue;
|
|
218
|
+
extractedConstraints.push({
|
|
219
|
+
content: trimmed,
|
|
220
|
+
type,
|
|
221
|
+
confidence,
|
|
222
|
+
triggers: extractTriggers(trimmed),
|
|
223
|
+
source_turns: [turn.originalIndex ?? 0],
|
|
224
|
+
reasoning: `Extracted from user turn ${(turn.originalIndex ?? 0) + 1} with ${type} pattern`,
|
|
225
|
+
});
|
|
226
|
+
}
|
|
227
|
+
}
|
|
228
|
+
return extractedConstraints;
|
|
229
|
+
}
|
|
230
|
+
/**
|
|
231
|
+
* Deduplicate similar constraints using simple similarity
|
|
232
|
+
*/
|
|
233
|
+
function deduplicateConstraints(constraints) {
|
|
234
|
+
const result = [];
|
|
235
|
+
for (const constraint of constraints) {
|
|
236
|
+
const isDuplicate = result.some(existing => {
|
|
237
|
+
const similarity = calculateTextSimilarity(constraint.content, existing.content);
|
|
238
|
+
return similarity > 0.7;
|
|
239
|
+
});
|
|
240
|
+
if (!isDuplicate) {
|
|
241
|
+
result.push(constraint);
|
|
242
|
+
}
|
|
243
|
+
}
|
|
244
|
+
return result;
|
|
245
|
+
}
|
|
246
|
+
/**
|
|
247
|
+
* Calculate simple text similarity (Jaccard)
|
|
248
|
+
*/
|
|
249
|
+
function calculateTextSimilarity(text1, text2) {
|
|
250
|
+
const words1 = new Set(text1.toLowerCase().split(/\s+/));
|
|
251
|
+
const words2 = new Set(text2.toLowerCase().split(/\s+/));
|
|
252
|
+
const intersection = new Set([...words1].filter(x => words2.has(x)));
|
|
253
|
+
const union = new Set([...words1, ...words2]);
|
|
254
|
+
return intersection.size / union.size;
|
|
255
|
+
}
|
|
256
|
+
/**
|
|
257
|
+
* Generate summary of extracted constraints
|
|
258
|
+
*/
|
|
259
|
+
function generateSummary(constraints) {
|
|
260
|
+
if (constraints.length === 0) {
|
|
261
|
+
return 'No durable constraints or preferences detected in the conversation.';
|
|
262
|
+
}
|
|
263
|
+
const byType = {
|
|
264
|
+
constraint: constraints.filter(c => c.type === 'constraint' || c.type === 'rule'),
|
|
265
|
+
preference: constraints.filter(c => c.type === 'preference'),
|
|
266
|
+
workflow: constraints.filter(c => c.type === 'workflow')
|
|
267
|
+
};
|
|
268
|
+
const parts = [];
|
|
269
|
+
if (byType.constraint.length > 0) {
|
|
270
|
+
parts.push(`${byType.constraint.length} constraint(s)`);
|
|
271
|
+
}
|
|
272
|
+
if (byType.preference.length > 0) {
|
|
273
|
+
parts.push(`${byType.preference.length} preference(s)`);
|
|
274
|
+
}
|
|
275
|
+
if (byType.workflow.length > 0) {
|
|
276
|
+
parts.push(`${byType.workflow.length} workflow(s)`);
|
|
277
|
+
}
|
|
278
|
+
return `Extracted ${parts.join(', ')} from conversation.`;
|
|
279
|
+
}
|
|
280
|
+
//# sourceMappingURL=memory-extract.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-extract.js","sourceRoot":"","sources":["../../../src/mcp/tools/memory-extract.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IACzC,kBAAkB,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;QACnC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QACnC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;KACpB,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,+BAA+B,CAAC;IACpD,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wCAAwC,CAAC;IAC/F,YAAY,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,kBAAkB,EAAE,kBAAkB,CAAC,CAAC,CAAC,QAAQ,EAAE;SAC7E,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,iBAAiB,CAAC;IAC7C,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC;SAC7D,QAAQ,CAAC,8BAA8B,CAAC;IAC3C,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;SAC1C,QAAQ,CAAC,yEAAyE,CAAC;CACvF,CAAC,CAAC;AA0BH;;GAEG;AACH,MAAM,qBAAqB,GAAG;IAC5B,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO;IACtD,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,WAAW;IACxD,cAAc,EAAE,YAAY,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU;IAChE,QAAQ,EAAE,QAAQ,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ;IAClD,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE,UAAU;IACrD,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,QAAQ;CACjD,CAAC;AAEF,MAAM,qBAAqB,GAAG;IAC5B,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,WAAW;IACtD,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS;IACtD,UAAU,EAAE,gBAAgB,EAAE,OAAO,EAAE,SAAS;CACjD,CAAC;AAEF,MAAM,mBAAmB,GAAG;IAC1B,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ;IACpD,SAAS,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO;IACvD,YAAY,EAAE,UAAU,EAAE,UAAU,EAAE,aAAa;IACnD,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS;CACvC,CAAC;AAEF,MAAM,eAAe,GAAG;IACtB,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,UAAU,EAAE,YAAY;IACxD,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY;IAC1C,aAAa,EAAE,eAAe,EAAE,gBAAgB;CACjD,CAAC;AAEF;;GAEG;AACH,MAAM,kBAAkB,GAAG;IACzB,SAAS,EAAE,WAAW,EAAE,WAAW,EAAE,QAAQ;IAC7C,SAAS,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS;IAC1C,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU;IAC3C,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,SAAS;IAC1C,MAAM,EAAE,MAAM,EAAE,WAAW;CAC5B,CAAC;AAEF;;GAEG;AACH,SAAS,4BAA4B,CAAC,IAAY;IAChD,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACrC,OAAO,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;AAChF,CAAC;AAED;;GAEG;AACH,SAAS,gBAAgB,CAAC,IAAY;IACpC,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACrC,6CAA6C;IAC7C,IAAI,kBAAkB,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC5D,OAAO,IAAI,CAAC;IACd,CAAC;IACD,qCAAqC;IACrC,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjC,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAS,kBAAkB,CAAC,IAAY;IACtC,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IAErC,IAAI,mBAAmB,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC7D,OAAO,UAAU,CAAC;IACpB,CAAC;IACD,IAAI,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QACzD,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,IAAI,qBAAqB,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;QAC/D,OAAO,YAAY,CAAC;IACtB,CAAC;IACD,OAAO,YAAY,CAAC;AACtB,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,IAAY;IACnC,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IAErC,yBAAyB;IACzB,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAC/C,IAAI,aAAa,EAAE,CAAC;QAClB,QAAQ,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,0BAA0B;IAC1B,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,qDAAqD,CAAC,CAAC;IACpF,IAAI,SAAS,EAAE,CAAC;QACd,QAAQ,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACxD,CAAC;IAED,wCAAwC;IACxC,MAAM,kBAAkB,GAAG,SAAS,CAAC,KAAK,CAAC,+CAA+C,CAAC,CAAC;IAC5F,IAAI,kBAAkB,EAAE,CAAC;QACvB,QAAQ,CAAC,IAAI,CAAC,GAAG,kBAAkB,CAAC,CAAC;IACvC,CAAC;IAED,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AAC5C,CAAC;AAED;;GAEG;AACH,SAAS,mBAAmB,CAAC,IAAY,EAAE,UAAkB,EAAE,UAAkB;IAC/E,IAAI,UAAU,GAAG,GAAG,CAAC;IAErB,4CAA4C;IAC5C,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;QACnB,UAAU,IAAI,GAAG,CAAC;IACpB,CAAC;IAED,mDAAmD;IACnD,IAAI,4BAA4B,CAAC,IAAI,CAAC,EAAE,CAAC;QACvC,UAAU,IAAI,GAAG,CAAC;IACpB,CAAC;IAED,6CAA6C;IAC7C,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;QAC5B,UAAU,IAAI,GAAG,CAAC;IACpB,CAAC;IAED,4CAA4C;IAC5C,IAAI,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;QAClC,UAAU,IAAI,GAAG,CAAC;IACpB,CAAC;IAED,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;AACnC,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CAAC,KAA6B;IACnE,MAAM,EAAE,kBAAkB,EAAE,YAAY,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAE,OAAO,GAAG,IAAI,EAAE,GAAG,KAAK,CAAC;IAEhH,wBAAwB;IACxB,IAAI,OAAO,EAAE,CAAC;QACZ,IAAI,CAAC;YACH,MAAM,EAAE,yBAAyB,EAAE,GAAG,MAAM,MAAM,CAAC,wCAAwC,CAAC,CAAC;YAC7F,MAAM,SAAS,GAAG,MAAM,yBAAyB,CAAC,kBAAkB,EAAE,YAAY,IAAI,KAAK,EAAE,cAAc,CAAC,CAAC;YAE7G,IAAI,SAAS,CAAC,QAAQ,IAAI,SAAS,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3D,sBAAsB;gBACtB,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,kBAAkB,EAAE,YAAY,IAAI,KAAK,EAAE,cAAc,CAAC,CAAC;gBAC1G,MAAM,iBAAiB,GAAG,CAAC,GAAG,SAAS,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;wBAC5D,OAAO,EAAE,CAAC,CAAC,OAAO;wBAClB,IAAI,EAAE,CAAC,CAAC,IAAI;wBACZ,UAAU,EAAE,CAAC,CAAC,UAAU;wBACxB,QAAQ,EAAE,CAAC,CAAC,QAAQ;wBACpB,YAAY,EAAE,EAAE;wBAChB,SAAS,EAAE,CAAC,CAAC,SAAS;qBACvB,CAAC,CAAC,EAAE,GAAG,kBAAkB,CAAC,CAAC;gBAC5B,MAAM,YAAY,GAAG,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;gBAC/D,OAAO;oBACL,WAAW,EAAE,YAAY;oBACzB,OAAO,EAAE,GAAG,SAAS,CAAC,OAAO,4BAA4B,kBAAkB,CAAC,MAAM,2BAA2B;oBAC7G,eAAe,EAAE,YAAY,IAAI,KAAK;oBACtC,oBAAoB,EAAE,kBAAkB,CAAC,MAAM;iBAChD,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,oBAAoB;QACtB,CAAC;IACH,CAAC;IAED,cAAc;IACd,MAAM,WAAW,GAAG,mBAAmB,CAAC,kBAAkB,EAAE,YAAY,IAAI,KAAK,EAAE,cAAc,CAAC,CAAC;IACnG,MAAM,YAAY,GAAG,sBAAsB,CAAC,WAAW,CAAC,CAAC;IACzD,MAAM,OAAO,GAAG,eAAe,CAAC,YAAY,CAAC,CAAC;IAE9C,OAAO;QACL,WAAW,EAAE,YAAY;QACzB,OAAO;QACP,eAAe,EAAE,YAAY,IAAI,KAAK;QACtC,oBAAoB,EAAE,kBAAkB,CAAC,MAAM;KAChD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,SAAS,mBAAmB,CAC1B,kBAAkG,EAClG,YAA6D,EAC7D,cAAsB;IAEtB,4DAA4D;IAC5D,MAAM,SAAS,GAAG,kBAAkB;SACjC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;SACpC,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;IAE7D,MAAM,oBAAoB,GAA0B,EAAE,CAAC;IAEvD,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAE7B,yBAAyB;QACzB,IAAI,gBAAgB,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9B,SAAS;QACX,CAAC;QAED,+BAA+B;QAC/B,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;QAE5E,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;YAChC,IAAI,OAAO,CAAC,MAAM,GAAG,EAAE;gBAAE,SAAS;YAElC,MAAM,IAAI,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;YACzC,MAAM,UAAU,GAAG,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,IAAI,CAAC,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;YAE3F,gCAAgC;YAChC,IAAI,UAAU,GAAG,cAAc;gBAAE,SAAS;YAE1C,IAAI,YAAY,KAAK,kBAAkB,IAAI,IAAI,KAAK,YAAY,IAAI,IAAI,KAAK,MAAM;gBAAE,SAAS;YAC9F,IAAI,YAAY,KAAK,kBAAkB,IAAI,IAAI,KAAK,YAAY;gBAAE,SAAS;YAE3E,oBAAoB,CAAC,IAAI,CAAC;gBACxB,OAAO,EAAE,OAAO;gBAChB,IAAI;gBACJ,UAAU;gBACV,QAAQ,EAAE,eAAe,CAAC,OAAO,CAAC;gBAClC,YAAY,EAAE,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,CAAC;gBACvC,SAAS,EAAE,4BAA4B,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,IAAI,UAAU;aAC5F,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED;;GAEG;AACH,SAAS,sBAAsB,CAAC,WAAkC;IAChE,MAAM,MAAM,GAA0B,EAAE,CAAC;IAEzC,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;QACrC,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE;YACzC,MAAM,UAAU,GAAG,uBAAuB,CAAC,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YACjF,OAAO,UAAU,GAAG,GAAG,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,SAAS,uBAAuB,CAAC,KAAa,EAAE,KAAa;IAC3D,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IACzD,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IAEzD,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACrE,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC;IAE9C,OAAO,YAAY,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;AACxC,CAAC;AAED;;GAEG;AACH,SAAS,eAAe,CAAC,WAAkC;IACzD,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,qEAAqE,CAAC;IAC/E,CAAC;IAED,MAAM,MAAM,GAAG;QACb,UAAU,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,YAAY,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;QACjF,UAAU,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,YAAY,CAAC;QAC5D,QAAQ,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC;KACzD,CAAC;IAEF,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,gBAAgB,CAAC,CAAC;IAC1D,CAAC;IACD,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,gBAAgB,CAAC,CAAC;IAC1D,CAAC;IACD,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,KAAK,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,cAAc,CAAC,CAAC;IACtD,CAAC;IAED,OAAO,aAAa,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,qBAAqB,CAAC;AAC5D,CAAC"}
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
export declare const MemoryRecallInput: z.ZodObject<{
|
|
3
|
+
query: z.ZodString;
|
|
4
|
+
project_tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
|
|
5
|
+
limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
|
|
6
|
+
search_mode: z.ZodDefault<z.ZodOptional<z.ZodEnum<["keyword", "semantic", "hybrid", "auto"]>>>;
|
|
7
|
+
user_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
8
|
+
agent_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
9
|
+
project_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
10
|
+
}, "strip", z.ZodTypeAny, {
|
|
11
|
+
query: string;
|
|
12
|
+
limit: number;
|
|
13
|
+
search_mode: "auto" | "keyword" | "semantic" | "hybrid";
|
|
14
|
+
user_id: string;
|
|
15
|
+
agent_id: string;
|
|
16
|
+
project_id: string;
|
|
17
|
+
project_tags?: string[] | undefined;
|
|
18
|
+
}, {
|
|
19
|
+
query: string;
|
|
20
|
+
project_tags?: string[] | undefined;
|
|
21
|
+
limit?: number | undefined;
|
|
22
|
+
search_mode?: "auto" | "keyword" | "semantic" | "hybrid" | undefined;
|
|
23
|
+
user_id?: string | undefined;
|
|
24
|
+
agent_id?: string | undefined;
|
|
25
|
+
project_id?: string | undefined;
|
|
26
|
+
}>;
|
|
27
|
+
export type MemoryRecallInputType = z.infer<typeof MemoryRecallInput>;
|
|
28
|
+
export interface MemoryRecallOutput {
|
|
29
|
+
memories: Array<{
|
|
30
|
+
id: string;
|
|
31
|
+
content: string;
|
|
32
|
+
priority: string;
|
|
33
|
+
project_tags: string[];
|
|
34
|
+
anchored: boolean;
|
|
35
|
+
access_count: number;
|
|
36
|
+
created_at: string;
|
|
37
|
+
updated_at: string;
|
|
38
|
+
}>;
|
|
39
|
+
search_mode: string;
|
|
40
|
+
}
|
|
41
|
+
export declare function memoryRecallTool(input: MemoryRecallInputType): Promise<MemoryRecallOutput>;
|
|
42
|
+
//# sourceMappingURL=memory-recall.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-recall.d.ts","sourceRoot":"","sources":["../../../src/mcp/tools/memory-recall.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;EAS5B,CAAC;AAEH,MAAM,MAAM,qBAAqB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAEtE,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,KAAK,CAAC;QACd,EAAE,EAAE,MAAM,CAAC;QACX,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC;QACjB,YAAY,EAAE,MAAM,EAAE,CAAC;QACvB,QAAQ,EAAE,OAAO,CAAC;QAClB,YAAY,EAAE,MAAM,CAAC;QACrB,UAAU,EAAE,MAAM,CAAC;QACnB,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC,CAAC;IACH,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,wBAAsB,gBAAgB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAuBhG"}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { getMemoryService } from '../../services/memory.js';
|
|
3
|
+
const memoryService = getMemoryService();
|
|
4
|
+
export const MemoryRecallInput = z.object({
|
|
5
|
+
query: z.string().describe('Search query for memories'),
|
|
6
|
+
project_tags: z.array(z.string()).optional().describe('Filter by project tags'),
|
|
7
|
+
limit: z.number().int().positive().optional().default(10).describe('Maximum number of results'),
|
|
8
|
+
search_mode: z.enum(['keyword', 'semantic', 'hybrid', 'auto']).optional().default('auto').describe('Search mode: keyword (BM25), semantic (vector), hybrid (both), auto (detect)'),
|
|
9
|
+
// Storage isolation fields
|
|
10
|
+
user_id: z.string().optional().default('default').describe('User ID for isolation'),
|
|
11
|
+
agent_id: z.string().optional().default('default').describe('Agent ID for isolation'),
|
|
12
|
+
project_id: z.string().optional().default('default').describe('Project ID for isolation')
|
|
13
|
+
});
|
|
14
|
+
export async function memoryRecallTool(input) {
|
|
15
|
+
const memories = memoryService.recallMemories({
|
|
16
|
+
query: input.query,
|
|
17
|
+
projectTags: input.project_tags,
|
|
18
|
+
limit: input.limit,
|
|
19
|
+
userId: input.user_id,
|
|
20
|
+
agentId: input.agent_id,
|
|
21
|
+
projectId: input.project_id
|
|
22
|
+
}, input.search_mode);
|
|
23
|
+
return {
|
|
24
|
+
memories: memories.map(m => ({
|
|
25
|
+
id: m.id,
|
|
26
|
+
content: m.content,
|
|
27
|
+
priority: m.priority,
|
|
28
|
+
project_tags: m.projectTags,
|
|
29
|
+
anchored: m.anchored,
|
|
30
|
+
access_count: m.accessCount,
|
|
31
|
+
created_at: m.createdAt.toISOString(),
|
|
32
|
+
updated_at: m.updatedAt.toISOString()
|
|
33
|
+
})),
|
|
34
|
+
search_mode: input.search_mode || 'auto'
|
|
35
|
+
};
|
|
36
|
+
}
|
|
37
|
+
//# sourceMappingURL=memory-recall.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-recall.js","sourceRoot":"","sources":["../../../src/mcp/tools/memory-recall.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,gBAAgB,EAAc,MAAM,0BAA0B,CAAC;AAExE,MAAM,aAAa,GAAG,gBAAgB,EAAE,CAAC;AAEzC,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;IACvD,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;IAC/E,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;IAC/F,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,8EAA8E,CAAC;IAClL,2BAA2B;IAC3B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,uBAAuB,CAAC;IACnF,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,wBAAwB,CAAC;IACrF,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,0BAA0B,CAAC;CAC1F,CAAC,CAAC;AAkBH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,KAA4B;IACjE,MAAM,QAAQ,GAAG,aAAa,CAAC,cAAc,CAAC;QAC5C,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,WAAW,EAAE,KAAK,CAAC,YAAY;QAC/B,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,MAAM,EAAE,KAAK,CAAC,OAAO;QACrB,OAAO,EAAE,KAAK,CAAC,QAAQ;QACvB,SAAS,EAAE,KAAK,CAAC,UAAU;KAC5B,EAAE,KAAK,CAAC,WAAyB,CAAC,CAAC;IAEpC,OAAO;QACL,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC3B,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;YACpB,YAAY,EAAE,CAAC,CAAC,WAAW;YAC3B,QAAQ,EAAE,CAAC,CAAC,QAAQ;YACpB,YAAY,EAAE,CAAC,CAAC,WAAW;YAC3B,UAAU,EAAE,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE;YACrC,UAAU,EAAE,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE;SACtC,CAAC,CAAC;QACH,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,MAAM;KACzC,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
export declare const MemoryReportUsageInput: z.ZodObject<{
|
|
3
|
+
used_tokens: z.ZodNumber;
|
|
4
|
+
max_tokens: z.ZodNumber;
|
|
5
|
+
}, "strip", z.ZodTypeAny, {
|
|
6
|
+
used_tokens: number;
|
|
7
|
+
max_tokens: number;
|
|
8
|
+
}, {
|
|
9
|
+
used_tokens: number;
|
|
10
|
+
max_tokens: number;
|
|
11
|
+
}>;
|
|
12
|
+
export type MemoryReportUsageInputType = z.infer<typeof MemoryReportUsageInput>;
|
|
13
|
+
export declare function memoryReportUsageTool(input: MemoryReportUsageInputType): Promise<{
|
|
14
|
+
should_compress: boolean;
|
|
15
|
+
current_ratio: number;
|
|
16
|
+
}>;
|
|
17
|
+
//# sourceMappingURL=memory-report-usage.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-report-usage.d.ts","sourceRoot":"","sources":["../../../src/mcp/tools/memory-report-usage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,eAAO,MAAM,sBAAsB;;;;;;;;;EAGjC,CAAC;AAEH,MAAM,MAAM,0BAA0B,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAEhF,wBAAsB,qBAAqB,CAAC,KAAK,EAAE,0BAA0B,GAAG,OAAO,CAAC;IACtF,eAAe,EAAE,OAAO,CAAC;IACzB,aAAa,EAAE,MAAM,CAAA;CACtB,CAAC,CAQD"}
|
|
@@ -0,0 +1,15 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { getThresholdDetector } from '../../services/compressor/threshold.js';
|
|
3
|
+
export const MemoryReportUsageInput = z.object({
|
|
4
|
+
used_tokens: z.number().int().nonnegative().describe('Current token usage'),
|
|
5
|
+
max_tokens: z.number().int().positive().describe('Maximum token limit')
|
|
6
|
+
});
|
|
7
|
+
export async function memoryReportUsageTool(input) {
|
|
8
|
+
const detector = getThresholdDetector();
|
|
9
|
+
const result = detector.reportUsage(input.used_tokens, input.max_tokens);
|
|
10
|
+
return {
|
|
11
|
+
should_compress: result.shouldCompress,
|
|
12
|
+
current_ratio: result.currentRatio
|
|
13
|
+
};
|
|
14
|
+
}
|
|
15
|
+
//# sourceMappingURL=memory-report-usage.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-report-usage.js","sourceRoot":"","sources":["../../../src/mcp/tools/memory-report-usage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,oBAAoB,EAAE,MAAM,wCAAwC,CAAC;AAE9E,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7C,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;IAC3E,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;CACxE,CAAC,CAAC;AAIH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,KAAiC;IAI3E,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;IACxC,MAAM,MAAM,GAAG,QAAQ,CAAC,WAAW,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;IAEzE,OAAO;QACL,eAAe,EAAE,MAAM,CAAC,cAAc;QACtC,aAAa,EAAE,MAAM,CAAC,YAAY;KACnC,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
export declare const MemorySearchInput: z.ZodObject<{
|
|
3
|
+
query: z.ZodString;
|
|
4
|
+
project_tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
|
|
5
|
+
limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
|
|
6
|
+
search_mode: z.ZodDefault<z.ZodOptional<z.ZodEnum<["keyword", "semantic", "hybrid", "auto"]>>>;
|
|
7
|
+
user_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
8
|
+
agent_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
9
|
+
project_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
10
|
+
}, "strip", z.ZodTypeAny, {
|
|
11
|
+
query: string;
|
|
12
|
+
limit: number;
|
|
13
|
+
search_mode: "auto" | "keyword" | "semantic" | "hybrid";
|
|
14
|
+
user_id: string;
|
|
15
|
+
agent_id: string;
|
|
16
|
+
project_id: string;
|
|
17
|
+
project_tags?: string[] | undefined;
|
|
18
|
+
}, {
|
|
19
|
+
query: string;
|
|
20
|
+
project_tags?: string[] | undefined;
|
|
21
|
+
limit?: number | undefined;
|
|
22
|
+
search_mode?: "auto" | "keyword" | "semantic" | "hybrid" | undefined;
|
|
23
|
+
user_id?: string | undefined;
|
|
24
|
+
agent_id?: string | undefined;
|
|
25
|
+
project_id?: string | undefined;
|
|
26
|
+
}>;
|
|
27
|
+
export type MemorySearchInputType = z.infer<typeof MemorySearchInput>;
|
|
28
|
+
export interface MemorySearchOutput {
|
|
29
|
+
memories: Array<{
|
|
30
|
+
id: string;
|
|
31
|
+
content: string;
|
|
32
|
+
priority: string;
|
|
33
|
+
project_tags: string[];
|
|
34
|
+
anchored: boolean;
|
|
35
|
+
access_count: number;
|
|
36
|
+
created_at: string;
|
|
37
|
+
updated_at: string;
|
|
38
|
+
}>;
|
|
39
|
+
search_mode: string;
|
|
40
|
+
total: number;
|
|
41
|
+
}
|
|
42
|
+
export declare function memorySearchTool(input: MemorySearchInputType): Promise<MemorySearchOutput>;
|
|
43
|
+
//# sourceMappingURL=memory-search.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-search.d.ts","sourceRoot":"","sources":["../../../src/mcp/tools/memory-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;EAS5B,CAAC;AAEH,MAAM,MAAM,qBAAqB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAEtE,MAAM,WAAW,kBAAkB;IACjC,QAAQ,EAAE,KAAK,CAAC;QACd,EAAE,EAAE,MAAM,CAAC;QACX,OAAO,EAAE,MAAM,CAAC;QAChB,QAAQ,EAAE,MAAM,CAAC;QACjB,YAAY,EAAE,MAAM,EAAE,CAAC;QACvB,QAAQ,EAAE,OAAO,CAAC;QAClB,YAAY,EAAE,MAAM,CAAC;QACrB,UAAU,EAAE,MAAM,CAAC;QACnB,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC,CAAC;IACH,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,wBAAsB,gBAAgB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,kBAAkB,CAAC,CA2BhG"}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { getMemoryService } from '../../services/memory.js';
|
|
3
|
+
const memoryService = getMemoryService();
|
|
4
|
+
export const MemorySearchInput = z.object({
|
|
5
|
+
query: z.string().describe('Search query'),
|
|
6
|
+
project_tags: z.array(z.string()).optional().describe('Filter by project tags'),
|
|
7
|
+
limit: z.number().int().positive().optional().default(10).describe('Maximum number of results'),
|
|
8
|
+
search_mode: z.enum(['keyword', 'semantic', 'hybrid', 'auto']).optional().default('auto').describe('Search mode: keyword (BM25), semantic (vector), hybrid (both), or auto (detect)'),
|
|
9
|
+
// Storage isolation fields
|
|
10
|
+
user_id: z.string().optional().default('default').describe('User ID for isolation'),
|
|
11
|
+
agent_id: z.string().optional().default('default').describe('Agent ID for isolation'),
|
|
12
|
+
project_id: z.string().optional().default('default').describe('Project ID for isolation')
|
|
13
|
+
});
|
|
14
|
+
export async function memorySearchTool(input) {
|
|
15
|
+
const memories = memoryService.recallMemories({
|
|
16
|
+
query: input.query,
|
|
17
|
+
projectTags: input.project_tags,
|
|
18
|
+
limit: input.limit,
|
|
19
|
+
userId: input.user_id,
|
|
20
|
+
agentId: input.agent_id,
|
|
21
|
+
projectId: input.project_id
|
|
22
|
+
}, input.search_mode);
|
|
23
|
+
return {
|
|
24
|
+
memories: memories.map(m => ({
|
|
25
|
+
id: m.id,
|
|
26
|
+
content: m.content,
|
|
27
|
+
priority: m.priority,
|
|
28
|
+
project_tags: m.projectTags,
|
|
29
|
+
anchored: m.anchored,
|
|
30
|
+
access_count: m.accessCount,
|
|
31
|
+
created_at: m.createdAt.toISOString(),
|
|
32
|
+
updated_at: m.updatedAt.toISOString()
|
|
33
|
+
})),
|
|
34
|
+
search_mode: input.search_mode,
|
|
35
|
+
total: memories.length
|
|
36
|
+
};
|
|
37
|
+
}
|
|
38
|
+
//# sourceMappingURL=memory-search.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-search.js","sourceRoot":"","sources":["../../../src/mcp/tools/memory-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,gBAAgB,EAAc,MAAM,0BAA0B,CAAC;AAExE,MAAM,aAAa,GAAG,gBAAgB,EAAE,CAAC;AAEzC,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;IAC1C,YAAY,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;IAC/E,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;IAC/F,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,iFAAiF,CAAC;IACrL,2BAA2B;IAC3B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,uBAAuB,CAAC;IACnF,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,wBAAwB,CAAC;IACrF,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,QAAQ,CAAC,0BAA0B,CAAC;CAC1F,CAAC,CAAC;AAmBH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,KAA4B;IACjE,MAAM,QAAQ,GAAG,aAAa,CAAC,cAAc,CAC3C;QACE,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,WAAW,EAAE,KAAK,CAAC,YAAY;QAC/B,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,MAAM,EAAE,KAAK,CAAC,OAAO;QACrB,OAAO,EAAE,KAAK,CAAC,QAAQ;QACvB,SAAS,EAAE,KAAK,CAAC,UAAU;KAC5B,EACD,KAAK,CAAC,WAAyB,CAChC,CAAC;IAEF,OAAO;QACL,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC3B,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,QAAQ,EAAE,CAAC,CAAC,QAAQ;YACpB,YAAY,EAAE,CAAC,CAAC,WAAW;YAC3B,QAAQ,EAAE,CAAC,CAAC,QAAQ;YACpB,YAAY,EAAE,CAAC,CAAC,WAAW;YAC3B,UAAU,EAAE,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE;YACrC,UAAU,EAAE,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE;SACtC,CAAC,CAAC;QACH,WAAW,EAAE,KAAK,CAAC,WAAW;QAC9B,KAAK,EAAE,QAAQ,CAAC,MAAM;KACvB,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,118 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
export declare const MemorySessionStoreInput: z.ZodObject<{
|
|
3
|
+
key: z.ZodString;
|
|
4
|
+
value: z.ZodString;
|
|
5
|
+
scope: z.ZodDefault<z.ZodEnum<["session", "short", "long", "archival"]>>;
|
|
6
|
+
expires_in_hours: z.ZodOptional<z.ZodNumber>;
|
|
7
|
+
meta: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
|
|
8
|
+
user_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
9
|
+
agent_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
10
|
+
project_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
11
|
+
}, "strip", z.ZodTypeAny, {
|
|
12
|
+
value: string;
|
|
13
|
+
key: string;
|
|
14
|
+
scope: "session" | "short" | "long" | "archival";
|
|
15
|
+
user_id: string;
|
|
16
|
+
agent_id: string;
|
|
17
|
+
project_id: string;
|
|
18
|
+
expires_in_hours?: number | undefined;
|
|
19
|
+
meta?: Record<string, unknown> | undefined;
|
|
20
|
+
}, {
|
|
21
|
+
value: string;
|
|
22
|
+
key: string;
|
|
23
|
+
scope?: "session" | "short" | "long" | "archival" | undefined;
|
|
24
|
+
user_id?: string | undefined;
|
|
25
|
+
agent_id?: string | undefined;
|
|
26
|
+
project_id?: string | undefined;
|
|
27
|
+
expires_in_hours?: number | undefined;
|
|
28
|
+
meta?: Record<string, unknown> | undefined;
|
|
29
|
+
}>;
|
|
30
|
+
export type MemorySessionStoreInputType = z.infer<typeof MemorySessionStoreInput>;
|
|
31
|
+
/** 存储会话级记忆 */
|
|
32
|
+
export declare function memorySessionStoreTool(input: MemorySessionStoreInputType): Promise<{
|
|
33
|
+
id: string;
|
|
34
|
+
key: string;
|
|
35
|
+
scope: string;
|
|
36
|
+
}>;
|
|
37
|
+
/** 获取会话级记忆 */
|
|
38
|
+
export declare const MemorySessionGetInput: z.ZodObject<{
|
|
39
|
+
key: z.ZodString;
|
|
40
|
+
user_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
41
|
+
agent_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
42
|
+
project_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
43
|
+
scope: z.ZodDefault<z.ZodEnum<["session", "short", "long", "archival"]>>;
|
|
44
|
+
}, "strip", z.ZodTypeAny, {
|
|
45
|
+
key: string;
|
|
46
|
+
scope: "session" | "short" | "long" | "archival";
|
|
47
|
+
user_id: string;
|
|
48
|
+
agent_id: string;
|
|
49
|
+
project_id: string;
|
|
50
|
+
}, {
|
|
51
|
+
key: string;
|
|
52
|
+
scope?: "session" | "short" | "long" | "archival" | undefined;
|
|
53
|
+
user_id?: string | undefined;
|
|
54
|
+
agent_id?: string | undefined;
|
|
55
|
+
project_id?: string | undefined;
|
|
56
|
+
}>;
|
|
57
|
+
export type MemorySessionGetInputType = z.infer<typeof MemorySessionGetInput>;
|
|
58
|
+
export declare function memorySessionGetTool(input: MemorySessionGetInputType): Promise<{
|
|
59
|
+
id: string;
|
|
60
|
+
key: string;
|
|
61
|
+
value: string | null;
|
|
62
|
+
scope: string;
|
|
63
|
+
} | null>;
|
|
64
|
+
/** 列出会话级记忆 */
|
|
65
|
+
export declare const MemorySessionListInput: z.ZodObject<{
|
|
66
|
+
user_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
67
|
+
agent_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
68
|
+
project_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
69
|
+
scope: z.ZodOptional<z.ZodEnum<["session", "short", "long", "archival"]>>;
|
|
70
|
+
limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
|
|
71
|
+
}, "strip", z.ZodTypeAny, {
|
|
72
|
+
limit: number;
|
|
73
|
+
user_id: string;
|
|
74
|
+
agent_id: string;
|
|
75
|
+
project_id: string;
|
|
76
|
+
scope?: "session" | "short" | "long" | "archival" | undefined;
|
|
77
|
+
}, {
|
|
78
|
+
limit?: number | undefined;
|
|
79
|
+
scope?: "session" | "short" | "long" | "archival" | undefined;
|
|
80
|
+
user_id?: string | undefined;
|
|
81
|
+
agent_id?: string | undefined;
|
|
82
|
+
project_id?: string | undefined;
|
|
83
|
+
}>;
|
|
84
|
+
export type MemorySessionListInputType = z.infer<typeof MemorySessionListInput>;
|
|
85
|
+
export declare function memorySessionListTool(input: MemorySessionListInputType): Promise<{
|
|
86
|
+
sessions: Array<{
|
|
87
|
+
id: string;
|
|
88
|
+
key: string;
|
|
89
|
+
scope: string;
|
|
90
|
+
value: string | null;
|
|
91
|
+
updatedAt: string;
|
|
92
|
+
}>;
|
|
93
|
+
}>;
|
|
94
|
+
/** 删除会话级记忆 */
|
|
95
|
+
export declare const MemorySessionDeleteInput: z.ZodObject<{
|
|
96
|
+
key: z.ZodString;
|
|
97
|
+
user_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
98
|
+
agent_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
99
|
+
project_id: z.ZodDefault<z.ZodOptional<z.ZodString>>;
|
|
100
|
+
scope: z.ZodDefault<z.ZodEnum<["session", "short", "long", "archival"]>>;
|
|
101
|
+
}, "strip", z.ZodTypeAny, {
|
|
102
|
+
key: string;
|
|
103
|
+
scope: "session" | "short" | "long" | "archival";
|
|
104
|
+
user_id: string;
|
|
105
|
+
agent_id: string;
|
|
106
|
+
project_id: string;
|
|
107
|
+
}, {
|
|
108
|
+
key: string;
|
|
109
|
+
scope?: "session" | "short" | "long" | "archival" | undefined;
|
|
110
|
+
user_id?: string | undefined;
|
|
111
|
+
agent_id?: string | undefined;
|
|
112
|
+
project_id?: string | undefined;
|
|
113
|
+
}>;
|
|
114
|
+
export type MemorySessionDeleteInputType = z.infer<typeof MemorySessionDeleteInput>;
|
|
115
|
+
export declare function memorySessionDeleteTool(input: MemorySessionDeleteInputType): Promise<{
|
|
116
|
+
deleted: boolean;
|
|
117
|
+
}>;
|
|
118
|
+
//# sourceMappingURL=memory-session.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory-session.d.ts","sourceRoot":"","sources":["../../../src/mcp/tools/memory-session.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AASxB,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;;;;;;;;;;;;EAUlC,CAAC;AAEH,MAAM,MAAM,2BAA2B,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAElF,cAAc;AACd,wBAAsB,sBAAsB,CAAC,KAAK,EAAE,2BAA2B,GAAG,OAAO,CAAC;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC,CAqBpI;AAED,cAAc;AACd,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;EAMhC,CAAC;AAEH,MAAM,MAAM,yBAAyB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AAE9E,wBAAsB,oBAAoB,CAAC,KAAK,EAAE,yBAAyB,GAAG,OAAO,CAAC;IAAE,EAAE,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAAC,CAW7J;AAED,cAAc;AACd,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;EAMjC,CAAC;AAEH,MAAM,MAAM,0BAA0B,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAEhF,wBAAsB,qBAAqB,CAAC,KAAK,EAAE,0BAA0B,GAAG,OAAO,CAAC;IAAE,QAAQ,EAAE,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;CAAE,CAAC,CAkBhM;AAED,cAAc;AACd,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;;;;EAMnC,CAAC;AAEH,MAAM,MAAM,4BAA4B,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,CAAC;AAEpF,wBAAsB,uBAAuB,CAAC,KAAK,EAAE,4BAA4B,GAAG,OAAO,CAAC;IAAE,OAAO,EAAE,OAAO,CAAA;CAAE,CAAC,CAYhH"}
|