agents 0.2.2 → 0.2.4

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.
@@ -4,9 +4,10 @@ import { Connection, WSMessage } from "partyserver";
4
4
  import "cloudflare:workers";
5
5
  import "@modelcontextprotocol/sdk/client/index.js";
6
6
  import "@modelcontextprotocol/sdk/types.js";
7
- import "./client-BohGLma8.js";
7
+ import "./client-DVoPb3-C.js";
8
8
  import "zod";
9
9
  import "@modelcontextprotocol/sdk/shared/protocol.js";
10
+ import "./mcp-BH1fJeiU.js";
10
11
  import "@modelcontextprotocol/sdk/client/sse.js";
11
12
  import "@modelcontextprotocol/sdk/client/streamableHttp.js";
12
13
  import "./mcp/do-oauth-client-provider.js";
@@ -3,14 +3,19 @@ import {
3
3
  } from "./chunk-UJVEAURM.js";
4
4
  import {
5
5
  Agent
6
- } from "./chunk-RS5OCNEQ.js";
7
- import "./chunk-QEPGNUG6.js";
8
- import "./chunk-XFS5ERG3.js";
6
+ } from "./chunk-254F4GDT.js";
7
+ import "./chunk-3OT2NNEW.js";
8
+ import "./chunk-Z44WASMA.js";
9
9
  import "./chunk-QEVM4BVL.js";
10
10
  import "./chunk-BER7KXUJ.js";
11
11
  import "./chunk-PR4QN5HX.js";
12
12
 
13
13
  // src/ai-chat-agent.ts
14
+ import {
15
+ getToolName,
16
+ isToolUIPart,
17
+ parsePartialJson
18
+ } from "ai";
14
19
  var decoder = new TextDecoder();
15
20
  var AIChatAgent = class extends Agent {
16
21
  constructor(ctx, env) {
@@ -187,9 +192,84 @@ var AIChatAgent = class extends Agent {
187
192
  return;
188
193
  }
189
194
  const reader = response.body.getReader();
190
- let fullResponseText = "";
191
- let fullReasoningText = "";
192
- const toolCalls = /* @__PURE__ */ new Map();
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
+ }
193
273
  try {
194
274
  while (true) {
195
275
  const { done, value } = await reader.read();
@@ -212,33 +292,270 @@ var AIChatAgent = class extends Agent {
212
292
  try {
213
293
  const data = JSON.parse(line.slice(6));
214
294
  switch (data.type) {
215
- // SSE event signaling the tool input is ready. We track by
216
- // `toolCallId` so we can persist it as a tool part in the message.
217
- case "tool-input-available": {
218
- const { toolCallId, toolName, input } = data;
219
- toolCalls.set(toolCallId, {
220
- toolCallId,
221
- toolName,
222
- input,
223
- type: toolName ? `tool-${toolName}` : "dynamic-tool",
224
- state: "input-available"
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
225
358
  });
226
359
  break;
227
360
  }
228
- // SSE event signaling the tool output is ready. We should've
229
- // already received the input in a previous event so an entry
230
- // with `toolCallId` should already be present
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
+ }
231
465
  case "tool-output-available": {
232
- const { toolCallId, output, isError, errorText } = data;
233
- const toolPart = toolCalls.get(toolCallId);
234
- if (toolPart)
235
- toolCalls.set(toolCallId, {
236
- ...toolPart,
237
- output,
238
- isError,
239
- errorText,
240
- state: "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
241
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);
242
559
  break;
243
560
  }
244
561
  case "error": {
@@ -249,14 +566,6 @@ var AIChatAgent = class extends Agent {
249
566
  id,
250
567
  type: "cf_agent_use_chat_response" /* CF_AGENT_USE_CHAT_RESPONSE */
251
568
  });
252
- return;
253
- }
254
- case "reasoning-delta": {
255
- if (data.delta) fullReasoningText += data.delta;
256
- break;
257
- }
258
- case "text-delta": {
259
- if (data.delta) fullResponseText += data.delta;
260
569
  break;
261
570
  }
262
571
  }
@@ -272,7 +581,7 @@ var AIChatAgent = class extends Agent {
272
581
  }
273
582
  } else {
274
583
  if (chunk.length > 0) {
275
- fullResponseText += chunk;
584
+ message.parts.push({ type: "text", text: chunk });
276
585
  this._broadcastChatMessage({
277
586
  body: JSON.stringify({ type: "text-delta", delta: chunk }),
278
587
  done: false,
@@ -285,25 +594,8 @@ var AIChatAgent = class extends Agent {
285
594
  } finally {
286
595
  reader.releaseLock();
287
596
  }
288
- const messageParts = [];
289
- Array.from(toolCalls.values()).forEach((t) => {
290
- messageParts.push(t);
291
- });
292
- if (fullReasoningText.trim()) {
293
- messageParts.push({ type: "reasoning", text: fullReasoningText });
294
- }
295
- if (fullResponseText.trim()) {
296
- messageParts.push({ type: "text", text: fullResponseText });
297
- }
298
- if (messageParts.length > 0) {
299
- await this.persistMessages([
300
- ...this.messages,
301
- {
302
- id: `assistant_${Date.now()}_${Math.random().toString(36).slice(2, 11)}`,
303
- role: "assistant",
304
- parts: messageParts
305
- }
306
- ]);
597
+ if (message.parts.length > 0) {
598
+ await this.persistMessages([...this.messages, message]);
307
599
  }
308
600
  });
309
601
  }
@@ -1 +1 @@
1
- {"version":3,"sources":["../src/ai-chat-agent.ts"],"sourcesContent":["import type {\n UIMessage as ChatMessage,\n StreamTextOnFinishCallback,\n ToolSet\n} from \"ai\";\nimport { Agent, type AgentContext, type Connection, type WSMessage } from \"./\";\nimport {\n MessageType,\n type IncomingMessage,\n type OutgoingMessage\n} from \"./ai-types\";\nimport { autoTransformMessages } from \"./ai-chat-v5-migration\";\n\nconst decoder = new TextDecoder();\n\n/**\n * Extension of Agent with built-in chat capabilities\n * @template Env Environment type containing bindings\n */\nexport class AIChatAgent<Env = unknown, State = unknown> extends Agent<\n Env,\n State\n> {\n /**\n * Map of message `id`s to `AbortController`s\n * useful to propagate request cancellation signals for any external calls made by the agent\n */\n private _chatMessageAbortControllers: Map<string, AbortController>;\n /** Array of chat messages for the current conversation */\n messages: ChatMessage[];\n constructor(ctx: AgentContext, env: Env) {\n super(ctx, env);\n this.sql`create table if not exists cf_ai_chat_agent_messages (\n id text primary key,\n message text not null,\n created_at datetime default current_timestamp\n )`;\n\n // Load messages and automatically transform them to v5 format\n const rawMessages = (\n this.sql`select * from cf_ai_chat_agent_messages` || []\n ).map((row) => {\n return JSON.parse(row.message as string);\n });\n\n // Automatic migration following https://jhak.im/blog/ai-sdk-migration-handling-previously-saved-messages\n this.messages = autoTransformMessages(rawMessages);\n\n this._chatMessageAbortControllers = new Map();\n }\n\n private _broadcastChatMessage(message: OutgoingMessage, exclude?: string[]) {\n this.broadcast(JSON.stringify(message), exclude);\n }\n\n override async onMessage(connection: Connection, message: WSMessage) {\n if (typeof message === \"string\") {\n let data: IncomingMessage;\n try {\n data = JSON.parse(message) as IncomingMessage;\n } catch (_error) {\n // silently ignore invalid messages for now\n // TODO: log errors with log levels\n return;\n }\n if (\n data.type === MessageType.CF_AGENT_USE_CHAT_REQUEST &&\n data.init.method === \"POST\"\n ) {\n const {\n // method,\n // keepalive,\n // headers,\n body // we're reading this\n //\n // // these might not exist?\n // dispatcher,\n // duplex\n } = data.init;\n const { messages } = JSON.parse(body as string);\n\n // Automatically transform any incoming messages\n const transformedMessages = autoTransformMessages(messages);\n\n this._broadcastChatMessage(\n {\n messages: transformedMessages,\n type: MessageType.CF_AGENT_CHAT_MESSAGES\n },\n [connection.id]\n );\n\n await this.persistMessages(transformedMessages, [connection.id]);\n\n this.observability?.emit(\n {\n displayMessage: \"Chat message request\",\n id: data.id,\n payload: {},\n timestamp: Date.now(),\n type: \"message:request\"\n },\n this.ctx\n );\n\n const chatMessageId = data.id;\n const abortSignal = this._getAbortSignal(chatMessageId);\n\n return this._tryCatchChat(async () => {\n const response = await this.onChatMessage(\n async (_finishResult) => {\n this._removeAbortController(chatMessageId);\n\n this.observability?.emit(\n {\n displayMessage: \"Chat message response\",\n id: data.id,\n payload: {},\n timestamp: Date.now(),\n type: \"message:response\"\n },\n this.ctx\n );\n\n // Note: Message persistence now happens in the _reply method\n // after the complete response text has been accumulated\n },\n abortSignal ? { abortSignal } : undefined\n );\n\n if (response) {\n await this._reply(data.id, response);\n } else {\n // Log a warning for observability\n console.warn(\n `[AIChatAgent] onChatMessage returned no response for chatMessageId: ${chatMessageId}`\n );\n // Send a fallback message to the client\n this._broadcastChatMessage(\n {\n body: \"No response was generated by the agent.\",\n done: true,\n id: data.id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n },\n [connection.id]\n );\n }\n });\n }\n if (data.type === MessageType.CF_AGENT_CHAT_CLEAR) {\n this._destroyAbortControllers();\n this.sql`delete from cf_ai_chat_agent_messages`;\n this.messages = [];\n this._broadcastChatMessage(\n {\n type: MessageType.CF_AGENT_CHAT_CLEAR\n },\n [connection.id]\n );\n } else if (data.type === MessageType.CF_AGENT_CHAT_MESSAGES) {\n // replace the messages with the new ones, automatically transformed\n const transformedMessages = autoTransformMessages(data.messages);\n await this.persistMessages(transformedMessages, [connection.id]);\n } else if (data.type === MessageType.CF_AGENT_CHAT_REQUEST_CANCEL) {\n // propagate an abort signal for the associated request\n this._cancelChatRequest(data.id);\n }\n }\n }\n\n override async onRequest(request: Request): Promise<Response> {\n return this._tryCatchChat(() => {\n const url = new URL(request.url);\n if (url.pathname.endsWith(\"/get-messages\")) {\n const messages = (\n this.sql`select * from cf_ai_chat_agent_messages` || []\n ).map((row) => {\n return JSON.parse(row.message as string);\n });\n return Response.json(messages);\n }\n return super.onRequest(request);\n });\n }\n\n private async _tryCatchChat<T>(fn: () => T | Promise<T>) {\n try {\n return await fn();\n } catch (e) {\n throw this.onError(e);\n }\n }\n\n /**\n * Handle incoming chat messages and generate a response\n * @param onFinish Callback to be called when the response is finished\n * @param options.signal A signal to pass to any child requests which can be used to cancel them\n * @returns Response to send to the client or undefined\n */\n async onChatMessage(\n // biome-ignore lint/correctness/noUnusedFunctionParameters: overridden later\n onFinish: StreamTextOnFinishCallback<ToolSet>,\n // biome-ignore lint/correctness/noUnusedFunctionParameters: overridden later\n options?: { abortSignal: AbortSignal | undefined }\n ): Promise<Response | undefined> {\n throw new Error(\n \"recieved a chat message, override onChatMessage and return a Response to send to the client\"\n );\n }\n\n /**\n * Save messages on the server side\n * @param messages Chat messages to save\n */\n async saveMessages(messages: ChatMessage[]) {\n await this.persistMessages(messages);\n }\n\n async persistMessages(\n messages: ChatMessage[],\n excludeBroadcastIds: string[] = []\n ) {\n this.sql`delete from cf_ai_chat_agent_messages`;\n for (const message of messages) {\n this.sql`insert into cf_ai_chat_agent_messages (id, message) values (${\n message.id\n },${JSON.stringify(message)})`;\n }\n this.messages = messages;\n this._broadcastChatMessage(\n {\n messages: messages,\n type: MessageType.CF_AGENT_CHAT_MESSAGES\n },\n excludeBroadcastIds\n );\n }\n\n private async _reply(id: string, response: Response) {\n return this._tryCatchChat(async () => {\n if (!response.body) {\n // Send empty response if no body\n this._broadcastChatMessage({\n body: \"\",\n done: true,\n id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n });\n return;\n }\n\n const reader = response.body.getReader();\n let fullResponseText = \"\"; // Accumulate the assistant's response text\n let fullReasoningText = \"\"; // Accumulate the assistant's reasoning\n // Track tool calls by toolCallid, so we can persist them as parts later\n const toolCalls = new Map<\n string,\n {\n type: string;\n state: string;\n toolCallId: string;\n toolName: string;\n input?: unknown;\n output?: unknown;\n isError?: boolean;\n errorText?: string;\n }\n >();\n try {\n while (true) {\n const { done, value } = await reader.read();\n if (done) {\n // Send final completion signal\n this._broadcastChatMessage({\n body: \"\",\n done: true,\n id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n });\n break;\n }\n\n const chunk = decoder.decode(value);\n\n // Determine response format based on content-type\n const contentType = response.headers.get(\"content-type\") || \"\";\n const isSSE = contentType.includes(\"text/event-stream\");\n\n if (isSSE) {\n // Parse AI SDK v5 SSE format and extract text deltas\n const lines = chunk.split(\"\\n\");\n for (const line of lines) {\n if (line.startsWith(\"data: \") && line !== \"data: [DONE]\") {\n try {\n const data = JSON.parse(line.slice(6)); // Remove 'data: ' prefix\n\n switch (data.type) {\n // SSE event signaling the tool input is ready. We track by\n // `toolCallId` so we can persist it as a tool part in the message.\n case \"tool-input-available\": {\n const { toolCallId, toolName, input } = data;\n toolCalls.set(toolCallId, {\n toolCallId,\n toolName,\n input,\n type: toolName ? `tool-${toolName}` : \"dynamic-tool\",\n state: \"input-available\"\n });\n break;\n }\n\n // SSE event signaling the tool output is ready. We should've\n // already received the input in a previous event so an entry\n // with `toolCallId` should already be present\n case \"tool-output-available\": {\n const { toolCallId, output, isError, errorText } = data;\n const toolPart = toolCalls.get(toolCallId);\n if (toolPart)\n toolCalls.set(toolCallId, {\n ...toolPart,\n output,\n isError,\n errorText,\n state: \"output-available\"\n });\n break;\n }\n\n case \"error\": {\n // Non-tool errors, we set `error: true` and terminate early\n this._broadcastChatMessage({\n error: true,\n body: data.errorText ?? JSON.stringify(data),\n done: false,\n id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n });\n return;\n }\n\n case \"reasoning-delta\": {\n if (data.delta) fullReasoningText += data.delta;\n break;\n }\n\n case \"text-delta\": {\n if (data.delta) fullResponseText += data.delta;\n break;\n }\n }\n\n // Always forward the raw part to the client\n this._broadcastChatMessage({\n body: JSON.stringify(data),\n done: false,\n id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n });\n } catch (_e) {\n // Skip malformed JSON lines silently\n }\n }\n }\n } else {\n // Handle plain text responses (e.g., from generateText)\n // Treat the entire chunk as a text delta to preserve exact formatting\n if (chunk.length > 0) {\n fullResponseText += chunk;\n // Synthesize a text-delta event so clients can stream-render\n this._broadcastChatMessage({\n body: JSON.stringify({ type: \"text-delta\", delta: chunk }),\n done: false,\n id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n });\n }\n }\n }\n } finally {\n reader.releaseLock();\n }\n\n // After streaming is complete, persist the complete assistant's response\n const messageParts: ChatMessage[\"parts\"] = [];\n\n Array.from(toolCalls.values()).forEach((t) => {\n messageParts.push(t as ChatMessage[\"parts\"][number]);\n });\n\n if (fullReasoningText.trim()) {\n messageParts.push({ type: \"reasoning\", text: fullReasoningText });\n }\n\n if (fullResponseText.trim()) {\n messageParts.push({ type: \"text\", text: fullResponseText });\n }\n\n if (messageParts.length > 0) {\n await this.persistMessages([\n ...this.messages,\n {\n id: `assistant_${Date.now()}_${Math.random().toString(36).slice(2, 11)}`,\n role: \"assistant\",\n parts: messageParts\n }\n ]);\n }\n });\n }\n\n /**\n * For the given message id, look up its associated AbortController\n * If the AbortController does not exist, create and store one in memory\n *\n * returns the AbortSignal associated with the AbortController\n */\n private _getAbortSignal(id: string): AbortSignal | undefined {\n // Defensive check, since we're coercing message types at the moment\n if (typeof id !== \"string\") {\n return undefined;\n }\n\n if (!this._chatMessageAbortControllers.has(id)) {\n this._chatMessageAbortControllers.set(id, new AbortController());\n }\n\n return this._chatMessageAbortControllers.get(id)?.signal;\n }\n\n /**\n * Remove an abort controller from the cache of pending message responses\n */\n private _removeAbortController(id: string) {\n this._chatMessageAbortControllers.delete(id);\n }\n\n /**\n * Propagate an abort signal for any requests associated with the given message id\n */\n private _cancelChatRequest(id: string) {\n if (this._chatMessageAbortControllers.has(id)) {\n const abortController = this._chatMessageAbortControllers.get(id);\n abortController?.abort();\n }\n }\n\n /**\n * Abort all pending requests and clear the cache of AbortControllers\n */\n private _destroyAbortControllers() {\n for (const controller of this._chatMessageAbortControllers.values()) {\n controller?.abort();\n }\n this._chatMessageAbortControllers.clear();\n }\n\n /**\n * When the DO is destroyed, cancel all pending requests\n */\n async destroy() {\n this._destroyAbortControllers();\n await super.destroy();\n }\n}\n"],"mappings":";;;;;;;;;;;;;AAaA,IAAM,UAAU,IAAI,YAAY;AAMzB,IAAM,cAAN,cAA0D,MAG/D;AAAA,EAQA,YAAY,KAAmB,KAAU;AACvC,UAAM,KAAK,GAAG;AACd,SAAK;AAAA;AAAA;AAAA;AAAA;AAOL,UAAM,eACJ,KAAK,gDAAgD,CAAC,GACtD,IAAI,CAAC,QAAQ;AACb,aAAO,KAAK,MAAM,IAAI,OAAiB;AAAA,IACzC,CAAC;AAGD,SAAK,WAAW,sBAAsB,WAAW;AAEjD,SAAK,+BAA+B,oBAAI,IAAI;AAAA,EAC9C;AAAA,EAEQ,sBAAsB,SAA0B,SAAoB;AAC1E,SAAK,UAAU,KAAK,UAAU,OAAO,GAAG,OAAO;AAAA,EACjD;AAAA,EAEA,MAAe,UAAU,YAAwB,SAAoB;AACnE,QAAI,OAAO,YAAY,UAAU;AAC/B,UAAI;AACJ,UAAI;AACF,eAAO,KAAK,MAAM,OAAO;AAAA,MAC3B,SAAS,QAAQ;AAGf;AAAA,MACF;AACA,UACE,KAAK,wEACL,KAAK,KAAK,WAAW,QACrB;AACA,cAAM;AAAA;AAAA;AAAA;AAAA,UAIJ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QAKF,IAAI,KAAK;AACT,cAAM,EAAE,SAAS,IAAI,KAAK,MAAM,IAAc;AAG9C,cAAM,sBAAsB,sBAAsB,QAAQ;AAE1D,aAAK;AAAA,UACH;AAAA,YACE,UAAU;AAAA,YACV;AAAA,UACF;AAAA,UACA,CAAC,WAAW,EAAE;AAAA,QAChB;AAEA,cAAM,KAAK,gBAAgB,qBAAqB,CAAC,WAAW,EAAE,CAAC;AAE/D,aAAK,eAAe;AAAA,UAClB;AAAA,YACE,gBAAgB;AAAA,YAChB,IAAI,KAAK;AAAA,YACT,SAAS,CAAC;AAAA,YACV,WAAW,KAAK,IAAI;AAAA,YACpB,MAAM;AAAA,UACR;AAAA,UACA,KAAK;AAAA,QACP;AAEA,cAAM,gBAAgB,KAAK;AAC3B,cAAM,cAAc,KAAK,gBAAgB,aAAa;AAEtD,eAAO,KAAK,cAAc,YAAY;AACpC,gBAAM,WAAW,MAAM,KAAK;AAAA,YAC1B,OAAO,kBAAkB;AACvB,mBAAK,uBAAuB,aAAa;AAEzC,mBAAK,eAAe;AAAA,gBAClB;AAAA,kBACE,gBAAgB;AAAA,kBAChB,IAAI,KAAK;AAAA,kBACT,SAAS,CAAC;AAAA,kBACV,WAAW,KAAK,IAAI;AAAA,kBACpB,MAAM;AAAA,gBACR;AAAA,gBACA,KAAK;AAAA,cACP;AAAA,YAIF;AAAA,YACA,cAAc,EAAE,YAAY,IAAI;AAAA,UAClC;AAEA,cAAI,UAAU;AACZ,kBAAM,KAAK,OAAO,KAAK,IAAI,QAAQ;AAAA,UACrC,OAAO;AAEL,oBAAQ;AAAA,cACN,uEAAuE,aAAa;AAAA,YACtF;AAEA,iBAAK;AAAA,cACH;AAAA,gBACE,MAAM;AAAA,gBACN,MAAM;AAAA,gBACN,IAAI,KAAK;AAAA,gBACT;AAAA,cACF;AAAA,cACA,CAAC,WAAW,EAAE;AAAA,YAChB;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AACA,UAAI,KAAK,0DAA0C;AACjD,aAAK,yBAAyB;AAC9B,aAAK;AACL,aAAK,WAAW,CAAC;AACjB,aAAK;AAAA,UACH;AAAA,YACE;AAAA,UACF;AAAA,UACA,CAAC,WAAW,EAAE;AAAA,QAChB;AAAA,MACF,WAAW,KAAK,gEAA6C;AAE3D,cAAM,sBAAsB,sBAAsB,KAAK,QAAQ;AAC/D,cAAM,KAAK,gBAAgB,qBAAqB,CAAC,WAAW,EAAE,CAAC;AAAA,MACjE,WAAW,KAAK,4EAAmD;AAEjE,aAAK,mBAAmB,KAAK,EAAE;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAe,UAAU,SAAqC;AAC5D,WAAO,KAAK,cAAc,MAAM;AAC9B,YAAM,MAAM,IAAI,IAAI,QAAQ,GAAG;AAC/B,UAAI,IAAI,SAAS,SAAS,eAAe,GAAG;AAC1C,cAAM,YACJ,KAAK,gDAAgD,CAAC,GACtD,IAAI,CAAC,QAAQ;AACb,iBAAO,KAAK,MAAM,IAAI,OAAiB;AAAA,QACzC,CAAC;AACD,eAAO,SAAS,KAAK,QAAQ;AAAA,MAC/B;AACA,aAAO,MAAM,UAAU,OAAO;AAAA,IAChC,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,cAAiB,IAA0B;AACvD,QAAI;AACF,aAAO,MAAM,GAAG;AAAA,IAClB,SAAS,GAAG;AACV,YAAM,KAAK,QAAQ,CAAC;AAAA,IACtB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,cAEJ,UAEA,SAC+B;AAC/B,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,aAAa,UAAyB;AAC1C,UAAM,KAAK,gBAAgB,QAAQ;AAAA,EACrC;AAAA,EAEA,MAAM,gBACJ,UACA,sBAAgC,CAAC,GACjC;AACA,SAAK;AACL,eAAW,WAAW,UAAU;AAC9B,WAAK,kEACH,QAAQ,EACV,IAAI,KAAK,UAAU,OAAO,CAAC;AAAA,IAC7B;AACA,SAAK,WAAW;AAChB,SAAK;AAAA,MACH;AAAA,QACE;AAAA,QACA;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,OAAO,IAAY,UAAoB;AACnD,WAAO,KAAK,cAAc,YAAY;AACpC,UAAI,CAAC,SAAS,MAAM;AAElB,aAAK,sBAAsB;AAAA,UACzB,MAAM;AAAA,UACN,MAAM;AAAA,UACN;AAAA,UACA;AAAA,QACF,CAAC;AACD;AAAA,MACF;AAEA,YAAM,SAAS,SAAS,KAAK,UAAU;AACvC,UAAI,mBAAmB;AACvB,UAAI,oBAAoB;AAExB,YAAM,YAAY,oBAAI,IAYpB;AACF,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,MAAM;AAER,iBAAK,sBAAsB;AAAA,cACzB,MAAM;AAAA,cACN,MAAM;AAAA,cACN;AAAA,cACA;AAAA,YACF,CAAC;AACD;AAAA,UACF;AAEA,gBAAM,QAAQ,QAAQ,OAAO,KAAK;AAGlC,gBAAM,cAAc,SAAS,QAAQ,IAAI,cAAc,KAAK;AAC5D,gBAAM,QAAQ,YAAY,SAAS,mBAAmB;AAEtD,cAAI,OAAO;AAET,kBAAM,QAAQ,MAAM,MAAM,IAAI;AAC9B,uBAAW,QAAQ,OAAO;AACxB,kBAAI,KAAK,WAAW,QAAQ,KAAK,SAAS,gBAAgB;AACxD,oBAAI;AACF,wBAAM,OAAO,KAAK,MAAM,KAAK,MAAM,CAAC,CAAC;AAErC,0BAAQ,KAAK,MAAM;AAAA;AAAA;AAAA,oBAGjB,KAAK,wBAAwB;AAC3B,4BAAM,EAAE,YAAY,UAAU,MAAM,IAAI;AACxC,gCAAU,IAAI,YAAY;AAAA,wBACxB;AAAA,wBACA;AAAA,wBACA;AAAA,wBACA,MAAM,WAAW,QAAQ,QAAQ,KAAK;AAAA,wBACtC,OAAO;AAAA,sBACT,CAAC;AACD;AAAA,oBACF;AAAA;AAAA;AAAA;AAAA,oBAKA,KAAK,yBAAyB;AAC5B,4BAAM,EAAE,YAAY,QAAQ,SAAS,UAAU,IAAI;AACnD,4BAAM,WAAW,UAAU,IAAI,UAAU;AACzC,0BAAI;AACF,kCAAU,IAAI,YAAY;AAAA,0BACxB,GAAG;AAAA,0BACH;AAAA,0BACA;AAAA,0BACA;AAAA,0BACA,OAAO;AAAA,wBACT,CAAC;AACH;AAAA,oBACF;AAAA,oBAEA,KAAK,SAAS;AAEZ,2BAAK,sBAAsB;AAAA,wBACzB,OAAO;AAAA,wBACP,MAAM,KAAK,aAAa,KAAK,UAAU,IAAI;AAAA,wBAC3C,MAAM;AAAA,wBACN;AAAA,wBACA;AAAA,sBACF,CAAC;AACD;AAAA,oBACF;AAAA,oBAEA,KAAK,mBAAmB;AACtB,0BAAI,KAAK,MAAO,sBAAqB,KAAK;AAC1C;AAAA,oBACF;AAAA,oBAEA,KAAK,cAAc;AACjB,0BAAI,KAAK,MAAO,qBAAoB,KAAK;AACzC;AAAA,oBACF;AAAA,kBACF;AAGA,uBAAK,sBAAsB;AAAA,oBACzB,MAAM,KAAK,UAAU,IAAI;AAAA,oBACzB,MAAM;AAAA,oBACN;AAAA,oBACA;AAAA,kBACF,CAAC;AAAA,gBACH,SAAS,IAAI;AAAA,gBAEb;AAAA,cACF;AAAA,YACF;AAAA,UACF,OAAO;AAGL,gBAAI,MAAM,SAAS,GAAG;AACpB,kCAAoB;AAEpB,mBAAK,sBAAsB;AAAA,gBACzB,MAAM,KAAK,UAAU,EAAE,MAAM,cAAc,OAAO,MAAM,CAAC;AAAA,gBACzD,MAAM;AAAA,gBACN;AAAA,gBACA;AAAA,cACF,CAAC;AAAA,YACH;AAAA,UACF;AAAA,QACF;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAGA,YAAM,eAAqC,CAAC;AAE5C,YAAM,KAAK,UAAU,OAAO,CAAC,EAAE,QAAQ,CAAC,MAAM;AAC5C,qBAAa,KAAK,CAAiC;AAAA,MACrD,CAAC;AAED,UAAI,kBAAkB,KAAK,GAAG;AAC5B,qBAAa,KAAK,EAAE,MAAM,aAAa,MAAM,kBAAkB,CAAC;AAAA,MAClE;AAEA,UAAI,iBAAiB,KAAK,GAAG;AAC3B,qBAAa,KAAK,EAAE,MAAM,QAAQ,MAAM,iBAAiB,CAAC;AAAA,MAC5D;AAEA,UAAI,aAAa,SAAS,GAAG;AAC3B,cAAM,KAAK,gBAAgB;AAAA,UACzB,GAAG,KAAK;AAAA,UACR;AAAA,YACE,IAAI,aAAa,KAAK,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,MAAM,GAAG,EAAE,CAAC;AAAA,YACtE,MAAM;AAAA,YACN,OAAO;AAAA,UACT;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQQ,gBAAgB,IAAqC;AAE3D,QAAI,OAAO,OAAO,UAAU;AAC1B,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,KAAK,6BAA6B,IAAI,EAAE,GAAG;AAC9C,WAAK,6BAA6B,IAAI,IAAI,IAAI,gBAAgB,CAAC;AAAA,IACjE;AAEA,WAAO,KAAK,6BAA6B,IAAI,EAAE,GAAG;AAAA,EACpD;AAAA;AAAA;AAAA;AAAA,EAKQ,uBAAuB,IAAY;AACzC,SAAK,6BAA6B,OAAO,EAAE;AAAA,EAC7C;AAAA;AAAA;AAAA;AAAA,EAKQ,mBAAmB,IAAY;AACrC,QAAI,KAAK,6BAA6B,IAAI,EAAE,GAAG;AAC7C,YAAM,kBAAkB,KAAK,6BAA6B,IAAI,EAAE;AAChE,uBAAiB,MAAM;AAAA,IACzB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,2BAA2B;AACjC,eAAW,cAAc,KAAK,6BAA6B,OAAO,GAAG;AACnE,kBAAY,MAAM;AAAA,IACpB;AACA,SAAK,6BAA6B,MAAM;AAAA,EAC1C;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,UAAU;AACd,SAAK,yBAAyB;AAC9B,UAAM,MAAM,QAAQ;AAAA,EACtB;AACF;","names":[]}
1
+ {"version":3,"sources":["../src/ai-chat-agent.ts"],"sourcesContent":["import {\n getToolName,\n isToolUIPart,\n parsePartialJson,\n type UIMessage as ChatMessage,\n type DynamicToolUIPart,\n type ProviderMetadata,\n type ReasoningUIPart,\n type StreamTextOnFinishCallback,\n type TextUIPart,\n type ToolSet,\n type ToolUIPart,\n type UIMessageChunk\n} from \"ai\";\nimport { Agent, type AgentContext, type Connection, type WSMessage } from \"./\";\nimport {\n MessageType,\n type IncomingMessage,\n type OutgoingMessage\n} from \"./ai-types\";\nimport { autoTransformMessages } from \"./ai-chat-v5-migration\";\n\nconst decoder = new TextDecoder();\n\n/**\n * Extension of Agent with built-in chat capabilities\n * @template Env Environment type containing bindings\n */\nexport class AIChatAgent<Env = unknown, State = unknown> extends Agent<\n Env,\n State\n> {\n /**\n * Map of message `id`s to `AbortController`s\n * useful to propagate request cancellation signals for any external calls made by the agent\n */\n private _chatMessageAbortControllers: Map<string, AbortController>;\n /** Array of chat messages for the current conversation */\n messages: ChatMessage[];\n constructor(ctx: AgentContext, env: Env) {\n super(ctx, env);\n this.sql`create table if not exists cf_ai_chat_agent_messages (\n id text primary key,\n message text not null,\n created_at datetime default current_timestamp\n )`;\n\n // Load messages and automatically transform them to v5 format\n const rawMessages = (\n this.sql`select * from cf_ai_chat_agent_messages` || []\n ).map((row) => {\n return JSON.parse(row.message as string);\n });\n\n // Automatic migration following https://jhak.im/blog/ai-sdk-migration-handling-previously-saved-messages\n this.messages = autoTransformMessages(rawMessages);\n\n this._chatMessageAbortControllers = new Map();\n }\n\n private _broadcastChatMessage(message: OutgoingMessage, exclude?: string[]) {\n this.broadcast(JSON.stringify(message), exclude);\n }\n\n override async onMessage(connection: Connection, message: WSMessage) {\n if (typeof message === \"string\") {\n let data: IncomingMessage;\n try {\n data = JSON.parse(message) as IncomingMessage;\n } catch (_error) {\n // silently ignore invalid messages for now\n // TODO: log errors with log levels\n return;\n }\n if (\n data.type === MessageType.CF_AGENT_USE_CHAT_REQUEST &&\n data.init.method === \"POST\"\n ) {\n const {\n // method,\n // keepalive,\n // headers,\n body // we're reading this\n //\n // // these might not exist?\n // dispatcher,\n // duplex\n } = data.init;\n const { messages } = JSON.parse(body as string);\n\n // Automatically transform any incoming messages\n const transformedMessages = autoTransformMessages(messages);\n\n this._broadcastChatMessage(\n {\n messages: transformedMessages,\n type: MessageType.CF_AGENT_CHAT_MESSAGES\n },\n [connection.id]\n );\n\n await this.persistMessages(transformedMessages, [connection.id]);\n\n this.observability?.emit(\n {\n displayMessage: \"Chat message request\",\n id: data.id,\n payload: {},\n timestamp: Date.now(),\n type: \"message:request\"\n },\n this.ctx\n );\n\n const chatMessageId = data.id;\n const abortSignal = this._getAbortSignal(chatMessageId);\n\n return this._tryCatchChat(async () => {\n const response = await this.onChatMessage(\n async (_finishResult) => {\n this._removeAbortController(chatMessageId);\n\n this.observability?.emit(\n {\n displayMessage: \"Chat message response\",\n id: data.id,\n payload: {},\n timestamp: Date.now(),\n type: \"message:response\"\n },\n this.ctx\n );\n\n // Note: Message persistence now happens in the _reply method\n // after the complete response text has been accumulated\n },\n abortSignal ? { abortSignal } : undefined\n );\n\n if (response) {\n await this._reply(data.id, response);\n } else {\n // Log a warning for observability\n console.warn(\n `[AIChatAgent] onChatMessage returned no response for chatMessageId: ${chatMessageId}`\n );\n // Send a fallback message to the client\n this._broadcastChatMessage(\n {\n body: \"No response was generated by the agent.\",\n done: true,\n id: data.id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n },\n [connection.id]\n );\n }\n });\n }\n if (data.type === MessageType.CF_AGENT_CHAT_CLEAR) {\n this._destroyAbortControllers();\n this.sql`delete from cf_ai_chat_agent_messages`;\n this.messages = [];\n this._broadcastChatMessage(\n {\n type: MessageType.CF_AGENT_CHAT_CLEAR\n },\n [connection.id]\n );\n } else if (data.type === MessageType.CF_AGENT_CHAT_MESSAGES) {\n // replace the messages with the new ones, automatically transformed\n const transformedMessages = autoTransformMessages(data.messages);\n await this.persistMessages(transformedMessages, [connection.id]);\n } else if (data.type === MessageType.CF_AGENT_CHAT_REQUEST_CANCEL) {\n // propagate an abort signal for the associated request\n this._cancelChatRequest(data.id);\n }\n }\n }\n\n override async onRequest(request: Request): Promise<Response> {\n return this._tryCatchChat(() => {\n const url = new URL(request.url);\n if (url.pathname.endsWith(\"/get-messages\")) {\n const messages = (\n this.sql`select * from cf_ai_chat_agent_messages` || []\n ).map((row) => {\n return JSON.parse(row.message as string);\n });\n return Response.json(messages);\n }\n return super.onRequest(request);\n });\n }\n\n private async _tryCatchChat<T>(fn: () => T | Promise<T>) {\n try {\n return await fn();\n } catch (e) {\n throw this.onError(e);\n }\n }\n\n /**\n * Handle incoming chat messages and generate a response\n * @param onFinish Callback to be called when the response is finished\n * @param options.signal A signal to pass to any child requests which can be used to cancel them\n * @returns Response to send to the client or undefined\n */\n async onChatMessage(\n // biome-ignore lint/correctness/noUnusedFunctionParameters: overridden later\n onFinish: StreamTextOnFinishCallback<ToolSet>,\n // biome-ignore lint/correctness/noUnusedFunctionParameters: overridden later\n options?: { abortSignal: AbortSignal | undefined }\n ): Promise<Response | undefined> {\n throw new Error(\n \"recieved a chat message, override onChatMessage and return a Response to send to the client\"\n );\n }\n\n /**\n * Save messages on the server side\n * @param messages Chat messages to save\n */\n async saveMessages(messages: ChatMessage[]) {\n await this.persistMessages(messages);\n }\n\n async persistMessages(\n messages: ChatMessage[],\n excludeBroadcastIds: string[] = []\n ) {\n this.sql`delete from cf_ai_chat_agent_messages`;\n for (const message of messages) {\n this.sql`insert into cf_ai_chat_agent_messages (id, message) values (${\n message.id\n },${JSON.stringify(message)})`;\n }\n this.messages = messages;\n this._broadcastChatMessage(\n {\n messages: messages,\n type: MessageType.CF_AGENT_CHAT_MESSAGES\n },\n excludeBroadcastIds\n );\n }\n\n private async _reply(id: string, response: Response) {\n return this._tryCatchChat(async () => {\n if (!response.body) {\n // Send empty response if no body\n this._broadcastChatMessage({\n body: \"\",\n done: true,\n id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n });\n return;\n }\n\n const reader = response.body.getReader();\n\n // Parsing state adapted from:\n // https://github.com/vercel/ai/blob/main/packages/ai/src/ui-message-stream/ui-message-chunks.ts#L295\n const message: ChatMessage = {\n id: `assistant_${Date.now()}_${Math.random().toString(36).slice(2, 11)}`, // default\n role: \"assistant\",\n parts: []\n };\n let activeTextParts: Record<string, TextUIPart> = {};\n let activeReasoningParts: Record<string, ReasoningUIPart> = {};\n const partialToolCalls: Record<\n string,\n { text: string; index: number; toolName: string; dynamic?: boolean }\n > = {};\n\n function updateDynamicToolPart(\n options: {\n toolName: string;\n toolCallId: string;\n providerExecuted?: boolean;\n } & (\n | {\n state: \"input-streaming\";\n input: unknown;\n }\n | {\n state: \"input-available\";\n input: unknown;\n providerMetadata?: ProviderMetadata;\n }\n | {\n state: \"output-available\";\n input: unknown;\n output: unknown;\n preliminary: boolean | undefined;\n }\n | {\n state: \"output-error\";\n input: unknown;\n errorText: string;\n providerMetadata?: ProviderMetadata;\n }\n )\n ) {\n const part = message.parts.find(\n (part) =>\n part.type === \"dynamic-tool\" &&\n part.toolCallId === options.toolCallId\n ) as DynamicToolUIPart | undefined;\n\n const anyOptions = options as Record<string, unknown>;\n const anyPart = part as Record<string, unknown>;\n\n if (part != null) {\n part.state = options.state;\n anyPart.toolName = options.toolName;\n anyPart.input = anyOptions.input;\n anyPart.output = anyOptions.output;\n anyPart.errorText = anyOptions.errorText;\n anyPart.rawInput = anyOptions.rawInput ?? anyPart.rawInput;\n anyPart.preliminary = anyOptions.preliminary;\n\n if (\n anyOptions.providerMetadata != null &&\n part.state === \"input-available\"\n ) {\n part.callProviderMetadata =\n anyOptions.providerMetadata as ProviderMetadata;\n }\n } else {\n message.parts.push({\n type: \"dynamic-tool\",\n toolName: options.toolName,\n toolCallId: options.toolCallId,\n state: options.state,\n input: anyOptions.input,\n output: anyOptions.output,\n errorText: anyOptions.errorText,\n preliminary: anyOptions.preliminary,\n ...(anyOptions.providerMetadata != null\n ? { callProviderMetadata: anyOptions.providerMetadata }\n : {})\n } as DynamicToolUIPart);\n }\n }\n\n function updateToolPart(\n options: {\n toolName: string;\n toolCallId: string;\n providerExecuted?: boolean;\n } & (\n | {\n state: \"input-streaming\";\n input: unknown;\n providerExecuted?: boolean;\n }\n | {\n state: \"input-available\";\n input: unknown;\n providerExecuted?: boolean;\n providerMetadata?: ProviderMetadata;\n }\n | {\n state: \"output-available\";\n input: unknown;\n output: unknown;\n providerExecuted?: boolean;\n preliminary?: boolean;\n }\n | {\n state: \"output-error\";\n input: unknown;\n rawInput?: unknown;\n errorText: string;\n providerExecuted?: boolean;\n providerMetadata?: ProviderMetadata;\n }\n )\n ) {\n const part = message.parts.find(\n (part) => isToolUIPart(part) && part.toolCallId === options.toolCallId\n ) as ToolUIPart | undefined;\n\n const anyOptions = options as Record<string, unknown>;\n const anyPart = part as Record<string, unknown>;\n\n if (part != null) {\n part.state = options.state;\n anyPart.input = anyOptions.input;\n anyPart.output = anyOptions.output;\n anyPart.errorText = anyOptions.errorText;\n anyPart.rawInput = anyOptions.rawInput;\n anyPart.preliminary = anyOptions.preliminary;\n\n // once providerExecuted is set, it stays for streaming\n anyPart.providerExecuted =\n anyOptions.providerExecuted ?? part.providerExecuted;\n\n if (\n anyOptions.providerMetadata != null &&\n part.state === \"input-available\"\n ) {\n part.callProviderMetadata =\n anyOptions.providerMetadata as ProviderMetadata;\n }\n } else {\n message.parts.push({\n type: `tool-${options.toolName}`,\n toolCallId: options.toolCallId,\n state: options.state,\n input: anyOptions.input,\n output: anyOptions.output,\n rawInput: anyOptions.rawInput,\n errorText: anyOptions.errorText,\n providerExecuted: anyOptions.providerExecuted,\n preliminary: anyOptions.preliminary,\n ...(anyOptions.providerMetadata != null\n ? { callProviderMetadata: anyOptions.providerMetadata }\n : {})\n } as ToolUIPart);\n }\n }\n\n async function updateMessageMetadata(metadata: unknown) {\n if (metadata != null) {\n const mergedMetadata =\n message.metadata != null\n ? { ...message.metadata, ...metadata } // TODO: do proper merging\n : metadata;\n\n message.metadata = mergedMetadata;\n }\n }\n\n try {\n while (true) {\n const { done, value } = await reader.read();\n if (done) {\n // Send final completion signal\n this._broadcastChatMessage({\n body: \"\",\n done: true,\n id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n });\n break;\n }\n\n const chunk = decoder.decode(value);\n\n // Determine response format based on content-type\n const contentType = response.headers.get(\"content-type\") || \"\";\n const isSSE = contentType.includes(\"text/event-stream\");\n\n // After streaming is complete, persist the complete assistant's response\n if (isSSE) {\n // Parse AI SDK v5 SSE format and extract text deltas\n const lines = chunk.split(\"\\n\");\n for (const line of lines) {\n if (line.startsWith(\"data: \") && line !== \"data: [DONE]\") {\n try {\n const data: UIMessageChunk = JSON.parse(line.slice(6)); // Remove 'data: ' prefix\n switch (data.type) {\n case \"text-start\": {\n const textPart: TextUIPart = {\n type: \"text\",\n text: \"\",\n providerMetadata: data.providerMetadata,\n state: \"streaming\"\n };\n activeTextParts[data.id] = textPart;\n message.parts.push(textPart);\n break;\n }\n\n case \"text-delta\": {\n const textPart = activeTextParts[data.id];\n textPart.text += data.delta;\n textPart.providerMetadata =\n data.providerMetadata ?? textPart.providerMetadata;\n break;\n }\n\n case \"text-end\": {\n const textPart = activeTextParts[data.id];\n textPart.state = \"done\";\n textPart.providerMetadata =\n data.providerMetadata ?? textPart.providerMetadata;\n delete activeTextParts[data.id];\n break;\n }\n\n case \"reasoning-start\": {\n const reasoningPart: ReasoningUIPart = {\n type: \"reasoning\",\n text: \"\",\n providerMetadata: data.providerMetadata,\n state: \"streaming\"\n };\n activeReasoningParts[data.id] = reasoningPart;\n message.parts.push(reasoningPart);\n break;\n }\n\n case \"reasoning-delta\": {\n const reasoningPart = activeReasoningParts[data.id];\n reasoningPart.text += data.delta;\n reasoningPart.providerMetadata =\n data.providerMetadata ?? reasoningPart.providerMetadata;\n break;\n }\n\n case \"reasoning-end\": {\n const reasoningPart = activeReasoningParts[data.id];\n reasoningPart.providerMetadata =\n data.providerMetadata ?? reasoningPart.providerMetadata;\n reasoningPart.state = \"done\";\n delete activeReasoningParts[data.id];\n\n break;\n }\n\n case \"file\": {\n message.parts.push({\n type: \"file\",\n mediaType: data.mediaType,\n url: data.url\n });\n\n break;\n }\n\n case \"source-url\": {\n message.parts.push({\n type: \"source-url\",\n sourceId: data.sourceId,\n url: data.url,\n title: data.title,\n providerMetadata: data.providerMetadata\n });\n\n break;\n }\n\n case \"source-document\": {\n message.parts.push({\n type: \"source-document\",\n sourceId: data.sourceId,\n mediaType: data.mediaType,\n title: data.title,\n filename: data.filename,\n providerMetadata: data.providerMetadata\n });\n\n break;\n }\n\n case \"tool-input-start\": {\n const toolInvocations =\n message.parts.filter(isToolUIPart);\n\n // add the partial tool call to the map\n partialToolCalls[data.toolCallId] = {\n text: \"\",\n toolName: data.toolName,\n index: toolInvocations.length,\n dynamic: data.dynamic\n };\n\n if (data.dynamic) {\n updateDynamicToolPart({\n toolCallId: data.toolCallId,\n toolName: data.toolName,\n state: \"input-streaming\",\n input: undefined\n });\n } else {\n updateToolPart({\n toolCallId: data.toolCallId,\n toolName: data.toolName,\n state: \"input-streaming\",\n input: undefined\n });\n }\n\n break;\n }\n\n case \"tool-input-delta\": {\n const partialToolCall = partialToolCalls[data.toolCallId];\n\n partialToolCall.text += data.inputTextDelta;\n\n const { value: partialArgs } = await parsePartialJson(\n partialToolCall.text\n );\n\n if (partialToolCall.dynamic) {\n updateDynamicToolPart({\n toolCallId: data.toolCallId,\n toolName: partialToolCall.toolName,\n state: \"input-streaming\",\n input: partialArgs\n });\n } else {\n updateToolPart({\n toolCallId: data.toolCallId,\n toolName: partialToolCall.toolName,\n state: \"input-streaming\",\n input: partialArgs\n });\n }\n\n break;\n }\n\n case \"tool-input-available\": {\n if (data.dynamic) {\n updateDynamicToolPart({\n toolCallId: data.toolCallId,\n toolName: data.toolName,\n state: \"input-available\",\n input: data.input,\n providerMetadata: data.providerMetadata\n });\n } else {\n updateToolPart({\n toolCallId: data.toolCallId,\n toolName: data.toolName,\n state: \"input-available\",\n input: data.input,\n providerExecuted: data.providerExecuted,\n providerMetadata: data.providerMetadata\n });\n }\n\n // TODO: Do we want to expose onToolCall?\n\n // invoke the onToolCall callback if it exists. This is blocking.\n // In the future we should make this non-blocking, which\n // requires additional state management for error handling etc.\n // Skip calling onToolCall for provider-executed tools since they are already executed\n // if (onToolCall && !data.providerExecuted) {\n // await onToolCall({\n // toolCall: data\n // });\n // }\n break;\n }\n\n case \"tool-input-error\": {\n if (data.dynamic) {\n updateDynamicToolPart({\n toolCallId: data.toolCallId,\n toolName: data.toolName,\n state: \"output-error\",\n input: data.input,\n errorText: data.errorText,\n providerMetadata: data.providerMetadata\n });\n } else {\n updateToolPart({\n toolCallId: data.toolCallId,\n toolName: data.toolName,\n state: \"output-error\",\n input: undefined,\n rawInput: data.input,\n errorText: data.errorText,\n providerExecuted: data.providerExecuted,\n providerMetadata: data.providerMetadata\n });\n }\n\n break;\n }\n\n case \"tool-output-available\": {\n if (data.dynamic) {\n const toolInvocations = message.parts.filter(\n (part) => part.type === \"dynamic-tool\"\n ) as DynamicToolUIPart[];\n\n const toolInvocation = toolInvocations.find(\n (invocation) =>\n invocation.toolCallId === data.toolCallId\n );\n\n if (!toolInvocation)\n throw new Error(\"Tool invocation not found\");\n\n updateDynamicToolPart({\n toolCallId: data.toolCallId,\n toolName: toolInvocation.toolName,\n state: \"output-available\",\n input: toolInvocation.input,\n output: data.output,\n preliminary: data.preliminary\n });\n } else {\n const toolInvocations =\n message.parts.filter(isToolUIPart);\n\n const toolInvocation = toolInvocations.find(\n (invocation) =>\n invocation.toolCallId === data.toolCallId\n );\n\n if (!toolInvocation)\n throw new Error(\"Tool invocation not found\");\n\n updateToolPart({\n toolCallId: data.toolCallId,\n toolName: getToolName(toolInvocation),\n state: \"output-available\",\n input: toolInvocation.input,\n output: data.output,\n providerExecuted: data.providerExecuted,\n preliminary: data.preliminary\n });\n }\n\n break;\n }\n\n case \"tool-output-error\": {\n if (data.dynamic) {\n const toolInvocations = message.parts.filter(\n (part) => part.type === \"dynamic-tool\"\n ) as DynamicToolUIPart[];\n\n const toolInvocation = toolInvocations.find(\n (invocation) =>\n invocation.toolCallId === data.toolCallId\n );\n\n if (!toolInvocation)\n throw new Error(\"Tool invocation not found\");\n\n updateDynamicToolPart({\n toolCallId: data.toolCallId,\n toolName: toolInvocation.toolName,\n state: \"output-error\",\n input: toolInvocation.input,\n errorText: data.errorText\n });\n } else {\n const toolInvocations =\n message.parts.filter(isToolUIPart);\n\n const toolInvocation = toolInvocations.find(\n (invocation) =>\n invocation.toolCallId === data.toolCallId\n );\n\n if (!toolInvocation)\n throw new Error(\"Tool invocation not found\");\n updateToolPart({\n toolCallId: data.toolCallId,\n toolName: getToolName(toolInvocation),\n state: \"output-error\",\n input: toolInvocation.input,\n rawInput:\n \"rawInput\" in toolInvocation\n ? toolInvocation.rawInput\n : undefined,\n errorText: data.errorText\n });\n }\n\n break;\n }\n\n case \"start-step\": {\n // add a step boundary part to the message\n message.parts.push({ type: \"step-start\" });\n break;\n }\n\n case \"finish-step\": {\n // reset the current text and reasoning parts\n activeTextParts = {};\n activeReasoningParts = {};\n break;\n }\n\n case \"start\": {\n if (data.messageId != null) {\n message.id = data.messageId;\n }\n\n await updateMessageMetadata(data.messageMetadata);\n\n break;\n }\n\n case \"finish\": {\n await updateMessageMetadata(data.messageMetadata);\n break;\n }\n\n case \"message-metadata\": {\n await updateMessageMetadata(data.messageMetadata);\n break;\n }\n\n case \"error\": {\n this._broadcastChatMessage({\n error: true,\n body: data.errorText ?? JSON.stringify(data),\n done: false,\n id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n });\n\n break;\n }\n // Do we want to handle data parts?\n }\n\n // Always forward the raw part to the client\n this._broadcastChatMessage({\n body: JSON.stringify(data),\n done: false,\n id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n });\n } catch (_e) {\n // Skip malformed JSON lines silently\n }\n }\n }\n } else {\n // Handle plain text responses (e.g., from generateText)\n // Treat the entire chunk as a text delta to preserve exact formatting\n if (chunk.length > 0) {\n message.parts.push({ type: \"text\", text: chunk });\n // Synthesize a text-delta event so clients can stream-render\n this._broadcastChatMessage({\n body: JSON.stringify({ type: \"text-delta\", delta: chunk }),\n done: false,\n id,\n type: MessageType.CF_AGENT_USE_CHAT_RESPONSE\n });\n }\n }\n }\n } finally {\n reader.releaseLock();\n }\n\n if (message.parts.length > 0) {\n await this.persistMessages([...this.messages, message]);\n }\n });\n }\n\n /**\n * For the given message id, look up its associated AbortController\n * If the AbortController does not exist, create and store one in memory\n *\n * returns the AbortSignal associated with the AbortController\n */\n private _getAbortSignal(id: string): AbortSignal | undefined {\n // Defensive check, since we're coercing message types at the moment\n if (typeof id !== \"string\") {\n return undefined;\n }\n\n if (!this._chatMessageAbortControllers.has(id)) {\n this._chatMessageAbortControllers.set(id, new AbortController());\n }\n\n return this._chatMessageAbortControllers.get(id)?.signal;\n }\n\n /**\n * Remove an abort controller from the cache of pending message responses\n */\n private _removeAbortController(id: string) {\n this._chatMessageAbortControllers.delete(id);\n }\n\n /**\n * Propagate an abort signal for any requests associated with the given message id\n */\n private _cancelChatRequest(id: string) {\n if (this._chatMessageAbortControllers.has(id)) {\n const abortController = this._chatMessageAbortControllers.get(id);\n abortController?.abort();\n }\n }\n\n /**\n * Abort all pending requests and clear the cache of AbortControllers\n */\n private _destroyAbortControllers() {\n for (const controller of this._chatMessageAbortControllers.values()) {\n controller?.abort();\n }\n this._chatMessageAbortControllers.clear();\n }\n\n /**\n * When the DO is destroyed, cancel all pending requests\n */\n async destroy() {\n this._destroyAbortControllers();\n await super.destroy();\n }\n}\n"],"mappings":";;;;;;;;;;;;;AAAA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAUK;AASP,IAAM,UAAU,IAAI,YAAY;AAMzB,IAAM,cAAN,cAA0D,MAG/D;AAAA,EAQA,YAAY,KAAmB,KAAU;AACvC,UAAM,KAAK,GAAG;AACd,SAAK;AAAA;AAAA;AAAA;AAAA;AAOL,UAAM,eACJ,KAAK,gDAAgD,CAAC,GACtD,IAAI,CAAC,QAAQ;AACb,aAAO,KAAK,MAAM,IAAI,OAAiB;AAAA,IACzC,CAAC;AAGD,SAAK,WAAW,sBAAsB,WAAW;AAEjD,SAAK,+BAA+B,oBAAI,IAAI;AAAA,EAC9C;AAAA,EAEQ,sBAAsB,SAA0B,SAAoB;AAC1E,SAAK,UAAU,KAAK,UAAU,OAAO,GAAG,OAAO;AAAA,EACjD;AAAA,EAEA,MAAe,UAAU,YAAwB,SAAoB;AACnE,QAAI,OAAO,YAAY,UAAU;AAC/B,UAAI;AACJ,UAAI;AACF,eAAO,KAAK,MAAM,OAAO;AAAA,MAC3B,SAAS,QAAQ;AAGf;AAAA,MACF;AACA,UACE,KAAK,wEACL,KAAK,KAAK,WAAW,QACrB;AACA,cAAM;AAAA;AAAA;AAAA;AAAA,UAIJ;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,QAKF,IAAI,KAAK;AACT,cAAM,EAAE,SAAS,IAAI,KAAK,MAAM,IAAc;AAG9C,cAAM,sBAAsB,sBAAsB,QAAQ;AAE1D,aAAK;AAAA,UACH;AAAA,YACE,UAAU;AAAA,YACV;AAAA,UACF;AAAA,UACA,CAAC,WAAW,EAAE;AAAA,QAChB;AAEA,cAAM,KAAK,gBAAgB,qBAAqB,CAAC,WAAW,EAAE,CAAC;AAE/D,aAAK,eAAe;AAAA,UAClB;AAAA,YACE,gBAAgB;AAAA,YAChB,IAAI,KAAK;AAAA,YACT,SAAS,CAAC;AAAA,YACV,WAAW,KAAK,IAAI;AAAA,YACpB,MAAM;AAAA,UACR;AAAA,UACA,KAAK;AAAA,QACP;AAEA,cAAM,gBAAgB,KAAK;AAC3B,cAAM,cAAc,KAAK,gBAAgB,aAAa;AAEtD,eAAO,KAAK,cAAc,YAAY;AACpC,gBAAM,WAAW,MAAM,KAAK;AAAA,YAC1B,OAAO,kBAAkB;AACvB,mBAAK,uBAAuB,aAAa;AAEzC,mBAAK,eAAe;AAAA,gBAClB;AAAA,kBACE,gBAAgB;AAAA,kBAChB,IAAI,KAAK;AAAA,kBACT,SAAS,CAAC;AAAA,kBACV,WAAW,KAAK,IAAI;AAAA,kBACpB,MAAM;AAAA,gBACR;AAAA,gBACA,KAAK;AAAA,cACP;AAAA,YAIF;AAAA,YACA,cAAc,EAAE,YAAY,IAAI;AAAA,UAClC;AAEA,cAAI,UAAU;AACZ,kBAAM,KAAK,OAAO,KAAK,IAAI,QAAQ;AAAA,UACrC,OAAO;AAEL,oBAAQ;AAAA,cACN,uEAAuE,aAAa;AAAA,YACtF;AAEA,iBAAK;AAAA,cACH;AAAA,gBACE,MAAM;AAAA,gBACN,MAAM;AAAA,gBACN,IAAI,KAAK;AAAA,gBACT;AAAA,cACF;AAAA,cACA,CAAC,WAAW,EAAE;AAAA,YAChB;AAAA,UACF;AAAA,QACF,CAAC;AAAA,MACH;AACA,UAAI,KAAK,0DAA0C;AACjD,aAAK,yBAAyB;AAC9B,aAAK;AACL,aAAK,WAAW,CAAC;AACjB,aAAK;AAAA,UACH;AAAA,YACE;AAAA,UACF;AAAA,UACA,CAAC,WAAW,EAAE;AAAA,QAChB;AAAA,MACF,WAAW,KAAK,gEAA6C;AAE3D,cAAM,sBAAsB,sBAAsB,KAAK,QAAQ;AAC/D,cAAM,KAAK,gBAAgB,qBAAqB,CAAC,WAAW,EAAE,CAAC;AAAA,MACjE,WAAW,KAAK,4EAAmD;AAEjE,aAAK,mBAAmB,KAAK,EAAE;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAe,UAAU,SAAqC;AAC5D,WAAO,KAAK,cAAc,MAAM;AAC9B,YAAM,MAAM,IAAI,IAAI,QAAQ,GAAG;AAC/B,UAAI,IAAI,SAAS,SAAS,eAAe,GAAG;AAC1C,cAAM,YACJ,KAAK,gDAAgD,CAAC,GACtD,IAAI,CAAC,QAAQ;AACb,iBAAO,KAAK,MAAM,IAAI,OAAiB;AAAA,QACzC,CAAC;AACD,eAAO,SAAS,KAAK,QAAQ;AAAA,MAC/B;AACA,aAAO,MAAM,UAAU,OAAO;AAAA,IAChC,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,cAAiB,IAA0B;AACvD,QAAI;AACF,aAAO,MAAM,GAAG;AAAA,IAClB,SAAS,GAAG;AACV,YAAM,KAAK,QAAQ,CAAC;AAAA,IACtB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,cAEJ,UAEA,SAC+B;AAC/B,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,aAAa,UAAyB;AAC1C,UAAM,KAAK,gBAAgB,QAAQ;AAAA,EACrC;AAAA,EAEA,MAAM,gBACJ,UACA,sBAAgC,CAAC,GACjC;AACA,SAAK;AACL,eAAW,WAAW,UAAU;AAC9B,WAAK,kEACH,QAAQ,EACV,IAAI,KAAK,UAAU,OAAO,CAAC;AAAA,IAC7B;AACA,SAAK,WAAW;AAChB,SAAK;AAAA,MACH;AAAA,QACE;AAAA,QACA;AAAA,MACF;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,OAAO,IAAY,UAAoB;AACnD,WAAO,KAAK,cAAc,YAAY;AACpC,UAAI,CAAC,SAAS,MAAM;AAElB,aAAK,sBAAsB;AAAA,UACzB,MAAM;AAAA,UACN,MAAM;AAAA,UACN;AAAA,UACA;AAAA,QACF,CAAC;AACD;AAAA,MACF;AAEA,YAAM,SAAS,SAAS,KAAK,UAAU;AAIvC,YAAM,UAAuB;AAAA,QAC3B,IAAI,aAAa,KAAK,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,MAAM,GAAG,EAAE,CAAC;AAAA;AAAA,QACtE,MAAM;AAAA,QACN,OAAO,CAAC;AAAA,MACV;AACA,UAAI,kBAA8C,CAAC;AACnD,UAAI,uBAAwD,CAAC;AAC7D,YAAM,mBAGF,CAAC;AAEL,eAAS,sBACP,SA2BA;AACA,cAAM,OAAO,QAAQ,MAAM;AAAA,UACzB,CAACA,UACCA,MAAK,SAAS,kBACdA,MAAK,eAAe,QAAQ;AAAA,QAChC;AAEA,cAAM,aAAa;AACnB,cAAM,UAAU;AAEhB,YAAI,QAAQ,MAAM;AAChB,eAAK,QAAQ,QAAQ;AACrB,kBAAQ,WAAW,QAAQ;AAC3B,kBAAQ,QAAQ,WAAW;AAC3B,kBAAQ,SAAS,WAAW;AAC5B,kBAAQ,YAAY,WAAW;AAC/B,kBAAQ,WAAW,WAAW,YAAY,QAAQ;AAClD,kBAAQ,cAAc,WAAW;AAEjC,cACE,WAAW,oBAAoB,QAC/B,KAAK,UAAU,mBACf;AACA,iBAAK,uBACH,WAAW;AAAA,UACf;AAAA,QACF,OAAO;AACL,kBAAQ,MAAM,KAAK;AAAA,YACjB,MAAM;AAAA,YACN,UAAU,QAAQ;AAAA,YAClB,YAAY,QAAQ;AAAA,YACpB,OAAO,QAAQ;AAAA,YACf,OAAO,WAAW;AAAA,YAClB,QAAQ,WAAW;AAAA,YACnB,WAAW,WAAW;AAAA,YACtB,aAAa,WAAW;AAAA,YACxB,GAAI,WAAW,oBAAoB,OAC/B,EAAE,sBAAsB,WAAW,iBAAiB,IACpD,CAAC;AAAA,UACP,CAAsB;AAAA,QACxB;AAAA,MACF;AAEA,eAAS,eACP,SAgCA;AACA,cAAM,OAAO,QAAQ,MAAM;AAAA,UACzB,CAACA,UAAS,aAAaA,KAAI,KAAKA,MAAK,eAAe,QAAQ;AAAA,QAC9D;AAEA,cAAM,aAAa;AACnB,cAAM,UAAU;AAEhB,YAAI,QAAQ,MAAM;AAChB,eAAK,QAAQ,QAAQ;AACrB,kBAAQ,QAAQ,WAAW;AAC3B,kBAAQ,SAAS,WAAW;AAC5B,kBAAQ,YAAY,WAAW;AAC/B,kBAAQ,WAAW,WAAW;AAC9B,kBAAQ,cAAc,WAAW;AAGjC,kBAAQ,mBACN,WAAW,oBAAoB,KAAK;AAEtC,cACE,WAAW,oBAAoB,QAC/B,KAAK,UAAU,mBACf;AACA,iBAAK,uBACH,WAAW;AAAA,UACf;AAAA,QACF,OAAO;AACL,kBAAQ,MAAM,KAAK;AAAA,YACjB,MAAM,QAAQ,QAAQ,QAAQ;AAAA,YAC9B,YAAY,QAAQ;AAAA,YACpB,OAAO,QAAQ;AAAA,YACf,OAAO,WAAW;AAAA,YAClB,QAAQ,WAAW;AAAA,YACnB,UAAU,WAAW;AAAA,YACrB,WAAW,WAAW;AAAA,YACtB,kBAAkB,WAAW;AAAA,YAC7B,aAAa,WAAW;AAAA,YACxB,GAAI,WAAW,oBAAoB,OAC/B,EAAE,sBAAsB,WAAW,iBAAiB,IACpD,CAAC;AAAA,UACP,CAAe;AAAA,QACjB;AAAA,MACF;AAEA,qBAAe,sBAAsB,UAAmB;AACtD,YAAI,YAAY,MAAM;AACpB,gBAAM,iBACJ,QAAQ,YAAY,OAChB,EAAE,GAAG,QAAQ,UAAU,GAAG,SAAS,IACnC;AAEN,kBAAQ,WAAW;AAAA,QACrB;AAAA,MACF;AAEA,UAAI;AACF,eAAO,MAAM;AACX,gBAAM,EAAE,MAAM,MAAM,IAAI,MAAM,OAAO,KAAK;AAC1C,cAAI,MAAM;AAER,iBAAK,sBAAsB;AAAA,cACzB,MAAM;AAAA,cACN,MAAM;AAAA,cACN;AAAA,cACA;AAAA,YACF,CAAC;AACD;AAAA,UACF;AAEA,gBAAM,QAAQ,QAAQ,OAAO,KAAK;AAGlC,gBAAM,cAAc,SAAS,QAAQ,IAAI,cAAc,KAAK;AAC5D,gBAAM,QAAQ,YAAY,SAAS,mBAAmB;AAGtD,cAAI,OAAO;AAET,kBAAM,QAAQ,MAAM,MAAM,IAAI;AAC9B,uBAAW,QAAQ,OAAO;AACxB,kBAAI,KAAK,WAAW,QAAQ,KAAK,SAAS,gBAAgB;AACxD,oBAAI;AACF,wBAAM,OAAuB,KAAK,MAAM,KAAK,MAAM,CAAC,CAAC;AACrD,0BAAQ,KAAK,MAAM;AAAA,oBACjB,KAAK,cAAc;AACjB,4BAAM,WAAuB;AAAA,wBAC3B,MAAM;AAAA,wBACN,MAAM;AAAA,wBACN,kBAAkB,KAAK;AAAA,wBACvB,OAAO;AAAA,sBACT;AACA,sCAAgB,KAAK,EAAE,IAAI;AAC3B,8BAAQ,MAAM,KAAK,QAAQ;AAC3B;AAAA,oBACF;AAAA,oBAEA,KAAK,cAAc;AACjB,4BAAM,WAAW,gBAAgB,KAAK,EAAE;AACxC,+BAAS,QAAQ,KAAK;AACtB,+BAAS,mBACP,KAAK,oBAAoB,SAAS;AACpC;AAAA,oBACF;AAAA,oBAEA,KAAK,YAAY;AACf,4BAAM,WAAW,gBAAgB,KAAK,EAAE;AACxC,+BAAS,QAAQ;AACjB,+BAAS,mBACP,KAAK,oBAAoB,SAAS;AACpC,6BAAO,gBAAgB,KAAK,EAAE;AAC9B;AAAA,oBACF;AAAA,oBAEA,KAAK,mBAAmB;AACtB,4BAAM,gBAAiC;AAAA,wBACrC,MAAM;AAAA,wBACN,MAAM;AAAA,wBACN,kBAAkB,KAAK;AAAA,wBACvB,OAAO;AAAA,sBACT;AACA,2CAAqB,KAAK,EAAE,IAAI;AAChC,8BAAQ,MAAM,KAAK,aAAa;AAChC;AAAA,oBACF;AAAA,oBAEA,KAAK,mBAAmB;AACtB,4BAAM,gBAAgB,qBAAqB,KAAK,EAAE;AAClD,oCAAc,QAAQ,KAAK;AAC3B,oCAAc,mBACZ,KAAK,oBAAoB,cAAc;AACzC;AAAA,oBACF;AAAA,oBAEA,KAAK,iBAAiB;AACpB,4BAAM,gBAAgB,qBAAqB,KAAK,EAAE;AAClD,oCAAc,mBACZ,KAAK,oBAAoB,cAAc;AACzC,oCAAc,QAAQ;AACtB,6BAAO,qBAAqB,KAAK,EAAE;AAEnC;AAAA,oBACF;AAAA,oBAEA,KAAK,QAAQ;AACX,8BAAQ,MAAM,KAAK;AAAA,wBACjB,MAAM;AAAA,wBACN,WAAW,KAAK;AAAA,wBAChB,KAAK,KAAK;AAAA,sBACZ,CAAC;AAED;AAAA,oBACF;AAAA,oBAEA,KAAK,cAAc;AACjB,8BAAQ,MAAM,KAAK;AAAA,wBACjB,MAAM;AAAA,wBACN,UAAU,KAAK;AAAA,wBACf,KAAK,KAAK;AAAA,wBACV,OAAO,KAAK;AAAA,wBACZ,kBAAkB,KAAK;AAAA,sBACzB,CAAC;AAED;AAAA,oBACF;AAAA,oBAEA,KAAK,mBAAmB;AACtB,8BAAQ,MAAM,KAAK;AAAA,wBACjB,MAAM;AAAA,wBACN,UAAU,KAAK;AAAA,wBACf,WAAW,KAAK;AAAA,wBAChB,OAAO,KAAK;AAAA,wBACZ,UAAU,KAAK;AAAA,wBACf,kBAAkB,KAAK;AAAA,sBACzB,CAAC;AAED;AAAA,oBACF;AAAA,oBAEA,KAAK,oBAAoB;AACvB,4BAAM,kBACJ,QAAQ,MAAM,OAAO,YAAY;AAGnC,uCAAiB,KAAK,UAAU,IAAI;AAAA,wBAClC,MAAM;AAAA,wBACN,UAAU,KAAK;AAAA,wBACf,OAAO,gBAAgB;AAAA,wBACvB,SAAS,KAAK;AAAA,sBAChB;AAEA,0BAAI,KAAK,SAAS;AAChB,8CAAsB;AAAA,0BACpB,YAAY,KAAK;AAAA,0BACjB,UAAU,KAAK;AAAA,0BACf,OAAO;AAAA,0BACP,OAAO;AAAA,wBACT,CAAC;AAAA,sBACH,OAAO;AACL,uCAAe;AAAA,0BACb,YAAY,KAAK;AAAA,0BACjB,UAAU,KAAK;AAAA,0BACf,OAAO;AAAA,0BACP,OAAO;AAAA,wBACT,CAAC;AAAA,sBACH;AAEA;AAAA,oBACF;AAAA,oBAEA,KAAK,oBAAoB;AACvB,4BAAM,kBAAkB,iBAAiB,KAAK,UAAU;AAExD,sCAAgB,QAAQ,KAAK;AAE7B,4BAAM,EAAE,OAAO,YAAY,IAAI,MAAM;AAAA,wBACnC,gBAAgB;AAAA,sBAClB;AAEA,0BAAI,gBAAgB,SAAS;AAC3B,8CAAsB;AAAA,0BACpB,YAAY,KAAK;AAAA,0BACjB,UAAU,gBAAgB;AAAA,0BAC1B,OAAO;AAAA,0BACP,OAAO;AAAA,wBACT,CAAC;AAAA,sBACH,OAAO;AACL,uCAAe;AAAA,0BACb,YAAY,KAAK;AAAA,0BACjB,UAAU,gBAAgB;AAAA,0BAC1B,OAAO;AAAA,0BACP,OAAO;AAAA,wBACT,CAAC;AAAA,sBACH;AAEA;AAAA,oBACF;AAAA,oBAEA,KAAK,wBAAwB;AAC3B,0BAAI,KAAK,SAAS;AAChB,8CAAsB;AAAA,0BACpB,YAAY,KAAK;AAAA,0BACjB,UAAU,KAAK;AAAA,0BACf,OAAO;AAAA,0BACP,OAAO,KAAK;AAAA,0BACZ,kBAAkB,KAAK;AAAA,wBACzB,CAAC;AAAA,sBACH,OAAO;AACL,uCAAe;AAAA,0BACb,YAAY,KAAK;AAAA,0BACjB,UAAU,KAAK;AAAA,0BACf,OAAO;AAAA,0BACP,OAAO,KAAK;AAAA,0BACZ,kBAAkB,KAAK;AAAA,0BACvB,kBAAkB,KAAK;AAAA,wBACzB,CAAC;AAAA,sBACH;AAaA;AAAA,oBACF;AAAA,oBAEA,KAAK,oBAAoB;AACvB,0BAAI,KAAK,SAAS;AAChB,8CAAsB;AAAA,0BACpB,YAAY,KAAK;AAAA,0BACjB,UAAU,KAAK;AAAA,0BACf,OAAO;AAAA,0BACP,OAAO,KAAK;AAAA,0BACZ,WAAW,KAAK;AAAA,0BAChB,kBAAkB,KAAK;AAAA,wBACzB,CAAC;AAAA,sBACH,OAAO;AACL,uCAAe;AAAA,0BACb,YAAY,KAAK;AAAA,0BACjB,UAAU,KAAK;AAAA,0BACf,OAAO;AAAA,0BACP,OAAO;AAAA,0BACP,UAAU,KAAK;AAAA,0BACf,WAAW,KAAK;AAAA,0BAChB,kBAAkB,KAAK;AAAA,0BACvB,kBAAkB,KAAK;AAAA,wBACzB,CAAC;AAAA,sBACH;AAEA;AAAA,oBACF;AAAA,oBAEA,KAAK,yBAAyB;AAC5B,0BAAI,KAAK,SAAS;AAChB,8BAAM,kBAAkB,QAAQ,MAAM;AAAA,0BACpC,CAAC,SAAS,KAAK,SAAS;AAAA,wBAC1B;AAEA,8BAAM,iBAAiB,gBAAgB;AAAA,0BACrC,CAAC,eACC,WAAW,eAAe,KAAK;AAAA,wBACnC;AAEA,4BAAI,CAAC;AACH,gCAAM,IAAI,MAAM,2BAA2B;AAE7C,8CAAsB;AAAA,0BACpB,YAAY,KAAK;AAAA,0BACjB,UAAU,eAAe;AAAA,0BACzB,OAAO;AAAA,0BACP,OAAO,eAAe;AAAA,0BACtB,QAAQ,KAAK;AAAA,0BACb,aAAa,KAAK;AAAA,wBACpB,CAAC;AAAA,sBACH,OAAO;AACL,8BAAM,kBACJ,QAAQ,MAAM,OAAO,YAAY;AAEnC,8BAAM,iBAAiB,gBAAgB;AAAA,0BACrC,CAAC,eACC,WAAW,eAAe,KAAK;AAAA,wBACnC;AAEA,4BAAI,CAAC;AACH,gCAAM,IAAI,MAAM,2BAA2B;AAE7C,uCAAe;AAAA,0BACb,YAAY,KAAK;AAAA,0BACjB,UAAU,YAAY,cAAc;AAAA,0BACpC,OAAO;AAAA,0BACP,OAAO,eAAe;AAAA,0BACtB,QAAQ,KAAK;AAAA,0BACb,kBAAkB,KAAK;AAAA,0BACvB,aAAa,KAAK;AAAA,wBACpB,CAAC;AAAA,sBACH;AAEA;AAAA,oBACF;AAAA,oBAEA,KAAK,qBAAqB;AACxB,0BAAI,KAAK,SAAS;AAChB,8BAAM,kBAAkB,QAAQ,MAAM;AAAA,0BACpC,CAAC,SAAS,KAAK,SAAS;AAAA,wBAC1B;AAEA,8BAAM,iBAAiB,gBAAgB;AAAA,0BACrC,CAAC,eACC,WAAW,eAAe,KAAK;AAAA,wBACnC;AAEA,4BAAI,CAAC;AACH,gCAAM,IAAI,MAAM,2BAA2B;AAE7C,8CAAsB;AAAA,0BACpB,YAAY,KAAK;AAAA,0BACjB,UAAU,eAAe;AAAA,0BACzB,OAAO;AAAA,0BACP,OAAO,eAAe;AAAA,0BACtB,WAAW,KAAK;AAAA,wBAClB,CAAC;AAAA,sBACH,OAAO;AACL,8BAAM,kBACJ,QAAQ,MAAM,OAAO,YAAY;AAEnC,8BAAM,iBAAiB,gBAAgB;AAAA,0BACrC,CAAC,eACC,WAAW,eAAe,KAAK;AAAA,wBACnC;AAEA,4BAAI,CAAC;AACH,gCAAM,IAAI,MAAM,2BAA2B;AAC7C,uCAAe;AAAA,0BACb,YAAY,KAAK;AAAA,0BACjB,UAAU,YAAY,cAAc;AAAA,0BACpC,OAAO;AAAA,0BACP,OAAO,eAAe;AAAA,0BACtB,UACE,cAAc,iBACV,eAAe,WACf;AAAA,0BACN,WAAW,KAAK;AAAA,wBAClB,CAAC;AAAA,sBACH;AAEA;AAAA,oBACF;AAAA,oBAEA,KAAK,cAAc;AAEjB,8BAAQ,MAAM,KAAK,EAAE,MAAM,aAAa,CAAC;AACzC;AAAA,oBACF;AAAA,oBAEA,KAAK,eAAe;AAElB,wCAAkB,CAAC;AACnB,6CAAuB,CAAC;AACxB;AAAA,oBACF;AAAA,oBAEA,KAAK,SAAS;AACZ,0BAAI,KAAK,aAAa,MAAM;AAC1B,gCAAQ,KAAK,KAAK;AAAA,sBACpB;AAEA,4BAAM,sBAAsB,KAAK,eAAe;AAEhD;AAAA,oBACF;AAAA,oBAEA,KAAK,UAAU;AACb,4BAAM,sBAAsB,KAAK,eAAe;AAChD;AAAA,oBACF;AAAA,oBAEA,KAAK,oBAAoB;AACvB,4BAAM,sBAAsB,KAAK,eAAe;AAChD;AAAA,oBACF;AAAA,oBAEA,KAAK,SAAS;AACZ,2BAAK,sBAAsB;AAAA,wBACzB,OAAO;AAAA,wBACP,MAAM,KAAK,aAAa,KAAK,UAAU,IAAI;AAAA,wBAC3C,MAAM;AAAA,wBACN;AAAA,wBACA;AAAA,sBACF,CAAC;AAED;AAAA,oBACF;AAAA,kBAEF;AAGA,uBAAK,sBAAsB;AAAA,oBACzB,MAAM,KAAK,UAAU,IAAI;AAAA,oBACzB,MAAM;AAAA,oBACN;AAAA,oBACA;AAAA,kBACF,CAAC;AAAA,gBACH,SAAS,IAAI;AAAA,gBAEb;AAAA,cACF;AAAA,YACF;AAAA,UACF,OAAO;AAGL,gBAAI,MAAM,SAAS,GAAG;AACpB,sBAAQ,MAAM,KAAK,EAAE,MAAM,QAAQ,MAAM,MAAM,CAAC;AAEhD,mBAAK,sBAAsB;AAAA,gBACzB,MAAM,KAAK,UAAU,EAAE,MAAM,cAAc,OAAO,MAAM,CAAC;AAAA,gBACzD,MAAM;AAAA,gBACN;AAAA,gBACA;AAAA,cACF,CAAC;AAAA,YACH;AAAA,UACF;AAAA,QACF;AAAA,MACF,UAAE;AACA,eAAO,YAAY;AAAA,MACrB;AAEA,UAAI,QAAQ,MAAM,SAAS,GAAG;AAC5B,cAAM,KAAK,gBAAgB,CAAC,GAAG,KAAK,UAAU,OAAO,CAAC;AAAA,MACxD;AAAA,IACF,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQQ,gBAAgB,IAAqC;AAE3D,QAAI,OAAO,OAAO,UAAU;AAC1B,aAAO;AAAA,IACT;AAEA,QAAI,CAAC,KAAK,6BAA6B,IAAI,EAAE,GAAG;AAC9C,WAAK,6BAA6B,IAAI,IAAI,IAAI,gBAAgB,CAAC;AAAA,IACjE;AAEA,WAAO,KAAK,6BAA6B,IAAI,EAAE,GAAG;AAAA,EACpD;AAAA;AAAA;AAAA;AAAA,EAKQ,uBAAuB,IAAY;AACzC,SAAK,6BAA6B,OAAO,EAAE;AAAA,EAC7C;AAAA;AAAA;AAAA;AAAA,EAKQ,mBAAmB,IAAY;AACrC,QAAI,KAAK,6BAA6B,IAAI,EAAE,GAAG;AAC7C,YAAM,kBAAkB,KAAK,6BAA6B,IAAI,EAAE;AAChE,uBAAiB,MAAM;AAAA,IACzB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,2BAA2B;AACjC,eAAW,cAAc,KAAK,6BAA6B,OAAO,GAAG;AACnE,kBAAY,MAAM;AAAA,IACpB;AACA,SAAK,6BAA6B,MAAM;AAAA,EAC1C;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,UAAU;AACd,SAAK,yBAAyB;AAC9B,UAAM,MAAM,QAAQ;AAAA,EACtB;AACF;","names":["part"]}
@@ -8,9 +8,10 @@ import "cloudflare:workers";
8
8
  import "@modelcontextprotocol/sdk/client/index.js";
9
9
  import "@modelcontextprotocol/sdk/types.js";
10
10
  import "partyserver";
11
- import "./client-BohGLma8.js";
11
+ import "./client-DVoPb3-C.js";
12
12
  import "zod";
13
13
  import "@modelcontextprotocol/sdk/shared/protocol.js";
14
+ import "./mcp-BH1fJeiU.js";
14
15
  import "@modelcontextprotocol/sdk/client/sse.js";
15
16
  import "@modelcontextprotocol/sdk/client/streamableHttp.js";
16
17
  import "./mcp/do-oauth-client-provider.js";