@arden-hq/loop 0.0.0-stage → 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 (223) hide show
  1. package/LICENSE +202 -0
  2. package/README.md +121 -2
  3. package/dist/abort.d.ts +9 -0
  4. package/dist/abort.d.ts.map +1 -0
  5. package/dist/abort.js +31 -0
  6. package/dist/abort.js.map +1 -0
  7. package/dist/agent/agent.d.ts +48 -0
  8. package/dist/agent/agent.d.ts.map +1 -0
  9. package/dist/agent/agent.js +146 -0
  10. package/dist/agent/agent.js.map +1 -0
  11. package/dist/agent/background-tools.d.ts +51 -0
  12. package/dist/agent/background-tools.d.ts.map +1 -0
  13. package/dist/agent/background-tools.js +257 -0
  14. package/dist/agent/background-tools.js.map +1 -0
  15. package/dist/agent/budget.d.ts +5 -0
  16. package/dist/agent/budget.d.ts.map +1 -0
  17. package/dist/agent/budget.js +48 -0
  18. package/dist/agent/budget.js.map +1 -0
  19. package/dist/agent/config.d.ts +61 -0
  20. package/dist/agent/config.d.ts.map +1 -0
  21. package/dist/agent/config.js +99 -0
  22. package/dist/agent/config.js.map +1 -0
  23. package/dist/agent/guards.d.ts +10 -0
  24. package/dist/agent/guards.d.ts.map +1 -0
  25. package/dist/agent/guards.js +136 -0
  26. package/dist/agent/guards.js.map +1 -0
  27. package/dist/agent/history-journal.d.ts +18 -0
  28. package/dist/agent/history-journal.d.ts.map +1 -0
  29. package/dist/agent/history-journal.js +87 -0
  30. package/dist/agent/history-journal.js.map +1 -0
  31. package/dist/agent/history-projection.d.ts +6 -0
  32. package/dist/agent/history-projection.d.ts.map +1 -0
  33. package/dist/agent/history-projection.js +62 -0
  34. package/dist/agent/history-projection.js.map +1 -0
  35. package/dist/agent/interactions.d.ts +14 -0
  36. package/dist/agent/interactions.d.ts.map +1 -0
  37. package/dist/agent/interactions.js +85 -0
  38. package/dist/agent/interactions.js.map +1 -0
  39. package/dist/agent/middleware.d.ts +5 -0
  40. package/dist/agent/middleware.d.ts.map +1 -0
  41. package/dist/agent/middleware.js +55 -0
  42. package/dist/agent/middleware.js.map +1 -0
  43. package/dist/agent/model-output-projection.d.ts +27 -0
  44. package/dist/agent/model-output-projection.d.ts.map +1 -0
  45. package/dist/agent/model-output-projection.js +138 -0
  46. package/dist/agent/model-output-projection.js.map +1 -0
  47. package/dist/agent/model-output.d.ts +19 -0
  48. package/dist/agent/model-output.d.ts.map +1 -0
  49. package/dist/agent/model-output.js +172 -0
  50. package/dist/agent/model-output.js.map +1 -0
  51. package/dist/agent/model-turn.d.ts +29 -0
  52. package/dist/agent/model-turn.d.ts.map +1 -0
  53. package/dist/agent/model-turn.js +401 -0
  54. package/dist/agent/model-turn.js.map +1 -0
  55. package/dist/agent/policy.d.ts +28 -0
  56. package/dist/agent/policy.d.ts.map +1 -0
  57. package/dist/agent/policy.js +109 -0
  58. package/dist/agent/policy.js.map +1 -0
  59. package/dist/agent/provider-capabilities.d.ts +12 -0
  60. package/dist/agent/provider-capabilities.d.ts.map +1 -0
  61. package/dist/agent/provider-capabilities.js +76 -0
  62. package/dist/agent/provider-capabilities.js.map +1 -0
  63. package/dist/agent/recovery.d.ts +8 -0
  64. package/dist/agent/recovery.d.ts.map +1 -0
  65. package/dist/agent/recovery.js +21 -0
  66. package/dist/agent/recovery.js.map +1 -0
  67. package/dist/agent/replies.d.ts +3 -0
  68. package/dist/agent/replies.d.ts.map +1 -0
  69. package/dist/agent/replies.js +24 -0
  70. package/dist/agent/replies.js.map +1 -0
  71. package/dist/agent/run-state.d.ts +98 -0
  72. package/dist/agent/run-state.d.ts.map +1 -0
  73. package/dist/agent/run-state.js +454 -0
  74. package/dist/agent/run-state.js.map +1 -0
  75. package/dist/agent/run.d.ts +6 -0
  76. package/dist/agent/run.d.ts.map +1 -0
  77. package/dist/agent/run.js +462 -0
  78. package/dist/agent/run.js.map +1 -0
  79. package/dist/agent/streamed-call-history.d.ts +20 -0
  80. package/dist/agent/streamed-call-history.d.ts.map +1 -0
  81. package/dist/agent/streamed-call-history.js +71 -0
  82. package/dist/agent/streamed-call-history.js.map +1 -0
  83. package/dist/agent/streaming-tools.d.ts +23 -0
  84. package/dist/agent/streaming-tools.d.ts.map +1 -0
  85. package/dist/agent/streaming-tools.js +67 -0
  86. package/dist/agent/streaming-tools.js.map +1 -0
  87. package/dist/agent/structured-output.d.ts +27 -0
  88. package/dist/agent/structured-output.d.ts.map +1 -0
  89. package/dist/agent/structured-output.js +26 -0
  90. package/dist/agent/structured-output.js.map +1 -0
  91. package/dist/agent/tool-admission.d.ts +18 -0
  92. package/dist/agent/tool-admission.d.ts.map +1 -0
  93. package/dist/agent/tool-admission.js +22 -0
  94. package/dist/agent/tool-admission.js.map +1 -0
  95. package/dist/contracts/events.d.ts +236 -0
  96. package/dist/contracts/events.d.ts.map +1 -0
  97. package/dist/contracts/events.js +13 -0
  98. package/dist/contracts/events.js.map +1 -0
  99. package/dist/contracts/messages.d.ts +234 -0
  100. package/dist/contracts/messages.d.ts.map +1 -0
  101. package/dist/contracts/messages.js +11 -0
  102. package/dist/contracts/messages.js.map +1 -0
  103. package/dist/contracts/provider.d.ts +177 -0
  104. package/dist/contracts/provider.d.ts.map +1 -0
  105. package/dist/contracts/provider.js +26 -0
  106. package/dist/contracts/provider.js.map +1 -0
  107. package/dist/contracts/tools.d.ts +217 -0
  108. package/dist/contracts/tools.d.ts.map +1 -0
  109. package/dist/contracts/tools.js +16 -0
  110. package/dist/contracts/tools.js.map +1 -0
  111. package/dist/errors.d.ts +112 -0
  112. package/dist/errors.d.ts.map +1 -0
  113. package/dist/errors.js +190 -0
  114. package/dist/errors.js.map +1 -0
  115. package/dist/index.d.ts +34 -0
  116. package/dist/index.d.ts.map +1 -0
  117. package/dist/index.js +20 -0
  118. package/dist/index.js.map +1 -0
  119. package/dist/internal/concurrency.d.ts +11 -0
  120. package/dist/internal/concurrency.d.ts.map +1 -0
  121. package/dist/internal/concurrency.js +86 -0
  122. package/dist/internal/concurrency.js.map +1 -0
  123. package/dist/internal/stream.d.ts +25 -0
  124. package/dist/internal/stream.d.ts.map +1 -0
  125. package/dist/internal/stream.js +97 -0
  126. package/dist/internal/stream.js.map +1 -0
  127. package/dist/internal/validate.d.ts +30 -0
  128. package/dist/internal/validate.d.ts.map +1 -0
  129. package/dist/internal/validate.js +480 -0
  130. package/dist/internal/validate.js.map +1 -0
  131. package/dist/providers/kit/errors.d.ts +40 -0
  132. package/dist/providers/kit/errors.d.ts.map +1 -0
  133. package/dist/providers/kit/errors.js +48 -0
  134. package/dist/providers/kit/errors.js.map +1 -0
  135. package/dist/providers/kit/index.d.ts +10 -0
  136. package/dist/providers/kit/index.d.ts.map +1 -0
  137. package/dist/providers/kit/index.js +10 -0
  138. package/dist/providers/kit/index.js.map +1 -0
  139. package/dist/providers/kit/retry.d.ts +43 -0
  140. package/dist/providers/kit/retry.d.ts.map +1 -0
  141. package/dist/providers/kit/retry.js +118 -0
  142. package/dist/providers/kit/retry.js.map +1 -0
  143. package/dist/providers/kit/sse.d.ts +22 -0
  144. package/dist/providers/kit/sse.d.ts.map +1 -0
  145. package/dist/providers/kit/sse.js +70 -0
  146. package/dist/providers/kit/sse.js.map +1 -0
  147. package/dist/providers/openai-responses/citations.d.ts +21 -0
  148. package/dist/providers/openai-responses/citations.d.ts.map +1 -0
  149. package/dist/providers/openai-responses/citations.js +65 -0
  150. package/dist/providers/openai-responses/citations.js.map +1 -0
  151. package/dist/providers/openai-responses/continuation.d.ts +19 -0
  152. package/dist/providers/openai-responses/continuation.d.ts.map +1 -0
  153. package/dist/providers/openai-responses/continuation.js +28 -0
  154. package/dist/providers/openai-responses/continuation.js.map +1 -0
  155. package/dist/providers/openai-responses/errors.d.ts +28 -0
  156. package/dist/providers/openai-responses/errors.d.ts.map +1 -0
  157. package/dist/providers/openai-responses/errors.js +103 -0
  158. package/dist/providers/openai-responses/errors.js.map +1 -0
  159. package/dist/providers/openai-responses/history.d.ts +8 -0
  160. package/dist/providers/openai-responses/history.d.ts.map +1 -0
  161. package/dist/providers/openai-responses/history.js +93 -0
  162. package/dist/providers/openai-responses/history.js.map +1 -0
  163. package/dist/providers/openai-responses/items.d.ts +106 -0
  164. package/dist/providers/openai-responses/items.d.ts.map +1 -0
  165. package/dist/providers/openai-responses/items.js +124 -0
  166. package/dist/providers/openai-responses/items.js.map +1 -0
  167. package/dist/providers/openai-responses/parameter-schema.d.ts +11 -0
  168. package/dist/providers/openai-responses/parameter-schema.d.ts.map +1 -0
  169. package/dist/providers/openai-responses/parameter-schema.js +186 -0
  170. package/dist/providers/openai-responses/parameter-schema.js.map +1 -0
  171. package/dist/providers/openai-responses/provider.d.ts +54 -0
  172. package/dist/providers/openai-responses/provider.d.ts.map +1 -0
  173. package/dist/providers/openai-responses/provider.js +270 -0
  174. package/dist/providers/openai-responses/provider.js.map +1 -0
  175. package/dist/providers/openai-responses/request.d.ts +41 -0
  176. package/dist/providers/openai-responses/request.d.ts.map +1 -0
  177. package/dist/providers/openai-responses/request.js +276 -0
  178. package/dist/providers/openai-responses/request.js.map +1 -0
  179. package/dist/providers/openai-responses/response.d.ts +7 -0
  180. package/dist/providers/openai-responses/response.d.ts.map +1 -0
  181. package/dist/providers/openai-responses/response.js +354 -0
  182. package/dist/providers/openai-responses/response.js.map +1 -0
  183. package/dist/providers/openai-responses/stored-items.d.ts +9 -0
  184. package/dist/providers/openai-responses/stored-items.d.ts.map +1 -0
  185. package/dist/providers/openai-responses/stored-items.js +133 -0
  186. package/dist/providers/openai-responses/stored-items.js.map +1 -0
  187. package/dist/providers/openai-responses/tool-search.d.ts +12 -0
  188. package/dist/providers/openai-responses/tool-search.d.ts.map +1 -0
  189. package/dist/providers/openai-responses/tool-search.js +68 -0
  190. package/dist/providers/openai-responses/tool-search.js.map +1 -0
  191. package/dist/providers/openai-responses/tools.d.ts +33 -0
  192. package/dist/providers/openai-responses/tools.d.ts.map +1 -0
  193. package/dist/providers/openai-responses/tools.js +53 -0
  194. package/dist/providers/openai-responses/tools.js.map +1 -0
  195. package/dist/providers/openai-responses/web-search.d.ts +5 -0
  196. package/dist/providers/openai-responses/web-search.d.ts.map +1 -0
  197. package/dist/providers/openai-responses/web-search.js +114 -0
  198. package/dist/providers/openai-responses/web-search.js.map +1 -0
  199. package/dist/providers/openai-responses/websocket.d.ts +29 -0
  200. package/dist/providers/openai-responses/websocket.d.ts.map +1 -0
  201. package/dist/providers/openai-responses/websocket.js +400 -0
  202. package/dist/providers/openai-responses/websocket.js.map +1 -0
  203. package/dist/tools/dispatch.d.ts +65 -0
  204. package/dist/tools/dispatch.d.ts.map +1 -0
  205. package/dist/tools/dispatch.js +536 -0
  206. package/dist/tools/dispatch.js.map +1 -0
  207. package/dist/tools/display-title.d.ts +18 -0
  208. package/dist/tools/display-title.d.ts.map +1 -0
  209. package/dist/tools/display-title.js +50 -0
  210. package/dist/tools/display-title.js.map +1 -0
  211. package/dist/tools/search.d.ts +36 -0
  212. package/dist/tools/search.d.ts.map +1 -0
  213. package/dist/tools/search.js +224 -0
  214. package/dist/tools/search.js.map +1 -0
  215. package/dist/tools/tool.d.ts +200 -0
  216. package/dist/tools/tool.d.ts.map +1 -0
  217. package/dist/tools/tool.js +275 -0
  218. package/dist/tools/tool.js.map +1 -0
  219. package/dist/tools/truncate.d.ts +18 -0
  220. package/dist/tools/truncate.d.ts.map +1 -0
  221. package/dist/tools/truncate.js +76 -0
  222. package/dist/tools/truncate.js.map +1 -0
  223. package/package.json +67 -4
@@ -0,0 +1,401 @@
1
+ import { isAbortError, raceAbort } from "../abort.js";
2
+ import { InvalidModelResponseError, ModelProviderError } from "../errors.js";
3
+ import { messageId as newMessageId } from "../contracts/messages.js";
4
+ import { parseModelItem, validateCompletion } from "../internal/validate.js";
5
+ import { assistantMessage, fallbackOutputParts, persistedToolCall, providerCompleted, providerStarted, startKey } from "./model-output.js";
6
+ import { StreamedCallHistory } from "./streamed-call-history.js";
7
+ import { ModelOutputProjection } from "./model-output-projection.js";
8
+ import { errorMessage } from "@arden-hq/std/errors";
9
+ const openProviderStream = async (input) => {
10
+ let source;
11
+ try {
12
+ source = await input.provider.stream(input.request);
13
+ }
14
+ catch (cause) {
15
+ if (isAbortError(cause))
16
+ throw cause;
17
+ throw new ModelProviderError(`model provider failed: ${errorMessage(cause)}`, cause);
18
+ }
19
+ if (Symbol.asyncIterator in source)
20
+ return source[Symbol.asyncIterator]();
21
+ return (async function* () {
22
+ yield* source;
23
+ })();
24
+ };
25
+ export const runModelTurn = async (input) => {
26
+ const readyCalls = new StreamedCallHistory(input.history);
27
+ const minter = input.mintMessageId;
28
+ const mintId = (kind, index) => minter === undefined ? newMessageId(kind) : minter(kind, { fragment: readyCalls.count, index });
29
+ let response;
30
+ const projection = new ModelOutputProjection(input.emit, mintId);
31
+ let streamedToolInputs = new Map();
32
+ let streamedToolInputIds = new Set();
33
+ let streamedProviderToolIds = new Set();
34
+ let completedProviderToolIds = new Set();
35
+ let toolStarts = new Map();
36
+ let cutAtCommit = false;
37
+ const now = Date.now;
38
+ const emitToolStart = async (event, execution) => {
39
+ const timestamp = now();
40
+ const parentId = event.parent_id ?? null;
41
+ toolStarts.set(startKey(execution, event.tool_id), {
42
+ timestamp,
43
+ parentId,
44
+ persisted: input.persistHost?.(event.host) ?? {}
45
+ });
46
+ await input.emit({ ...event, timestamp });
47
+ };
48
+ const completeProviderTool = async (call) => {
49
+ if (completedProviderToolIds.has(call.id))
50
+ return;
51
+ const start = toolStarts.get(startKey("provider", call.id));
52
+ const duration = start === undefined ? 0 : Math.max(0, Math.trunc(now() - start.timestamp));
53
+ await input.emit(providerCompleted(call, input.metas, duration));
54
+ completedProviderToolIds.add(call.id);
55
+ };
56
+ const resetProjection = () => {
57
+ projection.beginFragment();
58
+ streamedToolInputs = new Map();
59
+ streamedToolInputIds = new Set(readyCalls.persistedIds);
60
+ streamedProviderToolIds = new Set(readyCalls.providerCalls.keys());
61
+ completedProviderToolIds = new Set([...completedProviderToolIds].filter((id) => readyCalls.providerCalls.has(id)));
62
+ toolStarts = new Map([...toolStarts].filter(([key]) => [...readyCalls.persistedIds].some((id) => key === startKey("runtime", id)) ||
63
+ [...readyCalls.providerCalls.keys()].some((id) => key === startKey("provider", id))));
64
+ };
65
+ const emitProjectionReset = async () => {
66
+ await input.emit({
67
+ _tag: "ModelResponseReset",
68
+ depth: 0,
69
+ parent_id: null,
70
+ message_ids: [...projection.messageIds],
71
+ tool_call_ids: [...new Set([...streamedToolInputIds, ...streamedProviderToolIds])]
72
+ .filter((id) => !readyCalls.persistedIds.has(id) && !readyCalls.providerCalls.has(id))
73
+ .sort()
74
+ });
75
+ };
76
+ const toolInput = (index) => {
77
+ const existing = streamedToolInputs.get(index);
78
+ if (existing !== undefined)
79
+ return existing;
80
+ const created = {
81
+ toolId: null,
82
+ name: null,
83
+ arguments: "",
84
+ started: false,
85
+ ended: false
86
+ };
87
+ streamedToolInputs.set(index, created);
88
+ return created;
89
+ };
90
+ const reconcileOutputParts = async (parts, completionResponse) => {
91
+ const message = completionResponse.message;
92
+ const runtimeCalls = new Map((message.tool_calls ?? []).map((call) => [call.id, call]));
93
+ const providerCalls = new Map((message.provider_tool_calls ?? []).map((call) => [call.id, call]));
94
+ if (!projection.canExtend(parts)) {
95
+ await emitProjectionReset();
96
+ resetProjection();
97
+ }
98
+ if (parts.length > 0 && parts.every((part) => part.type === "tool_call")) {
99
+ await projection.ensureText();
100
+ await projection.close();
101
+ }
102
+ for (const [index, part] of parts.entries()) {
103
+ const streamed = projection.parts[index] !== undefined;
104
+ switch (part.type) {
105
+ case "reasoning":
106
+ case "text":
107
+ await projection.completePart(part, index);
108
+ break;
109
+ case "tool_call": {
110
+ if (!streamed) {
111
+ await projection.close();
112
+ projection.parts.push({ ...part });
113
+ }
114
+ if (part.execution === "provider") {
115
+ const call = providerCalls.get(part.tool_call_id);
116
+ if (call === undefined) {
117
+ throw new InvalidModelResponseError("output_parts references a missing provider tool call");
118
+ }
119
+ if (!streamedProviderToolIds.has(call.id)) {
120
+ streamedProviderToolIds.add(call.id);
121
+ await emitToolStart(providerStarted(call, input.metas), "provider");
122
+ }
123
+ await completeProviderTool(call);
124
+ break;
125
+ }
126
+ const call = runtimeCalls.get(part.tool_call_id);
127
+ if (call === undefined) {
128
+ throw new InvalidModelResponseError("output_parts references a missing runtime tool call");
129
+ }
130
+ if (streamedToolInputIds.has(call.id)) {
131
+ const state = [...streamedToolInputs.values()].find((value) => value.toolId === call.id);
132
+ if (state !== undefined && !state.ended) {
133
+ state.ended = true;
134
+ await input.emit({
135
+ _tag: "ToolInputEnded",
136
+ depth: 0,
137
+ parent_id: null,
138
+ tool_id: call.id
139
+ });
140
+ }
141
+ break;
142
+ }
143
+ streamedToolInputIds.add(call.id);
144
+ await emitToolStart({
145
+ _tag: "ToolInputStarted",
146
+ depth: 0,
147
+ parent_id: null,
148
+ tool_id: call.id,
149
+ name: call.function.name,
150
+ host: { ...(input.metas.get(call.function.name)?.host ?? {}) }
151
+ }, "runtime");
152
+ if (call.function.arguments.length > 0) {
153
+ await input.emit({
154
+ _tag: "ToolInputDelta",
155
+ depth: 0,
156
+ parent_id: null,
157
+ tool_id: call.id,
158
+ delta: call.function.arguments
159
+ });
160
+ }
161
+ await input.emit({ _tag: "ToolInputEnded", depth: 0, parent_id: null, tool_id: call.id });
162
+ break;
163
+ }
164
+ }
165
+ }
166
+ };
167
+ const processItem = async (raw) => {
168
+ const item = parseModelItem(raw);
169
+ switch (item._tag) {
170
+ case "TextDelta":
171
+ await projection.delta("text", item.content);
172
+ return;
173
+ case "ReasoningDelta":
174
+ await projection.delta("reasoning", item.content);
175
+ return;
176
+ case "ToolCallDelta": {
177
+ const delta = item.delta;
178
+ const state = toolInput(delta.index);
179
+ if (delta.tool_id !== null)
180
+ state.toolId = delta.tool_id;
181
+ if (delta.name !== null)
182
+ state.name = delta.name;
183
+ if (delta.arguments_delta !== null)
184
+ state.arguments += delta.arguments_delta;
185
+ if (state.toolId !== null && state.name !== null && !state.started) {
186
+ await projection.close();
187
+ state.started = true;
188
+ streamedToolInputIds.add(state.toolId);
189
+ projection.parts.push({
190
+ type: "tool_call",
191
+ tool_call_id: state.toolId,
192
+ execution: "runtime"
193
+ });
194
+ await emitToolStart({
195
+ _tag: "ToolInputStarted",
196
+ depth: 0,
197
+ parent_id: null,
198
+ tool_id: state.toolId,
199
+ name: state.name,
200
+ host: { ...(input.metas.get(state.name)?.host ?? {}) }
201
+ }, "runtime");
202
+ if (state.arguments.length > 0) {
203
+ await input.emit({
204
+ _tag: "ToolInputDelta",
205
+ depth: 0,
206
+ parent_id: null,
207
+ tool_id: state.toolId,
208
+ delta: state.arguments
209
+ });
210
+ }
211
+ }
212
+ else if (delta.arguments_delta !== null && state.started && state.toolId !== null) {
213
+ await input.emit({
214
+ _tag: "ToolInputDelta",
215
+ depth: 0,
216
+ parent_id: null,
217
+ tool_id: state.toolId,
218
+ delta: delta.arguments_delta
219
+ });
220
+ }
221
+ if (delta.done && state.started && !state.ended && state.toolId !== null) {
222
+ state.ended = true;
223
+ await input.emit({
224
+ _tag: "ToolInputEnded",
225
+ depth: 0,
226
+ parent_id: null,
227
+ tool_id: state.toolId
228
+ });
229
+ }
230
+ return;
231
+ }
232
+ case "ProviderToolCall": {
233
+ const call = item.call;
234
+ if (!call.done && streamedProviderToolIds.has(call.id))
235
+ return;
236
+ if (!streamedProviderToolIds.has(call.id)) {
237
+ await projection.close();
238
+ streamedProviderToolIds.add(call.id);
239
+ projection.parts.push({
240
+ type: "tool_call",
241
+ tool_call_id: call.id,
242
+ execution: "provider"
243
+ });
244
+ await emitToolStart(providerStarted(call, input.metas), "provider");
245
+ }
246
+ return;
247
+ }
248
+ case "OutputItemReady": {
249
+ const commit = async () => {
250
+ const fragment = {
251
+ _tag: "Completed",
252
+ model: input.request.model,
253
+ finish_reason: "tool_calls",
254
+ usage: { completion_tokens: 0 },
255
+ message: item.message
256
+ };
257
+ for (const call of item.message.tool_calls ?? []) {
258
+ if (readyCalls.calls.has(call.id))
259
+ throw new InvalidModelResponseError("a completed tool call was emitted twice");
260
+ readyCalls.add(call);
261
+ }
262
+ const parts = item.message.output_parts ?? fallbackOutputParts(fragment, projection.parts);
263
+ await reconcileOutputParts(parts, fragment);
264
+ await projection.close();
265
+ readyCalls.persist(assistantMessage(fragment, projection.fragmentId, parts.some((part) => part.type === "text") ? projection.fragmentId : null, parts, toolStarts, input.displayTitle));
266
+ for (const [index, part] of parts.entries()) {
267
+ const messageId = projection.textMessageIdsByPartIndex.get(index);
268
+ if (part.type !== "text" || part.phase === undefined || messageId === undefined)
269
+ continue;
270
+ await input.emit({
271
+ _tag: "TextPhaseResolved",
272
+ depth: 0,
273
+ parent_id: null,
274
+ message_id: messageId,
275
+ phase: part.phase
276
+ });
277
+ }
278
+ projection.beginFragment();
279
+ for (const call of item.message.tool_calls ?? []) {
280
+ await input.onToolCallReady?.(persistedToolCall(call, toolStarts.get(startKey("runtime", call.id)), input.displayTitle), async () => {
281
+ await input.checkpoint?.();
282
+ });
283
+ }
284
+ };
285
+ await (input.commitOutput === undefined ? commit() : input.commitOutput(commit));
286
+ return;
287
+ }
288
+ case "ProviderStreamRetry": {
289
+ if (readyCalls.persisted) {
290
+ if (projection.messageIds.size > 0 ||
291
+ [...streamedToolInputIds].some((id) => !readyCalls.persistedIds.has(id)) ||
292
+ [...streamedProviderToolIds].some((id) => !readyCalls.providerCalls.has(id))) {
293
+ await emitProjectionReset();
294
+ }
295
+ resetProjection();
296
+ response = readyCalls.committed(input.request.model);
297
+ cutAtCommit = true;
298
+ return;
299
+ }
300
+ readyCalls.reset();
301
+ await input.emit({
302
+ _tag: "ModelStreamRetry",
303
+ depth: 0,
304
+ parent_id: null,
305
+ attempt: item.attempt,
306
+ max_attempts: item.maxAttempts
307
+ });
308
+ if (projection.messageIds.size > 0 ||
309
+ streamedToolInputIds.size > 0 ||
310
+ streamedProviderToolIds.size > 0) {
311
+ await emitProjectionReset();
312
+ }
313
+ response = undefined;
314
+ resetProjection();
315
+ return;
316
+ }
317
+ case "Completed": {
318
+ if (response !== undefined) {
319
+ throw new InvalidModelResponseError("model stream emitted more than one CompletionResponse");
320
+ }
321
+ response = item;
322
+ return;
323
+ }
324
+ }
325
+ };
326
+ const iterator = await openProviderStream(input);
327
+ try {
328
+ while (true) {
329
+ let next;
330
+ try {
331
+ next = await raceAbort(iterator.next(), input.request.signal);
332
+ }
333
+ catch (cause) {
334
+ if (isAbortError(cause) || cause instanceof ModelProviderError)
335
+ throw cause;
336
+ throw new ModelProviderError(`model provider stream failed: ${errorMessage(cause)}`, cause);
337
+ }
338
+ if (next.done)
339
+ break;
340
+ await processItem(next.value);
341
+ if (cutAtCommit) {
342
+ void Promise.resolve()
343
+ .then(() => iterator.return?.())
344
+ .catch(() => undefined);
345
+ break;
346
+ }
347
+ }
348
+ }
349
+ catch (cause) {
350
+ await projection.close().catch(() => undefined);
351
+ void Promise.resolve()
352
+ .then(() => iterator.return?.())
353
+ .catch(() => undefined);
354
+ throw cause;
355
+ }
356
+ if (response === undefined) {
357
+ await projection.close();
358
+ throw new InvalidModelResponseError("model stream ended without a CompletionResponse");
359
+ }
360
+ let validResponse;
361
+ try {
362
+ validResponse = validateCompletion(response);
363
+ }
364
+ catch (cause) {
365
+ await projection.close().catch(() => undefined);
366
+ throw cause;
367
+ }
368
+ readyCalls.validate(validResponse.message);
369
+ const message = validResponse.message;
370
+ const canonicalOutputParts = message.output_parts ?? fallbackOutputParts(validResponse, projection.parts);
371
+ if (!readyCalls.persisted) {
372
+ await reconcileOutputParts(canonicalOutputParts, validResponse);
373
+ await projection.ensureText();
374
+ }
375
+ const resolvedTextPhases = (readyCalls.persisted ? [] : canonicalOutputParts).flatMap((part, index) => {
376
+ if (part.type !== "text" || part.phase === undefined)
377
+ return [];
378
+ const messageId = projection.textMessageIdsByPartIndex.get(index);
379
+ if (messageId === undefined) {
380
+ throw new InvalidModelResponseError("text output part is missing its projected message id");
381
+ }
382
+ return [{ messageId, phase: part.phase }];
383
+ });
384
+ readyCalls.finish(assistantMessage(validResponse, projection.fragmentId, canonicalOutputParts.some((part) => part.type === "text") ? projection.fragmentId : null, canonicalOutputParts, toolStarts, input.displayTitle));
385
+ await projection.close();
386
+ for (const resolved of resolvedTextPhases) {
387
+ await input.emit({
388
+ _tag: "TextPhaseResolved",
389
+ depth: 0,
390
+ parent_id: null,
391
+ message_id: resolved.messageId,
392
+ phase: resolved.phase
393
+ });
394
+ }
395
+ return {
396
+ response: validResponse,
397
+ streamedToolInputIds,
398
+ lastTextId: projection.lastTextId
399
+ };
400
+ };
401
+ //# sourceMappingURL=model-turn.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"model-turn.js","sourceRoot":"","sources":["../../src/agent/model-turn.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,aAAa,CAAA;AACrD,OAAO,EAAE,yBAAyB,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAA;AAC5E,OAAO,EACL,SAAS,IAAI,YAAY,EAO1B,MAAM,0BAA0B,CAAA;AASjC,OAAO,EAAE,cAAc,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAA;AAE5E,OAAO,EACL,gBAAgB,EAChB,mBAAmB,EACnB,iBAAiB,EACjB,iBAAiB,EACjB,eAAe,EACf,QAAQ,EAGT,MAAM,mBAAmB,CAAA;AAC1B,OAAO,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAA;AAChE,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAA;AACpE,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAA;AAmCnD,MAAM,kBAAkB,GAAG,KAAK,EAC9B,KAAqB,EACoB,EAAE;IAC3C,IAAI,MAAkE,CAAA;IACtE,IAAI,CAAC;QACH,MAAM,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;IACrD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,YAAY,CAAC,KAAK,CAAC;YAAE,MAAM,KAAK,CAAA;QACpC,MAAM,IAAI,kBAAkB,CAAC,0BAA0B,YAAY,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAA;IACtF,CAAC;IACD,IAAI,MAAM,CAAC,aAAa,IAAI,MAAM;QAAE,OAAO,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAA;IACzE,OAAO,CAAC,KAAK,SAAS,CAAC;QACrB,KAAK,CAAC,CAAC,MAAM,CAAA;IACf,CAAC,CAAC,EAAE,CAAA;AACN,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,KAAK,EAAE,KAAqB,EAA4B,EAAE;IACpF,MAAM,UAAU,GAAG,IAAI,mBAAmB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;IACzD,MAAM,MAAM,GAAG,KAAK,CAAC,aAAa,CAAA;IAClC,MAAM,MAAM,GAAG,CAAC,IAA0B,EAAE,KAAa,EAAU,EAAE,CACnE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,QAAQ,EAAE,UAAU,CAAC,KAAK,EAAE,KAAK,EAAE,CAAC,CAAA;IACjG,IAAI,QAAwC,CAAA;IAC5C,MAAM,UAAU,GAAG,IAAI,qBAAqB,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;IAChE,IAAI,kBAAkB,GAAG,IAAI,GAAG,EAA6B,CAAA;IAC7D,IAAI,oBAAoB,GAAG,IAAI,GAAG,EAAU,CAAA;IAC5C,IAAI,uBAAuB,GAAG,IAAI,GAAG,EAAU,CAAA;IAC/C,IAAI,wBAAwB,GAAG,IAAI,GAAG,EAAU,CAAA;IAChD,IAAI,UAAU,GAAG,IAAI,GAAG,EAA6B,CAAA;IACrD,IAAI,WAAW,GAAG,KAAK,CAAA;IACvB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAA;IAEpB,MAAM,aAAa,GAAG,KAAK,EACzB,KAAqB,EACrB,SAAiC,EAClB,EAAE;QACjB,MAAM,SAAS,GAAG,GAAG,EAAE,CAAA;QACvB,MAAM,QAAQ,GAAG,KAAK,CAAC,SAAS,IAAI,IAAI,CAAA;QACxC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE;YACjD,SAAS;YACT,QAAQ;YACR,SAAS,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE;SACjD,CAAC,CAAA;QACF,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;IAC3C,CAAC,CAAA;IAED,MAAM,oBAAoB,GAAG,KAAK,EAAE,IAAsB,EAAiB,EAAE;QAC3E,IAAI,wBAAwB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAAE,OAAM;QACjD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,UAAU,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC,CAAA;QAC3D,MAAM,QAAQ,GAAG,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAA;QAC3F,MAAM,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAA;QAChE,wBAAwB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IACvC,CAAC,CAAA;IAED,MAAM,eAAe,GAAG,GAAS,EAAE;QACjC,UAAU,CAAC,aAAa,EAAE,CAAA;QAC1B,kBAAkB,GAAG,IAAI,GAAG,EAAE,CAAA;QAC9B,oBAAoB,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,YAAY,CAAC,CAAA;QACvD,uBAAuB,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAA;QAClE,wBAAwB,GAAG,IAAI,GAAG,CAChC,CAAC,GAAG,wBAAwB,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAC/E,CAAA;QACD,UAAU,GAAG,IAAI,GAAG,CAClB,CAAC,GAAG,UAAU,CAAC,CAAC,MAAM,CACpB,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CACR,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,KAAK,QAAQ,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YAC1E,CAAC,GAAG,UAAU,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,KAAK,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,CACtF,CACF,CAAA;IACH,CAAC,CAAA;IAED,MAAM,mBAAmB,GAAG,KAAK,IAAmB,EAAE;QACpD,MAAM,KAAK,CAAC,IAAI,CAAC;YACf,IAAI,EAAE,oBAAoB;YAC1B,KAAK,EAAE,CAAC;YACR,SAAS,EAAE,IAAI;YACf,WAAW,EAAE,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC;YACvC,aAAa,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,oBAAoB,EAAE,GAAG,uBAAuB,CAAC,CAAC,CAAC;iBAC/E,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;iBACrF,IAAI,EAAE;SACV,CAAC,CAAA;IACJ,CAAC,CAAA;IAED,MAAM,SAAS,GAAG,CAAC,KAAa,EAAqB,EAAE;QACrD,MAAM,QAAQ,GAAG,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QAC9C,IAAI,QAAQ,KAAK,SAAS;YAAE,OAAO,QAAQ,CAAA;QAC3C,MAAM,OAAO,GAAsB;YACjC,MAAM,EAAE,IAAI;YACZ,IAAI,EAAE,IAAI;YACV,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,KAAK;SACb,CAAA;QACD,kBAAkB,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAA;QACtC,OAAO,OAAO,CAAA;IAChB,CAAC,CAAA;IAED,MAAM,oBAAoB,GAAG,KAAK,EAChC,KAAyC,EACzC,kBAAsC,EACvB,EAAE;QACjB,MAAM,OAAO,GAAG,kBAAkB,CAAC,OAAO,CAAA;QAC1C,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC,CAAA;QACvF,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,CAAC,OAAO,CAAC,mBAAmB,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CACnE,CAAA;QACD,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;YACjC,MAAM,mBAAmB,EAAE,CAAA;YAC3B,eAAe,EAAE,CAAA;QACnB,CAAC;QACD,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,WAAW,CAAC,EAAE,CAAC;YACzE,MAAM,UAAU,CAAC,UAAU,EAAE,CAAA;YAC7B,MAAM,UAAU,CAAC,KAAK,EAAE,CAAA;QAC1B,CAAC;QACD,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;YAC5C,MAAM,QAAQ,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,SAAS,CAAA;YACtD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,WAAW,CAAC;gBACjB,KAAK,MAAM;oBACT,MAAM,UAAU,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;oBAC1C,MAAK;gBACP,KAAK,WAAW,EAAE,CAAC;oBACjB,IAAI,CAAC,QAAQ,EAAE,CAAC;wBACd,MAAM,UAAU,CAAC,KAAK,EAAE,CAAA;wBACxB,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAA;oBACpC,CAAC;oBACD,IAAI,IAAI,CAAC,SAAS,KAAK,UAAU,EAAE,CAAC;wBAClC,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;wBACjD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;4BACvB,MAAM,IAAI,yBAAyB,CACjC,sDAAsD,CACvD,CAAA;wBACH,CAAC;wBACD,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;4BAC1C,uBAAuB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;4BACpC,MAAM,aAAa,CAAC,eAAe,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,UAAU,CAAC,CAAA;wBACrE,CAAC;wBACD,MAAM,oBAAoB,CAAC,IAAI,CAAC,CAAA;wBAChC,MAAK;oBACP,CAAC;oBACD,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;oBAChD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;wBACvB,MAAM,IAAI,yBAAyB,CACjC,qDAAqD,CACtD,CAAA;oBACH,CAAC;oBACD,IAAI,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;wBACtC,MAAM,KAAK,GAAG,CAAC,GAAG,kBAAkB,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,IAAI,CAAC,EAAE,CAAC,CAAA;wBACxF,IAAI,KAAK,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;4BACxC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAA;4BAClB,MAAM,KAAK,CAAC,IAAI,CAAC;gCACf,IAAI,EAAE,gBAAgB;gCACtB,KAAK,EAAE,CAAC;gCACR,SAAS,EAAE,IAAI;gCACf,OAAO,EAAE,IAAI,CAAC,EAAE;6BACjB,CAAC,CAAA;wBACJ,CAAC;wBACD,MAAK;oBACP,CAAC;oBACD,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;oBACjC,MAAM,aAAa,CACjB;wBACE,IAAI,EAAE,kBAAkB;wBACxB,KAAK,EAAE,CAAC;wBACR,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,IAAI,CAAC,EAAE;wBAChB,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI;wBACxB,IAAI,EAAE,EAAE,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC,EAAE;qBAC/D,EACD,SAAS,CACV,CAAA;oBACD,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACvC,MAAM,KAAK,CAAC,IAAI,CAAC;4BACf,IAAI,EAAE,gBAAgB;4BACtB,KAAK,EAAE,CAAC;4BACR,SAAS,EAAE,IAAI;4BACf,OAAO,EAAE,IAAI,CAAC,EAAE;4BAChB,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS;yBAC/B,CAAC,CAAA;oBACJ,CAAC;oBACD,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAA;oBACzF,MAAK;gBACP,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAA;IAED,MAAM,WAAW,GAAG,KAAK,EAAE,GAAoB,EAAiB,EAAE;QAChE,MAAM,IAAI,GAAG,cAAc,CAAC,GAAG,CAAC,CAAA;QAEhC,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,WAAW;gBACd,MAAM,UAAU,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAA;gBAC5C,OAAM;YAER,KAAK,gBAAgB;gBACnB,MAAM,UAAU,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,CAAA;gBACjD,OAAM;YAER,KAAK,eAAe,EAAE,CAAC;gBACrB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAA;gBACxB,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBACpC,IAAI,KAAK,CAAC,OAAO,KAAK,IAAI;oBAAE,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,OAAO,CAAA;gBACxD,IAAI,KAAK,CAAC,IAAI,KAAK,IAAI;oBAAE,KAAK,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAA;gBAChD,IAAI,KAAK,CAAC,eAAe,KAAK,IAAI;oBAAE,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,eAAe,CAAA;gBAC5E,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,IAAI,KAAK,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;oBACnE,MAAM,UAAU,CAAC,KAAK,EAAE,CAAA;oBACxB,KAAK,CAAC,OAAO,GAAG,IAAI,CAAA;oBACpB,oBAAoB,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;oBACtC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC;wBACpB,IAAI,EAAE,WAAW;wBACjB,YAAY,EAAE,KAAK,CAAC,MAAM;wBAC1B,SAAS,EAAE,SAAS;qBACrB,CAAC,CAAA;oBACF,MAAM,aAAa,CACjB;wBACE,IAAI,EAAE,kBAAkB;wBACxB,KAAK,EAAE,CAAC;wBACR,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,KAAK,CAAC,MAAM;wBACrB,IAAI,EAAE,KAAK,CAAC,IAAI;wBAChB,IAAI,EAAE,EAAE,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC,EAAE;qBACvD,EACD,SAAS,CACV,CAAA;oBACD,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC/B,MAAM,KAAK,CAAC,IAAI,CAAC;4BACf,IAAI,EAAE,gBAAgB;4BACtB,KAAK,EAAE,CAAC;4BACR,SAAS,EAAE,IAAI;4BACf,OAAO,EAAE,KAAK,CAAC,MAAM;4BACrB,KAAK,EAAE,KAAK,CAAC,SAAS;yBACvB,CAAC,CAAA;oBACJ,CAAC;gBACH,CAAC;qBAAM,IAAI,KAAK,CAAC,eAAe,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;oBACpF,MAAM,KAAK,CAAC,IAAI,CAAC;wBACf,IAAI,EAAE,gBAAgB;wBACtB,KAAK,EAAE,CAAC;wBACR,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,KAAK,CAAC,MAAM;wBACrB,KAAK,EAAE,KAAK,CAAC,eAAe;qBAC7B,CAAC,CAAA;gBACJ,CAAC;gBACD,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;oBACzE,KAAK,CAAC,KAAK,GAAG,IAAI,CAAA;oBAClB,MAAM,KAAK,CAAC,IAAI,CAAC;wBACf,IAAI,EAAE,gBAAgB;wBACtB,KAAK,EAAE,CAAC;wBACR,SAAS,EAAE,IAAI;wBACf,OAAO,EAAE,KAAK,CAAC,MAAM;qBACtB,CAAC,CAAA;gBACJ,CAAC;gBACD,OAAM;YACR,CAAC;YAED,KAAK,kBAAkB,EAAE,CAAC;gBACxB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAA;gBACtB,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,uBAAuB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;oBAAE,OAAM;gBAC9D,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC1C,MAAM,UAAU,CAAC,KAAK,EAAE,CAAA;oBACxB,uBAAuB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;oBACpC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC;wBACpB,IAAI,EAAE,WAAW;wBACjB,YAAY,EAAE,IAAI,CAAC,EAAE;wBACrB,SAAS,EAAE,UAAU;qBACtB,CAAC,CAAA;oBACF,MAAM,aAAa,CAAC,eAAe,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,EAAE,UAAU,CAAC,CAAA;gBACrE,CAAC;gBACD,OAAM;YACR,CAAC;YAED,KAAK,iBAAiB,EAAE,CAAC;gBACvB,MAAM,MAAM,GAAG,KAAK,IAAmB,EAAE;oBACvC,MAAM,QAAQ,GAAuB;wBACnC,IAAI,EAAE,WAAW;wBACjB,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK;wBAC1B,aAAa,EAAE,YAAY;wBAC3B,KAAK,EAAE,EAAE,iBAAiB,EAAE,CAAC,EAAE;wBAC/B,OAAO,EAAE,IAAI,CAAC,OAAO;qBACtB,CAAA;oBACD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,IAAI,EAAE,EAAE,CAAC;wBACjD,IAAI,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;4BAC/B,MAAM,IAAI,yBAAyB,CAAC,yCAAyC,CAAC,CAAA;wBAChF,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;oBACtB,CAAC;oBACD,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,IAAI,mBAAmB,CAAC,QAAQ,EAAE,UAAU,CAAC,KAAK,CAAC,CAAA;oBAC1F,MAAM,oBAAoB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAA;oBAC3C,MAAM,UAAU,CAAC,KAAK,EAAE,CAAA;oBACxB,UAAU,CAAC,OAAO,CAChB,gBAAgB,CACd,QAAQ,EACR,UAAU,CAAC,UAAU,EACrB,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,EACzE,KAAK,EACL,UAAU,EACV,KAAK,CAAC,YAAY,CACnB,CACF,CAAA;oBACD,KAAK,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC;wBAC5C,MAAM,SAAS,GAAG,UAAU,CAAC,yBAAyB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;wBACjE,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,IAAI,SAAS,KAAK,SAAS;4BAC7E,SAAQ;wBACV,MAAM,KAAK,CAAC,IAAI,CAAC;4BACf,IAAI,EAAE,mBAAmB;4BACzB,KAAK,EAAE,CAAC;4BACR,SAAS,EAAE,IAAI;4BACf,UAAU,EAAE,SAAS;4BACrB,KAAK,EAAE,IAAI,CAAC,KAAK;yBAClB,CAAC,CAAA;oBACJ,CAAC;oBACD,UAAU,CAAC,aAAa,EAAE,CAAA;oBAC1B,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,IAAI,EAAE,EAAE,CAAC;wBACjD,MAAM,KAAK,CAAC,eAAe,EAAE,CAC3B,iBAAiB,CACf,IAAI,EACJ,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC,EAC5C,KAAK,CAAC,YAAY,CACnB,EACD,KAAK,IAAI,EAAE;4BACT,MAAM,KAAK,CAAC,UAAU,EAAE,EAAE,CAAA;wBAC5B,CAAC,CACF,CAAA;oBACH,CAAC;gBACH,CAAC,CAAA;gBACD,MAAM,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAA;gBAChF,OAAM;YACR,CAAC;YAED,KAAK,qBAAqB,EAAE,CAAC;gBAC3B,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;oBACzB,IACE,UAAU,CAAC,UAAU,CAAC,IAAI,GAAG,CAAC;wBAC9B,CAAC,GAAG,oBAAoB,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;wBACxE,CAAC,GAAG,uBAAuB,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,EAC5E,CAAC;wBACD,MAAM,mBAAmB,EAAE,CAAA;oBAC7B,CAAC;oBACD,eAAe,EAAE,CAAA;oBACjB,QAAQ,GAAG,UAAU,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;oBACpD,WAAW,GAAG,IAAI,CAAA;oBAClB,OAAM;gBACR,CAAC;gBACD,UAAU,CAAC,KAAK,EAAE,CAAA;gBAElB,MAAM,KAAK,CAAC,IAAI,CAAC;oBACf,IAAI,EAAE,kBAAkB;oBACxB,KAAK,EAAE,CAAC;oBACR,SAAS,EAAE,IAAI;oBACf,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,YAAY,EAAE,IAAI,CAAC,WAAW;iBAC/B,CAAC,CAAA;gBACF,IACE,UAAU,CAAC,UAAU,CAAC,IAAI,GAAG,CAAC;oBAC9B,oBAAoB,CAAC,IAAI,GAAG,CAAC;oBAC7B,uBAAuB,CAAC,IAAI,GAAG,CAAC,EAChC,CAAC;oBACD,MAAM,mBAAmB,EAAE,CAAA;gBAC7B,CAAC;gBACD,QAAQ,GAAG,SAAS,CAAA;gBACpB,eAAe,EAAE,CAAA;gBACjB,OAAM;YACR,CAAC;YAED,KAAK,WAAW,EAAE,CAAC;gBACjB,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;oBAC3B,MAAM,IAAI,yBAAyB,CACjC,uDAAuD,CACxD,CAAA;gBACH,CAAC;gBACD,QAAQ,GAAG,IAAI,CAAA;gBACf,OAAM;YACR,CAAC;QACH,CAAC;IACH,CAAC,CAAA;IAED,MAAM,QAAQ,GAAG,MAAM,kBAAkB,CAAC,KAAK,CAAC,CAAA;IAChD,IAAI,CAAC;QACH,OAAO,IAAI,EAAE,CAAC;YACZ,IAAI,IAAqC,CAAA;YACzC,IAAI,CAAC;gBACH,IAAI,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;YAC/D,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,YAAY,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,kBAAkB;oBAAE,MAAM,KAAK,CAAA;gBAC3E,MAAM,IAAI,kBAAkB,CAAC,iCAAiC,YAAY,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,CAAA;YAC7F,CAAC;YACD,IAAI,IAAI,CAAC,IAAI;gBAAE,MAAK;YACpB,MAAM,WAAW,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YAC7B,IAAI,WAAW,EAAE,CAAC;gBAChB,KAAK,OAAO,CAAC,OAAO,EAAE;qBACnB,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;qBAC/B,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAA;gBACzB,MAAK;YACP,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAA;QAC/C,KAAK,OAAO,CAAC,OAAO,EAAE;aACnB,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;aAC/B,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAA;QACzB,MAAM,KAAK,CAAA;IACb,CAAC;IAED,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,MAAM,UAAU,CAAC,KAAK,EAAE,CAAA;QACxB,MAAM,IAAI,yBAAyB,CAAC,iDAAiD,CAAC,CAAA;IACxF,CAAC;IACD,IAAI,aAAiC,CAAA;IACrC,IAAI,CAAC;QACH,aAAa,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAA;IAC9C,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAA;QAC/C,MAAM,KAAK,CAAA;IACb,CAAC;IAED,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,OAAO,CAAC,CAAA;IAC1C,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,CAAA;IACrC,MAAM,oBAAoB,GACxB,OAAO,CAAC,YAAY,IAAI,mBAAmB,CAAC,aAAa,EAAE,UAAU,CAAC,KAAK,CAAC,CAAA;IAC9E,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,CAAC;QAC1B,MAAM,oBAAoB,CAAC,oBAAoB,EAAE,aAAa,CAAC,CAAA;QAC/D,MAAM,UAAU,CAAC,UAAU,EAAE,CAAA;IAC/B,CAAC;IAED,MAAM,kBAAkB,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,OAAO,CACnF,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;QACd,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS;YAAE,OAAO,EAAE,CAAA;QAC/D,MAAM,SAAS,GAAG,UAAU,CAAC,yBAAyB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QACjE,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,IAAI,yBAAyB,CAAC,sDAAsD,CAAC,CAAA;QAC7F,CAAC;QACD,OAAO,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAA;IAC3C,CAAC,CACF,CAAA;IAED,UAAU,CAAC,MAAM,CACf,gBAAgB,CACd,aAAa,EACb,UAAU,CAAC,UAAU,EACrB,oBAAoB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,EACxF,oBAAoB,EACpB,UAAU,EACV,KAAK,CAAC,YAAY,CACnB,CACF,CAAA;IACD,MAAM,UAAU,CAAC,KAAK,EAAE,CAAA;IACxB,KAAK,MAAM,QAAQ,IAAI,kBAAkB,EAAE,CAAC;QAC1C,MAAM,KAAK,CAAC,IAAI,CAAC;YACf,IAAI,EAAE,mBAAmB;YACzB,KAAK,EAAE,CAAC;YACR,SAAS,EAAE,IAAI;YACf,UAAU,EAAE,QAAQ,CAAC,SAAS;YAC9B,KAAK,EAAE,QAAQ,CAAC,KAAK;SACtB,CAAC,CAAA;IACJ,CAAC;IAED,OAAO;QACL,QAAQ,EAAE,aAAa;QACvB,oBAAoB;QACpB,UAAU,EAAE,UAAU,CAAC,UAAU;KAClC,CAAA;AACH,CAAC,CAAA"}
@@ -0,0 +1,28 @@
1
+ import { type ChatMessage, type ToolCall } from "../contracts/messages.js";
2
+ import { type PendingToolCall, type ToolMeta, type ToolResult } from "../contracts/tools.js";
3
+ import { StopReason, type RunBudgetSnapshot, type RunPolicy, type ToolCompleted } from "../contracts/events.js";
4
+ import type { AgentConfig } from "./config.js";
5
+ interface BudgetStopInput {
6
+ readonly config: AgentConfig;
7
+ readonly budget: RunBudgetSnapshot;
8
+ readonly startedAtMs: number;
9
+ readonly nowMs: number;
10
+ readonly cost: number;
11
+ }
12
+ export declare const budgetStopReason: (input: BudgetStopInput) => StopReason | null;
13
+ export declare const turnEndingToolBatchRejection: (calls: ReadonlyArray<PendingToolCall>, metas: ReadonlyMap<string, ToolMeta>) => ToolResult | null;
14
+ export interface TurnEndingBatchRefusal {
15
+ readonly rejection: ToolResult;
16
+ readonly refused: ReadonlyArray<PendingToolCall>;
17
+ readonly refusedRaw: ReadonlyArray<ToolCall>;
18
+ }
19
+ export declare const turnEndingBatchRefusal: (calls: ReadonlyArray<PendingToolCall>, rawCalls: ReadonlyArray<ToolCall>, metas: ReadonlyMap<string, ToolMeta>) => TurnEndingBatchRefusal | null;
20
+ export declare const nextNoProgressCount: (current: number, completions: ReadonlyArray<ToolCompleted>) => number;
21
+ export declare const terminalCompletionText: (completions: ReadonlyArray<ToolCompleted>) => string | undefined;
22
+ export declare const budgetDenialResult: (reason: StopReason, policy: RunPolicy) => Required<ToolResult>;
23
+ export declare const appendBudgetDenials: (messages: ChatMessage[], calls: ReadonlyArray<ToolCall>, reason: StopReason, policy: RunPolicy) => void;
24
+ export declare const withBudgetReminder: (messages: ReadonlyArray<ChatMessage>, reminder: string) => ReadonlyArray<ChatMessage>;
25
+ export declare const loadedInHistory: (messages: ReadonlyArray<ChatMessage>) => Set<string>;
26
+ export declare const recoveredTerminalText: (calls: ReadonlyArray<PendingToolCall>, recovered: ReadonlyMap<string, Required<ToolResult>>, metas: ReadonlyMap<string, ToolMeta>) => string | undefined;
27
+ export {};
28
+ //# sourceMappingURL=policy.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"policy.d.ts","sourceRoot":"","sources":["../../src/agent/policy.ts"],"names":[],"mappings":"AAAA,OAAO,EAAa,KAAK,WAAW,EAAE,KAAK,QAAQ,EAAE,MAAM,0BAA0B,CAAA;AACrF,OAAO,EAEL,KAAK,eAAe,EACpB,KAAK,QAAQ,EACb,KAAK,UAAU,EAChB,MAAM,uBAAuB,CAAA;AAC9B,OAAO,EACL,UAAU,EACV,KAAK,iBAAiB,EACtB,KAAK,SAAS,EACd,KAAK,aAAa,EACnB,MAAM,wBAAwB,CAAA;AAC/B,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAA;AAG9C,UAAU,eAAe;IACvB,QAAQ,CAAC,MAAM,EAAE,WAAW,CAAA;IAC5B,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAA;IAClC,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAA;IAC5B,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAA;IACtB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAA;CACtB;AAED,eAAO,MAAM,gBAAgB,UAAW,eAAe,KAAG,UAAU,GAAG,IAqBtE,CAAA;AAED,eAAO,MAAM,4BAA4B,UAChC,aAAa,CAAC,eAAe,CAAC,SAC9B,WAAW,CAAC,MAAM,EAAE,QAAQ,CAAC,KACnC,UAAU,GAAG,IAYf,CAAA;AAED,MAAM,WAAW,sBAAsB;IACrC,QAAQ,CAAC,SAAS,EAAE,UAAU,CAAA;IAC9B,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC,CAAA;IAChD,QAAQ,CAAC,UAAU,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAA;CAC7C;AAED,eAAO,MAAM,sBAAsB,UAC1B,aAAa,CAAC,eAAe,CAAC,YAC3B,aAAa,CAAC,QAAQ,CAAC,SAC1B,WAAW,CAAC,MAAM,EAAE,QAAQ,CAAC,KACnC,sBAAsB,GAAG,IAM3B,CAAA;AAED,eAAO,MAAM,mBAAmB,YACrB,MAAM,eACF,aAAa,CAAC,aAAa,CAAC,KACxC,MAGF,CAAA;AAED,eAAO,MAAM,sBAAsB,gBACpB,aAAa,CAAC,aAAa,CAAC,KACxC,MAAM,GAAG,SAIX,CAAA;AAED,eAAO,MAAM,kBAAkB,WAAY,UAAU,UAAU,SAAS,KAAG,QAAQ,CAAC,UAAU,CAQ1F,CAAA;AAEJ,eAAO,MAAM,mBAAmB,aACpB,WAAW,EAAE,SAChB,aAAa,CAAC,QAAQ,CAAC,UACtB,UAAU,UACV,SAAS,KAChB,IAUF,CAAA;AAED,eAAO,MAAM,kBAAkB,aACnB,aAAa,CAAC,WAAW,CAAC,YAC1B,MAAM,KACf,aAAa,CAAC,WAAW,CAS3B,CAAA;AAED,eAAO,MAAM,eAAe,aAAc,aAAa,CAAC,WAAW,CAAC,KAAG,GAAG,CAAC,MAAM,CAchF,CAAA;AAED,eAAO,MAAM,qBAAqB,UACzB,aAAa,CAAC,eAAe,CAAC,aAC1B,WAAW,CAAC,MAAM,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC,SAC7C,WAAW,CAAC,MAAM,EAAE,QAAQ,CAAC,KACnC,MAAM,GAAG,SASX,CAAA"}
@@ -0,0 +1,109 @@
1
+ import { messageId } from "../contracts/messages.js";
2
+ import { callEndsTurn } from "../contracts/tools.js";
3
+ import { StopReason } from "../contracts/events.js";
4
+ import { normalizeToolResult } from "../internal/validate.js";
5
+ export const budgetStopReason = (input) => {
6
+ if (input.config.maxToolCalls !== undefined &&
7
+ input.budget.toolCalls >= input.config.maxToolCalls) {
8
+ return StopReason.MAX_TOOL_CALLS;
9
+ }
10
+ if (input.config.maxWallTimeMs !== undefined &&
11
+ input.nowMs - input.startedAtMs >= input.config.maxWallTimeMs) {
12
+ return StopReason.MAX_WALL_TIME;
13
+ }
14
+ if (input.config.maxCost !== undefined && input.cost >= input.config.maxCost)
15
+ return StopReason.MAX_COST;
16
+ if (input.budget.totalOutputTokens !== undefined &&
17
+ input.budget.outputTokens >= input.budget.totalOutputTokens)
18
+ return StopReason.MAX_TOKEN_BUDGET;
19
+ return null;
20
+ };
21
+ export const turnEndingToolBatchRejection = (calls, metas) => {
22
+ if (calls.length > 1 && calls.some((call) => callEndsTurn(metas.get(call.name), call.args))) {
23
+ return {
24
+ content: "Not run: a turn-ending tool must be the only call in its step. " +
25
+ "Call it again by itself once the other work is done and its results are in.",
26
+ preview: "Call it by itself",
27
+ is_error: true,
28
+ retryable: false
29
+ };
30
+ }
31
+ return null;
32
+ };
33
+ export const turnEndingBatchRefusal = (calls, rawCalls, metas) => {
34
+ const rejection = turnEndingToolBatchRejection(calls, metas);
35
+ if (rejection === null)
36
+ return null;
37
+ const refused = calls.filter((call) => callEndsTurn(metas.get(call.name), call.args));
38
+ const refusedIds = new Set(refused.map((call) => call.toolCall.id));
39
+ return { rejection, refused, refusedRaw: rawCalls.filter((call) => refusedIds.has(call.id)) };
40
+ };
41
+ export const nextNoProgressCount = (current, completions) => {
42
+ if (completions.length === 0)
43
+ return 0;
44
+ return completions.every((event) => event.is_error && event.retryable === false) ? current + 1 : 0;
45
+ };
46
+ export const terminalCompletionText = (completions) => {
47
+ const terminal = completions.find((event) => event.ends_turn && !event.is_error);
48
+ if (terminal === undefined)
49
+ return undefined;
50
+ return terminal.result.trim() || terminal.preview.trim() || undefined;
51
+ };
52
+ export const budgetDenialResult = (reason, policy) => normalizeToolResult({
53
+ content: policy.budgetDenialContent?.(reason) ??
54
+ `Tool call denied: the run hit its ${reason.replaceAll("_", " ")} limit. Stop calling tools and summarize.`,
55
+ preview: "Tool call denied",
56
+ is_error: true,
57
+ retryable: false
58
+ });
59
+ export const appendBudgetDenials = (messages, calls, reason, policy) => {
60
+ const denial = budgetDenialResult(reason, policy);
61
+ for (const call of calls) {
62
+ messages.push({
63
+ message_id: messageId("tool"),
64
+ role: "tool",
65
+ tool_call_id: call.id,
66
+ content: denial.content
67
+ });
68
+ }
69
+ };
70
+ export const withBudgetReminder = (messages, reminder) => {
71
+ const first = messages[0];
72
+ if (first?.role !== "system")
73
+ return [{ role: "system", content: reminder }, ...messages];
74
+ const content = first.content;
75
+ const system = typeof content === "string"
76
+ ? { ...first, content: `${content}\n\n${reminder}` }
77
+ : { ...first, content: [...content, { type: "text", text: reminder }] };
78
+ return [system, ...messages.slice(1)];
79
+ };
80
+ export const loadedInHistory = (messages) => {
81
+ const loaded = new Set();
82
+ for (const message of messages) {
83
+ if (message.role === "developer") {
84
+ for (const name of message.loaded_tools ?? [])
85
+ loaded.add(name);
86
+ for (const name of message.removed_tools ?? [])
87
+ loaded.delete(name);
88
+ }
89
+ if (message.role === "assistant") {
90
+ for (const call of message.provider_tool_calls ?? []) {
91
+ for (const name of call.loaded_tool_names ?? [])
92
+ loaded.add(name);
93
+ }
94
+ }
95
+ }
96
+ return loaded;
97
+ };
98
+ export const recoveredTerminalText = (calls, recovered, metas) => {
99
+ for (const call of calls) {
100
+ const result = recovered.get(call.toolCall.id);
101
+ if (!callEndsTurn(metas.get(call.name), call.args) || result === undefined || result.is_error)
102
+ continue;
103
+ const text = result.content.trim() || result.preview.trim();
104
+ if (text.length > 0)
105
+ return text;
106
+ }
107
+ return undefined;
108
+ };
109
+ //# sourceMappingURL=policy.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"policy.js","sourceRoot":"","sources":["../../src/agent/policy.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAmC,MAAM,0BAA0B,CAAA;AACrF,OAAO,EACL,YAAY,EAIb,MAAM,uBAAuB,CAAA;AAC9B,OAAO,EACL,UAAU,EAIX,MAAM,wBAAwB,CAAA;AAE/B,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAA;AAU7D,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,KAAsB,EAAqB,EAAE;IAC5E,IACE,KAAK,CAAC,MAAM,CAAC,YAAY,KAAK,SAAS;QACvC,KAAK,CAAC,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC,MAAM,CAAC,YAAY,EACnD,CAAC;QACD,OAAO,UAAU,CAAC,cAAc,CAAA;IAClC,CAAC;IACD,IACE,KAAK,CAAC,MAAM,CAAC,aAAa,KAAK,SAAS;QACxC,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,MAAM,CAAC,aAAa,EAC7D,CAAC;QACD,OAAO,UAAU,CAAC,aAAa,CAAA;IACjC,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,OAAO;QAC1E,OAAO,UAAU,CAAC,QAAQ,CAAA;IAC5B,IACE,KAAK,CAAC,MAAM,CAAC,iBAAiB,KAAK,SAAS;QAC5C,KAAK,CAAC,MAAM,CAAC,YAAY,IAAI,KAAK,CAAC,MAAM,CAAC,iBAAiB;QAE3D,OAAO,UAAU,CAAC,gBAAgB,CAAA;IACpC,OAAO,IAAI,CAAA;AACb,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAC1C,KAAqC,EACrC,KAAoC,EACjB,EAAE;IACrB,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC;QAC5F,OAAO;YACL,OAAO,EACL,iEAAiE;gBACjE,6EAA6E;YAC/E,OAAO,EAAE,mBAAmB;YAC5B,QAAQ,EAAE,IAAI;YACd,SAAS,EAAE,KAAK;SACjB,CAAA;IACH,CAAC;IACD,OAAO,IAAI,CAAA;AACb,CAAC,CAAA;AAQD,MAAM,CAAC,MAAM,sBAAsB,GAAG,CACpC,KAAqC,EACrC,QAAiC,EACjC,KAAoC,EACL,EAAE;IACjC,MAAM,SAAS,GAAG,4BAA4B,CAAC,KAAK,EAAE,KAAK,CAAC,CAAA;IAC5D,IAAI,SAAS,KAAK,IAAI;QAAE,OAAO,IAAI,CAAA;IACnC,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;IACrF,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAA;IACnE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,CAAA;AAC/F,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,mBAAmB,GAAG,CACjC,OAAe,EACf,WAAyC,EACjC,EAAE;IACV,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,CAAC,CAAA;IACtC,OAAO,WAAW,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,SAAS,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;AACpG,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,sBAAsB,GAAG,CACpC,WAAyC,EACrB,EAAE;IACtB,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;IAChF,IAAI,QAAQ,KAAK,SAAS;QAAE,OAAO,SAAS,CAAA;IAC5C,OAAO,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,SAAS,CAAA;AACvE,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,MAAkB,EAAE,MAAiB,EAAwB,EAAE,CAChG,mBAAmB,CAAC;IAClB,OAAO,EACL,MAAM,CAAC,mBAAmB,EAAE,CAAC,MAAM,CAAC;QACpC,qCAAqC,MAAM,CAAC,UAAU,CAAC,GAAG,EAAE,GAAG,CAAC,2CAA2C;IAC7G,OAAO,EAAE,kBAAkB;IAC3B,QAAQ,EAAE,IAAI;IACd,SAAS,EAAE,KAAK;CACjB,CAAC,CAAA;AAEJ,MAAM,CAAC,MAAM,mBAAmB,GAAG,CACjC,QAAuB,EACvB,KAA8B,EAC9B,MAAkB,EAClB,MAAiB,EACX,EAAE;IACR,MAAM,MAAM,GAAG,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IACjD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,QAAQ,CAAC,IAAI,CAAC;YACZ,UAAU,EAAE,SAAS,CAAC,MAAM,CAAC;YAC7B,IAAI,EAAE,MAAM;YACZ,YAAY,EAAE,IAAI,CAAC,EAAE;YACrB,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC,CAAA;IACJ,CAAC;AACH,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAChC,QAAoC,EACpC,QAAgB,EACY,EAAE;IAC9B,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAA;IACzB,IAAI,KAAK,EAAE,IAAI,KAAK,QAAQ;QAAE,OAAO,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,GAAG,QAAQ,CAAC,CAAA;IACzF,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAA;IAC7B,MAAM,MAAM,GACV,OAAO,OAAO,KAAK,QAAQ;QACzB,CAAC,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,GAAG,OAAO,OAAO,QAAQ,EAAE,EAAE;QACpD,CAAC,CAAC,EAAE,GAAG,KAAK,EAAE,OAAO,EAAE,CAAC,GAAG,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAA;IAC3E,OAAO,CAAC,MAAM,EAAE,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;AACvC,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,QAAoC,EAAe,EAAE;IACnF,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAA;IAChC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACjC,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,YAAY,IAAI,EAAE;gBAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;YAC/D,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,aAAa,IAAI,EAAE;gBAAE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;QACrE,CAAC;QACD,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACjC,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,mBAAmB,IAAI,EAAE,EAAE,CAAC;gBACrD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,iBAAiB,IAAI,EAAE;oBAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;YACnE,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAA;AACf,CAAC,CAAA;AAED,MAAM,CAAC,MAAM,qBAAqB,GAAG,CACnC,KAAqC,EACrC,SAAoD,EACpD,KAAoC,EAChB,EAAE;IACtB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAA;QAC9C,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,QAAQ;YAC3F,SAAQ;QACV,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAA;QAC3D,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,IAAI,CAAA;IAClC,CAAC;IACD,OAAO,SAAS,CAAA;AAClB,CAAC,CAAA"}
@@ -0,0 +1,12 @@
1
+ import { type ToolEntry } from "../contracts/tools.js";
2
+ import type { ModelCapabilities, ModelProvider, ModelRequest } from "../contracts/provider.js";
3
+ export declare const resolveModelCapabilities: (options: {
4
+ readonly provider: ModelProvider;
5
+ readonly model: string;
6
+ }) => Promise<ModelCapabilities>;
7
+ export declare const prepareProviderRequest: (options: {
8
+ readonly provider: ModelProvider;
9
+ readonly request: ModelRequest;
10
+ readonly entries: ReadonlyArray<ToolEntry>;
11
+ }) => Promise<ModelRequest>;
12
+ //# sourceMappingURL=provider-capabilities.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"provider-capabilities.d.ts","sourceRoot":"","sources":["../../src/agent/provider-capabilities.ts"],"names":[],"mappings":"AAIA,OAAO,EAA4C,KAAK,SAAS,EAAE,MAAM,uBAAuB,CAAA;AAChG,OAAO,KAAK,EAAE,iBAAiB,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAA;AAO9F,eAAO,MAAM,wBAAwB,YAAmB;IACtD,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAA;IAChC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAA;CACvB,KAAG,OAAO,CAAC,iBAAiB,CACoD,CAAA;AAEjF,eAAO,MAAM,sBAAsB,YAAmB;IACpD,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAA;IAChC,QAAQ,CAAC,OAAO,EAAE,YAAY,CAAA;IAC9B,QAAQ,CAAC,OAAO,EAAE,aAAa,CAAC,SAAS,CAAC,CAAA;CAC3C,KAAG,OAAO,CAAC,YAAY,CA8EvB,CAAA"}