@cloudflare/ai-chat 0.0.7 → 0.0.8

package/CHANGELOG.md

@@ -1,5 +1,21 @@
 # @cloudflare/ai-chat
 
+## 0.0.8
+
+### Patch Changes
+
+- [#882](https://github.com/cloudflare/agents/pull/882) [`584cebe`](https://github.com/cloudflare/agents/commit/584cebe882f437a685b96b26b15200dc50ba70e1) Thanks [@alexanderjacobsen](https://github.com/alexanderjacobsen)! - Fix multi-step client tool calling: pass stored client tool schemas to `onChatMessage` during tool continuations so the LLM can call additional client tools after auto-continuation. Also add a re-trigger mechanism to the client-side tool resolution effect to handle tool calls arriving during active resolution.
+
+- [#891](https://github.com/cloudflare/agents/pull/891) [`0723b99`](https://github.com/cloudflare/agents/commit/0723b9909f037d494e0c7db43e031c952578c82e) Thanks [@ask-bonk](https://github.com/apps/ask-bonk)! - Fix `getCurrentAgent()` returning `undefined` connection when used with `@cloudflare/ai-chat` and Vite SSR
+
+  Re-export `agentContext` as `__DO_NOT_USE_WILL_BREAK__agentContext` from the main `agents` entry point and update `@cloudflare/ai-chat` to import it from `agents` instead of the `agents/internal_context` subpath export. This prevents Vite SSR pre-bundling from creating two separate `AsyncLocalStorage` instances, which caused `getCurrentAgent().connection` to be `undefined` inside `onChatMessage` and tool `execute` functions.
+
+  The `agents/internal_context` subpath export has been removed from `package.json` and the deprecated `agentContext` alias has been removed from `internal_context.ts`. This was never a public API.
+
+- [#886](https://github.com/cloudflare/agents/pull/886) [`4292f6b`](https://github.com/cloudflare/agents/commit/4292f6ba6d49201c88b09553452c3b243620f35b) Thanks [@whoiskatrin](https://github.com/whoiskatrin)! - Forward custom body fields from client requests to `onChatMessage` options
+
+  Custom data sent via `prepareSendMessagesRequest` or the AI SDK's `body` option in `sendMessage` is now available in the `onChatMessage` handler through `options.body`. This allows passing dynamic context (e.g., model selection, temperature, custom metadata) from the client to the server without workarounds.
+
 ## 0.0.7
 
 ### Patch Changes

package/dist/index.d.ts

@@ -38,6 +38,14 @@ type OnChatMessageOptions = {
    * in `useAgentChat` for client-side execution.
    */
   clientTools?: ClientToolSchema[];
+  /**
+   * Custom body data sent from the client via `prepareSendMessagesRequest`
+   * or the AI SDK's `body` option in `sendMessage`.
+   *
+   * Contains all fields from the request body except `messages` and `clientTools`,
+   * which are handled separately.
+   */
+  body?: Record<string, unknown>;
 };
 /**
  * Converts client tool schemas to AI SDK tool format.
@@ -138,6 +146,12 @@ declare class AIChatAgent<
    * @internal
    */
   private _pendingResumeConnections;
+  /**
+   * Client tool schemas from the most recent chat request.
+   * Stored so they can be passed to onChatMessage during tool continuations.
+   * @internal
+   */
+  private _lastClientTools;
   /** Array of chat messages for the current conversation */
   messages: UIMessage[];
   constructor(ctx: AgentContext, env: Env);

package/dist/index.js

@@ -1,9 +1,8 @@
 import { MessageType } from "./types.js";
 import { autoTransformMessages } from "./ai-chat-v5-migration.js";
 import { jsonSchema, tool } from "ai";
-import { Agent } from "agents";
+import { Agent, __DO_NOT_USE_WILL_BREAK__agentContext } from "agents";
 import { nanoid } from "nanoid";
-import { agentContext } from "agents/internal_context";
 
 //#region src/index.ts
 /**
@@ -126,7 +125,8 @@ var AIChatAgent = class extends Agent {
 				}
 				if (data.type === MessageType.CF_AGENT_USE_CHAT_REQUEST && data.init.method === "POST") {
 					const { body } = data.init;
-					const { messages, clientTools } = JSON.parse(body);
+					const { messages, clientTools, ...customBody } = JSON.parse(body);
+					this._lastClientTools = clientTools?.length ? clientTools : void 0;
 					const transformedMessages = autoTransformMessages(messages);
 					this._broadcastChatMessage({
 						messages: transformedMessages,
@@ -143,7 +143,7 @@ var AIChatAgent = class extends Agent {
 					const chatMessageId = data.id;
 					const abortSignal = this._getAbortSignal(chatMessageId);
 					return this._tryCatchChat(async () => {
-						return agentContext.run({
+						return __DO_NOT_USE_WILL_BREAK__agentContext.run({
 							agent: this,
 							connection,
 							request: void 0,
@@ -160,7 +160,8 @@ var AIChatAgent = class extends Agent {
 								}, this.ctx);
 							}, {
 								abortSignal,
-								clientTools
+								clientTools,
+								body: Object.keys(customBody).length > 0 ? customBody : void 0
 							});
 							if (response) await this._reply(data.id, response, [connection.id]);
 							else {
@@ -184,6 +185,7 @@ var AIChatAgent = class extends Agent {
 					this._activeRequestId = null;
 					this._streamChunkIndex = 0;
 					this._pendingResumeConnections.clear();
+					this._lastClientTools = void 0;
 					this.messages = [];
 					this._broadcastChatMessage({ type: MessageType.CF_AGENT_CHAT_CLEAR }, [connection.id]);
 					return;
@@ -214,7 +216,7 @@ var AIChatAgent = class extends Agent {
 								const continuationId = nanoid();
 								const abortSignal = this._getAbortSignal(continuationId);
 								this._tryCatchChat(async () => {
-									return agentContext.run({
+									return __DO_NOT_USE_WILL_BREAK__agentContext.run({
 										agent: this,
 										connection,
 										request: void 0,
@@ -229,7 +231,10 @@ var AIChatAgent = class extends Agent {
 												timestamp: Date.now(),
 												type: "message:response"
 											}, this.ctx);
-										}, { abortSignal });
+										}, {
+											abortSignal,
+											clientTools: this._lastClientTools
+										});
 										if (response) await this._reply(continuationId, response, [], { continuation: true });
 									});
 								});