daedalus-cli 0.5.24 → 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/AGENTS.md +12 -2
- package/CHANGELOG.md +54 -0
- package/CONTRIBUTING.md +3 -2
- package/README.md +4 -3
- package/dist/agents/orchestrator.js +7 -5
- package/dist/agents/orchestrator.js.map +1 -1
- package/dist/agents/orchestrator.test.d.ts +2 -0
- package/dist/agents/orchestrator.test.d.ts.map +1 -0
- package/dist/agents/orchestrator.test.js +48 -0
- package/dist/agents/orchestrator.test.js.map +1 -0
- package/dist/agents/roles.test.d.ts +2 -0
- package/dist/agents/roles.test.d.ts.map +1 -0
- package/dist/agents/roles.test.js +108 -0
- package/dist/agents/roles.test.js.map +1 -0
- package/dist/config/index.test.d.ts +2 -0
- package/dist/config/index.test.d.ts.map +1 -0
- package/dist/config/index.test.js +220 -0
- package/dist/config/index.test.js.map +1 -0
- package/dist/extraction.test.d.ts +2 -0
- package/dist/extraction.test.d.ts.map +1 -0
- package/dist/extraction.test.js +105 -0
- package/dist/extraction.test.js.map +1 -0
- package/dist/highlight.test.d.ts +2 -0
- package/dist/highlight.test.d.ts.map +1 -0
- package/dist/highlight.test.js +56 -0
- package/dist/highlight.test.js.map +1 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +19 -2
- package/dist/index.js.map +1 -1
- package/dist/indexing/fts.test.d.ts +2 -0
- package/dist/indexing/fts.test.d.ts.map +1 -0
- package/dist/indexing/fts.test.js +104 -0
- package/dist/indexing/fts.test.js.map +1 -0
- package/dist/indexing/indexer.d.ts +21 -0
- package/dist/indexing/indexer.d.ts.map +1 -1
- package/dist/indexing/indexer.js +4 -4
- package/dist/indexing/indexer.js.map +1 -1
- package/dist/indexing/indexer.test.d.ts +2 -0
- package/dist/indexing/indexer.test.d.ts.map +1 -0
- package/dist/indexing/indexer.test.js +192 -0
- package/dist/indexing/indexer.test.js.map +1 -0
- package/dist/profile.d.ts.map +1 -1
- package/dist/profile.js +12 -7
- package/dist/profile.js.map +1 -1
- package/dist/profile.test.d.ts +2 -0
- package/dist/profile.test.d.ts.map +1 -0
- package/dist/profile.test.js +70 -0
- package/dist/profile.test.js.map +1 -0
- package/dist/router/health.test.d.ts +2 -0
- package/dist/router/health.test.d.ts.map +1 -0
- package/dist/router/health.test.js +87 -0
- package/dist/router/health.test.js.map +1 -0
- package/dist/router/index.test.d.ts +2 -0
- package/dist/router/index.test.d.ts.map +1 -0
- package/dist/router/index.test.js +136 -0
- package/dist/router/index.test.js.map +1 -0
- package/dist/router/rate-limiter.test.d.ts +2 -0
- package/dist/router/rate-limiter.test.d.ts.map +1 -0
- package/dist/router/rate-limiter.test.js +90 -0
- package/dist/router/rate-limiter.test.js.map +1 -0
- package/dist/session/jsonl.d.ts +0 -2
- package/dist/session/jsonl.d.ts.map +1 -1
- package/dist/session/jsonl.js +3 -7
- package/dist/session/jsonl.js.map +1 -1
- package/dist/session/jsonl.test.d.ts +2 -0
- package/dist/session/jsonl.test.d.ts.map +1 -0
- package/dist/session/jsonl.test.js +88 -0
- package/dist/session/jsonl.test.js.map +1 -0
- package/dist/session/manager.d.ts +2 -0
- package/dist/session/manager.d.ts.map +1 -1
- package/dist/session/manager.js +14 -0
- package/dist/session/manager.js.map +1 -1
- package/dist/session/manager.test.d.ts +2 -0
- package/dist/session/manager.test.d.ts.map +1 -0
- package/dist/session/manager.test.js +120 -0
- package/dist/session/manager.test.js.map +1 -0
- package/dist/session/memory.test.d.ts +2 -0
- package/dist/session/memory.test.d.ts.map +1 -0
- package/dist/session/memory.test.js +100 -0
- package/dist/session/memory.test.js.map +1 -0
- package/dist/session/sqlite.test.d.ts +2 -0
- package/dist/session/sqlite.test.d.ts.map +1 -0
- package/dist/session/sqlite.test.js +151 -0
- package/dist/session/sqlite.test.js.map +1 -0
- package/dist/tools/builtin/delegation.test.d.ts +2 -0
- package/dist/tools/builtin/delegation.test.d.ts.map +1 -0
- package/dist/tools/builtin/delegation.test.js +87 -0
- package/dist/tools/builtin/delegation.test.js.map +1 -0
- package/dist/tools/builtin/eval.d.ts +7 -0
- package/dist/tools/builtin/eval.d.ts.map +1 -0
- package/dist/tools/builtin/eval.js +92 -0
- package/dist/tools/builtin/eval.js.map +1 -0
- package/dist/tools/builtin/eval.test.d.ts +2 -0
- package/dist/tools/builtin/eval.test.d.ts.map +1 -0
- package/dist/tools/builtin/eval.test.js +24 -0
- package/dist/tools/builtin/eval.test.js.map +1 -0
- package/dist/tools/builtin/git.test.d.ts +2 -0
- package/dist/tools/builtin/git.test.d.ts.map +1 -0
- package/dist/tools/builtin/git.test.js +49 -0
- package/dist/tools/builtin/git.test.js.map +1 -0
- package/dist/tools/builtin/impact.d.ts +5 -0
- package/dist/tools/builtin/impact.d.ts.map +1 -0
- package/dist/tools/builtin/impact.js +129 -0
- package/dist/tools/builtin/impact.js.map +1 -0
- package/dist/tools/builtin/impact.test.d.ts +2 -0
- package/dist/tools/builtin/impact.test.d.ts.map +1 -0
- package/dist/tools/builtin/impact.test.js +50 -0
- package/dist/tools/builtin/impact.test.js.map +1 -0
- package/dist/tools/builtin/indexing.test.d.ts +2 -0
- package/dist/tools/builtin/indexing.test.d.ts.map +1 -0
- package/dist/tools/builtin/indexing.test.js +30 -0
- package/dist/tools/builtin/indexing.test.js.map +1 -0
- package/dist/tools/builtin/lsp.d.ts +16 -0
- package/dist/tools/builtin/lsp.d.ts.map +1 -0
- package/dist/tools/builtin/lsp.js +143 -0
- package/dist/tools/builtin/lsp.js.map +1 -0
- package/dist/tools/builtin/lsp.test.d.ts +2 -0
- package/dist/tools/builtin/lsp.test.d.ts.map +1 -0
- package/dist/tools/builtin/lsp.test.js +34 -0
- package/dist/tools/builtin/lsp.test.js.map +1 -0
- package/dist/tools/builtin/process-watcher.d.ts +14 -0
- package/dist/tools/builtin/process-watcher.d.ts.map +1 -0
- package/dist/tools/builtin/process-watcher.js +87 -0
- package/dist/tools/builtin/process-watcher.js.map +1 -0
- package/dist/tools/builtin/process-watcher.test.d.ts +2 -0
- package/dist/tools/builtin/process-watcher.test.d.ts.map +1 -0
- package/dist/tools/builtin/process-watcher.test.js +53 -0
- package/dist/tools/builtin/process-watcher.test.js.map +1 -0
- package/dist/tools/builtin/project-config.test.d.ts +2 -0
- package/dist/tools/builtin/project-config.test.d.ts.map +1 -0
- package/dist/tools/builtin/project-config.test.js +81 -0
- package/dist/tools/builtin/project-config.test.js.map +1 -0
- package/dist/tools/builtin/screenshot.d.ts +8 -0
- package/dist/tools/builtin/screenshot.d.ts.map +1 -0
- package/dist/tools/builtin/screenshot.js +76 -0
- package/dist/tools/builtin/screenshot.js.map +1 -0
- package/dist/tools/builtin/todo.test.d.ts +2 -0
- package/dist/tools/builtin/todo.test.d.ts.map +1 -0
- package/dist/tools/builtin/todo.test.js +66 -0
- package/dist/tools/builtin/todo.test.js.map +1 -0
- package/dist/tools/builtin/web.test.d.ts +2 -0
- package/dist/tools/builtin/web.test.d.ts.map +1 -0
- package/dist/tools/builtin/web.test.js +78 -0
- package/dist/tools/builtin/web.test.js.map +1 -0
- package/dist/tools/definitions.d.ts +1 -0
- package/dist/tools/definitions.d.ts.map +1 -1
- package/dist/tools/definitions.js +157 -0
- package/dist/tools/definitions.js.map +1 -1
- package/dist/tools/executor.test.d.ts +2 -0
- package/dist/tools/executor.test.d.ts.map +1 -0
- package/dist/tools/executor.test.js +79 -0
- package/dist/tools/executor.test.js.map +1 -0
- package/dist/tools/mcp/registry.test.d.ts +2 -0
- package/dist/tools/mcp/registry.test.d.ts.map +1 -0
- package/dist/tools/mcp/registry.test.js +18 -0
- package/dist/tools/mcp/registry.test.js.map +1 -0
- package/package.json +12 -1
- package/dist/manual-test.d.ts +0 -2
- package/dist/manual-test.d.ts.map +0 -1
- package/dist/manual-test.js.map +0 -1
|
@@ -0,0 +1,120 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
|
|
2
|
+
import fs from 'fs';
|
|
3
|
+
import path from 'path';
|
|
4
|
+
import os from 'os';
|
|
5
|
+
import { SessionManager } from './manager.js';
|
|
6
|
+
describe('SessionManager', () => {
|
|
7
|
+
let tmpDir;
|
|
8
|
+
let manager;
|
|
9
|
+
beforeEach(() => {
|
|
10
|
+
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'daedalus-manager-test-'));
|
|
11
|
+
process.chdir(tmpDir);
|
|
12
|
+
vi.stubEnv('HOME', tmpDir);
|
|
13
|
+
vi.stubEnv('USERPROFILE', tmpDir);
|
|
14
|
+
manager = new SessionManager(tmpDir);
|
|
15
|
+
});
|
|
16
|
+
afterEach(() => {
|
|
17
|
+
manager.close();
|
|
18
|
+
manager = null;
|
|
19
|
+
process.chdir(os.tmpdir());
|
|
20
|
+
for (let i = 0; i < 5; i++) {
|
|
21
|
+
try {
|
|
22
|
+
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
23
|
+
break;
|
|
24
|
+
}
|
|
25
|
+
catch { }
|
|
26
|
+
}
|
|
27
|
+
vi.unstubAllEnvs();
|
|
28
|
+
});
|
|
29
|
+
it('init creates directories and index database', () => {
|
|
30
|
+
manager.init();
|
|
31
|
+
expect(fs.existsSync(path.join(tmpDir, '.daedalus', 'sessions'))).toBe(true);
|
|
32
|
+
expect(fs.existsSync(path.join(tmpDir, '.daedalus', 'memory'))).toBe(true);
|
|
33
|
+
expect(fs.existsSync(path.join(tmpDir, '.daedalus', 'sessions', 'index.sqlite'))).toBe(true);
|
|
34
|
+
});
|
|
35
|
+
it('startSession creates a new session with id and title', () => {
|
|
36
|
+
manager.init();
|
|
37
|
+
const session = manager.startSession();
|
|
38
|
+
expect(session.sessionId).toBeTruthy();
|
|
39
|
+
expect(session.turns).toEqual([]);
|
|
40
|
+
expect(session.activeFiles.size).toBe(0);
|
|
41
|
+
expect(session.todos).toEqual([]);
|
|
42
|
+
});
|
|
43
|
+
it('startSession loads existing session when sessionId is provided', () => {
|
|
44
|
+
manager.init();
|
|
45
|
+
const created = manager.startSession();
|
|
46
|
+
const loaded = manager.startSession(created.sessionId);
|
|
47
|
+
expect(loaded.sessionId).toBe(created.sessionId);
|
|
48
|
+
});
|
|
49
|
+
it('saveSessionState persists messages, files, and todos', () => {
|
|
50
|
+
manager.init();
|
|
51
|
+
const session = manager.startSession();
|
|
52
|
+
const messages = [
|
|
53
|
+
{ role: 'system', content: 'sys prompt' },
|
|
54
|
+
{ role: 'user', content: 'hello' },
|
|
55
|
+
];
|
|
56
|
+
const activeFiles = new Map([['/test/file.ts', 'file.ts']]);
|
|
57
|
+
const todos = [{ id: '1', content: 'test', status: 'pending' }];
|
|
58
|
+
manager.saveSessionState(messages, activeFiles, todos);
|
|
59
|
+
const loaded = manager.startSession(session.sessionId);
|
|
60
|
+
expect(loaded.turns).toHaveLength(2);
|
|
61
|
+
expect(loaded.activeFiles.size).toBe(1);
|
|
62
|
+
expect(loaded.todos).toHaveLength(1);
|
|
63
|
+
});
|
|
64
|
+
it('getSessionsForProject returns session list', () => {
|
|
65
|
+
manager.init();
|
|
66
|
+
manager.startSession();
|
|
67
|
+
const sessions = manager.getSessionsForProject();
|
|
68
|
+
expect(sessions.length).toBeGreaterThan(0);
|
|
69
|
+
});
|
|
70
|
+
it('getChatMessages returns formatted chat messages', () => {
|
|
71
|
+
manager.init();
|
|
72
|
+
manager.startSession();
|
|
73
|
+
manager.saveSessionState([
|
|
74
|
+
{ role: 'user', content: 'hi' },
|
|
75
|
+
{ role: 'assistant', content: 'hello', tool_calls: [{ id: '1', type: 'function', function: { name: 'test', arguments: '{}' } }] },
|
|
76
|
+
], new Map(), []);
|
|
77
|
+
const msgs = manager.getChatMessages();
|
|
78
|
+
expect(msgs).toHaveLength(2);
|
|
79
|
+
expect(msgs[0].role).toBe('user');
|
|
80
|
+
});
|
|
81
|
+
it('deleteSession removes db and jsonl files', () => {
|
|
82
|
+
manager.init();
|
|
83
|
+
const session = manager.startSession();
|
|
84
|
+
manager.saveSessionState([{ role: 'user', content: 'x' }], new Map(), []);
|
|
85
|
+
manager.deleteSession(session.sessionId);
|
|
86
|
+
const serviceDir = path.join(tmpDir, '.daedalus', 'sessions', manager.projectHash);
|
|
87
|
+
expect(fs.existsSync(path.join(serviceDir, `${session.sessionId}.sqlite`))).toBe(false);
|
|
88
|
+
});
|
|
89
|
+
it('addFact and setConvention delegate to memory module', () => {
|
|
90
|
+
manager.init();
|
|
91
|
+
manager.startSession();
|
|
92
|
+
manager.addFact('framework', 'vitest');
|
|
93
|
+
manager.setConvention('style', 'no comments');
|
|
94
|
+
const mem = manager.loadMemory();
|
|
95
|
+
expect(mem.facts.length).toBeGreaterThan(0);
|
|
96
|
+
expect(mem.conventions.style).toBe('no comments');
|
|
97
|
+
});
|
|
98
|
+
it('getMemoryPrompt returns non-empty after adding facts', () => {
|
|
99
|
+
manager.init();
|
|
100
|
+
manager.startSession();
|
|
101
|
+
manager.addFact('lang', 'typescript');
|
|
102
|
+
const prompt = manager.getMemoryPrompt();
|
|
103
|
+
expect(prompt).toContain('typescript');
|
|
104
|
+
});
|
|
105
|
+
it('importSessionFromJsonl loads turns from JSONL', () => {
|
|
106
|
+
manager.init();
|
|
107
|
+
manager.startSession();
|
|
108
|
+
const jsonlPath = path.join(tmpDir, 'test.jsonl');
|
|
109
|
+
fs.writeFileSync(jsonlPath, JSON.stringify({ turn: 1, role: 'user', content: 'imported' }) + '\n', 'utf8');
|
|
110
|
+
const turns = manager.importSessionFromJsonl(jsonlPath);
|
|
111
|
+
expect(turns).toHaveLength(1);
|
|
112
|
+
expect(turns[0].content).toBe('imported');
|
|
113
|
+
});
|
|
114
|
+
it('importSessionFromJsonl throws without active session', () => {
|
|
115
|
+
const freshManager = new SessionManager(tmpDir);
|
|
116
|
+
freshManager.init();
|
|
117
|
+
expect(() => freshManager.importSessionFromJsonl('test.jsonl')).toThrow('No active session');
|
|
118
|
+
});
|
|
119
|
+
});
|
|
120
|
+
//# sourceMappingURL=manager.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"manager.test.js","sourceRoot":"","sources":["../../src/session/manager.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAE9C,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,MAAc,CAAC;IACnB,IAAI,OAAuB,CAAC;IAE5B,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,wBAAwB,CAAC,CAAC,CAAC;QAC1E,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACtB,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC3B,EAAE,CAAC,OAAO,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QAClC,OAAO,GAAG,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,KAAK,EAAE,CAAC;QAChB,OAAO,GAAG,IAAK,CAAC;QAChB,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC;QAC3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3B,IAAI,CAAC;gBAAC,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBAAC,MAAM;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,CAAC;QAC/E,CAAC;QACD,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,OAAO,CAAC,IAAI,EAAE,CAAC;QACf,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7E,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3E,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,OAAO,CAAC,IAAI,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,UAAU,EAAE,CAAC;QACvC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,OAAO,CAAC,IAAI,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,OAAO,CAAC,IAAI,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;QAEvC,MAAM,QAAQ,GAAG;YACf,EAAE,IAAI,EAAE,QAAiB,EAAE,OAAO,EAAE,YAAY,EAAE;YAClD,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,OAAO,EAAE;SAC5C,CAAC;QACF,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;QAC5D,MAAM,KAAK,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC;QAEhE,OAAO,CAAC,gBAAgB,CAAC,QAAQ,EAAE,WAAW,EAAE,KAAY,CAAC,CAAC;QAE9D,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,OAAO,CAAC,IAAI,EAAE,CAAC;QACf,OAAO,CAAC,YAAY,EAAE,CAAC;QACvB,MAAM,QAAQ,GAAG,OAAO,CAAC,qBAAqB,EAAE,CAAC;QACjD,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,OAAO,CAAC,IAAI,EAAE,CAAC;QACf,OAAO,CAAC,YAAY,EAAE,CAAC;QAEvB,OAAO,CAAC,gBAAgB,CACtB;YACE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE;YAC/B,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE;SAClI,EACD,IAAI,GAAG,EAAE,EACT,EAAE,CACH,CAAC;QAEF,MAAM,IAAI,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,OAAO,CAAC,IAAI,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;QACvC,OAAO,CAAC,gBAAgB,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,IAAI,GAAG,EAAE,EAAE,EAAS,CAAC,CAAC;QACjF,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAEzC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;QACnF,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,OAAO,CAAC,SAAS,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,OAAO,CAAC,IAAI,EAAE,CAAC;QACf,OAAO,CAAC,YAAY,EAAE,CAAC;QACvB,OAAO,CAAC,OAAO,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QACvC,OAAO,CAAC,aAAa,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;QAC9C,MAAM,GAAG,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC;QACjC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,OAAO,CAAC,IAAI,EAAE,CAAC;QACf,OAAO,CAAC,YAAY,EAAE,CAAC;QACvB,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QACtC,MAAM,MAAM,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,OAAO,CAAC,IAAI,EAAE,CAAC;QACf,OAAO,CAAC,YAAY,EAAE,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAClD,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,GAAG,IAAI,EAAE,MAAM,CAAC,CAAC;QAC3G,MAAM,KAAK,GAAG,OAAO,CAAC,sBAAsB,CAAC,SAAS,CAAC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,YAAY,GAAG,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;QAChD,YAAY,CAAC,IAAI,EAAE,CAAC;QACpB,MAAM,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,sBAAsB,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;IAC/F,CAAC,CAAC,CAAC;AAEL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.test.d.ts","sourceRoot":"","sources":["../../src/session/memory.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,100 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import fs from 'fs';
|
|
3
|
+
import path from 'path';
|
|
4
|
+
import os from 'os';
|
|
5
|
+
import { loadMemory, saveMemory, addFact, setConvention, getMemoryAsPrompt, } from './memory.js';
|
|
6
|
+
function makeTempDir() {
|
|
7
|
+
return fs.mkdtempSync(path.join(os.tmpdir(), 'daedalus-memory-test-'));
|
|
8
|
+
}
|
|
9
|
+
describe('Project memory', () => {
|
|
10
|
+
let tmpDir;
|
|
11
|
+
let memoryPath;
|
|
12
|
+
const projectHash = 'testhash123';
|
|
13
|
+
beforeEach(() => {
|
|
14
|
+
tmpDir = makeTempDir();
|
|
15
|
+
memoryPath = path.join(tmpDir, 'memory.json');
|
|
16
|
+
});
|
|
17
|
+
afterEach(() => {
|
|
18
|
+
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
19
|
+
});
|
|
20
|
+
it('loadMemory returns empty memory when no file exists', () => {
|
|
21
|
+
const mem = loadMemory(memoryPath, projectHash);
|
|
22
|
+
expect(mem.projectHash).toBe(projectHash);
|
|
23
|
+
expect(mem.facts).toEqual([]);
|
|
24
|
+
expect(mem.conventions).toEqual({});
|
|
25
|
+
});
|
|
26
|
+
it('loadMemory returns empty memory for corrupt file', () => {
|
|
27
|
+
fs.writeFileSync(memoryPath, '{ broken json', 'utf8');
|
|
28
|
+
const mem = loadMemory(memoryPath, projectHash);
|
|
29
|
+
expect(mem.facts).toEqual([]);
|
|
30
|
+
});
|
|
31
|
+
it('loadMemory reads existing memory file', () => {
|
|
32
|
+
const existing = {
|
|
33
|
+
projectHash,
|
|
34
|
+
facts: [{ key: 'framework', value: 'vitest', source: 'user', created: 100 }],
|
|
35
|
+
conventions: { 'no comments': 'true' },
|
|
36
|
+
updated_at: 100,
|
|
37
|
+
};
|
|
38
|
+
saveMemory(memoryPath, existing);
|
|
39
|
+
const loaded = loadMemory(memoryPath, projectHash);
|
|
40
|
+
expect(loaded.facts).toHaveLength(1);
|
|
41
|
+
expect(loaded.facts[0].key).toBe('framework');
|
|
42
|
+
expect(loaded.conventions['no comments']).toBe('true');
|
|
43
|
+
});
|
|
44
|
+
it('addFact creates a new fact', () => {
|
|
45
|
+
addFact(memoryPath, projectHash, 'test-framework', 'vitest', 'user');
|
|
46
|
+
const mem = loadMemory(memoryPath, projectHash);
|
|
47
|
+
expect(mem.facts).toHaveLength(1);
|
|
48
|
+
expect(mem.facts[0].key).toBe('test-framework');
|
|
49
|
+
expect(mem.facts[0].value).toBe('vitest');
|
|
50
|
+
expect(mem.facts[0].source).toBe('user');
|
|
51
|
+
});
|
|
52
|
+
it('addFact updates existing fact with same key', () => {
|
|
53
|
+
addFact(memoryPath, projectHash, 'key', 'old', 'user');
|
|
54
|
+
addFact(memoryPath, projectHash, 'key', 'new', 'agent');
|
|
55
|
+
const mem = loadMemory(memoryPath, projectHash);
|
|
56
|
+
expect(mem.facts).toHaveLength(1);
|
|
57
|
+
expect(mem.facts[0].value).toBe('new');
|
|
58
|
+
expect(mem.facts[0].source).toBe('agent');
|
|
59
|
+
});
|
|
60
|
+
it('addFact accumulates multiple facts', () => {
|
|
61
|
+
addFact(memoryPath, projectHash, 'a', '1', 'user');
|
|
62
|
+
addFact(memoryPath, projectHash, 'b', '2', 'agent');
|
|
63
|
+
addFact(memoryPath, projectHash, 'c', '3', 'user');
|
|
64
|
+
const mem = loadMemory(memoryPath, projectHash);
|
|
65
|
+
expect(mem.facts).toHaveLength(3);
|
|
66
|
+
});
|
|
67
|
+
it('setConvention creates or updates a convention', () => {
|
|
68
|
+
setConvention(memoryPath, projectHash, 'style', 'no comments');
|
|
69
|
+
setConvention(memoryPath, projectHash, 'lint', 'strict');
|
|
70
|
+
const mem = loadMemory(memoryPath, projectHash);
|
|
71
|
+
expect(mem.conventions.style).toBe('no comments');
|
|
72
|
+
expect(mem.conventions.lint).toBe('strict');
|
|
73
|
+
});
|
|
74
|
+
it('setConvention overwrites existing convention', () => {
|
|
75
|
+
setConvention(memoryPath, projectHash, 'style', 'tabs');
|
|
76
|
+
setConvention(memoryPath, projectHash, 'style', 'spaces');
|
|
77
|
+
const mem = loadMemory(memoryPath, projectHash);
|
|
78
|
+
expect(mem.conventions.style).toBe('spaces');
|
|
79
|
+
});
|
|
80
|
+
it('getMemoryAsPrompt returns empty string for empty memory', () => {
|
|
81
|
+
const prompt = getMemoryAsPrompt(memoryPath, projectHash);
|
|
82
|
+
expect(prompt).toBe('');
|
|
83
|
+
});
|
|
84
|
+
it('getMemoryAsPrompt includes conventions and facts', () => {
|
|
85
|
+
setConvention(memoryPath, projectHash, 'style', 'no comments');
|
|
86
|
+
addFact(memoryPath, projectHash, 'framework', 'vitest', 'agent');
|
|
87
|
+
const prompt = getMemoryAsPrompt(memoryPath, projectHash);
|
|
88
|
+
expect(prompt).toContain('no comments');
|
|
89
|
+
expect(prompt).toContain('vitest');
|
|
90
|
+
expect(prompt).toContain('PROJECT CONTEXT');
|
|
91
|
+
});
|
|
92
|
+
it('getMemoryAsPrompt includes only conventions when no facts', () => {
|
|
93
|
+
setConvention(memoryPath, projectHash, 'lint', 'strict');
|
|
94
|
+
const prompt = getMemoryAsPrompt(memoryPath, projectHash);
|
|
95
|
+
expect(prompt).toContain('strict');
|
|
96
|
+
expect(prompt).toContain('Conventions');
|
|
97
|
+
expect(prompt).not.toContain('Facts');
|
|
98
|
+
});
|
|
99
|
+
});
|
|
100
|
+
//# sourceMappingURL=memory.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.test.js","sourceRoot":"","sources":["../../src/session/memory.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EACL,UAAU,EACV,UAAU,EACV,OAAO,EACP,aAAa,EACb,iBAAiB,GAElB,MAAM,aAAa,CAAC;AAErB,SAAS,WAAW;IAClB,OAAO,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAC;AACzE,CAAC;AAED,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,MAAc,CAAC;IACnB,IAAI,UAAkB,CAAC;IACvB,MAAM,WAAW,GAAG,aAAa,CAAC;IAElC,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,WAAW,EAAE,CAAC;QACvB,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,GAAG,GAAG,UAAU,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1C,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9B,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,eAAe,EAAE,MAAM,CAAC,CAAC;QACtD,MAAM,GAAG,GAAG,UAAU,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,QAAQ,GAAkB;YAC9B,WAAW;YACX,KAAK,EAAE,CAAC,EAAE,GAAG,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;YAC5E,WAAW,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;YACtC,UAAU,EAAE,GAAG;SAChB,CAAC;QACF,UAAU,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAEjC,MAAM,MAAM,GAAG,UAAU,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,OAAO,CAAC,UAAU,EAAE,WAAW,EAAE,gBAAgB,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;QACrE,MAAM,GAAG,GAAG,UAAU,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,OAAO,CAAC,UAAU,EAAE,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;QACvD,OAAO,CAAC,UAAU,EAAE,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;QACxD,MAAM,GAAG,GAAG,UAAU,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,OAAO,CAAC,UAAU,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACnD,OAAO,CAAC,UAAU,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QACpD,OAAO,CAAC,UAAU,EAAE,WAAW,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACnD,MAAM,GAAG,GAAG,UAAU,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,aAAa,CAAC,UAAU,EAAE,WAAW,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;QAC/D,aAAa,CAAC,UAAU,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACzD,MAAM,GAAG,GAAG,UAAU,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAClD,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,aAAa,CAAC,UAAU,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QACxD,aAAa,CAAC,UAAU,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;QAC1D,MAAM,GAAG,GAAG,UAAU,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAChD,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,MAAM,GAAG,iBAAiB,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,aAAa,CAAC,UAAU,EAAE,WAAW,EAAE,OAAO,EAAE,aAAa,CAAC,CAAC;QAC/D,OAAO,CAAC,UAAU,EAAE,WAAW,EAAE,WAAW,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;QACjE,MAAM,MAAM,GAAG,iBAAiB,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,aAAa,CAAC,UAAU,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACzD,MAAM,MAAM,GAAG,iBAAiB,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;AAEL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sqlite.test.d.ts","sourceRoot":"","sources":["../../src/session/sqlite.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,151 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import fs from 'fs';
|
|
3
|
+
import path from 'path';
|
|
4
|
+
import os from 'os';
|
|
5
|
+
import { initIndexDb, initSessionDb, registerSession, listSessionsForProject, deleteSessionFromIndex, saveTurn, getTurns, clearTurns, saveActiveFiles, getActiveFiles, saveTodos, getTodos, saveState, getState, } from './sqlite.js';
|
|
6
|
+
describe('SQLite session database (index)', () => {
|
|
7
|
+
let db;
|
|
8
|
+
let dbPath;
|
|
9
|
+
beforeEach(() => {
|
|
10
|
+
const dir = fs.mkdtempSync(path.join(os.tmpdir(), 'daedalus-sqlite-test-'));
|
|
11
|
+
dbPath = path.join(dir, 'index.sqlite');
|
|
12
|
+
db = initIndexDb(dbPath);
|
|
13
|
+
});
|
|
14
|
+
afterEach(() => {
|
|
15
|
+
db.close();
|
|
16
|
+
fs.rmSync(path.dirname(dbPath), { recursive: true, force: true });
|
|
17
|
+
});
|
|
18
|
+
it('creates sessions table on init', () => {
|
|
19
|
+
const result = db.prepare("SELECT name FROM sqlite_master WHERE type='table' AND name='sessions'").get();
|
|
20
|
+
expect(result).toBeTruthy();
|
|
21
|
+
});
|
|
22
|
+
it('registers a session', () => {
|
|
23
|
+
registerSession(db, {
|
|
24
|
+
id: 'test-1', project_path: '/test', project_hash: 'abc123',
|
|
25
|
+
title: 'Test Session', created_at: 1000, updated_at: 1000,
|
|
26
|
+
});
|
|
27
|
+
const rows = db.prepare('SELECT * FROM sessions').all();
|
|
28
|
+
expect(rows).toHaveLength(1);
|
|
29
|
+
});
|
|
30
|
+
it('upserts session on duplicate id', () => {
|
|
31
|
+
registerSession(db, {
|
|
32
|
+
id: 'test-1', project_path: '/test', project_hash: 'abc123',
|
|
33
|
+
title: 'Original', created_at: 1000, updated_at: 1000,
|
|
34
|
+
});
|
|
35
|
+
registerSession(db, {
|
|
36
|
+
id: 'test-1', project_path: '/test', project_hash: 'abc123',
|
|
37
|
+
title: 'Updated', created_at: 1000, updated_at: 2000,
|
|
38
|
+
});
|
|
39
|
+
const rows = db.prepare('SELECT * FROM sessions').all();
|
|
40
|
+
expect(rows).toHaveLength(1);
|
|
41
|
+
expect(rows[0].title).toBe('Updated');
|
|
42
|
+
});
|
|
43
|
+
it('lists sessions for a project ordered by updated_at desc', () => {
|
|
44
|
+
registerSession(db, { id: 'a', project_path: '/p', project_hash: 'abc', title: 'A', created_at: 1, updated_at: 3 });
|
|
45
|
+
registerSession(db, { id: 'b', project_path: '/p', project_hash: 'abc', title: 'B', created_at: 2, updated_at: 1 });
|
|
46
|
+
registerSession(db, { id: 'c', project_path: '/p2', project_hash: 'xyz', title: 'C', created_at: 3, updated_at: 3 });
|
|
47
|
+
const result = listSessionsForProject(db, 'abc');
|
|
48
|
+
expect(result).toHaveLength(2);
|
|
49
|
+
expect(result[0].id).toBe('a');
|
|
50
|
+
expect(result[1].id).toBe('b');
|
|
51
|
+
});
|
|
52
|
+
it('deletes a session from index', () => {
|
|
53
|
+
registerSession(db, { id: 'del', project_path: '/p', project_hash: 'abc', title: 'Del', created_at: 1, updated_at: 1 });
|
|
54
|
+
deleteSessionFromIndex(db, 'del');
|
|
55
|
+
expect(listSessionsForProject(db, 'abc')).toHaveLength(0);
|
|
56
|
+
});
|
|
57
|
+
});
|
|
58
|
+
describe('SQLite session database (session)', () => {
|
|
59
|
+
let db;
|
|
60
|
+
let dbPath;
|
|
61
|
+
beforeEach(() => {
|
|
62
|
+
const dir = fs.mkdtempSync(path.join(os.tmpdir(), 'daedalus-sqlite-test-'));
|
|
63
|
+
dbPath = path.join(dir, 'session.sqlite');
|
|
64
|
+
db = initSessionDb(dbPath);
|
|
65
|
+
});
|
|
66
|
+
afterEach(() => {
|
|
67
|
+
db.close();
|
|
68
|
+
fs.rmSync(path.dirname(dbPath), { recursive: true, force: true });
|
|
69
|
+
});
|
|
70
|
+
it('creates all tables on init', () => {
|
|
71
|
+
const tables = db.prepare("SELECT name FROM sqlite_master WHERE type='table'").all().map((r) => r.name);
|
|
72
|
+
expect(tables).toContain('turns');
|
|
73
|
+
expect(tables).toContain('agent_state');
|
|
74
|
+
expect(tables).toContain('active_files');
|
|
75
|
+
expect(tables).toContain('todos');
|
|
76
|
+
});
|
|
77
|
+
it('saves and retrieves turns', () => {
|
|
78
|
+
saveTurn(db, { role: 'user', content: 'hello' });
|
|
79
|
+
saveTurn(db, { role: 'assistant', content: 'world' });
|
|
80
|
+
const turns = getTurns(db);
|
|
81
|
+
expect(turns).toHaveLength(2);
|
|
82
|
+
expect(turns[0].role).toBe('user');
|
|
83
|
+
expect(turns[1].content).toBe('world');
|
|
84
|
+
});
|
|
85
|
+
it('clearTurns removes all turns', () => {
|
|
86
|
+
saveTurn(db, { role: 'user', content: 'test' });
|
|
87
|
+
clearTurns(db);
|
|
88
|
+
expect(getTurns(db)).toHaveLength(0);
|
|
89
|
+
});
|
|
90
|
+
it('saves and retrieves active files', () => {
|
|
91
|
+
const files = new Map([
|
|
92
|
+
['/path/to/file.ts', 'file.ts'],
|
|
93
|
+
['/path/to/other.ts', 'other.ts'],
|
|
94
|
+
]);
|
|
95
|
+
saveActiveFiles(db, files);
|
|
96
|
+
const loaded = getActiveFiles(db);
|
|
97
|
+
expect(loaded.size).toBe(2);
|
|
98
|
+
expect(loaded.get('/path/to/file.ts')).toBe('file.ts');
|
|
99
|
+
});
|
|
100
|
+
it('replaces active files on save', () => {
|
|
101
|
+
saveActiveFiles(db, new Map([['/a.ts', 'a.ts']]));
|
|
102
|
+
saveActiveFiles(db, new Map([['/b.ts', 'b.ts']]));
|
|
103
|
+
expect(getActiveFiles(db).size).toBe(1);
|
|
104
|
+
expect(getActiveFiles(db).get('/a.ts')).toBeUndefined();
|
|
105
|
+
});
|
|
106
|
+
it('saves and retrieves todos', () => {
|
|
107
|
+
const todos = [
|
|
108
|
+
{ id: '1', content: 'Task 1', status: 'pending', created_at: 100, updated_at: 100 },
|
|
109
|
+
{ id: '2', content: 'Task 2', status: 'completed', created_at: 200, updated_at: 200 },
|
|
110
|
+
];
|
|
111
|
+
saveTodos(db, todos);
|
|
112
|
+
const loaded = getTodos(db);
|
|
113
|
+
expect(loaded).toHaveLength(2);
|
|
114
|
+
expect(loaded[0].status).toBe('pending');
|
|
115
|
+
});
|
|
116
|
+
it('replaces all todos on save', () => {
|
|
117
|
+
saveTodos(db, [{ id: '1', content: 'old', status: 'pending', created_at: 1, updated_at: 1 }]);
|
|
118
|
+
saveTodos(db, [{ id: '2', content: 'new', status: 'completed', created_at: 2, updated_at: 2 }]);
|
|
119
|
+
expect(getTodos(db)).toHaveLength(1);
|
|
120
|
+
});
|
|
121
|
+
it('saves and retrieves agent state', () => {
|
|
122
|
+
saveState(db, 'myKey', { nested: { value: 42 } });
|
|
123
|
+
expect(getState(db, 'myKey')).toEqual({ nested: { value: 42 } });
|
|
124
|
+
});
|
|
125
|
+
it('returns null for missing state key', () => {
|
|
126
|
+
expect(getState(db, 'nonexistent')).toBeNull();
|
|
127
|
+
});
|
|
128
|
+
it('overwrites existing state key', () => {
|
|
129
|
+
saveState(db, 'key', 'original');
|
|
130
|
+
saveState(db, 'key', 'updated');
|
|
131
|
+
expect(getState(db, 'key')).toBe('updated');
|
|
132
|
+
});
|
|
133
|
+
it('saves turn with all optional fields', () => {
|
|
134
|
+
saveTurn(db, {
|
|
135
|
+
role: 'assistant',
|
|
136
|
+
content: 'response',
|
|
137
|
+
tool_calls: JSON.stringify([{ id: 'call_1', type: 'function', function: { name: 'test', arguments: '{}' } }]),
|
|
138
|
+
tool_call_id: 'call_1',
|
|
139
|
+
name: 'test',
|
|
140
|
+
model: 'gpt-4',
|
|
141
|
+
tokens_input: 100,
|
|
142
|
+
tokens_output: 50,
|
|
143
|
+
latency_ms: 500,
|
|
144
|
+
});
|
|
145
|
+
const turns = getTurns(db);
|
|
146
|
+
expect(turns).toHaveLength(1);
|
|
147
|
+
expect(turns[0].tool_calls).toContain('call_1');
|
|
148
|
+
expect(turns[0].model).toBe('gpt-4');
|
|
149
|
+
});
|
|
150
|
+
});
|
|
151
|
+
//# sourceMappingURL=sqlite.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sqlite.test.js","sourceRoot":"","sources":["../../src/session/sqlite.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,MAAM,IAAI,CAAC;AAEpB,OAAO,EACL,WAAW,EACX,aAAa,EACb,eAAe,EACf,sBAAsB,EACtB,sBAAsB,EACtB,QAAQ,EACR,QAAQ,EACR,UAAU,EACV,eAAe,EACf,cAAc,EACd,SAAS,EACT,QAAQ,EACR,SAAS,EACT,QAAQ,GACT,MAAM,aAAa,CAAC;AAErB,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;IAC/C,IAAI,EAAqB,CAAC;IAC1B,IAAI,MAAc,CAAC;IAEnB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAC;QAC5E,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;QACxC,EAAE,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,MAAM,GAAG,EAAE,CAAC,OAAO,CAAC,uEAAuE,CAAC,CAAC,GAAG,EAAE,CAAC;QACzG,MAAM,CAAC,MAAM,CAAC,CAAC,UAAU,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,GAAG,EAAE;QAC7B,eAAe,CAAC,EAAE,EAAE;YAClB,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,YAAY,EAAE,QAAQ;YAC3D,KAAK,EAAE,cAAc,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI;SAC1D,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,CAAC;QACxD,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,eAAe,CAAC,EAAE,EAAE;YAClB,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,YAAY,EAAE,QAAQ;YAC3D,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI;SACtD,CAAC,CAAC;QACH,eAAe,CAAC,EAAE,EAAE;YAClB,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,YAAY,EAAE,QAAQ;YAC3D,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI;SACrD,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,CAAC;QACxD,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC7B,MAAM,CAAE,IAAI,CAAC,CAAC,CAAS,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,eAAe,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,YAAY,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QACpH,eAAe,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,YAAY,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QACpH,eAAe,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QAErH,MAAM,MAAM,GAAG,sBAAsB,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,eAAe,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QACxH,sBAAsB,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAClC,MAAM,CAAC,sBAAsB,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IACjD,IAAI,EAAqB,CAAC;IAC1B,IAAI,MAAc,CAAC;IAEnB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC,CAAC;QAC5E,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,CAAC,CAAC;QAC1C,EAAE,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,MAAM,GAAG,EAAE,CAAC,OAAO,CAAC,mDAAmD,CAAC,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAC7G,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,QAAQ,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QACjD,QAAQ,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QACtD,MAAM,KAAK,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,QAAQ,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAChD,UAAU,CAAC,EAAE,CAAC,CAAC;QACf,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,KAAK,GAAG,IAAI,GAAG,CAAiB;YACpC,CAAC,kBAAkB,EAAE,SAAS,CAAC;YAC/B,CAAC,mBAAmB,EAAE,UAAU,CAAC;SAClC,CAAC,CAAC;QACH,eAAe,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QAE3B,MAAM,MAAM,GAAG,cAAc,CAAC,EAAE,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,eAAe,CAAC,EAAE,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAClD,eAAe,CAAC,EAAE,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,KAAK,GAAG;YACZ,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,EAAE,GAAG,EAAE;YACnF,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,GAAG,EAAE,UAAU,EAAE,GAAG,EAAE;SACtF,CAAC;QACF,SAAS,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACrB,MAAM,MAAM,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC5B,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,SAAS,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9F,SAAS,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAChG,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,SAAS,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QAClD,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,SAAS,CAAC,EAAE,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;QACjC,SAAS,CAAC,EAAE,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;QAChC,MAAM,CAAC,QAAQ,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,QAAQ,CAAC,EAAE,EAAE;YACX,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,UAAU;YACnB,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;YAC7G,YAAY,EAAE,QAAQ;YACtB,IAAI,EAAE,MAAM;YACZ,KAAK,EAAE,OAAO;YACd,YAAY,EAAE,GAAG;YACjB,aAAa,EAAE,EAAE;YACjB,UAAU,EAAE,GAAG;SAChB,CAAC,CAAC;QACH,MAAM,KAAK,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC3B,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAChD,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;AAEL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"delegation.test.d.ts","sourceRoot":"","sources":["../../../src/tools/builtin/delegation.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,87 @@
|
|
|
1
|
+
import { describe, it, expect, vi, afterEach } from 'vitest';
|
|
2
|
+
import { manage, setRouterClient } from './delegation.js';
|
|
3
|
+
describe('Delegation tool', () => {
|
|
4
|
+
const mockContext = {
|
|
5
|
+
sessionId: 'test-session',
|
|
6
|
+
projectRoot: process.cwd(),
|
|
7
|
+
projectHash: 'testhash',
|
|
8
|
+
activeFiles: new Map(),
|
|
9
|
+
agentRole: 'coder',
|
|
10
|
+
abortSignal: new AbortController().signal,
|
|
11
|
+
autoApplyEdits: 'all',
|
|
12
|
+
patchHistory: [],
|
|
13
|
+
};
|
|
14
|
+
afterEach(() => {
|
|
15
|
+
vi.restoreAllMocks();
|
|
16
|
+
setRouterClient(null);
|
|
17
|
+
});
|
|
18
|
+
it('returns error when router client is not set', async () => {
|
|
19
|
+
setRouterClient(null);
|
|
20
|
+
const result = await manage({
|
|
21
|
+
goal: 'do something',
|
|
22
|
+
role: 'coder',
|
|
23
|
+
}, mockContext);
|
|
24
|
+
expect(result.success).toBe(false);
|
|
25
|
+
expect(result.error).toContain('not initialized');
|
|
26
|
+
});
|
|
27
|
+
it('delegates and returns sub-agent result', async () => {
|
|
28
|
+
const mockRouterClient = {
|
|
29
|
+
chat: {
|
|
30
|
+
completions: {
|
|
31
|
+
create: vi.fn()
|
|
32
|
+
.mockResolvedValueOnce({
|
|
33
|
+
choices: [{ message: { content: 'I will do the task', role: 'assistant', tool_calls: [] } }],
|
|
34
|
+
}),
|
|
35
|
+
},
|
|
36
|
+
},
|
|
37
|
+
};
|
|
38
|
+
setRouterClient(mockRouterClient);
|
|
39
|
+
const result = await manage({
|
|
40
|
+
goal: 'implement feature',
|
|
41
|
+
role: 'coder',
|
|
42
|
+
}, mockContext);
|
|
43
|
+
expect(result.success).toBe(true);
|
|
44
|
+
expect(mockRouterClient.chat.completions.create).toHaveBeenCalled();
|
|
45
|
+
});
|
|
46
|
+
it('reaches max turns for long-running agents', async () => {
|
|
47
|
+
const mockRouterClient = {
|
|
48
|
+
chat: {
|
|
49
|
+
completions: {
|
|
50
|
+
create: vi.fn().mockResolvedValue({
|
|
51
|
+
choices: [{
|
|
52
|
+
message: {
|
|
53
|
+
content: 'working...',
|
|
54
|
+
role: 'assistant',
|
|
55
|
+
tool_calls: [
|
|
56
|
+
{ id: 'call_1', type: 'function', function: { name: 'read_file', arguments: '{"path": "test.ts"}' } },
|
|
57
|
+
],
|
|
58
|
+
},
|
|
59
|
+
}],
|
|
60
|
+
}),
|
|
61
|
+
},
|
|
62
|
+
},
|
|
63
|
+
};
|
|
64
|
+
setRouterClient(mockRouterClient);
|
|
65
|
+
const result = await manage({
|
|
66
|
+
goal: 'do something',
|
|
67
|
+
role: 'coder',
|
|
68
|
+
}, mockContext);
|
|
69
|
+
expect(result.content).toContain('max turns');
|
|
70
|
+
});
|
|
71
|
+
it('handles router errors gracefully', async () => {
|
|
72
|
+
const mockRouterClient = {
|
|
73
|
+
chat: {
|
|
74
|
+
completions: {
|
|
75
|
+
create: vi.fn().mockRejectedValue(new Error('API error')),
|
|
76
|
+
},
|
|
77
|
+
},
|
|
78
|
+
};
|
|
79
|
+
setRouterClient(mockRouterClient);
|
|
80
|
+
const result = await manage({
|
|
81
|
+
goal: 'do something',
|
|
82
|
+
}, mockContext);
|
|
83
|
+
expect(result.success).toBe(false);
|
|
84
|
+
expect(result.error).toContain('Delegation failed');
|
|
85
|
+
});
|
|
86
|
+
});
|
|
87
|
+
//# sourceMappingURL=delegation.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"delegation.test.js","sourceRoot":"","sources":["../../../src/tools/builtin/delegation.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAc,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAG1D,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAE/B,MAAM,WAAW,GAAgB;QAC/B,SAAS,EAAE,cAAc;QACzB,WAAW,EAAE,OAAO,CAAC,GAAG,EAAE;QAC1B,WAAW,EAAE,UAAU;QACvB,WAAW,EAAE,IAAI,GAAG,EAAE;QACtB,SAAS,EAAE,OAAO;QAClB,WAAW,EAAE,IAAI,eAAe,EAAE,CAAC,MAAM;QACzC,cAAc,EAAE,KAAK;QACrB,YAAY,EAAE,EAAE;KACF,CAAC;IAEjB,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;QACrB,eAAe,CAAC,IAAW,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,eAAe,CAAC,IAAW,CAAC,CAAC;QAC7B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC;YAC1B,IAAI,EAAE,cAAc;YACpB,IAAI,EAAE,OAAO;SACd,EAAE,WAAW,CAAC,CAAC;QAChB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,gBAAgB,GAAG;YACvB,IAAI,EAAE;gBACJ,WAAW,EAAE;oBACX,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;yBACZ,qBAAqB,CAAC;wBACrB,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,OAAO,EAAE,oBAAoB,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC;qBAC7F,CAAC;iBACL;aACF;SACF,CAAC;QACF,eAAe,CAAC,gBAAuB,CAAC,CAAC;QAEzC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC;YAC1B,IAAI,EAAE,mBAAmB;YACzB,IAAI,EAAE,OAAO;SACd,EAAE,WAAW,CAAC,CAAC;QAEhB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,gBAAgB,EAAE,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;QACzD,MAAM,gBAAgB,GAAG;YACvB,IAAI,EAAE;gBACJ,WAAW,EAAE;oBACX,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;wBAChC,OAAO,EAAE,CAAC;gCACR,OAAO,EAAE;oCACP,OAAO,EAAE,YAAY;oCACrB,IAAI,EAAE,WAAW;oCACjB,UAAU,EAAE;wCACV,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE,qBAAqB,EAAE,EAAE;qCACtG;iCACF;6BACF,CAAC;qBACH,CAAC;iBACH;aACF;SACF,CAAC;QACF,eAAe,CAAC,gBAAuB,CAAC,CAAC;QAEzC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC;YAC1B,IAAI,EAAE,cAAc;YACpB,IAAI,EAAE,OAAO;SACd,EAAE,WAAW,CAAC,CAAC;QAEhB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,MAAM,gBAAgB,GAAG;YACvB,IAAI,EAAE;gBACJ,WAAW,EAAE;oBACX,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;iBAC1D;aACF;SACF,CAAC;QACF,eAAe,CAAC,gBAAuB,CAAC,CAAC;QAEzC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC;YAC1B,IAAI,EAAE,cAAc;SACrB,EAAE,WAAW,CAAC,CAAC;QAEhB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;AAEL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"eval.d.ts","sourceRoot":"","sources":["../../../src/tools/builtin/eval.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAyBzD,wBAAsB,QAAQ,CAC5B,IAAI,EAAE;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IAAC,OAAO,CAAC,EAAE,MAAM,CAAA;CAAE,EACvD,OAAO,EAAE,WAAW,GACnB,OAAO,CAAC,UAAU,CAAC,CA8DrB"}
|
|
@@ -0,0 +1,92 @@
|
|
|
1
|
+
import { spawnSync } from 'child_process';
|
|
2
|
+
import ts from 'typescript';
|
|
3
|
+
function formatError(error) {
|
|
4
|
+
return { toolCallId: '', name: 'eval_code', success: false, content: '', error };
|
|
5
|
+
}
|
|
6
|
+
async function promptApproval(code) {
|
|
7
|
+
if (!process.stdin.isTTY)
|
|
8
|
+
return false;
|
|
9
|
+
process.stdout.write(`\n[EVAL] About to execute:\n${'-'.repeat(50)}\n${code.slice(0, 500)}\n${'-'.repeat(50)}\nProceed? [y/n]: `);
|
|
10
|
+
return new Promise(resolve => {
|
|
11
|
+
if (process.stdin.isTTY)
|
|
12
|
+
process.stdin.setRawMode?.(true);
|
|
13
|
+
const onKey = (key) => {
|
|
14
|
+
const ch = key.toString().toLowerCase();
|
|
15
|
+
if (ch === 'y') {
|
|
16
|
+
cleanup();
|
|
17
|
+
process.stdout.write('Y\n');
|
|
18
|
+
resolve(true);
|
|
19
|
+
}
|
|
20
|
+
else if (ch === 'n' || ch === '\u0003') {
|
|
21
|
+
cleanup();
|
|
22
|
+
process.stdout.write('N\n');
|
|
23
|
+
resolve(false);
|
|
24
|
+
}
|
|
25
|
+
};
|
|
26
|
+
process.stdin.on('data', onKey);
|
|
27
|
+
const cleanup = () => {
|
|
28
|
+
process.stdin.off('data', onKey);
|
|
29
|
+
if (process.stdin.isTTY)
|
|
30
|
+
process.stdin.setRawMode?.(false);
|
|
31
|
+
};
|
|
32
|
+
});
|
|
33
|
+
}
|
|
34
|
+
export async function evalCode(args, context) {
|
|
35
|
+
const approved = await promptApproval(args.code);
|
|
36
|
+
if (!approved) {
|
|
37
|
+
return { toolCallId: '', name: 'eval_code', success: false, content: '', error: 'Eval cancelled by user.' };
|
|
38
|
+
}
|
|
39
|
+
try {
|
|
40
|
+
const lang = args.lang ?? 'ts';
|
|
41
|
+
const timeoutMs = (args.timeout ?? 10) * 1000;
|
|
42
|
+
let jsCode = args.code;
|
|
43
|
+
if (lang === 'ts' || lang === 'typescript') {
|
|
44
|
+
const result = ts.transpileModule(args.code, {
|
|
45
|
+
compilerOptions: {
|
|
46
|
+
target: ts.ScriptTarget.ESNext,
|
|
47
|
+
module: ts.ModuleKind.CommonJS,
|
|
48
|
+
esModuleInterop: true,
|
|
49
|
+
},
|
|
50
|
+
});
|
|
51
|
+
if (result.diagnostics && result.diagnostics.length > 0) {
|
|
52
|
+
const errors = result.diagnostics.map(d => ts.flattenDiagnosticMessageText(d.messageText, '\n')).join('\n');
|
|
53
|
+
return formatError(`TypeScript transpile errors:\n${errors}`);
|
|
54
|
+
}
|
|
55
|
+
jsCode = result.outputText;
|
|
56
|
+
}
|
|
57
|
+
const wrappedCode = `
|
|
58
|
+
const __result = (async () => {
|
|
59
|
+
${jsCode}
|
|
60
|
+
})();
|
|
61
|
+
__result.then(v => { if (v !== undefined) process.stdout.write(String(v) + '\\n'); }).catch(e => { process.stderr.write(e.message + '\\n'); process.exit(1); });
|
|
62
|
+
`;
|
|
63
|
+
const proc = spawnSync(process.execPath, ['--input-type=commonjs'], {
|
|
64
|
+
input: wrappedCode,
|
|
65
|
+
cwd: context.projectRoot,
|
|
66
|
+
timeout: timeoutMs,
|
|
67
|
+
encoding: 'utf8',
|
|
68
|
+
env: { ...process.env },
|
|
69
|
+
});
|
|
70
|
+
const stdout = (proc.stdout ?? '').trim();
|
|
71
|
+
const stderr = (proc.stderr ?? '').trim();
|
|
72
|
+
if (proc.signal === 'SIGTERM' || proc.error?.message?.includes('ETIMEDOUT')) {
|
|
73
|
+
return formatError(`eval_code timed out after ${args.timeout ?? 10}s`);
|
|
74
|
+
}
|
|
75
|
+
const parts = [];
|
|
76
|
+
if (stdout)
|
|
77
|
+
parts.push(`stdout:\n${stdout}`);
|
|
78
|
+
if (stderr)
|
|
79
|
+
parts.push(`stderr:\n${stderr}`);
|
|
80
|
+
if (parts.length === 0)
|
|
81
|
+
parts.push('(no output)');
|
|
82
|
+
return {
|
|
83
|
+
toolCallId: '', name: 'eval_code',
|
|
84
|
+
success: proc.status === 0,
|
|
85
|
+
content: parts.join('\n\n'),
|
|
86
|
+
};
|
|
87
|
+
}
|
|
88
|
+
catch (err) {
|
|
89
|
+
return formatError(`eval_code failed: ${err.message}`);
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
//# sourceMappingURL=eval.js.map
|