neoctl 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.
Files changed (218) hide show
  1. package/README.md +146 -0
  2. package/dist/agents/agent-definition.d.ts +39 -0
  3. package/dist/agents/agent-definition.js +57 -0
  4. package/dist/agents/agent-definition.js.map +1 -0
  5. package/dist/agents/agent-tool.d.ts +38 -0
  6. package/dist/agents/agent-tool.js +336 -0
  7. package/dist/agents/agent-tool.js.map +1 -0
  8. package/dist/agents/local-agent-task.d.ts +52 -0
  9. package/dist/agents/local-agent-task.js +63 -0
  10. package/dist/agents/local-agent-task.js.map +1 -0
  11. package/dist/agents/smoke-agents.d.ts +1 -0
  12. package/dist/agents/smoke-agents.js +142 -0
  13. package/dist/agents/smoke-agents.js.map +1 -0
  14. package/dist/agents/team.d.ts +10 -0
  15. package/dist/agents/team.js +2 -0
  16. package/dist/agents/team.js.map +1 -0
  17. package/dist/app/app-state.d.ts +21 -0
  18. package/dist/app/app-state.js +24 -0
  19. package/dist/app/app-state.js.map +1 -0
  20. package/dist/context/compaction.d.ts +49 -0
  21. package/dist/context/compaction.js +334 -0
  22. package/dist/context/compaction.js.map +1 -0
  23. package/dist/context/context-manager.d.ts +53 -0
  24. package/dist/context/context-manager.js +98 -0
  25. package/dist/context/context-manager.js.map +1 -0
  26. package/dist/context/prompts.d.ts +22 -0
  27. package/dist/context/prompts.js +84 -0
  28. package/dist/context/prompts.js.map +1 -0
  29. package/dist/context/smoke-context.d.ts +1 -0
  30. package/dist/context/smoke-context.js +151 -0
  31. package/dist/context/smoke-context.js.map +1 -0
  32. package/dist/core/assistant-output-filter.d.ts +9 -0
  33. package/dist/core/assistant-output-filter.js +78 -0
  34. package/dist/core/assistant-output-filter.js.map +1 -0
  35. package/dist/core/context-metrics.d.ts +10 -0
  36. package/dist/core/context-metrics.js +77 -0
  37. package/dist/core/context-metrics.js.map +1 -0
  38. package/dist/core/message-pipeline.d.ts +10 -0
  39. package/dist/core/message-pipeline.js +138 -0
  40. package/dist/core/message-pipeline.js.map +1 -0
  41. package/dist/core/query-engine.d.ts +86 -0
  42. package/dist/core/query-engine.js +337 -0
  43. package/dist/core/query-engine.js.map +1 -0
  44. package/dist/core/query.d.ts +47 -0
  45. package/dist/core/query.js +408 -0
  46. package/dist/core/query.js.map +1 -0
  47. package/dist/core/run-agent.d.ts +48 -0
  48. package/dist/core/run-agent.js +150 -0
  49. package/dist/core/run-agent.js.map +1 -0
  50. package/dist/core/smoke-core-loop.d.ts +1 -0
  51. package/dist/core/smoke-core-loop.js +42 -0
  52. package/dist/core/smoke-core-loop.js.map +1 -0
  53. package/dist/core/state.d.ts +37 -0
  54. package/dist/core/state.js +31 -0
  55. package/dist/core/state.js.map +1 -0
  56. package/dist/index.d.ts +42 -0
  57. package/dist/index.js +43 -0
  58. package/dist/index.js.map +1 -0
  59. package/dist/model/communication-logger.d.ts +30 -0
  60. package/dist/model/communication-logger.js +218 -0
  61. package/dist/model/communication-logger.js.map +1 -0
  62. package/dist/model/config.d.ts +25 -0
  63. package/dist/model/config.js +65 -0
  64. package/dist/model/config.js.map +1 -0
  65. package/dist/model/context-window.d.ts +32 -0
  66. package/dist/model/context-window.js +76 -0
  67. package/dist/model/context-window.js.map +1 -0
  68. package/dist/model/credentials.d.ts +13 -0
  69. package/dist/model/credentials.js +19 -0
  70. package/dist/model/credentials.js.map +1 -0
  71. package/dist/model/env.d.ts +8 -0
  72. package/dist/model/env.js +36 -0
  73. package/dist/model/env.js.map +1 -0
  74. package/dist/model/errors.d.ts +42 -0
  75. package/dist/model/errors.js +74 -0
  76. package/dist/model/errors.js.map +1 -0
  77. package/dist/model/http-transport.d.ts +26 -0
  78. package/dist/model/http-transport.js +118 -0
  79. package/dist/model/http-transport.js.map +1 -0
  80. package/dist/model/model-gateway.d.ts +103 -0
  81. package/dist/model/model-gateway.js +15 -0
  82. package/dist/model/model-gateway.js.map +1 -0
  83. package/dist/model/model-metadata.json +677 -0
  84. package/dist/model/openai-adapter.d.ts +34 -0
  85. package/dist/model/openai-adapter.js +152 -0
  86. package/dist/model/openai-adapter.js.map +1 -0
  87. package/dist/model/openai-chat-mapper.d.ts +11 -0
  88. package/dist/model/openai-chat-mapper.js +114 -0
  89. package/dist/model/openai-chat-mapper.js.map +1 -0
  90. package/dist/model/openai-mappers.d.ts +21 -0
  91. package/dist/model/openai-mappers.js +261 -0
  92. package/dist/model/openai-mappers.js.map +1 -0
  93. package/dist/model/openai-responses-adapter.d.ts +2 -0
  94. package/dist/model/openai-responses-adapter.js +2 -0
  95. package/dist/model/openai-responses-adapter.js.map +1 -0
  96. package/dist/model/openai-responses-mapper.d.ts +11 -0
  97. package/dist/model/openai-responses-mapper.js +218 -0
  98. package/dist/model/openai-responses-mapper.js.map +1 -0
  99. package/dist/model/provider-adapter.d.ts +17 -0
  100. package/dist/model/provider-adapter.js +2 -0
  101. package/dist/model/provider-adapter.js.map +1 -0
  102. package/dist/model/provider-factory.d.ts +4 -0
  103. package/dist/model/provider-factory.js +29 -0
  104. package/dist/model/provider-factory.js.map +1 -0
  105. package/dist/model/retry-runner.d.ts +8 -0
  106. package/dist/model/retry-runner.js +28 -0
  107. package/dist/model/retry-runner.js.map +1 -0
  108. package/dist/model/smoke-openai.d.ts +1 -0
  109. package/dist/model/smoke-openai.js +44 -0
  110. package/dist/model/smoke-openai.js.map +1 -0
  111. package/dist/model/smoke-responses-mapper.d.ts +1 -0
  112. package/dist/model/smoke-responses-mapper.js +72 -0
  113. package/dist/model/smoke-responses-mapper.js.map +1 -0
  114. package/dist/model/sse-decoder.d.ts +5 -0
  115. package/dist/model/sse-decoder.js +79 -0
  116. package/dist/model/sse-decoder.js.map +1 -0
  117. package/dist/repl/clipboard.d.ts +14 -0
  118. package/dist/repl/clipboard.js +92 -0
  119. package/dist/repl/clipboard.js.map +1 -0
  120. package/dist/repl/commands.d.ts +38 -0
  121. package/dist/repl/commands.js +86 -0
  122. package/dist/repl/commands.js.map +1 -0
  123. package/dist/repl/index.d.ts +2 -0
  124. package/dist/repl/index.js +2836 -0
  125. package/dist/repl/index.js.map +1 -0
  126. package/dist/repl/markdown-renderer.d.ts +81 -0
  127. package/dist/repl/markdown-renderer.js +546 -0
  128. package/dist/repl/markdown-renderer.js.map +1 -0
  129. package/dist/repl/render.d.ts +2 -0
  130. package/dist/repl/render.js +45 -0
  131. package/dist/repl/render.js.map +1 -0
  132. package/dist/repl/status-line.d.ts +19 -0
  133. package/dist/repl/status-line.js +160 -0
  134. package/dist/repl/status-line.js.map +1 -0
  135. package/dist/safety/audit.d.ts +10 -0
  136. package/dist/safety/audit.js +2 -0
  137. package/dist/safety/audit.js.map +1 -0
  138. package/dist/safety/permissions.d.ts +10 -0
  139. package/dist/safety/permissions.js +2 -0
  140. package/dist/safety/permissions.js.map +1 -0
  141. package/dist/safety/sandbox.d.ts +6 -0
  142. package/dist/safety/sandbox.js +2 -0
  143. package/dist/safety/sandbox.js.map +1 -0
  144. package/dist/session/session-store.d.ts +98 -0
  145. package/dist/session/session-store.js +249 -0
  146. package/dist/session/session-store.js.map +1 -0
  147. package/dist/session/smoke-session.d.ts +1 -0
  148. package/dist/session/smoke-session.js +85 -0
  149. package/dist/session/smoke-session.js.map +1 -0
  150. package/dist/session/tool-result-memory.d.ts +64 -0
  151. package/dist/session/tool-result-memory.js +303 -0
  152. package/dist/session/tool-result-memory.js.map +1 -0
  153. package/dist/skills/skill-tool.d.ts +28 -0
  154. package/dist/skills/skill-tool.js +107 -0
  155. package/dist/skills/skill-tool.js.map +1 -0
  156. package/dist/skills/smoke-skills.d.ts +1 -0
  157. package/dist/skills/smoke-skills.js +60 -0
  158. package/dist/skills/smoke-skills.js.map +1 -0
  159. package/dist/tasks/task-store.d.ts +43 -0
  160. package/dist/tasks/task-store.js +150 -0
  161. package/dist/tasks/task-store.js.map +1 -0
  162. package/dist/tasks/task-tools.d.ts +31 -0
  163. package/dist/tasks/task-tools.js +232 -0
  164. package/dist/tasks/task-tools.js.map +1 -0
  165. package/dist/tools/builtins/echo-tool.d.ts +4 -0
  166. package/dist/tools/builtins/echo-tool.js +26 -0
  167. package/dist/tools/builtins/echo-tool.js.map +1 -0
  168. package/dist/tools/builtins/edit-tool.d.ts +30 -0
  169. package/dist/tools/builtins/edit-tool.js +423 -0
  170. package/dist/tools/builtins/edit-tool.js.map +1 -0
  171. package/dist/tools/builtins/exec-tool.d.ts +18 -0
  172. package/dist/tools/builtins/exec-tool.js +290 -0
  173. package/dist/tools/builtins/exec-tool.js.map +1 -0
  174. package/dist/tools/builtins/filesystem-tools.d.ts +16 -0
  175. package/dist/tools/builtins/filesystem-tools.js +306 -0
  176. package/dist/tools/builtins/filesystem-tools.js.map +1 -0
  177. package/dist/tools/builtins/grep-tool.d.ts +56 -0
  178. package/dist/tools/builtins/grep-tool.js +320 -0
  179. package/dist/tools/builtins/grep-tool.js.map +1 -0
  180. package/dist/tools/builtins/ripgrep-binary.d.ts +6 -0
  181. package/dist/tools/builtins/ripgrep-binary.js +51 -0
  182. package/dist/tools/builtins/ripgrep-binary.js.map +1 -0
  183. package/dist/tools/builtins/search-providers.d.ts +49 -0
  184. package/dist/tools/builtins/search-providers.js +240 -0
  185. package/dist/tools/builtins/search-providers.js.map +1 -0
  186. package/dist/tools/builtins/search-tool.d.ts +18 -0
  187. package/dist/tools/builtins/search-tool.js +171 -0
  188. package/dist/tools/builtins/search-tool.js.map +1 -0
  189. package/dist/tools/registry.d.ts +18 -0
  190. package/dist/tools/registry.js +69 -0
  191. package/dist/tools/registry.js.map +1 -0
  192. package/dist/tools/run-tool-use.d.ts +13 -0
  193. package/dist/tools/run-tool-use.js +113 -0
  194. package/dist/tools/run-tool-use.js.map +1 -0
  195. package/dist/tools/schema.d.ts +2 -0
  196. package/dist/tools/schema.js +61 -0
  197. package/dist/tools/schema.js.map +1 -0
  198. package/dist/tools/smoke-tool-system.d.ts +1 -0
  199. package/dist/tools/smoke-tool-system.js +204 -0
  200. package/dist/tools/smoke-tool-system.js.map +1 -0
  201. package/dist/tools/streaming-tool-executor.d.ts +19 -0
  202. package/dist/tools/streaming-tool-executor.js +89 -0
  203. package/dist/tools/streaming-tool-executor.js.map +1 -0
  204. package/dist/tools/tool-orchestration.d.ts +15 -0
  205. package/dist/tools/tool-orchestration.js +89 -0
  206. package/dist/tools/tool-orchestration.js.map +1 -0
  207. package/dist/tools/tool.d.ts +121 -0
  208. package/dist/tools/tool.js +4 -0
  209. package/dist/tools/tool.js.map +1 -0
  210. package/dist/types/events.d.ts +66 -0
  211. package/dist/types/events.js +2 -0
  212. package/dist/types/events.js.map +1 -0
  213. package/dist/types/messages.d.ts +57 -0
  214. package/dist/types/messages.js +42 -0
  215. package/dist/types/messages.js.map +1 -0
  216. package/package.json +49 -0
  217. package/scripts/copy-model-metadata.mjs +4 -0
  218. package/scripts/install-ripgrep.cjs +152 -0
@@ -0,0 +1,408 @@
1
+ import { InMemoryAppState } from "../app/app-state.js";
2
+ import { ModelDrivenCompactor } from "../context/compaction.js";
3
+ import { DefaultContextManager } from "../context/context-manager.js";
4
+ import { ModelAPIError } from "../model/errors.js";
5
+ import { runTools } from "../tools/tool-orchestration.js";
6
+ import { createTextMessage, createThinkingMessage } from "../types/messages.js";
7
+ import { appendSystemContext, applyToolResultBudget, ensureToolResultPairing, getMessagesAfterCompactBoundary, prependUserContext, } from "./message-pipeline.js";
8
+ import { createInitialState, nextTracking, } from "./state.js";
9
+ import { AssistantOutputFilter } from "./assistant-output-filter.js";
10
+ import { buildContextMetrics } from "./context-metrics.js";
11
+ export async function* query(messages, dependencies, options) {
12
+ const initialState = createInitialState(messages, {
13
+ model: options.model,
14
+ fallbackModel: options.fallbackModel,
15
+ });
16
+ initialState.maxOutputTokensOverride = options.maxOutputTokensOverride;
17
+ const terminal = yield* queryLoop(initialState, dependencies, options);
18
+ yield { type: "terminal", reason: terminal };
19
+ return terminal;
20
+ }
21
+ async function* queryLoop(initialState, dependencies, options) {
22
+ const contextManager = dependencies.contextManager ?? new DefaultContextManager();
23
+ const compactor = dependencies.compactor ?? new ModelDrivenCompactor(dependencies.modelGateway);
24
+ const appState = new InMemoryAppState(options.agentId, options.workspaceCwd);
25
+ const maxTurns = options.maxTurns;
26
+ let state = initialState;
27
+ let toolContext = {
28
+ agentId: options.agentId,
29
+ abortSignal: options.abortSignal,
30
+ tools: dependencies.tools,
31
+ appState,
32
+ toolResultMemory: dependencies.toolResultMemory,
33
+ recordContentReplacements: dependencies.recordContentReplacements,
34
+ emit: () => undefined,
35
+ };
36
+ while (true) {
37
+ if (options.abortSignal?.aborted)
38
+ return "aborted_streaming";
39
+ if (maxTurns !== undefined && state.turnCount >= maxTurns)
40
+ return "max_turns";
41
+ state = beginTurn(state);
42
+ toolContext = { ...toolContext, queryTracking: state.queryTracking, messages: state.messages };
43
+ yield { type: "state", phase: state.phase, detail: `turn ${state.turnCount + 1} started (${state.transition.reason})` };
44
+ const context = await contextManager.build({
45
+ agentId: options.agentId,
46
+ messages: state.messages,
47
+ cwd: options.workspaceCwd,
48
+ enabledTools: dependencies.tools.definitions(toolContext).map((tool) => tool.name),
49
+ toolUseContext: toolContext,
50
+ });
51
+ const toolDefinitions = dependencies.tools.definitions(toolContext);
52
+ const systemPrompt = appendSystemContext(context.systemPrompt, context.systemContext);
53
+ const prepared = await prepareMessagesForQuery(state, context, dependencies, compactor, {
54
+ model: state.currentModel ?? options.model,
55
+ systemPrompt,
56
+ toolDefinitions,
57
+ toolUseContext: toolContext,
58
+ });
59
+ if (prepared.compaction?.changed) {
60
+ state = { ...state, messages: prepared.compactedMessages };
61
+ yield { type: "state", phase: "compacting", detail: formatCompactionDetail(prepared.compaction) };
62
+ for (const message of prepared.compactedMessages.filter((message) => message.metadata?.compactBoundary === true)) {
63
+ yield { type: "message", message };
64
+ }
65
+ }
66
+ const modelOutput = yield* callModelForTurn(state, context, prepared.messagesForQuery, dependencies, options, toolContext, {
67
+ systemPrompt,
68
+ toolDefinitions,
69
+ metrics: prepared.metrics,
70
+ });
71
+ if (modelOutput.reactiveCompact) {
72
+ state = modelOutput.reactiveCompact;
73
+ continue;
74
+ }
75
+ if (modelOutput.terminal)
76
+ return modelOutput.terminal;
77
+ if (!modelOutput.output)
78
+ return "model_error";
79
+ const { assistantMessages, toolUses, previousResponseId, incompleteReason } = modelOutput.output;
80
+ if (toolUses.length === 0) {
81
+ const recovery = maybeRecoverWithoutTools(state, incompleteReason, assistantMessages, previousResponseId, options);
82
+ if (recovery) {
83
+ state = recovery;
84
+ continue;
85
+ }
86
+ return "completed";
87
+ }
88
+ const toolResult = yield* executeToolsForTurn(toolUses, dependencies, options, toolContext);
89
+ if (toolResult.terminal)
90
+ return toolResult.terminal;
91
+ toolContext = toolResult.context;
92
+ const taskNotifications = collectTaskNotifications(dependencies.taskNotificationSource);
93
+ for (const notification of taskNotifications) {
94
+ yield { type: "message", message: notification };
95
+ }
96
+ state = buildNextTurnState(state, {
97
+ assistantMessages,
98
+ toolResults: [...toolResult.messages, ...taskNotifications],
99
+ previousResponseId,
100
+ });
101
+ }
102
+ }
103
+ function beginTurn(state) {
104
+ return {
105
+ ...state,
106
+ phase: "preparing",
107
+ queryTracking: nextTracking(state.queryTracking),
108
+ };
109
+ }
110
+ async function prepareMessagesForQuery(state, context, dependencies, compactor, telemetry) {
111
+ const baseMessages = state.modelInputMessages ?? getMessagesAfterCompactBoundary(state.messages);
112
+ const budgetResult = telemetry.toolUseContext?.toolResultMemory
113
+ ? await telemetry.toolUseContext.toolResultMemory.applyBudget(baseMessages, {
114
+ maxSerializedLength: dependencies.maxToolResultSerializedLength,
115
+ })
116
+ : {
117
+ messages: applyToolResultBudget(baseMessages, {
118
+ maxSerializedLength: dependencies.maxToolResultSerializedLength,
119
+ }),
120
+ records: [],
121
+ };
122
+ if (budgetResult.records.length)
123
+ telemetry.toolUseContext?.recordContentReplacements?.(budgetResult.records);
124
+ const budgeted = budgetResult.messages;
125
+ const pairedBudgeted = ensureToolResultPairing(budgeted);
126
+ const metricsBeforeCompact = buildContextMetrics({
127
+ model: telemetry.model,
128
+ messages: prependUserContext(pairedBudgeted, context.userContext),
129
+ systemPrompt: telemetry.systemPrompt,
130
+ tools: telemetry.toolDefinitions,
131
+ });
132
+ const compaction = await compactor.compact(pairedBudgeted, {
133
+ ...dependencies.contextBudget,
134
+ estimatedInputTokens: metricsBeforeCompact.estimatedInputTokens,
135
+ contextWindowTokens: metricsBeforeCompact.contextWindowTokens,
136
+ });
137
+ const compactedMessages = ensureToolResultPairing(compaction.messages);
138
+ const messagesForQuery = prependUserContext(compactedMessages, context.userContext);
139
+ const metrics = buildContextMetrics({
140
+ model: telemetry.model,
141
+ messages: messagesForQuery,
142
+ systemPrompt: telemetry.systemPrompt,
143
+ tools: telemetry.toolDefinitions,
144
+ });
145
+ return {
146
+ messagesForQuery,
147
+ compactedMessages,
148
+ compaction: compaction.changed ? compaction : undefined,
149
+ metrics,
150
+ };
151
+ }
152
+ async function* callModelForTurn(state, context, messagesForQuery, dependencies, options, toolContext, telemetry) {
153
+ const assistantMessages = [];
154
+ const toolUses = [];
155
+ const outputFilter = new AssistantOutputFilter();
156
+ const thinkingParts = [];
157
+ let previousResponseId = state.previousResponseId;
158
+ let incompleteReason;
159
+ let activeModel = state.currentModel ?? options.model;
160
+ yield { type: "state", phase: "compacting", detail: "messages prepared for model" };
161
+ yield { type: "context.metrics", metrics: telemetry.metrics };
162
+ yield { type: "state", phase: "calling_model", detail: activeModel ? `model stream opened (${activeModel})` : "model stream opened" };
163
+ try {
164
+ for await (const event of dependencies.modelGateway.stream({
165
+ model: activeModel,
166
+ fallbackModel: state.fallbackModel ?? options.fallbackModel,
167
+ messages: messagesForQuery,
168
+ systemPrompt: telemetry.systemPrompt,
169
+ tools: telemetry.toolDefinitions,
170
+ stream: true,
171
+ maxOutputTokens: state.maxOutputTokensOverride ?? options.maxOutputTokensOverride,
172
+ reasoning: options.reasoning,
173
+ previousResponseId: state.previousResponseId,
174
+ queryOrigin: options.queryOrigin,
175
+ cancellation: options.abortSignal,
176
+ })) {
177
+ if (options.abortSignal?.aborted)
178
+ return { terminal: "aborted_streaming" };
179
+ const handled = yield* handleModelEvent(event, assistantMessages, toolUses, outputFilter, thinkingParts);
180
+ previousResponseId = handled.previousResponseId ?? previousResponseId;
181
+ incompleteReason = handled.incompleteReason ?? incompleteReason;
182
+ if (event.type === "fallback_started") {
183
+ activeModel = event.toModel;
184
+ assistantMessages.length = 0;
185
+ toolUses.length = 0;
186
+ yield { type: "message", message: createTextMessage("progress", `Fallback model started: ${event.fromModel} -> ${event.toModel}`) };
187
+ }
188
+ }
189
+ }
190
+ catch (error) {
191
+ if (isContextLengthError(error) && !state.hasAttemptedReactiveCompact) {
192
+ const compactor = dependencies.compactor ?? new ModelDrivenCompactor(dependencies.modelGateway);
193
+ const normalized = error instanceof Error ? error : new Error(String(error));
194
+ const reactiveMetrics = buildContextMetrics({
195
+ model: activeModel,
196
+ messages: state.messages,
197
+ systemPrompt: telemetry.systemPrompt,
198
+ tools: telemetry.toolDefinitions,
199
+ });
200
+ const reactiveBudget = {
201
+ ...dependencies.contextBudget,
202
+ estimatedInputTokens: reactiveMetrics.estimatedInputTokens,
203
+ contextWindowTokens: reactiveMetrics.contextWindowTokens,
204
+ };
205
+ const compacted = await (compactor.reactiveCompact?.(state.messages, normalized, reactiveBudget) ?? compactor.compact(state.messages, reactiveBudget));
206
+ yield { type: "state", phase: "compacting", detail: "reactive compact retry after prompt-too-long" };
207
+ for (const message of compacted.messages.filter((message) => message.metadata?.compactBoundary === true)) {
208
+ yield { type: "message", message };
209
+ }
210
+ return {
211
+ reactiveCompact: {
212
+ ...state,
213
+ messages: compacted.messages,
214
+ modelInputMessages: compacted.messages,
215
+ hasAttemptedReactiveCompact: true,
216
+ turnCount: state.turnCount + 1,
217
+ transition: { reason: "reactive_compact_retry", detail: normalized.message },
218
+ },
219
+ };
220
+ }
221
+ const terminal = terminalForModelError(error);
222
+ yield { type: "error", error: error instanceof Error ? error : new Error(String(error)) };
223
+ return { terminal };
224
+ }
225
+ if (options.abortSignal?.aborted)
226
+ return { terminal: "aborted_streaming" };
227
+ appendSyntheticToolUseMessage(assistantMessages, toolUses);
228
+ return { output: { assistantMessages, toolUses, previousResponseId, incompleteReason } };
229
+ }
230
+ async function* handleModelEvent(event, assistantMessages, toolUses, outputFilter, thinkingParts) {
231
+ if (event.type === "assistant_delta") {
232
+ const text = outputFilter.push(event.text);
233
+ if (text)
234
+ yield { type: "assistant.delta", text };
235
+ return {};
236
+ }
237
+ if (event.type === "thinking_delta") {
238
+ thinkingParts.push(event.text);
239
+ yield { type: "thinking.delta", text: event.text };
240
+ return {};
241
+ }
242
+ if (event.type === "assistant_message") {
243
+ const message = outputFilter.sanitizeMessage(event.message);
244
+ assistantMessages.push(message);
245
+ for (const toolUse of extractToolUses(message))
246
+ toolUses.push(toolUse);
247
+ yield { type: "message", message };
248
+ return {};
249
+ }
250
+ if (event.type === "tool_use") {
251
+ toolUses.push(event.toolUse);
252
+ return {};
253
+ }
254
+ if (event.type === "tool_call_delta") {
255
+ yield {
256
+ type: "tool_call.delta",
257
+ callId: event.callId,
258
+ name: event.name,
259
+ argumentsDelta: event.argumentsDelta,
260
+ };
261
+ return {};
262
+ }
263
+ if (event.type === "usage") {
264
+ yield { type: "usage", usage: event.usage };
265
+ return {};
266
+ }
267
+ if (event.type === "retrying") {
268
+ yield { type: "retrying", attempt: event.attempt, delayMs: event.delayMs, error: event.error };
269
+ return {};
270
+ }
271
+ if (event.type === "response_completed") {
272
+ const thinkingMessage = finalizeThinkingMessage(assistantMessages, thinkingParts);
273
+ if (thinkingMessage)
274
+ yield { type: "message", message: thinkingMessage };
275
+ if (event.usage)
276
+ yield { type: "usage", usage: event.usage };
277
+ return { previousResponseId: event.responseId };
278
+ }
279
+ if (event.type === "response_incomplete") {
280
+ const thinkingMessage = finalizeThinkingMessage(assistantMessages, thinkingParts);
281
+ if (thinkingMessage)
282
+ yield { type: "message", message: thinkingMessage };
283
+ if (event.usage)
284
+ yield { type: "usage", usage: event.usage };
285
+ return { previousResponseId: event.responseId, incompleteReason: event.reason };
286
+ }
287
+ if (event.type === "error") {
288
+ yield { type: "error", error: event.error };
289
+ return {};
290
+ }
291
+ return {};
292
+ }
293
+ function finalizeThinkingMessage(assistantMessages, thinkingParts) {
294
+ const text = thinkingParts.join("").trim();
295
+ if (!text)
296
+ return undefined;
297
+ const alreadyIncluded = assistantMessages.some((message) => message.blocks.some((block) => block.type === "thinking" && (block.text === text || block.text.startsWith(text) || text.startsWith(block.text))));
298
+ thinkingParts.length = 0;
299
+ if (alreadyIncluded)
300
+ return undefined;
301
+ const message = createThinkingMessage(text);
302
+ assistantMessages.push(message);
303
+ return message;
304
+ }
305
+ async function* executeToolsForTurn(toolUses, dependencies, options, context) {
306
+ yield { type: "state", phase: "running_tools", detail: `${toolUses.length} tool call(s)` };
307
+ for (const toolUse of toolUses)
308
+ yield { type: "tool.started", toolUse };
309
+ if (options.abortSignal?.aborted)
310
+ return { terminal: "aborted_tools", messages: [], context };
311
+ const result = await runTools(toolUses, context, { canUseTool: dependencies.canUseTool });
312
+ for (const message of result.messages) {
313
+ context.appState.appendMessage(message);
314
+ yield { type: "message", message };
315
+ }
316
+ for (const toolUse of toolUses) {
317
+ const resultMessage = result.messages.find((message) => message.blocks.some((block) => block.type === "tool_result" && block.toolUseId === toolUse.id));
318
+ yield { type: "tool.finished", toolUse, ok: resultMessage ? toolResultOk(resultMessage) : false };
319
+ }
320
+ return { messages: result.messages, context: result.context };
321
+ }
322
+ function maybeRecoverWithoutTools(state, incompleteReason, assistantMessages, previousResponseId, options) {
323
+ if (incompleteReason === "max_output_tokens" && state.maxOutputTokensRecoveryCount === 0) {
324
+ return {
325
+ ...state,
326
+ messages: [...state.messages, ...assistantMessages],
327
+ modelInputMessages: [createTextMessage("user", "Continue from the exact point where the previous response stopped.")],
328
+ previousResponseId,
329
+ maxOutputTokensOverride: Math.max(options.maxOutputTokensOverride ?? 0, 64000),
330
+ maxOutputTokensRecoveryCount: state.maxOutputTokensRecoveryCount + 1,
331
+ turnCount: state.turnCount + 1,
332
+ transition: { reason: "max_output_tokens_escalate", detail: incompleteReason },
333
+ };
334
+ }
335
+ return undefined;
336
+ }
337
+ function buildNextTurnState(state, input) {
338
+ const allMessages = [...state.messages, ...input.assistantMessages, ...input.toolResults];
339
+ return {
340
+ ...state,
341
+ phase: "injecting_context",
342
+ messages: allMessages,
343
+ modelInputMessages: undefined,
344
+ previousResponseId: undefined,
345
+ turnCount: state.turnCount + 1,
346
+ maxOutputTokensRecoveryCount: 0,
347
+ hasAttemptedReactiveCompact: false,
348
+ maxOutputTokensOverride: undefined,
349
+ transition: { reason: "next_turn" },
350
+ };
351
+ }
352
+ function appendSyntheticToolUseMessage(assistantMessages, toolUses) {
353
+ const missing = toolUses.filter((toolUse) => !assistantMessages.some((message) => message.blocks.some((block) => block.type === "tool_use" && block.id === toolUse.id)));
354
+ if (!missing.length)
355
+ return;
356
+ assistantMessages.push({
357
+ id: `tool-use-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`,
358
+ role: "assistant",
359
+ createdAt: new Date().toISOString(),
360
+ blocks: missing.map((toolUse) => ({
361
+ type: "tool_use",
362
+ id: toolUse.id,
363
+ name: toolUse.name,
364
+ input: toolUse.input,
365
+ })),
366
+ isMeta: true,
367
+ metadata: { syntheticToolUse: true },
368
+ });
369
+ }
370
+ function extractToolUses(message) {
371
+ return message.blocks
372
+ .filter((block) => block.type === "tool_use")
373
+ .map((block) => ({ id: block.id, name: block.name, input: block.input }));
374
+ }
375
+ function toolResultOk(message) {
376
+ return message.blocks.every((block) => block.type !== "tool_result" || block.ok);
377
+ }
378
+ function formatCompactionDetail(compaction) {
379
+ if (compaction.reason === "microcompact" && compaction.summary)
380
+ return compaction.summary;
381
+ return `${compaction.reason ?? "compact"} reduced context by ${compaction.tokensFreed ?? 0} chars`;
382
+ }
383
+ function terminalForModelError(error) {
384
+ if (error instanceof ModelAPIError) {
385
+ if (error.category === "context_length")
386
+ return "prompt_too_long";
387
+ if (error.category === "user_abort")
388
+ return "aborted_streaming";
389
+ if (error.category === "unsupported_image_input")
390
+ return "image_error";
391
+ if (error.category === "max_output_tokens")
392
+ return "model_error";
393
+ }
394
+ return "model_error";
395
+ }
396
+ function isContextLengthError(error) {
397
+ return error instanceof ModelAPIError && error.category === "context_length";
398
+ }
399
+ function collectTaskNotifications(source) {
400
+ if (!source)
401
+ return [];
402
+ const completed = source.collectUnnotifiedCompletions();
403
+ return completed.map((task) => {
404
+ source.markNotified(task.taskId);
405
+ return createTextMessage("user", `<task-notification task_id="${task.taskId}" agent_id="${task.agentId}" status="${task.status}" type="${task.type}">\n${task.content}\n</task-notification>`);
406
+ });
407
+ }
408
+ //# sourceMappingURL=query.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"query.js","sourceRoot":"","sources":["../../src/core/query.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEvD,OAAO,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAEhE,OAAO,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AAEtE,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAEnD,OAAO,EAAE,QAAQ,EAAE,MAAM,gCAAgC,CAAC;AAG1D,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAqC,MAAM,sBAAsB,CAAC;AACnH,OAAO,EACL,mBAAmB,EACnB,qBAAqB,EACrB,uBAAuB,EACvB,+BAA+B,EAC/B,kBAAkB,GACnB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EACL,kBAAkB,EAClB,YAAY,GAGb,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AACrE,OAAO,EAAE,mBAAmB,EAAE,MAAM,sBAAsB,CAAC;AAiD3D,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,KAAK,CAC1B,QAAmB,EACnB,YAA+B,EAC/B,OAAqB;IAErB,MAAM,YAAY,GAAG,kBAAkB,CAAC,QAAQ,EAAE;QAChD,KAAK,EAAE,OAAO,CAAC,KAAK;QACpB,aAAa,EAAE,OAAO,CAAC,aAAa;KACrC,CAAC,CAAC;IACH,YAAY,CAAC,uBAAuB,GAAG,OAAO,CAAC,uBAAuB,CAAC;IAEvE,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,EAAE,OAAO,CAAC,CAAC;IACvE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC;IAC7C,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,SAAS,CACvB,YAAwB,EACxB,YAA+B,EAC/B,OAAqB;IAErB,MAAM,cAAc,GAAG,YAAY,CAAC,cAAc,IAAI,IAAI,qBAAqB,EAAE,CAAC;IAClF,MAAM,SAAS,GAAG,YAAY,CAAC,SAAS,IAAI,IAAI,oBAAoB,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;IAChG,MAAM,QAAQ,GAAG,IAAI,gBAAgB,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,YAAY,CAAC,CAAC;IAC7E,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;IAClC,IAAI,KAAK,GAAG,YAAY,CAAC;IACzB,IAAI,WAAW,GAAmB;QAChC,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,KAAK,EAAE,YAAY,CAAC,KAAK;QACzB,QAAQ;QACR,gBAAgB,EAAE,YAAY,CAAC,gBAAgB;QAC/C,yBAAyB,EAAE,YAAY,CAAC,yBAAyB;QACjE,IAAI,EAAE,GAAG,EAAE,CAAC,SAAS;KACtB,CAAC;IAEF,OAAO,IAAI,EAAE,CAAC;QACZ,IAAI,OAAO,CAAC,WAAW,EAAE,OAAO;YAAE,OAAO,mBAAmB,CAAC;QAC7D,IAAI,QAAQ,KAAK,SAAS,IAAI,KAAK,CAAC,SAAS,IAAI,QAAQ;YAAE,OAAO,WAAW,CAAC;QAE9E,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;QACzB,WAAW,GAAG,EAAE,GAAG,WAAW,EAAE,aAAa,EAAE,KAAK,CAAC,aAAa,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC;QAC/F,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,KAAK,CAAC,SAAS,GAAG,CAAC,aAAa,KAAK,CAAC,UAAU,CAAC,MAAM,GAAG,EAAE,CAAC;QAExH,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,KAAK,CAAC;YACzC,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,GAAG,EAAE,OAAO,CAAC,YAAY;YACzB,YAAY,EAAE,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;YAClF,cAAc,EAAE,WAAW;SAC5B,CAAC,CAAC;QACH,MAAM,eAAe,GAAG,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;QACpE,MAAM,YAAY,GAAG,mBAAmB,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;QACtF,MAAM,QAAQ,GAAG,MAAM,uBAAuB,CAAC,KAAK,EAAE,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE;YACtF,KAAK,EAAE,KAAK,CAAC,YAAY,IAAI,OAAO,CAAC,KAAK;YAC1C,YAAY;YACZ,eAAe;YACf,cAAc,EAAE,WAAW;SAC5B,CAAC,CAAC;QACH,IAAI,QAAQ,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC;YACjC,KAAK,GAAG,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,QAAQ,CAAC,iBAAiB,EAAE,CAAC;YAC3D,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,sBAAsB,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAClG,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,eAAe,KAAK,IAAI,CAAC,EAAE,CAAC;gBACjH,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;YACrC,CAAC;QACH,CAAC;QAED,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,QAAQ,CAAC,gBAAgB,EAAE,YAAY,EAAE,OAAO,EAAE,WAAW,EAAE;YACzH,YAAY;YACZ,eAAe;YACf,OAAO,EAAE,QAAQ,CAAC,OAAO;SAC1B,CAAC,CAAC;QACH,IAAI,WAAW,CAAC,eAAe,EAAE,CAAC;YAChC,KAAK,GAAG,WAAW,CAAC,eAAe,CAAC;YACpC,SAAS;QACX,CAAC;QACD,IAAI,WAAW,CAAC,QAAQ;YAAE,OAAO,WAAW,CAAC,QAAQ,CAAC;QACtD,IAAI,CAAC,WAAW,CAAC,MAAM;YAAE,OAAO,aAAa,CAAC;QAE9C,MAAM,EAAE,iBAAiB,EAAE,QAAQ,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,GAAG,WAAW,CAAC,MAAM,CAAC;QAEjG,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,wBAAwB,CAAC,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,kBAAkB,EAAE,OAAO,CAAC,CAAC;YACnH,IAAI,QAAQ,EAAE,CAAC;gBACb,KAAK,GAAG,QAAQ,CAAC;gBACjB,SAAS;YACX,CAAC;YACD,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,mBAAmB,CAAC,QAAQ,EAAE,YAAY,EAAE,OAAO,EAAE,WAAW,CAAC,CAAC;QAC5F,IAAI,UAAU,CAAC,QAAQ;YAAE,OAAO,UAAU,CAAC,QAAQ,CAAC;QACpD,WAAW,GAAG,UAAU,CAAC,OAAO,CAAC;QAEjC,MAAM,iBAAiB,GAAG,wBAAwB,CAAC,YAAY,CAAC,sBAAsB,CAAC,CAAC;QACxF,KAAK,MAAM,YAAY,IAAI,iBAAiB,EAAE,CAAC;YAC7C,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC;QACnD,CAAC;QAED,KAAK,GAAG,kBAAkB,CAAC,KAAK,EAAE;YAChC,iBAAiB;YACjB,WAAW,EAAE,CAAC,GAAG,UAAU,CAAC,QAAQ,EAAE,GAAG,iBAAiB,CAAC;YAC3D,kBAAkB;SACnB,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,SAAS,SAAS,CAAC,KAAiB;IAClC,OAAO;QACL,GAAG,KAAK;QACR,KAAK,EAAE,WAAW;QAClB,aAAa,EAAE,YAAY,CAAC,KAAK,CAAC,aAAa,CAAC;KACjD,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,uBAAuB,CACpC,KAAiB,EACjB,OAAuB,EACvB,YAA+B,EAC/B,SAAoB,EACpB,SAA8I;IAE9I,MAAM,YAAY,GAAG,KAAK,CAAC,kBAAkB,IAAI,+BAA+B,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACjG,MAAM,YAAY,GAAG,SAAS,CAAC,cAAc,EAAE,gBAAgB;QAC7D,CAAC,CAAC,MAAM,SAAS,CAAC,cAAc,CAAC,gBAAgB,CAAC,WAAW,CAAC,YAAY,EAAE;YACxE,mBAAmB,EAAE,YAAY,CAAC,6BAA6B;SAChE,CAAC;QACJ,CAAC,CAAC;YACE,QAAQ,EAAE,qBAAqB,CAAC,YAAY,EAAE;gBAC5C,mBAAmB,EAAE,YAAY,CAAC,6BAA6B;aAChE,CAAC;YACF,OAAO,EAAE,EAAE;SACZ,CAAC;IACN,IAAI,YAAY,CAAC,OAAO,CAAC,MAAM;QAAE,SAAS,CAAC,cAAc,EAAE,yBAAyB,EAAE,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IAC7G,MAAM,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAC;IACvC,MAAM,cAAc,GAAG,uBAAuB,CAAC,QAAQ,CAAC,CAAC;IACzD,MAAM,oBAAoB,GAAG,mBAAmB,CAAC;QAC/C,KAAK,EAAE,SAAS,CAAC,KAAK;QACtB,QAAQ,EAAE,kBAAkB,CAAC,cAAc,EAAE,OAAO,CAAC,WAAW,CAAC;QACjE,YAAY,EAAE,SAAS,CAAC,YAAY;QACpC,KAAK,EAAE,SAAS,CAAC,eAAe;KACjC,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,cAAc,EAAE;QACzD,GAAG,YAAY,CAAC,aAAa;QAC7B,oBAAoB,EAAE,oBAAoB,CAAC,oBAAoB;QAC/D,mBAAmB,EAAE,oBAAoB,CAAC,mBAAmB;KAC9D,CAAC,CAAC;IACH,MAAM,iBAAiB,GAAG,uBAAuB,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACvE,MAAM,gBAAgB,GAAG,kBAAkB,CAAC,iBAAiB,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IACpF,MAAM,OAAO,GAAG,mBAAmB,CAAC;QAClC,KAAK,EAAE,SAAS,CAAC,KAAK;QACtB,QAAQ,EAAE,gBAAgB;QAC1B,YAAY,EAAE,SAAS,CAAC,YAAY;QACpC,KAAK,EAAE,SAAS,CAAC,eAAe;KACjC,CAAC,CAAC;IACH,OAAO;QACL,gBAAgB;QAChB,iBAAiB;QACjB,UAAU,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;QACvD,OAAO;KACR,CAAC;AACJ,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,gBAAgB,CAC9B,KAAiB,EACjB,OAAuB,EACvB,gBAA2B,EAC3B,YAA+B,EAC/B,OAAqB,EACrB,WAA2B,EAC3B,SAA8I;IAE9I,MAAM,iBAAiB,GAAc,EAAE,CAAC;IACxC,MAAM,QAAQ,GAAqB,EAAE,CAAC;IACtC,MAAM,YAAY,GAAG,IAAI,qBAAqB,EAAE,CAAC;IACjD,MAAM,aAAa,GAAa,EAAE,CAAC;IACnC,IAAI,kBAAkB,GAAG,KAAK,CAAC,kBAAkB,CAAC;IAClD,IAAI,gBAAoC,CAAC;IACzC,IAAI,WAAW,GAAG,KAAK,CAAC,YAAY,IAAI,OAAO,CAAC,KAAK,CAAC;IAEtD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,6BAA6B,EAAE,CAAC;IACpF,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,SAAS,CAAC,OAAO,EAAE,CAAC;IAC9D,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,eAAe,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,wBAAwB,WAAW,GAAG,CAAC,CAAC,CAAC,qBAAqB,EAAE,CAAC;IAEtI,IAAI,CAAC;QACH,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,YAAY,CAAC,YAAY,CAAC,MAAM,CAAC;YACzD,KAAK,EAAE,WAAW;YAClB,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,OAAO,CAAC,aAAa;YAC3D,QAAQ,EAAE,gBAAgB;YAC1B,YAAY,EAAE,SAAS,CAAC,YAAY;YACpC,KAAK,EAAE,SAAS,CAAC,eAAe;YAChC,MAAM,EAAE,IAAI;YACZ,eAAe,EAAE,KAAK,CAAC,uBAAuB,IAAI,OAAO,CAAC,uBAAuB;YACjF,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,kBAAkB,EAAE,KAAK,CAAC,kBAAkB;YAC5C,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,YAAY,EAAE,OAAO,CAAC,WAAW;SAClC,CAAC,EAAE,CAAC;YACH,IAAI,OAAO,CAAC,WAAW,EAAE,OAAO;gBAAE,OAAO,EAAE,QAAQ,EAAE,mBAAmB,EAAE,CAAC;YAC3E,MAAM,OAAO,GAAG,KAAK,CAAC,CAAC,gBAAgB,CAAC,KAAK,EAAE,iBAAiB,EAAE,QAAQ,EAAE,YAAY,EAAE,aAAa,CAAC,CAAC;YACzG,kBAAkB,GAAG,OAAO,CAAC,kBAAkB,IAAI,kBAAkB,CAAC;YACtE,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,gBAAgB,CAAC;YAEhE,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;gBACtC,WAAW,GAAG,KAAK,CAAC,OAAO,CAAC;gBAC5B,iBAAiB,CAAC,MAAM,GAAG,CAAC,CAAC;gBAC7B,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;gBACpB,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,CAAC,UAAU,EAAE,2BAA2B,KAAK,CAAC,SAAS,OAAO,KAAK,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YACtI,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,2BAA2B,EAAE,CAAC;YACtE,MAAM,SAAS,GAAG,YAAY,CAAC,SAAS,IAAI,IAAI,oBAAoB,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC;YAChG,MAAM,UAAU,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YAC7E,MAAM,eAAe,GAAG,mBAAmB,CAAC;gBAC1C,KAAK,EAAE,WAAW;gBAClB,QAAQ,EAAE,KAAK,CAAC,QAAQ;gBACxB,YAAY,EAAE,SAAS,CAAC,YAAY;gBACpC,KAAK,EAAE,SAAS,CAAC,eAAe;aACjC,CAAC,CAAC;YACH,MAAM,cAAc,GAAyB;gBAC3C,GAAG,YAAY,CAAC,aAAa;gBAC7B,oBAAoB,EAAE,eAAe,CAAC,oBAAoB;gBAC1D,mBAAmB,EAAE,eAAe,CAAC,mBAAmB;aACzD,CAAC;YACF,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC,KAAK,CAAC,QAAQ,EAAE,UAAU,EAAE,cAAc,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC;YACvJ,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,MAAM,EAAE,8CAA8C,EAAE,CAAC;YACrG,KAAK,MAAM,OAAO,IAAI,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,QAAQ,EAAE,eAAe,KAAK,IAAI,CAAC,EAAE,CAAC;gBACzG,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;YACrC,CAAC;YACD,OAAO;gBACL,eAAe,EAAE;oBACf,GAAG,KAAK;oBACR,QAAQ,EAAE,SAAS,CAAC,QAAQ;oBAC5B,kBAAkB,EAAE,SAAS,CAAC,QAAQ;oBACtC,2BAA2B,EAAE,IAAI;oBACjC,SAAS,EAAE,KAAK,CAAC,SAAS,GAAG,CAAC;oBAC9B,UAAU,EAAE,EAAE,MAAM,EAAE,wBAAwB,EAAE,MAAM,EAAE,UAAU,CAAC,OAAO,EAAE;iBAC7E;aACF,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;QAC9C,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;QAC1F,OAAO,EAAE,QAAQ,EAAE,CAAC;IACtB,CAAC;IAED,IAAI,OAAO,CAAC,WAAW,EAAE,OAAO;QAAE,OAAO,EAAE,QAAQ,EAAE,mBAAmB,EAAE,CAAC;IAE3E,6BAA6B,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC;IAC3D,OAAO,EAAE,MAAM,EAAE,EAAE,iBAAiB,EAAE,QAAQ,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,EAAE,CAAC;AAC3F,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,gBAAgB,CAC9B,KAAuB,EACvB,iBAA4B,EAC5B,QAA0B,EAC1B,YAAmC,EACnC,aAAuB;IAEvB,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;QACrC,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3C,IAAI,IAAI;YAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC;QAClD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;QACpC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC/B,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;QACnD,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,YAAY,CAAC,eAAe,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC5D,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChC,KAAK,MAAM,OAAO,IAAI,eAAe,CAAC,OAAO,CAAC;YAAE,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;QACnC,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAC9B,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC7B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;QACrC,MAAM;YACJ,IAAI,EAAE,iBAAiB;YACvB,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,cAAc,EAAE,KAAK,CAAC,cAAc;SACrC,CAAC;QACF,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC3B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;QAC5C,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;QAC9B,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;QAC/F,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;QACxC,MAAM,eAAe,GAAG,uBAAuB,CAAC,iBAAiB,EAAE,aAAa,CAAC,CAAC;QAClF,IAAI,eAAe;YAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;QACzE,IAAI,KAAK,CAAC,KAAK;YAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;QAC7D,OAAO,EAAE,kBAAkB,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC;IAClD,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,qBAAqB,EAAE,CAAC;QACzC,MAAM,eAAe,GAAG,uBAAuB,CAAC,iBAAiB,EAAE,aAAa,CAAC,CAAC;QAClF,IAAI,eAAe;YAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;QACzE,IAAI,KAAK,CAAC,KAAK;YAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;QAC7D,OAAO,EAAE,kBAAkB,EAAE,KAAK,CAAC,UAAU,EAAE,gBAAgB,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC;IAClF,CAAC;IAED,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QAC3B,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC;QAC5C,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,uBAAuB,CAAC,iBAA4B,EAAE,aAAuB;IACpF,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;IAC3C,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5B,MAAM,eAAe,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CACzD,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CACjJ,CAAC;IACF,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC;IACzB,IAAI,eAAe;QAAE,OAAO,SAAS,CAAC;IACtC,MAAM,OAAO,GAAG,qBAAqB,CAAC,IAAI,CAAC,CAAC;IAC5C,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAChC,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,mBAAmB,CACjC,QAA0B,EAC1B,YAA+B,EAC/B,OAAqB,EACrB,OAAuB;IAEvB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,eAAe,EAAE,MAAM,EAAE,GAAG,QAAQ,CAAC,MAAM,eAAe,EAAE,CAAC;IAC3F,KAAK,MAAM,OAAO,IAAI,QAAQ;QAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,CAAC;IAExE,IAAI,OAAO,CAAC,WAAW,EAAE,OAAO;QAAE,OAAO,EAAE,QAAQ,EAAE,eAAe,EAAE,QAAQ,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC;IAE9F,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,YAAY,CAAC,UAAU,EAAE,CAAC,CAAC;IAE1F,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACtC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QACxC,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;IACrC,CAAC;IAED,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CACrD,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,SAAS,KAAK,OAAO,CAAC,EAAE,CAAC,CAC/F,CAAC;QACF,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,EAAE,EAAE,aAAa,CAAC,CAAC,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;IACpG,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;AAChE,CAAC;AAED,SAAS,wBAAwB,CAC/B,KAAiB,EACjB,gBAAoC,EACpC,iBAA4B,EAC5B,kBAAsC,EACtC,OAAqB;IAErB,IAAI,gBAAgB,KAAK,mBAAmB,IAAI,KAAK,CAAC,4BAA4B,KAAK,CAAC,EAAE,CAAC;QACzF,OAAO;YACL,GAAG,KAAK;YACR,QAAQ,EAAE,CAAC,GAAG,KAAK,CAAC,QAAQ,EAAE,GAAG,iBAAiB,CAAC;YACnD,kBAAkB,EAAE,CAAC,iBAAiB,CAAC,MAAM,EAAE,oEAAoE,CAAC,CAAC;YACrH,kBAAkB;YAClB,uBAAuB,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,uBAAuB,IAAI,CAAC,EAAE,KAAK,CAAC;YAC9E,4BAA4B,EAAE,KAAK,CAAC,4BAA4B,GAAG,CAAC;YACpE,SAAS,EAAE,KAAK,CAAC,SAAS,GAAG,CAAC;YAC9B,UAAU,EAAE,EAAE,MAAM,EAAE,4BAA4B,EAAE,MAAM,EAAE,gBAAgB,EAAE;SAC/E,CAAC;IACJ,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,kBAAkB,CACzB,KAAiB,EACjB,KAIC;IAED,MAAM,WAAW,GAAG,CAAC,GAAG,KAAK,CAAC,QAAQ,EAAE,GAAG,KAAK,CAAC,iBAAiB,EAAE,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC;IAC1F,OAAO;QACL,GAAG,KAAK;QACR,KAAK,EAAE,mBAAmB;QAC1B,QAAQ,EAAE,WAAW;QACrB,kBAAkB,EAAE,SAAS;QAC7B,kBAAkB,EAAE,SAAS;QAC7B,SAAS,EAAE,KAAK,CAAC,SAAS,GAAG,CAAC;QAC9B,4BAA4B,EAAE,CAAC;QAC/B,2BAA2B,EAAE,KAAK;QAClC,uBAAuB,EAAE,SAAS;QAClC,UAAU,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE;KACpC,CAAC;AACJ,CAAC;AAED,SAAS,6BAA6B,CAAC,iBAA4B,EAAE,QAA0B;IAC7F,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAC1C,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAClC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,EAAE,KAAK,OAAO,CAAC,EAAE,CAAC,CACrF,CACF,CAAC;IACF,IAAI,CAAC,OAAO,CAAC,MAAM;QAAE,OAAO;IAE5B,iBAAiB,CAAC,IAAI,CAAC;QACrB,EAAE,EAAE,YAAY,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE;QACnF,IAAI,EAAE,WAAW;QACjB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAChC,IAAI,EAAE,UAAU;YAChB,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,KAAK,EAAE,OAAO,CAAC,KAAK;SACrB,CAAC,CAAC;QACH,MAAM,EAAE,IAAI;QACZ,QAAQ,EAAE,EAAE,gBAAgB,EAAE,IAAI,EAAE;KACrC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,eAAe,CAAC,OAAgB;IACvC,OAAO,OAAO,CAAC,MAAM;SAClB,MAAM,CAAC,CAAC,KAAK,EAA2E,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC;SACrH,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,YAAY,CAAC,OAAgB;IACpC,OAAO,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,EAAE,CAAC,CAAC;AACnF,CAAC;AAED,SAAS,sBAAsB,CAAC,UAA4B;IAC1D,IAAI,UAAU,CAAC,MAAM,KAAK,cAAc,IAAI,UAAU,CAAC,OAAO;QAAE,OAAO,UAAU,CAAC,OAAO,CAAC;IAC1F,OAAO,GAAG,UAAU,CAAC,MAAM,IAAI,SAAS,uBAAuB,UAAU,CAAC,WAAW,IAAI,CAAC,QAAQ,CAAC;AACrG,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAc;IAC3C,IAAI,KAAK,YAAY,aAAa,EAAE,CAAC;QACnC,IAAI,KAAK,CAAC,QAAQ,KAAK,gBAAgB;YAAE,OAAO,iBAAiB,CAAC;QAClE,IAAI,KAAK,CAAC,QAAQ,KAAK,YAAY;YAAE,OAAO,mBAAmB,CAAC;QAChE,IAAI,KAAK,CAAC,QAAQ,KAAK,yBAAyB;YAAE,OAAO,aAAa,CAAC;QACvE,IAAI,KAAK,CAAC,QAAQ,KAAK,mBAAmB;YAAE,OAAO,aAAa,CAAC;IACnE,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAc;IAC1C,OAAO,KAAK,YAAY,aAAa,IAAI,KAAK,CAAC,QAAQ,KAAK,gBAAgB,CAAC;AAC/E,CAAC;AAED,SAAS,wBAAwB,CAAC,MAA+B;IAC/D,IAAI,CAAC,MAAM;QAAE,OAAO,EAAE,CAAC;IACvB,MAAM,SAAS,GAAG,MAAM,CAAC,4BAA4B,EAAE,CAAC;IACxD,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QAC5B,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjC,OAAO,iBAAiB,CACtB,MAAM,EACN,+BAA+B,IAAI,CAAC,MAAM,eAAe,IAAI,CAAC,OAAO,aAAa,IAAI,CAAC,MAAM,WAAW,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,OAAO,wBAAwB,CAC7J,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC"}
@@ -0,0 +1,48 @@
1
+ import { type ContextManager } from "../context/context-manager.js";
2
+ import type { Compactor, ContextBudgetOptions } from "../context/compaction.js";
3
+ import type { ModelGateway, ModelUsage } from "../model/model-gateway.js";
4
+ import { ToolRegistry } from "../tools/registry.js";
5
+ import type { CanUseTool, ToolUseContext } from "../tools/tool.js";
6
+ import type { AgentEvent } from "../types/events.js";
7
+ import { type Message } from "../types/messages.js";
8
+ import type { AgentDefinition } from "../agents/agent-definition.js";
9
+ import type { AgentToolResult } from "../agents/local-agent-task.js";
10
+ export interface RunAgentDependencies {
11
+ modelGateway: ModelGateway;
12
+ tools: ToolRegistry;
13
+ contextManager?: ContextManager;
14
+ compactor?: Compactor;
15
+ contextBudget?: ContextBudgetOptions;
16
+ canUseTool?: CanUseTool;
17
+ maxToolResultSerializedLength?: number;
18
+ }
19
+ export interface RunAgentOptions {
20
+ agentId: string;
21
+ agent: AgentDefinition;
22
+ prompt: string;
23
+ parentContext?: ToolUseContext;
24
+ parentMessages?: readonly Message[];
25
+ dependencies: RunAgentDependencies;
26
+ model?: string;
27
+ fallbackModel?: string;
28
+ maxTurns?: number;
29
+ abortSignal?: AbortSignal;
30
+ fork?: boolean;
31
+ existingMessages?: Message[];
32
+ /** Resolved absolute cwd for file/exec tools in this subagent session. */
33
+ workspaceCwd?: string;
34
+ }
35
+ export interface RunAgentCompleted {
36
+ result: AgentToolResult;
37
+ messages: Message[];
38
+ terminalReason?: string;
39
+ }
40
+ export declare function runAgent(options: RunAgentOptions): AsyncGenerator<AgentEvent, RunAgentCompleted, void>;
41
+ export declare function resolveAgentTools(parentTools: ToolRegistry, agent: AgentDefinition): ToolRegistry;
42
+ export declare function finalizeAgentTool(input: {
43
+ agentId: string;
44
+ agentType: string;
45
+ messages: readonly Message[];
46
+ durationMs: number;
47
+ usage?: ModelUsage;
48
+ }): AgentToolResult;
@@ -0,0 +1,150 @@
1
+ import { DefaultContextManager } from "../context/context-manager.js";
2
+ import { ToolRegistry } from "../tools/registry.js";
3
+ import { createTextMessage } from "../types/messages.js";
4
+ import { buildForkChildPrompt, FORK_AGENT } from "../agents/agent-definition.js";
5
+ import { query } from "./query.js";
6
+ export async function* runAgent(options) {
7
+ const startedAt = Date.now();
8
+ const messages = options.existingMessages?.length
9
+ ? [...options.existingMessages, ...buildResumeMessages(options)]
10
+ : buildInitialAgentMessages(options);
11
+ const agentMessages = [];
12
+ let terminalReason;
13
+ let lastUsage;
14
+ const dependencies = {
15
+ modelGateway: options.dependencies.modelGateway,
16
+ tools: resolveAgentTools(options.dependencies.tools, options.agent),
17
+ contextManager: createAgentContextManager(options),
18
+ compactor: options.dependencies.compactor,
19
+ contextBudget: options.dependencies.contextBudget,
20
+ canUseTool: options.dependencies.canUseTool,
21
+ maxToolResultSerializedLength: options.dependencies.maxToolResultSerializedLength,
22
+ };
23
+ for await (const event of query(messages, dependencies, {
24
+ agentId: options.agentId,
25
+ model: resolveAgentModel(options.agent, options.model),
26
+ fallbackModel: options.fallbackModel,
27
+ maxTurns: resolveSubagentMaxTurns(options),
28
+ queryOrigin: "subagent",
29
+ abortSignal: options.abortSignal,
30
+ workspaceCwd: options.workspaceCwd,
31
+ })) {
32
+ if (event.type === "message")
33
+ agentMessages.push(event.message);
34
+ if (event.type === "usage")
35
+ lastUsage = event.usage;
36
+ if (event.type === "terminal")
37
+ terminalReason = event.reason;
38
+ yield event;
39
+ }
40
+ return {
41
+ result: finalizeAgentTool({
42
+ agentId: options.agentId,
43
+ agentType: options.agent.agentType,
44
+ messages: agentMessages,
45
+ durationMs: Date.now() - startedAt,
46
+ usage: lastUsage,
47
+ }),
48
+ messages: agentMessages,
49
+ terminalReason,
50
+ };
51
+ }
52
+ export function resolveAgentTools(parentTools, agent) {
53
+ const registry = new ToolRegistry();
54
+ const allowed = agent.tools;
55
+ const disallowed = new Set(agent.disallowedTools ?? []);
56
+ for (const tool of parentTools.list(undefined, { includeDeferred: true })) {
57
+ if (disallowed.has(tool.name))
58
+ continue;
59
+ if (allowed && !allowed.includes("*") && !allowed.includes(tool.name))
60
+ continue;
61
+ registry.register(tool);
62
+ }
63
+ return registry;
64
+ }
65
+ export function finalizeAgentTool(input) {
66
+ return {
67
+ agent_id: input.agentId,
68
+ agent_type: input.agentType,
69
+ content: extractFinalText(input.messages),
70
+ total_duration_ms: input.durationMs,
71
+ total_tokens: input.usage?.totalTokens,
72
+ total_tool_use_count: countToolUses(input.messages),
73
+ usage: input.usage,
74
+ };
75
+ }
76
+ function buildInitialAgentMessages(options) {
77
+ const prompt = options.fork ? buildForkChildPrompt(options.prompt) : options.prompt;
78
+ const messages = [];
79
+ if (options.fork && options.parentMessages?.length) {
80
+ messages.push(...options.parentMessages.map(cloneMessage));
81
+ }
82
+ if (options.agent.initialPrompt)
83
+ messages.push(createTextMessage("user", options.agent.initialPrompt));
84
+ messages.push(createTextMessage("user", prompt));
85
+ return messages;
86
+ }
87
+ function buildResumeMessages(options) {
88
+ return [createTextMessage("user", `[Resumed] ${options.prompt}`)];
89
+ }
90
+ function createAgentContextManager(options) {
91
+ const parent = options.dependencies.contextManager;
92
+ const inheritedCwd = options.workspaceCwd ?? options.parentContext?.appState.snapshot().cwd;
93
+ if (!parent)
94
+ return new DefaultContextManager({ cwd: inheritedCwd });
95
+ return {
96
+ async build(input) {
97
+ const runtime = await parent.build({
98
+ ...input,
99
+ cwd: options.workspaceCwd ?? input.cwd,
100
+ omitProjectMemory: options.agent.omitProjectMemory ?? input.omitProjectMemory,
101
+ agentPrompt: options.agent.buildSystemPrompt?.(options.parentContext),
102
+ agentPromptMode: options.agent.agentType === FORK_AGENT.agentType ? "proactive_append" : "replace",
103
+ });
104
+ return runtime;
105
+ },
106
+ };
107
+ }
108
+ function resolveSubagentMaxTurns(options) {
109
+ if (options.maxTurns !== undefined)
110
+ return options.maxTurns;
111
+ const raw = process.env.AGENT_SUBAGENT_MAX_TURNS;
112
+ if (raw !== undefined && raw !== "") {
113
+ const n = Number(raw);
114
+ if (Number.isFinite(n) && n >= 1)
115
+ return Math.floor(n);
116
+ }
117
+ return options.agent.maxTurns;
118
+ }
119
+ function resolveAgentModel(agent, override) {
120
+ if (override)
121
+ return override;
122
+ if (!agent.model || agent.model === "inherit")
123
+ return undefined;
124
+ return agent.model;
125
+ }
126
+ function extractFinalText(messages) {
127
+ for (let index = messages.length - 1; index >= 0; index -= 1) {
128
+ if (messages[index].role !== "assistant")
129
+ continue;
130
+ const text = messages[index].blocks
131
+ .filter((block) => block.type === "text")
132
+ .map((block) => block.text)
133
+ .join("\n")
134
+ .trim();
135
+ if (text)
136
+ return text;
137
+ }
138
+ return "";
139
+ }
140
+ function countToolUses(messages) {
141
+ return messages.reduce((total, message) => total + message.blocks.filter((block) => block.type === "tool_use").length, 0);
142
+ }
143
+ function cloneMessage(message) {
144
+ return {
145
+ ...message,
146
+ blocks: message.blocks.map((block) => ({ ...block })),
147
+ metadata: message.metadata ? { ...message.metadata } : undefined,
148
+ };
149
+ }
150
+ //# sourceMappingURL=run-agent.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"run-agent.js","sourceRoot":"","sources":["../../src/core/run-agent.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAuB,MAAM,+BAA+B,CAAC;AAG3F,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAGpD,OAAO,EAAE,iBAAiB,EAAgB,MAAM,sBAAsB,CAAC;AAEvE,OAAO,EAAE,oBAAoB,EAAE,UAAU,EAAE,MAAM,+BAA+B,CAAC;AAEjF,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAmCnC,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,QAAQ,CAAC,OAAwB;IACtD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,QAAQ,GAAG,OAAO,CAAC,gBAAgB,EAAE,MAAM;QAC/C,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,gBAAgB,EAAE,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAChE,CAAC,CAAC,yBAAyB,CAAC,OAAO,CAAC,CAAC;IACvC,MAAM,aAAa,GAAc,EAAE,CAAC;IACpC,IAAI,cAAkC,CAAC;IACvC,IAAI,SAAiC,CAAC;IAEtC,MAAM,YAAY,GAAG;QACnB,YAAY,EAAE,OAAO,CAAC,YAAY,CAAC,YAAY;QAC/C,KAAK,EAAE,iBAAiB,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC;QACnE,cAAc,EAAE,yBAAyB,CAAC,OAAO,CAAC;QAClD,SAAS,EAAE,OAAO,CAAC,YAAY,CAAC,SAAS;QACzC,aAAa,EAAE,OAAO,CAAC,YAAY,CAAC,aAAa;QACjD,UAAU,EAAE,OAAO,CAAC,YAAY,CAAC,UAAU;QAC3C,6BAA6B,EAAE,OAAO,CAAC,YAAY,CAAC,6BAA6B;KAClF,CAAC;IAEF,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,KAAK,CAAC,QAAQ,EAAE,YAAY,EAAE;QACtD,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,KAAK,EAAE,iBAAiB,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC;QACtD,aAAa,EAAE,OAAO,CAAC,aAAa;QACpC,QAAQ,EAAE,uBAAuB,CAAC,OAAO,CAAC;QAC1C,WAAW,EAAE,UAAU;QACvB,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,YAAY,EAAE,OAAO,CAAC,YAAY;KACnC,CAAC,EAAE,CAAC;QACH,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS;YAAE,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAChE,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;YAAE,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC;QACpD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;YAAE,cAAc,GAAG,KAAK,CAAC,MAAM,CAAC;QAC7D,MAAM,KAAK,CAAC;IACd,CAAC;IAED,OAAO;QACL,MAAM,EAAE,iBAAiB,CAAC;YACxB,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,SAAS,EAAE,OAAO,CAAC,KAAK,CAAC,SAAS;YAClC,QAAQ,EAAE,aAAa;YACvB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;YAClC,KAAK,EAAE,SAAS;SACjB,CAAC;QACF,QAAQ,EAAE,aAAa;QACvB,cAAc;KACf,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,WAAyB,EAAE,KAAsB;IACjF,MAAM,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;IACpC,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC;IAC5B,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC;IAExD,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,IAAI,CAAC,SAAS,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAW,EAAE,CAAC;QACpF,IAAI,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;YAAE,SAAS;QACxC,IAAI,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;YAAE,SAAS;QAChF,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,KAMjC;IACC,OAAO;QACL,QAAQ,EAAE,KAAK,CAAC,OAAO;QACvB,UAAU,EAAE,KAAK,CAAC,SAAS;QAC3B,OAAO,EAAE,gBAAgB,CAAC,KAAK,CAAC,QAAQ,CAAC;QACzC,iBAAiB,EAAE,KAAK,CAAC,UAAU;QACnC,YAAY,EAAE,KAAK,CAAC,KAAK,EAAE,WAAW;QACtC,oBAAoB,EAAE,aAAa,CAAC,KAAK,CAAC,QAAQ,CAAC;QACnD,KAAK,EAAE,KAAK,CAAC,KAAK;KACnB,CAAC;AACJ,CAAC;AAED,SAAS,yBAAyB,CAAC,OAAwB;IACzD,MAAM,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;IACpF,MAAM,QAAQ,GAAc,EAAE,CAAC;IAC/B,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,cAAc,EAAE,MAAM,EAAE,CAAC;QACnD,QAAQ,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC;IAC7D,CAAC;IACD,IAAI,OAAO,CAAC,KAAK,CAAC,aAAa;QAAE,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC;IACvG,QAAQ,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IACjD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,mBAAmB,CAAC,OAAwB;IACnD,OAAO,CAAC,iBAAiB,CAAC,MAAM,EAAE,aAAa,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;AACpE,CAAC;AAED,SAAS,yBAAyB,CAAC,OAAwB;IACzD,MAAM,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,cAAc,CAAC;IACnD,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,aAAa,EAAE,QAAQ,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC;IAC5F,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,qBAAqB,CAAC,EAAE,GAAG,EAAE,YAAY,EAAE,CAAC,CAAC;IAErE,OAAO;QACL,KAAK,CAAC,KAAK,CAAC,KAAK;YACf,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;gBACjC,GAAG,KAAK;gBACR,GAAG,EAAE,OAAO,CAAC,YAAY,IAAI,KAAK,CAAC,GAAG;gBACtC,iBAAiB,EAAE,OAAO,CAAC,KAAK,CAAC,iBAAiB,IAAI,KAAK,CAAC,iBAAiB;gBAC7E,WAAW,EAAE,OAAO,CAAC,KAAK,CAAC,iBAAiB,EAAE,CAAC,OAAO,CAAC,aAAa,CAAC;gBACrE,eAAe,EAAE,OAAO,CAAC,KAAK,CAAC,SAAS,KAAK,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,SAAS;aACnG,CAAC,CAAC;YACH,OAAO,OAAO,CAAC;QACjB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,OAAwB;IACvD,IAAI,OAAO,CAAC,QAAQ,KAAK,SAAS;QAAE,OAAO,OAAO,CAAC,QAAQ,CAAC;IAC5D,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC;IACjD,IAAI,GAAG,KAAK,SAAS,IAAI,GAAG,KAAK,EAAE,EAAE,CAAC;QACpC,MAAM,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QACtB,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC;IACD,OAAO,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC;AAChC,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAsB,EAAE,QAAiB;IAClE,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC9B,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IAChE,OAAO,KAAK,CAAC,KAAK,CAAC;AACrB,CAAC;AAED,SAAS,gBAAgB,CAAC,QAA4B;IACpD,KAAK,IAAI,KAAK,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QAC7D,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,WAAW;YAAE,SAAS;QACnD,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,MAAM;aAChC,MAAM,CAAC,CAAC,KAAK,EAA2C,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,MAAM,CAAC;aACjF,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC;aAC1B,IAAI,CAAC,IAAI,CAAC;aACV,IAAI,EAAE,CAAC;QACV,IAAI,IAAI;YAAE,OAAO,IAAI,CAAC;IACxB,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,aAAa,CAAC,QAA4B;IACjD,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;AAC5H,CAAC;AAED,SAAS,YAAY,CAAC,OAAgB;IACpC,OAAO;QACL,GAAG,OAAO;QACV,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;QACrD,QAAQ,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,SAAS;KACjE,CAAC;AACJ,CAAC"}
@@ -0,0 +1 @@
1
+ export {};