@snoglobe/helios 0.3.6 → 0.4.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/dist/__tests__/db-helper.d.ts +8 -0
- package/dist/__tests__/db-helper.d.ts.map +1 -0
- package/dist/__tests__/db-helper.js +15 -0
- package/dist/__tests__/db-helper.js.map +1 -0
- package/dist/config/project.test.d.ts +2 -0
- package/dist/config/project.test.d.ts.map +1 -0
- package/dist/config/project.test.js +219 -0
- package/dist/config/project.test.js.map +1 -0
- package/dist/core/orchestrator-integration.test.d.ts +2 -0
- package/dist/core/orchestrator-integration.test.d.ts.map +1 -0
- package/dist/core/orchestrator-integration.test.js +674 -0
- package/dist/core/orchestrator-integration.test.js.map +1 -0
- package/dist/core/orchestrator.d.ts +1 -0
- package/dist/core/orchestrator.d.ts.map +1 -1
- package/dist/core/orchestrator.js +81 -37
- package/dist/core/orchestrator.js.map +1 -1
- package/dist/core/orchestrator.test.d.ts +2 -0
- package/dist/core/orchestrator.test.d.ts.map +1 -0
- package/dist/core/orchestrator.test.js +982 -0
- package/dist/core/orchestrator.test.js.map +1 -0
- package/dist/core/state-machine.d.ts.map +1 -1
- package/dist/core/state-machine.js +3 -0
- package/dist/core/state-machine.js.map +1 -1
- package/dist/core/state-machine.test.d.ts +2 -0
- package/dist/core/state-machine.test.d.ts.map +1 -0
- package/dist/core/state-machine.test.js +95 -0
- package/dist/core/state-machine.test.js.map +1 -0
- package/dist/core/stickies.d.ts.map +1 -1
- package/dist/core/stickies.js +3 -0
- package/dist/core/stickies.js.map +1 -1
- package/dist/core/stickies.test.d.ts +2 -0
- package/dist/core/stickies.test.d.ts.map +1 -0
- package/dist/core/stickies.test.js +63 -0
- package/dist/core/stickies.test.js.map +1 -0
- package/dist/core/task-poller.d.ts.map +1 -1
- package/dist/core/task-poller.js +2 -1
- package/dist/core/task-poller.js.map +1 -1
- package/dist/init.d.ts +2 -0
- package/dist/init.d.ts.map +1 -1
- package/dist/init.js +19 -0
- package/dist/init.js.map +1 -1
- package/dist/memory/context-gate.test.d.ts +2 -0
- package/dist/memory/context-gate.test.d.ts.map +1 -0
- package/dist/memory/context-gate.test.js +318 -0
- package/dist/memory/context-gate.test.js.map +1 -0
- package/dist/memory/experiment-tracker.test.d.ts +2 -0
- package/dist/memory/experiment-tracker.test.d.ts.map +1 -0
- package/dist/memory/experiment-tracker.test.js +325 -0
- package/dist/memory/experiment-tracker.test.js.map +1 -0
- package/dist/memory/memory-store.d.ts +2 -0
- package/dist/memory/memory-store.d.ts.map +1 -1
- package/dist/memory/memory-store.js +35 -28
- package/dist/memory/memory-store.js.map +1 -1
- package/dist/memory/memory-store.test.d.ts +2 -0
- package/dist/memory/memory-store.test.d.ts.map +1 -0
- package/dist/memory/memory-store.test.js +144 -0
- package/dist/memory/memory-store.test.js.map +1 -0
- package/dist/memory/token-estimator.test.d.ts +2 -0
- package/dist/memory/token-estimator.test.d.ts.map +1 -0
- package/dist/memory/token-estimator.test.js +42 -0
- package/dist/memory/token-estimator.test.js.map +1 -0
- package/dist/metrics/analyzer.test.d.ts +2 -0
- package/dist/metrics/analyzer.test.d.ts.map +1 -0
- package/dist/metrics/analyzer.test.js +65 -0
- package/dist/metrics/analyzer.test.js.map +1 -0
- package/dist/metrics/parser.test.d.ts +2 -0
- package/dist/metrics/parser.test.d.ts.map +1 -0
- package/dist/metrics/parser.test.js +106 -0
- package/dist/metrics/parser.test.js.map +1 -0
- package/dist/metrics/store.d.ts +2 -0
- package/dist/metrics/store.d.ts.map +1 -1
- package/dist/metrics/store.js +26 -50
- package/dist/metrics/store.js.map +1 -1
- package/dist/metrics/store.test.d.ts +2 -0
- package/dist/metrics/store.test.d.ts.map +1 -0
- package/dist/metrics/store.test.js +148 -0
- package/dist/metrics/store.test.js.map +1 -0
- package/dist/paths.js +5 -5
- package/dist/paths.js.map +1 -1
- package/dist/providers/auth/auth-manager.d.ts +1 -0
- package/dist/providers/auth/auth-manager.d.ts.map +1 -1
- package/dist/providers/auth/auth-manager.js +8 -1
- package/dist/providers/auth/auth-manager.js.map +1 -1
- package/dist/providers/auth/auth-manager.test.d.ts +2 -0
- package/dist/providers/auth/auth-manager.test.d.ts.map +1 -0
- package/dist/providers/auth/auth-manager.test.js +364 -0
- package/dist/providers/auth/auth-manager.test.js.map +1 -0
- package/dist/providers/auth/token-store.js +2 -2
- package/dist/providers/auth/token-store.js.map +1 -1
- package/dist/providers/claude/history.test.d.ts +2 -0
- package/dist/providers/claude/history.test.d.ts.map +1 -0
- package/dist/providers/claude/history.test.js +755 -0
- package/dist/providers/claude/history.test.js.map +1 -0
- package/dist/providers/claude/provider.d.ts +3 -1
- package/dist/providers/claude/provider.d.ts.map +1 -1
- package/dist/providers/claude/provider.js +33 -19
- package/dist/providers/claude/provider.js.map +1 -1
- package/dist/providers/claude/provider.test.d.ts +2 -0
- package/dist/providers/claude/provider.test.d.ts.map +1 -0
- package/dist/providers/claude/provider.test.js +1167 -0
- package/dist/providers/claude/provider.test.js.map +1 -0
- package/dist/providers/openai/history.test.d.ts +2 -0
- package/dist/providers/openai/history.test.d.ts.map +1 -0
- package/dist/providers/openai/history.test.js +655 -0
- package/dist/providers/openai/history.test.js.map +1 -0
- package/dist/providers/openai/provider.d.ts +2 -1
- package/dist/providers/openai/provider.d.ts.map +1 -1
- package/dist/providers/openai/provider.js +14 -8
- package/dist/providers/openai/provider.js.map +1 -1
- package/dist/providers/openai/provider.test.d.ts +2 -0
- package/dist/providers/openai/provider.test.d.ts.map +1 -0
- package/dist/providers/openai/provider.test.js +1089 -0
- package/dist/providers/openai/provider.test.js.map +1 -0
- package/dist/providers/retry.test.d.ts +2 -0
- package/dist/providers/retry.test.d.ts.map +1 -0
- package/dist/providers/retry.test.js +194 -0
- package/dist/providers/retry.test.js.map +1 -0
- package/dist/providers/sse.d.ts +1 -0
- package/dist/providers/sse.d.ts.map +1 -1
- package/dist/providers/sse.js +29 -20
- package/dist/providers/sse.js.map +1 -1
- package/dist/providers/sse.test.d.ts +2 -0
- package/dist/providers/sse.test.d.ts.map +1 -0
- package/dist/providers/sse.test.js +79 -0
- package/dist/providers/sse.test.js.map +1 -0
- package/dist/providers/types.test.d.ts +2 -0
- package/dist/providers/types.test.d.ts.map +1 -0
- package/dist/providers/types.test.js +112 -0
- package/dist/providers/types.test.js.map +1 -0
- package/dist/remote/connection-pool.d.ts.map +1 -1
- package/dist/remote/connection-pool.js +24 -3
- package/dist/remote/connection-pool.js.map +1 -1
- package/dist/remote/file-sync.d.ts.map +1 -1
- package/dist/remote/file-sync.js +4 -3
- package/dist/remote/file-sync.js.map +1 -1
- package/dist/scheduler/sleep-manager.d.ts.map +1 -1
- package/dist/scheduler/sleep-manager.js +8 -1
- package/dist/scheduler/sleep-manager.js.map +1 -1
- package/dist/scheduler/sleep-manager.test.d.ts +2 -0
- package/dist/scheduler/sleep-manager.test.d.ts.map +1 -0
- package/dist/scheduler/sleep-manager.test.js +491 -0
- package/dist/scheduler/sleep-manager.test.js.map +1 -0
- package/dist/scheduler/ssh-batcher.d.ts.map +1 -1
- package/dist/scheduler/ssh-batcher.js +6 -4
- package/dist/scheduler/ssh-batcher.js.map +1 -1
- package/dist/scheduler/ssh-batcher.test.d.ts +2 -0
- package/dist/scheduler/ssh-batcher.test.d.ts.map +1 -0
- package/dist/scheduler/ssh-batcher.test.js +76 -0
- package/dist/scheduler/ssh-batcher.test.js.map +1 -0
- package/dist/scheduler/state-store.d.ts.map +1 -1
- package/dist/scheduler/state-store.js +1 -2
- package/dist/scheduler/state-store.js.map +1 -1
- package/dist/scheduler/trigger-scheduler.d.ts.map +1 -1
- package/dist/scheduler/trigger-scheduler.js +59 -36
- package/dist/scheduler/trigger-scheduler.js.map +1 -1
- package/dist/scheduler/trigger-scheduler.test.d.ts +2 -0
- package/dist/scheduler/trigger-scheduler.test.d.ts.map +1 -0
- package/dist/scheduler/trigger-scheduler.test.js +483 -0
- package/dist/scheduler/trigger-scheduler.test.js.map +1 -0
- package/dist/scheduler/triggers/file.d.ts.map +1 -1
- package/dist/scheduler/triggers/file.js +12 -3
- package/dist/scheduler/triggers/file.js.map +1 -1
- package/dist/scheduler/triggers/file.test.d.ts +2 -0
- package/dist/scheduler/triggers/file.test.d.ts.map +1 -0
- package/dist/scheduler/triggers/file.test.js +294 -0
- package/dist/scheduler/triggers/file.test.js.map +1 -0
- package/dist/scheduler/triggers/metric.d.ts +3 -1
- package/dist/scheduler/triggers/metric.d.ts.map +1 -1
- package/dist/scheduler/triggers/metric.js +12 -8
- package/dist/scheduler/triggers/metric.js.map +1 -1
- package/dist/scheduler/triggers/metric.test.d.ts +2 -0
- package/dist/scheduler/triggers/metric.test.d.ts.map +1 -0
- package/dist/scheduler/triggers/metric.test.js +533 -0
- package/dist/scheduler/triggers/metric.test.js.map +1 -0
- package/dist/scheduler/triggers/process-exit.d.ts.map +1 -1
- package/dist/scheduler/triggers/process-exit.js +2 -1
- package/dist/scheduler/triggers/process-exit.js.map +1 -1
- package/dist/scheduler/triggers/process-exit.test.d.ts +2 -0
- package/dist/scheduler/triggers/process-exit.test.d.ts.map +1 -0
- package/dist/scheduler/triggers/process-exit.test.js +118 -0
- package/dist/scheduler/triggers/process-exit.test.js.map +1 -0
- package/dist/scheduler/triggers/resource.d.ts.map +1 -1
- package/dist/scheduler/triggers/resource.js +2 -10
- package/dist/scheduler/triggers/resource.js.map +1 -1
- package/dist/scheduler/triggers/resource.test.d.ts +2 -0
- package/dist/scheduler/triggers/resource.test.d.ts.map +1 -0
- package/dist/scheduler/triggers/resource.test.js +225 -0
- package/dist/scheduler/triggers/resource.test.js.map +1 -0
- package/dist/scheduler/triggers/timer.test.d.ts +2 -0
- package/dist/scheduler/triggers/timer.test.d.ts.map +1 -0
- package/dist/scheduler/triggers/timer.test.js +56 -0
- package/dist/scheduler/triggers/timer.test.js.map +1 -0
- package/dist/scheduler/triggers/types.d.ts +4 -2
- package/dist/scheduler/triggers/types.d.ts.map +1 -1
- package/dist/scheduler/triggers/types.js +0 -1
- package/dist/scheduler/triggers/types.js.map +1 -1
- package/dist/skills/executor.d.ts.map +1 -1
- package/dist/skills/executor.js +13 -15
- package/dist/skills/executor.js.map +1 -1
- package/dist/store/database.d.ts +5 -0
- package/dist/store/database.d.ts.map +1 -1
- package/dist/store/database.js +17 -1
- package/dist/store/database.js.map +1 -1
- package/dist/store/migrations.test.d.ts +2 -0
- package/dist/store/migrations.test.d.ts.map +1 -0
- package/dist/store/migrations.test.js +278 -0
- package/dist/store/migrations.test.js.map +1 -0
- package/dist/store/session-store-edge.test.d.ts +2 -0
- package/dist/store/session-store-edge.test.d.ts.map +1 -0
- package/dist/store/session-store-edge.test.js +522 -0
- package/dist/store/session-store-edge.test.js.map +1 -0
- package/dist/store/session-store.d.ts +6 -0
- package/dist/store/session-store.d.ts.map +1 -1
- package/dist/store/session-store.js +25 -24
- package/dist/store/session-store.js.map +1 -1
- package/dist/store/session-store.test.d.ts +2 -0
- package/dist/store/session-store.test.d.ts.map +1 -0
- package/dist/store/session-store.test.js +125 -0
- package/dist/store/session-store.test.js.map +1 -0
- package/dist/subagent/executor.d.ts +21 -0
- package/dist/subagent/executor.d.ts.map +1 -0
- package/dist/subagent/executor.js +136 -0
- package/dist/subagent/executor.js.map +1 -0
- package/dist/subagent/manager.d.ts +20 -0
- package/dist/subagent/manager.d.ts.map +1 -0
- package/dist/subagent/manager.js +98 -0
- package/dist/subagent/manager.js.map +1 -0
- package/dist/subagent/scoped-memory.d.ts +28 -0
- package/dist/subagent/scoped-memory.d.ts.map +1 -0
- package/dist/subagent/scoped-memory.js +122 -0
- package/dist/subagent/scoped-memory.js.map +1 -0
- package/dist/subagent/types.d.ts +27 -0
- package/dist/subagent/types.d.ts.map +1 -0
- package/dist/subagent/types.js +2 -0
- package/dist/subagent/types.js.map +1 -0
- package/dist/tools/file-ops.d.ts.map +1 -1
- package/dist/tools/file-ops.js +23 -13
- package/dist/tools/file-ops.js.map +1 -1
- package/dist/tools/file-ops.test.d.ts +2 -0
- package/dist/tools/file-ops.test.d.ts.map +1 -0
- package/dist/tools/file-ops.test.js +656 -0
- package/dist/tools/file-ops.test.js.map +1 -0
- package/dist/tools/memory-tools.test.d.ts +2 -0
- package/dist/tools/memory-tools.test.d.ts.map +1 -0
- package/dist/tools/memory-tools.test.js +273 -0
- package/dist/tools/memory-tools.test.js.map +1 -0
- package/dist/tools/subagent.d.ts +10 -0
- package/dist/tools/subagent.d.ts.map +1 -0
- package/dist/tools/subagent.js +164 -0
- package/dist/tools/subagent.js.map +1 -0
- package/dist/tools/task-output.d.ts.map +1 -1
- package/dist/tools/task-output.js +13 -2
- package/dist/tools/task-output.js.map +1 -1
- package/dist/ui/components/input-bar.d.ts +1 -1
- package/dist/ui/components/input-bar.d.ts.map +1 -1
- package/dist/ui/components/input-bar.js +3 -3
- package/dist/ui/components/input-bar.js.map +1 -1
- package/dist/ui/components/input-bar.test.d.ts +2 -0
- package/dist/ui/components/input-bar.test.d.ts.map +1 -0
- package/dist/ui/components/input-bar.test.js +380 -0
- package/dist/ui/components/input-bar.test.js.map +1 -0
- package/dist/ui/components/key-hint-rule.d.ts +2 -1
- package/dist/ui/components/key-hint-rule.d.ts.map +1 -1
- package/dist/ui/components/key-hint-rule.js +3 -3
- package/dist/ui/components/key-hint-rule.js.map +1 -1
- package/dist/ui/components/status-bar.d.ts +1 -1
- package/dist/ui/components/status-bar.d.ts.map +1 -1
- package/dist/ui/components/status-bar.js +11 -10
- package/dist/ui/components/status-bar.js.map +1 -1
- package/dist/ui/components/status-bar.test.d.ts +2 -0
- package/dist/ui/components/status-bar.test.d.ts.map +1 -0
- package/dist/ui/components/status-bar.test.js +206 -0
- package/dist/ui/components/status-bar.test.js.map +1 -0
- package/dist/ui/format.d.ts +9 -0
- package/dist/ui/format.d.ts.map +1 -1
- package/dist/ui/format.js +21 -0
- package/dist/ui/format.js.map +1 -1
- package/dist/ui/format.test.d.ts +2 -0
- package/dist/ui/format.test.d.ts.map +1 -0
- package/dist/ui/format.test.js +122 -0
- package/dist/ui/format.test.js.map +1 -0
- package/dist/ui/layout.d.ts.map +1 -1
- package/dist/ui/layout.js +63 -11
- package/dist/ui/layout.js.map +1 -1
- package/dist/ui/markdown.test.d.ts +2 -0
- package/dist/ui/markdown.test.d.ts.map +1 -0
- package/dist/ui/markdown.test.js +133 -0
- package/dist/ui/markdown.test.js.map +1 -0
- package/dist/ui/mouse-filter.test.d.ts +2 -0
- package/dist/ui/mouse-filter.test.d.ts.map +1 -0
- package/dist/ui/mouse-filter.test.js +231 -0
- package/dist/ui/mouse-filter.test.js.map +1 -0
- package/dist/ui/overlays/metrics-overlay.test.d.ts +2 -0
- package/dist/ui/overlays/metrics-overlay.test.d.ts.map +1 -0
- package/dist/ui/overlays/metrics-overlay.test.js +248 -0
- package/dist/ui/overlays/metrics-overlay.test.js.map +1 -0
- package/dist/ui/overlays/task-overlay.test.d.ts +2 -0
- package/dist/ui/overlays/task-overlay.test.d.ts.map +1 -0
- package/dist/ui/overlays/task-overlay.test.js +238 -0
- package/dist/ui/overlays/task-overlay.test.js.map +1 -0
- package/dist/ui/panels/conversation.d.ts.map +1 -1
- package/dist/ui/panels/conversation.js +60 -5
- package/dist/ui/panels/conversation.js.map +1 -1
- package/dist/ui/panels/conversation.test.d.ts +2 -0
- package/dist/ui/panels/conversation.test.d.ts.map +1 -0
- package/dist/ui/panels/conversation.test.js +381 -0
- package/dist/ui/panels/conversation.test.js.map +1 -0
- package/dist/ui/panels/metrics-dashboard.test.d.ts +2 -0
- package/dist/ui/panels/metrics-dashboard.test.d.ts.map +1 -0
- package/dist/ui/panels/metrics-dashboard.test.js +191 -0
- package/dist/ui/panels/metrics-dashboard.test.js.map +1 -0
- package/dist/ui/panels/sleep-panel.test.d.ts +2 -0
- package/dist/ui/panels/sleep-panel.test.d.ts.map +1 -0
- package/dist/ui/panels/sleep-panel.test.js +376 -0
- package/dist/ui/panels/sleep-panel.test.js.map +1 -0
- package/dist/ui/panels/task-list.d.ts.map +1 -1
- package/dist/ui/panels/task-list.js +3 -2
- package/dist/ui/panels/task-list.js.map +1 -1
- package/dist/ui/panels/task-list.test.d.ts +2 -0
- package/dist/ui/panels/task-list.test.d.ts.map +1 -0
- package/dist/ui/panels/task-list.test.js +210 -0
- package/dist/ui/panels/task-list.test.js.map +1 -0
- package/dist/ui/theme.test.d.ts +2 -0
- package/dist/ui/theme.test.d.ts.map +1 -0
- package/dist/ui/theme.test.js +159 -0
- package/dist/ui/theme.test.js.map +1 -0
- package/dist/ui/types.d.ts +3 -0
- package/dist/ui/types.d.ts.map +1 -1
- package/package.json +6 -2
|
@@ -1,21 +1,27 @@
|
|
|
1
1
|
import { nanoid } from "nanoid";
|
|
2
|
-
import {
|
|
2
|
+
import { StmtCache } from "./database.js";
|
|
3
3
|
import { truncate } from "../ui/format.js";
|
|
4
|
+
const EPHEMERAL_PREFIX = "eph-";
|
|
4
5
|
/** Create an in-memory session that is NOT persisted to the database. */
|
|
5
6
|
export function createEphemeralSession(providerId) {
|
|
6
7
|
const now = Date.now();
|
|
7
|
-
return { id:
|
|
8
|
+
return { id: `${EPHEMERAL_PREFIX}${nanoid()}`, providerId, createdAt: now, lastActiveAt: now };
|
|
9
|
+
}
|
|
10
|
+
/** Check if a session is ephemeral (not persisted). */
|
|
11
|
+
export function isEphemeralSession(session) {
|
|
12
|
+
return session.id.startsWith(EPHEMERAL_PREFIX);
|
|
8
13
|
}
|
|
9
14
|
export class SessionStore {
|
|
10
15
|
agentId;
|
|
16
|
+
stmts = new StmtCache();
|
|
17
|
+
stmt(sql) { return this.stmts.stmt(sql); }
|
|
11
18
|
constructor(agentId = "") {
|
|
12
19
|
this.agentId = agentId;
|
|
13
20
|
}
|
|
14
21
|
createSession(provider, model, providerSessionId) {
|
|
15
22
|
const id = nanoid();
|
|
16
23
|
const now = Date.now();
|
|
17
|
-
|
|
18
|
-
db.prepare(`INSERT INTO sessions (id, provider, provider_session_id, model, status, created_at, last_active_at, agent_id)
|
|
24
|
+
this.stmt(`INSERT INTO sessions (id, provider, provider_session_id, model, status, created_at, last_active_at, agent_id)
|
|
19
25
|
VALUES (?, ?, ?, ?, 'active', ?, ?, ?)`).run(id, provider, providerSessionId ?? null, model ?? null, now, now, this.agentId);
|
|
20
26
|
return {
|
|
21
27
|
id,
|
|
@@ -26,9 +32,7 @@ export class SessionStore {
|
|
|
26
32
|
};
|
|
27
33
|
}
|
|
28
34
|
getSession(id) {
|
|
29
|
-
const
|
|
30
|
-
const row = db
|
|
31
|
-
.prepare("SELECT * FROM sessions WHERE id = ?")
|
|
35
|
+
const row = this.stmt("SELECT * FROM sessions WHERE id = ?")
|
|
32
36
|
.get(id);
|
|
33
37
|
if (!row)
|
|
34
38
|
return null;
|
|
@@ -41,26 +45,27 @@ export class SessionStore {
|
|
|
41
45
|
};
|
|
42
46
|
}
|
|
43
47
|
updateProviderSessionId(sessionId, providerSessionId) {
|
|
44
|
-
|
|
45
|
-
|
|
48
|
+
this.stmt("UPDATE sessions SET provider_session_id = ? WHERE id = ?").run(providerSessionId, sessionId);
|
|
49
|
+
}
|
|
50
|
+
updateProvider(sessionId, provider) {
|
|
51
|
+
this.stmt("UPDATE sessions SET provider = ?, provider_session_id = NULL WHERE id = ?").run(provider, sessionId);
|
|
46
52
|
}
|
|
47
53
|
updateLastActive(sessionId) {
|
|
48
|
-
|
|
49
|
-
db.prepare("UPDATE sessions SET last_active_at = ? WHERE id = ?").run(Date.now(), sessionId);
|
|
54
|
+
this.stmt("UPDATE sessions SET last_active_at = ? WHERE id = ?").run(Date.now(), sessionId);
|
|
50
55
|
}
|
|
51
56
|
addCost(sessionId, costUsd, inputTokens, outputTokens) {
|
|
52
|
-
|
|
53
|
-
db.prepare(`UPDATE sessions SET cost_usd = cost_usd + ?, input_tokens = input_tokens + ?, output_tokens = output_tokens + ? WHERE id = ?`).run(costUsd, inputTokens, outputTokens, sessionId);
|
|
57
|
+
this.stmt(`UPDATE sessions SET cost_usd = cost_usd + ?, input_tokens = input_tokens + ?, output_tokens = output_tokens + ? WHERE id = ?`).run(costUsd, inputTokens, outputTokens, sessionId);
|
|
54
58
|
}
|
|
55
59
|
addMessage(sessionId, role, content, toolCalls, tokenCount) {
|
|
56
|
-
|
|
57
|
-
db.prepare(`INSERT INTO messages (session_id, role, content, tool_calls, token_count, timestamp)
|
|
60
|
+
this.stmt(`INSERT INTO messages (session_id, role, content, tool_calls, token_count, timestamp)
|
|
58
61
|
VALUES (?, ?, ?, ?, ?, ?)`).run(sessionId, role, content, toolCalls ?? null, tokenCount ?? null, Date.now());
|
|
59
62
|
}
|
|
63
|
+
hasMessages(sessionId) {
|
|
64
|
+
const row = this.stmt("SELECT 1 FROM messages WHERE session_id = ? LIMIT 1").get(sessionId);
|
|
65
|
+
return row !== undefined;
|
|
66
|
+
}
|
|
60
67
|
getMessages(sessionId, limit = 100) {
|
|
61
|
-
const
|
|
62
|
-
const rows = db
|
|
63
|
-
.prepare(`SELECT * FROM messages WHERE session_id = ? ORDER BY timestamp ASC LIMIT ?`)
|
|
68
|
+
const rows = this.stmt(`SELECT * FROM messages WHERE session_id = ? ORDER BY timestamp ASC LIMIT ?`)
|
|
64
69
|
.all(sessionId, limit);
|
|
65
70
|
return rows.map((row) => ({
|
|
66
71
|
id: row.id,
|
|
@@ -73,9 +78,7 @@ export class SessionStore {
|
|
|
73
78
|
}));
|
|
74
79
|
}
|
|
75
80
|
listSessions(limit = 20) {
|
|
76
|
-
const
|
|
77
|
-
const rows = db
|
|
78
|
-
.prepare(`SELECT * FROM sessions s
|
|
81
|
+
const rows = this.stmt(`SELECT * FROM sessions s
|
|
79
82
|
WHERE s.agent_id = ?
|
|
80
83
|
AND EXISTS (SELECT 1 FROM messages m WHERE m.session_id = s.id)
|
|
81
84
|
ORDER BY s.last_active_at DESC LIMIT ?`)
|
|
@@ -89,9 +92,7 @@ export class SessionStore {
|
|
|
89
92
|
}));
|
|
90
93
|
}
|
|
91
94
|
listSessionSummaries(limit = 20) {
|
|
92
|
-
const
|
|
93
|
-
const rows = db
|
|
94
|
-
.prepare(`SELECT
|
|
95
|
+
const rows = this.stmt(`SELECT
|
|
95
96
|
s.id,
|
|
96
97
|
s.provider,
|
|
97
98
|
s.model,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"session-store.js","sourceRoot":"","sources":["../../src/store/session-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"session-store.js","sourceRoot":"","sources":["../../src/store/session-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE1C,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAyB3C,MAAM,gBAAgB,GAAG,MAAM,CAAC;AAEhC,yEAAyE;AACzE,MAAM,UAAU,sBAAsB,CAAC,UAAkB;IACvD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,OAAO,EAAE,EAAE,EAAE,GAAG,gBAAgB,GAAG,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC;AACjG,CAAC;AAED,uDAAuD;AACvD,MAAM,UAAU,kBAAkB,CAAC,OAAgB;IACjD,OAAO,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;AACjD,CAAC;AAED,MAAM,OAAO,YAAY;IACf,OAAO,CAAS;IAChB,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;IACxB,IAAI,CAAC,GAAW,IAAI,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IAE1D,YAAY,OAAO,GAAG,EAAE;QACtB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,aAAa,CACX,QAAgB,EAChB,KAAc,EACd,iBAA0B;QAE1B,MAAM,EAAE,GAAG,MAAM,EAAE,CAAC;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,IAAI,CAAC,IAAI,CACP;8CACwC,CACzC,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,iBAAiB,IAAI,IAAI,EAAE,KAAK,IAAI,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAEtF,OAAO;YACL,EAAE;YACF,UAAU,EAAE,QAAQ;YACpB,iBAAiB;YACjB,SAAS,EAAE,GAAG;YACd,YAAY,EAAE,GAAG;SAClB,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,EAAU;QACnB,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,qCAAqC,CAAC;aACzD,GAAG,CAAC,EAAE,CAAwC,CAAC;QAElD,IAAI,CAAC,GAAG;YAAE,OAAO,IAAI,CAAC;QAEtB,OAAO;YACL,EAAE,EAAE,GAAG,CAAC,EAAY;YACpB,UAAU,EAAE,GAAG,CAAC,QAAkB;YAClC,iBAAiB,EAAE,GAAG,CAAC,mBAAyC;YAChE,SAAS,EAAE,GAAG,CAAC,UAAoB;YACnC,YAAY,EAAE,GAAG,CAAC,cAAwB;SAC3C,CAAC;IACJ,CAAC;IAED,uBAAuB,CAAC,SAAiB,EAAE,iBAAyB;QAClE,IAAI,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAC,GAAG,CACvE,iBAAiB,EACjB,SAAS,CACV,CAAC;IACJ,CAAC;IAED,cAAc,CAAC,SAAiB,EAAE,QAAgB;QAChD,IAAI,CAAC,IAAI,CAAC,2EAA2E,CAAC,CAAC,GAAG,CACxF,QAAQ,EACR,SAAS,CACV,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,SAAiB;QAChC,IAAI,CAAC,IAAI,CAAC,qDAAqD,CAAC,CAAC,GAAG,CAClE,IAAI,CAAC,GAAG,EAAE,EACV,SAAS,CACV,CAAC;IACJ,CAAC;IAED,OAAO,CAAC,SAAiB,EAAE,OAAe,EAAE,WAAmB,EAAE,YAAoB;QACnF,IAAI,CAAC,IAAI,CACP,8HAA8H,CAC/H,CAAC,GAAG,CAAC,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IACvD,CAAC;IAED,UAAU,CACR,SAAiB,EACjB,IAA2B,EAC3B,OAAe,EACf,SAAkB,EAClB,UAAmB;QAEnB,IAAI,CAAC,IAAI,CACP;iCAC2B,CAC5B,CAAC,GAAG,CACH,SAAS,EACT,IAAI,EACJ,OAAO,EACP,SAAS,IAAI,IAAI,EACjB,UAAU,IAAI,IAAI,EAClB,IAAI,CAAC,GAAG,EAAE,CACX,CAAC;IACJ,CAAC;IAED,WAAW,CAAC,SAAiB;QAC3B,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,qDAAqD,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC5F,OAAO,GAAG,KAAK,SAAS,CAAC;IAC3B,CAAC;IAED,WAAW,CAAC,SAAiB,EAAE,KAAK,GAAG,GAAG;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAClB,4EAA4E,CAC7E;aACA,GAAG,CAAC,SAAS,EAAE,KAAK,CAA8B,CAAC;QAEtD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACxB,EAAE,EAAE,GAAG,CAAC,EAAY;YACpB,SAAS,EAAE,GAAG,CAAC,UAAoB;YACnC,IAAI,EAAE,GAAG,CAAC,IAA6B;YACvC,OAAO,EAAE,GAAG,CAAC,OAAiB;YAC9B,SAAS,EAAE,GAAG,CAAC,UAAgC;YAC/C,UAAU,EAAE,GAAG,CAAC,WAAiC;YACjD,SAAS,EAAE,GAAG,CAAC,SAAmB;SACnC,CAAC,CAAC,CAAC;IACN,CAAC;IAED,YAAY,CAAC,KAAK,GAAG,EAAE;QACrB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAClB;;;gDAGwC,CACzC;aACA,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAA8B,CAAC;QAEzD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACxB,EAAE,EAAE,GAAG,CAAC,EAAY;YACpB,UAAU,EAAE,GAAG,CAAC,QAAkB;YAClC,iBAAiB,EAAE,GAAG,CAAC,mBAAyC;YAChE,SAAS,EAAE,GAAG,CAAC,UAAoB;YACnC,YAAY,EAAE,GAAG,CAAC,cAAwB;SAC3C,CAAC,CAAC,CAAC;IACN,CAAC;IAED,oBAAoB,CAAC,KAAK,GAAG,EAAE;QAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAClB;;;;;;;;;;;;;;;;;;;iBAmBS,CACV;aACA,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAA8B,CAAC;QAEzD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACxB,EAAE,EAAE,GAAG,CAAC,EAAY;YACpB,QAAQ,EAAE,GAAG,CAAC,QAAkB;YAChC,KAAK,EAAG,GAAG,CAAC,KAAuB,IAAI,IAAI;YAC3C,SAAS,EAAE,GAAG,CAAC,UAAoB;YACnC,YAAY,EAAE,GAAG,CAAC,cAAwB;YAC1C,YAAY,EAAE,GAAG,CAAC,aAAuB;YACzC,OAAO,EAAG,GAAG,CAAC,QAAmB,IAAI,CAAC;YACtC,WAAW,EAAG,GAAG,CAAC,YAAuB,IAAI,CAAC;YAC9C,YAAY,EAAG,GAAG,CAAC,aAAwB,IAAI,CAAC;YAChD,gBAAgB,EAAE,GAAG,CAAC,kBAAkB;gBACtC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,kBAA4B,EAAE,EAAE,EAAE,IAAI,CAAC;gBACtD,CAAC,CAAC,IAAI;SACT,CAAC,CAAC,CAAC;IACN,CAAC;CACF"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-store.test.d.ts","sourceRoot":"","sources":["../../src/store/session-store.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,125 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, vi } from "vitest";
|
|
2
|
+
import { createTestDb } from "../__tests__/db-helper.js";
|
|
3
|
+
const mockDb = { current: createTestDb() };
|
|
4
|
+
vi.mock("./database.js", () => {
|
|
5
|
+
const getDb = () => mockDb.current;
|
|
6
|
+
class StmtCache {
|
|
7
|
+
cache = new Map();
|
|
8
|
+
stmt(sql) {
|
|
9
|
+
let s = this.cache.get(sql);
|
|
10
|
+
if (!s) {
|
|
11
|
+
s = getDb().prepare(sql);
|
|
12
|
+
this.cache.set(sql, s);
|
|
13
|
+
}
|
|
14
|
+
return s;
|
|
15
|
+
}
|
|
16
|
+
}
|
|
17
|
+
return { getDb, StmtCache, getHeliosDir: () => "/tmp/helios-test" };
|
|
18
|
+
});
|
|
19
|
+
const { SessionStore, createEphemeralSession } = await import("./session-store.js");
|
|
20
|
+
describe("SessionStore", () => {
|
|
21
|
+
beforeEach(() => {
|
|
22
|
+
mockDb.current = createTestDb();
|
|
23
|
+
});
|
|
24
|
+
it("creates and retrieves a session", () => {
|
|
25
|
+
const store = new SessionStore("agent1");
|
|
26
|
+
const session = store.createSession("claude", "opus");
|
|
27
|
+
expect(session.id).toBeTruthy();
|
|
28
|
+
expect(session.providerId).toBe("claude");
|
|
29
|
+
const retrieved = store.getSession(session.id);
|
|
30
|
+
expect(retrieved).not.toBeNull();
|
|
31
|
+
expect(retrieved.id).toBe(session.id);
|
|
32
|
+
expect(retrieved.providerId).toBe("claude");
|
|
33
|
+
});
|
|
34
|
+
it("returns null for missing session", () => {
|
|
35
|
+
const store = new SessionStore("agent1");
|
|
36
|
+
expect(store.getSession("nonexistent")).toBeNull();
|
|
37
|
+
});
|
|
38
|
+
it("updates provider session ID", () => {
|
|
39
|
+
const store = new SessionStore("agent1");
|
|
40
|
+
const session = store.createSession("claude");
|
|
41
|
+
store.updateProviderSessionId(session.id, "sdk-123");
|
|
42
|
+
const retrieved = store.getSession(session.id);
|
|
43
|
+
expect(retrieved.providerSessionId).toBe("sdk-123");
|
|
44
|
+
});
|
|
45
|
+
it("adds and retrieves messages", () => {
|
|
46
|
+
const store = new SessionStore("agent1");
|
|
47
|
+
const session = store.createSession("claude");
|
|
48
|
+
store.addMessage(session.id, "user", "Hello");
|
|
49
|
+
store.addMessage(session.id, "assistant", "Hi there");
|
|
50
|
+
const messages = store.getMessages(session.id);
|
|
51
|
+
expect(messages).toHaveLength(2);
|
|
52
|
+
expect(messages[0].role).toBe("user");
|
|
53
|
+
expect(messages[0].content).toBe("Hello");
|
|
54
|
+
expect(messages[1].role).toBe("assistant");
|
|
55
|
+
});
|
|
56
|
+
it("getMessages respects limit", () => {
|
|
57
|
+
const store = new SessionStore("agent1");
|
|
58
|
+
const session = store.createSession("claude");
|
|
59
|
+
for (let i = 0; i < 10; i++) {
|
|
60
|
+
store.addMessage(session.id, "user", `msg ${i}`);
|
|
61
|
+
}
|
|
62
|
+
const messages = store.getMessages(session.id, 3);
|
|
63
|
+
expect(messages).toHaveLength(3);
|
|
64
|
+
});
|
|
65
|
+
it("tracks cost accumulation", () => {
|
|
66
|
+
const store = new SessionStore("agent1");
|
|
67
|
+
const session = store.createSession("claude");
|
|
68
|
+
store.addCost(session.id, 0.01, 100, 50);
|
|
69
|
+
store.addCost(session.id, 0.02, 200, 100);
|
|
70
|
+
const summaries = store.listSessionSummaries();
|
|
71
|
+
// No messages yet, so summary might not show. Add a message first.
|
|
72
|
+
store.addMessage(session.id, "user", "test");
|
|
73
|
+
const s = store.listSessionSummaries();
|
|
74
|
+
expect(s).toHaveLength(1);
|
|
75
|
+
expect(s[0].costUsd).toBeCloseTo(0.03);
|
|
76
|
+
expect(s[0].inputTokens).toBe(300);
|
|
77
|
+
expect(s[0].outputTokens).toBe(150);
|
|
78
|
+
});
|
|
79
|
+
it("listSessions filters by agent ID", () => {
|
|
80
|
+
const store1 = new SessionStore("agent1");
|
|
81
|
+
const store2 = new SessionStore("agent2");
|
|
82
|
+
const s1 = store1.createSession("claude");
|
|
83
|
+
store1.addMessage(s1.id, "user", "hi");
|
|
84
|
+
const s2 = store2.createSession("openai");
|
|
85
|
+
store2.addMessage(s2.id, "user", "hi");
|
|
86
|
+
expect(store1.listSessions()).toHaveLength(1);
|
|
87
|
+
expect(store1.listSessions()[0].id).toBe(s1.id);
|
|
88
|
+
});
|
|
89
|
+
it("listSessions excludes empty sessions", () => {
|
|
90
|
+
const store = new SessionStore("agent1");
|
|
91
|
+
store.createSession("claude"); // no messages
|
|
92
|
+
expect(store.listSessions()).toHaveLength(0);
|
|
93
|
+
});
|
|
94
|
+
it("listSessionSummaries includes first user message", () => {
|
|
95
|
+
const store = new SessionStore("agent1");
|
|
96
|
+
const session = store.createSession("claude");
|
|
97
|
+
store.addMessage(session.id, "user", "Train a model to classify cats");
|
|
98
|
+
store.addMessage(session.id, "assistant", "Sure!");
|
|
99
|
+
const summaries = store.listSessionSummaries();
|
|
100
|
+
expect(summaries[0].firstUserMessage).toContain("Train a model");
|
|
101
|
+
});
|
|
102
|
+
it("updates lastActiveAt", () => {
|
|
103
|
+
const store = new SessionStore("agent1");
|
|
104
|
+
const session = store.createSession("claude");
|
|
105
|
+
const before = session.lastActiveAt;
|
|
106
|
+
// Small delay to ensure timestamp changes
|
|
107
|
+
store.updateLastActive(session.id);
|
|
108
|
+
const updated = store.getSession(session.id);
|
|
109
|
+
expect(updated.lastActiveAt).toBeGreaterThanOrEqual(before);
|
|
110
|
+
});
|
|
111
|
+
});
|
|
112
|
+
describe("createEphemeralSession", () => {
|
|
113
|
+
it("creates session with eph- prefix", () => {
|
|
114
|
+
const session = createEphemeralSession("claude");
|
|
115
|
+
expect(session.id).toMatch(/^eph-/);
|
|
116
|
+
expect(session.providerId).toBe("claude");
|
|
117
|
+
});
|
|
118
|
+
it("sets timestamps", () => {
|
|
119
|
+
const before = Date.now();
|
|
120
|
+
const session = createEphemeralSession("openai");
|
|
121
|
+
expect(session.createdAt).toBeGreaterThanOrEqual(before);
|
|
122
|
+
expect(session.lastActiveAt).toBeGreaterThanOrEqual(before);
|
|
123
|
+
});
|
|
124
|
+
});
|
|
125
|
+
//# sourceMappingURL=session-store.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-store.test.js","sourceRoot":"","sources":["../../src/store/session-store.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AAEzD,MAAM,MAAM,GAAG,EAAE,OAAO,EAAE,YAAY,EAAE,EAAE,CAAC;AAC3C,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE;IAC5B,MAAM,KAAK,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC;IACnC,MAAM,SAAS;QACL,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;QAC1B,IAAI,CAAC,GAAW;YACd,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC5B,IAAI,CAAC,CAAC,EAAE,CAAC;gBAAC,CAAC,GAAG,KAAK,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;gBAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;YAAC,CAAC;YAC7D,OAAO,CAAC,CAAC;QACX,CAAC;KACF;IACD,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC,kBAAkB,EAAE,CAAC;AACtE,CAAC,CAAC,CAAC;AAEH,MAAM,EAAE,YAAY,EAAE,sBAAsB,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;AAEpF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,UAAU,CAAC,GAAG,EAAE;QACd,MAAM,CAAC,OAAO,GAAG,YAAY,EAAE,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAEtD,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,UAAU,EAAE,CAAC;QAChC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAE1C,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QACjC,MAAM,CAAC,SAAU,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACvC,MAAM,CAAC,SAAU,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAE9C,KAAK,CAAC,uBAAuB,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QACrD,MAAM,SAAS,GAAG,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,SAAU,CAAC,iBAAiB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,GAAG,EAAE;QACrC,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAE9C,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QAC9C,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC;QAEtD,MAAM,QAAQ,GAAG,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAE9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5B,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;QACnD,CAAC;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;QAClC,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAE9C,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QACzC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;QAE1C,MAAM,SAAS,GAAG,KAAK,CAAC,oBAAoB,EAAE,CAAC;QAC/C,mEAAmE;QACnE,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QAC7C,MAAM,CAAC,GAAG,KAAK,CAAC,oBAAoB,EAAE,CAAC;QACvC,MAAM,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC1B,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QAE1C,MAAM,EAAE,GAAG,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QACvC,MAAM,EAAE,GAAG,MAAM,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;QAEvC,MAAM,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QACzC,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC,cAAc;QAE7C,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC9C,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,MAAM,EAAE,gCAAgC,CAAC,CAAC;QACvE,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;QAEnD,MAAM,SAAS,GAAG,KAAK,CAAC,oBAAoB,EAAE,CAAC;QAC/C,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,KAAK,GAAG,IAAI,YAAY,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC;QAEpC,0CAA0C;QAC1C,KAAK,CAAC,gBAAgB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACnC,MAAM,OAAO,GAAG,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7C,MAAM,CAAC,OAAQ,CAAC,YAAY,CAAC,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iBAAiB,EAAE,GAAG,EAAE;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC;QACzD,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,sBAAsB,CAAC,MAAM,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import type { SubagentInfo, SubagentSpawnConfig } from "./types.js";
|
|
2
|
+
import type { ToolDefinition } from "../providers/types.js";
|
|
3
|
+
import type { Orchestrator } from "../core/orchestrator.js";
|
|
4
|
+
import type { MemoryStore } from "../memory/memory-store.js";
|
|
5
|
+
import type { SubagentManager } from "./manager.js";
|
|
6
|
+
export interface SubagentExecContext {
|
|
7
|
+
orchestrator: Orchestrator;
|
|
8
|
+
allTools: ToolDefinition[];
|
|
9
|
+
parentMemory: MemoryStore;
|
|
10
|
+
parentSessionId: string;
|
|
11
|
+
depth: number;
|
|
12
|
+
subagentManager: SubagentManager;
|
|
13
|
+
}
|
|
14
|
+
/** Infer provider from model name. */
|
|
15
|
+
export declare function resolveProviderForModel(model: string | undefined, provider: string | undefined, orchestrator: Orchestrator): {
|
|
16
|
+
provider: "claude" | "openai";
|
|
17
|
+
model: string;
|
|
18
|
+
};
|
|
19
|
+
/** Run a subagent in the background. Mutates `info` with status/result/cost. */
|
|
20
|
+
export declare function runSubagent(info: SubagentInfo, config: SubagentSpawnConfig, ctx: SubagentExecContext): Promise<void>;
|
|
21
|
+
//# sourceMappingURL=executor.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor.d.ts","sourceRoot":"","sources":["../../src/subagent/executor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AACpE,OAAO,KAAK,EAAE,cAAc,EAAiB,MAAM,uBAAuB,CAAC;AAC3E,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAC7D,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AASpD,MAAM,WAAW,mBAAmB;IAClC,YAAY,EAAE,YAAY,CAAC;IAC3B,QAAQ,EAAE,cAAc,EAAE,CAAC;IAC3B,YAAY,EAAE,WAAW,CAAC;IAC1B,eAAe,EAAE,MAAM,CAAC;IACxB,KAAK,EAAE,MAAM,CAAC;IACd,eAAe,EAAE,eAAe,CAAC;CAClC;AAED,sCAAsC;AACtC,wBAAgB,uBAAuB,CACrC,KAAK,EAAE,MAAM,GAAG,SAAS,EACzB,QAAQ,EAAE,MAAM,GAAG,SAAS,EAC5B,YAAY,EAAE,YAAY,GACzB;IAAE,QAAQ,EAAE,QAAQ,GAAG,QAAQ,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAkBlD;AAoDD,gFAAgF;AAChF,wBAAsB,WAAW,CAC/B,IAAI,EAAE,YAAY,EAClB,MAAM,EAAE,mBAAmB,EAC3B,GAAG,EAAE,mBAAmB,GACvB,OAAO,CAAC,IAAI,CAAC,CAuFf"}
|
|
@@ -0,0 +1,136 @@
|
|
|
1
|
+
import { ScopedMemoryStore } from "./scoped-memory.js";
|
|
2
|
+
import { createMemoryTools } from "../tools/memory-tools.js";
|
|
3
|
+
import { formatError } from "../ui/format.js";
|
|
4
|
+
import { debugLog } from "../paths.js";
|
|
5
|
+
/** Tools that subagents should never have access to. */
|
|
6
|
+
const ALWAYS_DENY = new Set(["sleep", "start_monitor", "stop_monitor"]);
|
|
7
|
+
/** Infer provider from model name. */
|
|
8
|
+
export function resolveProviderForModel(model, provider, orchestrator) {
|
|
9
|
+
if (provider === "claude" || provider === "openai") {
|
|
10
|
+
return {
|
|
11
|
+
provider,
|
|
12
|
+
model: model ?? (provider === "claude" ? "claude-sonnet-4-6" : "gpt-5.2"),
|
|
13
|
+
};
|
|
14
|
+
}
|
|
15
|
+
if (model) {
|
|
16
|
+
if (model.startsWith("claude"))
|
|
17
|
+
return { provider: "claude", model };
|
|
18
|
+
if (model.startsWith("gpt-") || model.startsWith("o1") || model.startsWith("o3") || model.startsWith("o4")) {
|
|
19
|
+
return { provider: "openai", model };
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
// Inherit parent
|
|
23
|
+
return {
|
|
24
|
+
provider: orchestrator.currentProvider?.name ?? "claude",
|
|
25
|
+
model: model ?? orchestrator.currentModel ?? "claude-sonnet-4-6",
|
|
26
|
+
};
|
|
27
|
+
}
|
|
28
|
+
/** Build the filtered tool set for a subagent. */
|
|
29
|
+
function buildTools(allTools, config, scopedMemory, depth, maxDepth, subagentManager, orchestrator, parentMemory) {
|
|
30
|
+
let tools = allTools.filter((t) => !ALWAYS_DENY.has(t.name));
|
|
31
|
+
// Apply user deny list
|
|
32
|
+
if (config.tools_deny) {
|
|
33
|
+
const denied = new Set(config.tools_deny);
|
|
34
|
+
tools = tools.filter((t) => !denied.has(t.name));
|
|
35
|
+
}
|
|
36
|
+
// Replace memory tools with scoped versions
|
|
37
|
+
tools = tools.filter((t) => !t.name.startsWith("memory_"));
|
|
38
|
+
tools.push(...createMemoryTools(scopedMemory));
|
|
39
|
+
// Replace subagent tools — include if depth allows
|
|
40
|
+
tools = tools.filter((t) => !t.name.startsWith("subagent"));
|
|
41
|
+
if (maxDepth === 0 || depth + 1 < maxDepth) {
|
|
42
|
+
// Lazy import to avoid circular dependency — the tool factories are lightweight
|
|
43
|
+
const { createSubagentTools } = require("../tools/subagent.js");
|
|
44
|
+
tools.push(...createSubagentTools(subagentManager, orchestrator, parentMemory));
|
|
45
|
+
}
|
|
46
|
+
return tools;
|
|
47
|
+
}
|
|
48
|
+
function buildSystemPrompt(info, config) {
|
|
49
|
+
return `You are a Helios subagent (ID: ${info.id}, depth: ${info.depth}).
|
|
50
|
+
|
|
51
|
+
## Your Task
|
|
52
|
+
${config.task}
|
|
53
|
+
|
|
54
|
+
## Memory
|
|
55
|
+
Your memory is scoped — writes go to /subagents/${info.id}/ in the parent's memory tree.
|
|
56
|
+
You can read the parent's memory for context (goal, observations, experiments, etc.).
|
|
57
|
+
When done, write a clear result summary to memory.
|
|
58
|
+
|
|
59
|
+
## Completion
|
|
60
|
+
When your task is complete, state your conclusion clearly in your final message.
|
|
61
|
+
Be thorough but concise — the parent agent will read your result.`;
|
|
62
|
+
}
|
|
63
|
+
/** Run a subagent in the background. Mutates `info` with status/result/cost. */
|
|
64
|
+
export async function runSubagent(info, config, ctx) {
|
|
65
|
+
const providerInstance = ctx.orchestrator.getProvider(info.provider);
|
|
66
|
+
if (!providerInstance)
|
|
67
|
+
throw new Error(`Provider "${info.provider}" not available`);
|
|
68
|
+
// Ensure authenticated
|
|
69
|
+
if (!(await providerInstance.isAuthenticated())) {
|
|
70
|
+
await providerInstance.authenticate();
|
|
71
|
+
}
|
|
72
|
+
// Temporarily set model for session creation
|
|
73
|
+
const originalModel = providerInstance.currentModel;
|
|
74
|
+
providerInstance.currentModel = info.model;
|
|
75
|
+
const session = await providerInstance.createSession({
|
|
76
|
+
systemPrompt: buildSystemPrompt(info, config),
|
|
77
|
+
model: info.model,
|
|
78
|
+
ephemeral: true,
|
|
79
|
+
});
|
|
80
|
+
// Restore model immediately — session already captured it
|
|
81
|
+
providerInstance.currentModel = originalModel;
|
|
82
|
+
const scopedMemory = new ScopedMemoryStore(ctx.parentMemory, info.id);
|
|
83
|
+
const tools = buildTools(ctx.allTools, config, scopedMemory, ctx.depth, ctx.subagentManager.maxDepth, ctx.subagentManager, ctx.orchestrator, ctx.parentMemory);
|
|
84
|
+
const maxTurns = config.max_turns ?? 50;
|
|
85
|
+
let lastText = "";
|
|
86
|
+
try {
|
|
87
|
+
// The provider's send() handles the full tool loop internally.
|
|
88
|
+
// We just drain events and track cost.
|
|
89
|
+
for (let turn = 0; turn < maxTurns; turn++) {
|
|
90
|
+
if (info.abortController.signal.aborted)
|
|
91
|
+
break;
|
|
92
|
+
let turnText = "";
|
|
93
|
+
let hadToolCalls = false;
|
|
94
|
+
const message = turn === 0 ? config.task : "Continue.";
|
|
95
|
+
for await (const event of providerInstance.send(session, message, tools)) {
|
|
96
|
+
if (info.abortController.signal.aborted)
|
|
97
|
+
break;
|
|
98
|
+
if (event.type === "text" && event.delta) {
|
|
99
|
+
turnText += event.delta;
|
|
100
|
+
}
|
|
101
|
+
if (event.type === "tool_call") {
|
|
102
|
+
hadToolCalls = true;
|
|
103
|
+
}
|
|
104
|
+
if (event.type === "done" && event.usage) {
|
|
105
|
+
const cost = event.usage.costUsd ?? 0;
|
|
106
|
+
info.costUsd += cost;
|
|
107
|
+
info.inputTokens += event.usage.inputTokens;
|
|
108
|
+
info.outputTokens += event.usage.outputTokens;
|
|
109
|
+
ctx.orchestrator.addCost(cost, event.usage.inputTokens, event.usage.outputTokens);
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
if (turnText)
|
|
113
|
+
lastText = turnText;
|
|
114
|
+
// If the provider didn't make any tool calls, the agent is done
|
|
115
|
+
if (!hadToolCalls)
|
|
116
|
+
break;
|
|
117
|
+
}
|
|
118
|
+
const result = lastText || "(subagent produced no text output)";
|
|
119
|
+
info.result = result;
|
|
120
|
+
scopedMemory.write("/result", `Subagent ${info.id} result`, result);
|
|
121
|
+
info.status = "completed";
|
|
122
|
+
info.completedAt = Date.now();
|
|
123
|
+
debugLog("subagent", "completed", { id: info.id, cost: info.costUsd });
|
|
124
|
+
}
|
|
125
|
+
catch (err) {
|
|
126
|
+
info.error = formatError(err);
|
|
127
|
+
info.status = info.abortController.signal.aborted ? "cancelled" : "failed";
|
|
128
|
+
info.completedAt = Date.now();
|
|
129
|
+
scopedMemory.write("/result", `Subagent ${info.id} failed`, `Error: ${info.error}`);
|
|
130
|
+
debugLog("subagent", "failed", { id: info.id, error: info.error });
|
|
131
|
+
}
|
|
132
|
+
finally {
|
|
133
|
+
await providerInstance.closeSession(session).catch(() => { });
|
|
134
|
+
}
|
|
135
|
+
}
|
|
136
|
+
//# sourceMappingURL=executor.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"executor.js","sourceRoot":"","sources":["../../src/subagent/executor.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAEvC,wDAAwD;AACxD,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,CAAC,OAAO,EAAE,eAAe,EAAE,cAAc,CAAC,CAAC,CAAC;AAWxE,sCAAsC;AACtC,MAAM,UAAU,uBAAuB,CACrC,KAAyB,EACzB,QAA4B,EAC5B,YAA0B;IAE1B,IAAI,QAAQ,KAAK,QAAQ,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACnD,OAAO;YACL,QAAQ;YACR,KAAK,EAAE,KAAK,IAAI,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,SAAS,CAAC;SAC1E,CAAC;IACJ,CAAC;IACD,IAAI,KAAK,EAAE,CAAC;QACV,IAAI,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACrE,IAAI,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3G,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACvC,CAAC;IACH,CAAC;IACD,iBAAiB;IACjB,OAAO;QACL,QAAQ,EAAG,YAAY,CAAC,eAAe,EAAE,IAA4B,IAAI,QAAQ;QACjF,KAAK,EAAE,KAAK,IAAI,YAAY,CAAC,YAAY,IAAI,mBAAmB;KACjE,CAAC;AACJ,CAAC;AAED,kDAAkD;AAClD,SAAS,UAAU,CACjB,QAA0B,EAC1B,MAA2B,EAC3B,YAA+B,EAC/B,KAAa,EACb,QAAgB,EAChB,eAAgC,EAChC,YAA0B,EAC1B,YAAyB;IAEzB,IAAI,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAE7D,uBAAuB;IACvB,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QACtB,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAC1C,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IACnD,CAAC;IAED,4CAA4C;IAC5C,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3D,KAAK,CAAC,IAAI,CAAC,GAAG,iBAAiB,CAAC,YAAY,CAAC,CAAC,CAAC;IAE/C,mDAAmD;IACnD,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC;IAC5D,IAAI,QAAQ,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,GAAG,QAAQ,EAAE,CAAC;QAC3C,gFAAgF;QAChF,MAAM,EAAE,mBAAmB,EAAE,GAAG,OAAO,CAAC,sBAAsB,CAA0C,CAAC;QACzG,KAAK,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,eAAe,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC,CAAC;IAClF,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAkB,EAAE,MAA2B;IACxE,OAAO,kCAAkC,IAAI,CAAC,EAAE,YAAY,IAAI,CAAC,KAAK;;;EAGtE,MAAM,CAAC,IAAI;;;kDAGqC,IAAI,CAAC,EAAE;;;;;;kEAMS,CAAC;AACnE,CAAC;AAED,gFAAgF;AAChF,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,IAAkB,EAClB,MAA2B,EAC3B,GAAwB;IAExB,MAAM,gBAAgB,GAAG,GAAG,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAyB,CAAC;IAC7F,IAAI,CAAC,gBAAgB;QAAE,MAAM,IAAI,KAAK,CAAC,aAAa,IAAI,CAAC,QAAQ,iBAAiB,CAAC,CAAC;IAEpF,uBAAuB;IACvB,IAAI,CAAC,CAAC,MAAM,gBAAgB,CAAC,eAAe,EAAE,CAAC,EAAE,CAAC;QAChD,MAAM,gBAAgB,CAAC,YAAY,EAAE,CAAC;IACxC,CAAC;IAED,6CAA6C;IAC7C,MAAM,aAAa,GAAG,gBAAgB,CAAC,YAAY,CAAC;IACpD,gBAAgB,CAAC,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC;IAE3C,MAAM,OAAO,GAAG,MAAM,gBAAgB,CAAC,aAAa,CAAC;QACnD,YAAY,EAAE,iBAAiB,CAAC,IAAI,EAAE,MAAM,CAAC;QAC7C,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,SAAS,EAAE,IAAI;KAChB,CAAC,CAAC;IAEH,0DAA0D;IAC1D,gBAAgB,CAAC,YAAY,GAAG,aAAa,CAAC;IAE9C,MAAM,YAAY,GAAG,IAAI,iBAAiB,CAAC,GAAG,CAAC,YAAY,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IACtE,MAAM,KAAK,GAAG,UAAU,CACtB,GAAG,CAAC,QAAQ,EACZ,MAAM,EACN,YAAY,EACZ,GAAG,CAAC,KAAK,EACT,GAAG,CAAC,eAAe,CAAC,QAAQ,EAC5B,GAAG,CAAC,eAAe,EACnB,GAAG,CAAC,YAAY,EAChB,GAAG,CAAC,YAAY,CACjB,CAAC;IAEF,MAAM,QAAQ,GAAG,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC;IACxC,IAAI,QAAQ,GAAG,EAAE,CAAC;IAElB,IAAI,CAAC;QACH,+DAA+D;QAC/D,uCAAuC;QACvC,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,QAAQ,EAAE,IAAI,EAAE,EAAE,CAAC;YAC3C,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO;gBAAE,MAAM;YAE/C,IAAI,QAAQ,GAAG,EAAE,CAAC;YAClB,IAAI,YAAY,GAAG,KAAK,CAAC;YAEzB,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC;YAEvD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,gBAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,EAAE,CAAC;gBACzE,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO;oBAAE,MAAM;gBAE/C,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oBACzC,QAAQ,IAAI,KAAK,CAAC,KAAK,CAAC;gBAC1B,CAAC;gBACD,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC/B,YAAY,GAAG,IAAI,CAAC;gBACtB,CAAC;gBACD,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;oBACzC,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,OAAO,IAAI,CAAC,CAAC;oBACtC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC;oBACrB,IAAI,CAAC,WAAW,IAAI,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;oBAC5C,IAAI,CAAC,YAAY,IAAI,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC;oBAC9C,GAAG,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBACpF,CAAC;YACH,CAAC;YAED,IAAI,QAAQ;gBAAE,QAAQ,GAAG,QAAQ,CAAC;YAElC,gEAAgE;YAChE,IAAI,CAAC,YAAY;gBAAE,MAAM;QAC3B,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,IAAI,oCAAoC,CAAC;QAChE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,YAAY,CAAC,KAAK,CAAC,SAAS,EAAE,YAAY,IAAI,CAAC,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;QACpE,IAAI,CAAC,MAAM,GAAG,WAAW,CAAC;QAC1B,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC9B,QAAQ,CAAC,UAAU,EAAE,WAAW,EAAE,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IACzE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC;QAC3E,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC9B,YAAY,CAAC,KAAK,CAAC,SAAS,EAAE,YAAY,IAAI,CAAC,EAAE,SAAS,EAAE,UAAU,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;QACpF,QAAQ,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;IACrE,CAAC;YAAS,CAAC;QACT,MAAM,gBAAgB,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;IAC/D,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { EventEmitter } from "node:events";
|
|
2
|
+
import type { SubagentInfo, SubagentSpawnConfig } from "./types.js";
|
|
3
|
+
import type { Orchestrator } from "../core/orchestrator.js";
|
|
4
|
+
import type { MemoryStore } from "../memory/memory-store.js";
|
|
5
|
+
import type { ToolDefinition } from "../providers/types.js";
|
|
6
|
+
export declare class SubagentManager extends EventEmitter {
|
|
7
|
+
private agents;
|
|
8
|
+
readonly maxDepth: number;
|
|
9
|
+
constructor(maxDepth?: number);
|
|
10
|
+
/** Spawn a subagent. Returns immediately — the agent runs in the background. */
|
|
11
|
+
spawn(config: SubagentSpawnConfig, orchestrator: Orchestrator, allTools: ToolDefinition[], parentMemory: MemoryStore, parentSessionId: string, depth?: number): SubagentInfo;
|
|
12
|
+
get(id: string): SubagentInfo | undefined;
|
|
13
|
+
listActive(): SubagentInfo[];
|
|
14
|
+
listAll(): SubagentInfo[];
|
|
15
|
+
listForSession(parentSessionId: string): SubagentInfo[];
|
|
16
|
+
cancel(id: string): boolean;
|
|
17
|
+
/** Remove completed/failed agents older than maxAgeMs (default: 30 min). */
|
|
18
|
+
prune(maxAgeMs?: number): void;
|
|
19
|
+
}
|
|
20
|
+
//# sourceMappingURL=manager.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../src/subagent/manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,OAAO,KAAK,EAAE,YAAY,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAEpE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAC7D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAG5D,qBAAa,eAAgB,SAAQ,YAAY;IAC/C,OAAO,CAAC,MAAM,CAAmC;IACjD,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;gBAEd,QAAQ,SAAI;IAKxB,gFAAgF;IAChF,KAAK,CACH,MAAM,EAAE,mBAAmB,EAC3B,YAAY,EAAE,YAAY,EAC1B,QAAQ,EAAE,cAAc,EAAE,EAC1B,YAAY,EAAE,WAAW,EACzB,eAAe,EAAE,MAAM,EACvB,KAAK,SAAI,GACR,YAAY;IA6Df,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAIzC,UAAU,IAAI,YAAY,EAAE;IAI5B,OAAO,IAAI,YAAY,EAAE;IAIzB,cAAc,CAAC,eAAe,EAAE,MAAM,GAAG,YAAY,EAAE;IAIvD,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO;IAO3B,4EAA4E;IAC5E,KAAK,CAAC,QAAQ,SAAc,GAAG,IAAI;CAQpC"}
|
|
@@ -0,0 +1,98 @@
|
|
|
1
|
+
import { EventEmitter } from "node:events";
|
|
2
|
+
import { nanoid } from "nanoid";
|
|
3
|
+
import { runSubagent, resolveProviderForModel } from "./executor.js";
|
|
4
|
+
import { debugLog } from "../paths.js";
|
|
5
|
+
export class SubagentManager extends EventEmitter {
|
|
6
|
+
agents = new Map();
|
|
7
|
+
maxDepth;
|
|
8
|
+
constructor(maxDepth = 0) {
|
|
9
|
+
super();
|
|
10
|
+
this.maxDepth = maxDepth;
|
|
11
|
+
}
|
|
12
|
+
/** Spawn a subagent. Returns immediately — the agent runs in the background. */
|
|
13
|
+
spawn(config, orchestrator, allTools, parentMemory, parentSessionId, depth = 0) {
|
|
14
|
+
if (this.maxDepth > 0 && depth >= this.maxDepth) {
|
|
15
|
+
throw new Error(`Subagent depth limit reached (max: ${this.maxDepth})`);
|
|
16
|
+
}
|
|
17
|
+
const resolved = resolveProviderForModel(config.model, config.provider, orchestrator);
|
|
18
|
+
const id = nanoid(8);
|
|
19
|
+
const info = {
|
|
20
|
+
id,
|
|
21
|
+
parentSessionId,
|
|
22
|
+
depth,
|
|
23
|
+
task: config.task,
|
|
24
|
+
model: resolved.model,
|
|
25
|
+
provider: resolved.provider,
|
|
26
|
+
status: "running",
|
|
27
|
+
createdAt: Date.now(),
|
|
28
|
+
costUsd: 0,
|
|
29
|
+
inputTokens: 0,
|
|
30
|
+
outputTokens: 0,
|
|
31
|
+
memoryPrefix: `/subagents/${id}`,
|
|
32
|
+
abortController: new AbortController(),
|
|
33
|
+
};
|
|
34
|
+
this.agents.set(id, info);
|
|
35
|
+
debugLog("subagent", "spawned", { id, model: resolved.model, provider: resolved.provider, depth });
|
|
36
|
+
this.emit("spawned", info);
|
|
37
|
+
// Fire-and-forget — run in background
|
|
38
|
+
const ctx = {
|
|
39
|
+
orchestrator,
|
|
40
|
+
allTools,
|
|
41
|
+
parentMemory,
|
|
42
|
+
parentSessionId,
|
|
43
|
+
depth,
|
|
44
|
+
subagentManager: this,
|
|
45
|
+
};
|
|
46
|
+
runSubagent(info, config, ctx)
|
|
47
|
+
.then(() => {
|
|
48
|
+
if (info.status === "completed") {
|
|
49
|
+
this.emit("completed", info);
|
|
50
|
+
}
|
|
51
|
+
else if (info.status === "cancelled") {
|
|
52
|
+
this.emit("cancelled", info);
|
|
53
|
+
}
|
|
54
|
+
else {
|
|
55
|
+
this.emit("failed", info);
|
|
56
|
+
}
|
|
57
|
+
})
|
|
58
|
+
.catch((err) => {
|
|
59
|
+
// Safety net — runSubagent should handle its own errors
|
|
60
|
+
if (info.status === "running") {
|
|
61
|
+
info.status = "failed";
|
|
62
|
+
info.error = String(err);
|
|
63
|
+
info.completedAt = Date.now();
|
|
64
|
+
this.emit("failed", info);
|
|
65
|
+
}
|
|
66
|
+
});
|
|
67
|
+
return info;
|
|
68
|
+
}
|
|
69
|
+
get(id) {
|
|
70
|
+
return this.agents.get(id);
|
|
71
|
+
}
|
|
72
|
+
listActive() {
|
|
73
|
+
return [...this.agents.values()].filter((a) => a.status === "running");
|
|
74
|
+
}
|
|
75
|
+
listAll() {
|
|
76
|
+
return [...this.agents.values()];
|
|
77
|
+
}
|
|
78
|
+
listForSession(parentSessionId) {
|
|
79
|
+
return [...this.agents.values()].filter((a) => a.parentSessionId === parentSessionId);
|
|
80
|
+
}
|
|
81
|
+
cancel(id) {
|
|
82
|
+
const info = this.agents.get(id);
|
|
83
|
+
if (!info || info.status !== "running")
|
|
84
|
+
return false;
|
|
85
|
+
info.abortController.abort();
|
|
86
|
+
return true;
|
|
87
|
+
}
|
|
88
|
+
/** Remove completed/failed agents older than maxAgeMs (default: 30 min). */
|
|
89
|
+
prune(maxAgeMs = 30 * 60_000) {
|
|
90
|
+
const cutoff = Date.now() - maxAgeMs;
|
|
91
|
+
for (const [id, info] of this.agents) {
|
|
92
|
+
if (info.status !== "running" && (info.completedAt ?? 0) < cutoff) {
|
|
93
|
+
this.agents.delete(id);
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
}
|
|
97
|
+
}
|
|
98
|
+
//# sourceMappingURL=manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../src/subagent/manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAEhC,OAAO,EAAE,WAAW,EAAE,uBAAuB,EAA4B,MAAM,eAAe,CAAC;AAI/F,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAEvC,MAAM,OAAO,eAAgB,SAAQ,YAAY;IACvC,MAAM,GAAG,IAAI,GAAG,EAAwB,CAAC;IACxC,QAAQ,CAAS;IAE1B,YAAY,QAAQ,GAAG,CAAC;QACtB,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,gFAAgF;IAChF,KAAK,CACH,MAA2B,EAC3B,YAA0B,EAC1B,QAA0B,EAC1B,YAAyB,EACzB,eAAuB,EACvB,KAAK,GAAG,CAAC;QAET,IAAI,IAAI,CAAC,QAAQ,GAAG,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAChD,MAAM,IAAI,KAAK,CAAC,sCAAsC,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC;QAC1E,CAAC;QAED,MAAM,QAAQ,GAAG,uBAAuB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QACtF,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QAErB,MAAM,IAAI,GAAiB;YACzB,EAAE;YACF,eAAe;YACf,KAAK;YACL,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,QAAQ,EAAE,QAAQ,CAAC,QAAQ;YAC3B,MAAM,EAAE,SAAS;YACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,OAAO,EAAE,CAAC;YACV,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,cAAc,EAAE,EAAE;YAChC,eAAe,EAAE,IAAI,eAAe,EAAE;SACvC,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC;QAC1B,QAAQ,CAAC,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QACnG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAE3B,sCAAsC;QACtC,MAAM,GAAG,GAAwB;YAC/B,YAAY;YACZ,QAAQ;YACR,YAAY;YACZ,eAAe;YACf,KAAK;YACL,eAAe,EAAE,IAAI;SACtB,CAAC;QAEF,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC;aAC3B,IAAI,CAAC,GAAG,EAAE;YACT,IAAI,IAAI,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBAChC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;YAC/B,CAAC;iBAAM,IAAI,IAAI,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBACvC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACb,wDAAwD;YACxD,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC9B,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC;gBACvB,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;gBACzB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC9B,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC,CAAC;QAEL,OAAO,IAAI,CAAC;IACd,CAAC;IAED,GAAG,CAAC,EAAU;QACZ,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC;IAED,UAAU;QACR,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC;IACzE,CAAC;IAED,OAAO;QACL,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IACnC,CAAC;IAED,cAAc,CAAC,eAAuB;QACpC,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,eAAe,KAAK,eAAe,CAAC,CAAC;IACxF,CAAC;IAED,MAAM,CAAC,EAAU;QACf,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACjC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS;YAAE,OAAO,KAAK,CAAC;QACrD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,4EAA4E;IAC5E,KAAK,CAAC,QAAQ,GAAG,EAAE,GAAG,MAAM;QAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC;QACrC,KAAK,MAAM,CAAC,EAAE,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACrC,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC;gBAClE,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACzB,CAAC;QACH,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { MemoryStore, type MemoryNode } from "../memory/memory-store.js";
|
|
2
|
+
/**
|
|
3
|
+
* Memory wrapper for subagents. Writes are scoped under /subagents/{id}/
|
|
4
|
+
* in the parent's memory. Reads check own scope first, then fall through
|
|
5
|
+
* to the parent for read-only access to the broader context.
|
|
6
|
+
*
|
|
7
|
+
* Follows the same override pattern as GlobalMemoryRouter.
|
|
8
|
+
*/
|
|
9
|
+
export declare class ScopedMemoryStore extends MemoryStore {
|
|
10
|
+
private parent;
|
|
11
|
+
private prefix;
|
|
12
|
+
constructor(parent: MemoryStore, subagentId: string);
|
|
13
|
+
private toScoped;
|
|
14
|
+
private isOwn;
|
|
15
|
+
write(path: string, gist: string, content?: string | null): void;
|
|
16
|
+
read(path: string): MemoryNode | null;
|
|
17
|
+
exists(path: string): boolean;
|
|
18
|
+
ls(dirPath?: string): MemoryNode[];
|
|
19
|
+
tree(dirPath?: string): Array<{
|
|
20
|
+
path: string;
|
|
21
|
+
gist: string;
|
|
22
|
+
isDir: boolean;
|
|
23
|
+
}>;
|
|
24
|
+
rm(path: string): number;
|
|
25
|
+
clear(): void;
|
|
26
|
+
count(): number;
|
|
27
|
+
}
|
|
28
|
+
//# sourceMappingURL=scoped-memory.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"scoped-memory.d.ts","sourceRoot":"","sources":["../../src/subagent/scoped-memory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,KAAK,UAAU,EAAE,MAAM,2BAA2B,CAAC;AAEzE;;;;;;GAMG;AACH,qBAAa,iBAAkB,SAAQ,WAAW;IAChD,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM;IAOnD,OAAO,CAAC,QAAQ;IAKhB,OAAO,CAAC,KAAK;IAMJ,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,IAAI;IAMhE,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,UAAU,GAAG,IAAI;IAarC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAO7B,EAAE,CAAC,OAAO,SAAM,GAAG,UAAU,EAAE;IAkC/B,IAAI,CAAC,OAAO,SAAM,GAAG,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,OAAO,CAAA;KAAE,CAAC;IAgC1E,EAAE,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAIxB,KAAK,IAAI,IAAI;IAIb,KAAK,IAAI,MAAM;CAIzB"}
|