agents 0.0.0-cb5957e → 0.0.0-cccbd0f

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 (99) hide show
  1. package/README.md +5 -3
  2. package/dist/ai-chat-agent.d.ts +91 -20
  3. package/dist/ai-chat-agent.js +798 -644
  4. package/dist/ai-chat-agent.js.map +1 -1
  5. package/dist/ai-chat-v5-migration-BSiGZmYU.js +155 -0
  6. package/dist/ai-chat-v5-migration-BSiGZmYU.js.map +1 -0
  7. package/dist/ai-chat-v5-migration.d.ts +7 -4
  8. package/dist/ai-chat-v5-migration.js +3 -20
  9. package/dist/ai-react.d.ts +19 -24
  10. package/dist/ai-react.js +393 -349
  11. package/dist/ai-react.js.map +1 -1
  12. package/dist/ai-types-81H_-Uxh.d.ts +103 -0
  13. package/dist/ai-types-CrMqkwc_.js +24 -0
  14. package/dist/ai-types-CrMqkwc_.js.map +1 -0
  15. package/dist/ai-types.d.ts +6 -92
  16. package/dist/ai-types.js +3 -8
  17. package/dist/cli.d.ts +8 -0
  18. package/dist/cli.js +27 -0
  19. package/dist/cli.js.map +1 -0
  20. package/dist/client-B3SR12TQ.js +117 -0
  21. package/dist/client-B3SR12TQ.js.map +1 -0
  22. package/dist/client-BAQA84dr.d.ts +104 -0
  23. package/dist/client-BZ-xTxF5.js +901 -0
  24. package/dist/client-BZ-xTxF5.js.map +1 -0
  25. package/dist/client-ctTw3KHG.d.ts +1440 -0
  26. package/dist/client.d.ts +11 -92
  27. package/dist/client.js +4 -13
  28. package/dist/codemode/ai.d.ts +19 -17
  29. package/dist/codemode/ai.js +90 -5051
  30. package/dist/codemode/ai.js.map +1 -1
  31. package/dist/do-oauth-client-provider-Cs9QpXYp.js +93 -0
  32. package/dist/do-oauth-client-provider-Cs9QpXYp.js.map +1 -0
  33. package/dist/do-oauth-client-provider-UhQDpDb8.d.ts +134 -0
  34. package/dist/index-BUle9RiP.d.ts +58 -0
  35. package/dist/index-D2dnUH0r.d.ts +578 -0
  36. package/dist/index.d.ts +55 -542
  37. package/dist/index.js +7 -32
  38. package/dist/mcp/client.d.ts +4 -12
  39. package/dist/mcp/client.js +4 -10
  40. package/dist/mcp/do-oauth-client-provider.d.ts +2 -42
  41. package/dist/mcp/do-oauth-client-provider.js +3 -8
  42. package/dist/mcp/index.d.ts +190 -76
  43. package/dist/mcp/index.js +1428 -1080
  44. package/dist/mcp/index.js.map +1 -1
  45. package/dist/mcp/x402.d.ts +22 -27
  46. package/dist/mcp/x402.js +190 -3187
  47. package/dist/mcp/x402.js.map +1 -1
  48. package/dist/{mcp-BH1fJeiU.d.ts → mcp-BwPscEiF.d.ts} +6 -3
  49. package/dist/observability/index.d.ts +3 -34
  50. package/dist/observability/index.js +7 -12
  51. package/dist/react-B_ENAdCe.d.ts +113 -0
  52. package/dist/react.d.ts +10 -127
  53. package/dist/react.js +173 -200
  54. package/dist/react.js.map +1 -1
  55. package/dist/schedule.d.ts +13 -10
  56. package/dist/schedule.js +43 -33
  57. package/dist/schedule.js.map +1 -1
  58. package/dist/serializable-faDkMCai.d.ts +39 -0
  59. package/dist/serializable.d.ts +7 -32
  60. package/dist/serializable.js +1 -1
  61. package/dist/src-C6rC6ZpH.js +1248 -0
  62. package/dist/src-C6rC6ZpH.js.map +1 -0
  63. package/package.json +58 -32
  64. package/dist/_esm-LV5FJ3HK.js +0 -3922
  65. package/dist/_esm-LV5FJ3HK.js.map +0 -1
  66. package/dist/ai-chat-v5-migration.js.map +0 -1
  67. package/dist/ai-types.js.map +0 -1
  68. package/dist/ccip-CMBYN64O.js +0 -15
  69. package/dist/ccip-CMBYN64O.js.map +0 -1
  70. package/dist/chunk-254F4GDT.js +0 -1343
  71. package/dist/chunk-254F4GDT.js.map +0 -1
  72. package/dist/chunk-3OT2NNEW.js +0 -941
  73. package/dist/chunk-3OT2NNEW.js.map +0 -1
  74. package/dist/chunk-5Y6BEZDY.js +0 -276
  75. package/dist/chunk-5Y6BEZDY.js.map +0 -1
  76. package/dist/chunk-BER7KXUJ.js +0 -18
  77. package/dist/chunk-BER7KXUJ.js.map +0 -1
  78. package/dist/chunk-JJBFIGUC.js +0 -5202
  79. package/dist/chunk-JJBFIGUC.js.map +0 -1
  80. package/dist/chunk-PR4QN5HX.js +0 -43
  81. package/dist/chunk-PR4QN5HX.js.map +0 -1
  82. package/dist/chunk-QEVM4BVL.js +0 -116
  83. package/dist/chunk-QEVM4BVL.js.map +0 -1
  84. package/dist/chunk-TYAY6AU6.js +0 -159
  85. package/dist/chunk-TYAY6AU6.js.map +0 -1
  86. package/dist/chunk-UJVEAURM.js +0 -150
  87. package/dist/chunk-UJVEAURM.js.map +0 -1
  88. package/dist/chunk-Z44WASMA.js +0 -114
  89. package/dist/chunk-Z44WASMA.js.map +0 -1
  90. package/dist/client-DVoPb3-C.d.ts +0 -5120
  91. package/dist/client.js.map +0 -1
  92. package/dist/index.js.map +0 -1
  93. package/dist/mcp/client.js.map +0 -1
  94. package/dist/mcp/do-oauth-client-provider.js.map +0 -1
  95. package/dist/observability/index.js.map +0 -1
  96. package/dist/secp256k1-M22GZP2U.js +0 -2193
  97. package/dist/secp256k1-M22GZP2U.js.map +0 -1
  98. package/dist/serializable.js.map +0 -1
  99. package/src/index.ts +0 -2031
package/dist/ai-react.js CHANGED
@@ -1,356 +1,400 @@
1
- import "./chunk-BER7KXUJ.js";
2
- import "./chunk-PR4QN5HX.js";
3
-
4
- // src/ai-react.tsx
5
- import { useChat } from "@ai-sdk/react";
6
- import { getToolName, isToolUIPart } from "ai";
7
- import { DefaultChatTransport } from "ai";
1
+ import { t as MessageType } from "./ai-types-CrMqkwc_.js";
8
2
  import { nanoid } from "nanoid";
3
+ import { useChat } from "@ai-sdk/react";
4
+ import { DefaultChatTransport, getToolName, isToolUIPart } from "ai";
9
5
  import { use, useCallback, useEffect, useMemo, useRef } from "react";
10
- var requestCache = /* @__PURE__ */ new Map();
6
+
7
+ //#region src/ai-react.tsx
8
+ const requestCache = /* @__PURE__ */ new Map();
9
+ /**
10
+ * React hook for building AI chat interfaces using an Agent
11
+ * @param options Chat options including the agent connection
12
+ * @returns Chat interface controls and state with added clearHistory method
13
+ */
14
+ /**
15
+ * Automatically detects which tools require confirmation based on their configuration.
16
+ * Tools require confirmation if they have no execute function AND are not server-executed.
17
+ * @param tools - Record of tool name to tool definition
18
+ * @returns Array of tool names that require confirmation
19
+ */
11
20
  function detectToolsRequiringConfirmation(tools) {
12
- if (!tools) return [];
13
- return Object.entries(tools).filter(([_name, tool]) => !tool.execute).map(([name]) => name);
21
+ if (!tools) return [];
22
+ return Object.entries(tools).filter(([_name, tool$1]) => !tool$1.execute).map(([name]) => name);
14
23
  }
15
24
  function useAgentChat(options) {
16
- const {
17
- agent,
18
- getInitialMessages,
19
- messages: optionsInitialMessages,
20
- experimental_automaticToolResolution,
21
- tools,
22
- toolsRequiringConfirmation: manualToolsRequiringConfirmation,
23
- autoSendAfterAllConfirmationsResolved = true,
24
- ...rest
25
- } = options;
26
- const toolsRequiringConfirmation = manualToolsRequiringConfirmation ?? detectToolsRequiringConfirmation(tools);
27
- const agentUrl = new URL(
28
- `${// @ts-expect-error we're using a protected _url property that includes query params
29
- (agent._url || agent._pkurl)?.replace("ws://", "http://").replace("wss://", "https://")}`
30
- );
31
- agentUrl.searchParams.delete("_pk");
32
- const agentUrlString = agentUrl.toString();
33
- const agentRef = useRef(agent);
34
- useEffect(() => {
35
- agentRef.current = agent;
36
- }, [agent]);
37
- async function defaultGetInitialMessagesFetch({
38
- url
39
- }) {
40
- const getMessagesUrl = new URL(url);
41
- getMessagesUrl.pathname += "/get-messages";
42
- const response = await fetch(getMessagesUrl.toString(), {
43
- credentials: options.credentials,
44
- headers: options.headers
45
- });
46
- if (!response.ok) {
47
- console.warn(
48
- `Failed to fetch initial messages: ${response.status} ${response.statusText}`
49
- );
50
- return [];
51
- }
52
- const text = await response.text();
53
- if (!text.trim()) {
54
- return [];
55
- }
56
- try {
57
- return JSON.parse(text);
58
- } catch (error) {
59
- console.warn("Failed to parse initial messages JSON:", error);
60
- return [];
61
- }
62
- }
63
- const getInitialMessagesFetch = getInitialMessages || defaultGetInitialMessagesFetch;
64
- function doGetInitialMessages(getInitialMessagesOptions) {
65
- if (requestCache.has(agentUrlString)) {
66
- return requestCache.get(agentUrlString);
67
- }
68
- const promise = getInitialMessagesFetch(getInitialMessagesOptions);
69
- requestCache.set(agentUrlString, promise);
70
- return promise;
71
- }
72
- const initialMessagesPromise = getInitialMessages === null ? null : doGetInitialMessages({
73
- agent: agent.agent,
74
- name: agent.name,
75
- url: agentUrlString
76
- });
77
- const initialMessages = initialMessagesPromise ? use(initialMessagesPromise) : optionsInitialMessages ?? [];
78
- useEffect(() => {
79
- if (!initialMessagesPromise) {
80
- return;
81
- }
82
- requestCache.set(agentUrlString, initialMessagesPromise);
83
- return () => {
84
- if (requestCache.get(agentUrlString) === initialMessagesPromise) {
85
- requestCache.delete(agentUrlString);
86
- }
87
- };
88
- }, [agentUrlString, initialMessagesPromise]);
89
- const aiFetch = useCallback(
90
- async (request, options2 = {}) => {
91
- const {
92
- method,
93
- keepalive,
94
- headers,
95
- body,
96
- redirect,
97
- integrity,
98
- signal,
99
- credentials,
100
- mode,
101
- referrer,
102
- referrerPolicy,
103
- window
104
- } = options2;
105
- const id = nanoid(8);
106
- const abortController = new AbortController();
107
- let controller;
108
- let isToolCallInProgress = false;
109
- const currentAgent = agentRef.current;
110
- signal?.addEventListener("abort", () => {
111
- currentAgent.send(
112
- JSON.stringify({
113
- id,
114
- type: "cf_agent_chat_request_cancel" /* CF_AGENT_CHAT_REQUEST_CANCEL */
115
- })
116
- );
117
- abortController.abort();
118
- if (!isToolCallInProgress) {
119
- controller.close();
120
- }
121
- });
122
- currentAgent.addEventListener(
123
- "message",
124
- (event) => {
125
- let data;
126
- try {
127
- data = JSON.parse(event.data);
128
- } catch (_error) {
129
- return;
130
- }
131
- if (data.type === "cf_agent_use_chat_response" /* CF_AGENT_USE_CHAT_RESPONSE */) {
132
- if (data.id === id) {
133
- if (data.error) {
134
- controller.error(new Error(data.body));
135
- abortController.abort();
136
- } else {
137
- if (data.body?.trim()) {
138
- if (data.body.includes('"tool_calls"')) {
139
- isToolCallInProgress = true;
140
- }
141
- controller.enqueue(
142
- new TextEncoder().encode(`data: ${data.body}
143
-
144
- `)
145
- );
146
- }
147
- if (data.done && !isToolCallInProgress) {
148
- controller.close();
149
- abortController.abort();
150
- }
151
- }
152
- }
153
- }
154
- },
155
- { signal: abortController.signal }
156
- );
157
- const stream = new ReadableStream({
158
- start(c) {
159
- controller = c;
160
- }
161
- });
162
- currentAgent.send(
163
- JSON.stringify({
164
- id,
165
- init: {
166
- body,
167
- credentials,
168
- headers,
169
- integrity,
170
- keepalive,
171
- method,
172
- mode,
173
- redirect,
174
- referrer,
175
- referrerPolicy,
176
- window
177
- },
178
- type: "cf_agent_use_chat_request" /* CF_AGENT_USE_CHAT_REQUEST */,
179
- url: request.toString()
180
- })
181
- );
182
- return new Response(stream);
183
- },
184
- []
185
- );
186
- const customTransport = useMemo(
187
- () => ({
188
- sendMessages: async (options2) => {
189
- const transport = new DefaultChatTransport({
190
- api: agentUrlString,
191
- fetch: aiFetch
192
- });
193
- return transport.sendMessages(options2);
194
- },
195
- reconnectToStream: async (options2) => {
196
- const transport = new DefaultChatTransport({
197
- api: agentUrlString,
198
- fetch: aiFetch
199
- });
200
- return transport.reconnectToStream(options2);
201
- }
202
- }),
203
- [agentUrlString, aiFetch]
204
- );
205
- const useChatHelpers = useChat({
206
- ...rest,
207
- messages: initialMessages,
208
- transport: customTransport,
209
- id: agent._pk
210
- });
211
- const processedToolCalls = useRef(/* @__PURE__ */ new Set());
212
- const lastMessage = useChatHelpers.messages[useChatHelpers.messages.length - 1];
213
- const pendingConfirmations = (() => {
214
- if (!lastMessage || lastMessage.role !== "assistant") {
215
- return { messageId: void 0, toolCallIds: /* @__PURE__ */ new Set() };
216
- }
217
- const pendingIds = /* @__PURE__ */ new Set();
218
- for (const part of lastMessage.parts ?? []) {
219
- if (isToolUIPart(part) && part.state === "input-available" && toolsRequiringConfirmation.includes(getToolName(part))) {
220
- pendingIds.add(part.toolCallId);
221
- }
222
- }
223
- return { messageId: lastMessage.id, toolCallIds: pendingIds };
224
- })();
225
- const pendingConfirmationsRef = useRef(pendingConfirmations);
226
- pendingConfirmationsRef.current = pendingConfirmations;
227
- useEffect(() => {
228
- if (!experimental_automaticToolResolution) {
229
- return;
230
- }
231
- const lastMessage2 = useChatHelpers.messages[useChatHelpers.messages.length - 1];
232
- if (!lastMessage2 || lastMessage2.role !== "assistant") {
233
- return;
234
- }
235
- const toolCalls = lastMessage2.parts.filter(
236
- (part) => isToolUIPart(part) && part.state === "input-available" && !processedToolCalls.current.has(part.toolCallId)
237
- );
238
- if (toolCalls.length > 0) {
239
- (async () => {
240
- const toolCallsToResolve = toolCalls.filter(
241
- (part) => isToolUIPart(part) && !toolsRequiringConfirmation.includes(getToolName(part)) && tools?.[getToolName(part)]?.execute
242
- // Only execute if client has execute function
243
- );
244
- if (toolCallsToResolve.length > 0) {
245
- for (const part of toolCallsToResolve) {
246
- if (isToolUIPart(part)) {
247
- processedToolCalls.current.add(part.toolCallId);
248
- let toolOutput = null;
249
- const toolName = getToolName(part);
250
- const tool = tools?.[toolName];
251
- if (tool?.execute && part.input) {
252
- try {
253
- toolOutput = await tool.execute(part.input);
254
- } catch (error) {
255
- toolOutput = `Error executing tool: ${error instanceof Error ? error.message : String(error)}`;
256
- }
257
- }
258
- await useChatHelpers.addToolResult({
259
- toolCallId: part.toolCallId,
260
- tool: toolName,
261
- output: toolOutput
262
- });
263
- }
264
- }
265
- if (pendingConfirmationsRef.current.toolCallIds.size === 0) {
266
- useChatHelpers.sendMessage();
267
- }
268
- }
269
- })();
270
- }
271
- }, [
272
- useChatHelpers.messages,
273
- experimental_automaticToolResolution,
274
- useChatHelpers.addToolResult,
275
- useChatHelpers.sendMessage,
276
- toolsRequiringConfirmation
277
- ]);
278
- useEffect(() => {
279
- function onClearHistory(event) {
280
- if (typeof event.data !== "string") return;
281
- let data;
282
- try {
283
- data = JSON.parse(event.data);
284
- } catch (_error) {
285
- return;
286
- }
287
- if (data.type === "cf_agent_chat_clear" /* CF_AGENT_CHAT_CLEAR */) {
288
- useChatHelpers.setMessages([]);
289
- }
290
- }
291
- function onMessages(event) {
292
- if (typeof event.data !== "string") return;
293
- let data;
294
- try {
295
- data = JSON.parse(event.data);
296
- } catch (_error) {
297
- return;
298
- }
299
- if (data.type === "cf_agent_chat_messages" /* CF_AGENT_CHAT_MESSAGES */) {
300
- useChatHelpers.setMessages(data.messages);
301
- }
302
- }
303
- agent.addEventListener("message", onClearHistory);
304
- agent.addEventListener("message", onMessages);
305
- return () => {
306
- agent.removeEventListener("message", onClearHistory);
307
- agent.removeEventListener("message", onMessages);
308
- };
309
- }, [agent, useChatHelpers.setMessages]);
310
- const addToolResultAndSendMessage = async (args) => {
311
- const { toolCallId } = args;
312
- await useChatHelpers.addToolResult(args);
313
- if (!autoSendAfterAllConfirmationsResolved) {
314
- useChatHelpers.sendMessage();
315
- return;
316
- }
317
- const pending = pendingConfirmationsRef.current?.toolCallIds;
318
- if (!pending) {
319
- useChatHelpers.sendMessage();
320
- return;
321
- }
322
- const wasLast = pending.size === 1 && pending.has(toolCallId);
323
- if (pending.has(toolCallId)) {
324
- pending.delete(toolCallId);
325
- }
326
- if (wasLast || pending.size === 0) {
327
- useChatHelpers.sendMessage();
328
- }
329
- };
330
- return {
331
- ...useChatHelpers,
332
- addToolResult: addToolResultAndSendMessage,
333
- clearHistory: () => {
334
- useChatHelpers.setMessages([]);
335
- agent.send(
336
- JSON.stringify({
337
- type: "cf_agent_chat_clear" /* CF_AGENT_CHAT_CLEAR */
338
- })
339
- );
340
- },
341
- setMessages: (messages) => {
342
- useChatHelpers.setMessages(messages);
343
- agent.send(
344
- JSON.stringify({
345
- messages: Array.isArray(messages) ? messages : [],
346
- type: "cf_agent_chat_messages" /* CF_AGENT_CHAT_MESSAGES */
347
- })
348
- );
349
- }
350
- };
25
+ const { agent, getInitialMessages, messages: optionsInitialMessages, experimental_automaticToolResolution, tools, toolsRequiringConfirmation: manualToolsRequiringConfirmation, autoSendAfterAllConfirmationsResolved = true, resume = true, ...rest } = options;
26
+ const toolsRequiringConfirmation = manualToolsRequiringConfirmation ?? detectToolsRequiringConfirmation(tools);
27
+ const agentUrl = new URL(`${(agent._url || agent._pkurl)?.replace("ws://", "http://").replace("wss://", "https://")}`);
28
+ agentUrl.searchParams.delete("_pk");
29
+ const agentUrlString = agentUrl.toString();
30
+ const initialMessagesCacheKey = `${agentUrlString}|${agent.agent ?? ""}|${agent.name ?? ""}`;
31
+ const agentRef = useRef(agent);
32
+ useEffect(() => {
33
+ agentRef.current = agent;
34
+ }, [agent]);
35
+ async function defaultGetInitialMessagesFetch({ url }) {
36
+ const getMessagesUrl = new URL(url);
37
+ getMessagesUrl.pathname += "/get-messages";
38
+ const response = await fetch(getMessagesUrl.toString(), {
39
+ credentials: options.credentials,
40
+ headers: options.headers
41
+ });
42
+ if (!response.ok) {
43
+ console.warn(`Failed to fetch initial messages: ${response.status} ${response.statusText}`);
44
+ return [];
45
+ }
46
+ const text = await response.text();
47
+ if (!text.trim()) return [];
48
+ try {
49
+ return JSON.parse(text);
50
+ } catch (error) {
51
+ console.warn("Failed to parse initial messages JSON:", error);
52
+ return [];
53
+ }
54
+ }
55
+ const getInitialMessagesFetch = getInitialMessages || defaultGetInitialMessagesFetch;
56
+ function doGetInitialMessages(getInitialMessagesOptions, cacheKey) {
57
+ if (requestCache.has(cacheKey)) return requestCache.get(cacheKey);
58
+ const promise = getInitialMessagesFetch(getInitialMessagesOptions);
59
+ requestCache.set(cacheKey, promise);
60
+ return promise;
61
+ }
62
+ const initialMessagesPromise = getInitialMessages === null ? null : doGetInitialMessages({
63
+ agent: agent.agent,
64
+ name: agent.name,
65
+ url: agentUrlString
66
+ }, initialMessagesCacheKey);
67
+ const initialMessages = initialMessagesPromise ? use(initialMessagesPromise) : optionsInitialMessages ?? [];
68
+ useEffect(() => {
69
+ if (!initialMessagesPromise) return;
70
+ requestCache.set(initialMessagesCacheKey, initialMessagesPromise);
71
+ return () => {
72
+ if (requestCache.get(initialMessagesCacheKey) === initialMessagesPromise) requestCache.delete(initialMessagesCacheKey);
73
+ };
74
+ }, [initialMessagesCacheKey, initialMessagesPromise]);
75
+ const aiFetch = useCallback(async (request, options$1 = {}) => {
76
+ const { method, keepalive, headers, body, redirect, integrity, signal, credentials, mode, referrer, referrerPolicy, window } = options$1;
77
+ const id = nanoid(8);
78
+ const abortController = new AbortController();
79
+ let controller;
80
+ const currentAgent = agentRef.current;
81
+ signal?.addEventListener("abort", () => {
82
+ currentAgent.send(JSON.stringify({
83
+ id,
84
+ type: MessageType.CF_AGENT_CHAT_REQUEST_CANCEL
85
+ }));
86
+ abortController.abort();
87
+ try {
88
+ controller.close();
89
+ } catch {}
90
+ });
91
+ currentAgent.addEventListener("message", (event) => {
92
+ let data;
93
+ try {
94
+ data = JSON.parse(event.data);
95
+ } catch (_error) {
96
+ return;
97
+ }
98
+ if (data.type === MessageType.CF_AGENT_USE_CHAT_RESPONSE) {
99
+ if (data.id === id) if (data.error) {
100
+ controller.error(new Error(data.body));
101
+ abortController.abort();
102
+ } else {
103
+ if (data.body?.trim()) controller.enqueue(new TextEncoder().encode(`data: ${data.body}\n\n`));
104
+ if (data.done) {
105
+ try {
106
+ controller.close();
107
+ } catch {}
108
+ abortController.abort();
109
+ }
110
+ }
111
+ }
112
+ }, { signal: abortController.signal });
113
+ const stream = new ReadableStream({
114
+ start(c) {
115
+ controller = c;
116
+ },
117
+ cancel(reason) {
118
+ console.warn("[agents/ai-react] cancelling stream", id, reason || "no reason");
119
+ }
120
+ });
121
+ currentAgent.send(JSON.stringify({
122
+ id,
123
+ init: {
124
+ body,
125
+ credentials,
126
+ headers,
127
+ integrity,
128
+ keepalive,
129
+ method,
130
+ mode,
131
+ redirect,
132
+ referrer,
133
+ referrerPolicy,
134
+ window
135
+ },
136
+ type: MessageType.CF_AGENT_USE_CHAT_REQUEST,
137
+ url: request.toString()
138
+ }));
139
+ return new Response(stream);
140
+ }, []);
141
+ const customTransport = useMemo(() => ({
142
+ sendMessages: async (options$1) => {
143
+ return new DefaultChatTransport({
144
+ api: agentUrlString,
145
+ fetch: aiFetch
146
+ }).sendMessages(options$1);
147
+ },
148
+ reconnectToStream: async () => null
149
+ }), [agentUrlString, aiFetch]);
150
+ const useChatHelpers = useChat({
151
+ ...rest,
152
+ messages: initialMessages,
153
+ transport: customTransport,
154
+ id: agent._pk
155
+ });
156
+ const processedToolCalls = useRef(/* @__PURE__ */ new Set());
157
+ const lastMessage = useChatHelpers.messages[useChatHelpers.messages.length - 1];
158
+ const pendingConfirmations = (() => {
159
+ if (!lastMessage || lastMessage.role !== "assistant") return {
160
+ messageId: void 0,
161
+ toolCallIds: /* @__PURE__ */ new Set()
162
+ };
163
+ const pendingIds = /* @__PURE__ */ new Set();
164
+ for (const part of lastMessage.parts ?? []) if (isToolUIPart(part) && part.state === "input-available" && toolsRequiringConfirmation.includes(getToolName(part))) pendingIds.add(part.toolCallId);
165
+ return {
166
+ messageId: lastMessage.id,
167
+ toolCallIds: pendingIds
168
+ };
169
+ })();
170
+ const pendingConfirmationsRef = useRef(pendingConfirmations);
171
+ pendingConfirmationsRef.current = pendingConfirmations;
172
+ useEffect(() => {
173
+ if (!experimental_automaticToolResolution) return;
174
+ const lastMessage$1 = useChatHelpers.messages[useChatHelpers.messages.length - 1];
175
+ if (!lastMessage$1 || lastMessage$1.role !== "assistant") return;
176
+ const toolCalls = lastMessage$1.parts.filter((part) => isToolUIPart(part) && part.state === "input-available" && !processedToolCalls.current.has(part.toolCallId));
177
+ if (toolCalls.length > 0) (async () => {
178
+ const toolCallsToResolve = toolCalls.filter((part) => isToolUIPart(part) && !toolsRequiringConfirmation.includes(getToolName(part)) && tools?.[getToolName(part)]?.execute);
179
+ if (toolCallsToResolve.length > 0) {
180
+ for (const part of toolCallsToResolve) if (isToolUIPart(part)) {
181
+ processedToolCalls.current.add(part.toolCallId);
182
+ let toolOutput = null;
183
+ const toolName = getToolName(part);
184
+ const tool$1 = tools?.[toolName];
185
+ if (tool$1?.execute && part.input) try {
186
+ toolOutput = await tool$1.execute(part.input);
187
+ } catch (error) {
188
+ toolOutput = `Error executing tool: ${error instanceof Error ? error.message : String(error)}`;
189
+ }
190
+ await useChatHelpers.addToolResult({
191
+ toolCallId: part.toolCallId,
192
+ tool: toolName,
193
+ output: toolOutput
194
+ });
195
+ }
196
+ if (pendingConfirmationsRef.current.toolCallIds.size === 0) useChatHelpers.sendMessage();
197
+ }
198
+ })();
199
+ }, [
200
+ useChatHelpers.messages,
201
+ experimental_automaticToolResolution,
202
+ useChatHelpers.addToolResult,
203
+ useChatHelpers.sendMessage,
204
+ toolsRequiringConfirmation
205
+ ]);
206
+ /**
207
+ * Contains the request ID, accumulated message parts, and a unique message ID.
208
+ */
209
+ const resumedStreamRef = useRef(null);
210
+ useEffect(() => {
211
+ /**
212
+ * Unified message handler that parses JSON once and dispatches based on type.
213
+ * Avoids duplicate parsing overhead from separate listeners.
214
+ */
215
+ function onAgentMessage(event) {
216
+ if (typeof event.data !== "string") return;
217
+ let data;
218
+ try {
219
+ data = JSON.parse(event.data);
220
+ } catch (_error) {
221
+ return;
222
+ }
223
+ switch (data.type) {
224
+ case MessageType.CF_AGENT_CHAT_CLEAR:
225
+ useChatHelpers.setMessages([]);
226
+ break;
227
+ case MessageType.CF_AGENT_CHAT_MESSAGES:
228
+ useChatHelpers.setMessages(data.messages);
229
+ break;
230
+ case MessageType.CF_AGENT_STREAM_RESUMING:
231
+ if (!resume) return;
232
+ resumedStreamRef.current = null;
233
+ resumedStreamRef.current = {
234
+ id: data.id,
235
+ messageId: nanoid(),
236
+ parts: []
237
+ };
238
+ agentRef.current.send(JSON.stringify({
239
+ type: MessageType.CF_AGENT_STREAM_RESUME_ACK,
240
+ id: data.id
241
+ }));
242
+ break;
243
+ case MessageType.CF_AGENT_USE_CHAT_RESPONSE:
244
+ if (!resume || !resumedStreamRef.current) return;
245
+ if (data.id !== resumedStreamRef.current.id) return;
246
+ if (data.body?.trim()) try {
247
+ const chunkData = JSON.parse(data.body);
248
+ const resumedMsg = resumedStreamRef.current;
249
+ switch (chunkData.type) {
250
+ case "text-start":
251
+ resumedMsg.parts.push({
252
+ type: "text",
253
+ text: "",
254
+ state: "streaming"
255
+ });
256
+ break;
257
+ case "text-delta": {
258
+ const lastTextPart = [...resumedMsg.parts].reverse().find((p) => p.type === "text");
259
+ if (lastTextPart && lastTextPart.type === "text") lastTextPart.text += chunkData.delta;
260
+ else resumedMsg.parts.push({
261
+ type: "text",
262
+ text: chunkData.delta
263
+ });
264
+ break;
265
+ }
266
+ case "text-end": {
267
+ const lastTextPart = [...resumedMsg.parts].reverse().find((p) => p.type === "text");
268
+ if (lastTextPart && "state" in lastTextPart) lastTextPart.state = "done";
269
+ break;
270
+ }
271
+ case "reasoning-start":
272
+ resumedMsg.parts.push({
273
+ type: "reasoning",
274
+ text: "",
275
+ state: "streaming"
276
+ });
277
+ break;
278
+ case "reasoning-delta": {
279
+ const lastReasoningPart = [...resumedMsg.parts].reverse().find((p) => p.type === "reasoning");
280
+ if (lastReasoningPart && lastReasoningPart.type === "reasoning") lastReasoningPart.text += chunkData.delta;
281
+ break;
282
+ }
283
+ case "reasoning-end": {
284
+ const lastReasoningPart = [...resumedMsg.parts].reverse().find((p) => p.type === "reasoning");
285
+ if (lastReasoningPart && "state" in lastReasoningPart) lastReasoningPart.state = "done";
286
+ break;
287
+ }
288
+ case "file":
289
+ resumedMsg.parts.push({
290
+ type: "file",
291
+ mediaType: chunkData.mediaType,
292
+ url: chunkData.url
293
+ });
294
+ break;
295
+ case "source-url":
296
+ resumedMsg.parts.push({
297
+ type: "source-url",
298
+ sourceId: chunkData.sourceId,
299
+ url: chunkData.url,
300
+ title: chunkData.title
301
+ });
302
+ break;
303
+ case "source-document":
304
+ resumedMsg.parts.push({
305
+ type: "source-document",
306
+ sourceId: chunkData.sourceId,
307
+ mediaType: chunkData.mediaType,
308
+ title: chunkData.title,
309
+ filename: chunkData.filename
310
+ });
311
+ break;
312
+ case "tool-input-available":
313
+ resumedMsg.parts.push({
314
+ type: `tool-${chunkData.toolName}`,
315
+ toolCallId: chunkData.toolCallId,
316
+ toolName: chunkData.toolName,
317
+ state: "input-available",
318
+ input: chunkData.input
319
+ });
320
+ break;
321
+ case "tool-output-available": {
322
+ const toolPart = resumedMsg.parts.find((p) => "toolCallId" in p && p.toolCallId === chunkData.toolCallId);
323
+ if (toolPart && "state" in toolPart) {
324
+ toolPart.state = "output-available";
325
+ toolPart.output = chunkData.output;
326
+ }
327
+ break;
328
+ }
329
+ case "step-start":
330
+ resumedMsg.parts.push({ type: "step-start" });
331
+ break;
332
+ }
333
+ useChatHelpers.setMessages((prevMessages) => {
334
+ if (!resumedMsg) return prevMessages;
335
+ const existingIdx = prevMessages.findIndex((m) => m.id === resumedMsg.messageId);
336
+ const partialMessage = {
337
+ id: resumedMsg.messageId,
338
+ role: "assistant",
339
+ parts: [...resumedMsg.parts]
340
+ };
341
+ if (existingIdx >= 0) {
342
+ const updated = [...prevMessages];
343
+ updated[existingIdx] = partialMessage;
344
+ return updated;
345
+ }
346
+ return [...prevMessages, partialMessage];
347
+ });
348
+ } catch (parseError) {
349
+ console.warn("[useAgentChat] Failed to parse resumed stream chunk:", parseError instanceof Error ? parseError.message : parseError, "body:", data.body?.slice(0, 100));
350
+ }
351
+ if (data.done || data.error) resumedStreamRef.current = null;
352
+ break;
353
+ }
354
+ }
355
+ agent.addEventListener("message", onAgentMessage);
356
+ return () => {
357
+ agent.removeEventListener("message", onAgentMessage);
358
+ resumedStreamRef.current = null;
359
+ };
360
+ }, [
361
+ agent,
362
+ useChatHelpers.setMessages,
363
+ resume
364
+ ]);
365
+ const addToolResultAndSendMessage = async (args) => {
366
+ const { toolCallId } = args;
367
+ await useChatHelpers.addToolResult(args);
368
+ if (!autoSendAfterAllConfirmationsResolved) {
369
+ useChatHelpers.sendMessage();
370
+ return;
371
+ }
372
+ const pending = pendingConfirmationsRef.current?.toolCallIds;
373
+ if (!pending) {
374
+ useChatHelpers.sendMessage();
375
+ return;
376
+ }
377
+ const wasLast = pending.size === 1 && pending.has(toolCallId);
378
+ if (pending.has(toolCallId)) pending.delete(toolCallId);
379
+ if (wasLast || pending.size === 0) useChatHelpers.sendMessage();
380
+ };
381
+ return {
382
+ ...useChatHelpers,
383
+ addToolResult: addToolResultAndSendMessage,
384
+ clearHistory: () => {
385
+ useChatHelpers.setMessages([]);
386
+ agent.send(JSON.stringify({ type: MessageType.CF_AGENT_CHAT_CLEAR }));
387
+ },
388
+ setMessages: (messages) => {
389
+ useChatHelpers.setMessages(messages);
390
+ agent.send(JSON.stringify({
391
+ messages: Array.isArray(messages) ? messages : [],
392
+ type: MessageType.CF_AGENT_CHAT_MESSAGES
393
+ }));
394
+ }
395
+ };
351
396
  }
352
- export {
353
- detectToolsRequiringConfirmation,
354
- useAgentChat
355
- };
397
+
398
+ //#endregion
399
+ export { detectToolsRequiringConfirmation, useAgentChat };
356
400
  //# sourceMappingURL=ai-react.js.map