expxagents 0.6.0 → 0.7.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/dist/__tests__/cli.test.d.ts +1 -0
- package/dist/__tests__/cli.test.js +23 -0
- package/dist/cli/src/commands/init.js +12 -2
- package/dist/cli/src/commands/virtual-office.js +50 -5
- package/dist/cli/src/utils/config.d.ts +5 -0
- package/dist/cli/src/utils/config.js +1 -1
- package/dist/commands/create.d.ts +1 -0
- package/dist/commands/create.js +34 -0
- package/dist/commands/init.d.ts +1 -0
- package/dist/commands/init.js +58 -0
- package/dist/commands/install.d.ts +1 -0
- package/dist/commands/install.js +16 -0
- package/dist/commands/list.d.ts +1 -0
- package/dist/commands/list.js +58 -0
- package/dist/commands/run.d.ts +1 -0
- package/dist/commands/run.js +173 -0
- package/dist/commands/server.d.ts +1 -0
- package/dist/commands/server.js +22 -0
- package/dist/commands/stop.d.ts +1 -0
- package/dist/commands/stop.js +23 -0
- package/dist/commands/uninstall.d.ts +1 -0
- package/dist/commands/uninstall.js +12 -0
- package/dist/dashboard/assets/BufferResource-Cf2Uo4_f.js +185 -0
- package/dist/dashboard/assets/CanvasRenderer-DFdMBORe.js +1 -0
- package/dist/dashboard/assets/RenderTargetSystem-CMh8XRf_.js +172 -0
- package/dist/dashboard/assets/WebGLRenderer-B5huw0RY.js +156 -0
- package/dist/dashboard/assets/WebGPURenderer-BdIKurkV.js +41 -0
- package/dist/dashboard/assets/browserAll-BjVJrv1L.js +14 -0
- package/dist/dashboard/assets/index-DwTFo09S.js +344 -0
- package/dist/dashboard/assets/webworkerAll-DMtK63GZ.js +83 -0
- package/dist/dashboard/index.html +16 -0
- package/dist/index.d.ts +1 -0
- package/dist/index.js +47 -0
- package/dist/server/api/health-routes.d.ts +3 -0
- package/dist/server/api/health-routes.d.ts.map +1 -0
- package/dist/server/api/health-routes.js +6 -0
- package/dist/server/api/health-routes.js.map +1 -0
- package/dist/server/api/squads-routes.d.ts +7 -0
- package/dist/server/api/squads-routes.d.ts.map +1 -0
- package/dist/server/api/squads-routes.js +91 -0
- package/dist/server/api/squads-routes.js.map +1 -0
- package/dist/server/api/users-routes.d.ts +8 -0
- package/dist/server/api/users-routes.d.ts.map +1 -0
- package/dist/server/api/users-routes.js +45 -0
- package/dist/server/api/users-routes.js.map +1 -0
- package/dist/server/app.d.ts +7 -0
- package/dist/server/app.d.ts.map +1 -0
- package/dist/server/app.js +109 -0
- package/dist/server/app.js.map +1 -0
- package/dist/server/auth/auth-middleware.d.ts +13 -0
- package/dist/server/auth/auth-middleware.d.ts.map +1 -0
- package/dist/server/auth/auth-middleware.js +25 -0
- package/dist/server/auth/auth-middleware.js.map +1 -0
- package/dist/server/auth/auth-routes.d.ts +9 -0
- package/dist/server/auth/auth-routes.d.ts.map +1 -0
- package/dist/server/auth/auth-routes.js +149 -0
- package/dist/server/auth/auth-routes.js.map +1 -0
- package/dist/server/auth/jwt.d.ts +14 -0
- package/dist/server/auth/jwt.d.ts.map +1 -0
- package/dist/server/auth/jwt.js +16 -0
- package/dist/server/auth/jwt.js.map +1 -0
- package/dist/server/auth/password.d.ts +3 -0
- package/dist/server/auth/password.d.ts.map +1 -0
- package/dist/server/auth/password.js +9 -0
- package/dist/server/auth/password.js.map +1 -0
- package/dist/server/bridge/__tests__/chat-handler.test.d.ts +2 -0
- package/dist/server/bridge/__tests__/chat-handler.test.d.ts.map +1 -0
- package/dist/server/bridge/__tests__/chat-handler.test.js +132 -0
- package/dist/server/bridge/__tests__/chat-handler.test.js.map +1 -0
- package/dist/server/bridge/__tests__/chat-integration.test.d.ts +2 -0
- package/dist/server/bridge/__tests__/chat-integration.test.d.ts.map +1 -0
- package/dist/server/bridge/__tests__/chat-integration.test.js +141 -0
- package/dist/server/bridge/__tests__/chat-integration.test.js.map +1 -0
- package/dist/server/bridge/__tests__/claude-bridge.test.d.ts +2 -0
- package/dist/server/bridge/__tests__/claude-bridge.test.d.ts.map +1 -0
- package/dist/server/bridge/__tests__/claude-bridge.test.js +223 -0
- package/dist/server/bridge/__tests__/claude-bridge.test.js.map +1 -0
- package/dist/server/bridge/__tests__/conversation.test.d.ts +2 -0
- package/dist/server/bridge/__tests__/conversation.test.d.ts.map +1 -0
- package/dist/server/bridge/__tests__/conversation.test.js +168 -0
- package/dist/server/bridge/__tests__/conversation.test.js.map +1 -0
- package/dist/server/bridge/__tests__/stream-parser.test.d.ts +2 -0
- package/dist/server/bridge/__tests__/stream-parser.test.d.ts.map +1 -0
- package/dist/server/bridge/__tests__/stream-parser.test.js +66 -0
- package/dist/server/bridge/__tests__/stream-parser.test.js.map +1 -0
- package/dist/server/bridge/chat-handler.d.ts +19 -0
- package/dist/server/bridge/chat-handler.d.ts.map +1 -0
- package/dist/server/bridge/chat-handler.js +104 -0
- package/dist/server/bridge/chat-handler.js.map +1 -0
- package/dist/server/bridge/claude-bridge.d.ts +27 -0
- package/dist/server/bridge/claude-bridge.d.ts.map +1 -0
- package/dist/server/bridge/claude-bridge.js +91 -0
- package/dist/server/bridge/claude-bridge.js.map +1 -0
- package/dist/server/bridge/conversation.d.ts +44 -0
- package/dist/server/bridge/conversation.d.ts.map +1 -0
- package/dist/server/bridge/conversation.js +73 -0
- package/dist/server/bridge/conversation.js.map +1 -0
- package/dist/server/bridge/stream-parser.d.ts +14 -0
- package/dist/server/bridge/stream-parser.d.ts.map +1 -0
- package/dist/server/bridge/stream-parser.js +26 -0
- package/dist/server/bridge/stream-parser.js.map +1 -0
- package/dist/server/config.d.ts +10 -0
- package/dist/server/config.d.ts.map +1 -0
- package/dist/server/config.js +22 -0
- package/dist/server/config.js.map +1 -0
- package/dist/server/db/__tests__/chat-tables.test.d.ts +2 -0
- package/dist/server/db/__tests__/chat-tables.test.d.ts.map +1 -0
- package/dist/server/db/__tests__/chat-tables.test.js +82 -0
- package/dist/server/db/__tests__/chat-tables.test.js.map +1 -0
- package/dist/server/db/connection.d.ts +4 -0
- package/dist/server/db/connection.d.ts.map +1 -0
- package/dist/server/db/connection.js +21 -0
- package/dist/server/db/connection.js.map +1 -0
- package/dist/server/db/migrations.d.ts +4 -0
- package/dist/server/db/migrations.d.ts.map +1 -0
- package/dist/server/db/migrations.js +17 -0
- package/dist/server/db/migrations.js.map +1 -0
- package/dist/server/db/schema.d.ts +2 -0
- package/dist/server/db/schema.d.ts.map +1 -0
- package/dist/server/db/schema.js +44 -0
- package/dist/server/db/schema.js.map +1 -0
- package/dist/server/index.d.ts +2 -0
- package/dist/server/index.d.ts.map +1 -0
- package/dist/server/index.js +18 -0
- package/dist/server/index.js.map +1 -0
- package/dist/server/routes/__tests__/conversations.test.d.ts +2 -0
- package/dist/server/routes/__tests__/conversations.test.d.ts.map +1 -0
- package/dist/server/routes/__tests__/conversations.test.js +94 -0
- package/dist/server/routes/__tests__/conversations.test.js.map +1 -0
- package/dist/server/routes/conversations.d.ts +8 -0
- package/dist/server/routes/conversations.d.ts.map +1 -0
- package/dist/server/routes/conversations.js +25 -0
- package/dist/server/routes/conversations.js.map +1 -0
- package/dist/server/watcher/file-watcher.d.ts +12 -0
- package/dist/server/watcher/file-watcher.d.ts.map +1 -0
- package/dist/server/watcher/file-watcher.js +68 -0
- package/dist/server/watcher/file-watcher.js.map +1 -0
- package/dist/server/watcher/state-parser.d.ts +77 -0
- package/dist/server/watcher/state-parser.d.ts.map +1 -0
- package/dist/server/watcher/state-parser.js +74 -0
- package/dist/server/watcher/state-parser.js.map +1 -0
- package/dist/server/ws/ws-auth.d.ts +4 -0
- package/dist/server/ws/ws-auth.d.ts.map +1 -0
- package/dist/server/ws/ws-auth.js +42 -0
- package/dist/server/ws/ws-auth.js.map +1 -0
- package/dist/server/ws/ws-handler.d.ts +11 -0
- package/dist/server/ws/ws-handler.d.ts.map +1 -0
- package/dist/server/ws/ws-handler.js +107 -0
- package/dist/server/ws/ws-handler.js.map +1 -0
- package/dist/server/ws/ws-rooms.d.ts +12 -0
- package/dist/server/ws/ws-rooms.d.ts.map +1 -0
- package/dist/server/ws/ws-rooms.js +52 -0
- package/dist/server/ws/ws-rooms.js.map +1 -0
- package/dist/utils/config.d.ts +15 -0
- package/dist/utils/config.js +23 -0
- package/package.json +16 -5
|
@@ -0,0 +1,223 @@
|
|
|
1
|
+
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import { EventEmitter } from 'events';
|
|
3
|
+
import { ClaudeBridge } from '../claude-bridge.js';
|
|
4
|
+
// Mock child_process.spawn
|
|
5
|
+
vi.mock('child_process', () => {
|
|
6
|
+
const actual = vi.importActual('child_process');
|
|
7
|
+
return {
|
|
8
|
+
...actual,
|
|
9
|
+
spawn: vi.fn(),
|
|
10
|
+
};
|
|
11
|
+
});
|
|
12
|
+
import { spawn } from 'child_process';
|
|
13
|
+
const mockSpawn = vi.mocked(spawn);
|
|
14
|
+
function createMockProcess() {
|
|
15
|
+
const proc = new EventEmitter();
|
|
16
|
+
proc.stdout = new EventEmitter();
|
|
17
|
+
proc.stderr = new EventEmitter();
|
|
18
|
+
proc.kill = vi.fn();
|
|
19
|
+
proc.pid = 12345;
|
|
20
|
+
return proc;
|
|
21
|
+
}
|
|
22
|
+
describe('ClaudeBridge', () => {
|
|
23
|
+
let bridge;
|
|
24
|
+
beforeEach(() => {
|
|
25
|
+
bridge = new ClaudeBridge({ timeoutMs: 5000 });
|
|
26
|
+
vi.useFakeTimers();
|
|
27
|
+
});
|
|
28
|
+
afterEach(() => {
|
|
29
|
+
vi.useRealTimers();
|
|
30
|
+
vi.restoreAllMocks();
|
|
31
|
+
});
|
|
32
|
+
it('should spawn claude with correct arguments', () => {
|
|
33
|
+
const mockProc = createMockProcess();
|
|
34
|
+
mockSpawn.mockReturnValue(mockProc);
|
|
35
|
+
bridge.send({
|
|
36
|
+
prompt: 'Hello Claude',
|
|
37
|
+
cwd: '/tmp/test-squad',
|
|
38
|
+
onChunk: vi.fn(),
|
|
39
|
+
onDone: vi.fn(),
|
|
40
|
+
onError: vi.fn(),
|
|
41
|
+
});
|
|
42
|
+
expect(mockSpawn).toHaveBeenCalledWith('claude', ['-p', '--output-format', 'text', expect.any(String)], expect.objectContaining({
|
|
43
|
+
cwd: '/tmp/test-squad',
|
|
44
|
+
}));
|
|
45
|
+
});
|
|
46
|
+
it('should emit chunks from stdout', () => {
|
|
47
|
+
const mockProc = createMockProcess();
|
|
48
|
+
mockSpawn.mockReturnValue(mockProc);
|
|
49
|
+
const onChunk = vi.fn();
|
|
50
|
+
const onDone = vi.fn();
|
|
51
|
+
bridge.send({
|
|
52
|
+
prompt: 'Hello',
|
|
53
|
+
cwd: '/tmp/test-squad',
|
|
54
|
+
onChunk,
|
|
55
|
+
onDone,
|
|
56
|
+
onError: vi.fn(),
|
|
57
|
+
});
|
|
58
|
+
mockProc.stdout.emit('data', Buffer.from('Hello '));
|
|
59
|
+
mockProc.stdout.emit('data', Buffer.from('world'));
|
|
60
|
+
mockProc.emit('close', 0);
|
|
61
|
+
expect(onChunk).toHaveBeenCalledTimes(2);
|
|
62
|
+
expect(onChunk).toHaveBeenNthCalledWith(1, 'Hello ');
|
|
63
|
+
expect(onChunk).toHaveBeenNthCalledWith(2, 'world');
|
|
64
|
+
expect(onDone).toHaveBeenCalledWith('Hello world');
|
|
65
|
+
});
|
|
66
|
+
it('should call onError on non-zero exit', () => {
|
|
67
|
+
const mockProc = createMockProcess();
|
|
68
|
+
mockSpawn.mockReturnValue(mockProc);
|
|
69
|
+
const onError = vi.fn();
|
|
70
|
+
bridge.send({
|
|
71
|
+
prompt: 'Hello',
|
|
72
|
+
cwd: '/tmp/test-squad',
|
|
73
|
+
onChunk: vi.fn(),
|
|
74
|
+
onDone: vi.fn(),
|
|
75
|
+
onError,
|
|
76
|
+
});
|
|
77
|
+
mockProc.stderr.emit('data', Buffer.from('Something went wrong'));
|
|
78
|
+
mockProc.emit('close', 1);
|
|
79
|
+
expect(onError).toHaveBeenCalledWith(expect.objectContaining({
|
|
80
|
+
code: 1,
|
|
81
|
+
message: expect.stringContaining('Something went wrong'),
|
|
82
|
+
}));
|
|
83
|
+
});
|
|
84
|
+
it('should call onError on spawn error', () => {
|
|
85
|
+
const mockProc = createMockProcess();
|
|
86
|
+
mockSpawn.mockReturnValue(mockProc);
|
|
87
|
+
const onError = vi.fn();
|
|
88
|
+
bridge.send({
|
|
89
|
+
prompt: 'Hello',
|
|
90
|
+
cwd: '/tmp/test-squad',
|
|
91
|
+
onChunk: vi.fn(),
|
|
92
|
+
onDone: vi.fn(),
|
|
93
|
+
onError,
|
|
94
|
+
});
|
|
95
|
+
mockProc.emit('error', new Error('spawn ENOENT'));
|
|
96
|
+
expect(onError).toHaveBeenCalledWith(expect.objectContaining({
|
|
97
|
+
message: 'Claude Code CLI not available',
|
|
98
|
+
}));
|
|
99
|
+
});
|
|
100
|
+
it('should timeout and kill the process', () => {
|
|
101
|
+
const mockProc = createMockProcess();
|
|
102
|
+
mockSpawn.mockReturnValue(mockProc);
|
|
103
|
+
const onError = vi.fn();
|
|
104
|
+
const onDone = vi.fn();
|
|
105
|
+
bridge.send({
|
|
106
|
+
prompt: 'Hello',
|
|
107
|
+
cwd: '/tmp/test-squad',
|
|
108
|
+
onChunk: vi.fn(),
|
|
109
|
+
onDone,
|
|
110
|
+
onError,
|
|
111
|
+
});
|
|
112
|
+
mockProc.stdout.emit('data', Buffer.from('Partial response'));
|
|
113
|
+
// Advance past timeout
|
|
114
|
+
vi.advanceTimersByTime(5000);
|
|
115
|
+
expect(mockProc.kill).toHaveBeenCalledWith('SIGTERM');
|
|
116
|
+
// Advance past SIGKILL grace period (5s)
|
|
117
|
+
vi.advanceTimersByTime(5000);
|
|
118
|
+
expect(mockProc.kill).toHaveBeenCalledWith('SIGKILL');
|
|
119
|
+
});
|
|
120
|
+
it('should report busy when a process is already running for a squad', () => {
|
|
121
|
+
const mockProc = createMockProcess();
|
|
122
|
+
mockSpawn.mockReturnValue(mockProc);
|
|
123
|
+
const onError = vi.fn();
|
|
124
|
+
bridge.send({
|
|
125
|
+
prompt: 'First',
|
|
126
|
+
cwd: '/tmp/squad-a',
|
|
127
|
+
onChunk: vi.fn(),
|
|
128
|
+
onDone: vi.fn(),
|
|
129
|
+
onError: vi.fn(),
|
|
130
|
+
squadName: 'squad-a',
|
|
131
|
+
});
|
|
132
|
+
bridge.send({
|
|
133
|
+
prompt: 'Second',
|
|
134
|
+
cwd: '/tmp/squad-a',
|
|
135
|
+
onChunk: vi.fn(),
|
|
136
|
+
onDone: vi.fn(),
|
|
137
|
+
onError,
|
|
138
|
+
squadName: 'squad-a',
|
|
139
|
+
});
|
|
140
|
+
expect(onError).toHaveBeenCalledWith(expect.objectContaining({
|
|
141
|
+
message: expect.stringContaining('Pipeline is running'),
|
|
142
|
+
}));
|
|
143
|
+
});
|
|
144
|
+
it('should call onOutputChunk with agentId, agentName and delta when all three are provided', () => {
|
|
145
|
+
const mockProc = createMockProcess();
|
|
146
|
+
mockSpawn.mockReturnValue(mockProc);
|
|
147
|
+
const onChunk = vi.fn();
|
|
148
|
+
const onOutputChunk = vi.fn();
|
|
149
|
+
const onDone = vi.fn();
|
|
150
|
+
bridge.send({
|
|
151
|
+
prompt: 'Hello',
|
|
152
|
+
cwd: '/tmp/test-squad',
|
|
153
|
+
agentId: 'agent-001',
|
|
154
|
+
agentName: 'Developer',
|
|
155
|
+
onChunk,
|
|
156
|
+
onOutputChunk,
|
|
157
|
+
onDone,
|
|
158
|
+
onError: vi.fn(),
|
|
159
|
+
});
|
|
160
|
+
mockProc.stdout.emit('data', Buffer.from('Hello '));
|
|
161
|
+
mockProc.stdout.emit('data', Buffer.from('world'));
|
|
162
|
+
mockProc.emit('close', 0);
|
|
163
|
+
expect(onChunk).toHaveBeenCalledTimes(2);
|
|
164
|
+
expect(onOutputChunk).toHaveBeenCalledTimes(2);
|
|
165
|
+
expect(onOutputChunk).toHaveBeenNthCalledWith(1, 'agent-001', 'Developer', 'Hello ');
|
|
166
|
+
expect(onOutputChunk).toHaveBeenNthCalledWith(2, 'agent-001', 'Developer', 'world');
|
|
167
|
+
expect(onDone).toHaveBeenCalledWith('Hello world');
|
|
168
|
+
});
|
|
169
|
+
it('should not call onOutputChunk when agentId or agentName is missing', () => {
|
|
170
|
+
const mockProc = createMockProcess();
|
|
171
|
+
mockSpawn.mockReturnValue(mockProc);
|
|
172
|
+
const onOutputChunk = vi.fn();
|
|
173
|
+
bridge.send({
|
|
174
|
+
prompt: 'Hello',
|
|
175
|
+
cwd: '/tmp/test-squad',
|
|
176
|
+
agentId: 'agent-001',
|
|
177
|
+
// agentName is intentionally missing
|
|
178
|
+
onChunk: vi.fn(),
|
|
179
|
+
onOutputChunk,
|
|
180
|
+
onDone: vi.fn(),
|
|
181
|
+
onError: vi.fn(),
|
|
182
|
+
});
|
|
183
|
+
mockProc.stdout.emit('data', Buffer.from('Some output'));
|
|
184
|
+
mockProc.emit('close', 0);
|
|
185
|
+
expect(onOutputChunk).not.toHaveBeenCalled();
|
|
186
|
+
});
|
|
187
|
+
it('should not call onOutputChunk when callback is not provided', () => {
|
|
188
|
+
const mockProc = createMockProcess();
|
|
189
|
+
mockSpawn.mockReturnValue(mockProc);
|
|
190
|
+
const onChunk = vi.fn();
|
|
191
|
+
// No onOutputChunk provided - should not throw
|
|
192
|
+
bridge.send({
|
|
193
|
+
prompt: 'Hello',
|
|
194
|
+
cwd: '/tmp/test-squad',
|
|
195
|
+
agentId: 'agent-001',
|
|
196
|
+
agentName: 'Developer',
|
|
197
|
+
onChunk,
|
|
198
|
+
onDone: vi.fn(),
|
|
199
|
+
onError: vi.fn(),
|
|
200
|
+
});
|
|
201
|
+
mockProc.stdout.emit('data', Buffer.from('Some output'));
|
|
202
|
+
mockProc.emit('close', 0);
|
|
203
|
+
expect(onChunk).toHaveBeenCalledWith('Some output');
|
|
204
|
+
// No error thrown is the assertion here (test completes without throwing)
|
|
205
|
+
});
|
|
206
|
+
it('should pass full process environment', () => {
|
|
207
|
+
const mockProc = createMockProcess();
|
|
208
|
+
mockSpawn.mockReturnValue(mockProc);
|
|
209
|
+
bridge.send({
|
|
210
|
+
prompt: 'Hello',
|
|
211
|
+
cwd: '/tmp/test-squad',
|
|
212
|
+
onChunk: vi.fn(),
|
|
213
|
+
onDone: vi.fn(),
|
|
214
|
+
onError: vi.fn(),
|
|
215
|
+
});
|
|
216
|
+
const spawnCall = mockSpawn.mock.calls[0];
|
|
217
|
+
const opts = spawnCall[2];
|
|
218
|
+
const envKeys = Object.keys(opts.env);
|
|
219
|
+
expect(envKeys).toContain('PATH');
|
|
220
|
+
expect(envKeys).toContain('HOME');
|
|
221
|
+
});
|
|
222
|
+
});
|
|
223
|
+
//# sourceMappingURL=claude-bridge.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"claude-bridge.test.js","sourceRoot":"","sources":["../../../src/bridge/__tests__/claude-bridge.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAEnD,2BAA2B;AAC3B,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE;IAC5B,MAAM,MAAM,GAAG,EAAE,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;IAChD,OAAO;QACL,GAAG,MAAM;QACT,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;KACf,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AACtC,MAAM,SAAS,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAEnC,SAAS,iBAAiB;IACxB,MAAM,IAAI,GAAG,IAAI,YAAY,EAAS,CAAC;IACvC,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;IACjC,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,EAAE,CAAC;IACjC,IAAI,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACpB,IAAI,CAAC,GAAG,GAAG,KAAK,CAAC;IACjB,OAAO,IAAI,CAAC;AACd,CAAC;AAED,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,IAAI,MAAoB,CAAC;IAEzB,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;QACrC,SAAS,CAAC,eAAe,CAAC,QAAe,CAAC,CAAC;QAE3C,MAAM,CAAC,IAAI,CAAC;YACV,MAAM,EAAE,cAAc;YACtB,GAAG,EAAE,iBAAiB;YACtB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;SACjB,CAAC,CAAC;QAEH,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CACpC,QAAQ,EACR,CAAC,IAAI,EAAE,iBAAiB,EAAE,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,EACrD,MAAM,CAAC,gBAAgB,CAAC;YACtB,GAAG,EAAE,iBAAiB;SACvB,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;QACrC,SAAS,CAAC,eAAe,CAAC,QAAe,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAEvB,MAAM,CAAC,IAAI,CAAC;YACV,MAAM,EAAE,OAAO;YACf,GAAG,EAAE,iBAAiB;YACtB,OAAO;YACP,MAAM;YACN,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;SACjB,CAAC,CAAC;QAEH,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QACpD,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;QACnD,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAE1B,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;QACrC,SAAS,CAAC,eAAe,CAAC,QAAe,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAExB,MAAM,CAAC,IAAI,CAAC;YACV,MAAM,EAAE,OAAO;YACf,GAAG,EAAE,iBAAiB;YACtB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,OAAO;SACR,CAAC,CAAC;QAEH,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC,CAAC;QAClE,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAE1B,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAClC,MAAM,CAAC,gBAAgB,CAAC;YACtB,IAAI,EAAE,CAAC;YACP,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,sBAAsB,CAAC;SACzD,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;QACrC,SAAS,CAAC,eAAe,CAAC,QAAe,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAExB,MAAM,CAAC,IAAI,CAAC;YACV,MAAM,EAAE,OAAO;YACf,GAAG,EAAE,iBAAiB;YACtB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,OAAO;SACR,CAAC,CAAC;QAEH,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC;QAElD,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAClC,MAAM,CAAC,gBAAgB,CAAC;YACtB,OAAO,EAAE,+BAA+B;SACzC,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;QACrC,SAAS,CAAC,eAAe,CAAC,QAAe,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAEvB,MAAM,CAAC,IAAI,CAAC;YACV,MAAM,EAAE,OAAO;YACf,GAAG,EAAE,iBAAiB;YACtB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,MAAM;YACN,OAAO;SACR,CAAC,CAAC;QAEH,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAE9D,uBAAuB;QACvB,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAE7B,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QAEtD,yCAAyC;QACzC,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAE7B,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;QACrC,SAAS,CAAC,eAAe,CAAC,QAAe,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAExB,MAAM,CAAC,IAAI,CAAC;YACV,MAAM,EAAE,OAAO;YACf,GAAG,EAAE,cAAc;YACnB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,SAAS,EAAE,SAAS;SACrB,CAAC,CAAC;QAEH,MAAM,CAAC,IAAI,CAAC;YACV,MAAM,EAAE,QAAQ;YAChB,GAAG,EAAE,cAAc;YACnB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,OAAO;YACP,SAAS,EAAE,SAAS;SACrB,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAClC,MAAM,CAAC,gBAAgB,CAAC;YACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,qBAAqB,CAAC;SACxD,CAAC,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yFAAyF,EAAE,GAAG,EAAE;QACjG,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;QACrC,SAAS,CAAC,eAAe,CAAC,QAAe,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAEvB,MAAM,CAAC,IAAI,CAAC;YACV,MAAM,EAAE,OAAO;YACf,GAAG,EAAE,iBAAiB;YACtB,OAAO,EAAE,WAAW;YACpB,SAAS,EAAE,WAAW;YACtB,OAAO;YACP,aAAa;YACb,MAAM;YACN,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;SACjB,CAAC,CAAC;QAEH,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QACpD,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;QACnD,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAE1B,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,aAAa,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,WAAW,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;QACrF,MAAM,CAAC,aAAa,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,WAAW,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;QACpF,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;QAC5E,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;QACrC,SAAS,CAAC,eAAe,CAAC,QAAe,CAAC,CAAC;QAE3C,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAE9B,MAAM,CAAC,IAAI,CAAC;YACV,MAAM,EAAE,OAAO;YACf,GAAG,EAAE,iBAAiB;YACtB,OAAO,EAAE,WAAW;YACpB,qCAAqC;YACrC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,aAAa;YACb,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;SACjB,CAAC,CAAC;QAEH,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;QACzD,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAE1B,MAAM,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;QACrC,SAAS,CAAC,eAAe,CAAC,QAAe,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QAExB,+CAA+C;QAC/C,MAAM,CAAC,IAAI,CAAC;YACV,MAAM,EAAE,OAAO;YACf,GAAG,EAAE,iBAAiB;YACtB,OAAO,EAAE,WAAW;YACpB,SAAS,EAAE,WAAW;YACtB,OAAO;YACP,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;SACjB,CAAC,CAAC;QAEH,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;QACzD,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAE1B,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;QACpD,0EAA0E;IAC5E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;QACrC,SAAS,CAAC,eAAe,CAAC,QAAe,CAAC,CAAC;QAE3C,MAAM,CAAC,IAAI,CAAC;YACV,MAAM,EAAE,OAAO;YACf,GAAG,EAAE,iBAAiB;YACtB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YACf,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;SACjB,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC1C,MAAM,IAAI,GAAG,SAAS,CAAC,CAAC,CAAQ,CAAC;QACjC,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAClC,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"conversation.test.d.ts","sourceRoot":"","sources":["../../../src/bridge/__tests__/conversation.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,168 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach } from 'vitest';
|
|
2
|
+
import Database from 'better-sqlite3';
|
|
3
|
+
import { createConversation, getActiveConversation, getOrCreateConversation, insertMessage, getMessages, getConversations, generateId, } from '../conversation.js';
|
|
4
|
+
function createTestDb() {
|
|
5
|
+
const db = new Database(':memory:');
|
|
6
|
+
db.pragma('journal_mode = WAL');
|
|
7
|
+
db.exec(`
|
|
8
|
+
CREATE TABLE IF NOT EXISTS conversations (
|
|
9
|
+
id TEXT PRIMARY KEY,
|
|
10
|
+
squad_name TEXT NOT NULL,
|
|
11
|
+
created_at TEXT NOT NULL
|
|
12
|
+
);
|
|
13
|
+
CREATE TABLE IF NOT EXISTS chat_messages (
|
|
14
|
+
id TEXT PRIMARY KEY,
|
|
15
|
+
conversation_id TEXT NOT NULL REFERENCES conversations(id),
|
|
16
|
+
role TEXT NOT NULL,
|
|
17
|
+
content TEXT NOT NULL,
|
|
18
|
+
agent_id TEXT,
|
|
19
|
+
agent_name TEXT,
|
|
20
|
+
created_at TEXT NOT NULL
|
|
21
|
+
);
|
|
22
|
+
`);
|
|
23
|
+
return db;
|
|
24
|
+
}
|
|
25
|
+
describe('conversation module', () => {
|
|
26
|
+
let db;
|
|
27
|
+
beforeEach(() => {
|
|
28
|
+
db = createTestDb();
|
|
29
|
+
});
|
|
30
|
+
describe('generateId', () => {
|
|
31
|
+
it('should generate a timestamp-prefixed id', () => {
|
|
32
|
+
const id = generateId();
|
|
33
|
+
expect(id).toMatch(/^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\.\d{3}Z-[a-f0-9]+$/);
|
|
34
|
+
});
|
|
35
|
+
it('should generate unique ids', () => {
|
|
36
|
+
const ids = new Set(Array.from({ length: 100 }, () => generateId()));
|
|
37
|
+
expect(ids.size).toBe(100);
|
|
38
|
+
});
|
|
39
|
+
it('should be time-sortable', () => {
|
|
40
|
+
const id1 = generateId();
|
|
41
|
+
const id2 = `2099-01-01T00:00:00.000Z-ffffff`;
|
|
42
|
+
expect(id1 < id2).toBe(true);
|
|
43
|
+
});
|
|
44
|
+
});
|
|
45
|
+
describe('createConversation', () => {
|
|
46
|
+
it('should create and return a conversation', () => {
|
|
47
|
+
const conv = createConversation(db, 'content-team');
|
|
48
|
+
expect(conv.squad_name).toBe('content-team');
|
|
49
|
+
expect(conv.id).toMatch(/^\d{4}-/);
|
|
50
|
+
expect(conv.created_at).toBeDefined();
|
|
51
|
+
});
|
|
52
|
+
});
|
|
53
|
+
describe('getActiveConversation', () => {
|
|
54
|
+
it('should return null when no conversations exist', () => {
|
|
55
|
+
const conv = getActiveConversation(db, 'content-team');
|
|
56
|
+
expect(conv).toBeNull();
|
|
57
|
+
});
|
|
58
|
+
it('should return the newest conversation', () => {
|
|
59
|
+
const c1 = createConversation(db, 'content-team');
|
|
60
|
+
db.prepare('INSERT INTO conversations (id, squad_name, created_at) VALUES (?, ?, ?)').run('2099-12-31T23:59:59.999Z-zzz', 'content-team', '2099-12-31T23:59:59.999Z');
|
|
61
|
+
const active = getActiveConversation(db, 'content-team');
|
|
62
|
+
expect(active.id).toBe('2099-12-31T23:59:59.999Z-zzz');
|
|
63
|
+
});
|
|
64
|
+
});
|
|
65
|
+
describe('getOrCreateConversation', () => {
|
|
66
|
+
it('should create a new conversation if none exists', () => {
|
|
67
|
+
const result = getOrCreateConversation(db, 'content-team');
|
|
68
|
+
expect(result.conversation.squad_name).toBe('content-team');
|
|
69
|
+
expect(result.created).toBe(true);
|
|
70
|
+
});
|
|
71
|
+
it('should return existing conversation if one exists', () => {
|
|
72
|
+
const first = createConversation(db, 'content-team');
|
|
73
|
+
const result = getOrCreateConversation(db, 'content-team');
|
|
74
|
+
expect(result.conversation.id).toBe(first.id);
|
|
75
|
+
expect(result.created).toBe(false);
|
|
76
|
+
});
|
|
77
|
+
});
|
|
78
|
+
describe('insertMessage', () => {
|
|
79
|
+
it('should insert a user message', () => {
|
|
80
|
+
const conv = createConversation(db, 'content-team');
|
|
81
|
+
const msg = insertMessage(db, {
|
|
82
|
+
conversationId: conv.id,
|
|
83
|
+
role: 'user',
|
|
84
|
+
content: 'Hello',
|
|
85
|
+
});
|
|
86
|
+
expect(msg.id).toMatch(/^\d{4}-/);
|
|
87
|
+
expect(msg.role).toBe('user');
|
|
88
|
+
expect(msg.content).toBe('Hello');
|
|
89
|
+
expect(msg.agent_id).toBeNull();
|
|
90
|
+
});
|
|
91
|
+
it('should insert an assistant message', () => {
|
|
92
|
+
const conv = createConversation(db, 'content-team');
|
|
93
|
+
const msg = insertMessage(db, {
|
|
94
|
+
conversationId: conv.id,
|
|
95
|
+
role: 'assistant',
|
|
96
|
+
content: 'Hi there!',
|
|
97
|
+
});
|
|
98
|
+
expect(msg.role).toBe('assistant');
|
|
99
|
+
});
|
|
100
|
+
it('should insert an agent_dm message with agent metadata', () => {
|
|
101
|
+
const conv = createConversation(db, 'content-team');
|
|
102
|
+
const msg = insertMessage(db, {
|
|
103
|
+
conversationId: conv.id,
|
|
104
|
+
role: 'agent_dm',
|
|
105
|
+
content: 'Found 5 topics',
|
|
106
|
+
agentId: 'researcher',
|
|
107
|
+
agentName: 'Researcher',
|
|
108
|
+
});
|
|
109
|
+
expect(msg.role).toBe('agent_dm');
|
|
110
|
+
expect(msg.agent_id).toBe('researcher');
|
|
111
|
+
expect(msg.agent_name).toBe('Researcher');
|
|
112
|
+
});
|
|
113
|
+
});
|
|
114
|
+
describe('getMessages', () => {
|
|
115
|
+
it('should return messages oldest-first', () => {
|
|
116
|
+
const conv = createConversation(db, 'content-team');
|
|
117
|
+
// Use explicit IDs to guarantee ordering
|
|
118
|
+
db.prepare('INSERT INTO chat_messages (id, conversation_id, role, content, created_at) VALUES (?, ?, ?, ?, ?)').run('2026-03-13T10:00:01.000Z-aaa', conv.id, 'user', 'First', '2026-03-13T10:00:01.000Z');
|
|
119
|
+
db.prepare('INSERT INTO chat_messages (id, conversation_id, role, content, created_at) VALUES (?, ?, ?, ?, ?)').run('2026-03-13T10:00:02.000Z-bbb', conv.id, 'assistant', 'Second', '2026-03-13T10:00:02.000Z');
|
|
120
|
+
db.prepare('INSERT INTO chat_messages (id, conversation_id, role, content, created_at) VALUES (?, ?, ?, ?, ?)').run('2026-03-13T10:00:03.000Z-ccc', conv.id, 'user', 'Third', '2026-03-13T10:00:03.000Z');
|
|
121
|
+
const result = getMessages(db, conv.id, { limit: 50 });
|
|
122
|
+
expect(result.data).toHaveLength(3);
|
|
123
|
+
expect(result.data[0].content).toBe('First');
|
|
124
|
+
expect(result.data[2].content).toBe('Third');
|
|
125
|
+
});
|
|
126
|
+
it('should paginate with cursor', () => {
|
|
127
|
+
const conv = createConversation(db, 'content-team');
|
|
128
|
+
// Use explicit IDs to guarantee ordering
|
|
129
|
+
for (let i = 0; i < 5; i++) {
|
|
130
|
+
const ts = `2026-03-13T10:00:0${i}.000Z`;
|
|
131
|
+
db.prepare('INSERT INTO chat_messages (id, conversation_id, role, content, created_at) VALUES (?, ?, ?, ?, ?)').run(`${ts}-m${i}`, conv.id, 'user', `msg-${i}`, ts);
|
|
132
|
+
}
|
|
133
|
+
const page1 = getMessages(db, conv.id, { limit: 2 });
|
|
134
|
+
expect(page1.data).toHaveLength(2);
|
|
135
|
+
expect(page1.nextCursor).toBeDefined();
|
|
136
|
+
const page2 = getMessages(db, conv.id, { limit: 2, cursor: page1.nextCursor });
|
|
137
|
+
expect(page2.data).toHaveLength(2);
|
|
138
|
+
expect(page2.data[0].content).toBe('msg-2');
|
|
139
|
+
});
|
|
140
|
+
it('should return null nextCursor on last page', () => {
|
|
141
|
+
const conv = createConversation(db, 'content-team');
|
|
142
|
+
insertMessage(db, { conversationId: conv.id, role: 'user', content: 'only' });
|
|
143
|
+
const result = getMessages(db, conv.id, { limit: 50 });
|
|
144
|
+
expect(result.nextCursor).toBeNull();
|
|
145
|
+
});
|
|
146
|
+
});
|
|
147
|
+
describe('getConversations', () => {
|
|
148
|
+
it('should return conversations newest-first', () => {
|
|
149
|
+
createConversation(db, 'content-team');
|
|
150
|
+
db.prepare('INSERT INTO conversations (id, squad_name, created_at) VALUES (?, ?, ?)').run('2099-12-31T23:59:59.999Z-zzz', 'content-team', '2099-12-31T23:59:59.999Z');
|
|
151
|
+
const result = getConversations(db, 'content-team', { limit: 20 });
|
|
152
|
+
expect(result.data).toHaveLength(2);
|
|
153
|
+
expect(result.data[0].id).toBe('2099-12-31T23:59:59.999Z-zzz');
|
|
154
|
+
});
|
|
155
|
+
it('should paginate with cursor (newest-first, so cursor uses < comparison)', () => {
|
|
156
|
+
for (let i = 0; i < 5; i++) {
|
|
157
|
+
const ts = `2026-03-13T10:0${i}:00.000Z`;
|
|
158
|
+
db.prepare('INSERT INTO conversations (id, squad_name, created_at) VALUES (?, ?, ?)').run(`${ts}-c${i}`, 'content-team', ts);
|
|
159
|
+
}
|
|
160
|
+
const page1 = getConversations(db, 'content-team', { limit: 2 });
|
|
161
|
+
expect(page1.data).toHaveLength(2);
|
|
162
|
+
expect(page1.nextCursor).toBeDefined();
|
|
163
|
+
const page2 = getConversations(db, 'content-team', { limit: 2, cursor: page1.nextCursor });
|
|
164
|
+
expect(page2.data).toHaveLength(2);
|
|
165
|
+
});
|
|
166
|
+
});
|
|
167
|
+
});
|
|
168
|
+
//# sourceMappingURL=conversation.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"conversation.test.js","sourceRoot":"","sources":["../../../src/bridge/__tests__/conversation.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC1D,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EACL,kBAAkB,EAClB,qBAAqB,EACrB,uBAAuB,EACvB,aAAa,EACb,WAAW,EACX,gBAAgB,EAChB,UAAU,GACX,MAAM,oBAAoB,CAAC;AAE5B,SAAS,YAAY;IACnB,MAAM,EAAE,GAAG,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC;IACpC,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;IAChC,EAAE,CAAC,IAAI,CAAC;;;;;;;;;;;;;;;GAeP,CAAC,CAAC;IACH,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,EAAqB,CAAC;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,GAAG,YAAY,EAAE,CAAC;IACtB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;YACxB,MAAM,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,yDAAyD,CAAC,CAAC;QAChF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,GAAG,EAAE,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;YACrE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,GAAG,GAAG,UAAU,EAAE,CAAC;YACzB,MAAM,GAAG,GAAG,iCAAiC,CAAC;YAC9C,MAAM,CAAC,GAAG,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,IAAI,GAAG,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YACpD,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC7C,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;QACxC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,IAAI,GAAG,qBAAqB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YACvD,MAAM,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC1B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,EAAE,GAAG,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAClD,EAAE,CAAC,OAAO,CAAC,yEAAyE,CAAC,CAAC,GAAG,CACvF,8BAA8B,EAAE,cAAc,EAAE,0BAA0B,CAC3E,CAAC;YAEF,MAAM,MAAM,GAAG,qBAAqB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YACzD,MAAM,CAAC,MAAO,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,MAAM,GAAG,uBAAuB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,KAAK,GAAG,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YACrD,MAAM,MAAM,GAAG,uBAAuB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,IAAI,GAAG,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YACpD,MAAM,GAAG,GAAG,aAAa,CAAC,EAAE,EAAE;gBAC5B,cAAc,EAAE,IAAI,CAAC,EAAE;gBACvB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,OAAO;aACjB,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;YAClC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC9B,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAClC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,QAAQ,EAAE,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,IAAI,GAAG,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YACpD,MAAM,GAAG,GAAG,aAAa,CAAC,EAAE,EAAE;gBAC5B,cAAc,EAAE,IAAI,CAAC,EAAE;gBACvB,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,WAAW;aACrB,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,IAAI,GAAG,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YACpD,MAAM,GAAG,GAAG,aAAa,CAAC,EAAE,EAAE;gBAC5B,cAAc,EAAE,IAAI,CAAC,EAAE;gBACvB,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,gBAAgB;gBACzB,OAAO,EAAE,YAAY;gBACrB,SAAS,EAAE,YAAY;aACxB,CAAC,CAAC;YACH,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAClC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACxC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,IAAI,GAAG,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YACpD,yCAAyC;YACzC,EAAE,CAAC,OAAO,CACR,mGAAmG,CACpG,CAAC,GAAG,CAAC,8BAA8B,EAAE,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,0BAA0B,CAAC,CAAC;YAC5F,EAAE,CAAC,OAAO,CACR,mGAAmG,CACpG,CAAC,GAAG,CAAC,8BAA8B,EAAE,IAAI,CAAC,EAAE,EAAE,WAAW,EAAE,QAAQ,EAAE,0BAA0B,CAAC,CAAC;YAClG,EAAE,CAAC,OAAO,CACR,mGAAmG,CACpG,CAAC,GAAG,CAAC,8BAA8B,EAAE,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,0BAA0B,CAAC,CAAC;YAE5F,MAAM,MAAM,GAAG,WAAW,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;YACrC,MAAM,IAAI,GAAG,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YACpD,yCAAyC;YACzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3B,MAAM,EAAE,GAAG,qBAAqB,CAAC,OAAO,CAAC;gBACzC,EAAE,CAAC,OAAO,CACR,mGAAmG,CACpG,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YACxD,CAAC;YAED,MAAM,KAAK,GAAG,WAAW,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YACrD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;YAEvC,MAAM,KAAK,GAAG,WAAW,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,UAAW,EAAE,CAAC,CAAC;YAChF,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,IAAI,GAAG,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YACpD,aAAa,CAAC,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAE9E,MAAM,MAAM,GAAG,WAAW,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,QAAQ,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,kBAAkB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC;YACvC,EAAE,CAAC,OAAO,CAAC,yEAAyE,CAAC,CAAC,GAAG,CACvF,8BAA8B,EAAE,cAAc,EAAE,0BAA0B,CAC3E,CAAC;YAEF,MAAM,MAAM,GAAG,gBAAgB,CAAC,EAAE,EAAE,cAAc,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;YACjF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3B,MAAM,EAAE,GAAG,kBAAkB,CAAC,UAAU,CAAC;gBACzC,EAAE,CAAC,OAAO,CAAC,yEAAyE,CAAC,CAAC,GAAG,CACvF,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,cAAc,EAAE,EAAE,CAClC,CAAC;YACJ,CAAC;YAED,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,EAAE,cAAc,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YACjE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACnC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;YAEvC,MAAM,KAAK,GAAG,gBAAgB,CAAC,EAAE,EAAE,cAAc,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,UAAW,EAAE,CAAC,CAAC;YAC5F,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream-parser.test.d.ts","sourceRoot":"","sources":["../../../src/bridge/__tests__/stream-parser.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
import { describe, it, expect, vi } from 'vitest';
|
|
2
|
+
import { StreamParser } from '../stream-parser.js';
|
|
3
|
+
describe('StreamParser', () => {
|
|
4
|
+
it('should emit chunks as they arrive', () => {
|
|
5
|
+
const onChunk = vi.fn();
|
|
6
|
+
const onDone = vi.fn();
|
|
7
|
+
const parser = new StreamParser({ onChunk, onDone });
|
|
8
|
+
parser.write('Hello ');
|
|
9
|
+
parser.write('world');
|
|
10
|
+
parser.end();
|
|
11
|
+
expect(onChunk).toHaveBeenCalledTimes(2);
|
|
12
|
+
expect(onChunk).toHaveBeenNthCalledWith(1, 'Hello ');
|
|
13
|
+
expect(onChunk).toHaveBeenNthCalledWith(2, 'world');
|
|
14
|
+
expect(onDone).toHaveBeenCalledOnce();
|
|
15
|
+
expect(onDone).toHaveBeenCalledWith('Hello world');
|
|
16
|
+
});
|
|
17
|
+
it('should handle empty writes', () => {
|
|
18
|
+
const onChunk = vi.fn();
|
|
19
|
+
const onDone = vi.fn();
|
|
20
|
+
const parser = new StreamParser({ onChunk, onDone });
|
|
21
|
+
parser.write('');
|
|
22
|
+
parser.write('text');
|
|
23
|
+
parser.end();
|
|
24
|
+
expect(onChunk).toHaveBeenCalledTimes(1);
|
|
25
|
+
expect(onChunk).toHaveBeenCalledWith('text');
|
|
26
|
+
expect(onDone).toHaveBeenCalledWith('text');
|
|
27
|
+
});
|
|
28
|
+
it('should handle Buffer input', () => {
|
|
29
|
+
const onChunk = vi.fn();
|
|
30
|
+
const onDone = vi.fn();
|
|
31
|
+
const parser = new StreamParser({ onChunk, onDone });
|
|
32
|
+
parser.write(Buffer.from('Hello'));
|
|
33
|
+
parser.end();
|
|
34
|
+
expect(onChunk).toHaveBeenCalledWith('Hello');
|
|
35
|
+
expect(onDone).toHaveBeenCalledWith('Hello');
|
|
36
|
+
});
|
|
37
|
+
it('should accumulate full response for done callback', () => {
|
|
38
|
+
const onChunk = vi.fn();
|
|
39
|
+
const onDone = vi.fn();
|
|
40
|
+
const parser = new StreamParser({ onChunk, onDone });
|
|
41
|
+
parser.write('Part 1. ');
|
|
42
|
+
parser.write('Part 2. ');
|
|
43
|
+
parser.write('Part 3.');
|
|
44
|
+
parser.end();
|
|
45
|
+
expect(onDone).toHaveBeenCalledWith('Part 1. Part 2. Part 3.');
|
|
46
|
+
});
|
|
47
|
+
it('should handle end without any writes', () => {
|
|
48
|
+
const onChunk = vi.fn();
|
|
49
|
+
const onDone = vi.fn();
|
|
50
|
+
const parser = new StreamParser({ onChunk, onDone });
|
|
51
|
+
parser.end();
|
|
52
|
+
expect(onChunk).not.toHaveBeenCalled();
|
|
53
|
+
expect(onDone).toHaveBeenCalledWith('');
|
|
54
|
+
});
|
|
55
|
+
it('should ignore writes after end', () => {
|
|
56
|
+
const onChunk = vi.fn();
|
|
57
|
+
const onDone = vi.fn();
|
|
58
|
+
const parser = new StreamParser({ onChunk, onDone });
|
|
59
|
+
parser.write('before');
|
|
60
|
+
parser.end();
|
|
61
|
+
parser.write('after');
|
|
62
|
+
expect(onChunk).toHaveBeenCalledTimes(1);
|
|
63
|
+
expect(onDone).toHaveBeenCalledOnce();
|
|
64
|
+
});
|
|
65
|
+
});
|
|
66
|
+
//# sourceMappingURL=stream-parser.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream-parser.test.js","sourceRoot":"","sources":["../../../src/bridge/__tests__/stream-parser.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAEnD,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAErD,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACvB,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACtB,MAAM,CAAC,GAAG,EAAE,CAAC;QAEb,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,CAAC,uBAAuB,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,EAAE,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAErD,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACjB,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACrB,MAAM,CAAC,GAAG,EAAE,CAAC;QAEb,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;QAC7C,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAErD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,GAAG,EAAE,CAAC;QAEb,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAErD,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACzB,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QACzB,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACxB,MAAM,CAAC,GAAG,EAAE,CAAC;QAEb,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,yBAAyB,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAErD,MAAM,CAAC,GAAG,EAAE,CAAC;QAEb,MAAM,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,OAAO,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAErD,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACvB,MAAM,CAAC,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAEtB,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,CAAC,oBAAoB,EAAE,CAAC;IACxC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import type Database from 'better-sqlite3';
|
|
2
|
+
import type { ClaudeBridge } from './claude-bridge.js';
|
|
3
|
+
export interface ChatHandlerConfig {
|
|
4
|
+
db: Database.Database;
|
|
5
|
+
bridge: ClaudeBridge;
|
|
6
|
+
broadcast: (squadName: string, message: Record<string, unknown>) => void;
|
|
7
|
+
squadsDir: string;
|
|
8
|
+
}
|
|
9
|
+
export declare class ChatHandler {
|
|
10
|
+
private db;
|
|
11
|
+
private bridge;
|
|
12
|
+
private broadcast;
|
|
13
|
+
private squadsDir;
|
|
14
|
+
constructor(config: ChatHandlerConfig);
|
|
15
|
+
handleChatSend(squadName: string, message: string): void;
|
|
16
|
+
handleNewConversation(squadName: string): void;
|
|
17
|
+
private buildPrompt;
|
|
18
|
+
}
|
|
19
|
+
//# sourceMappingURL=chat-handler.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat-handler.d.ts","sourceRoot":"","sources":["../../src/bridge/chat-handler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,QAAQ,MAAM,gBAAgB,CAAC;AAC3C,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAUvD,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,QAAQ,CAAC,QAAQ,CAAC;IACtB,MAAM,EAAE,YAAY,CAAC;IACrB,SAAS,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;IACzE,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,EAAE,CAAoB;IAC9B,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,SAAS,CAAgE;IACjF,OAAO,CAAC,SAAS,CAAS;gBAEd,MAAM,EAAE,iBAAiB;IAOrC,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAgExD,qBAAqB,CAAC,SAAS,EAAE,MAAM,GAAG,IAAI;IAS9C,OAAO,CAAC,WAAW;CAyBpB"}
|
|
@@ -0,0 +1,104 @@
|
|
|
1
|
+
import { createConversation, getOrCreateConversation, getRecentMessages, insertMessage, } from './conversation.js';
|
|
2
|
+
import path from 'path';
|
|
3
|
+
export class ChatHandler {
|
|
4
|
+
db;
|
|
5
|
+
bridge;
|
|
6
|
+
broadcast;
|
|
7
|
+
squadsDir;
|
|
8
|
+
constructor(config) {
|
|
9
|
+
this.db = config.db;
|
|
10
|
+
this.bridge = config.bridge;
|
|
11
|
+
this.broadcast = config.broadcast;
|
|
12
|
+
this.squadsDir = config.squadsDir;
|
|
13
|
+
}
|
|
14
|
+
handleChatSend(squadName, message) {
|
|
15
|
+
const { conversation, created } = getOrCreateConversation(this.db, squadName);
|
|
16
|
+
if (created) {
|
|
17
|
+
this.broadcast(squadName, {
|
|
18
|
+
type: 'chat:conversation_created',
|
|
19
|
+
squadName,
|
|
20
|
+
conversationId: conversation.id,
|
|
21
|
+
});
|
|
22
|
+
}
|
|
23
|
+
insertMessage(this.db, {
|
|
24
|
+
conversationId: conversation.id,
|
|
25
|
+
role: 'user',
|
|
26
|
+
content: message,
|
|
27
|
+
});
|
|
28
|
+
const prompt = this.buildPrompt(squadName, conversation.id, message);
|
|
29
|
+
this.bridge.send({
|
|
30
|
+
prompt,
|
|
31
|
+
cwd: path.join(this.squadsDir, squadName),
|
|
32
|
+
squadName,
|
|
33
|
+
onOutputChunk: (agentId, agentName, delta) => {
|
|
34
|
+
this.broadcast(squadName, {
|
|
35
|
+
type: 'agent:output_chunk',
|
|
36
|
+
squadName,
|
|
37
|
+
agentId,
|
|
38
|
+
agentName,
|
|
39
|
+
delta,
|
|
40
|
+
});
|
|
41
|
+
},
|
|
42
|
+
onChunk: (text) => {
|
|
43
|
+
this.broadcast(squadName, {
|
|
44
|
+
type: 'chat:chunk',
|
|
45
|
+
squadName,
|
|
46
|
+
conversationId: conversation.id,
|
|
47
|
+
delta: text,
|
|
48
|
+
});
|
|
49
|
+
},
|
|
50
|
+
onDone: (fullText) => {
|
|
51
|
+
const assistantMsg = insertMessage(this.db, {
|
|
52
|
+
conversationId: conversation.id,
|
|
53
|
+
role: 'assistant',
|
|
54
|
+
content: fullText,
|
|
55
|
+
});
|
|
56
|
+
this.broadcast(squadName, {
|
|
57
|
+
type: 'chat:done',
|
|
58
|
+
squadName,
|
|
59
|
+
conversationId: conversation.id,
|
|
60
|
+
messageId: assistantMsg.id,
|
|
61
|
+
});
|
|
62
|
+
},
|
|
63
|
+
onError: (error) => {
|
|
64
|
+
this.broadcast(squadName, {
|
|
65
|
+
type: 'chat:error',
|
|
66
|
+
squadName,
|
|
67
|
+
error: error.message,
|
|
68
|
+
});
|
|
69
|
+
},
|
|
70
|
+
});
|
|
71
|
+
}
|
|
72
|
+
handleNewConversation(squadName) {
|
|
73
|
+
const conversation = createConversation(this.db, squadName);
|
|
74
|
+
this.broadcast(squadName, {
|
|
75
|
+
type: 'chat:conversation_created',
|
|
76
|
+
squadName,
|
|
77
|
+
conversationId: conversation.id,
|
|
78
|
+
});
|
|
79
|
+
}
|
|
80
|
+
buildPrompt(squadName, conversationId, userMessage) {
|
|
81
|
+
const history = getRecentMessages(this.db, conversationId, 10);
|
|
82
|
+
const historyText = history
|
|
83
|
+
.map((msg) => {
|
|
84
|
+
if (msg.role === 'user')
|
|
85
|
+
return `User: ${msg.content}`;
|
|
86
|
+
if (msg.role === 'assistant')
|
|
87
|
+
return `Assistant: ${msg.content}`;
|
|
88
|
+
if (msg.role === 'agent_dm')
|
|
89
|
+
return `[Agent ${msg.agent_name}]: ${msg.content}`;
|
|
90
|
+
return '';
|
|
91
|
+
})
|
|
92
|
+
.filter(Boolean)
|
|
93
|
+
.join('\n');
|
|
94
|
+
const parts = [
|
|
95
|
+
`You are assisting with squad '${squadName}'.`,
|
|
96
|
+
];
|
|
97
|
+
if (historyText) {
|
|
98
|
+
parts.push(`\nConversation:\n${historyText}`);
|
|
99
|
+
}
|
|
100
|
+
parts.push(`\nUser: ${userMessage}`);
|
|
101
|
+
return parts.join('');
|
|
102
|
+
}
|
|
103
|
+
}
|
|
104
|
+
//# sourceMappingURL=chat-handler.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"chat-handler.js","sourceRoot":"","sources":["../../src/bridge/chat-handler.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,kBAAkB,EAClB,uBAAuB,EACvB,iBAAiB,EACjB,aAAa,GACd,MAAM,mBAAmB,CAAC;AAE3B,OAAO,IAAI,MAAM,MAAM,CAAC;AASxB,MAAM,OAAO,WAAW;IACd,EAAE,CAAoB;IACtB,MAAM,CAAe;IACrB,SAAS,CAAgE;IACzE,SAAS,CAAS;IAE1B,YAAY,MAAyB;QACnC,IAAI,CAAC,EAAE,GAAG,MAAM,CAAC,EAAE,CAAC;QACpB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAClC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;IACpC,CAAC;IAED,cAAc,CAAC,SAAiB,EAAE,OAAe;QAC/C,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,GAAG,uBAAuB,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAE9E,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE;gBACxB,IAAI,EAAE,2BAA2B;gBACjC,SAAS;gBACT,cAAc,EAAE,YAAY,CAAC,EAAE;aAChC,CAAC,CAAC;QACL,CAAC;QAED,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE;YACrB,cAAc,EAAE,YAAY,CAAC,EAAE;YAC/B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,YAAY,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAErE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;YACf,MAAM;YACN,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,CAAC;YACzC,SAAS;YACT,aAAa,EAAE,CAAC,OAAe,EAAE,SAAiB,EAAE,KAAa,EAAE,EAAE;gBACnE,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE;oBACxB,IAAI,EAAE,oBAAoB;oBAC1B,SAAS;oBACT,OAAO;oBACP,SAAS;oBACT,KAAK;iBACN,CAAC,CAAC;YACL,CAAC;YACD,OAAO,EAAE,CAAC,IAAY,EAAE,EAAE;gBACxB,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE;oBACxB,IAAI,EAAE,YAAY;oBAClB,SAAS;oBACT,cAAc,EAAE,YAAY,CAAC,EAAE;oBAC/B,KAAK,EAAE,IAAI;iBACZ,CAAC,CAAC;YACL,CAAC;YACD,MAAM,EAAE,CAAC,QAAgB,EAAE,EAAE;gBAC3B,MAAM,YAAY,GAAG,aAAa,CAAC,IAAI,CAAC,EAAE,EAAE;oBAC1C,cAAc,EAAE,YAAY,CAAC,EAAE;oBAC/B,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,QAAQ;iBAClB,CAAC,CAAC;gBAEH,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE;oBACxB,IAAI,EAAE,WAAW;oBACjB,SAAS;oBACT,cAAc,EAAE,YAAY,CAAC,EAAE;oBAC/B,SAAS,EAAE,YAAY,CAAC,EAAE;iBAC3B,CAAC,CAAC;YACL,CAAC;YACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACjB,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE;oBACxB,IAAI,EAAE,YAAY;oBAClB,SAAS;oBACT,KAAK,EAAE,KAAK,CAAC,OAAO;iBACrB,CAAC,CAAC;YACL,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,qBAAqB,CAAC,SAAiB;QACrC,MAAM,YAAY,GAAG,kBAAkB,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAC5D,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE;YACxB,IAAI,EAAE,2BAA2B;YACjC,SAAS;YACT,cAAc,EAAE,YAAY,CAAC,EAAE;SAChC,CAAC,CAAC;IACL,CAAC;IAEO,WAAW,CAAC,SAAiB,EAAE,cAAsB,EAAE,WAAmB;QAChF,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,CAAC,EAAE,EAAE,cAAc,EAAE,EAAE,CAAC,CAAC;QAE/D,MAAM,WAAW,GAAG,OAAO;aACxB,GAAG,CAAC,CAAC,GAAgB,EAAE,EAAE;YACxB,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM;gBAAE,OAAO,SAAS,GAAG,CAAC,OAAO,EAAE,CAAC;YACvD,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW;gBAAE,OAAO,cAAc,GAAG,CAAC,OAAO,EAAE,CAAC;YACjE,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU;gBAAE,OAAO,UAAU,GAAG,CAAC,UAAU,MAAM,GAAG,CAAC,OAAO,EAAE,CAAC;YAChF,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC;aACD,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,CAAC,IAAI,CAAC,CAAC;QAEd,MAAM,KAAK,GAAa;YACtB,iCAAiC,SAAS,IAAI;SAC/C,CAAC;QAEF,IAAI,WAAW,EAAE,CAAC;YAChB,KAAK,CAAC,IAAI,CAAC,oBAAoB,WAAW,EAAE,CAAC,CAAC;QAChD,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,WAAW,WAAW,EAAE,CAAC,CAAC;QAErC,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxB,CAAC;CACF"}
|