agents 0.0.0-2a6e66e → 0.0.0-2b4c060

package/dist/ai-chat-v5-migration.d.ts

@@ -0,0 +1,152 @@
+import { UIMessage } from "ai";
+
+/**
+ * AI SDK v5 Migration following https://jhak.im/blog/ai-sdk-migration-handling-previously-saved-messages
+ * Using exact types from the official AI SDK documentation
+ */
+/**
+ * AI SDK v5 Message Part types reference (from official AI SDK documentation)
+ *
+ * The migration logic below transforms legacy messages to match these official AI SDK v5 formats:
+ * - TextUIPart: { type: "text", text: string, state?: "streaming" | "done" }
+ * - ReasoningUIPart: { type: "reasoning", text: string, state?: "streaming" | "done", providerMetadata?: Record<string, unknown> }
+ * - FileUIPart: { type: "file", mediaType: string, filename?: string, url: string }
+ * - ToolUIPart: { type: `tool-${string}`, toolCallId: string, state: "input-streaming" | "input-available" | "output-available" | "output-error", input?: Record<string, unknown>, output?: unknown, errorText?: string, providerExecuted?: boolean }
+ */
+/**
+ * Tool invocation from v4 format
+ */
+type ToolInvocation = {
+  toolCallId: string;
+  toolName: string;
+  args: Record<string, unknown>;
+  result?: unknown;
+  state: "partial-call" | "call" | "result" | "error";
+};
+/**
+ * Legacy part from v4 format
+ */
+type LegacyPart = {
+  type: string;
+  text?: string;
+  url?: string;
+  data?: string;
+  mimeType?: string;
+  mediaType?: string;
+  filename?: string;
+};
+/**
+ * Legacy message format from AI SDK v4
+ */
+type LegacyMessage = {
+  id?: string;
+  role: string;
+  content: string;
+  reasoning?: string;
+  toolInvocations?: ToolInvocation[];
+  parts?: LegacyPart[];
+  [key: string]: unknown;
+};
+/**
+ * Corrupt content item
+ */
+type CorruptContentItem = {
+  type: string;
+  text: string;
+};
+/**
+ * Corrupted message format - has content as array instead of parts
+ */
+type CorruptArrayMessage = {
+  id?: string;
+  role: string;
+  content: CorruptContentItem[];
+  reasoning?: string;
+  toolInvocations?: ToolInvocation[];
+  [key: string]: unknown;
+};
+/**
+ * Union type for messages that could be in any format
+ */
+type MigratableMessage = LegacyMessage | CorruptArrayMessage | UIMessage;
+/**
+ * Checks if a message is already in the UIMessage format (has parts array)
+ */
+declare function isUIMessage(message: unknown): message is UIMessage;
+/**
+ * Input message that could be in any format - using unknown for flexibility
+ */
+type InputMessage = {
+  id?: string;
+  role?: string;
+  content?: unknown;
+  reasoning?: string;
+  toolInvocations?: unknown[];
+  parts?: unknown[];
+  [key: string]: unknown;
+};
+/**
+ * Automatic message transformer following the blog post pattern
+ * Handles comprehensive migration from AI SDK v4 to v5 format
+ * @param message - Message in any legacy format
+ * @param index - Index for ID generation fallback
+ * @returns UIMessage in v5 format
+ */
+declare function autoTransformMessage(
+  message: InputMessage,
+  index?: number
+): UIMessage;
+/**
+ * Legacy single message migration for backward compatibility
+ */
+declare function migrateToUIMessage(message: MigratableMessage): UIMessage;
+/**
+ * Automatic message transformer for arrays following the blog post pattern
+ * @param messages - Array of messages in any format
+ * @returns Array of UIMessages in v5 format
+ */
+declare function autoTransformMessages(messages: unknown[]): UIMessage[];
+/**
+ * Migrates an array of messages to UIMessage format (legacy compatibility)
+ * @param messages - Array of messages in old or new format
+ * @returns Array of UIMessages in the new format
+ */
+declare function migrateMessagesToUIFormat(
+  messages: MigratableMessage[]
+): UIMessage[];
+/**
+ * Checks if any messages in an array need migration
+ * @param messages - Array of messages to check
+ * @returns true if any messages are not in proper UIMessage format
+ */
+declare function needsMigration(messages: unknown[]): boolean;
+/**
+ * Analyzes the corruption types in a message array for debugging
+ * @param messages - Array of messages to analyze
+ * @returns Statistics about corruption types found
+ */
+declare function analyzeCorruption(messages: unknown[]): {
+  total: number;
+  clean: number;
+  legacyString: number;
+  corruptArray: number;
+  unknown: number;
+  examples: {
+    legacyString?: unknown;
+    corruptArray?: unknown;
+    unknown?: unknown;
+  };
+};
+
+export {
+  type CorruptArrayMessage,
+  type LegacyMessage,
+  type MigratableMessage,
+  analyzeCorruption,
+  autoTransformMessage,
+  autoTransformMessages,
+  isUIMessage,
+  migrateMessagesToUIFormat,
+  migrateToUIMessage,
+  needsMigration
+};

package/dist/ai-chat-v5-migration.js

@@ -0,0 +1,19 @@
+import {
+  analyzeCorruption,
+  autoTransformMessage,
+  autoTransformMessages,
+  isUIMessage,
+  migrateMessagesToUIFormat,
+  migrateToUIMessage,
+  needsMigration
+} from "./chunk-UJVEAURM.js";
+export {
+  analyzeCorruption,
+  autoTransformMessage,
+  autoTransformMessages,
+  isUIMessage,
+  migrateMessagesToUIFormat,
+  migrateToUIMessage,
+  needsMigration
+};
+//# sourceMappingURL=ai-chat-v5-migration.js.map

package/dist/ai-chat-v5-migration.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/ai-react.d.ts

@@ -1,6 +1,5 @@
-import * as ai from "ai";
-import { Message } from "ai";
-import { useChat } from "@ai-sdk/react";
+import { UseChatOptions, useChat } from "@ai-sdk/react";
+import { UIMessage, ChatInit } from "ai";
 import { useAgent } from "./react.js";
 import "partysocket";
 import "partysocket/react";
@@ -9,7 +8,7 @@ import "cloudflare:workers";
 import "@modelcontextprotocol/sdk/client/index.js";
 import "@modelcontextprotocol/sdk/types.js";
 import "partyserver";
-import "./client-DgyzBU_8.js";
+import "./client-CvaJdLQA.js";
 import "zod";
 import "@modelcontextprotocol/sdk/shared/protocol.js";
 import "@modelcontextprotocol/sdk/client/sse.js";
@@ -18,87 +17,82 @@ import "./mcp/do-oauth-client-provider.js";
 import "@modelcontextprotocol/sdk/client/auth.js";
 import "@modelcontextprotocol/sdk/shared/auth.js";
 import "./observability/index.js";
+import "./ai-types.js";
 import "./client.js";
 import "./serializable.js";
 
+type AITool<Input = unknown, Output = unknown> = {
+  description?: string;
+  inputSchema?: unknown;
+  execute?: (input: Input) => Output | Promise<Output>;
+};
 type GetInitialMessagesOptions = {
   agent: string;
   name: string;
   url: string;
 };
+type UseChatParams<M extends UIMessage = UIMessage> = ChatInit<M> &
+  UseChatOptions<M>;
 /**
  * Options for the useAgentChat hook
  */
-type UseAgentChatOptions<State> = Omit<
-  Parameters<typeof useChat>[0] & {
-    /** Agent connection from useAgent */
-    agent: ReturnType<typeof useAgent<State>>;
-    getInitialMessages?:
-      | undefined
-      | null
-      | ((options: GetInitialMessagesOptions) => Promise<Message[]>);
-  },
-  "fetch"
->;
+type UseAgentChatOptions<
+  State,
+  ChatMessage extends UIMessage = UIMessage
+> = Omit<UseChatParams<ChatMessage>, "fetch"> & {
+  /** Agent connection from useAgent */
+  agent: ReturnType<typeof useAgent<State>>;
+  getInitialMessages?:
+    | undefined
+    | null
+    | ((options: GetInitialMessagesOptions) => Promise<ChatMessage[]>);
+  /** Request credentials */
+  credentials?: RequestCredentials;
+  /** Request headers */
+  headers?: HeadersInit;
+  /**
+   * @description Whether to automatically resolve tool calls that do not require human interaction.
+   * @experimental
+   */
+  experimental_automaticToolResolution?: boolean;
+  /**
+   * @description Tools object for automatic detection of confirmation requirements.
+   * Tools without execute function will require confirmation.
+   */
+  tools?: Record<string, AITool<unknown, unknown>>;
+  /**
+   * @description Manual override for tools requiring confirmation.
+   * If not provided, will auto-detect from tools object.
+   */
+  toolsRequiringConfirmation?: string[];
+  /**
+   * When true (default), automatically sends the next message only after
+   * all pending confirmation-required tool calls have been resolved.
+   * @default true
+   */
+  autoSendAfterAllConfirmationsResolved?: boolean;
+};
 /**
  * React hook for building AI chat interfaces using an Agent
  * @param options Chat options including the agent connection
  * @returns Chat interface controls and state with added clearHistory method
  */
-declare function useAgentChat<State = unknown>(
-  options: UseAgentChatOptions<State>
-): {
-  /**
-   * Clear chat history on both client and Agent
-   */
+/**
+ * Automatically detects which tools require confirmation based on their configuration.
+ * Tools require confirmation if they have no execute function AND are not server-executed.
+ * @param tools - Record of tool name to tool definition
+ * @returns Array of tool names that require confirmation
+ */
+declare function detectToolsRequiringConfirmation(
+  tools?: Record<string, AITool<unknown, unknown>>
+): string[];
+declare function useAgentChat<
+  State = unknown,
+  ChatMessage extends UIMessage = UIMessage
+>(
+  options: UseAgentChatOptions<State, ChatMessage>
+): ReturnType<typeof useChat<ChatMessage>> & {
   clearHistory: () => void;
-  /**
-   * Set the chat messages and synchronize with the Agent
-   * @param messages New messages to set
-   */
-  setMessages: (messages: Message[]) => void;
-  messages: ai.UIMessage[];
-  error: undefined | Error;
-  append: (
-    message: Message | ai.CreateMessage,
-    chatRequestOptions?: ai.ChatRequestOptions
-  ) => Promise<string | null | undefined>;
-  reload: (
-    chatRequestOptions?: ai.ChatRequestOptions
-  ) => Promise<string | null | undefined>;
-  stop: () => void;
-  experimental_resume: () => void;
-  input: string;
-  setInput: React.Dispatch<React.SetStateAction<string>>;
-  handleInputChange: (
-    e:
-      | React.ChangeEvent<HTMLInputElement>
-      | React.ChangeEvent<HTMLTextAreaElement>
-  ) => void;
-  handleSubmit: (
-    event?: {
-      preventDefault?: () => void;
-    },
-    chatRequestOptions?: ai.ChatRequestOptions
-  ) => void;
-  metadata?: Object;
-  isLoading: boolean;
-  status: "submitted" | "streaming" | "ready" | "error";
-  data?: ai.JSONValue[];
-  setData: (
-    data:
-      | ai.JSONValue[]
-      | undefined
-      | ((data: ai.JSONValue[] | undefined) => ai.JSONValue[] | undefined)
-  ) => void;
-  id: string;
-  addToolResult: ({
-    toolCallId,
-    result
-  }: {
-    toolCallId: string;
-    result: any;
-  }) => void;
 };
 
-export { useAgentChat };
+export { type AITool, detectToolsRequiringConfirmation, useAgentChat };

package/dist/ai-react.js

@@ -1,10 +1,28 @@
+import "./chunk-AVYJQSLW.js";
+
 // src/ai-react.tsx
 import { useChat } from "@ai-sdk/react";
+import { getToolName, isToolUIPart } from "ai";
+import { DefaultChatTransport } from "ai";
 import { nanoid } from "nanoid";
-import { use, useEffect } from "react";
+import { use, useEffect, useRef } from "react";
 var requestCache = /* @__PURE__ */ new Map();
+function detectToolsRequiringConfirmation(tools) {
+  if (!tools) return [];
+  return Object.entries(tools).filter(([_name, tool]) => !tool.execute).map(([name]) => name);
+}
 function useAgentChat(options) {
-  const { agent, getInitialMessages, ...rest } = options;
+  const {
+    agent,
+    getInitialMessages,
+    messages: optionsInitialMessages,
+    experimental_automaticToolResolution,
+    tools,
+    toolsRequiringConfirmation: manualToolsRequiringConfirmation,
+    autoSendAfterAllConfirmationsResolved = true,
+    ...rest
+  } = options;
+  const toolsRequiringConfirmation = manualToolsRequiringConfirmation ?? detectToolsRequiringConfirmation(tools);
   const agentUrl = new URL(
     `${// @ts-expect-error we're using a protected _url property that includes query params
     (agent._url || agent._pkurl)?.replace("ws://", "http://").replace("wss://", "https://")}`
@@ -20,7 +38,22 @@ function useAgentChat(options) {
       credentials: options.credentials,
       headers: options.headers
     });
-    return response.json();
+    if (!response.ok) {
+      console.warn(
+        `Failed to fetch initial messages: ${response.status} ${response.statusText}`
+      );
+      return [];
+    }
+    const text = await response.text();
+    if (!text.trim()) {
+      return [];
+    }
+    try {
+      return JSON.parse(text);
+    } catch (error) {
+      console.warn("Failed to parse initial messages JSON:", error);
+      return [];
+    }
   }
   const getInitialMessagesFetch = getInitialMessages || defaultGetInitialMessagesFetch;
   function doGetInitialMessages(getInitialMessagesOptions) {
@@ -36,7 +69,7 @@ function useAgentChat(options) {
     name: agent.name,
     url: agentUrlString
   });
-  const initialMessages = initialMessagesPromise ? use(initialMessagesPromise) : rest.initialMessages ?? [];
+  const initialMessages = initialMessagesPromise ? use(initialMessagesPromise) : optionsInitialMessages ?? [];
   useEffect(() => {
     if (!initialMessagesPromise) {
       return;
@@ -62,19 +95,22 @@ function useAgentChat(options) {
       referrer,
       referrerPolicy,
       window
-      //  dispatcher, duplex
     } = options2;
     const id = nanoid(8);
     const abortController = new AbortController();
+    let controller;
+    let isToolCallInProgress = false;
     signal?.addEventListener("abort", () => {
       agent.send(
         JSON.stringify({
           id,
-          type: "cf_agent_chat_request_cancel"
+          type: "cf_agent_chat_request_cancel" /* CF_AGENT_CHAT_REQUEST_CANCEL */
         })
       );
       abortController.abort();
-      controller.close();
+      if (!isToolCallInProgress) {
+        controller.close();
+      }
     });
     agent.addEventListener(
       "message",
@@ -85,19 +121,32 @@ function useAgentChat(options) {
         } catch (_error) {
           return;
         }
-        if (data.type === "cf_agent_use_chat_response") {
+        if (data.type === "cf_agent_use_chat_response" /* CF_AGENT_USE_CHAT_RESPONSE */) {
           if (data.id === id) {
-            controller.enqueue(new TextEncoder().encode(data.body));
-            if (data.done) {
-              controller.close();
+            if (data.error) {
+              controller.error(new Error(data.body));
               abortController.abort();
+            } else {
+              if (data.body?.trim()) {
+                if (data.body.includes('"tool_calls"')) {
+                  isToolCallInProgress = true;
+                }
+                controller.enqueue(
+                  new TextEncoder().encode(`data: ${data.body}
+
+`)
+                );
+              }
+              if (data.done && !isToolCallInProgress) {
+                controller.close();
+                abortController.abort();
+              }
             }
           }
         }
       },
       { signal: abortController.signal }
     );
-    let controller;
     const stream = new ReadableStream({
       start(c) {
         controller = c;
@@ -118,47 +167,123 @@ function useAgentChat(options) {
           referrer,
           referrerPolicy,
           window
-          // dispatcher,
-          // duplex
         },
-        type: "cf_agent_use_chat_request",
+        type: "cf_agent_use_chat_request" /* CF_AGENT_USE_CHAT_REQUEST */,
         url: request.toString()
       })
     );
     return new Response(stream);
   }
+  const customTransport = {
+    sendMessages: async (options2) => {
+      const transport = new DefaultChatTransport({
+        api: agentUrlString,
+        fetch: aiFetch
+      });
+      return transport.sendMessages(options2);
+    },
+    reconnectToStream: async (options2) => {
+      const transport = new DefaultChatTransport({
+        api: agentUrlString,
+        fetch: aiFetch
+      });
+      return transport.reconnectToStream(options2);
+    }
+  };
   const useChatHelpers = useChat({
-    fetch: aiFetch,
-    initialMessages,
-    sendExtraMessageFields: true,
-    ...rest
+    ...rest,
+    messages: initialMessages,
+    transport: customTransport
   });
+  const processedToolCalls = useRef(/* @__PURE__ */ new Set());
+  const lastMessage = useChatHelpers.messages[useChatHelpers.messages.length - 1];
+  const pendingConfirmations = (() => {
+    if (!lastMessage || lastMessage.role !== "assistant") {
+      return { messageId: void 0, toolCallIds: /* @__PURE__ */ new Set() };
+    }
+    const pendingIds = /* @__PURE__ */ new Set();
+    for (const part of lastMessage.parts ?? []) {
+      if (isToolUIPart(part) && part.state === "input-available" && toolsRequiringConfirmation.includes(getToolName(part))) {
+        pendingIds.add(part.toolCallId);
+      }
+    }
+    return { messageId: lastMessage.id, toolCallIds: pendingIds };
+  })();
+  const pendingConfirmationsRef = useRef(pendingConfirmations);
+  pendingConfirmationsRef.current = pendingConfirmations;
+  useEffect(() => {
+    if (!experimental_automaticToolResolution) {
+      return;
+    }
+    const lastMessage2 = useChatHelpers.messages[useChatHelpers.messages.length - 1];
+    if (!lastMessage2 || lastMessage2.role !== "assistant") {
+      return;
+    }
+    const toolCalls = lastMessage2.parts.filter(
+      (part) => isToolUIPart(part) && part.state === "input-available" && !processedToolCalls.current.has(part.toolCallId)
+    );
+    if (toolCalls.length > 0) {
+      (async () => {
+        const toolCallsToResolve = toolCalls.filter(
+          (part) => isToolUIPart(part) && !toolsRequiringConfirmation.includes(getToolName(part)) && tools?.[getToolName(part)]?.execute
+          // Only execute if client has execute function
+        );
+        if (toolCallsToResolve.length > 0) {
+          for (const part of toolCallsToResolve) {
+            if (isToolUIPart(part)) {
+              processedToolCalls.current.add(part.toolCallId);
+              let toolOutput = null;
+              const toolName = getToolName(part);
+              const tool = tools?.[toolName];
+              if (tool?.execute && part.input) {
+                try {
+                  toolOutput = await tool.execute(part.input);
+                } catch (error) {
+                  toolOutput = `Error executing tool: ${error instanceof Error ? error.message : String(error)}`;
+                }
+              }
+              await useChatHelpers.addToolResult({
+                toolCallId: part.toolCallId,
+                tool: toolName,
+                output: toolOutput
+              });
+            }
+          }
+          if (pendingConfirmationsRef.current.toolCallIds.size === 0) {
+            useChatHelpers.sendMessage();
+          }
+        }
+      })();
+    }
+  }, [
+    useChatHelpers.messages,
+    experimental_automaticToolResolution,
+    useChatHelpers.addToolResult,
+    useChatHelpers.sendMessage,
+    toolsRequiringConfirmation
+  ]);
   useEffect(() => {
     function onClearHistory(event) {
-      if (typeof event.data !== "string") {
-        return;
-      }
+      if (typeof event.data !== "string") return;
       let data;
       try {
         data = JSON.parse(event.data);
       } catch (_error) {
         return;
       }
-      if (data.type === "cf_agent_chat_clear") {
+      if (data.type === "cf_agent_chat_clear" /* CF_AGENT_CHAT_CLEAR */) {
         useChatHelpers.setMessages([]);
       }
     }
     function onMessages(event) {
-      if (typeof event.data !== "string") {
-        return;
-      }
+      if (typeof event.data !== "string") return;
       let data;
       try {
         data = JSON.parse(event.data);
       } catch (_error) {
         return;
       }
-      if (data.type === "cf_agent_chat_messages") {
+      if (data.type === "cf_agent_chat_messages" /* CF_AGENT_CHAT_MESSAGES */) {
         useChatHelpers.setMessages(data.messages);
       }
     }
@@ -169,35 +294,50 @@ function useAgentChat(options) {
       agent.removeEventListener("message", onMessages);
     };
   }, [agent, useChatHelpers.setMessages]);
+  const addToolResultAndSendMessage = async (args) => {
+    const { toolCallId } = args;
+    await useChatHelpers.addToolResult(args);
+    if (!autoSendAfterAllConfirmationsResolved) {
+      useChatHelpers.sendMessage();
+      return;
+    }
+    const pending = pendingConfirmationsRef.current?.toolCallIds;
+    if (!pending) {
+      useChatHelpers.sendMessage();
+      return;
+    }
+    const wasLast = pending.size === 1 && pending.has(toolCallId);
+    if (pending.has(toolCallId)) {
+      pending.delete(toolCallId);
+    }
+    if (wasLast || pending.size === 0) {
+      useChatHelpers.sendMessage();
+    }
+  };
   return {
     ...useChatHelpers,
-    /**
-     * Clear chat history on both client and Agent
-     */
+    addToolResult: addToolResultAndSendMessage,
     clearHistory: () => {
       useChatHelpers.setMessages([]);
       agent.send(
         JSON.stringify({
-          type: "cf_agent_chat_clear"
+          type: "cf_agent_chat_clear" /* CF_AGENT_CHAT_CLEAR */
         })
       );
     },
-    /**
-     * Set the chat messages and synchronize with the Agent
-     * @param messages New messages to set
-     */
     setMessages: (messages) => {
       useChatHelpers.setMessages(messages);
       agent.send(
         JSON.stringify({
-          messages,
-          type: "cf_agent_chat_messages"
+          messages: Array.isArray(messages) ? messages : [],
+          type: "cf_agent_chat_messages" /* CF_AGENT_CHAT_MESSAGES */
         })
       );
     }
   };
 }
 export {
+  detectToolsRequiringConfirmation,
   useAgentChat
 };
 //# sourceMappingURL=ai-react.js.map