@recursiv/mcp 0.1.1
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 +135 -0
- package/dist/__tests__/scopes.test.d.ts +2 -0
- package/dist/__tests__/scopes.test.d.ts.map +1 -0
- package/dist/__tests__/scopes.test.js +85 -0
- package/dist/__tests__/scopes.test.js.map +1 -0
- package/dist/index.d.ts +3 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +125 -0
- package/dist/index.js.map +1 -0
- package/dist/lib/approval-gate.d.ts +45 -0
- package/dist/lib/approval-gate.d.ts.map +1 -0
- package/dist/lib/approval-gate.js +143 -0
- package/dist/lib/approval-gate.js.map +1 -0
- package/dist/scopes.d.ts +99 -0
- package/dist/scopes.d.ts.map +1 -0
- package/dist/scopes.js +122 -0
- package/dist/scopes.js.map +1 -0
- package/dist/tools/__tests__/agents.test.d.ts +2 -0
- package/dist/tools/__tests__/agents.test.d.ts.map +1 -0
- package/dist/tools/__tests__/agents.test.js +248 -0
- package/dist/tools/__tests__/agents.test.js.map +1 -0
- package/dist/tools/__tests__/devtools.test.d.ts +2 -0
- package/dist/tools/__tests__/devtools.test.d.ts.map +1 -0
- package/dist/tools/__tests__/devtools.test.js +206 -0
- package/dist/tools/__tests__/devtools.test.js.map +1 -0
- package/dist/tools/__tests__/dispatcher.test.d.ts +2 -0
- package/dist/tools/__tests__/dispatcher.test.d.ts.map +1 -0
- package/dist/tools/__tests__/dispatcher.test.js +178 -0
- package/dist/tools/__tests__/dispatcher.test.js.map +1 -0
- package/dist/tools/__tests__/memory.test.d.ts +2 -0
- package/dist/tools/__tests__/memory.test.d.ts.map +1 -0
- package/dist/tools/__tests__/memory.test.js +151 -0
- package/dist/tools/__tests__/memory.test.js.map +1 -0
- package/dist/tools/__tests__/projects.test.d.ts +2 -0
- package/dist/tools/__tests__/projects.test.d.ts.map +1 -0
- package/dist/tools/__tests__/projects.test.js +168 -0
- package/dist/tools/__tests__/projects.test.js.map +1 -0
- package/dist/tools/__tests__/sandbox.test.d.ts +2 -0
- package/dist/tools/__tests__/sandbox.test.d.ts.map +1 -0
- package/dist/tools/__tests__/sandbox.test.js +113 -0
- package/dist/tools/__tests__/sandbox.test.js.map +1 -0
- package/dist/tools/__tests__/social.test.d.ts +2 -0
- package/dist/tools/__tests__/social.test.d.ts.map +1 -0
- package/dist/tools/__tests__/social.test.js +127 -0
- package/dist/tools/__tests__/social.test.js.map +1 -0
- package/dist/tools/__tests__/swarms.test.d.ts +2 -0
- package/dist/tools/__tests__/swarms.test.d.ts.map +1 -0
- package/dist/tools/__tests__/swarms.test.js +320 -0
- package/dist/tools/__tests__/swarms.test.js.map +1 -0
- package/dist/tools/__tests__/templates.test.d.ts +2 -0
- package/dist/tools/__tests__/templates.test.d.ts.map +1 -0
- package/dist/tools/__tests__/templates.test.js +176 -0
- package/dist/tools/__tests__/templates.test.js.map +1 -0
- package/dist/tools/agents.d.ts +4 -0
- package/dist/tools/agents.d.ts.map +1 -0
- package/dist/tools/agents.js +170 -0
- package/dist/tools/agents.js.map +1 -0
- package/dist/tools/devtools.d.ts +4 -0
- package/dist/tools/devtools.d.ts.map +1 -0
- package/dist/tools/devtools.js +511 -0
- package/dist/tools/devtools.js.map +1 -0
- package/dist/tools/dispatcher.d.ts +4 -0
- package/dist/tools/dispatcher.d.ts.map +1 -0
- package/dist/tools/dispatcher.js +647 -0
- package/dist/tools/dispatcher.js.map +1 -0
- package/dist/tools/memory.d.ts +4 -0
- package/dist/tools/memory.d.ts.map +1 -0
- package/dist/tools/memory.js +92 -0
- package/dist/tools/memory.js.map +1 -0
- package/dist/tools/platform.d.ts +4 -0
- package/dist/tools/platform.d.ts.map +1 -0
- package/dist/tools/platform.js +359 -0
- package/dist/tools/platform.js.map +1 -0
- package/dist/tools/projects.d.ts +4 -0
- package/dist/tools/projects.d.ts.map +1 -0
- package/dist/tools/projects.js +79 -0
- package/dist/tools/projects.js.map +1 -0
- package/dist/tools/rate-limiter.d.ts +53 -0
- package/dist/tools/rate-limiter.d.ts.map +1 -0
- package/dist/tools/rate-limiter.js +87 -0
- package/dist/tools/rate-limiter.js.map +1 -0
- package/dist/tools/remaining.d.ts +8 -0
- package/dist/tools/remaining.d.ts.map +1 -0
- package/dist/tools/remaining.js +283 -0
- package/dist/tools/remaining.js.map +1 -0
- package/dist/tools/sandbox.d.ts +4 -0
- package/dist/tools/sandbox.d.ts.map +1 -0
- package/dist/tools/sandbox.js +35 -0
- package/dist/tools/sandbox.js.map +1 -0
- package/dist/tools/social.d.ts +11 -0
- package/dist/tools/social.d.ts.map +1 -0
- package/dist/tools/social.js +136 -0
- package/dist/tools/social.js.map +1 -0
- package/dist/tools/swarms.d.ts +4 -0
- package/dist/tools/swarms.d.ts.map +1 -0
- package/dist/tools/swarms.js +184 -0
- package/dist/tools/swarms.js.map +1 -0
- package/dist/tools/templates.d.ts +4 -0
- package/dist/tools/templates.d.ts.map +1 -0
- package/dist/tools/templates.js +102 -0
- package/dist/tools/templates.js.map +1 -0
- package/dist/tools/utils.d.ts +9 -0
- package/dist/tools/utils.d.ts.map +1 -0
- package/dist/tools/utils.js +8 -0
- package/dist/tools/utils.js.map +1 -0
- package/package.json +50 -0
|
@@ -0,0 +1,206 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
2
|
+
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
3
|
+
import { registerDevTools } from '../devtools.js';
|
|
4
|
+
import { ScopedToolRegistry } from '../../scopes.js';
|
|
5
|
+
function createMockClient() {
|
|
6
|
+
return {
|
|
7
|
+
projectBrain: {
|
|
8
|
+
tasks: vi.fn(),
|
|
9
|
+
stats: vi.fn(),
|
|
10
|
+
decisions: vi.fn(),
|
|
11
|
+
milestones: vi.fn(),
|
|
12
|
+
teamActivity: vi.fn(),
|
|
13
|
+
usage: vi.fn(),
|
|
14
|
+
getSettings: vi.fn(),
|
|
15
|
+
updateSettings: vi.fn(),
|
|
16
|
+
inspectPage: vi.fn(),
|
|
17
|
+
screenshot: vi.fn(),
|
|
18
|
+
visualAudit: vi.fn(),
|
|
19
|
+
browserUseRunTask: vi.fn(),
|
|
20
|
+
browserUseTasks: vi.fn(),
|
|
21
|
+
browserUseTask: vi.fn(),
|
|
22
|
+
stopBrowserUseTask: vi.fn(),
|
|
23
|
+
},
|
|
24
|
+
storage: {
|
|
25
|
+
listBuckets: vi.fn(),
|
|
26
|
+
createBucket: vi.fn(),
|
|
27
|
+
listItems: vi.fn(),
|
|
28
|
+
getUploadUrl: vi.fn(),
|
|
29
|
+
getDownloadUrl: vi.fn(),
|
|
30
|
+
deleteObject: vi.fn(),
|
|
31
|
+
createFolder: vi.fn(),
|
|
32
|
+
},
|
|
33
|
+
databases: {
|
|
34
|
+
list: vi.fn(),
|
|
35
|
+
create: vi.fn(),
|
|
36
|
+
getCredentials: vi.fn(),
|
|
37
|
+
query: vi.fn(),
|
|
38
|
+
},
|
|
39
|
+
github: {
|
|
40
|
+
getRepository: vi.fn(),
|
|
41
|
+
getTree: vi.fn(),
|
|
42
|
+
getFileContent: vi.fn(),
|
|
43
|
+
getCommits: vi.fn(),
|
|
44
|
+
},
|
|
45
|
+
billing: {
|
|
46
|
+
get: vi.fn(),
|
|
47
|
+
getStatus: vi.fn(),
|
|
48
|
+
getUsage: vi.fn(),
|
|
49
|
+
getUsageByProject: vi.fn(),
|
|
50
|
+
},
|
|
51
|
+
freeTier: {
|
|
52
|
+
getTier: vi.fn(),
|
|
53
|
+
getUsage: vi.fn(),
|
|
54
|
+
getWarnings: vi.fn(),
|
|
55
|
+
},
|
|
56
|
+
integrations: {
|
|
57
|
+
listApps: vi.fn(),
|
|
58
|
+
getTools: vi.fn(),
|
|
59
|
+
listConnections: vi.fn(),
|
|
60
|
+
disconnect: vi.fn(),
|
|
61
|
+
listPendingExecutions: vi.fn(),
|
|
62
|
+
approveExecution: vi.fn(),
|
|
63
|
+
rejectExecution: vi.fn(),
|
|
64
|
+
},
|
|
65
|
+
agents: {
|
|
66
|
+
listDiscoverable: vi.fn(),
|
|
67
|
+
leaderboard: vi.fn(),
|
|
68
|
+
orchestratorStatus: vi.fn(),
|
|
69
|
+
grantProjectAccess: vi.fn(),
|
|
70
|
+
revokeProjectAccess: vi.fn(),
|
|
71
|
+
resetRequestCount: vi.fn(),
|
|
72
|
+
unreadCount: vi.fn(),
|
|
73
|
+
markAllRead: vi.fn(),
|
|
74
|
+
},
|
|
75
|
+
tags: {
|
|
76
|
+
list: vi.fn(),
|
|
77
|
+
create: vi.fn(),
|
|
78
|
+
delete: vi.fn(),
|
|
79
|
+
},
|
|
80
|
+
templates: {
|
|
81
|
+
delete: vi.fn(),
|
|
82
|
+
},
|
|
83
|
+
};
|
|
84
|
+
}
|
|
85
|
+
function extractTools(server) {
|
|
86
|
+
const tools = {};
|
|
87
|
+
const origTool = server.tool.bind(server);
|
|
88
|
+
server.tool = ((...args) => {
|
|
89
|
+
const name = args[0];
|
|
90
|
+
const handler = args[args.length - 1];
|
|
91
|
+
tools[name] = { handler };
|
|
92
|
+
return origTool(...args);
|
|
93
|
+
});
|
|
94
|
+
return tools;
|
|
95
|
+
}
|
|
96
|
+
describe('devtools', () => {
|
|
97
|
+
let client;
|
|
98
|
+
let server;
|
|
99
|
+
let tools;
|
|
100
|
+
beforeEach(() => {
|
|
101
|
+
client = createMockClient();
|
|
102
|
+
server = new McpServer({ name: 'test', version: '0.0.1' });
|
|
103
|
+
tools = extractTools(server);
|
|
104
|
+
registerDevTools(new ScopedToolRegistry(server, null), client);
|
|
105
|
+
});
|
|
106
|
+
it('registers browser inspection tools', () => {
|
|
107
|
+
expect(Object.keys(tools)).toEqual(expect.arrayContaining([
|
|
108
|
+
'browser_inspect_page',
|
|
109
|
+
'browser_screenshot',
|
|
110
|
+
'run_visual_audit',
|
|
111
|
+
'browser_use_run_task',
|
|
112
|
+
'browser_use_list_tasks',
|
|
113
|
+
'browser_use_get_task',
|
|
114
|
+
'browser_use_stop_task',
|
|
115
|
+
]));
|
|
116
|
+
});
|
|
117
|
+
it('routes browser_inspect_page through projectBrain.inspectPage', async () => {
|
|
118
|
+
const data = { data: { title: 'Homepage' } };
|
|
119
|
+
client.projectBrain.inspectPage.mockResolvedValue(data);
|
|
120
|
+
const result = await tools.browser_inspect_page.handler({
|
|
121
|
+
project_id: 'proj-1',
|
|
122
|
+
url: 'https://example.com',
|
|
123
|
+
viewport: 'desktop',
|
|
124
|
+
});
|
|
125
|
+
expect(client.projectBrain.inspectPage).toHaveBeenCalledWith('proj-1', {
|
|
126
|
+
url: 'https://example.com',
|
|
127
|
+
viewport: 'desktop',
|
|
128
|
+
});
|
|
129
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
130
|
+
});
|
|
131
|
+
it('routes browser_screenshot through projectBrain.screenshot', async () => {
|
|
132
|
+
const data = { data: { screenshotPath: '/tmp/example.png' } };
|
|
133
|
+
client.projectBrain.screenshot.mockResolvedValue(data);
|
|
134
|
+
const result = await tools.browser_screenshot.handler({
|
|
135
|
+
project_id: 'proj-1',
|
|
136
|
+
url: 'https://example.com',
|
|
137
|
+
viewport: 'mobile',
|
|
138
|
+
});
|
|
139
|
+
expect(client.projectBrain.screenshot).toHaveBeenCalledWith('proj-1', {
|
|
140
|
+
url: 'https://example.com',
|
|
141
|
+
viewport: 'mobile',
|
|
142
|
+
});
|
|
143
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
144
|
+
});
|
|
145
|
+
it('routes run_visual_audit through projectBrain.visualAudit', async () => {
|
|
146
|
+
const data = { data: { overallGrade: 'A' } };
|
|
147
|
+
client.projectBrain.visualAudit.mockResolvedValue(data);
|
|
148
|
+
const result = await tools.run_visual_audit.handler({
|
|
149
|
+
project_id: 'proj-1',
|
|
150
|
+
base_url: 'https://example.com',
|
|
151
|
+
lane: 'critical',
|
|
152
|
+
});
|
|
153
|
+
expect(client.projectBrain.visualAudit).toHaveBeenCalledWith('proj-1', {
|
|
154
|
+
base_url: 'https://example.com',
|
|
155
|
+
lane: 'critical',
|
|
156
|
+
});
|
|
157
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
158
|
+
});
|
|
159
|
+
it('routes browser_use_run_task through projectBrain.browserUseRunTask', async () => {
|
|
160
|
+
const data = { data: { taskId: 'task-1' } };
|
|
161
|
+
client.projectBrain.browserUseRunTask.mockResolvedValue(data);
|
|
162
|
+
const result = await tools.browser_use_run_task.handler({
|
|
163
|
+
project_id: 'proj-1',
|
|
164
|
+
task: 'Test the signup flow',
|
|
165
|
+
start_url: 'https://example.com',
|
|
166
|
+
max_steps: 25,
|
|
167
|
+
});
|
|
168
|
+
expect(client.projectBrain.browserUseRunTask).toHaveBeenCalledWith('proj-1', {
|
|
169
|
+
task: 'Test the signup flow',
|
|
170
|
+
start_url: 'https://example.com',
|
|
171
|
+
max_steps: 25,
|
|
172
|
+
});
|
|
173
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
174
|
+
});
|
|
175
|
+
it('routes browser_use_list_tasks through projectBrain.browserUseTasks', async () => {
|
|
176
|
+
const data = { data: [{ taskId: 'task-1' }] };
|
|
177
|
+
client.projectBrain.browserUseTasks.mockResolvedValue(data);
|
|
178
|
+
const result = await tools.browser_use_list_tasks.handler({
|
|
179
|
+
project_id: 'proj-1',
|
|
180
|
+
limit: 5,
|
|
181
|
+
});
|
|
182
|
+
expect(client.projectBrain.browserUseTasks).toHaveBeenCalledWith('proj-1', { limit: 5 });
|
|
183
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
184
|
+
});
|
|
185
|
+
it('routes browser_use_get_task through projectBrain.browserUseTask', async () => {
|
|
186
|
+
const data = { data: { taskId: 'task-1', status: 'started' } };
|
|
187
|
+
client.projectBrain.browserUseTask.mockResolvedValue(data);
|
|
188
|
+
const result = await tools.browser_use_get_task.handler({
|
|
189
|
+
project_id: 'proj-1',
|
|
190
|
+
task_id: 'task-1',
|
|
191
|
+
});
|
|
192
|
+
expect(client.projectBrain.browserUseTask).toHaveBeenCalledWith('proj-1', 'task-1');
|
|
193
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
194
|
+
});
|
|
195
|
+
it('routes browser_use_stop_task through projectBrain.stopBrowserUseTask', async () => {
|
|
196
|
+
const data = { data: { taskId: 'task-1', status: 'stopped' } };
|
|
197
|
+
client.projectBrain.stopBrowserUseTask.mockResolvedValue(data);
|
|
198
|
+
const result = await tools.browser_use_stop_task.handler({
|
|
199
|
+
project_id: 'proj-1',
|
|
200
|
+
task_id: 'task-1',
|
|
201
|
+
});
|
|
202
|
+
expect(client.projectBrain.stopBrowserUseTask).toHaveBeenCalledWith('proj-1', 'task-1');
|
|
203
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
204
|
+
});
|
|
205
|
+
});
|
|
206
|
+
//# sourceMappingURL=devtools.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"devtools.test.js","sourceRoot":"","sources":["../../../src/tools/__tests__/devtools.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAErD,SAAS,gBAAgB;IACvB,OAAO;QACL,YAAY,EAAE;YACZ,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;YAClB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YACnB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;YACrB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;YACpB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;YACvB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;YACpB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YACnB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;YACpB,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC1B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;YACxB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;YACvB,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;SAC5B;QACD,OAAO,EAAE;YACP,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;YACpB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;YACrB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;YAClB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;YACrB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;YACvB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;YACrB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE;SACtB;QACD,SAAS,EAAE;YACT,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;YACb,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;YACvB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACf;QACD,MAAM,EAAE;YACN,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;YACtB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;YACvB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;SACpB;QACD,OAAO,EAAE;YACP,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;YACZ,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;YAClB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;SAC3B;QACD,QAAQ,EAAE;YACR,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;SACrB;QACD,YAAY,EAAE;YACZ,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;YACxB,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YACnB,qBAAqB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC9B,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;YACzB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;SACzB;QACD,MAAM,EAAE;YACN,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;YACzB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;YACpB,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC3B,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC3B,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC5B,iBAAiB,EAAE,EAAE,CAAC,EAAE,EAAE;YAC1B,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;YACpB,WAAW,EAAE,EAAE,CAAC,EAAE,EAAE;SACrB;QACD,IAAI,EAAE;YACJ,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;YACb,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;SAChB;QACD,SAAS,EAAE;YACT,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;SAChB;KACqB,CAAC;AAC3B,CAAC;AAED,SAAS,YAAY,CAAC,MAAiB;IACrC,MAAM,KAAK,GAA0C,EAAE,CAAC;IACxD,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,IAAW,EAAE,EAAE;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAa,CAAC;QAClD,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC;QAC1B,OAAQ,QAAgB,CAAC,GAAG,IAAI,CAAC,CAAC;IACpC,CAAC,CAAQ,CAAC;IACV,OAAO,KAAK,CAAC;AACf,CAAC;AAED,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,IAAI,MAAgB,CAAC;IACrB,IAAI,MAAiB,CAAC;IACtB,IAAI,KAA4C,CAAC;IAEjD,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,gBAAgB,EAAE,CAAC;QAC5B,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAC3D,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QAC7B,gBAAgB,CAAC,IAAI,kBAAkB,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAChC,MAAM,CAAC,eAAe,CAAC;YACrB,sBAAsB;YACtB,oBAAoB;YACpB,kBAAkB;YAClB,sBAAsB;YACtB,wBAAwB;YACxB,sBAAsB;YACtB,uBAAuB;SACxB,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,CAAC;QAC5C,MAAM,CAAC,YAAY,CAAC,WAAmB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEjE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,oBAAoB,CAAC,OAAO,CAAC;YACtD,UAAU,EAAE,QAAQ;YACpB,GAAG,EAAE,qBAAqB;YAC1B,QAAQ,EAAE,SAAS;SACpB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE;YACrE,GAAG,EAAE,qBAAqB;YAC1B,QAAQ,EAAE,SAAS;SACpB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;QACzE,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,EAAE,CAAC;QAC7D,MAAM,CAAC,YAAY,CAAC,UAAkB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC;YACpD,UAAU,EAAE,QAAQ;YACpB,GAAG,EAAE,qBAAqB;YAC1B,QAAQ,EAAE,QAAQ;SACnB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE;YACpE,GAAG,EAAE,qBAAqB;YAC1B,QAAQ,EAAE,QAAQ;SACnB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,YAAY,EAAE,GAAG,EAAE,EAAE,CAAC;QAC5C,MAAM,CAAC,YAAY,CAAC,WAAmB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEjE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC;YAClD,UAAU,EAAE,QAAQ;YACpB,QAAQ,EAAE,qBAAqB;YAC/B,IAAI,EAAE,UAAU;SACjB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE;YACrE,QAAQ,EAAE,qBAAqB;YAC/B,IAAI,EAAE,UAAU;SACjB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,CAAC;QAC3C,MAAM,CAAC,YAAY,CAAC,iBAAyB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEvE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,oBAAoB,CAAC,OAAO,CAAC;YACtD,UAAU,EAAE,QAAQ;YACpB,IAAI,EAAE,sBAAsB;YAC5B,SAAS,EAAE,qBAAqB;YAChC,SAAS,EAAE,EAAE;SACd,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE;YAC3E,IAAI,EAAE,sBAAsB;YAC5B,SAAS,EAAE,qBAAqB;YAChC,SAAS,EAAE,EAAE;SACd,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QAC7C,MAAM,CAAC,YAAY,CAAC,eAAuB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAErE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,sBAAsB,CAAC,OAAO,CAAC;YACxD,UAAU,EAAE,QAAQ;YACpB,KAAK,EAAE,CAAC;SACT,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;QACzF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;QAC9D,MAAM,CAAC,YAAY,CAAC,cAAsB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEpE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,oBAAoB,CAAC,OAAO,CAAC;YACtD,UAAU,EAAE,QAAQ;YACpB,OAAO,EAAE,QAAQ;SAClB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACpF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC;QAC9D,MAAM,CAAC,YAAY,CAAC,kBAA0B,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAExE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,qBAAqB,CAAC,OAAO,CAAC;YACvD,UAAU,EAAE,QAAQ;YACpB,OAAO,EAAE,QAAQ;SAClB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACxF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"dispatcher.test.d.ts","sourceRoot":"","sources":["../../../src/tools/__tests__/dispatcher.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,178 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
2
|
+
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
3
|
+
import { registerDispatcherTools } from '../dispatcher.js';
|
|
4
|
+
import { ScopedToolRegistry } from '../../scopes.js';
|
|
5
|
+
function createMockClient() {
|
|
6
|
+
return {
|
|
7
|
+
dispatcher: {
|
|
8
|
+
tasks: vi.fn(),
|
|
9
|
+
claimNext: vi.fn(),
|
|
10
|
+
claim: vi.fn(),
|
|
11
|
+
heartbeat: vi.fn(),
|
|
12
|
+
complete: vi.fn(),
|
|
13
|
+
release: vi.fn(),
|
|
14
|
+
create: vi.fn(),
|
|
15
|
+
stats: vi.fn(),
|
|
16
|
+
},
|
|
17
|
+
};
|
|
18
|
+
}
|
|
19
|
+
// Extract tool handlers from the server
|
|
20
|
+
function extractTools(server) {
|
|
21
|
+
const tools = {};
|
|
22
|
+
const origTool = server.tool.bind(server);
|
|
23
|
+
server.tool = ((...args) => {
|
|
24
|
+
const name = args[0];
|
|
25
|
+
const handler = args[args.length - 1];
|
|
26
|
+
tools[name] = { schema: args.length === 4 ? args[2] : {}, handler };
|
|
27
|
+
return origTool(...args);
|
|
28
|
+
});
|
|
29
|
+
return tools;
|
|
30
|
+
}
|
|
31
|
+
describe('dispatcher tools', () => {
|
|
32
|
+
let client;
|
|
33
|
+
let server;
|
|
34
|
+
let tools;
|
|
35
|
+
beforeEach(() => {
|
|
36
|
+
client = createMockClient();
|
|
37
|
+
server = new McpServer({ name: 'test', version: '0.0.1' });
|
|
38
|
+
tools = extractTools(server);
|
|
39
|
+
const registry = new ScopedToolRegistry(server, null);
|
|
40
|
+
registerDispatcherTools(registry, client);
|
|
41
|
+
});
|
|
42
|
+
it('registers all 8 dispatcher tools', () => {
|
|
43
|
+
const names = Object.keys(tools);
|
|
44
|
+
expect(names).toContain('list_tasks');
|
|
45
|
+
expect(names).toContain('claim_next_task');
|
|
46
|
+
expect(names).toContain('claim_task');
|
|
47
|
+
expect(names).toContain('heartbeat_task');
|
|
48
|
+
expect(names).toContain('complete_task');
|
|
49
|
+
expect(names).toContain('release_task');
|
|
50
|
+
expect(names).toContain('create_task');
|
|
51
|
+
expect(names).toContain('get_dispatcher_stats');
|
|
52
|
+
});
|
|
53
|
+
describe('list_tasks', () => {
|
|
54
|
+
it('returns formatted task list', async () => {
|
|
55
|
+
client.dispatcher.tasks.mockResolvedValue({
|
|
56
|
+
data: [
|
|
57
|
+
{ id: 'task-1', title: 'Fix bug', status: 'pending', score: 80, effort: 3, owner: 'bill' },
|
|
58
|
+
{ id: 'task-2', title: 'Add feature', status: 'claimed', score: 60, effort: 5 },
|
|
59
|
+
],
|
|
60
|
+
});
|
|
61
|
+
const result = await tools.list_tasks.handler({ status: 'pending' });
|
|
62
|
+
expect(result.content[0].text).toContain('task-1');
|
|
63
|
+
expect(result.content[0].text).toContain('Fix bug');
|
|
64
|
+
expect(result.content[0].text).toContain('owner: bill');
|
|
65
|
+
expect(result.isError).toBeUndefined();
|
|
66
|
+
});
|
|
67
|
+
it('returns message when no tasks found', async () => {
|
|
68
|
+
client.dispatcher.tasks.mockResolvedValue({ data: [] });
|
|
69
|
+
const result = await tools.list_tasks.handler({});
|
|
70
|
+
expect(result.content[0].text).toContain('No tasks found');
|
|
71
|
+
});
|
|
72
|
+
it('returns error on failure', async () => {
|
|
73
|
+
client.dispatcher.tasks.mockRejectedValue(new Error('Network error'));
|
|
74
|
+
const result = await tools.list_tasks.handler({});
|
|
75
|
+
expect(result.content[0].text).toContain('Error listing tasks');
|
|
76
|
+
expect(result.isError).toBe(true);
|
|
77
|
+
});
|
|
78
|
+
});
|
|
79
|
+
describe('claim_next_task', () => {
|
|
80
|
+
it('returns claim details on success', async () => {
|
|
81
|
+
client.dispatcher.claimNext.mockResolvedValue({
|
|
82
|
+
data: {
|
|
83
|
+
task: { title: 'Fix bug' },
|
|
84
|
+
task_id: 'task-1',
|
|
85
|
+
claim_id: 'claim-1',
|
|
86
|
+
agent: 'claude-1',
|
|
87
|
+
},
|
|
88
|
+
});
|
|
89
|
+
const result = await tools.claim_next_task.handler({ agent: 'claude-1' });
|
|
90
|
+
expect(result.content[0].text).toContain('Claimed task: Fix bug');
|
|
91
|
+
expect(result.content[0].text).toContain('heartbeat');
|
|
92
|
+
});
|
|
93
|
+
it('returns friendly message on 409 (no available tasks)', async () => {
|
|
94
|
+
const err = new Error('Conflict');
|
|
95
|
+
err.status = 409;
|
|
96
|
+
client.dispatcher.claimNext.mockRejectedValue(err);
|
|
97
|
+
const result = await tools.claim_next_task.handler({ agent: 'claude-1' });
|
|
98
|
+
expect(result.content[0].text).toContain('No available tasks');
|
|
99
|
+
expect(result.isError).toBeUndefined();
|
|
100
|
+
});
|
|
101
|
+
});
|
|
102
|
+
describe('claim_task', () => {
|
|
103
|
+
it('returns claim details on success', async () => {
|
|
104
|
+
client.dispatcher.claim.mockResolvedValue({
|
|
105
|
+
data: {
|
|
106
|
+
task: { title: 'Fix bug' },
|
|
107
|
+
task_id: 'task-1',
|
|
108
|
+
claim_id: 'claim-1',
|
|
109
|
+
},
|
|
110
|
+
});
|
|
111
|
+
const result = await tools.claim_task.handler({ task_id: 'task-1', agent: 'claude-1' });
|
|
112
|
+
expect(result.content[0].text).toContain('Claimed task: Fix bug');
|
|
113
|
+
expect(client.dispatcher.claim).toHaveBeenCalledWith('task-1', { agent: 'claude-1' });
|
|
114
|
+
});
|
|
115
|
+
it('returns friendly message on 409 (already claimed)', async () => {
|
|
116
|
+
const err = new Error('Conflict');
|
|
117
|
+
err.status = 409;
|
|
118
|
+
client.dispatcher.claim.mockRejectedValue(err);
|
|
119
|
+
const result = await tools.claim_task.handler({ task_id: 'task-1', agent: 'claude-1' });
|
|
120
|
+
expect(result.content[0].text).toContain('already claimed');
|
|
121
|
+
});
|
|
122
|
+
});
|
|
123
|
+
describe('heartbeat_task', () => {
|
|
124
|
+
it('sends heartbeat successfully', async () => {
|
|
125
|
+
client.dispatcher.heartbeat.mockResolvedValue({});
|
|
126
|
+
const result = await tools.heartbeat_task.handler({ task_id: 'task-1', agent: 'claude-1' });
|
|
127
|
+
expect(result.content[0].text).toContain('Heartbeat sent');
|
|
128
|
+
expect(client.dispatcher.heartbeat).toHaveBeenCalledWith('task-1', { agent: 'claude-1' });
|
|
129
|
+
});
|
|
130
|
+
});
|
|
131
|
+
describe('complete_task', () => {
|
|
132
|
+
it('marks task done', async () => {
|
|
133
|
+
client.dispatcher.complete.mockResolvedValue({});
|
|
134
|
+
const result = await tools.complete_task.handler({ task_id: 'task-1', notes: 'Done!' });
|
|
135
|
+
expect(result.content[0].text).toContain('marked as done');
|
|
136
|
+
expect(client.dispatcher.complete).toHaveBeenCalledWith('task-1', { notes: 'Done!' });
|
|
137
|
+
});
|
|
138
|
+
});
|
|
139
|
+
describe('release_task', () => {
|
|
140
|
+
it('releases claim', async () => {
|
|
141
|
+
client.dispatcher.release.mockResolvedValue({});
|
|
142
|
+
const result = await tools.release_task.handler({ task_id: 'task-1' });
|
|
143
|
+
expect(result.content[0].text).toContain('Released claim');
|
|
144
|
+
});
|
|
145
|
+
});
|
|
146
|
+
describe('create_task', () => {
|
|
147
|
+
it('creates task with auto-generated ID', async () => {
|
|
148
|
+
client.dispatcher.create.mockResolvedValue({
|
|
149
|
+
data: { id: 'proj-fix-bug-abc123', title: 'Fix bug', score: 50 },
|
|
150
|
+
});
|
|
151
|
+
const result = await tools.create_task.handler({ title: 'Fix bug', effort: 3 });
|
|
152
|
+
expect(result.content[0].text).toContain('Created task: Fix bug');
|
|
153
|
+
expect(client.dispatcher.create).toHaveBeenCalledWith(expect.objectContaining({
|
|
154
|
+
title: 'Fix bug',
|
|
155
|
+
effort: 3,
|
|
156
|
+
id: expect.stringMatching(/^proj-fix-bug-/),
|
|
157
|
+
}));
|
|
158
|
+
});
|
|
159
|
+
});
|
|
160
|
+
describe('get_dispatcher_stats', () => {
|
|
161
|
+
it('formats stats output', async () => {
|
|
162
|
+
client.dispatcher.stats.mockResolvedValue({
|
|
163
|
+
data: {
|
|
164
|
+
tasks: { total: 10, pending: 3, in_progress: 2, done: 5, blocked: 0, archived: 0, stale: 0 },
|
|
165
|
+
activeClaims: 2,
|
|
166
|
+
byLayer: [{ layer: 'core', count: 7, done: 4 }],
|
|
167
|
+
byOwner: [{ owner: 'bill', count: 5, done: 3 }],
|
|
168
|
+
},
|
|
169
|
+
});
|
|
170
|
+
const result = await tools.get_dispatcher_stats.handler({});
|
|
171
|
+
expect(result.content[0].text).toContain('Total tasks: 10');
|
|
172
|
+
expect(result.content[0].text).toContain('Active claims: 2');
|
|
173
|
+
expect(result.content[0].text).toContain('core: 7');
|
|
174
|
+
expect(result.content[0].text).toContain('bill: 5');
|
|
175
|
+
});
|
|
176
|
+
});
|
|
177
|
+
});
|
|
178
|
+
//# sourceMappingURL=dispatcher.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"dispatcher.test.js","sourceRoot":"","sources":["../../../src/tools/__tests__/dispatcher.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,uBAAuB,EAAE,MAAM,kBAAkB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAErD,SAAS,gBAAgB;IACvB,OAAO;QACL,UAAU,EAAE;YACV,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;YAClB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;YACd,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;YAClB,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;YACjB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACf;KACqB,CAAC;AAC3B,CAAC;AAED,wCAAwC;AACxC,SAAS,YAAY,CAAC,MAAiB;IACrC,MAAM,KAAK,GAAuD,EAAE,CAAC;IACrE,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,IAAW,EAAE,EAAE;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAa,CAAC;QAClD,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,OAAO,EAAE,CAAC;QACpE,OAAQ,QAAgB,CAAC,GAAG,IAAI,CAAC,CAAC;IACpC,CAAC,CAAQ,CAAC;IACV,OAAO,KAAK,CAAC;AACf,CAAC;AAED,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,MAAgB,CAAC;IACrB,IAAI,MAAiB,CAAC;IACtB,IAAI,KAA4C,CAAC;IAEjD,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,gBAAgB,EAAE,CAAC;QAC5B,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAC3D,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QAC7B,MAAM,QAAQ,GAAG,IAAI,kBAAkB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACtD,uBAAuB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACtC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACtC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACxC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,CAAC,UAAU,CAAC,KAAa,CAAC,iBAAiB,CAAC;gBACjD,IAAI,EAAE;oBACJ,EAAE,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE;oBAC1F,EAAE,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE;iBAChF;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,CAAC,UAAU,CAAC,KAAa,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YACjE,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,CAAC,UAAU,CAAC,KAAa,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;YAC/E,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;YAChE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,CAAC,UAAU,CAAC,SAAiB,CAAC,iBAAiB,CAAC;gBACrD,IAAI,EAAE;oBACJ,IAAI,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;oBAC1B,OAAO,EAAE,QAAQ;oBACjB,QAAQ,EAAE,SAAS;oBACnB,KAAK,EAAE,UAAU;iBAClB;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,UAAU,CAAQ,CAAC;YACzC,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC;YAChB,MAAM,CAAC,UAAU,CAAC,SAAiB,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,eAAe,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,aAAa,EAAE,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,CAAC,UAAU,CAAC,KAAa,CAAC,iBAAiB,CAAC;gBACjD,IAAI,EAAE;oBACJ,IAAI,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;oBAC1B,OAAO,EAAE,QAAQ;oBACjB,QAAQ,EAAE,SAAS;iBACpB;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YACxF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,UAAU,CAAQ,CAAC;YACzC,GAAG,CAAC,MAAM,GAAG,GAAG,CAAC;YAChB,MAAM,CAAC,UAAU,CAAC,KAAa,CAAC,iBAAiB,CAAC,GAAG,CAAC,CAAC;YAExD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YACxF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,CAAC,UAAU,CAAC,SAAiB,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;YAC5F,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAC5F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,iBAAiB,EAAE,KAAK,IAAI,EAAE;YAC9B,MAAM,CAAC,UAAU,CAAC,QAAgB,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC1D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YACxF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;YAC7B,MAAM,CAAC,UAAU,CAAC,OAAe,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACzD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;YACvE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,CAAC,UAAU,CAAC,MAAc,CAAC,iBAAiB,CAAC;gBAClD,IAAI,EAAE,EAAE,EAAE,EAAE,qBAAqB,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE;aACjE,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;YAChF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CACnD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,KAAK,EAAE,SAAS;gBAChB,MAAM,EAAE,CAAC;gBACT,EAAE,EAAE,MAAM,CAAC,cAAc,CAAC,gBAAgB,CAAC;aAC5C,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;YACnC,MAAM,CAAC,UAAU,CAAC,KAAa,CAAC,iBAAiB,CAAC;gBACjD,IAAI,EAAE;oBACJ,KAAK,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;oBAC5F,YAAY,EAAE,CAAC;oBACf,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;oBAC/C,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;iBAChD;aACF,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,oBAAoB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACpD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.test.d.ts","sourceRoot":"","sources":["../../../src/tools/__tests__/memory.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,151 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach } from 'vitest';
|
|
2
|
+
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
3
|
+
import { registerMemoryTools } from '../memory.js';
|
|
4
|
+
import { ScopedToolRegistry } from '../../scopes.js';
|
|
5
|
+
function createMockClient() {
|
|
6
|
+
return {
|
|
7
|
+
memory: {
|
|
8
|
+
facts: {
|
|
9
|
+
add: vi.fn(),
|
|
10
|
+
list: vi.fn(),
|
|
11
|
+
remove: vi.fn(),
|
|
12
|
+
},
|
|
13
|
+
decisions: {
|
|
14
|
+
log: vi.fn(),
|
|
15
|
+
},
|
|
16
|
+
search: vi.fn(),
|
|
17
|
+
context: vi.fn(),
|
|
18
|
+
purge: vi.fn(),
|
|
19
|
+
},
|
|
20
|
+
};
|
|
21
|
+
}
|
|
22
|
+
function extractTools(server) {
|
|
23
|
+
const tools = {};
|
|
24
|
+
const origTool = server.tool.bind(server);
|
|
25
|
+
server.tool = ((...args) => {
|
|
26
|
+
const name = args[0];
|
|
27
|
+
const handler = args[args.length - 1];
|
|
28
|
+
tools[name] = { handler };
|
|
29
|
+
return origTool(...args);
|
|
30
|
+
});
|
|
31
|
+
return tools;
|
|
32
|
+
}
|
|
33
|
+
describe('memory tools', () => {
|
|
34
|
+
let client;
|
|
35
|
+
let server;
|
|
36
|
+
let tools;
|
|
37
|
+
beforeEach(() => {
|
|
38
|
+
client = createMockClient();
|
|
39
|
+
server = new McpServer({ name: 'test', version: '0.0.1' });
|
|
40
|
+
tools = extractTools(server);
|
|
41
|
+
const registry = new ScopedToolRegistry(server, null);
|
|
42
|
+
registerMemoryTools(registry, client);
|
|
43
|
+
});
|
|
44
|
+
it('registers all 7 memory tools', () => {
|
|
45
|
+
const names = Object.keys(tools);
|
|
46
|
+
expect(names).toEqual(expect.arrayContaining([
|
|
47
|
+
'remember', 'recall', 'forget', 'log_decision',
|
|
48
|
+
'search_memory', 'get_memory_context', 'purge_memory',
|
|
49
|
+
]));
|
|
50
|
+
});
|
|
51
|
+
describe('remember', () => {
|
|
52
|
+
it('stores a fact and returns JSON result', async () => {
|
|
53
|
+
const data = { data: { id: 'f1', fact: 'API uses tRPC' } };
|
|
54
|
+
client.memory.facts.add.mockResolvedValue(data);
|
|
55
|
+
const result = await tools.remember.handler({
|
|
56
|
+
fact: 'API uses tRPC',
|
|
57
|
+
tags: ['architecture'],
|
|
58
|
+
});
|
|
59
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
60
|
+
expect(client.memory.facts.add).toHaveBeenCalledWith(expect.objectContaining({
|
|
61
|
+
fact: 'API uses tRPC',
|
|
62
|
+
tags: ['architecture'],
|
|
63
|
+
source: 'manual',
|
|
64
|
+
}));
|
|
65
|
+
});
|
|
66
|
+
it('passes through explicit org and project IDs', async () => {
|
|
67
|
+
client.memory.facts.add.mockResolvedValue({ data: {} });
|
|
68
|
+
await tools.remember.handler({ fact: 'test', organization_id: 'org-1', project_id: 'proj-1' });
|
|
69
|
+
expect(client.memory.facts.add).toHaveBeenCalledWith(expect.objectContaining({
|
|
70
|
+
organization_id: 'org-1',
|
|
71
|
+
project_id: 'proj-1',
|
|
72
|
+
}));
|
|
73
|
+
});
|
|
74
|
+
});
|
|
75
|
+
describe('recall', () => {
|
|
76
|
+
it('returns stored facts and forwards query param', async () => {
|
|
77
|
+
const data = { data: [{ id: 'f1', fact: 'API uses tRPC' }] };
|
|
78
|
+
client.memory.facts.list.mockResolvedValue(data);
|
|
79
|
+
const result = await tools.recall.handler({ query: 'tRPC' });
|
|
80
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
81
|
+
expect(client.memory.facts.list).toHaveBeenCalledWith(expect.objectContaining({ query: 'tRPC' }));
|
|
82
|
+
});
|
|
83
|
+
});
|
|
84
|
+
describe('forget', () => {
|
|
85
|
+
it('removes a fact by ID and returns result', async () => {
|
|
86
|
+
const data = { data: { success: true } };
|
|
87
|
+
client.memory.facts.remove.mockResolvedValue(data);
|
|
88
|
+
const result = await tools.forget.handler({ fact_id: 'f1' });
|
|
89
|
+
expect(client.memory.facts.remove).toHaveBeenCalledWith('f1');
|
|
90
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
91
|
+
});
|
|
92
|
+
});
|
|
93
|
+
describe('log_decision', () => {
|
|
94
|
+
it('records a decision with all fields', async () => {
|
|
95
|
+
const data = { data: { id: 'd1', decision: 'Use Redis for caching' } };
|
|
96
|
+
client.memory.decisions.log.mockResolvedValue(data);
|
|
97
|
+
const result = await tools.log_decision.handler({
|
|
98
|
+
decision: 'Use Redis for caching',
|
|
99
|
+
context: 'Need fast session lookups',
|
|
100
|
+
tags: ['architecture'],
|
|
101
|
+
});
|
|
102
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
103
|
+
expect(client.memory.decisions.log).toHaveBeenCalledWith(expect.objectContaining({
|
|
104
|
+
decision: 'Use Redis for caching',
|
|
105
|
+
context: 'Need fast session lookups',
|
|
106
|
+
tags: ['architecture'],
|
|
107
|
+
}));
|
|
108
|
+
});
|
|
109
|
+
});
|
|
110
|
+
describe('search_memory', () => {
|
|
111
|
+
it('searches and forwards all params', async () => {
|
|
112
|
+
const data = { data: [{ type: 'fact', content: 'Uses tRPC' }] };
|
|
113
|
+
client.memory.search.mockResolvedValue(data);
|
|
114
|
+
const result = await tools.search_memory.handler({ query: 'tRPC', limit: 5 });
|
|
115
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
116
|
+
expect(client.memory.search).toHaveBeenCalledWith(expect.objectContaining({ query: 'tRPC', limit: 5 }));
|
|
117
|
+
});
|
|
118
|
+
});
|
|
119
|
+
describe('get_memory_context', () => {
|
|
120
|
+
it('returns context string directly (not JSON)', async () => {
|
|
121
|
+
const data = { data: { context: 'Relevant facts about architecture...' } };
|
|
122
|
+
client.memory.context.mockResolvedValue(data);
|
|
123
|
+
const result = await tools.get_memory_context.handler({
|
|
124
|
+
message: 'What architecture do we use?',
|
|
125
|
+
max_tokens: 1000,
|
|
126
|
+
});
|
|
127
|
+
expect(result.content[0].text).toBe('Relevant facts about architecture...');
|
|
128
|
+
expect(client.memory.context).toHaveBeenCalledWith(expect.objectContaining({ message: 'What architecture do we use?', max_tokens: 1000 }));
|
|
129
|
+
});
|
|
130
|
+
it('returns fallback when context is empty string', async () => {
|
|
131
|
+
client.memory.context.mockResolvedValue({ data: { context: '' } });
|
|
132
|
+
const result = await tools.get_memory_context.handler({ message: 'test' });
|
|
133
|
+
expect(result.content[0].text).toBe('(No relevant memories found)');
|
|
134
|
+
});
|
|
135
|
+
it('returns fallback when context is null', async () => {
|
|
136
|
+
client.memory.context.mockResolvedValue({ data: { context: null } });
|
|
137
|
+
const result = await tools.get_memory_context.handler({ message: 'test' });
|
|
138
|
+
expect(result.content[0].text).toBe('(No relevant memories found)');
|
|
139
|
+
});
|
|
140
|
+
});
|
|
141
|
+
describe('purge_memory', () => {
|
|
142
|
+
it('purges with project_id string arg and returns result', async () => {
|
|
143
|
+
const data = { data: { deleted: 5 } };
|
|
144
|
+
client.memory.purge.mockResolvedValue(data);
|
|
145
|
+
const result = await tools.purge_memory.handler({ project_id: 'proj-1' });
|
|
146
|
+
expect(client.memory.purge).toHaveBeenCalledWith('proj-1');
|
|
147
|
+
expect(result.content[0].text).toBe(JSON.stringify(data, null, 2));
|
|
148
|
+
});
|
|
149
|
+
});
|
|
150
|
+
});
|
|
151
|
+
//# sourceMappingURL=memory.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.test.js","sourceRoot":"","sources":["../../../src/tools/__tests__/memory.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEpE,OAAO,EAAE,mBAAmB,EAAE,MAAM,cAAc,CAAC;AACnD,OAAO,EAAE,kBAAkB,EAAE,MAAM,iBAAiB,CAAC;AAErD,SAAS,gBAAgB;IACvB,OAAO;QACL,MAAM,EAAE;YACN,KAAK,EAAE;gBACL,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;gBACZ,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE;gBACb,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;aAChB;YACD,SAAS,EAAE;gBACT,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;aACb;YACD,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACf;KACqB,CAAC;AAC3B,CAAC;AAED,SAAS,YAAY,CAAC,MAAiB;IACrC,MAAM,KAAK,GAA0C,EAAE,CAAC;IACxD,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC1C,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC,GAAG,IAAW,EAAE,EAAE;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,CAAW,CAAC;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAa,CAAC;QAClD,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,OAAO,EAAE,CAAC;QAC1B,OAAQ,QAAgB,CAAC,GAAG,IAAI,CAAC,CAAC;IACpC,CAAC,CAAQ,CAAC;IACV,OAAO,KAAK,CAAC;AACf,CAAC;AAED,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,MAAgB,CAAC;IACrB,IAAI,MAAiB,CAAC;IACtB,IAAI,KAA4C,CAAC;IAEjD,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,gBAAgB,EAAE,CAAC;QAC5B,MAAM,GAAG,IAAI,SAAS,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QAC3D,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QAC7B,MAAM,QAAQ,GAAG,IAAI,kBAAkB,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACtD,mBAAmB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CACnB,MAAM,CAAC,eAAe,CAAC;YACrB,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,cAAc;YAC9C,eAAe,EAAE,oBAAoB,EAAE,cAAc;SACtD,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,eAAe,EAAE,EAAE,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAW,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAC1C,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE,CAAC,cAAc,CAAC;aACvB,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAClD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,eAAe;gBACrB,IAAI,EAAE,CAAC,cAAc,CAAC;gBACtB,MAAM,EAAE,QAAQ;aACjB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAW,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YAEjE,MAAM,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,eAAe,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC/F,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAClD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,eAAe,EAAE,OAAO;gBACxB,UAAU,EAAE,QAAQ;aACrB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAY,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAE1D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,oBAAoB,CACnD,MAAM,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAC3C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,MAAc,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAE5D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,uBAAuB,EAAE,EAAE,CAAC;YACtE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,GAAW,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAE7D,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC;gBAC9C,QAAQ,EAAE,uBAAuB;gBACjC,OAAO,EAAE,2BAA2B;gBACpC,IAAI,EAAE,CAAC,cAAc,CAAC;aACvB,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,oBAAoB,CACtD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,QAAQ,EAAE,uBAAuB;gBACjC,OAAO,EAAE,2BAA2B;gBACpC,IAAI,EAAE,CAAC,cAAc,CAAC;aACvB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,MAAc,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAC9E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CACrD,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,sCAAsC,EAAE,EAAE,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,OAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEvD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC;gBACpD,OAAO,EAAE,8BAA8B;gBACvC,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;YAC5E,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAChD,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,8BAA8B,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,CACvF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAe,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;YAE5E,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAC3E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,CAAC,MAAM,CAAC,OAAe,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YAE9E,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,kBAAkB,CAAC,OAAO,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAC3E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,KAAa,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAErD,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"projects.test.d.ts","sourceRoot":"","sources":["../../../src/tools/__tests__/projects.test.ts"],"names":[],"mappings":""}
|