agents 0.0.0-f913299 → 0.0.0-f9f03b4

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 (77) hide show
  1. package/README.md +257 -33
  2. package/dist/_esm-LV5FJ3HK.js +3922 -0
  3. package/dist/_esm-LV5FJ3HK.js.map +1 -0
  4. package/dist/ai-chat-agent.d.ts +57 -6
  5. package/dist/ai-chat-agent.js +580 -98
  6. package/dist/ai-chat-agent.js.map +1 -1
  7. package/dist/ai-chat-v5-migration.d.ts +152 -0
  8. package/dist/ai-chat-v5-migration.js +20 -0
  9. package/dist/ai-react.d.ts +78 -65
  10. package/dist/ai-react.js +287 -120
  11. package/dist/ai-react.js.map +1 -1
  12. package/dist/ai-types.d.ts +41 -18
  13. package/dist/ai-types.js +7 -0
  14. package/dist/ccip-CMBYN64O.js +15 -0
  15. package/dist/ccip-CMBYN64O.js.map +1 -0
  16. package/dist/chunk-254F4GDT.js +1343 -0
  17. package/dist/chunk-254F4GDT.js.map +1 -0
  18. package/dist/chunk-3OT2NNEW.js +941 -0
  19. package/dist/chunk-3OT2NNEW.js.map +1 -0
  20. package/dist/chunk-5Y6BEZDY.js +276 -0
  21. package/dist/chunk-5Y6BEZDY.js.map +1 -0
  22. package/dist/chunk-BER7KXUJ.js +18 -0
  23. package/dist/chunk-BER7KXUJ.js.map +1 -0
  24. package/dist/chunk-JJBFIGUC.js +5202 -0
  25. package/dist/chunk-JJBFIGUC.js.map +1 -0
  26. package/dist/chunk-PR4QN5HX.js +43 -0
  27. package/dist/chunk-PR4QN5HX.js.map +1 -0
  28. package/dist/chunk-QEVM4BVL.js +116 -0
  29. package/dist/chunk-QEVM4BVL.js.map +1 -0
  30. package/dist/chunk-TYAY6AU6.js +159 -0
  31. package/dist/chunk-TYAY6AU6.js.map +1 -0
  32. package/dist/chunk-UJVEAURM.js +150 -0
  33. package/dist/chunk-UJVEAURM.js.map +1 -0
  34. package/dist/chunk-Z44WASMA.js +114 -0
  35. package/dist/chunk-Z44WASMA.js.map +1 -0
  36. package/dist/client-DVoPb3-C.d.ts +5120 -0
  37. package/dist/client.d.ts +16 -2
  38. package/dist/client.js +8 -133
  39. package/dist/client.js.map +1 -1
  40. package/dist/codemode/ai.d.ts +25 -0
  41. package/dist/codemode/ai.js +5112 -0
  42. package/dist/codemode/ai.js.map +1 -0
  43. package/dist/index.d.ts +277 -22
  44. package/dist/index.js +16 -4
  45. package/dist/mcp/client.d.ts +12 -0
  46. package/dist/mcp/client.js +10 -0
  47. package/dist/mcp/client.js.map +1 -0
  48. package/dist/mcp/do-oauth-client-provider.d.ts +42 -0
  49. package/dist/mcp/do-oauth-client-provider.js +8 -0
  50. package/dist/mcp/do-oauth-client-provider.js.map +1 -0
  51. package/dist/mcp/index.d.ts +79 -0
  52. package/dist/mcp/index.js +1099 -0
  53. package/dist/mcp/index.js.map +1 -0
  54. package/dist/mcp/x402.d.ts +39 -0
  55. package/dist/mcp/x402.js +3195 -0
  56. package/dist/mcp/x402.js.map +1 -0
  57. package/dist/mcp-BH1fJeiU.d.ts +58 -0
  58. package/dist/observability/index.d.ts +34 -0
  59. package/dist/observability/index.js +12 -0
  60. package/dist/observability/index.js.map +1 -0
  61. package/dist/react.d.ts +97 -9
  62. package/dist/react.js +153 -34
  63. package/dist/react.js.map +1 -1
  64. package/dist/schedule.d.ts +81 -7
  65. package/dist/schedule.js +20 -7
  66. package/dist/schedule.js.map +1 -1
  67. package/dist/secp256k1-M22GZP2U.js +2193 -0
  68. package/dist/secp256k1-M22GZP2U.js.map +1 -0
  69. package/dist/serializable.d.ts +32 -0
  70. package/dist/serializable.js +1 -0
  71. package/dist/serializable.js.map +1 -0
  72. package/package.json +107 -43
  73. package/src/index.ts +1327 -184
  74. package/dist/chunk-HMLY7DHA.js +0 -16
  75. package/dist/chunk-X6BBKLSC.js +0 -568
  76. package/dist/chunk-X6BBKLSC.js.map +0 -1
  77. /package/dist/{chunk-HMLY7DHA.js.map → ai-chat-v5-migration.js.map} +0 -0
@@ -1,86 +1,138 @@
1
1
  import {
2
- Agent
3
- } from "./chunk-X6BBKLSC.js";
2
+ autoTransformMessages
3
+ } from "./chunk-UJVEAURM.js";
4
4
  import {
5
- __privateAdd,
6
- __privateMethod
7
- } from "./chunk-HMLY7DHA.js";
5
+ Agent
6
+ } from "./chunk-254F4GDT.js";
7
+ import "./chunk-3OT2NNEW.js";
8
+ import "./chunk-Z44WASMA.js";
9
+ import "./chunk-QEVM4BVL.js";
10
+ import "./chunk-BER7KXUJ.js";
11
+ import "./chunk-PR4QN5HX.js";
8
12
 
9
13
  // src/ai-chat-agent.ts
10
- import { appendResponseMessages } from "ai";
14
+ import {
15
+ getToolName,
16
+ isToolUIPart,
17
+ parsePartialJson
18
+ } from "ai";
11
19
  var decoder = new TextDecoder();
12
- var _AIChatAgent_instances, sendChatMessage_fn, broadcastChatMessage_fn, tryCatch_fn, persistMessages_fn, reply_fn;
13
20
  var AIChatAgent = class extends Agent {
14
21
  constructor(ctx, env) {
15
22
  super(ctx, env);
16
- __privateAdd(this, _AIChatAgent_instances);
17
23
  this.sql`create table if not exists cf_ai_chat_agent_messages (
18
24
  id text primary key,
19
25
  message text not null,
20
26
  created_at datetime default current_timestamp
21
27
  )`;
22
- this.messages = (this.sql`select * from cf_ai_chat_agent_messages` || []).map((row) => {
28
+ const rawMessages = (this.sql`select * from cf_ai_chat_agent_messages` || []).map((row) => {
23
29
  return JSON.parse(row.message);
24
30
  });
31
+ this.messages = autoTransformMessages(rawMessages);
32
+ this._chatMessageAbortControllers = /* @__PURE__ */ new Map();
33
+ }
34
+ _broadcastChatMessage(message, exclude) {
35
+ this.broadcast(JSON.stringify(message), exclude);
25
36
  }
26
37
  async onMessage(connection, message) {
27
38
  if (typeof message === "string") {
28
39
  let data;
29
40
  try {
30
41
  data = JSON.parse(message);
31
- } catch (error) {
42
+ } catch (_error) {
32
43
  return;
33
44
  }
34
- if (data.type === "cf_agent_use_chat_request" && data.init.method === "POST") {
45
+ if (data.type === "cf_agent_use_chat_request" /* CF_AGENT_USE_CHAT_REQUEST */ && data.init.method === "POST") {
35
46
  const {
36
- method,
37
- keepalive,
38
- headers,
39
- body,
47
+ // method,
48
+ // keepalive,
49
+ // headers,
50
+ body
40
51
  // we're reading this
41
- redirect,
42
- integrity,
43
- credentials,
44
- mode,
45
- referrer,
46
- referrerPolicy,
47
- window
52
+ //
53
+ // // these might not exist?
48
54
  // dispatcher,
49
55
  // duplex
50
56
  } = data.init;
51
57
  const { messages } = JSON.parse(body);
52
- __privateMethod(this, _AIChatAgent_instances, broadcastChatMessage_fn).call(this, {
53
- type: "cf_agent_chat_messages",
54
- messages
55
- }, [connection.id]);
56
- await __privateMethod(this, _AIChatAgent_instances, persistMessages_fn).call(this, messages, [connection.id]);
57
- return __privateMethod(this, _AIChatAgent_instances, tryCatch_fn).call(this, async () => {
58
- const response = await this.onChatMessage(async ({ response: response2 }) => {
59
- const finalMessages = appendResponseMessages({
60
- messages,
61
- responseMessages: response2.messages
62
- });
63
- await __privateMethod(this, _AIChatAgent_instances, persistMessages_fn).call(this, finalMessages, [connection.id]);
64
- });
58
+ const transformedMessages = autoTransformMessages(messages);
59
+ this._broadcastChatMessage(
60
+ {
61
+ messages: transformedMessages,
62
+ type: "cf_agent_chat_messages" /* CF_AGENT_CHAT_MESSAGES */
63
+ },
64
+ [connection.id]
65
+ );
66
+ await this.persistMessages(transformedMessages, [connection.id]);
67
+ this.observability?.emit(
68
+ {
69
+ displayMessage: "Chat message request",
70
+ id: data.id,
71
+ payload: {},
72
+ timestamp: Date.now(),
73
+ type: "message:request"
74
+ },
75
+ this.ctx
76
+ );
77
+ const chatMessageId = data.id;
78
+ const abortSignal = this._getAbortSignal(chatMessageId);
79
+ return this._tryCatchChat(async () => {
80
+ const response = await this.onChatMessage(
81
+ async (_finishResult) => {
82
+ this._removeAbortController(chatMessageId);
83
+ this.observability?.emit(
84
+ {
85
+ displayMessage: "Chat message response",
86
+ id: data.id,
87
+ payload: {},
88
+ timestamp: Date.now(),
89
+ type: "message:response"
90
+ },
91
+ this.ctx
92
+ );
93
+ },
94
+ abortSignal ? { abortSignal } : void 0
95
+ );
65
96
  if (response) {
66
- await __privateMethod(this, _AIChatAgent_instances, reply_fn).call(this, data.id, response);
97
+ await this._reply(data.id, response);
98
+ } else {
99
+ console.warn(
100
+ `[AIChatAgent] onChatMessage returned no response for chatMessageId: ${chatMessageId}`
101
+ );
102
+ this._broadcastChatMessage(
103
+ {
104
+ body: "No response was generated by the agent.",
105
+ done: true,
106
+ id: data.id,
107
+ type: "cf_agent_use_chat_response" /* CF_AGENT_USE_CHAT_RESPONSE */
108
+ },
109
+ [connection.id]
110
+ );
67
111
  }
68
112
  });
69
113
  }
70
- if (data.type === "cf_agent_chat_clear") {
114
+ if (data.type === "cf_agent_chat_clear" /* CF_AGENT_CHAT_CLEAR */) {
115
+ this._destroyAbortControllers();
71
116
  this.sql`delete from cf_ai_chat_agent_messages`;
72
117
  this.messages = [];
73
- __privateMethod(this, _AIChatAgent_instances, broadcastChatMessage_fn).call(this, {
74
- type: "cf_agent_chat_clear"
75
- }, [connection.id]);
76
- } else if (data.type === "cf_agent_chat_messages") {
77
- await __privateMethod(this, _AIChatAgent_instances, persistMessages_fn).call(this, data.messages, [connection.id]);
118
+ this._broadcastChatMessage(
119
+ {
120
+ type: "cf_agent_chat_clear" /* CF_AGENT_CHAT_CLEAR */
121
+ },
122
+ [connection.id]
123
+ );
124
+ } else if (data.type === "cf_agent_chat_messages" /* CF_AGENT_CHAT_MESSAGES */) {
125
+ const transformedMessages = autoTransformMessages(data.messages);
126
+ await this.persistMessages(transformedMessages, [connection.id]);
127
+ } else if (data.type === "cf_agent_chat_request_cancel" /* CF_AGENT_CHAT_REQUEST_CANCEL */) {
128
+ this._cancelChatRequest(data.id);
78
129
  }
79
130
  }
80
131
  }
81
132
  async onRequest(request) {
82
- return __privateMethod(this, _AIChatAgent_instances, tryCatch_fn).call(this, () => {
83
- if (request.url.endsWith("/get-messages")) {
133
+ return this._tryCatchChat(() => {
134
+ const url = new URL(request.url);
135
+ if (url.pathname.endsWith("/get-messages")) {
84
136
  const messages = (this.sql`select * from cf_ai_chat_agent_messages` || []).map((row) => {
85
137
  return JSON.parse(row.message);
86
138
  });
@@ -89,80 +141,510 @@ var AIChatAgent = class extends Agent {
89
141
  return super.onRequest(request);
90
142
  });
91
143
  }
144
+ async _tryCatchChat(fn) {
145
+ try {
146
+ return await fn();
147
+ } catch (e) {
148
+ throw this.onError(e);
149
+ }
150
+ }
92
151
  /**
93
152
  * Handle incoming chat messages and generate a response
94
153
  * @param onFinish Callback to be called when the response is finished
154
+ * @param options.signal A signal to pass to any child requests which can be used to cancel them
95
155
  * @returns Response to send to the client or undefined
96
156
  */
97
- async onChatMessage(onFinish) {
157
+ async onChatMessage(onFinish, options) {
98
158
  throw new Error(
99
159
  "recieved a chat message, override onChatMessage and return a Response to send to the client"
100
160
  );
101
161
  }
102
162
  /**
103
- * Save messages on the server side and trigger AI response
163
+ * Save messages on the server side
104
164
  * @param messages Chat messages to save
105
165
  */
106
166
  async saveMessages(messages) {
107
- await __privateMethod(this, _AIChatAgent_instances, persistMessages_fn).call(this, messages);
108
- const response = await this.onChatMessage(async ({ response: response2 }) => {
109
- const finalMessages = appendResponseMessages({
167
+ await this.persistMessages(messages);
168
+ }
169
+ async persistMessages(messages, excludeBroadcastIds = []) {
170
+ this.sql`delete from cf_ai_chat_agent_messages`;
171
+ for (const message of messages) {
172
+ this.sql`insert into cf_ai_chat_agent_messages (id, message) values (${message.id},${JSON.stringify(message)})`;
173
+ }
174
+ this.messages = messages;
175
+ this._broadcastChatMessage(
176
+ {
110
177
  messages,
111
- responseMessages: response2.messages
112
- });
113
- await __privateMethod(this, _AIChatAgent_instances, persistMessages_fn).call(this, finalMessages, []);
114
- });
115
- if (response) {
116
- for await (const chunk of response.body) {
117
- decoder.decode(chunk);
178
+ type: "cf_agent_chat_messages" /* CF_AGENT_CHAT_MESSAGES */
179
+ },
180
+ excludeBroadcastIds
181
+ );
182
+ }
183
+ async _reply(id, response) {
184
+ return this._tryCatchChat(async () => {
185
+ if (!response.body) {
186
+ this._broadcastChatMessage({
187
+ body: "",
188
+ done: true,
189
+ id,
190
+ type: "cf_agent_use_chat_response" /* CF_AGENT_USE_CHAT_RESPONSE */
191
+ });
192
+ return;
193
+ }
194
+ const reader = response.body.getReader();
195
+ const message = {
196
+ id: `assistant_${Date.now()}_${Math.random().toString(36).slice(2, 11)}`,
197
+ // default
198
+ role: "assistant",
199
+ parts: []
200
+ };
201
+ let activeTextParts = {};
202
+ let activeReasoningParts = {};
203
+ const partialToolCalls = {};
204
+ function updateDynamicToolPart(options) {
205
+ const part = message.parts.find(
206
+ (part2) => part2.type === "dynamic-tool" && part2.toolCallId === options.toolCallId
207
+ );
208
+ const anyOptions = options;
209
+ const anyPart = part;
210
+ if (part != null) {
211
+ part.state = options.state;
212
+ anyPart.toolName = options.toolName;
213
+ anyPart.input = anyOptions.input;
214
+ anyPart.output = anyOptions.output;
215
+ anyPart.errorText = anyOptions.errorText;
216
+ anyPart.rawInput = anyOptions.rawInput ?? anyPart.rawInput;
217
+ anyPart.preliminary = anyOptions.preliminary;
218
+ if (anyOptions.providerMetadata != null && part.state === "input-available") {
219
+ part.callProviderMetadata = anyOptions.providerMetadata;
220
+ }
221
+ } else {
222
+ message.parts.push({
223
+ type: "dynamic-tool",
224
+ toolName: options.toolName,
225
+ toolCallId: options.toolCallId,
226
+ state: options.state,
227
+ input: anyOptions.input,
228
+ output: anyOptions.output,
229
+ errorText: anyOptions.errorText,
230
+ preliminary: anyOptions.preliminary,
231
+ ...anyOptions.providerMetadata != null ? { callProviderMetadata: anyOptions.providerMetadata } : {}
232
+ });
233
+ }
234
+ }
235
+ function updateToolPart(options) {
236
+ const part = message.parts.find(
237
+ (part2) => isToolUIPart(part2) && part2.toolCallId === options.toolCallId
238
+ );
239
+ const anyOptions = options;
240
+ const anyPart = part;
241
+ if (part != null) {
242
+ part.state = options.state;
243
+ anyPart.input = anyOptions.input;
244
+ anyPart.output = anyOptions.output;
245
+ anyPart.errorText = anyOptions.errorText;
246
+ anyPart.rawInput = anyOptions.rawInput;
247
+ anyPart.preliminary = anyOptions.preliminary;
248
+ anyPart.providerExecuted = anyOptions.providerExecuted ?? part.providerExecuted;
249
+ if (anyOptions.providerMetadata != null && part.state === "input-available") {
250
+ part.callProviderMetadata = anyOptions.providerMetadata;
251
+ }
252
+ } else {
253
+ message.parts.push({
254
+ type: `tool-${options.toolName}`,
255
+ toolCallId: options.toolCallId,
256
+ state: options.state,
257
+ input: anyOptions.input,
258
+ output: anyOptions.output,
259
+ rawInput: anyOptions.rawInput,
260
+ errorText: anyOptions.errorText,
261
+ providerExecuted: anyOptions.providerExecuted,
262
+ preliminary: anyOptions.preliminary,
263
+ ...anyOptions.providerMetadata != null ? { callProviderMetadata: anyOptions.providerMetadata } : {}
264
+ });
265
+ }
266
+ }
267
+ async function updateMessageMetadata(metadata) {
268
+ if (metadata != null) {
269
+ const mergedMetadata = message.metadata != null ? { ...message.metadata, ...metadata } : metadata;
270
+ message.metadata = mergedMetadata;
271
+ }
272
+ }
273
+ try {
274
+ while (true) {
275
+ const { done, value } = await reader.read();
276
+ if (done) {
277
+ this._broadcastChatMessage({
278
+ body: "",
279
+ done: true,
280
+ id,
281
+ type: "cf_agent_use_chat_response" /* CF_AGENT_USE_CHAT_RESPONSE */
282
+ });
283
+ break;
284
+ }
285
+ const chunk = decoder.decode(value);
286
+ const contentType = response.headers.get("content-type") || "";
287
+ const isSSE = contentType.includes("text/event-stream");
288
+ if (isSSE) {
289
+ const lines = chunk.split("\n");
290
+ for (const line of lines) {
291
+ if (line.startsWith("data: ") && line !== "data: [DONE]") {
292
+ try {
293
+ const data = JSON.parse(line.slice(6));
294
+ switch (data.type) {
295
+ case "text-start": {
296
+ const textPart = {
297
+ type: "text",
298
+ text: "",
299
+ providerMetadata: data.providerMetadata,
300
+ state: "streaming"
301
+ };
302
+ activeTextParts[data.id] = textPart;
303
+ message.parts.push(textPart);
304
+ break;
305
+ }
306
+ case "text-delta": {
307
+ const textPart = activeTextParts[data.id];
308
+ textPart.text += data.delta;
309
+ textPart.providerMetadata = data.providerMetadata ?? textPart.providerMetadata;
310
+ break;
311
+ }
312
+ case "text-end": {
313
+ const textPart = activeTextParts[data.id];
314
+ textPart.state = "done";
315
+ textPart.providerMetadata = data.providerMetadata ?? textPart.providerMetadata;
316
+ delete activeTextParts[data.id];
317
+ break;
318
+ }
319
+ case "reasoning-start": {
320
+ const reasoningPart = {
321
+ type: "reasoning",
322
+ text: "",
323
+ providerMetadata: data.providerMetadata,
324
+ state: "streaming"
325
+ };
326
+ activeReasoningParts[data.id] = reasoningPart;
327
+ message.parts.push(reasoningPart);
328
+ break;
329
+ }
330
+ case "reasoning-delta": {
331
+ const reasoningPart = activeReasoningParts[data.id];
332
+ reasoningPart.text += data.delta;
333
+ reasoningPart.providerMetadata = data.providerMetadata ?? reasoningPart.providerMetadata;
334
+ break;
335
+ }
336
+ case "reasoning-end": {
337
+ const reasoningPart = activeReasoningParts[data.id];
338
+ reasoningPart.providerMetadata = data.providerMetadata ?? reasoningPart.providerMetadata;
339
+ reasoningPart.state = "done";
340
+ delete activeReasoningParts[data.id];
341
+ break;
342
+ }
343
+ case "file": {
344
+ message.parts.push({
345
+ type: "file",
346
+ mediaType: data.mediaType,
347
+ url: data.url
348
+ });
349
+ break;
350
+ }
351
+ case "source-url": {
352
+ message.parts.push({
353
+ type: "source-url",
354
+ sourceId: data.sourceId,
355
+ url: data.url,
356
+ title: data.title,
357
+ providerMetadata: data.providerMetadata
358
+ });
359
+ break;
360
+ }
361
+ case "source-document": {
362
+ message.parts.push({
363
+ type: "source-document",
364
+ sourceId: data.sourceId,
365
+ mediaType: data.mediaType,
366
+ title: data.title,
367
+ filename: data.filename,
368
+ providerMetadata: data.providerMetadata
369
+ });
370
+ break;
371
+ }
372
+ case "tool-input-start": {
373
+ const toolInvocations = message.parts.filter(isToolUIPart);
374
+ partialToolCalls[data.toolCallId] = {
375
+ text: "",
376
+ toolName: data.toolName,
377
+ index: toolInvocations.length,
378
+ dynamic: data.dynamic
379
+ };
380
+ if (data.dynamic) {
381
+ updateDynamicToolPart({
382
+ toolCallId: data.toolCallId,
383
+ toolName: data.toolName,
384
+ state: "input-streaming",
385
+ input: void 0
386
+ });
387
+ } else {
388
+ updateToolPart({
389
+ toolCallId: data.toolCallId,
390
+ toolName: data.toolName,
391
+ state: "input-streaming",
392
+ input: void 0
393
+ });
394
+ }
395
+ break;
396
+ }
397
+ case "tool-input-delta": {
398
+ const partialToolCall = partialToolCalls[data.toolCallId];
399
+ partialToolCall.text += data.inputTextDelta;
400
+ const { value: partialArgs } = await parsePartialJson(
401
+ partialToolCall.text
402
+ );
403
+ if (partialToolCall.dynamic) {
404
+ updateDynamicToolPart({
405
+ toolCallId: data.toolCallId,
406
+ toolName: partialToolCall.toolName,
407
+ state: "input-streaming",
408
+ input: partialArgs
409
+ });
410
+ } else {
411
+ updateToolPart({
412
+ toolCallId: data.toolCallId,
413
+ toolName: partialToolCall.toolName,
414
+ state: "input-streaming",
415
+ input: partialArgs
416
+ });
417
+ }
418
+ break;
419
+ }
420
+ case "tool-input-available": {
421
+ if (data.dynamic) {
422
+ updateDynamicToolPart({
423
+ toolCallId: data.toolCallId,
424
+ toolName: data.toolName,
425
+ state: "input-available",
426
+ input: data.input,
427
+ providerMetadata: data.providerMetadata
428
+ });
429
+ } else {
430
+ updateToolPart({
431
+ toolCallId: data.toolCallId,
432
+ toolName: data.toolName,
433
+ state: "input-available",
434
+ input: data.input,
435
+ providerExecuted: data.providerExecuted,
436
+ providerMetadata: data.providerMetadata
437
+ });
438
+ }
439
+ break;
440
+ }
441
+ case "tool-input-error": {
442
+ if (data.dynamic) {
443
+ updateDynamicToolPart({
444
+ toolCallId: data.toolCallId,
445
+ toolName: data.toolName,
446
+ state: "output-error",
447
+ input: data.input,
448
+ errorText: data.errorText,
449
+ providerMetadata: data.providerMetadata
450
+ });
451
+ } else {
452
+ updateToolPart({
453
+ toolCallId: data.toolCallId,
454
+ toolName: data.toolName,
455
+ state: "output-error",
456
+ input: void 0,
457
+ rawInput: data.input,
458
+ errorText: data.errorText,
459
+ providerExecuted: data.providerExecuted,
460
+ providerMetadata: data.providerMetadata
461
+ });
462
+ }
463
+ break;
464
+ }
465
+ case "tool-output-available": {
466
+ if (data.dynamic) {
467
+ const toolInvocations = message.parts.filter(
468
+ (part) => part.type === "dynamic-tool"
469
+ );
470
+ const toolInvocation = toolInvocations.find(
471
+ (invocation) => invocation.toolCallId === data.toolCallId
472
+ );
473
+ if (!toolInvocation)
474
+ throw new Error("Tool invocation not found");
475
+ updateDynamicToolPart({
476
+ toolCallId: data.toolCallId,
477
+ toolName: toolInvocation.toolName,
478
+ state: "output-available",
479
+ input: toolInvocation.input,
480
+ output: data.output,
481
+ preliminary: data.preliminary
482
+ });
483
+ } else {
484
+ const toolInvocations = message.parts.filter(isToolUIPart);
485
+ const toolInvocation = toolInvocations.find(
486
+ (invocation) => invocation.toolCallId === data.toolCallId
487
+ );
488
+ if (!toolInvocation)
489
+ throw new Error("Tool invocation not found");
490
+ updateToolPart({
491
+ toolCallId: data.toolCallId,
492
+ toolName: getToolName(toolInvocation),
493
+ state: "output-available",
494
+ input: toolInvocation.input,
495
+ output: data.output,
496
+ providerExecuted: data.providerExecuted,
497
+ preliminary: data.preliminary
498
+ });
499
+ }
500
+ break;
501
+ }
502
+ case "tool-output-error": {
503
+ if (data.dynamic) {
504
+ const toolInvocations = message.parts.filter(
505
+ (part) => part.type === "dynamic-tool"
506
+ );
507
+ const toolInvocation = toolInvocations.find(
508
+ (invocation) => invocation.toolCallId === data.toolCallId
509
+ );
510
+ if (!toolInvocation)
511
+ throw new Error("Tool invocation not found");
512
+ updateDynamicToolPart({
513
+ toolCallId: data.toolCallId,
514
+ toolName: toolInvocation.toolName,
515
+ state: "output-error",
516
+ input: toolInvocation.input,
517
+ errorText: data.errorText
518
+ });
519
+ } else {
520
+ const toolInvocations = message.parts.filter(isToolUIPart);
521
+ const toolInvocation = toolInvocations.find(
522
+ (invocation) => invocation.toolCallId === data.toolCallId
523
+ );
524
+ if (!toolInvocation)
525
+ throw new Error("Tool invocation not found");
526
+ updateToolPart({
527
+ toolCallId: data.toolCallId,
528
+ toolName: getToolName(toolInvocation),
529
+ state: "output-error",
530
+ input: toolInvocation.input,
531
+ rawInput: "rawInput" in toolInvocation ? toolInvocation.rawInput : void 0,
532
+ errorText: data.errorText
533
+ });
534
+ }
535
+ break;
536
+ }
537
+ case "start-step": {
538
+ message.parts.push({ type: "step-start" });
539
+ break;
540
+ }
541
+ case "finish-step": {
542
+ activeTextParts = {};
543
+ activeReasoningParts = {};
544
+ break;
545
+ }
546
+ case "start": {
547
+ if (data.messageId != null) {
548
+ message.id = data.messageId;
549
+ }
550
+ await updateMessageMetadata(data.messageMetadata);
551
+ break;
552
+ }
553
+ case "finish": {
554
+ await updateMessageMetadata(data.messageMetadata);
555
+ break;
556
+ }
557
+ case "message-metadata": {
558
+ await updateMessageMetadata(data.messageMetadata);
559
+ break;
560
+ }
561
+ case "error": {
562
+ this._broadcastChatMessage({
563
+ error: true,
564
+ body: data.errorText ?? JSON.stringify(data),
565
+ done: false,
566
+ id,
567
+ type: "cf_agent_use_chat_response" /* CF_AGENT_USE_CHAT_RESPONSE */
568
+ });
569
+ break;
570
+ }
571
+ }
572
+ this._broadcastChatMessage({
573
+ body: JSON.stringify(data),
574
+ done: false,
575
+ id,
576
+ type: "cf_agent_use_chat_response" /* CF_AGENT_USE_CHAT_RESPONSE */
577
+ });
578
+ } catch (_e) {
579
+ }
580
+ }
581
+ }
582
+ } else {
583
+ if (chunk.length > 0) {
584
+ message.parts.push({ type: "text", text: chunk });
585
+ this._broadcastChatMessage({
586
+ body: JSON.stringify({ type: "text-delta", delta: chunk }),
587
+ done: false,
588
+ id,
589
+ type: "cf_agent_use_chat_response" /* CF_AGENT_USE_CHAT_RESPONSE */
590
+ });
591
+ }
592
+ }
593
+ }
594
+ } finally {
595
+ reader.releaseLock();
118
596
  }
119
- response.body?.cancel();
597
+ if (message.parts.length > 0) {
598
+ await this.persistMessages([...this.messages, message]);
599
+ }
600
+ });
601
+ }
602
+ /**
603
+ * For the given message id, look up its associated AbortController
604
+ * If the AbortController does not exist, create and store one in memory
605
+ *
606
+ * returns the AbortSignal associated with the AbortController
607
+ */
608
+ _getAbortSignal(id) {
609
+ if (typeof id !== "string") {
610
+ return void 0;
120
611
  }
612
+ if (!this._chatMessageAbortControllers.has(id)) {
613
+ this._chatMessageAbortControllers.set(id, new AbortController());
614
+ }
615
+ return this._chatMessageAbortControllers.get(id)?.signal;
121
616
  }
122
- };
123
- _AIChatAgent_instances = new WeakSet();
124
- sendChatMessage_fn = function(connection, message) {
125
- connection.send(JSON.stringify(message));
126
- };
127
- broadcastChatMessage_fn = function(message, exclude) {
128
- this.broadcast(JSON.stringify(message), exclude);
129
- };
130
- tryCatch_fn = async function(fn) {
131
- try {
132
- return await fn();
133
- } catch (e) {
134
- throw this.onError(e);
617
+ /**
618
+ * Remove an abort controller from the cache of pending message responses
619
+ */
620
+ _removeAbortController(id) {
621
+ this._chatMessageAbortControllers.delete(id);
135
622
  }
136
- };
137
- persistMessages_fn = async function(messages, excludeBroadcastIds = []) {
138
- this.sql`delete from cf_ai_chat_agent_messages`;
139
- for (const message of messages) {
140
- this.sql`insert into cf_ai_chat_agent_messages (id, message) values (${message.id},${JSON.stringify(message)})`;
623
+ /**
624
+ * Propagate an abort signal for any requests associated with the given message id
625
+ */
626
+ _cancelChatRequest(id) {
627
+ if (this._chatMessageAbortControllers.has(id)) {
628
+ const abortController = this._chatMessageAbortControllers.get(id);
629
+ abortController?.abort();
630
+ }
141
631
  }
142
- this.messages = messages;
143
- __privateMethod(this, _AIChatAgent_instances, broadcastChatMessage_fn).call(this, {
144
- type: "cf_agent_chat_messages",
145
- messages
146
- }, excludeBroadcastIds);
147
- };
148
- reply_fn = async function(id, response) {
149
- return __privateMethod(this, _AIChatAgent_instances, tryCatch_fn).call(this, async () => {
150
- for await (const chunk of response.body) {
151
- const body = decoder.decode(chunk);
152
- __privateMethod(this, _AIChatAgent_instances, broadcastChatMessage_fn).call(this, {
153
- id,
154
- type: "cf_agent_use_chat_response",
155
- body,
156
- done: false
157
- });
632
+ /**
633
+ * Abort all pending requests and clear the cache of AbortControllers
634
+ */
635
+ _destroyAbortControllers() {
636
+ for (const controller of this._chatMessageAbortControllers.values()) {
637
+ controller?.abort();
158
638
  }
159
- __privateMethod(this, _AIChatAgent_instances, broadcastChatMessage_fn).call(this, {
160
- id,
161
- type: "cf_agent_use_chat_response",
162
- body: "",
163
- done: true
164
- });
165
- });
639
+ this._chatMessageAbortControllers.clear();
640
+ }
641
+ /**
642
+ * When the DO is destroyed, cancel all pending requests
643
+ */
644
+ async destroy() {
645
+ this._destroyAbortControllers();
646
+ await super.destroy();
647
+ }
166
648
  };
167
649
  export {
168
650
  AIChatAgent