@primo-ai/core 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +114 -0
- package/dist/agent.d.ts +66 -0
- package/dist/agent.d.ts.map +1 -0
- package/dist/agent.js +260 -0
- package/dist/agent.js.map +1 -0
- package/dist/checkpoint-store.d.ts +19 -0
- package/dist/checkpoint-store.d.ts.map +1 -0
- package/dist/checkpoint-store.js +79 -0
- package/dist/checkpoint-store.js.map +1 -0
- package/dist/concurrency-controller.d.ts +9 -0
- package/dist/concurrency-controller.d.ts.map +1 -0
- package/dist/concurrency-controller.js +71 -0
- package/dist/concurrency-controller.js.map +1 -0
- package/dist/config-merge.d.ts +12 -0
- package/dist/config-merge.d.ts.map +1 -0
- package/dist/config-merge.js +47 -0
- package/dist/config-merge.js.map +1 -0
- package/dist/config.d.ts +16 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +144 -0
- package/dist/config.js.map +1 -0
- package/dist/context-builder.d.ts +29 -0
- package/dist/context-builder.d.ts.map +1 -0
- package/dist/context-builder.js +178 -0
- package/dist/context-builder.js.map +1 -0
- package/dist/dynamic-resolver.d.ts +7 -0
- package/dist/dynamic-resolver.d.ts.map +1 -0
- package/dist/dynamic-resolver.js +11 -0
- package/dist/dynamic-resolver.js.map +1 -0
- package/dist/errors.d.ts +30 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +40 -0
- package/dist/errors.js.map +1 -0
- package/dist/event-bus.d.ts +8 -0
- package/dist/event-bus.d.ts.map +1 -0
- package/dist/event-bus.js +36 -0
- package/dist/event-bus.js.map +1 -0
- package/dist/event-system.d.ts +19 -0
- package/dist/event-system.d.ts.map +1 -0
- package/dist/event-system.js +58 -0
- package/dist/event-system.js.map +1 -0
- package/dist/fallback-runner.d.ts +30 -0
- package/dist/fallback-runner.d.ts.map +1 -0
- package/dist/fallback-runner.js +40 -0
- package/dist/fallback-runner.js.map +1 -0
- package/dist/gateways/builtin-gateway.d.ts +17 -0
- package/dist/gateways/builtin-gateway.d.ts.map +1 -0
- package/dist/gateways/builtin-gateway.js +58 -0
- package/dist/gateways/builtin-gateway.js.map +1 -0
- package/dist/gateways/gateway-chain.d.ts +9 -0
- package/dist/gateways/gateway-chain.d.ts.map +1 -0
- package/dist/gateways/gateway-chain.js +18 -0
- package/dist/gateways/gateway-chain.js.map +1 -0
- package/dist/gateways/openai-compatible-gateway.d.ts +10 -0
- package/dist/gateways/openai-compatible-gateway.d.ts.map +1 -0
- package/dist/gateways/openai-compatible-gateway.js +26 -0
- package/dist/gateways/openai-compatible-gateway.js.map +1 -0
- package/dist/harness.d.ts +44 -0
- package/dist/harness.d.ts.map +1 -0
- package/dist/harness.js +70 -0
- package/dist/harness.js.map +1 -0
- package/dist/hook-manager.d.ts +22 -0
- package/dist/hook-manager.d.ts.map +1 -0
- package/dist/hook-manager.js +92 -0
- package/dist/hook-manager.js.map +1 -0
- package/dist/index.d.ts +34 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +34 -0
- package/dist/index.js.map +1 -0
- package/dist/llm-invoker.d.ts +34 -0
- package/dist/llm-invoker.d.ts.map +1 -0
- package/dist/llm-invoker.js +124 -0
- package/dist/llm-invoker.js.map +1 -0
- package/dist/loop-orchestrator.d.ts +48 -0
- package/dist/loop-orchestrator.d.ts.map +1 -0
- package/dist/loop-orchestrator.js +247 -0
- package/dist/loop-orchestrator.js.map +1 -0
- package/dist/model-factory.d.ts +13 -0
- package/dist/model-factory.d.ts.map +1 -0
- package/dist/model-factory.js +40 -0
- package/dist/model-factory.js.map +1 -0
- package/dist/model-profile.d.ts +17 -0
- package/dist/model-profile.d.ts.map +1 -0
- package/dist/model-profile.js +57 -0
- package/dist/model-profile.js.map +1 -0
- package/dist/model-resolver.d.ts +16 -0
- package/dist/model-resolver.d.ts.map +1 -0
- package/dist/model-resolver.js +28 -0
- package/dist/model-resolver.js.map +1 -0
- package/dist/parse-model.d.ts +6 -0
- package/dist/parse-model.d.ts.map +1 -0
- package/dist/parse-model.js +11 -0
- package/dist/parse-model.js.map +1 -0
- package/dist/pipeline.d.ts +28 -0
- package/dist/pipeline.d.ts.map +1 -0
- package/dist/pipeline.js +286 -0
- package/dist/pipeline.js.map +1 -0
- package/dist/plugin-manager.d.ts +42 -0
- package/dist/plugin-manager.d.ts.map +1 -0
- package/dist/plugin-manager.js +148 -0
- package/dist/plugin-manager.js.map +1 -0
- package/dist/processors/build-context.d.ts +3 -0
- package/dist/processors/build-context.d.ts.map +1 -0
- package/dist/processors/build-context.js +5 -0
- package/dist/processors/build-context.js.map +1 -0
- package/dist/processors/evaluate-iteration.d.ts +13 -0
- package/dist/processors/evaluate-iteration.d.ts.map +1 -0
- package/dist/processors/evaluate-iteration.js +196 -0
- package/dist/processors/evaluate-iteration.js.map +1 -0
- package/dist/processors/execute-tools.d.ts +4 -0
- package/dist/processors/execute-tools.d.ts.map +1 -0
- package/dist/processors/execute-tools.js +69 -0
- package/dist/processors/execute-tools.js.map +1 -0
- package/dist/processors/gate-tool.d.ts +3 -0
- package/dist/processors/gate-tool.d.ts.map +1 -0
- package/dist/processors/gate-tool.js +6 -0
- package/dist/processors/gate-tool.js.map +1 -0
- package/dist/processors/index.d.ts +10 -0
- package/dist/processors/index.d.ts.map +1 -0
- package/dist/processors/index.js +10 -0
- package/dist/processors/index.js.map +1 -0
- package/dist/processors/invoke-llm.d.ts +13 -0
- package/dist/processors/invoke-llm.d.ts.map +1 -0
- package/dist/processors/invoke-llm.js +138 -0
- package/dist/processors/invoke-llm.js.map +1 -0
- package/dist/processors/prepare-step.d.ts +3 -0
- package/dist/processors/prepare-step.d.ts.map +1 -0
- package/dist/processors/prepare-step.js +5 -0
- package/dist/processors/prepare-step.js.map +1 -0
- package/dist/processors/process-input.d.ts +3 -0
- package/dist/processors/process-input.d.ts.map +1 -0
- package/dist/processors/process-input.js +20 -0
- package/dist/processors/process-input.js.map +1 -0
- package/dist/processors/process-output.d.ts +3 -0
- package/dist/processors/process-output.d.ts.map +1 -0
- package/dist/processors/process-output.js +5 -0
- package/dist/processors/process-output.js.map +1 -0
- package/dist/processors/process-step-output.d.ts +3 -0
- package/dist/processors/process-step-output.d.ts.map +1 -0
- package/dist/processors/process-step-output.js +23 -0
- package/dist/processors/process-step-output.js.map +1 -0
- package/dist/processors/provider-history-compat.d.ts +7 -0
- package/dist/processors/provider-history-compat.d.ts.map +1 -0
- package/dist/processors/provider-history-compat.js +166 -0
- package/dist/processors/provider-history-compat.js.map +1 -0
- package/dist/provider-capabilities.d.ts +4 -0
- package/dist/provider-capabilities.d.ts.map +1 -0
- package/dist/provider-capabilities.js +47 -0
- package/dist/provider-capabilities.js.map +1 -0
- package/dist/retry.d.ts +6 -0
- package/dist/retry.d.ts.map +1 -0
- package/dist/retry.js +26 -0
- package/dist/retry.js.map +1 -0
- package/dist/serialize.d.ts +12 -0
- package/dist/serialize.d.ts.map +1 -0
- package/dist/serialize.js +22 -0
- package/dist/serialize.js.map +1 -0
- package/dist/session-manager.d.ts +17 -0
- package/dist/session-manager.d.ts.map +1 -0
- package/dist/session-manager.js +143 -0
- package/dist/session-manager.js.map +1 -0
- package/dist/session-persistence.d.ts +13 -0
- package/dist/session-persistence.d.ts.map +1 -0
- package/dist/session-persistence.js +63 -0
- package/dist/session-persistence.js.map +1 -0
- package/dist/session-storage.d.ts +19 -0
- package/dist/session-storage.d.ts.map +1 -0
- package/dist/session-storage.js +103 -0
- package/dist/session-storage.js.map +1 -0
- package/dist/state-machine.d.ts +13 -0
- package/dist/state-machine.d.ts.map +1 -0
- package/dist/state-machine.js +49 -0
- package/dist/state-machine.js.map +1 -0
- package/dist/storage-replay-backend.d.ts +7 -0
- package/dist/storage-replay-backend.d.ts.map +1 -0
- package/dist/storage-replay-backend.js +14 -0
- package/dist/storage-replay-backend.js.map +1 -0
- package/dist/sub-agent.d.ts +13 -0
- package/dist/sub-agent.d.ts.map +1 -0
- package/dist/sub-agent.js +87 -0
- package/dist/sub-agent.js.map +1 -0
- package/dist/task-manager.d.ts +22 -0
- package/dist/task-manager.d.ts.map +1 -0
- package/dist/task-manager.js +151 -0
- package/dist/task-manager.js.map +1 -0
- package/dist/token-counter.d.ts +9 -0
- package/dist/token-counter.d.ts.map +1 -0
- package/dist/token-counter.js +74 -0
- package/dist/token-counter.js.map +1 -0
- package/dist/tool-registry.d.ts +31 -0
- package/dist/tool-registry.d.ts.map +1 -0
- package/dist/tool-registry.js +147 -0
- package/dist/tool-registry.js.map +1 -0
- package/package.json +37 -0
|
@@ -0,0 +1,143 @@
|
|
|
1
|
+
export class SessionManagerImpl {
|
|
2
|
+
storage;
|
|
3
|
+
bus;
|
|
4
|
+
constructor(storage, bus) {
|
|
5
|
+
this.storage = storage;
|
|
6
|
+
this.bus = bus;
|
|
7
|
+
}
|
|
8
|
+
async start(input, options) {
|
|
9
|
+
const sessionId = crypto.randomUUID();
|
|
10
|
+
const now = new Date().toISOString();
|
|
11
|
+
const record = {
|
|
12
|
+
sessionId,
|
|
13
|
+
parentSessionId: options?.parentSessionId,
|
|
14
|
+
createdAt: now,
|
|
15
|
+
updatedAt: now,
|
|
16
|
+
status: 'active',
|
|
17
|
+
};
|
|
18
|
+
await this.storage.updateMeta(sessionId, record);
|
|
19
|
+
this.bus.emit('agent:start', { sessionId, input });
|
|
20
|
+
return record;
|
|
21
|
+
}
|
|
22
|
+
async restore(sessionId) {
|
|
23
|
+
const events = [];
|
|
24
|
+
for await (const event of this.storage.read(sessionId)) {
|
|
25
|
+
events.push(event);
|
|
26
|
+
}
|
|
27
|
+
if (events.length === 0) {
|
|
28
|
+
throw new Error(`Session not found: ${sessionId}`);
|
|
29
|
+
}
|
|
30
|
+
let input = '';
|
|
31
|
+
let lastStep = 0;
|
|
32
|
+
const messageHistory = [];
|
|
33
|
+
let agentConfig = {};
|
|
34
|
+
let promptFragments = [];
|
|
35
|
+
let toolDeclarations = [];
|
|
36
|
+
let totalTokenUsage;
|
|
37
|
+
let lastResponse;
|
|
38
|
+
const custom = {};
|
|
39
|
+
let toolCallIdx = 0;
|
|
40
|
+
for (const event of events) {
|
|
41
|
+
const payload = event.payload;
|
|
42
|
+
if (!payload)
|
|
43
|
+
continue;
|
|
44
|
+
switch (event.type) {
|
|
45
|
+
case 'agent:start': {
|
|
46
|
+
if (payload.input)
|
|
47
|
+
input = payload.input;
|
|
48
|
+
if (payload.agentConfig)
|
|
49
|
+
agentConfig = payload.agentConfig;
|
|
50
|
+
if (payload.promptFragments)
|
|
51
|
+
promptFragments = payload.promptFragments;
|
|
52
|
+
if (payload.toolDeclarations)
|
|
53
|
+
toolDeclarations = payload.toolDeclarations;
|
|
54
|
+
break;
|
|
55
|
+
}
|
|
56
|
+
case 'iteration:end':
|
|
57
|
+
case 'iteration.end': {
|
|
58
|
+
lastStep = payload.step + 1;
|
|
59
|
+
if (payload.response) {
|
|
60
|
+
lastResponse = payload.response;
|
|
61
|
+
const msg = { role: 'assistant', content: payload.response };
|
|
62
|
+
messageHistory.push(msg);
|
|
63
|
+
}
|
|
64
|
+
if (payload.tokenUsage) {
|
|
65
|
+
totalTokenUsage = payload.tokenUsage;
|
|
66
|
+
}
|
|
67
|
+
break;
|
|
68
|
+
}
|
|
69
|
+
case 'tool:before':
|
|
70
|
+
case 'tool.before': {
|
|
71
|
+
// tool.before is a pre-execution hook event; skip — tool:after carries the result
|
|
72
|
+
break;
|
|
73
|
+
}
|
|
74
|
+
case 'tool:after':
|
|
75
|
+
case 'tool.after': {
|
|
76
|
+
const toolName = payload.toolName;
|
|
77
|
+
const content = payload.error
|
|
78
|
+
? String(payload.error)
|
|
79
|
+
: typeof payload.result === 'string'
|
|
80
|
+
? payload.result
|
|
81
|
+
: JSON.stringify(payload.result ?? '');
|
|
82
|
+
const msg = {
|
|
83
|
+
role: 'tool',
|
|
84
|
+
content,
|
|
85
|
+
toolCallId: `restored_${toolName}_${toolCallIdx++}`,
|
|
86
|
+
toolName,
|
|
87
|
+
};
|
|
88
|
+
if (payload.error)
|
|
89
|
+
msg.error = String(payload.error);
|
|
90
|
+
if (payload.result !== undefined)
|
|
91
|
+
msg.result = payload.result;
|
|
92
|
+
messageHistory.push(msg);
|
|
93
|
+
break;
|
|
94
|
+
}
|
|
95
|
+
case 'llm:after':
|
|
96
|
+
case 'llm.after': {
|
|
97
|
+
if (payload.tokenUsage) {
|
|
98
|
+
const usage = payload.tokenUsage;
|
|
99
|
+
totalTokenUsage = totalTokenUsage
|
|
100
|
+
? { input: totalTokenUsage.input + usage.input, output: totalTokenUsage.output + usage.output }
|
|
101
|
+
: usage;
|
|
102
|
+
}
|
|
103
|
+
break;
|
|
104
|
+
}
|
|
105
|
+
case 'error': {
|
|
106
|
+
messageHistory.push({ role: 'assistant', content: `[Error] ${String(payload.error)}` });
|
|
107
|
+
break;
|
|
108
|
+
}
|
|
109
|
+
case 'session:suspended': {
|
|
110
|
+
custom.suspendReason = payload.reason;
|
|
111
|
+
break;
|
|
112
|
+
}
|
|
113
|
+
case 'stage:before':
|
|
114
|
+
case 'stage:after':
|
|
115
|
+
case 'llm:before':
|
|
116
|
+
case 'agent:end':
|
|
117
|
+
break;
|
|
118
|
+
}
|
|
119
|
+
}
|
|
120
|
+
return {
|
|
121
|
+
request: { input, sessionId },
|
|
122
|
+
agent: { config: agentConfig, promptFragments, toolDeclarations },
|
|
123
|
+
iteration: { step: lastStep, response: lastResponse },
|
|
124
|
+
session: { messageHistory, totalTokenUsage, custom },
|
|
125
|
+
};
|
|
126
|
+
}
|
|
127
|
+
async suspend(sessionId, reason) {
|
|
128
|
+
// Emit through EventBus so SessionPersistence handles seq + write
|
|
129
|
+
this.bus.emit('session:suspended', { sessionId, reason });
|
|
130
|
+
await this.storage.updateMeta(sessionId, { status: 'suspended' });
|
|
131
|
+
}
|
|
132
|
+
async resume(sessionId, input) {
|
|
133
|
+
// Mark original session as completed
|
|
134
|
+
await this.storage.updateMeta(sessionId, { status: 'completed' });
|
|
135
|
+
// Start a new continuation session linked to the original
|
|
136
|
+
const child = await this.start(input ?? '', { parentSessionId: sessionId });
|
|
137
|
+
return child.sessionId;
|
|
138
|
+
}
|
|
139
|
+
async list(filter) {
|
|
140
|
+
return this.storage.list(filter);
|
|
141
|
+
}
|
|
142
|
+
}
|
|
143
|
+
//# sourceMappingURL=session-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-manager.js","sourceRoot":"","sources":["../src/session-manager.ts"],"names":[],"mappings":"AAGA,MAAM,OAAO,kBAAkB;IAEnB;IACA;IAFV,YACU,OAAuB,EACvB,GAAa;QADb,YAAO,GAAP,OAAO,CAAgB;QACvB,QAAG,GAAH,GAAG,CAAU;IACpB,CAAC;IAEJ,KAAK,CAAC,KAAK,CAAC,KAAa,EAAE,OAAsC;QAC/D,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;QACtC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAErC,MAAM,MAAM,GAAkB;YAC5B,SAAS;YACT,eAAe,EAAE,OAAO,EAAE,eAAe;YACzC,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;YACd,MAAM,EAAE,QAAQ;SACjB,CAAC;QAEF,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAEjD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;QAEnD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,SAAiB;QAC7B,MAAM,MAAM,GAAmB,EAAE,CAAC;QAClC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YACvD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,sBAAsB,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,KAAK,GAAG,EAAE,CAAC;QACf,IAAI,QAAQ,GAAG,CAAC,CAAC;QACjB,MAAM,cAAc,GAAc,EAAE,CAAC;QACrC,IAAI,WAAW,GAA4B,EAAE,CAAC;QAC9C,IAAI,eAAe,GAAa,EAAE,CAAC;QACnC,IAAI,gBAAgB,GAAiD,EAAE,CAAC;QACxE,IAAI,eAA8D,CAAC;QACnE,IAAI,YAAgC,CAAC;QACrC,MAAM,MAAM,GAA4B,EAAE,CAAC;QAC3C,IAAI,WAAW,GAAG,CAAC,CAAC;QAEpB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,OAAO,GAAG,KAAK,CAAC,OAA8C,CAAC;YACrE,IAAI,CAAC,OAAO;gBAAE,SAAS;YAEvB,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,aAAa,CAAC,CAAC,CAAC;oBACnB,IAAI,OAAO,CAAC,KAAK;wBAAE,KAAK,GAAG,OAAO,CAAC,KAAe,CAAC;oBACnD,IAAI,OAAO,CAAC,WAAW;wBAAE,WAAW,GAAG,OAAO,CAAC,WAAsC,CAAC;oBACtF,IAAI,OAAO,CAAC,eAAe;wBAAE,eAAe,GAAG,OAAO,CAAC,eAA2B,CAAC;oBACnF,IAAI,OAAO,CAAC,gBAAgB;wBAAE,gBAAgB,GAAG,OAAO,CAAC,gBAAgE,CAAC;oBAC1H,MAAM;gBACR,CAAC;gBACD,KAAK,eAAe,CAAC;gBACrB,KAAK,eAAe,CAAC,CAAC,CAAC;oBACrB,QAAQ,GAAI,OAAO,CAAC,IAAe,GAAG,CAAC,CAAC;oBACxC,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;wBACrB,YAAY,GAAG,OAAO,CAAC,QAAkB,CAAC;wBAC1C,MAAM,GAAG,GAAY,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,CAAC,QAAkB,EAAE,CAAC;wBAChF,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAC3B,CAAC;oBACD,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;wBACvB,eAAe,GAAG,OAAO,CAAC,UAA+C,CAAC;oBAC5E,CAAC;oBACD,MAAM;gBACR,CAAC;gBACD,KAAK,aAAa,CAAC;gBACnB,KAAK,aAAa,CAAC,CAAC,CAAC;oBACnB,kFAAkF;oBAClF,MAAM;gBACR,CAAC;gBACD,KAAK,YAAY,CAAC;gBAClB,KAAK,YAAY,CAAC,CAAC,CAAC;oBAClB,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAkB,CAAC;oBAC5C,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK;wBAC3B,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC;wBACvB,CAAC,CAAC,OAAO,OAAO,CAAC,MAAM,KAAK,QAAQ;4BAClC,CAAC,CAAC,OAAO,CAAC,MAAM;4BAChB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;oBAC3C,MAAM,GAAG,GAAY;wBACnB,IAAI,EAAE,MAAM;wBACZ,OAAO;wBACP,UAAU,EAAE,YAAY,QAAQ,IAAI,WAAW,EAAE,EAAE;wBACnD,QAAQ;qBACT,CAAC;oBACF,IAAI,OAAO,CAAC,KAAK;wBAAG,GAAoC,CAAC,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBACvF,IAAI,OAAO,CAAC,MAAM,KAAK,SAAS;wBAAG,GAAsC,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;oBAClG,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBACzB,MAAM;gBACR,CAAC;gBACD,KAAK,WAAW,CAAC;gBACjB,KAAK,WAAW,CAAC,CAAC,CAAC;oBACjB,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;wBACvB,MAAM,KAAK,GAAG,OAAO,CAAC,UAA+C,CAAC;wBACtE,eAAe,GAAG,eAAe;4BAC/B,CAAC,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,eAAe,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,EAAE;4BAC/F,CAAC,CAAC,KAAK,CAAC;oBACZ,CAAC;oBACD,MAAM;gBACR,CAAC;gBACD,KAAK,OAAO,CAAC,CAAC,CAAC;oBACb,cAAc,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;oBACxF,MAAM;gBACR,CAAC;gBACD,KAAK,mBAAmB,CAAC,CAAC,CAAC;oBACzB,MAAM,CAAC,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC;oBACtC,MAAM;gBACR,CAAC;gBACD,KAAK,cAAc,CAAC;gBACpB,KAAK,aAAa,CAAC;gBACnB,KAAK,YAAY,CAAC;gBAClB,KAAK,WAAW;oBACd,MAAM;YACV,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;YAC7B,KAAK,EAAE,EAAE,MAAM,EAAE,WAA4D,EAAE,eAAe,EAAE,gBAAgB,EAAE;YAClH,SAAS,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE;YACrD,OAAO,EAAE,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,EAAE;SACrD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,SAAiB,EAAE,MAAc;QAC7C,kEAAkE;QAClE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,mBAAmB,EAAE,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;QAE1D,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IACpE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB,EAAE,KAAc;QAC5C,qCAAqC;QACrC,MAAM,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;QAElE,0DAA0D;QAC1D,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,EAAE,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC;QAC5E,OAAO,KAAK,CAAC,SAAS,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAAqC;QAC9C,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;CACF"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { SessionStorage } from '@primo-ai/sdk';
|
|
2
|
+
import type { EventBus } from './event-bus.js';
|
|
3
|
+
export declare class SessionPersistence {
|
|
4
|
+
private storage;
|
|
5
|
+
private unsubs;
|
|
6
|
+
private seqCounters;
|
|
7
|
+
private writeQueues;
|
|
8
|
+
private bus;
|
|
9
|
+
constructor(bus: EventBus, storage: SessionStorage);
|
|
10
|
+
stop(): Promise<void>;
|
|
11
|
+
private onEvent;
|
|
12
|
+
}
|
|
13
|
+
//# sourceMappingURL=session-persistence.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-persistence.d.ts","sourceRoot":"","sources":["../src/session-persistence.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAgB,MAAM,eAAe,CAAC;AAClE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAiB/C,qBAAa,kBAAkB;IAMF,OAAO,CAAC,OAAO;IAL1C,OAAO,CAAC,MAAM,CAAyB;IACvC,OAAO,CAAC,WAAW,CAA6B;IAChD,OAAO,CAAC,WAAW,CAAoC;IACvD,OAAO,CAAC,GAAG,CAAW;gBAEV,GAAG,EAAE,QAAQ,EAAU,OAAO,EAAE,cAAc;IAUpD,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IAM3B,OAAO,CAAC,OAAO;CA2BhB"}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
import { REPLAY_SENTINEL } from './event-system.js';
|
|
2
|
+
const SUBSCRIBED_EVENTS = [
|
|
3
|
+
'agent:start',
|
|
4
|
+
'stage:before',
|
|
5
|
+
'stage:after',
|
|
6
|
+
'agent:end',
|
|
7
|
+
'tool:before',
|
|
8
|
+
'tool:after',
|
|
9
|
+
'llm:before',
|
|
10
|
+
'llm:after',
|
|
11
|
+
'iteration:end',
|
|
12
|
+
'error',
|
|
13
|
+
'session:suspended',
|
|
14
|
+
];
|
|
15
|
+
export class SessionPersistence {
|
|
16
|
+
storage;
|
|
17
|
+
unsubs = [];
|
|
18
|
+
seqCounters = new Map();
|
|
19
|
+
writeQueues = new Map();
|
|
20
|
+
bus;
|
|
21
|
+
constructor(bus, storage) {
|
|
22
|
+
this.storage = storage;
|
|
23
|
+
this.bus = bus;
|
|
24
|
+
for (const eventType of SUBSCRIBED_EVENTS) {
|
|
25
|
+
const unsub = bus.subscribe(eventType, (data) => {
|
|
26
|
+
this.onEvent(eventType, data);
|
|
27
|
+
});
|
|
28
|
+
this.unsubs.push(unsub);
|
|
29
|
+
}
|
|
30
|
+
}
|
|
31
|
+
async stop() {
|
|
32
|
+
for (const unsub of this.unsubs)
|
|
33
|
+
unsub();
|
|
34
|
+
this.unsubs = [];
|
|
35
|
+
await Promise.all([...this.writeQueues.values()].map(p => p.catch(() => { })));
|
|
36
|
+
}
|
|
37
|
+
onEvent(eventType, data) {
|
|
38
|
+
// Skip replayed events to avoid duplicate writes
|
|
39
|
+
if (typeof data === 'object' && data !== null && REPLAY_SENTINEL in data) {
|
|
40
|
+
return;
|
|
41
|
+
}
|
|
42
|
+
const payload = data;
|
|
43
|
+
const sessionId = payload?.sessionId;
|
|
44
|
+
if (!sessionId)
|
|
45
|
+
return;
|
|
46
|
+
const seq = (this.seqCounters.get(sessionId) ?? 0) + 1;
|
|
47
|
+
this.seqCounters.set(sessionId, seq);
|
|
48
|
+
const event = {
|
|
49
|
+
seq,
|
|
50
|
+
timestamp: new Date().toISOString(),
|
|
51
|
+
type: eventType,
|
|
52
|
+
payload: data,
|
|
53
|
+
};
|
|
54
|
+
const prev = this.writeQueues.get(sessionId) ?? Promise.resolve();
|
|
55
|
+
const write = prev
|
|
56
|
+
.then(() => this.storage.append(sessionId, event))
|
|
57
|
+
.catch((err) => {
|
|
58
|
+
this.bus.emit('session:write_failed', { sessionId, error: err instanceof Error ? err.message : String(err) });
|
|
59
|
+
});
|
|
60
|
+
this.writeQueues.set(sessionId, write);
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
//# sourceMappingURL=session-persistence.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-persistence.js","sourceRoot":"","sources":["../src/session-persistence.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,MAAM,iBAAiB,GAAG;IACxB,aAAa;IACb,cAAc;IACd,aAAa;IACb,WAAW;IACX,aAAa;IACb,YAAY;IACZ,YAAY;IACZ,WAAW;IACX,eAAe;IACf,OAAO;IACP,mBAAmB;CACpB,CAAC;AAEF,MAAM,OAAO,kBAAkB;IAMM;IAL3B,MAAM,GAAsB,EAAE,CAAC;IAC/B,WAAW,GAAG,IAAI,GAAG,EAAkB,CAAC;IACxC,WAAW,GAAG,IAAI,GAAG,EAAyB,CAAC;IAC/C,GAAG,CAAW;IAEtB,YAAY,GAAa,EAAU,OAAuB;QAAvB,YAAO,GAAP,OAAO,CAAgB;QACxD,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;QACf,KAAK,MAAM,SAAS,IAAI,iBAAiB,EAAE,CAAC;YAC1C,MAAM,KAAK,GAAG,GAAG,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;gBAC9C,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI;QACR,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM;YAAE,KAAK,EAAE,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;QACjB,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAChF,CAAC;IAEO,OAAO,CAAC,SAAiB,EAAE,IAAa;QAC9C,iDAAiD;QACjD,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,IAAI,eAAe,IAAK,IAAgC,EAAE,CAAC;YACtG,OAAO;QACT,CAAC;QACD,MAAM,OAAO,GAAG,IAA2C,CAAC;QAC5D,MAAM,SAAS,GAAG,OAAO,EAAE,SAA+B,CAAC;QAC3D,IAAI,CAAC,SAAS;YAAE,OAAO;QAEvB,MAAM,GAAG,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACvD,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;QAErC,MAAM,KAAK,GAAiB;YAC1B,GAAG;YACH,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,IAAI;SACd,CAAC;QAEF,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QAClE,MAAM,KAAK,GAAG,IAAI;aACf,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;aACjD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACb,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,sBAAsB,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAChH,CAAC,CAAC,CAAC;QACL,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IACzC,CAAC;CACF"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import type { SessionEvent, SessionRecord, SessionStorage, SessionStatus } from '@primo-ai/sdk';
|
|
2
|
+
export declare class FilesystemSessionStorage implements SessionStorage {
|
|
3
|
+
private basePath;
|
|
4
|
+
constructor(basePath: string);
|
|
5
|
+
private validateSessionId;
|
|
6
|
+
append(sessionId: string, event: SessionEvent): Promise<void>;
|
|
7
|
+
read(sessionId: string): AsyncIterable<SessionEvent>;
|
|
8
|
+
list(filter?: {
|
|
9
|
+
parentSessionId?: string;
|
|
10
|
+
status?: SessionStatus;
|
|
11
|
+
}): Promise<SessionRecord[]>;
|
|
12
|
+
updateMeta(sessionId: string, meta: Partial<SessionRecord>): Promise<void>;
|
|
13
|
+
private sessionDir;
|
|
14
|
+
private eventsPath;
|
|
15
|
+
private metaPath;
|
|
16
|
+
private readMeta;
|
|
17
|
+
private defaultRecord;
|
|
18
|
+
}
|
|
19
|
+
//# sourceMappingURL=session-storage.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-storage.d.ts","sourceRoot":"","sources":["../src/session-storage.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAIhG,qBAAa,wBAAyB,YAAW,cAAc;IACjD,OAAO,CAAC,QAAQ;gBAAR,QAAQ,EAAE,MAAM;IAEpC,OAAO,CAAC,iBAAiB;IAMnB,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ5D,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,aAAa,CAAC,YAAY,CAAC;IAoBrD,IAAI,CAAC,MAAM,CAAC,EAAE;QAAE,eAAe,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,aAAa,CAAA;KAAE,GAAG,OAAO,CAAC,aAAa,EAAE,CAAC;IAmB7F,UAAU,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,aAAa,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAahF,OAAO,CAAC,UAAU;IAIlB,OAAO,CAAC,UAAU;IAIlB,OAAO,CAAC,QAAQ;YAIF,QAAQ;IAStB,OAAO,CAAC,aAAa;CAStB"}
|
|
@@ -0,0 +1,103 @@
|
|
|
1
|
+
import { appendFile, mkdir, readFile, readdir, writeFile } from 'node:fs/promises';
|
|
2
|
+
import { join } from 'node:path';
|
|
3
|
+
export class FilesystemSessionStorage {
|
|
4
|
+
basePath;
|
|
5
|
+
constructor(basePath) {
|
|
6
|
+
this.basePath = basePath;
|
|
7
|
+
}
|
|
8
|
+
validateSessionId(sessionId) {
|
|
9
|
+
if (new RegExp('[/\\\\]|\\.\\.').test(sessionId)) {
|
|
10
|
+
throw new Error(`Invalid sessionId: ${sessionId}`);
|
|
11
|
+
}
|
|
12
|
+
}
|
|
13
|
+
async append(sessionId, event) {
|
|
14
|
+
this.validateSessionId(sessionId);
|
|
15
|
+
const dir = this.sessionDir(sessionId);
|
|
16
|
+
await mkdir(dir, { recursive: true });
|
|
17
|
+
const line = JSON.stringify(event) + '\n';
|
|
18
|
+
await appendFile(this.eventsPath(sessionId), line, 'utf-8');
|
|
19
|
+
}
|
|
20
|
+
async *read(sessionId) {
|
|
21
|
+
this.validateSessionId(sessionId);
|
|
22
|
+
const filePath = this.eventsPath(sessionId);
|
|
23
|
+
let content;
|
|
24
|
+
try {
|
|
25
|
+
content = await readFile(filePath, 'utf-8');
|
|
26
|
+
}
|
|
27
|
+
catch {
|
|
28
|
+
return;
|
|
29
|
+
}
|
|
30
|
+
for (const line of content.split('\n')) {
|
|
31
|
+
const trimmed = line.trim();
|
|
32
|
+
if (trimmed.length === 0)
|
|
33
|
+
continue;
|
|
34
|
+
try {
|
|
35
|
+
yield JSON.parse(trimmed);
|
|
36
|
+
}
|
|
37
|
+
catch {
|
|
38
|
+
// Skip malformed lines (e.g. truncated writes from crash)
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
async list(filter) {
|
|
43
|
+
let entries;
|
|
44
|
+
try {
|
|
45
|
+
entries = await readdir(this.basePath);
|
|
46
|
+
}
|
|
47
|
+
catch {
|
|
48
|
+
return [];
|
|
49
|
+
}
|
|
50
|
+
const records = [];
|
|
51
|
+
for (const sessionId of entries) {
|
|
52
|
+
const meta = await this.readMeta(sessionId);
|
|
53
|
+
if (!meta)
|
|
54
|
+
continue;
|
|
55
|
+
if (filter?.parentSessionId && meta.parentSessionId !== filter.parentSessionId)
|
|
56
|
+
continue;
|
|
57
|
+
if (filter?.status && meta.status !== filter.status)
|
|
58
|
+
continue;
|
|
59
|
+
records.push(meta);
|
|
60
|
+
}
|
|
61
|
+
return records;
|
|
62
|
+
}
|
|
63
|
+
async updateMeta(sessionId, meta) {
|
|
64
|
+
this.validateSessionId(sessionId);
|
|
65
|
+
const existing = await this.readMeta(sessionId);
|
|
66
|
+
const updated = {
|
|
67
|
+
...(existing ?? this.defaultRecord(sessionId)),
|
|
68
|
+
...meta,
|
|
69
|
+
updatedAt: new Date().toISOString(),
|
|
70
|
+
};
|
|
71
|
+
const dir = this.sessionDir(sessionId);
|
|
72
|
+
await mkdir(dir, { recursive: true });
|
|
73
|
+
await writeFile(this.metaPath(sessionId), JSON.stringify(updated), 'utf-8');
|
|
74
|
+
}
|
|
75
|
+
sessionDir(sessionId) {
|
|
76
|
+
return join(this.basePath, sessionId);
|
|
77
|
+
}
|
|
78
|
+
eventsPath(sessionId) {
|
|
79
|
+
return join(this.sessionDir(sessionId), 'events.jsonl');
|
|
80
|
+
}
|
|
81
|
+
metaPath(sessionId) {
|
|
82
|
+
return join(this.sessionDir(sessionId), 'meta.json');
|
|
83
|
+
}
|
|
84
|
+
async readMeta(sessionId) {
|
|
85
|
+
try {
|
|
86
|
+
const content = await readFile(this.metaPath(sessionId), 'utf-8');
|
|
87
|
+
return JSON.parse(content);
|
|
88
|
+
}
|
|
89
|
+
catch {
|
|
90
|
+
return undefined;
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
defaultRecord(sessionId) {
|
|
94
|
+
const now = new Date().toISOString();
|
|
95
|
+
return {
|
|
96
|
+
sessionId,
|
|
97
|
+
createdAt: now,
|
|
98
|
+
updatedAt: now,
|
|
99
|
+
status: 'active',
|
|
100
|
+
};
|
|
101
|
+
}
|
|
102
|
+
}
|
|
103
|
+
//# sourceMappingURL=session-storage.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-storage.js","sourceRoot":"","sources":["../src/session-storage.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACnF,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,MAAM,OAAO,wBAAwB;IACf;IAApB,YAAoB,QAAgB;QAAhB,aAAQ,GAAR,QAAQ,CAAQ;IAAG,CAAC;IAEhC,iBAAiB,CAAC,SAAiB;QACzC,IAAI,IAAI,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,sBAAsB,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB,EAAE,KAAmB;QACjD,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAClC,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACtC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;QAC1C,MAAM,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,CAAC,IAAI,CAAC,SAAiB;QAC3B,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAC5C,IAAI,OAAe,CAAC;QACpB,IAAI,CAAC;YACH,OAAO,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC9C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;QACT,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YACvC,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YAC5B,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;gBAAE,SAAS;YACnC,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAiB,CAAC;YAC5C,CAAC;YAAC,MAAM,CAAC;gBACP,0DAA0D;YAC5D,CAAC;QACH,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,MAA6D;QACtE,IAAI,OAAiB,CAAC;QACtB,IAAI,CAAC;YACH,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,OAAO,GAAoB,EAAE,CAAC;QACpC,KAAK,MAAM,SAAS,IAAI,OAAO,EAAE,CAAC;YAChC,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YAC5C,IAAI,CAAC,IAAI;gBAAE,SAAS;YACpB,IAAI,MAAM,EAAE,eAAe,IAAI,IAAI,CAAC,eAAe,KAAK,MAAM,CAAC,eAAe;gBAAE,SAAS;YACzF,IAAI,MAAM,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,KAAK,MAAM,CAAC,MAAM;gBAAE,SAAS;YAC9D,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,SAAiB,EAAE,IAA4B;QAC9D,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAClC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAChD,MAAM,OAAO,GAAkB;YAC7B,GAAG,CAAC,QAAQ,IAAI,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAC9C,GAAG,IAAI;YACP,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;QACF,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACtC,MAAM,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,CAAC;IAC9E,CAAC;IAEO,UAAU,CAAC,SAAiB;QAClC,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IACxC,CAAC;IAEO,UAAU,CAAC,SAAiB;QAClC,OAAO,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,cAAc,CAAC,CAAC;IAC1D,CAAC;IAEO,QAAQ,CAAC,SAAiB;QAChC,OAAO,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,CAAC;IACvD,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,SAAiB;QACtC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,CAAC;YAClE,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAkB,CAAC;QAC9C,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAEO,aAAa,CAAC,SAAiB;QACrC,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,OAAO;YACL,SAAS;YACT,SAAS,EAAE,GAAG;YACd,SAAS,EAAE,GAAG;YACd,MAAM,EAAE,QAAQ;SACjB,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { AgentForgeError } from './errors.js';
|
|
2
|
+
export type AgentState = 'pending' | 'running' | 'paused' | 'completed' | 'cancelled' | 'error';
|
|
3
|
+
export declare class StateMachine {
|
|
4
|
+
private _current;
|
|
5
|
+
private _lastError?;
|
|
6
|
+
private listeners;
|
|
7
|
+
get current(): AgentState;
|
|
8
|
+
canTransition(to: AgentState, error?: AgentForgeError): boolean;
|
|
9
|
+
transition(to: AgentState, error?: AgentForgeError): void;
|
|
10
|
+
onTransition(cb: (from: AgentState, to: AgentState) => void): () => void;
|
|
11
|
+
private isRecoverable;
|
|
12
|
+
}
|
|
13
|
+
//# sourceMappingURL=state-machine.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"state-machine.d.ts","sourceRoot":"","sources":["../src/state-machine.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAEnD,MAAM,MAAM,UAAU,GAAG,SAAS,GAAG,SAAS,GAAG,QAAQ,GAAG,WAAW,GAAG,WAAW,GAAG,OAAO,CAAC;AAWhG,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAyB;IACzC,OAAO,CAAC,UAAU,CAAC,CAAkB;IACrC,OAAO,CAAC,SAAS,CAAyD;IAE1E,IAAI,OAAO,IAAI,UAAU,CAExB;IAED,aAAa,CAAC,EAAE,EAAE,UAAU,EAAE,KAAK,CAAC,EAAE,eAAe,GAAG,OAAO;IAO/D,UAAU,CAAC,EAAE,EAAE,UAAU,EAAE,KAAK,CAAC,EAAE,eAAe,GAAG,IAAI;IAUzD,YAAY,CAAC,EAAE,EAAE,CAAC,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,UAAU,KAAK,IAAI,GAAG,MAAM,IAAI;IAQxE,OAAO,CAAC,aAAa;CAMtB"}
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
const VALID_TRANSITIONS = {
|
|
2
|
+
pending: ['running'],
|
|
3
|
+
running: ['paused', 'completed', 'cancelled', 'error'],
|
|
4
|
+
paused: ['running', 'cancelled'],
|
|
5
|
+
completed: ['pending'],
|
|
6
|
+
cancelled: ['pending'],
|
|
7
|
+
error: ['pending'],
|
|
8
|
+
};
|
|
9
|
+
export class StateMachine {
|
|
10
|
+
_current = 'pending';
|
|
11
|
+
_lastError;
|
|
12
|
+
listeners = [];
|
|
13
|
+
get current() {
|
|
14
|
+
return this._current;
|
|
15
|
+
}
|
|
16
|
+
canTransition(to, error) {
|
|
17
|
+
if (to === 'running' && this._current === 'error') {
|
|
18
|
+
return this.isRecoverable(error ?? this._lastError);
|
|
19
|
+
}
|
|
20
|
+
return VALID_TRANSITIONS[this._current]?.includes(to) ?? false;
|
|
21
|
+
}
|
|
22
|
+
transition(to, error) {
|
|
23
|
+
if (!this.canTransition(to, error)) {
|
|
24
|
+
throw new Error(`Invalid state transition: ${this._current} -> ${to}`);
|
|
25
|
+
}
|
|
26
|
+
const from = this._current;
|
|
27
|
+
this._current = to;
|
|
28
|
+
if (to === 'error')
|
|
29
|
+
this._lastError = error;
|
|
30
|
+
for (const cb of this.listeners)
|
|
31
|
+
cb(from, to);
|
|
32
|
+
}
|
|
33
|
+
onTransition(cb) {
|
|
34
|
+
this.listeners.push(cb);
|
|
35
|
+
return () => {
|
|
36
|
+
const idx = this.listeners.indexOf(cb);
|
|
37
|
+
if (idx >= 0)
|
|
38
|
+
this.listeners.splice(idx, 1);
|
|
39
|
+
};
|
|
40
|
+
}
|
|
41
|
+
isRecoverable(error) {
|
|
42
|
+
if (!error || !error.recoverable)
|
|
43
|
+
return false;
|
|
44
|
+
const retryCount = error.retryCount ?? 0;
|
|
45
|
+
const maxRetries = error.maxRetries ?? 3;
|
|
46
|
+
return retryCount < maxRetries;
|
|
47
|
+
}
|
|
48
|
+
}
|
|
49
|
+
//# sourceMappingURL=state-machine.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"state-machine.js","sourceRoot":"","sources":["../src/state-machine.ts"],"names":[],"mappings":"AAIA,MAAM,iBAAiB,GAAqC;IAC1D,OAAO,EAAE,CAAC,SAAS,CAAC;IACpB,OAAO,EAAE,CAAC,QAAQ,EAAE,WAAW,EAAE,WAAW,EAAE,OAAO,CAAC;IACtD,MAAM,EAAE,CAAC,SAAS,EAAE,WAAW,CAAC;IAChC,SAAS,EAAE,CAAC,SAAS,CAAC;IACtB,SAAS,EAAE,CAAC,SAAS,CAAC;IACtB,KAAK,EAAE,CAAC,SAAS,CAAC;CACnB,CAAC;AAEF,MAAM,OAAO,YAAY;IACf,QAAQ,GAAe,SAAS,CAAC;IACjC,UAAU,CAAmB;IAC7B,SAAS,GAAsD,EAAE,CAAC;IAE1E,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,aAAa,CAAC,EAAc,EAAE,KAAuB;QACnD,IAAI,EAAE,KAAK,SAAS,IAAI,IAAI,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;YAClD,OAAO,IAAI,CAAC,aAAa,CAAC,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,EAAE,CAAC,IAAI,KAAK,CAAC;IACjE,CAAC;IAED,UAAU,CAAC,EAAc,EAAE,KAAuB;QAChD,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CAAC,6BAA6B,IAAI,CAAC,QAAQ,OAAO,EAAE,EAAE,CAAC,CAAC;QACzE,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC;QAC3B,IAAI,CAAC,QAAQ,GAAG,EAAE,CAAC;QACnB,IAAI,EAAE,KAAK,OAAO;YAAE,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;QAC5C,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,SAAS;YAAE,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAChD,CAAC;IAED,YAAY,CAAC,EAA8C;QACzD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxB,OAAO,GAAG,EAAE;YACV,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACvC,IAAI,GAAG,IAAI,CAAC;gBAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC;IACJ,CAAC;IAEO,aAAa,CAAC,KAAuB;QAC3C,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,WAAW;YAAE,OAAO,KAAK,CAAC;QAC/C,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,IAAI,CAAC,CAAC;QACzC,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,IAAI,CAAC,CAAC;QACzC,OAAO,UAAU,GAAG,UAAU,CAAC;IACjC,CAAC;CACF"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import type { ReplayBackend, SessionStorage, SessionEvent } from '@primo-ai/sdk';
|
|
2
|
+
export declare class StorageReplayBackend implements ReplayBackend {
|
|
3
|
+
private storage;
|
|
4
|
+
constructor(storage: SessionStorage);
|
|
5
|
+
query(sessionId: string): Promise<SessionEvent[]>;
|
|
6
|
+
}
|
|
7
|
+
//# sourceMappingURL=storage-replay-backend.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"storage-replay-backend.d.ts","sourceRoot":"","sources":["../src/storage-replay-backend.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAEjF,qBAAa,oBAAqB,YAAW,aAAa;IAC5C,OAAO,CAAC,OAAO;gBAAP,OAAO,EAAE,cAAc;IAErC,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;CAOxD"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
export class StorageReplayBackend {
|
|
2
|
+
storage;
|
|
3
|
+
constructor(storage) {
|
|
4
|
+
this.storage = storage;
|
|
5
|
+
}
|
|
6
|
+
async query(sessionId) {
|
|
7
|
+
const events = [];
|
|
8
|
+
for await (const event of this.storage.read(sessionId)) {
|
|
9
|
+
events.push(event);
|
|
10
|
+
}
|
|
11
|
+
return events;
|
|
12
|
+
}
|
|
13
|
+
}
|
|
14
|
+
//# sourceMappingURL=storage-replay-backend.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"storage-replay-backend.js","sourceRoot":"","sources":["../src/storage-replay-backend.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,oBAAoB;IACX;IAApB,YAAoB,OAAuB;QAAvB,YAAO,GAAP,OAAO,CAAgB;IAAG,CAAC;IAE/C,KAAK,CAAC,KAAK,CAAC,SAAiB;QAC3B,MAAM,MAAM,GAAmB,EAAE,CAAC;QAClC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YACvD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { SubAgentConfig, ToolDefinition, Tracer } from '@primo-ai/sdk';
|
|
2
|
+
interface SubAgentParentContext {
|
|
3
|
+
model: string;
|
|
4
|
+
tools: ToolDefinition[];
|
|
5
|
+
eventBus?: {
|
|
6
|
+
emit: (event: string, data: unknown) => void;
|
|
7
|
+
};
|
|
8
|
+
tracer?: Tracer;
|
|
9
|
+
getSessionState?: () => Record<string, unknown>;
|
|
10
|
+
}
|
|
11
|
+
export declare function createSubAgentTool(config: SubAgentConfig, parent: SubAgentParentContext): ToolDefinition;
|
|
12
|
+
export {};
|
|
13
|
+
//# sourceMappingURL=sub-agent.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sub-agent.d.ts","sourceRoot":"","sources":["../src/sub-agent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAkC,cAAc,EAAwB,cAAc,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKlI,UAAU,qBAAqB;IAC7B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,cAAc,EAAE,CAAC;IACxB,QAAQ,CAAC,EAAE;QAAE,IAAI,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,KAAK,IAAI,CAAA;KAAE,CAAC;IAC5D,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,eAAe,CAAC,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACjD;AAED,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,cAAc,EACtB,MAAM,EAAE,qBAAqB,GAC5B,cAAc,CAsEhB"}
|
|
@@ -0,0 +1,87 @@
|
|
|
1
|
+
import { Agent } from './agent.js';
|
|
2
|
+
export function createSubAgentTool(config, parent) {
|
|
3
|
+
const tool = {
|
|
4
|
+
name: config.name,
|
|
5
|
+
description: config.description ?? `Sub-agent: ${config.name}`,
|
|
6
|
+
inputSchema: config.inputSchema ?? {},
|
|
7
|
+
async execute(input) {
|
|
8
|
+
parent.eventBus?.emit('task:start', { name: config.name, input });
|
|
9
|
+
try {
|
|
10
|
+
const childModel = config.model ?? parent.model;
|
|
11
|
+
const childTools = config.tools ?? parent.tools;
|
|
12
|
+
const childAgent = new Agent({
|
|
13
|
+
model: childModel,
|
|
14
|
+
systemPrompt: config.systemPrompt,
|
|
15
|
+
tools: childTools,
|
|
16
|
+
maxIterations: config.maxIterations,
|
|
17
|
+
}, { tracer: parent.tracer });
|
|
18
|
+
if (config.contextPolicy === 'inherit' && parent.getSessionState) {
|
|
19
|
+
const parentState = parent.getSessionState();
|
|
20
|
+
childAgent.use({
|
|
21
|
+
stage: 'prepareStep',
|
|
22
|
+
execute: async (ctx) => {
|
|
23
|
+
if (ctx.iteration?.step === 0) {
|
|
24
|
+
return { ...ctx, session: mergeSessionState(ctx.session, parentState) };
|
|
25
|
+
}
|
|
26
|
+
return ctx;
|
|
27
|
+
},
|
|
28
|
+
});
|
|
29
|
+
}
|
|
30
|
+
if (config.contextPolicy === 'summary-only' && parent.getSessionState) {
|
|
31
|
+
const parentState = parent.getSessionState();
|
|
32
|
+
const summary = summarizeSessionState(parentState);
|
|
33
|
+
childAgent.use({
|
|
34
|
+
stage: 'prepareStep',
|
|
35
|
+
execute: async (ctx) => ({
|
|
36
|
+
...ctx,
|
|
37
|
+
session: { ...ctx.session, custom: { ...ctx.session.custom, parentContextSummary: summary } },
|
|
38
|
+
}),
|
|
39
|
+
});
|
|
40
|
+
}
|
|
41
|
+
const runResult = await childAgent.run(input.task);
|
|
42
|
+
const result = {
|
|
43
|
+
response: runResult.response,
|
|
44
|
+
tokenUsage: runResult.tokenUsage,
|
|
45
|
+
sessionId: runResult.sessionId,
|
|
46
|
+
};
|
|
47
|
+
parent.eventBus?.emit('task:end', { name: config.name, result });
|
|
48
|
+
return runResult.response;
|
|
49
|
+
}
|
|
50
|
+
catch (err) {
|
|
51
|
+
const errorSummary = `Sub-agent "${config.name}" failed: ${err instanceof Error ? err.message : String(err)}`;
|
|
52
|
+
parent.eventBus?.emit('task:end', {
|
|
53
|
+
name: config.name,
|
|
54
|
+
error: errorSummary,
|
|
55
|
+
});
|
|
56
|
+
const wrapper = new Error(errorSummary);
|
|
57
|
+
wrapper.cause = err;
|
|
58
|
+
throw wrapper;
|
|
59
|
+
}
|
|
60
|
+
},
|
|
61
|
+
};
|
|
62
|
+
return tool;
|
|
63
|
+
}
|
|
64
|
+
function summarizeSessionState(state) {
|
|
65
|
+
const history = state.messageHistory;
|
|
66
|
+
if (!history || !Array.isArray(history))
|
|
67
|
+
return '';
|
|
68
|
+
return history
|
|
69
|
+
.map((m) => `${m.role}: ${m.content}`)
|
|
70
|
+
.join('\n');
|
|
71
|
+
}
|
|
72
|
+
function mergeSessionState(child, parent) {
|
|
73
|
+
const merged = { ...child };
|
|
74
|
+
for (const [key, parentVal] of Object.entries(parent)) {
|
|
75
|
+
if (Array.isArray(parentVal) && Array.isArray(merged[key])) {
|
|
76
|
+
const childArr = merged[key];
|
|
77
|
+
const parentLen = parentVal.length;
|
|
78
|
+
const childExtras = childArr.slice(parentLen);
|
|
79
|
+
merged[key] = [...parentVal, ...childExtras];
|
|
80
|
+
}
|
|
81
|
+
else {
|
|
82
|
+
merged[key] = parentVal;
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
return merged;
|
|
86
|
+
}
|
|
87
|
+
//# sourceMappingURL=sub-agent.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sub-agent.js","sourceRoot":"","sources":["../src/sub-agent.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAYnC,MAAM,UAAU,kBAAkB,CAChC,MAAsB,EACtB,MAA6B;IAE7B,MAAM,IAAI,GAAmC;QAC3C,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,cAAc,MAAM,CAAC,IAAI,EAAE;QAC9D,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,EAAE;QAErC,KAAK,CAAC,OAAO,CAAC,KAAuB;YACnC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YAElE,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC;gBAChD,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC;gBAEhD,MAAM,UAAU,GAAG,IAAI,KAAK,CAAC;oBAC3B,KAAK,EAAE,UAAU;oBACjB,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,KAAK,EAAE,UAAU;oBACjB,aAAa,EAAE,MAAM,CAAC,aAAa;iBACpC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;gBAE9B,IAAI,MAAM,CAAC,aAAa,KAAK,SAAS,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;oBACjE,MAAM,WAAW,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC;oBAC7C,UAAU,CAAC,GAAG,CAAC;wBACb,KAAK,EAAE,aAAa;wBACpB,OAAO,EAAE,KAAK,EAAE,GAAoB,EAAE,EAAE;4BACtC,IAAI,GAAG,CAAC,SAAS,EAAE,IAAI,KAAK,CAAC,EAAE,CAAC;gCAC9B,OAAO,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,iBAAiB,CAAC,GAAG,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,CAAC;4BAC1E,CAAC;4BACD,OAAO,GAAG,CAAC;wBACb,CAAC;qBACF,CAAC,CAAC;gBACL,CAAC;gBAED,IAAI,MAAM,CAAC,aAAa,KAAK,cAAc,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;oBACtE,MAAM,WAAW,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC;oBAC7C,MAAM,OAAO,GAAG,qBAAqB,CAAC,WAAW,CAAC,CAAC;oBACnD,UAAU,CAAC,GAAG,CAAC;wBACb,KAAK,EAAE,aAAa;wBACpB,OAAO,EAAE,KAAK,EAAE,GAAoB,EAAE,EAAE,CAAC,CAAC;4BACxC,GAAG,GAAG;4BACN,OAAO,EAAE,EAAE,GAAG,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,oBAAoB,EAAE,OAAO,EAAE,EAAE;yBAC9F,CAAC;qBACH,CAAC,CAAC;gBACL,CAAC;gBAED,MAAM,SAAS,GAAG,MAAM,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAEnD,MAAM,MAAM,GAAmB;oBAC7B,QAAQ,EAAE,SAAS,CAAC,QAAQ;oBAC5B,UAAU,EAAE,SAAS,CAAC,UAAU;oBAChC,SAAS,EAAE,SAAS,CAAC,SAAS;iBAC/B,CAAC;gBAEF,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;gBAEjE,OAAO,SAAS,CAAC,QAAQ,CAAC;YAC5B,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,YAAY,GAAG,cAAc,MAAM,CAAC,IAAI,aAAa,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC9G,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,UAAU,EAAE;oBAChC,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,KAAK,EAAE,YAAY;iBACpB,CAAC,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;gBACxC,OAAO,CAAC,KAAK,GAAG,GAAG,CAAC;gBACpB,MAAM,OAAO,CAAC;YAChB,CAAC;QACH,CAAC;KACF,CAAC;IAEF,OAAO,IAAsB,CAAC;AAChC,CAAC;AAED,SAAS,qBAAqB,CAAC,KAA8B;IAC3D,MAAM,OAAO,GAAG,KAAK,CAAC,cAAsE,CAAC;IAC7F,IAAI,CAAC,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,CAAC;IACnD,OAAO,OAAO;SACX,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC;SACrC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,CAAC;AAED,SAAS,iBAAiB,CACxB,KAAoB,EACpB,MAA+B;IAE/B,MAAM,MAAM,GAA4B,EAAE,GAAG,KAAK,EAAE,CAAC;IACrD,KAAK,MAAM,CAAC,GAAG,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACtD,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;YAC3D,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAc,CAAC;YAC1C,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC;YACnC,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YAC9C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,SAAS,EAAE,GAAG,WAAW,CAAC,CAAC;QAC/C,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;QAC1B,CAAC;IACH,CAAC;IACD,OAAO,MAAkC,CAAC;AAC5C,CAAC"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { AsyncTaskConfig, AsyncTaskHandle, AgentConfig, SubAgentResult, TaskManager as ITaskManager } from '@primo-ai/sdk';
|
|
2
|
+
import type { EventBus } from './event-bus.js';
|
|
3
|
+
import type { ConcurrencyController } from './concurrency-controller.js';
|
|
4
|
+
export declare class TaskManagerImpl implements ITaskManager {
|
|
5
|
+
private handles;
|
|
6
|
+
private eventBus?;
|
|
7
|
+
private concurrencyController?;
|
|
8
|
+
private runAgentFn;
|
|
9
|
+
constructor(options?: {
|
|
10
|
+
eventBus?: EventBus;
|
|
11
|
+
concurrencyController?: ConcurrencyController;
|
|
12
|
+
runAgentFn?: (config: AgentConfig, input: string, signal?: AbortSignal) => Promise<SubAgentResult>;
|
|
13
|
+
});
|
|
14
|
+
launch(config: AsyncTaskConfig, prompt: string): Promise<AsyncTaskHandle>;
|
|
15
|
+
get(taskId: string): AsyncTaskHandle | undefined;
|
|
16
|
+
cancel(taskId: string): void;
|
|
17
|
+
list(filter?: {
|
|
18
|
+
parentSessionId?: string;
|
|
19
|
+
}): AsyncTaskHandle[];
|
|
20
|
+
private executeTask;
|
|
21
|
+
}
|
|
22
|
+
//# sourceMappingURL=task-manager.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"task-manager.d.ts","sourceRoot":"","sources":["../src/task-manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,eAAe,EACf,eAAe,EAEf,WAAW,EACX,cAAc,EACd,WAAW,IAAI,YAAY,EAC5B,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC/C,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,6BAA6B,CAAC;AA4EzE,qBAAa,eAAgB,YAAW,YAAY;IAClD,OAAO,CAAC,OAAO,CAA8C;IAC7D,OAAO,CAAC,QAAQ,CAAC,CAAW;IAC5B,OAAO,CAAC,qBAAqB,CAAC,CAAwB;IACtD,OAAO,CAAC,UAAU,CAIW;gBAEjB,OAAO,CAAC,EAAE;QACpB,QAAQ,CAAC,EAAE,QAAQ,CAAC;QACpB,qBAAqB,CAAC,EAAE,qBAAqB,CAAC;QAC9C,UAAU,CAAC,EAAE,CACX,MAAM,EAAE,WAAW,EACnB,KAAK,EAAE,MAAM,EACb,MAAM,CAAC,EAAE,WAAW,KACjB,OAAO,CAAC,cAAc,CAAC,CAAC;KAC9B;IAMK,MAAM,CAAC,MAAM,EAAE,eAAe,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAwB/E,GAAG,CAAC,MAAM,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAIhD,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAO5B,IAAI,CAAC,MAAM,CAAC,EAAE;QAAE,eAAe,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,eAAe,EAAE;YAYhD,WAAW;CAoE1B"}
|