@roackb2/heddle 0.0.7 → 0.0.9
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 +207 -3
- package/dist/examples/cyberloop-observer.d.ts +3 -0
- package/dist/examples/cyberloop-observer.d.ts.map +1 -0
- package/dist/examples/cyberloop-observer.js +79 -0
- package/dist/examples/cyberloop-observer.js.map +1 -0
- package/dist/examples/heartbeat.d.ts +2 -0
- package/dist/examples/heartbeat.d.ts.map +1 -0
- package/dist/examples/heartbeat.js +61 -0
- package/dist/examples/heartbeat.js.map +1 -0
- package/dist/examples/host-events.d.ts +15 -0
- package/dist/examples/host-events.d.ts.map +1 -0
- package/dist/examples/host-events.js +233 -0
- package/dist/examples/host-events.js.map +1 -0
- package/dist/examples/programmatic-loop.d.ts +2 -0
- package/dist/examples/programmatic-loop.d.ts.map +1 -0
- package/dist/examples/programmatic-loop.js +128 -0
- package/dist/examples/programmatic-loop.js.map +1 -0
- package/dist/src/cli/ask.d.ts.map +1 -1
- package/dist/src/cli/ask.js +10 -41
- package/dist/src/cli/ask.js.map +1 -1
- package/dist/src/cli/chat/App.d.ts.map +1 -1
- package/dist/src/cli/chat/App.js +48 -0
- package/dist/src/cli/chat/App.js.map +1 -1
- package/dist/src/cli/chat/hooks/useAgentRun.d.ts +9 -0
- package/dist/src/cli/chat/hooks/useAgentRun.d.ts.map +1 -1
- package/dist/src/cli/chat/hooks/useAgentRun.js +47 -26
- package/dist/src/cli/chat/hooks/useAgentRun.js.map +1 -1
- package/dist/src/cli/chat/state/local-commands.d.ts +3 -0
- package/dist/src/cli/chat/state/local-commands.d.ts.map +1 -1
- package/dist/src/cli/chat/state/local-commands.js +21 -0
- package/dist/src/cli/chat/state/local-commands.js.map +1 -1
- package/dist/src/cli/chat/state/storage.d.ts.map +1 -1
- package/dist/src/cli/chat/state/storage.js +2 -0
- package/dist/src/cli/chat/state/storage.js.map +1 -1
- package/dist/src/cli/chat/state/types.d.ts +1 -0
- package/dist/src/cli/chat/state/types.d.ts.map +1 -1
- package/dist/src/cli/chat/submit.d.ts +3 -0
- package/dist/src/cli/chat/submit.d.ts.map +1 -1
- package/dist/src/cli/chat/submit.js +3 -0
- package/dist/src/cli/chat/submit.js.map +1 -1
- package/dist/src/cli/chat/utils/format.d.ts.map +1 -1
- package/dist/src/cli/chat/utils/format.js +6 -0
- package/dist/src/cli/chat/utils/format.js.map +1 -1
- package/dist/src/cli/chat/utils/runtime.d.ts.map +1 -1
- package/dist/src/cli/chat/utils/runtime.js +3 -19
- package/dist/src/cli/chat/utils/runtime.js.map +1 -1
- package/dist/src/cli/main.js +0 -0
- package/dist/src/index.d.ts +16 -0
- package/dist/src/index.d.ts.map +1 -1
- package/dist/src/index.js +9 -0
- package/dist/src/index.js.map +1 -1
- package/dist/src/integrations/cyberloop-kinematics.d.ts +16 -0
- package/dist/src/integrations/cyberloop-kinematics.d.ts.map +1 -0
- package/dist/src/integrations/cyberloop-kinematics.js +77 -0
- package/dist/src/integrations/cyberloop-kinematics.js.map +1 -0
- package/dist/src/integrations/cyberloop.d.ts +88 -0
- package/dist/src/integrations/cyberloop.d.ts.map +1 -0
- package/dist/src/integrations/cyberloop.js +244 -0
- package/dist/src/integrations/cyberloop.js.map +1 -0
- package/dist/src/llm/factory.d.ts +1 -1
- package/dist/src/llm/factory.d.ts.map +1 -1
- package/dist/src/llm/factory.js +2 -19
- package/dist/src/llm/factory.js.map +1 -1
- package/dist/src/llm/providers.d.ts +3 -0
- package/dist/src/llm/providers.d.ts.map +1 -0
- package/dist/src/llm/providers.js +21 -0
- package/dist/src/llm/providers.js.map +1 -0
- package/dist/src/run-agent.d.ts +3 -1
- package/dist/src/run-agent.d.ts.map +1 -1
- package/dist/src/run-agent.js +10 -0
- package/dist/src/run-agent.js.map +1 -1
- package/dist/src/runtime/agent-loop.d.ts +42 -0
- package/dist/src/runtime/agent-loop.d.ts.map +1 -0
- package/dist/src/runtime/agent-loop.js +173 -0
- package/dist/src/runtime/agent-loop.js.map +1 -0
- package/dist/src/runtime/api-keys.d.ts +8 -0
- package/dist/src/runtime/api-keys.d.ts.map +1 -0
- package/dist/src/runtime/api-keys.js +25 -0
- package/dist/src/runtime/api-keys.js.map +1 -0
- package/dist/src/runtime/default-tools.d.ts +12 -0
- package/dist/src/runtime/default-tools.d.ts.map +1 -0
- package/dist/src/runtime/default-tools.js +40 -0
- package/dist/src/runtime/default-tools.js.map +1 -0
- package/dist/src/runtime/events.d.ts +128 -0
- package/dist/src/runtime/events.d.ts.map +1 -0
- package/dist/src/runtime/events.js +41 -0
- package/dist/src/runtime/events.js.map +1 -0
- package/dist/src/runtime/heartbeat-store.d.ts +20 -0
- package/dist/src/runtime/heartbeat-store.d.ts.map +1 -0
- package/dist/src/runtime/heartbeat-store.js +42 -0
- package/dist/src/runtime/heartbeat-store.js.map +1 -0
- package/dist/src/runtime/heartbeat.d.ts +39 -0
- package/dist/src/runtime/heartbeat.d.ts.map +1 -0
- package/dist/src/runtime/heartbeat.js +117 -0
- package/dist/src/runtime/heartbeat.js.map +1 -0
- package/dist/src/trace/format.d.ts.map +1 -1
- package/dist/src/trace/format.js +8 -0
- package/dist/src/trace/format.js.map +1 -1
- package/dist/src/types.d.ts +8 -0
- package/dist/src/types.d.ts.map +1 -1
- package/dist/src/utils/logger.d.ts.map +1 -1
- package/dist/src/utils/logger.js +2 -2
- package/dist/src/utils/logger.js.map +1 -1
- package/package.json +31 -2
- package/dist/src/__tests__/chat-activity-format.test.d.ts +0 -2
- package/dist/src/__tests__/chat-activity-format.test.d.ts.map +0 -1
- package/dist/src/__tests__/chat-activity-format.test.js +0 -41
- package/dist/src/__tests__/chat-activity-format.test.js.map +0 -1
- package/dist/src/__tests__/chat-compaction.test.d.ts +0 -2
- package/dist/src/__tests__/chat-compaction.test.d.ts.map +0 -1
- package/dist/src/__tests__/chat-compaction.test.js +0 -63
- package/dist/src/__tests__/chat-compaction.test.js.map +0 -1
- package/dist/src/__tests__/chat-format.test.d.ts +0 -2
- package/dist/src/__tests__/chat-format.test.d.ts.map +0 -1
- package/dist/src/__tests__/chat-format.test.js +0 -137
- package/dist/src/__tests__/chat-format.test.js.map +0 -1
- package/dist/src/__tests__/chat-runtime.test.d.ts +0 -2
- package/dist/src/__tests__/chat-runtime.test.d.ts.map +0 -1
- package/dist/src/__tests__/chat-runtime.test.js +0 -39
- package/dist/src/__tests__/chat-runtime.test.js.map +0 -1
- package/dist/src/__tests__/core-utils.test.d.ts +0 -2
- package/dist/src/__tests__/core-utils.test.d.ts.map +0 -1
- package/dist/src/__tests__/core-utils.test.js +0 -87
- package/dist/src/__tests__/core-utils.test.js.map +0 -1
- package/dist/src/__tests__/file-mentions.test.d.ts +0 -2
- package/dist/src/__tests__/file-mentions.test.d.ts.map +0 -1
- package/dist/src/__tests__/file-mentions.test.js +0 -29
- package/dist/src/__tests__/file-mentions.test.js.map +0 -1
- package/dist/src/__tests__/llm-factory.test.d.ts +0 -2
- package/dist/src/__tests__/llm-factory.test.d.ts.map +0 -1
- package/dist/src/__tests__/llm-factory.test.js +0 -45
- package/dist/src/__tests__/llm-factory.test.js.map +0 -1
- package/dist/src/__tests__/local-commands.test.d.ts +0 -2
- package/dist/src/__tests__/local-commands.test.d.ts.map +0 -1
- package/dist/src/__tests__/local-commands.test.js +0 -153
- package/dist/src/__tests__/local-commands.test.js.map +0 -1
- package/dist/src/__tests__/project-approval-rules.test.d.ts +0 -2
- package/dist/src/__tests__/project-approval-rules.test.d.ts.map +0 -1
- package/dist/src/__tests__/project-approval-rules.test.js +0 -135
- package/dist/src/__tests__/project-approval-rules.test.js.map +0 -1
- package/dist/src/__tests__/prompt-input.test.d.ts +0 -2
- package/dist/src/__tests__/prompt-input.test.d.ts.map +0 -1
- package/dist/src/__tests__/prompt-input.test.js +0 -57
- package/dist/src/__tests__/prompt-input.test.js.map +0 -1
- package/dist/src/__tests__/prompts.test.d.ts +0 -2
- package/dist/src/__tests__/prompts.test.d.ts.map +0 -1
- package/dist/src/__tests__/prompts.test.js +0 -46
- package/dist/src/__tests__/prompts.test.js.map +0 -1
- package/dist/src/__tests__/run-agent.test.d.ts +0 -2
- package/dist/src/__tests__/run-agent.test.d.ts.map +0 -1
- package/dist/src/__tests__/run-agent.test.js +0 -1276
- package/dist/src/__tests__/run-agent.test.js.map +0 -1
- package/dist/src/__tests__/run-shell.command.test.d.ts +0 -2
- package/dist/src/__tests__/run-shell.command.test.d.ts.map +0 -1
- package/dist/src/__tests__/run-shell.command.test.js +0 -188
- package/dist/src/__tests__/run-shell.command.test.js.map +0 -1
- package/dist/src/__tests__/smoke.test.d.ts +0 -2
- package/dist/src/__tests__/smoke.test.d.ts.map +0 -1
- package/dist/src/__tests__/smoke.test.js +0 -314
- package/dist/src/__tests__/smoke.test.js.map +0 -1
- package/dist/src/__tests__/tools.test.d.ts +0 -2
- package/dist/src/__tests__/tools.test.d.ts.map +0 -1
- package/dist/src/__tests__/tools.test.js +0 -698
- package/dist/src/__tests__/tools.test.js.map +0 -1
- package/dist/src/__tests__/trace-format.test.d.ts +0 -2
- package/dist/src/__tests__/trace-format.test.d.ts.map +0 -1
- package/dist/src/__tests__/trace-format.test.js +0 -148
- package/dist/src/__tests__/trace-format.test.js.map +0 -1
- package/dist/src/cli/chat/actions.d.ts +0 -47
- package/dist/src/cli/chat/actions.d.ts.map +0 -1
- package/dist/src/cli/chat/actions.js +0 -215
- package/dist/src/cli/chat/actions.js.map +0 -1
- package/dist/src/cli/chat/format.d.ts +0 -23
- package/dist/src/cli/chat/format.d.ts.map +0 -1
- package/dist/src/cli/chat/format.js +0 -243
- package/dist/src/cli/chat/format.js.map +0 -1
- package/dist/src/cli/chat/local-commands.d.ts +0 -17
- package/dist/src/cli/chat/local-commands.d.ts.map +0 -1
- package/dist/src/cli/chat/local-commands.js +0 -180
- package/dist/src/cli/chat/local-commands.js.map +0 -1
- package/dist/src/cli/chat/panels.d.ts +0 -37
- package/dist/src/cli/chat/panels.d.ts.map +0 -1
- package/dist/src/cli/chat/panels.js +0 -142
- package/dist/src/cli/chat/panels.js.map +0 -1
- package/dist/src/cli/chat/runtime.d.ts +0 -26
- package/dist/src/cli/chat/runtime.d.ts.map +0 -1
- package/dist/src/cli/chat/runtime.js +0 -28
- package/dist/src/cli/chat/runtime.js.map +0 -1
- package/dist/src/cli/chat/storage.d.ts +0 -13
- package/dist/src/cli/chat/storage.d.ts.map +0 -1
- package/dist/src/cli/chat/storage.js +0 -126
- package/dist/src/cli/chat/storage.js.map +0 -1
- package/dist/src/cli/chat/types.d.ts +0 -51
- package/dist/src/cli/chat/types.d.ts.map +0 -1
- package/dist/src/cli/chat/types.js +0 -2
- package/dist/src/cli/chat/types.js.map +0 -1
- package/dist/src/cli/chat/use-run-state.d.ts +0 -23
- package/dist/src/cli/chat/use-run-state.d.ts.map +0 -1
- package/dist/src/cli/chat/use-run-state.js +0 -118
- package/dist/src/cli/chat/use-run-state.js.map +0 -1
- package/dist/src/cli/chat/use-sessions.d.ts +0 -21
- package/dist/src/cli/chat/use-sessions.d.ts.map +0 -1
- package/dist/src/cli/chat/use-sessions.js +0 -111
- package/dist/src/cli/chat/use-sessions.js.map +0 -1
- package/dist/src/cli/chat-actions.d.ts +0 -47
- package/dist/src/cli/chat-actions.d.ts.map +0 -1
- package/dist/src/cli/chat-actions.js +0 -215
- package/dist/src/cli/chat-actions.js.map +0 -1
- package/dist/src/cli/chat-format.d.ts +0 -23
- package/dist/src/cli/chat-format.d.ts.map +0 -1
- package/dist/src/cli/chat-format.js +0 -243
- package/dist/src/cli/chat-format.js.map +0 -1
- package/dist/src/cli/chat-local-commands.d.ts +0 -17
- package/dist/src/cli/chat-local-commands.d.ts.map +0 -1
- package/dist/src/cli/chat-local-commands.js +0 -180
- package/dist/src/cli/chat-local-commands.js.map +0 -1
- package/dist/src/cli/chat-panels.d.ts +0 -37
- package/dist/src/cli/chat-panels.d.ts.map +0 -1
- package/dist/src/cli/chat-panels.js +0 -142
- package/dist/src/cli/chat-panels.js.map +0 -1
- package/dist/src/cli/chat-runtime.d.ts +0 -26
- package/dist/src/cli/chat-runtime.d.ts.map +0 -1
- package/dist/src/cli/chat-runtime.js +0 -28
- package/dist/src/cli/chat-runtime.js.map +0 -1
- package/dist/src/cli/chat-storage.d.ts +0 -13
- package/dist/src/cli/chat-storage.d.ts.map +0 -1
- package/dist/src/cli/chat-storage.js +0 -126
- package/dist/src/cli/chat-storage.js.map +0 -1
- package/dist/src/cli/chat-submit.d.ts +0 -28
- package/dist/src/cli/chat-submit.d.ts.map +0 -1
- package/dist/src/cli/chat-submit.js +0 -90
- package/dist/src/cli/chat-submit.js.map +0 -1
- package/dist/src/cli/chat-types.d.ts +0 -51
- package/dist/src/cli/chat-types.d.ts.map +0 -1
- package/dist/src/cli/chat-types.js +0 -2
- package/dist/src/cli/chat-types.js.map +0 -1
- package/dist/src/cli/chat.d.ts +0 -4
- package/dist/src/cli/chat.d.ts.map +0 -1
- package/dist/src/cli/chat.js +0 -153
- package/dist/src/cli/chat.js.map +0 -1
- package/dist/src/cli/useChatRunState.d.ts +0 -23
- package/dist/src/cli/useChatRunState.d.ts.map +0 -1
- package/dist/src/cli/useChatRunState.js +0 -118
- package/dist/src/cli/useChatRunState.js.map +0 -1
- package/dist/src/cli/useChatSessions.d.ts +0 -21
- package/dist/src/cli/useChatSessions.d.ts.map +0 -1
- package/dist/src/cli/useChatSessions.js +0 -111
- package/dist/src/cli/useChatSessions.js.map +0 -1
|
@@ -0,0 +1,173 @@
|
|
|
1
|
+
import { resolve } from 'node:path';
|
|
2
|
+
import { DEFAULT_OPENAI_MODEL } from '../config.js';
|
|
3
|
+
import { inferProviderFromModel } from '../llm/providers.js';
|
|
4
|
+
import { runAgent } from '../run-agent.js';
|
|
5
|
+
import { createLogger } from '../utils/logger.js';
|
|
6
|
+
import { resolveApiKeyForModel } from './api-keys.js';
|
|
7
|
+
import { createFinishedAgentLoopState, generateRunId, getHistoryFromAgentLoopCheckpoint, getHistoryFromAgentLoopState } from './events.js';
|
|
8
|
+
export async function runAgentLoop(options) {
|
|
9
|
+
const runId = generateRunId();
|
|
10
|
+
const model = options.model ?? options.llm?.info?.model ?? process.env.OPENAI_MODEL ?? process.env.ANTHROPIC_MODEL ?? DEFAULT_OPENAI_MODEL;
|
|
11
|
+
const provider = inferProviderFromModel(model);
|
|
12
|
+
const workspaceRoot = resolve(options.workspaceRoot ?? process.cwd());
|
|
13
|
+
const apiKey = options.apiKey ?? resolveApiKeyForModel(model);
|
|
14
|
+
const llm = options.llm ?? await createLoopLlmAdapter({ model, apiKey });
|
|
15
|
+
const logger = options.logger ?? createLogger({ pretty: false, level: 'info', console: false });
|
|
16
|
+
const tools = await resolveTools({
|
|
17
|
+
...options,
|
|
18
|
+
model,
|
|
19
|
+
apiKey,
|
|
20
|
+
workspaceRoot,
|
|
21
|
+
});
|
|
22
|
+
const now = () => new Date().toISOString();
|
|
23
|
+
const startedAt = now();
|
|
24
|
+
const resumeMetadata = getResumeMetadata(options.resumeFrom);
|
|
25
|
+
options.onEvent?.({
|
|
26
|
+
type: 'loop.started',
|
|
27
|
+
runId,
|
|
28
|
+
goal: options.goal,
|
|
29
|
+
model,
|
|
30
|
+
provider,
|
|
31
|
+
workspaceRoot,
|
|
32
|
+
resumedFromCheckpoint: resumeMetadata?.checkpointRunId,
|
|
33
|
+
timestamp: startedAt,
|
|
34
|
+
});
|
|
35
|
+
if (resumeMetadata) {
|
|
36
|
+
options.onEvent?.({
|
|
37
|
+
type: 'loop.resumed',
|
|
38
|
+
runId,
|
|
39
|
+
fromCheckpoint: resumeMetadata.checkpointRunId,
|
|
40
|
+
priorTraceEvents: resumeMetadata.priorTraceEvents,
|
|
41
|
+
timestamp: now(),
|
|
42
|
+
});
|
|
43
|
+
}
|
|
44
|
+
const result = await runAgent({
|
|
45
|
+
goal: options.goal,
|
|
46
|
+
llm,
|
|
47
|
+
tools,
|
|
48
|
+
maxSteps: options.maxSteps,
|
|
49
|
+
logger,
|
|
50
|
+
history: resolveHistory(options),
|
|
51
|
+
systemContext: options.systemContext,
|
|
52
|
+
onAssistantStream: (update) => {
|
|
53
|
+
options.onEvent?.({
|
|
54
|
+
type: 'assistant.stream',
|
|
55
|
+
runId,
|
|
56
|
+
...update,
|
|
57
|
+
timestamp: now(),
|
|
58
|
+
});
|
|
59
|
+
options.onAssistantStream?.(update);
|
|
60
|
+
},
|
|
61
|
+
onEvent: (event) => {
|
|
62
|
+
options.onEvent?.({ type: 'trace', runId, event, timestamp: now() });
|
|
63
|
+
options.onTraceEvent?.(event);
|
|
64
|
+
},
|
|
65
|
+
onToolCalling: (call, step, toolDef) => {
|
|
66
|
+
options.onEvent?.({
|
|
67
|
+
type: 'tool.calling',
|
|
68
|
+
runId,
|
|
69
|
+
step,
|
|
70
|
+
tool: call.tool,
|
|
71
|
+
toolCallId: call.id,
|
|
72
|
+
input: call.input,
|
|
73
|
+
requiresApproval: toolDef.requiresApproval ?? false,
|
|
74
|
+
timestamp: now(),
|
|
75
|
+
});
|
|
76
|
+
},
|
|
77
|
+
onToolCompleted: (call, result, step, durationMs) => {
|
|
78
|
+
options.onEvent?.({
|
|
79
|
+
type: 'tool.completed',
|
|
80
|
+
runId,
|
|
81
|
+
step,
|
|
82
|
+
tool: call.tool,
|
|
83
|
+
toolCallId: call.id,
|
|
84
|
+
result: { ok: result.ok, output: result.output, error: result.error },
|
|
85
|
+
durationMs,
|
|
86
|
+
timestamp: now(),
|
|
87
|
+
});
|
|
88
|
+
},
|
|
89
|
+
approveToolCall: options.approveToolCall,
|
|
90
|
+
shouldStop: options.shouldStop,
|
|
91
|
+
abortSignal: options.abortSignal,
|
|
92
|
+
});
|
|
93
|
+
const finishedAt = now();
|
|
94
|
+
const state = createFinishedAgentLoopState({
|
|
95
|
+
runId,
|
|
96
|
+
goal: options.goal,
|
|
97
|
+
model,
|
|
98
|
+
provider,
|
|
99
|
+
workspaceRoot,
|
|
100
|
+
startedAt,
|
|
101
|
+
finishedAt,
|
|
102
|
+
result,
|
|
103
|
+
});
|
|
104
|
+
options.onEvent?.({
|
|
105
|
+
type: 'loop.finished',
|
|
106
|
+
runId,
|
|
107
|
+
outcome: result.outcome,
|
|
108
|
+
summary: result.summary,
|
|
109
|
+
usage: result.usage,
|
|
110
|
+
state,
|
|
111
|
+
timestamp: finishedAt,
|
|
112
|
+
});
|
|
113
|
+
return {
|
|
114
|
+
...result,
|
|
115
|
+
model,
|
|
116
|
+
provider,
|
|
117
|
+
workspaceRoot,
|
|
118
|
+
state,
|
|
119
|
+
};
|
|
120
|
+
}
|
|
121
|
+
function resolveHistory(options) {
|
|
122
|
+
if (options.history) {
|
|
123
|
+
return options.history;
|
|
124
|
+
}
|
|
125
|
+
if (!options.resumeFrom) {
|
|
126
|
+
return undefined;
|
|
127
|
+
}
|
|
128
|
+
if ('version' in options.resumeFrom) {
|
|
129
|
+
return getHistoryFromAgentLoopCheckpoint(options.resumeFrom);
|
|
130
|
+
}
|
|
131
|
+
return getHistoryFromAgentLoopState(options.resumeFrom);
|
|
132
|
+
}
|
|
133
|
+
function getResumeMetadata(resumeFrom) {
|
|
134
|
+
if (!resumeFrom) {
|
|
135
|
+
return undefined;
|
|
136
|
+
}
|
|
137
|
+
if ('version' in resumeFrom) {
|
|
138
|
+
return {
|
|
139
|
+
checkpointRunId: resumeFrom.runId,
|
|
140
|
+
priorTraceEvents: resumeFrom.state.trace.length,
|
|
141
|
+
};
|
|
142
|
+
}
|
|
143
|
+
return {
|
|
144
|
+
checkpointRunId: resumeFrom.runId,
|
|
145
|
+
priorTraceEvents: resumeFrom.trace.length,
|
|
146
|
+
};
|
|
147
|
+
}
|
|
148
|
+
async function createLoopLlmAdapter(options) {
|
|
149
|
+
const { createLlmAdapter } = await import('../llm/factory.js');
|
|
150
|
+
return createLlmAdapter(options);
|
|
151
|
+
}
|
|
152
|
+
async function resolveTools(options) {
|
|
153
|
+
const providedTools = options.tools ?? [];
|
|
154
|
+
const extraTools = options.extraTools ?? [];
|
|
155
|
+
if (options.includeDefaultTools === false) {
|
|
156
|
+
return [...providedTools, ...extraTools];
|
|
157
|
+
}
|
|
158
|
+
const { createDefaultAgentTools } = await import('./default-tools.js');
|
|
159
|
+
return [
|
|
160
|
+
...createDefaultAgentTools({
|
|
161
|
+
model: options.model,
|
|
162
|
+
apiKey: options.apiKey,
|
|
163
|
+
workspaceRoot: options.workspaceRoot,
|
|
164
|
+
stateDir: options.stateDir,
|
|
165
|
+
memoryDir: options.memoryDir,
|
|
166
|
+
searchIgnoreDirs: options.searchIgnoreDirs,
|
|
167
|
+
includePlanTool: options.includePlanTool,
|
|
168
|
+
}),
|
|
169
|
+
...providedTools,
|
|
170
|
+
...extraTools,
|
|
171
|
+
];
|
|
172
|
+
}
|
|
173
|
+
//# sourceMappingURL=agent-loop.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"agent-loop.js","sourceRoot":"","sources":["../../../src/runtime/agent-loop.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEpC,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AACpD,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAE7D,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAG3C,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,EAAE,4BAA4B,EAAE,aAAa,EAAE,iCAAiC,EAAE,4BAA4B,EAAE,MAAM,aAAa,CAAC;AAsC3I,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,OAA4B;IAC7D,MAAM,KAAK,GAAG,aAAa,EAAE,CAAC;IAC9B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,GAAG,EAAE,IAAI,EAAE,KAAK,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,oBAAoB,CAAC;IAC3I,MAAM,QAAQ,GAAG,sBAAsB,CAAC,KAAK,CAAC,CAAC;IAC/C,MAAM,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC,aAAa,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IACtE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,qBAAqB,CAAC,KAAK,CAAC,CAAC;IAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,MAAM,oBAAoB,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;IACzE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,YAAY,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;IAChG,MAAM,KAAK,GAAG,MAAM,YAAY,CAAC;QAC/B,GAAG,OAAO;QACV,KAAK;QACL,MAAM;QACN,aAAa;KACd,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,GAAG,EAAE,CAAC,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAC3C,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC;IAExB,MAAM,cAAc,GAAG,iBAAiB,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAE7D,OAAO,CAAC,OAAO,EAAE,CAAC;QAChB,IAAI,EAAE,cAAc;QACpB,KAAK;QACL,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,KAAK;QACL,QAAQ;QACR,aAAa;QACb,qBAAqB,EAAE,cAAc,EAAE,eAAe;QACtD,SAAS,EAAE,SAAS;KACrB,CAAC,CAAC;IAEH,IAAI,cAAc,EAAE,CAAC;QACnB,OAAO,CAAC,OAAO,EAAE,CAAC;YAChB,IAAI,EAAE,cAAc;YACpB,KAAK;YACL,cAAc,EAAE,cAAc,CAAC,eAAe;YAC9C,gBAAgB,EAAE,cAAc,CAAC,gBAAgB;YACjD,SAAS,EAAE,GAAG,EAAE;SACjB,CAAC,CAAC;IACL,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC;QAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,GAAG;QACH,KAAK;QACL,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,MAAM;QACN,OAAO,EAAE,cAAc,CAAC,OAAO,CAAC;QAChC,aAAa,EAAE,OAAO,CAAC,aAAa;QACpC,iBAAiB,EAAE,CAAC,MAAM,EAAE,EAAE;YAC5B,OAAO,CAAC,OAAO,EAAE,CAAC;gBAChB,IAAI,EAAE,kBAAkB;gBACxB,KAAK;gBACL,GAAG,MAAM;gBACT,SAAS,EAAE,GAAG,EAAE;aACjB,CAAC,CAAC;YACH,OAAO,CAAC,iBAAiB,EAAE,CAAC,MAAM,CAAC,CAAC;QACtC,CAAC;QACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YACjB,OAAO,CAAC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;YACrE,OAAO,CAAC,YAAY,EAAE,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;QACD,aAAa,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE;YACrC,OAAO,CAAC,OAAO,EAAE,CAAC;gBAChB,IAAI,EAAE,cAAc;gBACpB,KAAK;gBACL,IAAI;gBACJ,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,UAAU,EAAE,IAAI,CAAC,EAAE;gBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,gBAAgB,EAAE,OAAO,CAAC,gBAAgB,IAAI,KAAK;gBACnD,SAAS,EAAE,GAAG,EAAE;aACjB,CAAC,CAAC;QACL,CAAC;QACD,eAAe,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE;YAClD,OAAO,CAAC,OAAO,EAAE,CAAC;gBAChB,IAAI,EAAE,gBAAgB;gBACtB,KAAK;gBACL,IAAI;gBACJ,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,UAAU,EAAE,IAAI,CAAC,EAAE;gBACnB,MAAM,EAAE,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE;gBACrE,UAAU;gBACV,SAAS,EAAE,GAAG,EAAE;aACjB,CAAC,CAAC;QACL,CAAC;QACD,eAAe,EAAE,OAAO,CAAC,eAAe;QACxC,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,WAAW,EAAE,OAAO,CAAC,WAAW;KACjC,CAAC,CAAC;IAEH,MAAM,UAAU,GAAG,GAAG,EAAE,CAAC;IACzB,MAAM,KAAK,GAAG,4BAA4B,CAAC;QACzC,KAAK;QACL,IAAI,EAAE,OAAO,CAAC,IAAI;QAClB,KAAK;QACL,QAAQ;QACR,aAAa;QACb,SAAS;QACT,UAAU;QACV,MAAM;KACP,CAAC,CAAC;IAEH,OAAO,CAAC,OAAO,EAAE,CAAC;QAChB,IAAI,EAAE,eAAe;QACrB,KAAK;QACL,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,KAAK;QACL,SAAS,EAAE,UAAU;KACtB,CAAC,CAAC;IAEH,OAAO;QACL,GAAG,MAAM;QACT,KAAK;QACL,QAAQ;QACR,aAAa;QACb,KAAK;KACN,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,OAA4D;IAClF,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,OAAO,OAAO,CAAC,OAAO,CAAC;IACzB,CAAC;IAED,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;QACxB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,SAAS,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;QACpC,OAAO,iCAAiC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,4BAA4B,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;AAC1D,CAAC;AAED,SAAS,iBAAiB,CACxB,UAA4D;IAE5D,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,SAAS,IAAI,UAAU,EAAE,CAAC;QAC5B,OAAO;YACL,eAAe,EAAE,UAAU,CAAC,KAAK;YACjC,gBAAgB,EAAE,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM;SAChD,CAAC;IACJ,CAAC;IAED,OAAO;QACL,eAAe,EAAE,UAAU,CAAC,KAAK;QACjC,gBAAgB,EAAE,UAAU,CAAC,KAAK,CAAC,MAAM;KAC1C,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,oBAAoB,CAAC,OAA2C;IAC7E,MAAM,EAAE,gBAAgB,EAAE,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;IAC/D,OAAO,gBAAgB,CAAC,OAAO,CAAC,CAAC;AACnC,CAAC;AAED,KAAK,UAAU,YAAY,CACzB,OAIC;IAED,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;IAC1C,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,EAAE,CAAC;IAC5C,IAAI,OAAO,CAAC,mBAAmB,KAAK,KAAK,EAAE,CAAC;QAC1C,OAAO,CAAC,GAAG,aAAa,EAAE,GAAG,UAAU,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,EAAE,uBAAuB,EAAE,GAAG,MAAM,MAAM,CAAC,oBAAoB,CAAC,CAAC;IACvE,OAAO;QACL,GAAG,uBAAuB,CAAC;YACzB,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,aAAa,EAAE,OAAO,CAAC,aAAa;YACpC,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,gBAAgB,EAAE,OAAO,CAAC,gBAAgB;YAC1C,eAAe,EAAE,OAAO,CAAC,eAAe;SACzC,CAAC;QACF,GAAG,aAAa;QAChB,GAAG,UAAU;KACd,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import type { LlmProvider } from '../llm/types.js';
|
|
2
|
+
export type ApiKeyRuntime = {
|
|
3
|
+
apiKey?: string;
|
|
4
|
+
apiKeyProvider?: LlmProvider | 'explicit';
|
|
5
|
+
};
|
|
6
|
+
export declare function resolveProviderApiKey(provider: LlmProvider): string | undefined;
|
|
7
|
+
export declare function resolveApiKeyForModel(model: string, runtime?: ApiKeyRuntime): string | undefined;
|
|
8
|
+
//# sourceMappingURL=api-keys.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"api-keys.d.ts","sourceRoot":"","sources":["../../../src/runtime/api-keys.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAGnD,MAAM,MAAM,aAAa,GAAG;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,cAAc,CAAC,EAAE,WAAW,GAAG,UAAU,CAAC;CAC3C,CAAC;AAEF,wBAAgB,qBAAqB,CAAC,QAAQ,EAAE,WAAW,GAAG,MAAM,GAAG,SAAS,CAS/E;AAED,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,MAAM,GAAG,SAAS,CAWhG"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
import { inferProviderFromModel } from '../llm/providers.js';
|
|
2
|
+
export function resolveProviderApiKey(provider) {
|
|
3
|
+
switch (provider) {
|
|
4
|
+
case 'openai':
|
|
5
|
+
return firstDefinedNonEmpty(process.env.OPENAI_API_KEY, process.env.PERSONAL_OPENAI_API_KEY);
|
|
6
|
+
case 'anthropic':
|
|
7
|
+
return firstDefinedNonEmpty(process.env.ANTHROPIC_API_KEY, process.env.PERSONAL_ANTHROPIC_API_KEY);
|
|
8
|
+
case 'google':
|
|
9
|
+
return undefined;
|
|
10
|
+
}
|
|
11
|
+
}
|
|
12
|
+
export function resolveApiKeyForModel(model, runtime) {
|
|
13
|
+
if (runtime?.apiKey && runtime.apiKeyProvider === 'explicit') {
|
|
14
|
+
return runtime.apiKey;
|
|
15
|
+
}
|
|
16
|
+
const provider = inferProviderFromModel(model);
|
|
17
|
+
if (runtime?.apiKey && runtime.apiKeyProvider === provider) {
|
|
18
|
+
return runtime.apiKey;
|
|
19
|
+
}
|
|
20
|
+
return resolveProviderApiKey(provider);
|
|
21
|
+
}
|
|
22
|
+
function firstDefinedNonEmpty(...values) {
|
|
23
|
+
return values.find((value) => typeof value === 'string' && value.trim().length > 0);
|
|
24
|
+
}
|
|
25
|
+
//# sourceMappingURL=api-keys.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"api-keys.js","sourceRoot":"","sources":["../../../src/runtime/api-keys.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAO7D,MAAM,UAAU,qBAAqB,CAAC,QAAqB;IACzD,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,QAAQ;YACX,OAAO,oBAAoB,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;QAC/F,KAAK,WAAW;YACd,OAAO,oBAAoB,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;QACrG,KAAK,QAAQ;YACX,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,KAAa,EAAE,OAAuB;IAC1E,IAAI,OAAO,EAAE,MAAM,IAAI,OAAO,CAAC,cAAc,KAAK,UAAU,EAAE,CAAC;QAC7D,OAAO,OAAO,CAAC,MAAM,CAAC;IACxB,CAAC;IAED,MAAM,QAAQ,GAAG,sBAAsB,CAAC,KAAK,CAAC,CAAC;IAC/C,IAAI,OAAO,EAAE,MAAM,IAAI,OAAO,CAAC,cAAc,KAAK,QAAQ,EAAE,CAAC;QAC3D,OAAO,OAAO,CAAC,MAAM,CAAC;IACxB,CAAC;IAED,OAAO,qBAAqB,CAAC,QAAQ,CAAC,CAAC;AACzC,CAAC;AAED,SAAS,oBAAoB,CAAC,GAAG,MAAiC;IAChE,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACtF,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { ToolDefinition } from '../types.js';
|
|
2
|
+
export type DefaultAgentToolsOptions = {
|
|
3
|
+
model: string;
|
|
4
|
+
apiKey?: string;
|
|
5
|
+
workspaceRoot?: string;
|
|
6
|
+
stateDir?: string;
|
|
7
|
+
memoryDir?: string;
|
|
8
|
+
searchIgnoreDirs?: string[];
|
|
9
|
+
includePlanTool?: boolean;
|
|
10
|
+
};
|
|
11
|
+
export declare function createDefaultAgentTools(options: DefaultAgentToolsOptions): ToolDefinition[];
|
|
12
|
+
//# sourceMappingURL=default-tools.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"default-tools.d.ts","sourceRoot":"","sources":["../../../src/runtime/default-tools.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAiBlD,MAAM,MAAM,wBAAwB,GAAG;IACrC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B,CAAC;AAEF,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,wBAAwB,GAAG,cAAc,EAAE,CA8B3F"}
|
|
@@ -0,0 +1,40 @@
|
|
|
1
|
+
import { join } from 'node:path';
|
|
2
|
+
import { editFileTool } from '../tools/edit-file.js';
|
|
3
|
+
import { listFilesTool } from '../tools/list-files.js';
|
|
4
|
+
import { createEditMemoryNoteTool, createListMemoryNotesTool, createReadMemoryNoteTool, createSearchMemoryNotesTool, } from '../tools/memory-notes.js';
|
|
5
|
+
import { readFileTool } from '../tools/read-file.js';
|
|
6
|
+
import { reportStateTool } from '../tools/report-state.js';
|
|
7
|
+
import { createRunShellInspectTool, createRunShellMutateTool } from '../tools/run-shell.js';
|
|
8
|
+
import { createSearchFilesTool } from '../tools/search-files.js';
|
|
9
|
+
import { updatePlanTool } from '../tools/update-plan.js';
|
|
10
|
+
import { createViewImageTool } from '../tools/view-image.js';
|
|
11
|
+
import { createWebSearchTool } from '../tools/web-search.js';
|
|
12
|
+
export function createDefaultAgentTools(options) {
|
|
13
|
+
const memoryRoot = options.memoryDir ??
|
|
14
|
+
join(options.workspaceRoot ?? process.cwd(), options.stateDir ?? '.heddle', 'memory');
|
|
15
|
+
const tools = [
|
|
16
|
+
listFilesTool,
|
|
17
|
+
readFileTool,
|
|
18
|
+
editFileTool,
|
|
19
|
+
createSearchFilesTool({ excludedDirs: options.searchIgnoreDirs }),
|
|
20
|
+
createWebSearchTool({
|
|
21
|
+
model: options.model,
|
|
22
|
+
apiKey: options.apiKey,
|
|
23
|
+
}),
|
|
24
|
+
createViewImageTool({
|
|
25
|
+
model: options.model,
|
|
26
|
+
apiKey: options.apiKey,
|
|
27
|
+
}),
|
|
28
|
+
createListMemoryNotesTool({ memoryRoot }),
|
|
29
|
+
createReadMemoryNoteTool({ memoryRoot }),
|
|
30
|
+
createSearchMemoryNotesTool({ memoryRoot }),
|
|
31
|
+
createEditMemoryNoteTool({ memoryRoot }),
|
|
32
|
+
reportStateTool,
|
|
33
|
+
];
|
|
34
|
+
if (options.includePlanTool ?? true) {
|
|
35
|
+
tools.push(updatePlanTool);
|
|
36
|
+
}
|
|
37
|
+
tools.push(createRunShellInspectTool(), createRunShellMutateTool());
|
|
38
|
+
return tools;
|
|
39
|
+
}
|
|
40
|
+
//# sourceMappingURL=default-tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"default-tools.js","sourceRoot":"","sources":["../../../src/runtime/default-tools.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EACL,wBAAwB,EACxB,yBAAyB,EACzB,wBAAwB,EACxB,2BAA2B,GAC5B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAE,yBAAyB,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AAC5F,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAY7D,MAAM,UAAU,uBAAuB,CAAC,OAAiC;IACvE,MAAM,UAAU,GACd,OAAO,CAAC,SAAS;QACjB,IAAI,CAAC,OAAO,CAAC,aAAa,IAAI,OAAO,CAAC,GAAG,EAAE,EAAE,OAAO,CAAC,QAAQ,IAAI,SAAS,EAAE,QAAQ,CAAC,CAAC;IACxF,MAAM,KAAK,GAAqB;QAC9B,aAAa;QACb,YAAY;QACZ,YAAY;QACZ,qBAAqB,CAAC,EAAE,YAAY,EAAE,OAAO,CAAC,gBAAgB,EAAE,CAAC;QACjE,mBAAmB,CAAC;YAClB,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC;QACF,mBAAmB,CAAC;YAClB,KAAK,EAAE,OAAO,CAAC,KAAK;YACpB,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC;QACF,yBAAyB,CAAC,EAAE,UAAU,EAAE,CAAC;QACzC,wBAAwB,CAAC,EAAE,UAAU,EAAE,CAAC;QACxC,2BAA2B,CAAC,EAAE,UAAU,EAAE,CAAC;QAC3C,wBAAwB,CAAC,EAAE,UAAU,EAAE,CAAC;QACxC,eAAe;KAChB,CAAC;IAEF,IAAI,OAAO,CAAC,eAAe,IAAI,IAAI,EAAE,CAAC;QACpC,KAAK,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,wBAAwB,EAAE,CAAC,CAAC;IACpE,OAAO,KAAK,CAAC;AACf,CAAC"}
|
|
@@ -0,0 +1,128 @@
|
|
|
1
|
+
import type { ChatMessage, LlmProvider, LlmUsage } from '../llm/types.js';
|
|
2
|
+
import type { RunResult, StopReason, TraceEvent } from '../types.js';
|
|
3
|
+
export type AgentLoopStatus = 'finished';
|
|
4
|
+
export type AgentLoopState = {
|
|
5
|
+
status: AgentLoopStatus;
|
|
6
|
+
runId: string;
|
|
7
|
+
goal: string;
|
|
8
|
+
model: string;
|
|
9
|
+
provider: LlmProvider;
|
|
10
|
+
workspaceRoot: string;
|
|
11
|
+
startedAt: string;
|
|
12
|
+
finishedAt: string;
|
|
13
|
+
outcome: StopReason;
|
|
14
|
+
summary: string;
|
|
15
|
+
usage?: LlmUsage;
|
|
16
|
+
transcript: ChatMessage[];
|
|
17
|
+
trace: TraceEvent[];
|
|
18
|
+
};
|
|
19
|
+
export type AgentLoopCheckpoint = {
|
|
20
|
+
version: 1;
|
|
21
|
+
runId: string;
|
|
22
|
+
createdAt: string;
|
|
23
|
+
state: AgentLoopState;
|
|
24
|
+
};
|
|
25
|
+
/**
|
|
26
|
+
* Host-facing agent loop events.
|
|
27
|
+
* Stable payload shapes for external integration (CyberLoop, Lucid, etc.)
|
|
28
|
+
*/
|
|
29
|
+
export type AgentLoopEvent = {
|
|
30
|
+
type: 'loop.started';
|
|
31
|
+
runId: string;
|
|
32
|
+
goal: string;
|
|
33
|
+
model: string;
|
|
34
|
+
provider: LlmProvider;
|
|
35
|
+
workspaceRoot: string;
|
|
36
|
+
resumedFromCheckpoint?: string;
|
|
37
|
+
timestamp: string;
|
|
38
|
+
} | {
|
|
39
|
+
type: 'loop.resumed';
|
|
40
|
+
runId: string;
|
|
41
|
+
fromCheckpoint: string;
|
|
42
|
+
priorTraceEvents: number;
|
|
43
|
+
timestamp: string;
|
|
44
|
+
} | {
|
|
45
|
+
type: 'assistant.stream';
|
|
46
|
+
runId: string;
|
|
47
|
+
step: number;
|
|
48
|
+
text: string;
|
|
49
|
+
done: boolean;
|
|
50
|
+
timestamp: string;
|
|
51
|
+
} | {
|
|
52
|
+
type: 'tool.calling';
|
|
53
|
+
runId: string;
|
|
54
|
+
step: number;
|
|
55
|
+
tool: string;
|
|
56
|
+
toolCallId: string;
|
|
57
|
+
input: unknown;
|
|
58
|
+
requiresApproval: boolean;
|
|
59
|
+
timestamp: string;
|
|
60
|
+
} | {
|
|
61
|
+
type: 'tool.completed';
|
|
62
|
+
runId: string;
|
|
63
|
+
step: number;
|
|
64
|
+
tool: string;
|
|
65
|
+
toolCallId: string;
|
|
66
|
+
result: {
|
|
67
|
+
ok: boolean;
|
|
68
|
+
output?: unknown;
|
|
69
|
+
error?: string;
|
|
70
|
+
};
|
|
71
|
+
durationMs: number;
|
|
72
|
+
timestamp: string;
|
|
73
|
+
} | {
|
|
74
|
+
type: 'trace';
|
|
75
|
+
runId: string;
|
|
76
|
+
event: TraceEvent;
|
|
77
|
+
timestamp: string;
|
|
78
|
+
} | {
|
|
79
|
+
type: 'checkpoint.saved';
|
|
80
|
+
runId: string;
|
|
81
|
+
checkpoint: AgentLoopCheckpoint;
|
|
82
|
+
step: number;
|
|
83
|
+
timestamp: string;
|
|
84
|
+
} | {
|
|
85
|
+
type: 'escalation.required';
|
|
86
|
+
runId: string;
|
|
87
|
+
task: string;
|
|
88
|
+
outcome: StopReason;
|
|
89
|
+
summary: string;
|
|
90
|
+
step: number;
|
|
91
|
+
timestamp: string;
|
|
92
|
+
} | {
|
|
93
|
+
type: 'heartbeat.decision';
|
|
94
|
+
runId: string;
|
|
95
|
+
decision: 'continue' | 'pause' | 'complete' | 'escalate';
|
|
96
|
+
outcome: StopReason;
|
|
97
|
+
summary: string;
|
|
98
|
+
timestamp: string;
|
|
99
|
+
} | {
|
|
100
|
+
type: 'loop.finished';
|
|
101
|
+
runId: string;
|
|
102
|
+
outcome: RunResult['outcome'];
|
|
103
|
+
summary: string;
|
|
104
|
+
usage: RunResult['usage'];
|
|
105
|
+
state: AgentLoopState;
|
|
106
|
+
timestamp: string;
|
|
107
|
+
};
|
|
108
|
+
export declare function createFinishedAgentLoopState(args: {
|
|
109
|
+
runId: string;
|
|
110
|
+
goal: string;
|
|
111
|
+
model: string;
|
|
112
|
+
provider: LlmProvider;
|
|
113
|
+
workspaceRoot: string;
|
|
114
|
+
startedAt: string;
|
|
115
|
+
finishedAt: string;
|
|
116
|
+
result: RunResult;
|
|
117
|
+
}): AgentLoopState;
|
|
118
|
+
export declare function createAgentLoopCheckpoint(state: AgentLoopState, options?: {
|
|
119
|
+
createdAt?: string;
|
|
120
|
+
}): AgentLoopCheckpoint;
|
|
121
|
+
/**
|
|
122
|
+
* Generate a unique run ID for event correlation.
|
|
123
|
+
* Format: run_<timestamp>_<random>
|
|
124
|
+
*/
|
|
125
|
+
export declare function generateRunId(): string;
|
|
126
|
+
export declare function getHistoryFromAgentLoopState(state: AgentLoopState): ChatMessage[];
|
|
127
|
+
export declare function getHistoryFromAgentLoopCheckpoint(checkpoint: AgentLoopCheckpoint): ChatMessage[];
|
|
128
|
+
//# sourceMappingURL=events.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"events.d.ts","sourceRoot":"","sources":["../../../src/runtime/events.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC1E,OAAO,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErE,MAAM,MAAM,eAAe,GAAG,UAAU,CAAC;AAEzC,MAAM,MAAM,cAAc,GAAG;IAC3B,MAAM,EAAE,eAAe,CAAC;IACxB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,WAAW,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,UAAU,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,QAAQ,CAAC;IACjB,UAAU,EAAE,WAAW,EAAE,CAAC;IAC1B,KAAK,EAAE,UAAU,EAAE,CAAC;CACrB,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,EAAE,CAAC,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,cAAc,CAAC;CACvB,CAAC;AAEF;;;GAGG;AACH,MAAM,MAAM,cAAc,GACtB;IACE,IAAI,EAAE,cAAc,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,WAAW,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IACE,IAAI,EAAE,cAAc,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,cAAc,EAAE,MAAM,CAAC;IACvB,gBAAgB,EAAE,MAAM,CAAC;IACzB,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IACE,IAAI,EAAE,kBAAkB,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,OAAO,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IACE,IAAI,EAAE,cAAc,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,OAAO,CAAC;IACf,gBAAgB,EAAE,OAAO,CAAC;IAC1B,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IACE,IAAI,EAAE,gBAAgB,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE;QAAE,EAAE,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,OAAO,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAC1D,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IACE,IAAI,EAAE,OAAO,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,UAAU,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IACE,IAAI,EAAE,kBAAkB,CAAC;IACzB,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,mBAAmB,CAAC;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IACE,IAAI,EAAE,qBAAqB,CAAC;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,UAAU,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IACE,IAAI,EAAE,oBAAoB,CAAC;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,UAAU,GAAG,OAAO,GAAG,UAAU,GAAG,UAAU,CAAC;IACzD,OAAO,EAAE,UAAU,CAAC;IACpB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IACE,IAAI,EAAE,eAAe,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,SAAS,CAAC,SAAS,CAAC,CAAC;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC;IAC1B,KAAK,EAAE,cAAc,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEN,wBAAgB,4BAA4B,CAAC,IAAI,EAAE;IACjD,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,WAAW,CAAC;IACtB,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,SAAS,CAAC;CACnB,GAAG,cAAc,CAgBjB;AAED,wBAAgB,yBAAyB,CACvC,KAAK,EAAE,cAAc,EACrB,OAAO,GAAE;IAAE,SAAS,CAAC,EAAE,MAAM,CAAA;CAAO,GACnC,mBAAmB,CAOrB;AAED;;;GAGG;AACH,wBAAgB,aAAa,IAAI,MAAM,CAItC;AAED,wBAAgB,4BAA4B,CAAC,KAAK,EAAE,cAAc,GAAG,WAAW,EAAE,CAEjF;AAED,wBAAgB,iCAAiC,CAAC,UAAU,EAAE,mBAAmB,GAAG,WAAW,EAAE,CAEhG"}
|
|
@@ -0,0 +1,41 @@
|
|
|
1
|
+
export function createFinishedAgentLoopState(args) {
|
|
2
|
+
return {
|
|
3
|
+
status: 'finished',
|
|
4
|
+
runId: args.runId,
|
|
5
|
+
goal: args.goal,
|
|
6
|
+
model: args.model,
|
|
7
|
+
provider: args.provider,
|
|
8
|
+
workspaceRoot: args.workspaceRoot,
|
|
9
|
+
startedAt: args.startedAt,
|
|
10
|
+
finishedAt: args.finishedAt,
|
|
11
|
+
outcome: args.result.outcome,
|
|
12
|
+
summary: args.result.summary,
|
|
13
|
+
usage: args.result.usage,
|
|
14
|
+
transcript: args.result.transcript,
|
|
15
|
+
trace: args.result.trace,
|
|
16
|
+
};
|
|
17
|
+
}
|
|
18
|
+
export function createAgentLoopCheckpoint(state, options = {}) {
|
|
19
|
+
return {
|
|
20
|
+
version: 1,
|
|
21
|
+
runId: state.runId,
|
|
22
|
+
createdAt: options.createdAt ?? new Date().toISOString(),
|
|
23
|
+
state,
|
|
24
|
+
};
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* Generate a unique run ID for event correlation.
|
|
28
|
+
* Format: run_<timestamp>_<random>
|
|
29
|
+
*/
|
|
30
|
+
export function generateRunId() {
|
|
31
|
+
const timestamp = Date.now().toString(36);
|
|
32
|
+
const random = Math.random().toString(36).slice(2, 6);
|
|
33
|
+
return `run_${timestamp}_${random}`;
|
|
34
|
+
}
|
|
35
|
+
export function getHistoryFromAgentLoopState(state) {
|
|
36
|
+
return state.transcript;
|
|
37
|
+
}
|
|
38
|
+
export function getHistoryFromAgentLoopCheckpoint(checkpoint) {
|
|
39
|
+
return getHistoryFromAgentLoopState(checkpoint.state);
|
|
40
|
+
}
|
|
41
|
+
//# sourceMappingURL=events.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"events.js","sourceRoot":"","sources":["../../../src/runtime/events.ts"],"names":[],"mappings":"AAsHA,MAAM,UAAU,4BAA4B,CAAC,IAS5C;IACC,OAAO;QACL,MAAM,EAAE,UAAU;QAClB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,aAAa,EAAE,IAAI,CAAC,aAAa;QACjC,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;QAC5B,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;QAC5B,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;QACxB,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU;QAClC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;KACzB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,KAAqB,EACrB,UAAkC,EAAE;IAEpC,OAAO;QACL,OAAO,EAAE,CAAC;QACV,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,SAAS,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACxD,KAAK;KACN,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,aAAa;IAC3B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACtD,OAAO,OAAO,SAAS,IAAI,MAAM,EAAE,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,4BAA4B,CAAC,KAAqB;IAChE,OAAO,KAAK,CAAC,UAAU,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,iCAAiC,CAAC,UAA+B;IAC/E,OAAO,4BAA4B,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;AACxD,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { AgentHeartbeatResult, RunAgentHeartbeatOptions } from './heartbeat.js';
|
|
2
|
+
import type { AgentLoopCheckpoint } from './events.js';
|
|
3
|
+
export type HeartbeatCheckpointStore = {
|
|
4
|
+
load: () => Promise<AgentLoopCheckpoint | undefined>;
|
|
5
|
+
save: (checkpoint: AgentLoopCheckpoint) => Promise<void>;
|
|
6
|
+
};
|
|
7
|
+
export type FileHeartbeatCheckpointStoreOptions = {
|
|
8
|
+
path: string;
|
|
9
|
+
};
|
|
10
|
+
export type RunStoredHeartbeatOptions = Omit<RunAgentHeartbeatOptions, 'checkpoint'> & {
|
|
11
|
+
store: HeartbeatCheckpointStore;
|
|
12
|
+
};
|
|
13
|
+
export type StoredHeartbeatResult = AgentHeartbeatResult & {
|
|
14
|
+
loadedCheckpoint: boolean;
|
|
15
|
+
nextDelayMs?: number;
|
|
16
|
+
};
|
|
17
|
+
export declare function createFileHeartbeatCheckpointStore(options: FileHeartbeatCheckpointStoreOptions): HeartbeatCheckpointStore;
|
|
18
|
+
export declare function runStoredHeartbeat(options: RunStoredHeartbeatOptions): Promise<StoredHeartbeatResult>;
|
|
19
|
+
export declare function suggestNextHeartbeatDelayMs(decision: AgentHeartbeatResult['decision']): number | undefined;
|
|
20
|
+
//# sourceMappingURL=heartbeat-store.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"heartbeat-store.d.ts","sourceRoot":"","sources":["../../../src/runtime/heartbeat-store.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,oBAAoB,EAAE,wBAAwB,EAAE,MAAM,gBAAgB,CAAC;AACrF,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAEvD,MAAM,MAAM,wBAAwB,GAAG;IACrC,IAAI,EAAE,MAAM,OAAO,CAAC,mBAAmB,GAAG,SAAS,CAAC,CAAC;IACrD,IAAI,EAAE,CAAC,UAAU,EAAE,mBAAmB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1D,CAAC;AAEF,MAAM,MAAM,mCAAmC,GAAG;IAChD,IAAI,EAAE,MAAM,CAAC;CACd,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG,IAAI,CAAC,wBAAwB,EAAE,YAAY,CAAC,GAAG;IACrF,KAAK,EAAE,wBAAwB,CAAC;CACjC,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG,oBAAoB,GAAG;IACzD,gBAAgB,EAAE,OAAO,CAAC;IAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB,CAAC;AAEF,wBAAgB,kCAAkC,CAAC,OAAO,EAAE,mCAAmC,GAAG,wBAAwB,CAczH;AAED,wBAAsB,kBAAkB,CAAC,OAAO,EAAE,yBAAyB,GAAG,OAAO,CAAC,qBAAqB,CAAC,CAa3G;AAED,wBAAgB,2BAA2B,CAAC,QAAQ,EAAE,oBAAoB,CAAC,UAAU,CAAC,GAAG,MAAM,GAAG,SAAS,CAU1G"}
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'node:fs';
|
|
2
|
+
import { dirname } from 'node:path';
|
|
3
|
+
import { runAgentHeartbeat } from './heartbeat.js';
|
|
4
|
+
export function createFileHeartbeatCheckpointStore(options) {
|
|
5
|
+
return {
|
|
6
|
+
async load() {
|
|
7
|
+
if (!existsSync(options.path)) {
|
|
8
|
+
return undefined;
|
|
9
|
+
}
|
|
10
|
+
return JSON.parse(readFileSync(options.path, 'utf8'));
|
|
11
|
+
},
|
|
12
|
+
async save(checkpoint) {
|
|
13
|
+
mkdirSync(dirname(options.path), { recursive: true });
|
|
14
|
+
writeFileSync(options.path, JSON.stringify(checkpoint, null, 2));
|
|
15
|
+
},
|
|
16
|
+
};
|
|
17
|
+
}
|
|
18
|
+
export async function runStoredHeartbeat(options) {
|
|
19
|
+
const checkpoint = await options.store.load();
|
|
20
|
+
const result = await runAgentHeartbeat({
|
|
21
|
+
...options,
|
|
22
|
+
checkpoint,
|
|
23
|
+
});
|
|
24
|
+
await options.store.save(result.checkpoint);
|
|
25
|
+
return {
|
|
26
|
+
...result,
|
|
27
|
+
loadedCheckpoint: Boolean(checkpoint),
|
|
28
|
+
nextDelayMs: suggestNextHeartbeatDelayMs(result.decision),
|
|
29
|
+
};
|
|
30
|
+
}
|
|
31
|
+
export function suggestNextHeartbeatDelayMs(decision) {
|
|
32
|
+
switch (decision) {
|
|
33
|
+
case 'continue':
|
|
34
|
+
return 60_000;
|
|
35
|
+
case 'pause':
|
|
36
|
+
return 15 * 60_000;
|
|
37
|
+
case 'complete':
|
|
38
|
+
case 'escalate':
|
|
39
|
+
return undefined;
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
//# sourceMappingURL=heartbeat-store.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"heartbeat-store.js","sourceRoot":"","sources":["../../../src/runtime/heartbeat-store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC7E,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AAsBnD,MAAM,UAAU,kCAAkC,CAAC,OAA4C;IAC7F,OAAO;QACL,KAAK,CAAC,IAAI;YACR,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9B,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAwB,CAAC;QAC/E,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,UAAU;YACnB,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACtD,aAAa,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;QACnE,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,OAAkC;IACzE,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;IAC9C,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC;QACrC,GAAG,OAAO;QACV,UAAU;KACX,CAAC,CAAC;IACH,MAAM,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAE5C,OAAO;QACL,GAAG,MAAM;QACT,gBAAgB,EAAE,OAAO,CAAC,UAAU,CAAC;QACrC,WAAW,EAAE,2BAA2B,CAAC,MAAM,CAAC,QAAQ,CAAC;KAC1D,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,QAA0C;IACpF,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,UAAU;YACb,OAAO,MAAM,CAAC;QAChB,KAAK,OAAO;YACV,OAAO,EAAE,GAAG,MAAM,CAAC;QACrB,KAAK,UAAU,CAAC;QAChB,KAAK,UAAU;YACb,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
import type { Logger } from 'pino';
|
|
2
|
+
import type { ChatMessage, LlmAdapter } from '../llm/types.js';
|
|
3
|
+
import type { ToolCall, ToolDefinition } from '../types.js';
|
|
4
|
+
import type { AgentLoopCheckpoint, AgentLoopEvent, AgentLoopState } from './events.js';
|
|
5
|
+
export type HeartbeatDecision = 'continue' | 'pause' | 'complete' | 'escalate';
|
|
6
|
+
export type RunAgentHeartbeatOptions = {
|
|
7
|
+
task: string;
|
|
8
|
+
checkpoint?: AgentLoopState | AgentLoopCheckpoint;
|
|
9
|
+
model?: string;
|
|
10
|
+
apiKey?: string;
|
|
11
|
+
maxSteps?: number;
|
|
12
|
+
workspaceRoot?: string;
|
|
13
|
+
stateDir?: string;
|
|
14
|
+
memoryDir?: string;
|
|
15
|
+
searchIgnoreDirs?: string[];
|
|
16
|
+
systemContext?: string;
|
|
17
|
+
history?: ChatMessage[];
|
|
18
|
+
llm?: LlmAdapter;
|
|
19
|
+
tools?: ToolDefinition[];
|
|
20
|
+
extraTools?: ToolDefinition[];
|
|
21
|
+
includeDefaultTools?: boolean;
|
|
22
|
+
includePlanTool?: boolean;
|
|
23
|
+
logger?: Logger;
|
|
24
|
+
onEvent?: (event: AgentLoopEvent) => void;
|
|
25
|
+
approveToolCall?: (call: ToolCall, tool: ToolDefinition) => Promise<{
|
|
26
|
+
approved: boolean;
|
|
27
|
+
reason?: string;
|
|
28
|
+
}>;
|
|
29
|
+
shouldStop?: () => boolean;
|
|
30
|
+
abortSignal?: AbortSignal;
|
|
31
|
+
};
|
|
32
|
+
export type AgentHeartbeatResult = {
|
|
33
|
+
decision: HeartbeatDecision;
|
|
34
|
+
summary: string;
|
|
35
|
+
checkpoint: AgentLoopCheckpoint;
|
|
36
|
+
state: AgentLoopState;
|
|
37
|
+
};
|
|
38
|
+
export declare function runAgentHeartbeat(options: RunAgentHeartbeatOptions): Promise<AgentHeartbeatResult>;
|
|
39
|
+
//# sourceMappingURL=heartbeat.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"heartbeat.d.ts","sourceRoot":"","sources":["../../../src/runtime/heartbeat.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AACnC,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAG5D,OAAO,KAAK,EAAE,mBAAmB,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAIvF,MAAM,MAAM,iBAAiB,GAAG,UAAU,GAAG,OAAO,GAAG,UAAU,GAAG,UAAU,CAAC;AAE/E,MAAM,MAAM,wBAAwB,GAAG;IACrC,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,CAAC,EAAE,cAAc,GAAG,mBAAmB,CAAC;IAClD,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,gBAAgB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC5B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,OAAO,CAAC,EAAE,WAAW,EAAE,CAAC;IACxB,GAAG,CAAC,EAAE,UAAU,CAAC;IACjB,KAAK,CAAC,EAAE,cAAc,EAAE,CAAC;IACzB,UAAU,CAAC,EAAE,cAAc,EAAE,CAAC;IAC9B,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,cAAc,KAAK,IAAI,CAAC;IAC1C,eAAe,CAAC,EAAE,CAAC,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,cAAc,KAAK,OAAO,CAAC;QAAE,QAAQ,EAAE,OAAO,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC5G,UAAU,CAAC,EAAE,MAAM,OAAO,CAAC;IAC3B,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B,CAAC;AAEF,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,mBAAmB,CAAC;IAChC,KAAK,EAAE,cAAc,CAAC;CACvB,CAAC;AAEF,wBAAsB,iBAAiB,CAAC,OAAO,EAAE,wBAAwB,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAiExG"}
|