@cloudflare/ai-chat 0.0.5 → 0.0.7

package/CHANGELOG.md

@@ -1,5 +1,24 @@
 # @cloudflare/ai-chat
 
+## 0.0.7
+
+### Patch Changes
+
+- [#859](https://github.com/cloudflare/agents/pull/859) [`3de98a3`](https://github.com/cloudflare/agents/commit/3de98a398d55aeca51c7b845ed4c5d6051887d6d) Thanks [@threepointone](https://github.com/threepointone)! - broaden peer deps
+
+- [#865](https://github.com/cloudflare/agents/pull/865) [`c3211d0`](https://github.com/cloudflare/agents/commit/c3211d0b0cc36aa294c15569ae650d3afeab9926) Thanks [@threepointone](https://github.com/threepointone)! - update dependencies
+
+## 0.0.6
+
+### Patch Changes
+
+- [#829](https://github.com/cloudflare/agents/pull/829) [`83f137f`](https://github.com/cloudflare/agents/commit/83f137f7046aeafc3b480b5aa4518f6290b14406) Thanks [@Muhammad-Bin-Ali](https://github.com/Muhammad-Bin-Ali)! - Fix duplicate assistant messages when using needsApproval tools
+
+  When calling `addToolApprovalResponse`, the original assistant message is now updated in place instead of creating a duplicate with a new ID.
+
+- Updated dependencies [[`68916bf`](https://github.com/cloudflare/agents/commit/68916bfa08358d4bb5d61aff37acd8dc4ffc950e), [`3f490d0`](https://github.com/cloudflare/agents/commit/3f490d045844e4884db741afbb66ca1fe65d4093)]:
+  - agents@0.3.10
+
 ## 0.0.5
 
 ### Patch Changes

package/dist/index.d.ts

@@ -20,8 +20,8 @@ import { Agent, AgentContext } from "agents";
  * and handle them via the `onToolCall` callback in `useAgentChat`.
  */
 type ClientToolSchema = {
-  /** Unique name for the tool */ name: string; /** Human-readable description of what the tool does */
-  description?: Tool["description"]; /** JSON Schema defining the tool's input parameters */
+  /** Unique name for the tool */ name: string /** Human-readable description of what the tool does */;
+  description?: Tool["description"] /** JSON Schema defining the tool's input parameters */;
   parameters?: JSONSchema7;
 };
 /**
@@ -296,6 +296,16 @@ declare class AIChatAgent<
   private updateDynamicToolPart;
   private updateToolPart;
   private updateMessageMetadata;
+  /**
+   * Applies a tool approval response from the client, updating the persisted message.
+   * This is called when the client sends CF_AGENT_TOOL_APPROVAL for tools with needsApproval.
+   * Updates the tool part state from input-available/approval-requested to approval-responded.
+   *
+   * @param toolCallId - The tool call ID this approval is for
+   * @param approved - Whether the tool execution was approved
+   * @returns true if the approval was applied, false if the message was not found
+   */
+  private _applyToolApproval;
   private _reply;
   /**
    * Mark a stream as errored and clean up state.

package/dist/index.js

@@ -238,6 +238,11 @@ var AIChatAgent = class extends Agent {
 					});
 					return;
 				}
+				if (data.type === MessageType.CF_AGENT_TOOL_APPROVAL) {
+					const { toolCallId, approved } = data;
+					this._applyToolApproval(toolCallId, approved);
+					return;
+				}
 			}
 			return _onMessage(connection, message);
 		};
@@ -542,7 +547,7 @@ var AIChatAgent = class extends Agent {
 	*/
 	_resolveMessageForToolMerge(message) {
 		if (message.role !== "assistant") return message;
-		for (const part of message.parts) if ("toolCallId" in part && "state" in part && part.state === "output-available") {
+		for (const part of message.parts) if ("toolCallId" in part && "state" in part && (part.state === "output-available" || part.state === "approval-responded" || part.state === "approval-requested")) {
 			const toolCallId = part.toolCallId;
 			const existingMessage = this._findMessageByToolCallId(toolCallId);
 			if (existingMessage && existingMessage.id !== message.id) return {
@@ -1068,6 +1073,79 @@ var AIChatAgent = class extends Agent {
 			...metadata
 		} : metadata;
 	}
+	/**
+	* Applies a tool approval response from the client, updating the persisted message.
+	* This is called when the client sends CF_AGENT_TOOL_APPROVAL for tools with needsApproval.
+	* Updates the tool part state from input-available/approval-requested to approval-responded.
+	*
+	* @param toolCallId - The tool call ID this approval is for
+	* @param approved - Whether the tool execution was approved
+	* @returns true if the approval was applied, false if the message was not found
+	*/
+	async _applyToolApproval(toolCallId, approved) {
+		let message;
+		if (this._streamingMessage) {
+			for (const part of this._streamingMessage.parts) if ("toolCallId" in part && part.toolCallId === toolCallId) {
+				message = this._streamingMessage;
+				break;
+			}
+		}
+		if (!message) for (let attempt = 0; attempt < 10; attempt++) {
+			message = this._findMessageByToolCallId(toolCallId);
+			if (message) break;
+			await new Promise((resolve) => setTimeout(resolve, 100));
+		}
+		if (!message) {
+			console.warn(`[AIChatAgent] _applyToolApproval: Could not find message with toolCallId ${toolCallId} after retries`);
+			return false;
+		}
+		const isStreamingMessage = message === this._streamingMessage;
+		let updated = false;
+		if (isStreamingMessage) {
+			for (const part of message.parts) if ("toolCallId" in part && part.toolCallId === toolCallId && "state" in part && (part.state === "input-available" || part.state === "approval-requested")) {
+				part.state = "approval-responded";
+				part.approval = { approved };
+				updated = true;
+				break;
+			}
+		} else {
+			const updatedParts = message.parts.map((part) => {
+				if ("toolCallId" in part && part.toolCallId === toolCallId && "state" in part && (part.state === "input-available" || part.state === "approval-requested")) {
+					updated = true;
+					return {
+						...part,
+						state: "approval-responded",
+						approval: { approved }
+					};
+				}
+				return part;
+			});
+			if (updated) {
+				const updatedMessage = this._sanitizeMessageForPersistence({
+					...message,
+					parts: updatedParts
+				});
+				this.sql`
+          update cf_ai_chat_agent_messages 
+          set message = ${JSON.stringify(updatedMessage)}
+          where id = ${message.id}
+        `;
+				this.messages = autoTransformMessages(this._loadMessagesFromDb());
+			}
+		}
+		if (!updated) {
+			console.warn(`[AIChatAgent] _applyToolApproval: Tool part with toolCallId ${toolCallId} not in input-available or approval-requested state`);
+			return false;
+		}
+		if (!isStreamingMessage) {
+			const broadcastMessage = this._findMessageByToolCallId(toolCallId);
+			if (broadcastMessage) this._broadcastChatMessage({
+				type: MessageType.CF_AGENT_MESSAGE_UPDATED,
+				message: broadcastMessage
+			});
+		}
+		return true;
+	}
 	async _reply(id, response, excludeBroadcastIds = [], options = {}) {
 		const { continuation = false } = options;
 		return this._tryCatchChat(async () => {