@blackms/aistack 1.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +244 -0
- package/dist/agents/definitions/analyst.d.ts +6 -0
- package/dist/agents/definitions/analyst.d.ts.map +1 -0
- package/dist/agents/definitions/analyst.js +53 -0
- package/dist/agents/definitions/analyst.js.map +1 -0
- package/dist/agents/definitions/architect.d.ts +6 -0
- package/dist/agents/definitions/architect.d.ts.map +1 -0
- package/dist/agents/definitions/architect.js +46 -0
- package/dist/agents/definitions/architect.js.map +1 -0
- package/dist/agents/definitions/coder.d.ts +6 -0
- package/dist/agents/definitions/coder.d.ts.map +1 -0
- package/dist/agents/definitions/coder.js +39 -0
- package/dist/agents/definitions/coder.js.map +1 -0
- package/dist/agents/definitions/coordinator.d.ts +6 -0
- package/dist/agents/definitions/coordinator.d.ts.map +1 -0
- package/dist/agents/definitions/coordinator.js +54 -0
- package/dist/agents/definitions/coordinator.js.map +1 -0
- package/dist/agents/definitions/index.d.ts +11 -0
- package/dist/agents/definitions/index.d.ts.map +1 -0
- package/dist/agents/definitions/index.js +11 -0
- package/dist/agents/definitions/index.js.map +1 -0
- package/dist/agents/definitions/researcher.d.ts +6 -0
- package/dist/agents/definitions/researcher.d.ts.map +1 -0
- package/dist/agents/definitions/researcher.js +39 -0
- package/dist/agents/definitions/researcher.js.map +1 -0
- package/dist/agents/definitions/reviewer.d.ts +6 -0
- package/dist/agents/definitions/reviewer.d.ts.map +1 -0
- package/dist/agents/definitions/reviewer.js +46 -0
- package/dist/agents/definitions/reviewer.js.map +1 -0
- package/dist/agents/definitions/tester.d.ts +6 -0
- package/dist/agents/definitions/tester.d.ts.map +1 -0
- package/dist/agents/definitions/tester.js +44 -0
- package/dist/agents/definitions/tester.js.map +1 -0
- package/dist/agents/index.d.ts +7 -0
- package/dist/agents/index.d.ts.map +1 -0
- package/dist/agents/index.js +7 -0
- package/dist/agents/index.js.map +1 -0
- package/dist/agents/registry.d.ts +41 -0
- package/dist/agents/registry.d.ts.map +1 -0
- package/dist/agents/registry.js +107 -0
- package/dist/agents/registry.js.map +1 -0
- package/dist/agents/spawner.d.ts +62 -0
- package/dist/agents/spawner.d.ts.map +1 -0
- package/dist/agents/spawner.js +169 -0
- package/dist/agents/spawner.js.map +1 -0
- package/dist/cli/commands/agent.d.ts +6 -0
- package/dist/cli/commands/agent.d.ts.map +1 -0
- package/dist/cli/commands/agent.js +144 -0
- package/dist/cli/commands/agent.js.map +1 -0
- package/dist/cli/commands/index.d.ts +10 -0
- package/dist/cli/commands/index.d.ts.map +1 -0
- package/dist/cli/commands/index.js +10 -0
- package/dist/cli/commands/index.js.map +1 -0
- package/dist/cli/commands/init.d.ts +6 -0
- package/dist/cli/commands/init.d.ts.map +1 -0
- package/dist/cli/commands/init.js +50 -0
- package/dist/cli/commands/init.js.map +1 -0
- package/dist/cli/commands/mcp.d.ts +6 -0
- package/dist/cli/commands/mcp.d.ts.map +1 -0
- package/dist/cli/commands/mcp.js +82 -0
- package/dist/cli/commands/mcp.js.map +1 -0
- package/dist/cli/commands/memory.d.ts +6 -0
- package/dist/cli/commands/memory.d.ts.map +1 -0
- package/dist/cli/commands/memory.js +157 -0
- package/dist/cli/commands/memory.js.map +1 -0
- package/dist/cli/commands/plugin.d.ts +6 -0
- package/dist/cli/commands/plugin.d.ts.map +1 -0
- package/dist/cli/commands/plugin.js +146 -0
- package/dist/cli/commands/plugin.js.map +1 -0
- package/dist/cli/commands/status.d.ts +6 -0
- package/dist/cli/commands/status.d.ts.map +1 -0
- package/dist/cli/commands/status.js +115 -0
- package/dist/cli/commands/status.js.map +1 -0
- package/dist/cli/index.d.ts +6 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +41 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/coordination/index.d.ts +7 -0
- package/dist/coordination/index.d.ts.map +1 -0
- package/dist/coordination/index.js +7 -0
- package/dist/coordination/index.js.map +1 -0
- package/dist/coordination/message-bus.d.ts +64 -0
- package/dist/coordination/message-bus.d.ts.map +1 -0
- package/dist/coordination/message-bus.js +148 -0
- package/dist/coordination/message-bus.js.map +1 -0
- package/dist/coordination/task-queue.d.ts +71 -0
- package/dist/coordination/task-queue.d.ts.map +1 -0
- package/dist/coordination/task-queue.js +140 -0
- package/dist/coordination/task-queue.js.map +1 -0
- package/dist/coordination/topology.d.ts +68 -0
- package/dist/coordination/topology.d.ts.map +1 -0
- package/dist/coordination/topology.js +195 -0
- package/dist/coordination/topology.js.map +1 -0
- package/dist/github/client.d.ts +56 -0
- package/dist/github/client.d.ts.map +1 -0
- package/dist/github/client.js +179 -0
- package/dist/github/client.js.map +1 -0
- package/dist/github/index.d.ts +5 -0
- package/dist/github/index.d.ts.map +1 -0
- package/dist/github/index.js +5 -0
- package/dist/github/index.js.map +1 -0
- package/dist/hooks/index.d.ts +29 -0
- package/dist/hooks/index.d.ts.map +1 -0
- package/dist/hooks/index.js +112 -0
- package/dist/hooks/index.js.map +1 -0
- package/dist/hooks/session.d.ts +16 -0
- package/dist/hooks/session.d.ts.map +1 -0
- package/dist/hooks/session.js +60 -0
- package/dist/hooks/session.js.map +1 -0
- package/dist/hooks/task.d.ts +16 -0
- package/dist/hooks/task.d.ts.map +1 -0
- package/dist/hooks/task.js +82 -0
- package/dist/hooks/task.js.map +1 -0
- package/dist/index.d.ts +16 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +32 -0
- package/dist/index.js.map +1 -0
- package/dist/mcp/index.d.ts +6 -0
- package/dist/mcp/index.d.ts.map +1 -0
- package/dist/mcp/index.js +6 -0
- package/dist/mcp/index.js.map +1 -0
- package/dist/mcp/server.d.ts +22 -0
- package/dist/mcp/server.d.ts.map +1 -0
- package/dist/mcp/server.js +128 -0
- package/dist/mcp/server.js.map +1 -0
- package/dist/mcp/tools/agent-tools.d.ts +174 -0
- package/dist/mcp/tools/agent-tools.d.ts.map +1 -0
- package/dist/mcp/tools/agent-tools.js +214 -0
- package/dist/mcp/tools/agent-tools.js.map +1 -0
- package/dist/mcp/tools/github-tools.d.ts +229 -0
- package/dist/mcp/tools/github-tools.d.ts.map +1 -0
- package/dist/mcp/tools/github-tools.js +286 -0
- package/dist/mcp/tools/github-tools.js.map +1 -0
- package/dist/mcp/tools/index.d.ts +10 -0
- package/dist/mcp/tools/index.d.ts.map +1 -0
- package/dist/mcp/tools/index.js +10 -0
- package/dist/mcp/tools/index.js.map +1 -0
- package/dist/mcp/tools/memory-tools.d.ts +190 -0
- package/dist/mcp/tools/memory-tools.d.ts.map +1 -0
- package/dist/mcp/tools/memory-tools.js +207 -0
- package/dist/mcp/tools/memory-tools.js.map +1 -0
- package/dist/mcp/tools/session-tools.d.ts +116 -0
- package/dist/mcp/tools/session-tools.d.ts.map +1 -0
- package/dist/mcp/tools/session-tools.js +151 -0
- package/dist/mcp/tools/session-tools.js.map +1 -0
- package/dist/mcp/tools/system-tools.d.ts +98 -0
- package/dist/mcp/tools/system-tools.d.ts.map +1 -0
- package/dist/mcp/tools/system-tools.js +165 -0
- package/dist/mcp/tools/system-tools.js.map +1 -0
- package/dist/mcp/tools/task-tools.d.ts +170 -0
- package/dist/mcp/tools/task-tools.d.ts.map +1 -0
- package/dist/mcp/tools/task-tools.js +188 -0
- package/dist/mcp/tools/task-tools.js.map +1 -0
- package/dist/memory/fts-search.d.ts +46 -0
- package/dist/memory/fts-search.d.ts.map +1 -0
- package/dist/memory/fts-search.js +136 -0
- package/dist/memory/fts-search.js.map +1 -0
- package/dist/memory/index.d.ts +77 -0
- package/dist/memory/index.d.ts.map +1 -0
- package/dist/memory/index.js +203 -0
- package/dist/memory/index.js.map +1 -0
- package/dist/memory/sqlite-store.d.ts +36 -0
- package/dist/memory/sqlite-store.d.ts.map +1 -0
- package/dist/memory/sqlite-store.js +371 -0
- package/dist/memory/sqlite-store.js.map +1 -0
- package/dist/memory/vector-search.d.ts +43 -0
- package/dist/memory/vector-search.d.ts.map +1 -0
- package/dist/memory/vector-search.js +179 -0
- package/dist/memory/vector-search.js.map +1 -0
- package/dist/plugins/index.d.ts +7 -0
- package/dist/plugins/index.d.ts.map +1 -0
- package/dist/plugins/index.js +7 -0
- package/dist/plugins/index.js.map +1 -0
- package/dist/plugins/loader.d.ts +33 -0
- package/dist/plugins/loader.d.ts.map +1 -0
- package/dist/plugins/loader.js +147 -0
- package/dist/plugins/loader.js.map +1 -0
- package/dist/plugins/registry.d.ts +48 -0
- package/dist/plugins/registry.d.ts.map +1 -0
- package/dist/plugins/registry.js +72 -0
- package/dist/plugins/registry.js.map +1 -0
- package/dist/plugins/types.d.ts +5 -0
- package/dist/plugins/types.d.ts.map +1 -0
- package/dist/plugins/types.js +5 -0
- package/dist/plugins/types.js.map +1 -0
- package/dist/providers/index.d.ts +49 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +254 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/types.d.ts +189 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +10 -0
- package/dist/types.js.map +1 -0
- package/dist/utils/config.d.ts +32 -0
- package/dist/utils/config.d.ts.map +1 -0
- package/dist/utils/config.js +194 -0
- package/dist/utils/config.js.map +1 -0
- package/dist/utils/embeddings.d.ts +23 -0
- package/dist/utils/embeddings.d.ts.map +1 -0
- package/dist/utils/embeddings.js +151 -0
- package/dist/utils/embeddings.js.map +1 -0
- package/dist/utils/index.d.ts +8 -0
- package/dist/utils/index.d.ts.map +1 -0
- package/dist/utils/index.js +8 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/utils/logger.d.ts +24 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +96 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/utils/validation.d.ts +264 -0
- package/dist/utils/validation.d.ts.map +1 -0
- package/dist/utils/validation.js +144 -0
- package/dist/utils/validation.js.map +1 -0
- package/package.json +84 -0
|
@@ -0,0 +1,165 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* System MCP tools - status, health, config
|
|
3
|
+
*/
|
|
4
|
+
import { getAgentCount as getActiveAgentCount } from '../../agents/spawner.js';
|
|
5
|
+
import { getAgentCount as getRegisteredAgentCount } from '../../agents/registry.js';
|
|
6
|
+
export function createSystemTools(memory, config) {
|
|
7
|
+
return {
|
|
8
|
+
system_status: {
|
|
9
|
+
name: 'system_status',
|
|
10
|
+
description: 'Get overall system status',
|
|
11
|
+
inputSchema: {
|
|
12
|
+
type: 'object',
|
|
13
|
+
properties: {},
|
|
14
|
+
},
|
|
15
|
+
handler: async () => {
|
|
16
|
+
const session = memory.getActiveSession();
|
|
17
|
+
const vectorStats = memory.getVectorStats();
|
|
18
|
+
const registeredAgents = getRegisteredAgentCount();
|
|
19
|
+
const activeAgents = getActiveAgentCount();
|
|
20
|
+
return {
|
|
21
|
+
version: config.version,
|
|
22
|
+
session: session
|
|
23
|
+
? {
|
|
24
|
+
id: session.id,
|
|
25
|
+
status: session.status,
|
|
26
|
+
startedAt: session.startedAt.toISOString(),
|
|
27
|
+
}
|
|
28
|
+
: null,
|
|
29
|
+
agents: {
|
|
30
|
+
registered: registeredAgents,
|
|
31
|
+
active: activeAgents,
|
|
32
|
+
},
|
|
33
|
+
memory: {
|
|
34
|
+
total: vectorStats.total,
|
|
35
|
+
indexed: vectorStats.indexed,
|
|
36
|
+
coverage: `${vectorStats.coverage}%`,
|
|
37
|
+
vectorEnabled: vectorStats.indexed > 0 || config.memory.vectorSearch.enabled,
|
|
38
|
+
},
|
|
39
|
+
providers: {
|
|
40
|
+
default: config.providers.default,
|
|
41
|
+
available: Object.keys(config.providers).filter(k => k !== 'default'),
|
|
42
|
+
},
|
|
43
|
+
};
|
|
44
|
+
},
|
|
45
|
+
},
|
|
46
|
+
system_health: {
|
|
47
|
+
name: 'system_health',
|
|
48
|
+
description: 'Check system health',
|
|
49
|
+
inputSchema: {
|
|
50
|
+
type: 'object',
|
|
51
|
+
properties: {},
|
|
52
|
+
},
|
|
53
|
+
handler: async () => {
|
|
54
|
+
const checks = {};
|
|
55
|
+
// Check memory
|
|
56
|
+
try {
|
|
57
|
+
const count = memory.count();
|
|
58
|
+
checks['memory'] = {
|
|
59
|
+
status: 'ok',
|
|
60
|
+
message: `${count} entries stored`,
|
|
61
|
+
};
|
|
62
|
+
}
|
|
63
|
+
catch (error) {
|
|
64
|
+
checks['memory'] = {
|
|
65
|
+
status: 'error',
|
|
66
|
+
message: error instanceof Error ? error.message : 'Memory check failed',
|
|
67
|
+
};
|
|
68
|
+
}
|
|
69
|
+
// Check vector search
|
|
70
|
+
try {
|
|
71
|
+
const stats = memory.getVectorStats();
|
|
72
|
+
if (config.memory.vectorSearch.enabled) {
|
|
73
|
+
if (stats.indexed > 0) {
|
|
74
|
+
checks['vector_search'] = {
|
|
75
|
+
status: 'ok',
|
|
76
|
+
message: `${stats.indexed}/${stats.total} entries indexed`,
|
|
77
|
+
};
|
|
78
|
+
}
|
|
79
|
+
else {
|
|
80
|
+
checks['vector_search'] = {
|
|
81
|
+
status: 'warn',
|
|
82
|
+
message: 'Enabled but no entries indexed',
|
|
83
|
+
};
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
else {
|
|
87
|
+
checks['vector_search'] = {
|
|
88
|
+
status: 'ok',
|
|
89
|
+
message: 'Disabled (using FTS only)',
|
|
90
|
+
};
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
catch (error) {
|
|
94
|
+
checks['vector_search'] = {
|
|
95
|
+
status: 'error',
|
|
96
|
+
message: error instanceof Error ? error.message : 'Vector check failed',
|
|
97
|
+
};
|
|
98
|
+
}
|
|
99
|
+
// Check agents
|
|
100
|
+
try {
|
|
101
|
+
const registered = getRegisteredAgentCount();
|
|
102
|
+
const active = getActiveAgentCount();
|
|
103
|
+
checks['agents'] = {
|
|
104
|
+
status: 'ok',
|
|
105
|
+
message: `${registered.total} types, ${active} active`,
|
|
106
|
+
};
|
|
107
|
+
}
|
|
108
|
+
catch (error) {
|
|
109
|
+
checks['agents'] = {
|
|
110
|
+
status: 'error',
|
|
111
|
+
message: error instanceof Error ? error.message : 'Agent check failed',
|
|
112
|
+
};
|
|
113
|
+
}
|
|
114
|
+
// Overall status
|
|
115
|
+
const hasError = Object.values(checks).some(c => c.status === 'error');
|
|
116
|
+
const hasWarn = Object.values(checks).some(c => c.status === 'warn');
|
|
117
|
+
return {
|
|
118
|
+
healthy: !hasError,
|
|
119
|
+
status: hasError ? 'unhealthy' : hasWarn ? 'degraded' : 'healthy',
|
|
120
|
+
checks,
|
|
121
|
+
};
|
|
122
|
+
},
|
|
123
|
+
},
|
|
124
|
+
system_config: {
|
|
125
|
+
name: 'system_config',
|
|
126
|
+
description: 'Get current configuration (sanitized)',
|
|
127
|
+
inputSchema: {
|
|
128
|
+
type: 'object',
|
|
129
|
+
properties: {},
|
|
130
|
+
},
|
|
131
|
+
handler: async () => {
|
|
132
|
+
// Return sanitized config (no API keys)
|
|
133
|
+
return {
|
|
134
|
+
version: config.version,
|
|
135
|
+
memory: {
|
|
136
|
+
path: config.memory.path,
|
|
137
|
+
defaultNamespace: config.memory.defaultNamespace,
|
|
138
|
+
vectorSearch: {
|
|
139
|
+
enabled: config.memory.vectorSearch.enabled,
|
|
140
|
+
provider: config.memory.vectorSearch.provider,
|
|
141
|
+
model: config.memory.vectorSearch.model,
|
|
142
|
+
},
|
|
143
|
+
},
|
|
144
|
+
agents: config.agents,
|
|
145
|
+
providers: {
|
|
146
|
+
default: config.providers.default,
|
|
147
|
+
anthropic: config.providers.anthropic ? { configured: true } : undefined,
|
|
148
|
+
openai: config.providers.openai ? { configured: true } : undefined,
|
|
149
|
+
ollama: config.providers.ollama
|
|
150
|
+
? { baseUrl: config.providers.ollama.baseUrl }
|
|
151
|
+
: undefined,
|
|
152
|
+
},
|
|
153
|
+
github: {
|
|
154
|
+
enabled: config.github.enabled,
|
|
155
|
+
useGhCli: config.github.useGhCli,
|
|
156
|
+
},
|
|
157
|
+
plugins: config.plugins,
|
|
158
|
+
mcp: config.mcp,
|
|
159
|
+
hooks: config.hooks,
|
|
160
|
+
};
|
|
161
|
+
},
|
|
162
|
+
},
|
|
163
|
+
};
|
|
164
|
+
}
|
|
165
|
+
//# sourceMappingURL=system-tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"system-tools.js","sourceRoot":"","sources":["../../../src/mcp/tools/system-tools.ts"],"names":[],"mappings":"AAAA;;GAEG;AAIH,OAAO,EAAE,aAAa,IAAI,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAC/E,OAAO,EAAE,aAAa,IAAI,uBAAuB,EAAE,MAAM,0BAA0B,CAAC;AAEpF,MAAM,UAAU,iBAAiB,CAAC,MAAqB,EAAE,MAAwB;IAC/E,OAAO;QACL,aAAa,EAAE;YACb,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,2BAA2B;YACxC,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;aACf;YACD,OAAO,EAAE,KAAK,IAAI,EAAE;gBAClB,MAAM,OAAO,GAAG,MAAM,CAAC,gBAAgB,EAAE,CAAC;gBAC1C,MAAM,WAAW,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;gBAC5C,MAAM,gBAAgB,GAAG,uBAAuB,EAAE,CAAC;gBACnD,MAAM,YAAY,GAAG,mBAAmB,EAAE,CAAC;gBAE3C,OAAO;oBACL,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,OAAO,EAAE,OAAO;wBACd,CAAC,CAAC;4BACE,EAAE,EAAE,OAAO,CAAC,EAAE;4BACd,MAAM,EAAE,OAAO,CAAC,MAAM;4BACtB,SAAS,EAAE,OAAO,CAAC,SAAS,CAAC,WAAW,EAAE;yBAC3C;wBACH,CAAC,CAAC,IAAI;oBACR,MAAM,EAAE;wBACN,UAAU,EAAE,gBAAgB;wBAC5B,MAAM,EAAE,YAAY;qBACrB;oBACD,MAAM,EAAE;wBACN,KAAK,EAAE,WAAW,CAAC,KAAK;wBACxB,OAAO,EAAE,WAAW,CAAC,OAAO;wBAC5B,QAAQ,EAAE,GAAG,WAAW,CAAC,QAAQ,GAAG;wBACpC,aAAa,EAAE,WAAW,CAAC,OAAO,GAAG,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO;qBAC7E;oBACD,SAAS,EAAE;wBACT,OAAO,EAAE,MAAM,CAAC,SAAS,CAAC,OAAO;wBACjC,SAAS,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,SAAS,CAAC;qBACtE;iBACF,CAAC;YACJ,CAAC;SACF;QAED,aAAa,EAAE;YACb,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,qBAAqB;YAClC,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;aACf;YACD,OAAO,EAAE,KAAK,IAAI,EAAE;gBAClB,MAAM,MAAM,GAAyE,EAAE,CAAC;gBAExF,eAAe;gBACf,IAAI,CAAC;oBACH,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;oBAC7B,MAAM,CAAC,QAAQ,CAAC,GAAG;wBACjB,MAAM,EAAE,IAAI;wBACZ,OAAO,EAAE,GAAG,KAAK,iBAAiB;qBACnC,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,QAAQ,CAAC,GAAG;wBACjB,MAAM,EAAE,OAAO;wBACf,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,qBAAqB;qBACxE,CAAC;gBACJ,CAAC;gBAED,sBAAsB;gBACtB,IAAI,CAAC;oBACH,MAAM,KAAK,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;oBACtC,IAAI,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;wBACvC,IAAI,KAAK,CAAC,OAAO,GAAG,CAAC,EAAE,CAAC;4BACtB,MAAM,CAAC,eAAe,CAAC,GAAG;gCACxB,MAAM,EAAE,IAAI;gCACZ,OAAO,EAAE,GAAG,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,KAAK,kBAAkB;6BAC3D,CAAC;wBACJ,CAAC;6BAAM,CAAC;4BACN,MAAM,CAAC,eAAe,CAAC,GAAG;gCACxB,MAAM,EAAE,MAAM;gCACd,OAAO,EAAE,gCAAgC;6BAC1C,CAAC;wBACJ,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,MAAM,CAAC,eAAe,CAAC,GAAG;4BACxB,MAAM,EAAE,IAAI;4BACZ,OAAO,EAAE,2BAA2B;yBACrC,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,eAAe,CAAC,GAAG;wBACxB,MAAM,EAAE,OAAO;wBACf,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,qBAAqB;qBACxE,CAAC;gBACJ,CAAC;gBAED,eAAe;gBACf,IAAI,CAAC;oBACH,MAAM,UAAU,GAAG,uBAAuB,EAAE,CAAC;oBAC7C,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;oBACrC,MAAM,CAAC,QAAQ,CAAC,GAAG;wBACjB,MAAM,EAAE,IAAI;wBACZ,OAAO,EAAE,GAAG,UAAU,CAAC,KAAK,WAAW,MAAM,SAAS;qBACvD,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,CAAC,QAAQ,CAAC,GAAG;wBACjB,MAAM,EAAE,OAAO;wBACf,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,oBAAoB;qBACvE,CAAC;gBACJ,CAAC;gBAED,iBAAiB;gBACjB,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC;gBACvE,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;gBAErE,OAAO;oBACL,OAAO,EAAE,CAAC,QAAQ;oBAClB,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;oBACjE,MAAM;iBACP,CAAC;YACJ,CAAC;SACF;QAED,aAAa,EAAE;YACb,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,uCAAuC;YACpD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;aACf;YACD,OAAO,EAAE,KAAK,IAAI,EAAE;gBAClB,wCAAwC;gBACxC,OAAO;oBACL,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,MAAM,EAAE;wBACN,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI;wBACxB,gBAAgB,EAAE,MAAM,CAAC,MAAM,CAAC,gBAAgB;wBAChD,YAAY,EAAE;4BACZ,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,OAAO;4BAC3C,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ;4BAC7C,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK;yBACxC;qBACF;oBACD,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,SAAS,EAAE;wBACT,OAAO,EAAE,MAAM,CAAC,SAAS,CAAC,OAAO;wBACjC,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS;wBACxE,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS;wBAClE,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,MAAM;4BAC7B,CAAC,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,OAAO,EAAE;4BAC9C,CAAC,CAAC,SAAS;qBACd;oBACD,MAAM,EAAE;wBACN,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO;wBAC9B,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,QAAQ;qBACjC;oBACD,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,GAAG,EAAE,MAAM,CAAC,GAAG;oBACf,KAAK,EAAE,MAAM,CAAC,KAAK;iBACpB,CAAC;YACJ,CAAC;SACF;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,170 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Task MCP tools - create, assign, complete, list
|
|
3
|
+
*/
|
|
4
|
+
import type { MemoryManager } from '../../memory/index.js';
|
|
5
|
+
export declare function createTaskTools(memory: MemoryManager): {
|
|
6
|
+
task_create: {
|
|
7
|
+
name: string;
|
|
8
|
+
description: string;
|
|
9
|
+
inputSchema: {
|
|
10
|
+
type: string;
|
|
11
|
+
properties: {
|
|
12
|
+
agentType: {
|
|
13
|
+
type: string;
|
|
14
|
+
description: string;
|
|
15
|
+
};
|
|
16
|
+
input: {
|
|
17
|
+
type: string;
|
|
18
|
+
description: string;
|
|
19
|
+
};
|
|
20
|
+
sessionId: {
|
|
21
|
+
type: string;
|
|
22
|
+
description: string;
|
|
23
|
+
};
|
|
24
|
+
};
|
|
25
|
+
required: string[];
|
|
26
|
+
};
|
|
27
|
+
handler: (params: Record<string, unknown>) => Promise<{
|
|
28
|
+
success: boolean;
|
|
29
|
+
task: {
|
|
30
|
+
id: string;
|
|
31
|
+
agentType: string;
|
|
32
|
+
status: import("../../types.js").TaskStatus;
|
|
33
|
+
createdAt: string;
|
|
34
|
+
sessionId: string | undefined;
|
|
35
|
+
};
|
|
36
|
+
error?: undefined;
|
|
37
|
+
} | {
|
|
38
|
+
success: boolean;
|
|
39
|
+
error: string;
|
|
40
|
+
task?: undefined;
|
|
41
|
+
}>;
|
|
42
|
+
};
|
|
43
|
+
task_assign: {
|
|
44
|
+
name: string;
|
|
45
|
+
description: string;
|
|
46
|
+
inputSchema: {
|
|
47
|
+
type: string;
|
|
48
|
+
properties: {
|
|
49
|
+
taskId: {
|
|
50
|
+
type: string;
|
|
51
|
+
description: string;
|
|
52
|
+
};
|
|
53
|
+
agentId: {
|
|
54
|
+
type: string;
|
|
55
|
+
description: string;
|
|
56
|
+
};
|
|
57
|
+
};
|
|
58
|
+
required: string[];
|
|
59
|
+
};
|
|
60
|
+
handler: (params: Record<string, unknown>) => Promise<{
|
|
61
|
+
success: boolean;
|
|
62
|
+
error: string;
|
|
63
|
+
message?: undefined;
|
|
64
|
+
} | {
|
|
65
|
+
success: boolean;
|
|
66
|
+
message: string;
|
|
67
|
+
error?: undefined;
|
|
68
|
+
}>;
|
|
69
|
+
};
|
|
70
|
+
task_complete: {
|
|
71
|
+
name: string;
|
|
72
|
+
description: string;
|
|
73
|
+
inputSchema: {
|
|
74
|
+
type: string;
|
|
75
|
+
properties: {
|
|
76
|
+
taskId: {
|
|
77
|
+
type: string;
|
|
78
|
+
description: string;
|
|
79
|
+
};
|
|
80
|
+
output: {
|
|
81
|
+
type: string;
|
|
82
|
+
description: string;
|
|
83
|
+
};
|
|
84
|
+
status: {
|
|
85
|
+
type: string;
|
|
86
|
+
enum: string[];
|
|
87
|
+
description: string;
|
|
88
|
+
};
|
|
89
|
+
};
|
|
90
|
+
required: string[];
|
|
91
|
+
};
|
|
92
|
+
handler: (params: Record<string, unknown>) => Promise<{
|
|
93
|
+
success: boolean;
|
|
94
|
+
error: string;
|
|
95
|
+
task?: undefined;
|
|
96
|
+
} | {
|
|
97
|
+
success: boolean;
|
|
98
|
+
task: {
|
|
99
|
+
id: string;
|
|
100
|
+
status: "completed" | "failed";
|
|
101
|
+
hasOutput: boolean;
|
|
102
|
+
};
|
|
103
|
+
error?: undefined;
|
|
104
|
+
}>;
|
|
105
|
+
};
|
|
106
|
+
task_list: {
|
|
107
|
+
name: string;
|
|
108
|
+
description: string;
|
|
109
|
+
inputSchema: {
|
|
110
|
+
type: string;
|
|
111
|
+
properties: {
|
|
112
|
+
sessionId: {
|
|
113
|
+
type: string;
|
|
114
|
+
description: string;
|
|
115
|
+
};
|
|
116
|
+
status: {
|
|
117
|
+
type: string;
|
|
118
|
+
enum: string[];
|
|
119
|
+
description: string;
|
|
120
|
+
};
|
|
121
|
+
};
|
|
122
|
+
};
|
|
123
|
+
handler: (params: Record<string, unknown>) => Promise<{
|
|
124
|
+
count: number;
|
|
125
|
+
tasks: {
|
|
126
|
+
id: string;
|
|
127
|
+
agentType: string;
|
|
128
|
+
status: import("../../types.js").TaskStatus;
|
|
129
|
+
hasInput: boolean;
|
|
130
|
+
hasOutput: boolean;
|
|
131
|
+
createdAt: string;
|
|
132
|
+
completedAt: string | undefined;
|
|
133
|
+
sessionId: string | undefined;
|
|
134
|
+
}[];
|
|
135
|
+
}>;
|
|
136
|
+
};
|
|
137
|
+
task_get: {
|
|
138
|
+
name: string;
|
|
139
|
+
description: string;
|
|
140
|
+
inputSchema: {
|
|
141
|
+
type: string;
|
|
142
|
+
properties: {
|
|
143
|
+
taskId: {
|
|
144
|
+
type: string;
|
|
145
|
+
description: string;
|
|
146
|
+
};
|
|
147
|
+
};
|
|
148
|
+
required: string[];
|
|
149
|
+
};
|
|
150
|
+
handler: (params: Record<string, unknown>) => Promise<{
|
|
151
|
+
found: boolean;
|
|
152
|
+
message: string;
|
|
153
|
+
task?: undefined;
|
|
154
|
+
} | {
|
|
155
|
+
found: boolean;
|
|
156
|
+
task: {
|
|
157
|
+
id: string;
|
|
158
|
+
agentType: string;
|
|
159
|
+
status: import("../../types.js").TaskStatus;
|
|
160
|
+
input: string | undefined;
|
|
161
|
+
output: string | undefined;
|
|
162
|
+
createdAt: string;
|
|
163
|
+
completedAt: string | undefined;
|
|
164
|
+
sessionId: string | undefined;
|
|
165
|
+
};
|
|
166
|
+
message?: undefined;
|
|
167
|
+
}>;
|
|
168
|
+
};
|
|
169
|
+
};
|
|
170
|
+
//# sourceMappingURL=task-tools.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"task-tools.d.ts","sourceRoot":"","sources":["../../../src/mcp/tools/task-tools.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AA6B3D,wBAAgB,eAAe,CAAC,MAAM,EAAE,aAAa;;;;;;;;;;;;;;;;;;;;;;0BAcvB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;0BAwCvB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;0BAgCvB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;0BAmCvB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;0BA8BvB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;;;;;;;;;;;;;;;;;;;EA2BpD"}
|
|
@@ -0,0 +1,188 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Task MCP tools - create, assign, complete, list
|
|
3
|
+
*/
|
|
4
|
+
import { z } from 'zod';
|
|
5
|
+
// Input schemas
|
|
6
|
+
const CreateInputSchema = z.object({
|
|
7
|
+
agentType: z.string().min(1).describe('Agent type for this task'),
|
|
8
|
+
input: z.string().optional().describe('Task input/description'),
|
|
9
|
+
sessionId: z.string().uuid().optional().describe('Session to associate with'),
|
|
10
|
+
});
|
|
11
|
+
const AssignInputSchema = z.object({
|
|
12
|
+
taskId: z.string().uuid().describe('Task ID'),
|
|
13
|
+
agentId: z.string().uuid().describe('Agent ID to assign'),
|
|
14
|
+
});
|
|
15
|
+
const CompleteInputSchema = z.object({
|
|
16
|
+
taskId: z.string().uuid().describe('Task ID'),
|
|
17
|
+
output: z.string().optional().describe('Task output'),
|
|
18
|
+
status: z.enum(['completed', 'failed']).optional().describe('Completion status'),
|
|
19
|
+
});
|
|
20
|
+
const ListInputSchema = z.object({
|
|
21
|
+
sessionId: z.string().uuid().optional().describe('Filter by session'),
|
|
22
|
+
status: z.enum(['pending', 'running', 'completed', 'failed']).optional().describe('Filter by status'),
|
|
23
|
+
});
|
|
24
|
+
const GetInputSchema = z.object({
|
|
25
|
+
taskId: z.string().uuid().describe('Task ID'),
|
|
26
|
+
});
|
|
27
|
+
export function createTaskTools(memory) {
|
|
28
|
+
return {
|
|
29
|
+
task_create: {
|
|
30
|
+
name: 'task_create',
|
|
31
|
+
description: 'Create a new task',
|
|
32
|
+
inputSchema: {
|
|
33
|
+
type: 'object',
|
|
34
|
+
properties: {
|
|
35
|
+
agentType: { type: 'string', description: 'Agent type for this task' },
|
|
36
|
+
input: { type: 'string', description: 'Task input/description' },
|
|
37
|
+
sessionId: { type: 'string', description: 'Session to associate with' },
|
|
38
|
+
},
|
|
39
|
+
required: ['agentType'],
|
|
40
|
+
},
|
|
41
|
+
handler: async (params) => {
|
|
42
|
+
const input = CreateInputSchema.parse(params);
|
|
43
|
+
try {
|
|
44
|
+
const task = memory.createTask(input.agentType, input.input, input.sessionId);
|
|
45
|
+
return {
|
|
46
|
+
success: true,
|
|
47
|
+
task: {
|
|
48
|
+
id: task.id,
|
|
49
|
+
agentType: task.agentType,
|
|
50
|
+
status: task.status,
|
|
51
|
+
createdAt: task.createdAt.toISOString(),
|
|
52
|
+
sessionId: task.sessionId,
|
|
53
|
+
},
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
catch (error) {
|
|
57
|
+
return {
|
|
58
|
+
success: false,
|
|
59
|
+
error: error instanceof Error ? error.message : String(error),
|
|
60
|
+
};
|
|
61
|
+
}
|
|
62
|
+
},
|
|
63
|
+
},
|
|
64
|
+
task_assign: {
|
|
65
|
+
name: 'task_assign',
|
|
66
|
+
description: 'Assign a task to an agent (marks as running)',
|
|
67
|
+
inputSchema: {
|
|
68
|
+
type: 'object',
|
|
69
|
+
properties: {
|
|
70
|
+
taskId: { type: 'string', description: 'Task ID' },
|
|
71
|
+
agentId: { type: 'string', description: 'Agent ID to assign' },
|
|
72
|
+
},
|
|
73
|
+
required: ['taskId', 'agentId'],
|
|
74
|
+
},
|
|
75
|
+
handler: async (params) => {
|
|
76
|
+
const input = AssignInputSchema.parse(params);
|
|
77
|
+
const task = memory.getTask(input.taskId);
|
|
78
|
+
if (!task) {
|
|
79
|
+
return {
|
|
80
|
+
success: false,
|
|
81
|
+
error: 'Task not found',
|
|
82
|
+
};
|
|
83
|
+
}
|
|
84
|
+
const updated = memory.updateTaskStatus(input.taskId, 'running');
|
|
85
|
+
return {
|
|
86
|
+
success: updated,
|
|
87
|
+
message: updated ? 'Task assigned and running' : 'Failed to update task',
|
|
88
|
+
};
|
|
89
|
+
},
|
|
90
|
+
},
|
|
91
|
+
task_complete: {
|
|
92
|
+
name: 'task_complete',
|
|
93
|
+
description: 'Mark a task as completed or failed',
|
|
94
|
+
inputSchema: {
|
|
95
|
+
type: 'object',
|
|
96
|
+
properties: {
|
|
97
|
+
taskId: { type: 'string', description: 'Task ID' },
|
|
98
|
+
output: { type: 'string', description: 'Task output' },
|
|
99
|
+
status: { type: 'string', enum: ['completed', 'failed'], description: 'Completion status' },
|
|
100
|
+
},
|
|
101
|
+
required: ['taskId'],
|
|
102
|
+
},
|
|
103
|
+
handler: async (params) => {
|
|
104
|
+
const input = CompleteInputSchema.parse(params);
|
|
105
|
+
const task = memory.getTask(input.taskId);
|
|
106
|
+
if (!task) {
|
|
107
|
+
return {
|
|
108
|
+
success: false,
|
|
109
|
+
error: 'Task not found',
|
|
110
|
+
};
|
|
111
|
+
}
|
|
112
|
+
const status = input.status ?? 'completed';
|
|
113
|
+
const updated = memory.updateTaskStatus(input.taskId, status, input.output);
|
|
114
|
+
return {
|
|
115
|
+
success: updated,
|
|
116
|
+
task: {
|
|
117
|
+
id: input.taskId,
|
|
118
|
+
status,
|
|
119
|
+
hasOutput: !!input.output,
|
|
120
|
+
},
|
|
121
|
+
};
|
|
122
|
+
},
|
|
123
|
+
},
|
|
124
|
+
task_list: {
|
|
125
|
+
name: 'task_list',
|
|
126
|
+
description: 'List tasks',
|
|
127
|
+
inputSchema: {
|
|
128
|
+
type: 'object',
|
|
129
|
+
properties: {
|
|
130
|
+
sessionId: { type: 'string', description: 'Filter by session' },
|
|
131
|
+
status: { type: 'string', enum: ['pending', 'running', 'completed', 'failed'], description: 'Filter by status' },
|
|
132
|
+
},
|
|
133
|
+
},
|
|
134
|
+
handler: async (params) => {
|
|
135
|
+
const input = ListInputSchema.parse(params);
|
|
136
|
+
const tasks = memory.listTasks(input.sessionId, input.status);
|
|
137
|
+
return {
|
|
138
|
+
count: tasks.length,
|
|
139
|
+
tasks: tasks.map(t => ({
|
|
140
|
+
id: t.id,
|
|
141
|
+
agentType: t.agentType,
|
|
142
|
+
status: t.status,
|
|
143
|
+
hasInput: !!t.input,
|
|
144
|
+
hasOutput: !!t.output,
|
|
145
|
+
createdAt: t.createdAt.toISOString(),
|
|
146
|
+
completedAt: t.completedAt?.toISOString(),
|
|
147
|
+
sessionId: t.sessionId,
|
|
148
|
+
})),
|
|
149
|
+
};
|
|
150
|
+
},
|
|
151
|
+
},
|
|
152
|
+
task_get: {
|
|
153
|
+
name: 'task_get',
|
|
154
|
+
description: 'Get a task by ID',
|
|
155
|
+
inputSchema: {
|
|
156
|
+
type: 'object',
|
|
157
|
+
properties: {
|
|
158
|
+
taskId: { type: 'string', description: 'Task ID' },
|
|
159
|
+
},
|
|
160
|
+
required: ['taskId'],
|
|
161
|
+
},
|
|
162
|
+
handler: async (params) => {
|
|
163
|
+
const input = GetInputSchema.parse(params);
|
|
164
|
+
const task = memory.getTask(input.taskId);
|
|
165
|
+
if (!task) {
|
|
166
|
+
return {
|
|
167
|
+
found: false,
|
|
168
|
+
message: 'Task not found',
|
|
169
|
+
};
|
|
170
|
+
}
|
|
171
|
+
return {
|
|
172
|
+
found: true,
|
|
173
|
+
task: {
|
|
174
|
+
id: task.id,
|
|
175
|
+
agentType: task.agentType,
|
|
176
|
+
status: task.status,
|
|
177
|
+
input: task.input,
|
|
178
|
+
output: task.output,
|
|
179
|
+
createdAt: task.createdAt.toISOString(),
|
|
180
|
+
completedAt: task.completedAt?.toISOString(),
|
|
181
|
+
sessionId: task.sessionId,
|
|
182
|
+
},
|
|
183
|
+
};
|
|
184
|
+
},
|
|
185
|
+
},
|
|
186
|
+
};
|
|
187
|
+
}
|
|
188
|
+
//# sourceMappingURL=task-tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"task-tools.js","sourceRoot":"","sources":["../../../src/mcp/tools/task-tools.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,gBAAgB;AAChB,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACjC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,0BAA0B,CAAC;IACjE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;IAC/D,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;CAC9E,CAAC,CAAC;AAEH,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACjC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;IAC7C,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC;CAC1D,CAAC,CAAC;AAEH,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;IAC7C,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC;IACrD,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;CACjF,CAAC,CAAC;AAEH,MAAM,eAAe,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/B,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;IACrE,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC;CACtG,CAAC,CAAC;AAEH,MAAM,cAAc,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC;CAC9C,CAAC,CAAC;AAEH,MAAM,UAAU,eAAe,CAAC,MAAqB;IACnD,OAAO;QACL,WAAW,EAAE;YACX,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,mBAAmB;YAChC,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,0BAA0B,EAAE;oBACtE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,wBAAwB,EAAE;oBAChE,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,2BAA2B,EAAE;iBACxE;gBACD,QAAQ,EAAE,CAAC,WAAW,CAAC;aACxB;YACD,OAAO,EAAE,KAAK,EAAE,MAA+B,EAAE,EAAE;gBACjD,MAAM,KAAK,GAAG,iBAAiB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAE9C,IAAI,CAAC;oBACH,MAAM,IAAI,GAAG,MAAM,CAAC,UAAU,CAC5B,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,KAAK,EACX,KAAK,CAAC,SAAS,CAChB,CAAC;oBAEF,OAAO;wBACL,OAAO,EAAE,IAAI;wBACb,IAAI,EAAE;4BACJ,EAAE,EAAE,IAAI,CAAC,EAAE;4BACX,SAAS,EAAE,IAAI,CAAC,SAAS;4BACzB,MAAM,EAAE,IAAI,CAAC,MAAM;4BACnB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE;4BACvC,SAAS,EAAE,IAAI,CAAC,SAAS;yBAC1B;qBACF,CAAC;gBACJ,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC9D,CAAC;gBACJ,CAAC;YACH,CAAC;SACF;QAED,WAAW,EAAE;YACX,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,8CAA8C;YAC3D,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE;oBAClD,OAAO,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,oBAAoB,EAAE;iBAC/D;gBACD,QAAQ,EAAE,CAAC,QAAQ,EAAE,SAAS,CAAC;aAChC;YACD,OAAO,EAAE,KAAK,EAAE,MAA+B,EAAE,EAAE;gBACjD,MAAM,KAAK,GAAG,iBAAiB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAE9C,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC1C,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,gBAAgB;qBACxB,CAAC;gBACJ,CAAC;gBAED,MAAM,OAAO,GAAG,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;gBAEjE,OAAO;oBACL,OAAO,EAAE,OAAO;oBAChB,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,uBAAuB;iBACzE,CAAC;YACJ,CAAC;SACF;QAED,aAAa,EAAE;YACb,IAAI,EAAE,eAAe;YACrB,WAAW,EAAE,oCAAoC;YACjD,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE;oBAClD,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,aAAa,EAAE;oBACtD,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,QAAQ,CAAC,EAAE,WAAW,EAAE,mBAAmB,EAAE;iBAC5F;gBACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;aACrB;YACD,OAAO,EAAE,KAAK,EAAE,MAA+B,EAAE,EAAE;gBACjD,MAAM,KAAK,GAAG,mBAAmB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAEhD,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC1C,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,gBAAgB;qBACxB,CAAC;gBACJ,CAAC;gBAED,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,IAAI,WAAW,CAAC;gBAC3C,MAAM,OAAO,GAAG,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;gBAE5E,OAAO;oBACL,OAAO,EAAE,OAAO;oBAChB,IAAI,EAAE;wBACJ,EAAE,EAAE,KAAK,CAAC,MAAM;wBAChB,MAAM;wBACN,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,MAAM;qBAC1B;iBACF,CAAC;YACJ,CAAC;SACF;QAED,SAAS,EAAE;YACT,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,YAAY;YACzB,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,mBAAmB,EAAE;oBAC/D,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,SAAS,EAAE,SAAS,EAAE,WAAW,EAAE,QAAQ,CAAC,EAAE,WAAW,EAAE,kBAAkB,EAAE;iBACjH;aACF;YACD,OAAO,EAAE,KAAK,EAAE,MAA+B,EAAE,EAAE;gBACjD,MAAM,KAAK,GAAG,eAAe,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC5C,MAAM,KAAK,GAAG,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;gBAE9D,OAAO;oBACL,KAAK,EAAE,KAAK,CAAC,MAAM;oBACnB,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;wBACrB,EAAE,EAAE,CAAC,CAAC,EAAE;wBACR,SAAS,EAAE,CAAC,CAAC,SAAS;wBACtB,MAAM,EAAE,CAAC,CAAC,MAAM;wBAChB,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK;wBACnB,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM;wBACrB,SAAS,EAAE,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE;wBACpC,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,WAAW,EAAE;wBACzC,SAAS,EAAE,CAAC,CAAC,SAAS;qBACvB,CAAC,CAAC;iBACJ,CAAC;YACJ,CAAC;SACF;QAED,QAAQ,EAAE;YACR,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,kBAAkB;YAC/B,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE;oBACV,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE;iBACnD;gBACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;aACrB;YACD,OAAO,EAAE,KAAK,EAAE,MAA+B,EAAE,EAAE;gBACjD,MAAM,KAAK,GAAG,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC3C,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAE1C,IAAI,CAAC,IAAI,EAAE,CAAC;oBACV,OAAO;wBACL,KAAK,EAAE,KAAK;wBACZ,OAAO,EAAE,gBAAgB;qBAC1B,CAAC;gBACJ,CAAC;gBAED,OAAO;oBACL,KAAK,EAAE,IAAI;oBACX,IAAI,EAAE;wBACJ,EAAE,EAAE,IAAI,CAAC,EAAE;wBACX,SAAS,EAAE,IAAI,CAAC,SAAS;wBACzB,MAAM,EAAE,IAAI,CAAC,MAAM;wBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;wBACjB,MAAM,EAAE,IAAI,CAAC,MAAM;wBACnB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE;wBACvC,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE;wBAC5C,SAAS,EAAE,IAAI,CAAC,SAAS;qBAC1B;iBACF,CAAC;YACJ,CAAC;SACF;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,46 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Full-text search using SQLite FTS5
|
|
3
|
+
*/
|
|
4
|
+
import type Database from 'better-sqlite3';
|
|
5
|
+
import type { MemorySearchResult } from '../types.js';
|
|
6
|
+
export interface FTSSearchOptions {
|
|
7
|
+
namespace?: string;
|
|
8
|
+
limit?: number;
|
|
9
|
+
highlightStart?: string;
|
|
10
|
+
highlightEnd?: string;
|
|
11
|
+
}
|
|
12
|
+
export declare class FTSSearch {
|
|
13
|
+
private db;
|
|
14
|
+
constructor(db: Database.Database);
|
|
15
|
+
/**
|
|
16
|
+
* Search memory using FTS5 full-text search
|
|
17
|
+
*/
|
|
18
|
+
search(query: string, options?: FTSSearchOptions): MemorySearchResult[];
|
|
19
|
+
/**
|
|
20
|
+
* Search with phrase matching (exact phrase)
|
|
21
|
+
*/
|
|
22
|
+
phraseSearch(phrase: string, options?: FTSSearchOptions): MemorySearchResult[];
|
|
23
|
+
/**
|
|
24
|
+
* Search with prefix matching (autocomplete)
|
|
25
|
+
*/
|
|
26
|
+
prefixSearch(prefix: string, options?: FTSSearchOptions): MemorySearchResult[];
|
|
27
|
+
/**
|
|
28
|
+
* Check if a term exists in the index
|
|
29
|
+
*/
|
|
30
|
+
termExists(term: string, namespace?: string): boolean;
|
|
31
|
+
/**
|
|
32
|
+
* Get suggestions based on existing content
|
|
33
|
+
*/
|
|
34
|
+
suggest(partial: string, limit?: number): string[];
|
|
35
|
+
/**
|
|
36
|
+
* Escape special FTS5 characters in query
|
|
37
|
+
*/
|
|
38
|
+
private escapeFTSQuery;
|
|
39
|
+
/**
|
|
40
|
+
* Normalize BM25 score to 0-1 range
|
|
41
|
+
* BM25 returns negative values where more negative = better match
|
|
42
|
+
*/
|
|
43
|
+
private normalizeScore;
|
|
44
|
+
private rowToEntry;
|
|
45
|
+
}
|
|
46
|
+
//# sourceMappingURL=fts-search.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fts-search.d.ts","sourceRoot":"","sources":["../../src/memory/fts-search.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAC3C,OAAO,KAAK,EAAe,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAEnE,MAAM,WAAW,gBAAgB;IAC/B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAeD,qBAAa,SAAS;IACpB,OAAO,CAAC,EAAE,CAAoB;gBAElB,EAAE,EAAE,QAAQ,CAAC,QAAQ;IAIjC;;OAEG;IACH,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,GAAE,gBAAqB,GAAG,kBAAkB,EAAE;IAyC3E;;OAEG;IACH,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,gBAAqB,GAAG,kBAAkB,EAAE;IAMlF;;OAEG;IACH,YAAY,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,gBAAqB,GAAG,kBAAkB,EAAE;IAUlF;;OAEG;IACH,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO;IAmBrD;;OAEG;IACH,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,GAAE,MAAU,GAAG,MAAM,EAAE;IAmBrD;;OAEG;IACH,OAAO,CAAC,cAAc;IAUtB;;;OAGG;IACH,OAAO,CAAC,cAAc;IAOtB,OAAO,CAAC,UAAU;CAcnB"}
|