ai 6.0.38 → 6.0.40

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # ai
 
+## 6.0.40
+
+### Patch Changes
+
+- ab57783: Add usage information to onFinish callback in useChat
+
+## 6.0.39
+
+### Patch Changes
+
+- 4e28ba0: fix(ai): propagate providerMetadata during input-streaming state
+
+  Provider-executed tools (like MCP tools) need to send metadata during the streaming phase, but the implementation only set `callProviderMetadata` when `part.state === "input-available"`. This fix removes the overly-restrictive state check and adds `callProviderMetadata` to the input-streaming state types and schemas.
+
 ## 6.0.38
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1417,6 +1417,7 @@ type UIToolInvocation<TOOL extends UITool | Tool> = {
     input: DeepPartial<asUITool<TOOL>['input']> | undefined;
     output?: never;
     errorText?: never;
+    callProviderMetadata?: ProviderMetadata;
     approval?: never;
 } | {
     state: 'input-available';
@@ -1508,6 +1509,7 @@ type DynamicToolUIPart = {
     input: unknown | undefined;
     output?: never;
     errorText?: never;
+    callProviderMetadata?: ProviderMetadata;
     approval?: never;
 } | {
     state: 'input-available';
@@ -1654,6 +1656,7 @@ declare const uiMessageChunkSchema: _ai_sdk_provider_utils.LazySchema<{
     toolCallId: string;
     toolName: string;
     providerExecuted?: boolean | undefined;
+    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
     dynamic?: boolean | undefined;
     title?: string | undefined;
 } | {
@@ -1746,6 +1749,7 @@ declare const uiMessageChunkSchema: _ai_sdk_provider_utils.LazySchema<{
 } | {
     type: "finish";
     finishReason?: "length" | "error" | "stop" | "content-filter" | "tool-calls" | "other" | undefined;
+    usage?: LanguageModelUsage | undefined;
     messageMetadata?: unknown;
 } | {
     type: "abort";
@@ -1835,6 +1839,7 @@ type UIMessageChunk<METADATA = unknown, DATA_TYPES extends UIDataTypes = UIDataT
     toolCallId: string;
     toolName: string;
     providerExecuted?: boolean;
+    providerMetadata?: ProviderMetadata;
     dynamic?: boolean;
     title?: string;
 } | {
@@ -1870,6 +1875,7 @@ type UIMessageChunk<METADATA = unknown, DATA_TYPES extends UIDataTypes = UIDataT
 } | {
     type: 'finish';
     finishReason?: FinishReason;
+    usage?: LanguageModelUsage;
     messageMetadata?: METADATA;
 } | {
     type: 'abort';
@@ -3213,6 +3219,7 @@ type ChatOnDataCallback<UI_MESSAGE extends UIMessage> = (dataPart: DataUIPart<In
  * @param isDisconnect Indicates whether the request has been ended by a network error.
  * @param isError Indicates whether the request has been ended by an error.
  * @param finishReason The reason why the generation finished.
+ * @param usage Token usage information for the response.
  */
 type ChatOnFinishCallback<UI_MESSAGE extends UIMessage> = (options: {
     message: UI_MESSAGE;
@@ -3221,6 +3228,7 @@ type ChatOnFinishCallback<UI_MESSAGE extends UIMessage> = (options: {
     isDisconnect: boolean;
     isError: boolean;
     finishReason?: FinishReason;
+    usage?: LanguageModelUsage;
 }) => void;
 interface ChatInit<UI_MESSAGE extends UIMessage> {
     /**

package/dist/index.d.ts

@@ -1417,6 +1417,7 @@ type UIToolInvocation<TOOL extends UITool | Tool> = {
     input: DeepPartial<asUITool<TOOL>['input']> | undefined;
     output?: never;
     errorText?: never;
+    callProviderMetadata?: ProviderMetadata;
     approval?: never;
 } | {
     state: 'input-available';
@@ -1508,6 +1509,7 @@ type DynamicToolUIPart = {
     input: unknown | undefined;
     output?: never;
     errorText?: never;
+    callProviderMetadata?: ProviderMetadata;
     approval?: never;
 } | {
     state: 'input-available';
@@ -1654,6 +1656,7 @@ declare const uiMessageChunkSchema: _ai_sdk_provider_utils.LazySchema<{
     toolCallId: string;
     toolName: string;
     providerExecuted?: boolean | undefined;
+    providerMetadata?: _ai_sdk_provider.SharedV3ProviderMetadata | undefined;
     dynamic?: boolean | undefined;
     title?: string | undefined;
 } | {
@@ -1746,6 +1749,7 @@ declare const uiMessageChunkSchema: _ai_sdk_provider_utils.LazySchema<{
 } | {
     type: "finish";
     finishReason?: "length" | "error" | "stop" | "content-filter" | "tool-calls" | "other" | undefined;
+    usage?: LanguageModelUsage | undefined;
     messageMetadata?: unknown;
 } | {
     type: "abort";
@@ -1835,6 +1839,7 @@ type UIMessageChunk<METADATA = unknown, DATA_TYPES extends UIDataTypes = UIDataT
     toolCallId: string;
     toolName: string;
     providerExecuted?: boolean;
+    providerMetadata?: ProviderMetadata;
     dynamic?: boolean;
     title?: string;
 } | {
@@ -1870,6 +1875,7 @@ type UIMessageChunk<METADATA = unknown, DATA_TYPES extends UIDataTypes = UIDataT
 } | {
     type: 'finish';
     finishReason?: FinishReason;
+    usage?: LanguageModelUsage;
     messageMetadata?: METADATA;
 } | {
     type: 'abort';
@@ -3213,6 +3219,7 @@ type ChatOnDataCallback<UI_MESSAGE extends UIMessage> = (dataPart: DataUIPart<In
  * @param isDisconnect Indicates whether the request has been ended by a network error.
  * @param isError Indicates whether the request has been ended by an error.
  * @param finishReason The reason why the generation finished.
+ * @param usage Token usage information for the response.
  */
 type ChatOnFinishCallback<UI_MESSAGE extends UIMessage> = (options: {
     message: UI_MESSAGE;
@@ -3221,6 +3228,7 @@ type ChatOnFinishCallback<UI_MESSAGE extends UIMessage> = (options: {
     isDisconnect: boolean;
     isError: boolean;
     finishReason?: FinishReason;
+    usage?: LanguageModelUsage;
 }) => void;
 interface ChatInit<UI_MESSAGE extends UIMessage> {
     /**