veryfront 0.1.981 → 0.1.983

package/esm/src/agent/react/use-agent-metadata.js

@@ -0,0 +1,133 @@
+import { useEffect, useState } from "../../../react/react.js";
+import { ensureError } from "../../errors/veryfront-error.js";
+function isRecord(value) {
+    return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+function getRequiredString(record, key) {
+    const value = record[key];
+    if (typeof value !== "string" || value.trim().length === 0) {
+        throw new Error(`Invalid agent metadata: ${key} is required`);
+    }
+    return value;
+}
+function getNullableString(record, key) {
+    const value = record[key];
+    if (value === undefined || value === null)
+        return null;
+    if (typeof value !== "string") {
+        throw new Error(`Invalid agent metadata: ${key} must be a string`);
+    }
+    return value;
+}
+function normalizeSuggestion(value) {
+    if (!isRecord(value)) {
+        throw new Error("Invalid agent metadata: suggestion must be an object");
+    }
+    if (value.type === "task") {
+        return {
+            type: "task",
+            id: getRequiredString(value, "id"),
+        };
+    }
+    if (value.type !== "prompt") {
+        throw new Error("Invalid agent metadata: unsupported suggestion type");
+    }
+    if (typeof value.id === "string") {
+        return {
+            id: getRequiredString(value, "id"),
+            type: "prompt",
+        };
+    }
+    return {
+        type: "prompt",
+        title: getRequiredString(value, "title"),
+        prompt: getRequiredString(value, "prompt"),
+    };
+}
+function normalizeSuggestions(value) {
+    if (value === undefined)
+        return undefined;
+    if (!isRecord(value)) {
+        throw new Error("Invalid agent metadata: suggestions must be an object");
+    }
+    const rawSuggestions = value.suggestions;
+    if (!Array.isArray(rawSuggestions)) {
+        throw new Error("Invalid agent metadata: suggestions must be an array");
+    }
+    const welcomeMessage = value.welcomeMessage;
+    return {
+        ...(typeof welcomeMessage === "string" && welcomeMessage.trim().length > 0
+            ? { welcomeMessage }
+            : {}),
+        suggestions: rawSuggestions.map(normalizeSuggestion),
+    };
+}
+/** Normalize the wire response from /api/agents/:id. */
+export function normalizeAgentMetadataResponse(value) {
+    if (!isRecord(value) || !isRecord(value.agent)) {
+        throw new Error("Invalid agent metadata response");
+    }
+    const agent = value.agent;
+    return {
+        id: getRequiredString(agent, "id"),
+        name: getRequiredString(agent, "name"),
+        description: getNullableString(agent, "description"),
+        avatarUrl: getNullableString(agent, "avatar_url"),
+        suggestions: normalizeSuggestions(agent.suggestions),
+    };
+}
+/** Return prompt text suggestions that the current Chat component can render. */
+export function getAgentPromptSuggestions(agent) {
+    return agent?.suggestions?.suggestions.flatMap((suggestion) => {
+        if (suggestion.type !== "prompt" || !("prompt" in suggestion))
+            return [];
+        return [suggestion.prompt];
+    }) ?? [];
+}
+/** React hook for browser-safe source-defined agent metadata. */
+export function useAgentMetadata(agentId) {
+    const [agent, setAgent] = useState(null);
+    const [isLoading, setIsLoading] = useState(Boolean(agentId));
+    const [error, setError] = useState(null);
+    useEffect(() => {
+        if (!agentId) {
+            setAgent(null);
+            setIsLoading(false);
+            setError(null);
+            return;
+        }
+        const resolvedAgentId = agentId;
+        const abortController = new AbortController();
+        setIsLoading(true);
+        setError(null);
+        async function loadAgentMetadata() {
+            try {
+                const response = await fetch(`/api/agents/${encodeURIComponent(resolvedAgentId)}`, {
+                    headers: { accept: "application/json" },
+                    signal: abortController.signal,
+                });
+                if (!response.ok) {
+                    throw new Error(`Agent metadata request failed: ${response.status}`);
+                }
+                const nextAgent = normalizeAgentMetadataResponse(await response.json());
+                setAgent(nextAgent);
+            }
+            catch (caught) {
+                if (caught instanceof Error && caught.name === "AbortError")
+                    return;
+                setAgent(null);
+                setError(ensureError(caught));
+            }
+            finally {
+                if (!abortController.signal.aborted) {
+                    setIsLoading(false);
+                }
+            }
+        }
+        void loadAgentMetadata();
+        return () => {
+            abortController.abort();
+        };
+    }, [agentId]);
+    return { agent, isLoading, error };
+}

package/esm/src/agent/react/use-chat/streaming/handler.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"handler.d.ts","sourceRoot":"","sources":["../../../../../../src/src/agent/react/use-chat/streaming/handler.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EAKV,kBAAkB,EAEnB,MAAM,YAAY,CAAC;AA8BpB,wBAAsB,uBAAuB,CAC3C,IAAI,EAAE,cAAc,EACpB,SAAS,EAAE,kBAAkB,GAC5B,OAAO,CAAC,IAAI,CAAC,CAyDf;AAED,wBAAsB,2BAA2B,CAC/C,IAAI,EAAE,cAAc,EACpB,SAAS,EAAE,kBAAkB,GAC5B,OAAO,CAAC,IAAI,CAAC,CA4Cf"}
+{"version":3,"file":"handler.d.ts","sourceRoot":"","sources":["../../../../../../src/src/agent/react/use-chat/streaming/handler.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,EAKV,kBAAkB,EAEnB,MAAM,YAAY,CAAC;AAgCpB,wBAAsB,uBAAuB,CAC3C,IAAI,EAAE,cAAc,EACpB,SAAS,EAAE,kBAAkB,GAC5B,OAAO,CAAC,IAAI,CAAC,CAyDf;AAED,wBAAsB,2BAA2B,CAC/C,IAAI,EAAE,cAAc,EACpB,SAAS,EAAE,kBAAkB,GAC5B,OAAO,CAAC,IAAI,CAAC,CA4Cf"}

package/esm/src/agent/react/use-chat/streaming/handler.js

@@ -1,4 +1,5 @@
 import { createAgUiChatEventDecoderState, decodeAgUiSseChunk, flushAgUiSseChunk, } from "../../../../chat/ag-ui.js";
+import { normalizeChatMessageMetadata } from "../../../../chat/chat-ui-message-helpers.js";
 import { createAssistantMessage, generateClientId } from "../utils.js";
 import { buildCurrentParts } from "./parts-builder.js";
 function createStreamingState() {
@@ -11,6 +12,7 @@ function createStreamingState() {
         dataParts: [],
         currentTextId: "",
         messageId: "",
+        messageMetadata: {},
         partOrderCounter: 0,
         currentStep: 0,
     };
@@ -101,7 +103,7 @@ function processStreamEvent(parsed, state, callbacks, getBuildParts) {
     const { onMessage, onData, onUpdate, onToolCall } = callbacks;
     switch (parsed.type) {
         case "message-start":
-            handleStart(parsed, state);
+            handleStart(parsed, state, callbacks.onUpdate, getBuildParts);
             return;
         case "step-start":
             handleStepStart(parsed, state, callbacks.onUpdate, getBuildParts);
@@ -161,9 +163,13 @@ function processChatStreamEvent(event, state, callbacks, getBuildParts) {
     const { onMessage, onData, onUpdate, onToolCall } = callbacks;
     switch (event.type) {
         case "start":
-            handleStart(event, state);
+            handleStart(event, state, callbacks.onUpdate, getBuildParts);
             return;
         case "message-metadata":
+            mergeMessageMetadata(state, event.messageMetadata);
+            if (state.messageId) {
+                emitUpdate(state, onUpdate, getBuildParts);
+            }
             onData(event.messageMetadata);
             return;
         case "start-step":
@@ -222,13 +228,32 @@ function processChatStreamEvent(event, state, callbacks, getBuildParts) {
             return;
     }
 }
-function handleStart(parsed, state) {
+function mergeMessageMetadata(state, value) {
+    state.messageMetadata = {
+        ...state.messageMetadata,
+        ...normalizeChatMessageMetadata(value),
+    };
+}
+function emitUpdate(state, onUpdate, getBuildParts) {
+    if (!state.messageId) {
+        return;
+    }
+    onUpdate?.(getBuildParts(), state.messageId, state.messageMetadata);
+}
+function handleStart(parsed, state, onUpdate, getBuildParts) {
     state.messageId = typeof parsed.messageId === "string" ? parsed.messageId : "";
     state.textBlocks.clear();
     state.toolCalls.clear();
     state.reasoningBlocks.clear();
     state.messageParts.length = 0;
     state.dataParts.length = 0;
+    state.messageMetadata = {};
+    if ("messageMetadata" in parsed) {
+        mergeMessageMetadata(state, parsed.messageMetadata);
+    }
+    if (state.messageId && Object.keys(state.messageMetadata).length > 0) {
+        emitUpdate(state, onUpdate, getBuildParts);
+    }
 }
 function handleTextStart(parsed, state) {
     state.currentTextId = parsed.id || generateClientId("text");
@@ -253,7 +278,7 @@ function handleTextDelta(parsed, state, onUpdate, getBuildParts) {
     if (block.order === null) {
         block.order = state.partOrderCounter++;
     }
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function handleTextEnd(parsed, state) {
     const textId = parsed.id || state.currentTextId;
@@ -279,7 +304,7 @@ function handleDataPart(parsed, state, onUpdate, getBuildParts) {
             data: parsed.data,
         },
     });
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function isRenderableDataPartType(type) {
     return type !== "data-state-snapshot" &&
@@ -300,7 +325,7 @@ function handleToolInputStart(parsed, state, onUpdate, getBuildParts) {
         order: state.partOrderCounter++,
     };
     state.toolCalls.set(toolCallId, toolCall);
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function handleToolInputDelta(parsed, state, onUpdate, getBuildParts) {
     const toolCallId = parsed.toolCallId;
@@ -308,7 +333,7 @@ function handleToolInputDelta(parsed, state, onUpdate, getBuildParts) {
     if (!toolCall)
         return;
     toolCall.inputText += (parsed.inputTextDelta ?? parsed.delta ?? "");
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function handleToolInputAvailable(parsed, state, onUpdate, onToolCall, getBuildParts) {
     const toolCallId = parsed.toolCallId;
@@ -360,7 +385,7 @@ function handleToolInputAvailable(parsed, state, onUpdate, onToolCall, getBuildP
             input: toolCall.input,
         });
     }
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function handleToolOutputAvailable(parsed, state, onUpdate, getBuildParts) {
     const toolCallId = parsed.toolCallId;
@@ -375,7 +400,7 @@ function handleToolOutputAvailable(parsed, state, onUpdate, getBuildParts) {
         toolName: toolCall.toolName,
         result: toolCall.output,
     });
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function handleToolError(parsed, state, onUpdate, getBuildParts) {
     const toolCallId = parsed.toolCallId;
@@ -386,7 +411,7 @@ function handleToolError(parsed, state, onUpdate, getBuildParts) {
     toolCall.error = parsed.errorText;
     if (parsed.dynamic === true)
         toolCall.dynamic = true;
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function handleReasoningStart(parsed, state, onUpdate, getBuildParts) {
     if (!state.messageId) {
@@ -400,7 +425,7 @@ function handleReasoningStart(parsed, state, onUpdate, getBuildParts) {
         order: state.partOrderCounter++,
     };
     state.reasoningBlocks.set(reasoningId, reasoning);
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function handleReasoningDelta(parsed, state, onUpdate, getBuildParts) {
     const reasoningId = parsed.id;
@@ -408,7 +433,7 @@ function handleReasoningDelta(parsed, state, onUpdate, getBuildParts) {
     if (!reasoning)
         return;
     reasoning.text += (parsed.delta ?? "");
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function handleReasoningEnd(parsed, state, onUpdate, getBuildParts) {
     const reasoningId = parsed.id;
@@ -429,7 +454,7 @@ function handleReasoningEnd(parsed, state, onUpdate, getBuildParts) {
         ...(typeof reasoning.redactedData === "string" ? { redactedData: reasoning.redactedData } : {}),
         state: "done",
     });
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function handleStepStart(_parsed, state, onUpdate, getBuildParts) {
     const stepIndex = state.currentStep;
@@ -439,7 +464,7 @@ function handleStepStart(_parsed, state, onUpdate, getBuildParts) {
         order: state.partOrderCounter++,
     };
     state.steps.set(stepIndex, step);
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function handleStepEnd(_parsed, state, onUpdate, getBuildParts) {
     const step = state.steps.get(state.currentStep);
@@ -449,11 +474,11 @@ function handleStepEnd(_parsed, state, onUpdate, getBuildParts) {
         step.order = state.partOrderCounter++;
     }
     state.currentStep++;
-    onUpdate?.(getBuildParts(), state.messageId);
+    emitUpdate(state, onUpdate, getBuildParts);
 }
 function handleFinish(state, onMessage, getBuildParts) {
     const finalParts = getBuildParts();
     if (finalParts.length > 0) {
-        onMessage(createAssistantMessage(state.messageId, finalParts));
+        onMessage(createAssistantMessage(state.messageId, finalParts, state.messageMetadata));
     }
 }

package/esm/src/agent/react/use-chat/streaming/types.d.ts

@@ -2,7 +2,7 @@ import type { ChatMessage, ChatMessagePart, ChatToolState, OnToolCallArg } from
 export interface StreamingCallbacks {
     onMessage: (message: ChatMessage) => void;
     onData: (data: unknown) => void;
-    onUpdate?: (parts: ChatMessagePart[], messageId: string) => void;
+    onUpdate?: (parts: ChatMessagePart[], messageId: string, metadata?: ChatMessage["metadata"]) => void;
     onToolCall?: (arg: OnToolCallArg) => void;
 }
 export interface StreamingToolCall {

package/esm/src/agent/react/use-chat/streaming/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../../../src/src/agent/react/use-chat/streaming/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,eAAe,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE9F,MAAM,WAAW,kBAAkB;IACjC,SAAS,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,IAAI,CAAC;IAC1C,MAAM,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,IAAI,CAAC;IAChC,QAAQ,CAAC,EAAE,CAAC,KAAK,EAAE,eAAe,EAAE,EAAE,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;IACjE,UAAU,CAAC,EAAE,CAAC,GAAG,EAAE,aAAa,KAAK,IAAI,CAAC;CAC3C;AAED,MAAM,WAAW,iBAAiB;IAChC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,aAAa,CAAC;IACrB,+DAA+D;IAC/D,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,kBAAkB;IACjC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,OAAO,CAAC;CACrB;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,WAAW,GAAG,MAAM,CAAC;IAC5B,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CACtB;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,OAAO,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,MAAM,eAAe,GAAG,iBAAiB,GAAG;IAAE,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC;AACpE,MAAM,MAAM,gBAAgB,GAAG,kBAAkB,GAAG;IAAE,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC;AACtE,MAAM,MAAM,kBAAkB,GAAG;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,eAAe,CAAA;CAAE,CAAC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../../../src/src/agent/react/use-chat/streaming/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,eAAe,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE9F,MAAM,WAAW,kBAAkB;IACjC,SAAS,EAAE,CAAC,OAAO,EAAE,WAAW,KAAK,IAAI,CAAC;IAC1C,MAAM,EAAE,CAAC,IAAI,EAAE,OAAO,KAAK,IAAI,CAAC;IAChC,QAAQ,CAAC,EAAE,CACT,KAAK,EAAE,eAAe,EAAE,EACxB,SAAS,EAAE,MAAM,EACjB,QAAQ,CAAC,EAAE,WAAW,CAAC,UAAU,CAAC,KAC/B,IAAI,CAAC;IACV,UAAU,CAAC,EAAE,CAAC,GAAG,EAAE,aAAa,KAAK,IAAI,CAAC;CAC3C;AAED,MAAM,WAAW,iBAAiB;IAChC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,aAAa,CAAC;IACrB,+DAA+D;IAC/D,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,kBAAkB;IACjC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,OAAO,CAAC;CACrB;AAED,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,WAAW,GAAG,MAAM,CAAC;IAC5B,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CACtB;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,OAAO,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,MAAM,eAAe,GAAG,iBAAiB,GAAG;IAAE,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC;AACpE,MAAM,MAAM,gBAAgB,GAAG,kBAAkB,GAAG;IAAE,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC;AACtE,MAAM,MAAM,kBAAkB,GAAG;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,eAAe,CAAA;CAAE,CAAC"}

package/esm/src/agent/react/use-chat/use-chat.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"use-chat.d.ts","sourceRoot":"","sources":["../../../../../src/src/agent/react/use-chat/use-chat.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAKH,OAAO,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AACnE,OAAO,KAAK,EAEV,WAAW,EAGX,cAAc,EACd,aAAa,EACd,MAAM,YAAY,CAAC;AAGpB,KAAK,oBAAoB,GAAG,OAAO,2BAA2B,CAAC;AAE/D,wDAAwD;AACxD,UAAU,MAAM;IACd,QAAQ,EAAE,WAAW,EAAE,CAAC;CACzB;AAED,sEAAsE;AACtE,UAAU,WAAW;IACnB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,WAAW,EAAE,CAAC;CAC7B;AAED,wBAAgB,eAAe,CAAC,eAAe,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAEnF;AAED,wBAAgB,gBAAgB,CAC9B,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,EACnC,oBAAoB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,GACxC,MAAM,GAAG,SAAS,CAGpB;AAED,wBAAgB,0BAA0B,CACxC,QAAQ,EAAE,WAAW,EAAE,EACvB,SAAS,EAAE,MAAM,EACjB,oBAAoB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,GACxC,MAAM,CAIR;AAED,wBAAgB,2BAA2B,CACzC,SAAS,EAAE,cAAc,CAAC,WAAW,CAAC,GACrC,oBAAoB,CAGtB;AAOD,wBAAgB,OAAO,CAAC,OAAO,GAAE,cAAmB,GAAG,aAAa,CAmZnE"}
+{"version":3,"file":"use-chat.d.ts","sourceRoot":"","sources":["../../../../../src/src/agent/react/use-chat/use-chat.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAKH,OAAO,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AACnE,OAAO,KAAK,EAEV,WAAW,EAGX,cAAc,EACd,aAAa,EACd,MAAM,YAAY,CAAC;AAGpB,KAAK,oBAAoB,GAAG,OAAO,2BAA2B,CAAC;AAE/D,wDAAwD;AACxD,UAAU,MAAM;IACd,QAAQ,EAAE,WAAW,EAAE,CAAC;CACzB;AAED,sEAAsE;AACtE,UAAU,WAAW;IACnB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,WAAW,EAAE,CAAC;CAC7B;AAED,wBAAgB,eAAe,CAAC,eAAe,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAEnF;AAED,wBAAgB,gBAAgB,CAC9B,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,EACnC,oBAAoB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,GACxC,MAAM,GAAG,SAAS,CAGpB;AAED,wBAAgB,0BAA0B,CACxC,QAAQ,EAAE,WAAW,EAAE,EACvB,SAAS,EAAE,MAAM,EACjB,oBAAoB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,GACxC,MAAM,CAIR;AAED,wBAAgB,2BAA2B,CACzC,SAAS,EAAE,cAAc,CAAC,WAAW,CAAC,GACrC,oBAAoB,CAGtB;AAOD,wBAAgB,OAAO,CAAC,OAAO,GAAE,cAAmB,GAAG,aAAa,CA8ZnE"}

package/esm/src/agent/react/use-chat/use-chat.js

@@ -172,22 +172,27 @@ export function useChat(options = {}) {
                         }
                     }
                 },
-                onUpdate: (parts, messageId) => {
+                onUpdate: (parts, messageId, messageMetadata) => {
                     const id = messageId ?? streamingMessageId;
+                    const metadata = { ...messageMetadata, model: serverModel };
                     if (messageId && messageId !== currentMessageIdRef.current) {
                         const oldId = currentMessageIdRef.current;
                         currentMessageIdRef.current = messageId;
                         if (hasAddedStreamingMessage) {
-                            setMessages((prev) => prev.map((m) => (m.id === oldId ? { ...m, id, parts } : m)));
+                            setMessages((prev) => prev.map((m) => m.id === oldId
+                                ? { ...m, id, parts, metadata: { ...m.metadata, ...metadata } }
+                                : m));
                             return;
                         }
                     }
                     if (!hasAddedStreamingMessage) {
                         hasAddedStreamingMessage = true;
-                        setMessages((prev) => [...prev, { id, role: "assistant", parts, metadata: { model: serverModel } }]);
+                        setMessages((prev) => [...prev, { id, role: "assistant", parts, metadata }]);
                         return;
                     }
-                    setMessages((prev) => prev.map((m) => (m.id === currentMessageIdRef.current ? { ...m, parts } : m)));
+                    setMessages((prev) => prev.map((m) => m.id === currentMessageIdRef.current
+                        ? { ...m, parts, metadata: { ...m.metadata, ...metadata } }
+                        : m));
                 },
                 onToolCall: options.onToolCall,
             });

package/esm/src/agent/react/use-chat/utils.d.ts

@@ -1,4 +1,4 @@
 import type { ChatMessage, ChatMessagePart } from "./types.js";
 export declare function generateClientId(prefix: string): string;
-export declare function createAssistantMessage(messageId: string, parts: ChatMessagePart[]): ChatMessage;
+export declare function createAssistantMessage(messageId: string, parts: ChatMessagePart[], metadata?: ChatMessage["metadata"]): ChatMessage;
 //# sourceMappingURL=utils.d.ts.map

package/esm/src/agent/react/use-chat/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../../../src/src/agent/react/use-chat/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAE/D,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAEvD;AAED,wBAAgB,sBAAsB,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,GAAG,WAAW,CAM/F"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../../../src/src/agent/react/use-chat/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAE/D,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAEvD;AAED,wBAAgB,sBAAsB,CACpC,SAAS,EAAE,MAAM,EACjB,KAAK,EAAE,eAAe,EAAE,EACxB,QAAQ,CAAC,EAAE,WAAW,CAAC,UAAU,CAAC,GACjC,WAAW,CAOb"}

package/esm/src/agent/react/use-chat/utils.js

@@ -1,10 +1,11 @@
 export function generateClientId(prefix) {
     return `${prefix}-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;
 }
-export function createAssistantMessage(messageId, parts) {
+export function createAssistantMessage(messageId, parts, metadata) {
     return {
         id: messageId || generateClientId("msg"),
         role: "assistant",
         parts,
+        ...(metadata && Object.keys(metadata).length > 0 ? { metadata } : {}),
     };
 }

package/esm/src/agent/runtime/agent-definition.d.ts

@@ -20,6 +20,7 @@ export declare const getRuntimeAgentMarkdownDefinitionSchema: () => import("../.
     id: import("../../internal-agents/schema.js").Schema<string>;
     name: import("../../internal-agents/schema.js").Schema<string>;
     description: import("../../internal-agents/schema.js").Schema<string>;
+    avatarUrl: import("../../internal-agents/schema.js").Schema<string | undefined>;
     instructions: import("../../internal-agents/schema.js").Schema<string>;
     thinking: import("../../internal-agents/schema.js").Schema<import("../../extensions/schema/schema-validator.js").InferShape<{
         enabled: import("../../internal-agents/schema.js").Schema<boolean>;
@@ -42,6 +43,7 @@ export declare const runtimeAgentMarkdownDefinitionSchema: import("../../interna
     id: import("../../internal-agents/schema.js").Schema<string>;
     name: import("../../internal-agents/schema.js").Schema<string>;
     description: import("../../internal-agents/schema.js").Schema<string>;
+    avatarUrl: import("../../internal-agents/schema.js").Schema<string | undefined>;
     instructions: import("../../internal-agents/schema.js").Schema<string>;
     thinking: import("../../internal-agents/schema.js").Schema<import("../../extensions/schema/schema-validator.js").InferShape<{
         enabled: import("../../internal-agents/schema.js").Schema<boolean>;

package/esm/src/agent/runtime/agent-definition.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-definition.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/runtime/agent-definition.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAG7D,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAGlE,wDAAwD;AACxD,eAAO,MAAM,mCAAmC;;;GAK/C,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,gCAAgC;;;GAAkD,CAAC;AAEhG,oDAAoD;AACpD,MAAM,MAAM,0BAA0B,GAAG,WAAW,CAClD,UAAU,CAAC,OAAO,mCAAmC,CAAC,CACvD,CAAC;AAEF,4DAA4D;AAC5D,eAAO,MAAM,uCAAuC;;;;;;;;;;;;;;;;GAenD,CAAC;AAEF,sDAAsD;AACtD,eAAO,MAAM,oCAAoC,uCAAuC,CAAC;AAEzF;;GAEG;AACH,eAAO,MAAM,oCAAoC;;;;;;;;;;;;;;;;GAEhD,CAAC;AAEF,6CAA6C;AAC7C,MAAM,MAAM,8BAA8B,GAAG,WAAW,CACtD,UAAU,CAAC,OAAO,uCAAuC,CAAC,CAC3D,CAAC;AAEF,wEAAwE;AACxE,eAAO,MAAM,iDAAiD;;;GAK7D,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,8CAA8C;;;GAE1D,CAAC;AAEF,iEAAiE;AACjE,MAAM,MAAM,wCAAwC,GAAG,WAAW,CAChE,UAAU,CAAC,OAAO,iDAAiD,CAAC,CACrE,CAAC;AAEF,8DAA8D;AAC9D,MAAM,MAAM,qCAAqC,GAAG;IAClD,KAAK,EAAE,8BAA8B,CAAC;IACtC,aAAa,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;IAClC,MAAM,CAAC,EAAE,SAAS,sBAAsB,EAAE,CAAC;IAC3C,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B,CAAC;AA+DF,mDAAmD;AACnD,wBAAgB,mCAAmC,CACjD,KAAK,EAAE,wCAAwC,GAC9C,8BAA8B,CA6BhC;AAkBD,4CAA4C;AAC5C,wBAAgB,gCAAgC,CAC9C,KAAK,EAAE,qCAAqC,GAC3C,iBAAiB,EAAE,CA2CrB"}
+{"version":3,"file":"agent-definition.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/runtime/agent-definition.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAG7D,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AAGlE,wDAAwD;AACxD,eAAO,MAAM,mCAAmC;;;GAK/C,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,gCAAgC;;;GAAkD,CAAC;AAEhG,oDAAoD;AACpD,MAAM,MAAM,0BAA0B,GAAG,WAAW,CAClD,UAAU,CAAC,OAAO,mCAAmC,CAAC,CACvD,CAAC;AAEF,4DAA4D;AAC5D,eAAO,MAAM,uCAAuC;;;;;;;;;;;;;;;;;GAgBnD,CAAC;AAEF,sDAAsD;AACtD,eAAO,MAAM,oCAAoC,uCAAuC,CAAC;AAEzF;;GAEG;AACH,eAAO,MAAM,oCAAoC;;;;;;;;;;;;;;;;;GAEhD,CAAC;AAEF,6CAA6C;AAC7C,MAAM,MAAM,8BAA8B,GAAG,WAAW,CACtD,UAAU,CAAC,OAAO,uCAAuC,CAAC,CAC3D,CAAC;AAEF,wEAAwE;AACxE,eAAO,MAAM,iDAAiD;;;GAK7D,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,8CAA8C;;;GAE1D,CAAC;AAEF,iEAAiE;AACjE,MAAM,MAAM,wCAAwC,GAAG,WAAW,CAChE,UAAU,CAAC,OAAO,iDAAiD,CAAC,CACrE,CAAC;AAEF,8DAA8D;AAC9D,MAAM,MAAM,qCAAqC,GAAG;IAClD,KAAK,EAAE,8BAA8B,CAAC;IACtC,aAAa,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;IAClC,MAAM,CAAC,EAAE,SAAS,sBAAsB,EAAE,CAAC;IAC3C,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC/B,CAAC;AA+DF,mDAAmD;AACnD,wBAAgB,mCAAmC,CACjD,KAAK,EAAE,wCAAwC,GAC9C,8BAA8B,CAqChC;AAkBD,4CAA4C;AAC5C,wBAAgB,gCAAgC,CAC9C,KAAK,EAAE,qCAAqC,GAC3C,iBAAiB,EAAE,CA2CrB"}

package/esm/src/agent/runtime/agent-definition.js

@@ -17,6 +17,7 @@ export const getRuntimeAgentMarkdownDefinitionSchema = defineSchema((v) => v.obj
     id: v.string().min(1),
     name: v.string().min(1),
     description: v.string(),
+    avatarUrl: v.string().url().optional(),
     instructions: v.string(),
     thinking: getRuntimeAgentThinkingConfigSchema().optional(),
     model: v.string().min(1).optional(),
@@ -101,6 +102,13 @@ export function parseRuntimeAgentMarkdownDefinition(input) {
     const { attrs, body } = extract(parsedInput.content);
     const name = typeof attrs.name === "string" && attrs.name.trim() ? attrs.name : parsedInput.id;
     const description = typeof attrs.description === "string" ? attrs.description : "";
+    const avatarUrl = typeof attrs["avatar-url"] === "string" && attrs["avatar-url"].trim()
+        ? attrs["avatar-url"]
+        : typeof attrs.avatarUrl === "string" && attrs.avatarUrl.trim()
+            ? attrs.avatarUrl
+            : typeof attrs.avatar_url === "string" && attrs.avatar_url.trim()
+                ? attrs.avatar_url
+                : undefined;
     const model = typeof attrs.model === "string" && attrs.model.trim() ? attrs.model : undefined;
     const thinking = parseThinking(attrs.thinking);
     const temperature = typeof attrs.temperature === "number" ? attrs.temperature : undefined;
@@ -114,6 +122,7 @@ export function parseRuntimeAgentMarkdownDefinition(input) {
         id: parsedInput.id,
         name,
         description,
+        ...(avatarUrl ? { avatarUrl } : {}),
         instructions: body.trim(),
         ...(model ? { model } : {}),
         ...(thinking ? { thinking } : {}),

package/esm/src/agent/runtime/agent-markdown-adapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-markdown-adapter.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/runtime/agent-markdown-adapter.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,uBAAuB,CAAC;AAK5E,yDAAyD;AACzD,wBAAgB,wCAAwC,CACtD,UAAU,EAAE,8BAA8B,GACzC,KAAK,CAuCP;AAED,iDAAiD;AACjD,wBAAgB,iCAAiC,CAC/C,YAAY,EAAE,KAAK,GAClB,8BAA8B,GAAG,IAAI,CAEvC;AAED,iEAAiE;AACjE,wBAAgB,2BAA2B,CAAC,YAAY,EAAE,KAAK,GAAG,OAAO,CAExE"}
+{"version":3,"file":"agent-markdown-adapter.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/runtime/agent-markdown-adapter.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,uBAAuB,CAAC;AAK5E,yDAAyD;AACzD,wBAAgB,wCAAwC,CACtD,UAAU,EAAE,8BAA8B,GACzC,KAAK,CAwCP;AAED,iDAAiD;AACjD,wBAAgB,iCAAiC,CAC/C,YAAY,EAAE,KAAK,GAClB,8BAA8B,GAAG,IAAI,CAEvC;AAED,iEAAiE;AACjE,wBAAgB,2BAA2B,CAAC,YAAY,EAAE,KAAK,GAAG,OAAO,CAExE"}

package/esm/src/agent/runtime/agent-markdown-adapter.js

@@ -25,6 +25,7 @@ export function createRuntimeAgentFromMarkdownDefinition(definition) {
         id: definition.id,
         name: definition.name,
         description: definition.description,
+        ...(definition.avatarUrl ? { avatarUrl: definition.avatarUrl } : {}),
         system: definition.instructions,
         ...(definition.model ? { model: definition.model } : {}),
         ...(definition.temperature === undefined ? {} : { temperature: definition.temperature }),

package/esm/src/agent/runtime/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/runtime/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EACL,KAAK,WAAW,EAEhB,KAAK,aAAa,EAGlB,KAAK,OAAO,EAGZ,KAAK,QAAQ,EAGd,MAAM,aAAa,CAAC;AAIrB,OAAO,EAAqB,KAAK,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAwDpE,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAC5E,OAAO,EACL,qBAAqB,EACrB,iBAAiB,EACjB,iBAAiB,EACjB,uBAAuB,EACvB,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EACV,8BAA8B,EAC9B,gBAAgB,EAChB,wBAAwB,GACzB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,qBAAqB,EACrB,iBAAiB,EACjB,aAAa,EACb,aAAa,GACd,MAAM,mBAAmB,CAAC;AAC3B,YAAY,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACzE,OAAO,EACL,sBAAsB,EACtB,KAAK,yBAAyB,EAC9B,KAAK,0BAA0B,EAC/B,KAAK,mBAAmB,EACxB,0BAA0B,EAC1B,iCAAiC,EACjC,6BAA6B,GAC9B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAChF,OAAO,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAC5E,YAAY,EACV,mBAAmB,EACnB,eAAe,EACf,iBAAiB,GAClB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,mBAAmB,EACnB,sBAAsB,GACvB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACL,4BAA4B,EAC5B,6BAA6B,EAC7B,2BAA2B,EAC3B,2BAA2B,EAC3B,gCAAgC,EAChC,4BAA4B,EAC5B,2BAA2B,EAC3B,6BAA6B,EAC7B,KAAK,+BAA+B,GACrC,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,kBAAkB,EAClB,kBAAkB,EAClB,KAAK,iBAAiB,GACvB,MAAM,+BAA+B,CAAC;AA8EvC,+BAA+B;AAC/B,qBAAa,YAAY;IACvB,OAAO,CAAC,EAAE,CAAS;IACnB,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,MAAM,CAAuB;gBAEzB,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW;IAW3C;;;;;;;OAOG;YACW,mBAAmB;YAMnB,qBAAqB;YAcrB,mBAAmB;YAsBnB,gBAAgB;IAS9B;;OAEG;IACG,QAAQ,CACZ,KAAK,EAAE,MAAM,GAAG,OAAO,EAAE,EACzB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACjC,aAAa,CAAC,EAAE,MAAM,EACtB,uBAAuB,CAAC,EAAE,MAAM,GAC/B,OAAO,CAAC,aAAa,CAAC;IAoDzB;;;OAGG;IACG,MAAM,CACV,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACjC,SAAS,CAAC,EAAE;QACV,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,KAAK,IAAI,CAAC;QAC1C,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;QAClC,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,aAAa,KAAK,IAAI,CAAC;KAC9C,EACD,aAAa,CAAC,EAAE,MAAM,EACtB,uBAAuB,CAAC,EAAE,MAAM,EAChC,WAAW,CAAC,EAAE,WAAW,GACxB,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;IAmJtC;;OAEG;YACW,gBAAgB;IAoV9B;;;;OAIG;YACW,yBAAyB;IA4fvC;;OAEG;YACW,eAAe;IAgC7B;;OAEG;YACW,mBAAmB;IAOjC;;OAEG;IACH,OAAO,CAAC,eAAe;IAKvB,OAAO,CAAC,kBAAkB;IAY1B,OAAO,CAAC,sBAAsB;IAoB9B;;OAEG;IACH,SAAS,IAAI,MAAM,CAAC,OAAO,CAAC;IAI5B;;OAEG;IACG,cAAc,IAAI,OAAO,CAAC;QAC9B,aAAa,EAAE,MAAM,CAAC;QACtB,eAAe,EAAE,MAAM,CAAC;QACxB,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;IAIF;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;CAGnC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/runtime/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EACL,KAAK,WAAW,EAEhB,KAAK,aAAa,EAGlB,KAAK,OAAO,EAGZ,KAAK,QAAQ,EAGd,MAAM,aAAa,CAAC;AAIrB,OAAO,EAAqB,KAAK,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAwDpE,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAC5E,OAAO,EACL,qBAAqB,EACrB,iBAAiB,EACjB,iBAAiB,EACjB,uBAAuB,EACvB,iBAAiB,EACjB,mBAAmB,GACpB,MAAM,qBAAqB,CAAC;AAC7B,YAAY,EACV,8BAA8B,EAC9B,gBAAgB,EAChB,wBAAwB,GACzB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,qBAAqB,EACrB,iBAAiB,EACjB,aAAa,EACb,aAAa,GACd,MAAM,mBAAmB,CAAC;AAC3B,YAAY,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACzE,OAAO,EACL,sBAAsB,EACtB,KAAK,yBAAyB,EAC9B,KAAK,0BAA0B,EAC/B,KAAK,mBAAmB,EACxB,0BAA0B,EAC1B,iCAAiC,EACjC,6BAA6B,GAC9B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAChF,OAAO,EAAE,iBAAiB,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAC5E,YAAY,EACV,mBAAmB,EACnB,eAAe,EACf,iBAAiB,GAClB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,mBAAmB,EACnB,sBAAsB,GACvB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACL,4BAA4B,EAC5B,6BAA6B,EAC7B,2BAA2B,EAC3B,2BAA2B,EAC3B,gCAAgC,EAChC,4BAA4B,EAC5B,2BAA2B,EAC3B,6BAA6B,EAC7B,KAAK,+BAA+B,GACrC,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACL,kBAAkB,EAClB,kBAAkB,EAClB,KAAK,iBAAiB,GACvB,MAAM,+BAA+B,CAAC;AA2IvC,+BAA+B;AAC/B,qBAAa,YAAY;IACvB,OAAO,CAAC,EAAE,CAAS;IACnB,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,MAAM,CAAuB;gBAEzB,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW;IAW3C;;;;;;;OAOG;YACW,mBAAmB;YAMnB,qBAAqB;YAcrB,mBAAmB;YAsBnB,gBAAgB;IAS9B;;OAEG;IACG,QAAQ,CACZ,KAAK,EAAE,MAAM,GAAG,OAAO,EAAE,EACzB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACjC,aAAa,CAAC,EAAE,MAAM,EACtB,uBAAuB,CAAC,EAAE,MAAM,GAC/B,OAAO,CAAC,aAAa,CAAC;IAoDzB;;;OAGG;IACG,MAAM,CACV,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACjC,SAAS,CAAC,EAAE;QACV,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,KAAK,IAAI,CAAC;QAC1C,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;QAClC,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,aAAa,KAAK,IAAI,CAAC;KAC9C,EACD,aAAa,CAAC,EAAE,MAAM,EACtB,uBAAuB,CAAC,EAAE,MAAM,EAChC,WAAW,CAAC,EAAE,WAAW,GACxB,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;IAyJtC;;OAEG;YACW,gBAAgB;IAoV9B;;;;OAIG;YACW,yBAAyB;IA4fvC;;OAEG;YACW,eAAe;IAgC7B;;OAEG;YACW,mBAAmB;IAOjC;;OAEG;IACH,OAAO,CAAC,eAAe;IAKvB,OAAO,CAAC,kBAAkB;IAY1B,OAAO,CAAC,sBAAsB;IAoB9B;;OAEG;IACH,SAAS,IAAI,MAAM,CAAC,OAAO,CAAC;IAI5B;;OAEG;IACG,cAAc,IAAI,OAAO,CAAC;QAC9B,aAAa,EAAE,MAAM,CAAC;QACtB,eAAe,EAAE,MAAM,CAAC;QACxB,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;IAIF;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;CAGnC"}

package/esm/src/agent/runtime/index.js

@@ -51,6 +51,62 @@ import { resolveTemperatureParameter } from "./model-capabilities.js";
 import { applySkillDelegationOverridesToToolInput, extractSkillDelegationOverrides, } from "./skill-delegation-overrides.js";
 import { resolveAgentModelTransport } from "./model-transport.js";
 const logger = serverLogger.component("agent");
+function buildStreamFinishUsage(usage) {
+    if (!usage) {
+        return undefined;
+    }
+    return {
+        inputTokens: usage.promptTokens,
+        outputTokens: usage.completionTokens,
+        totalTokens: usage.totalTokens,
+        ...(usage.cachedInputTokens !== undefined
+            ? { cachedInputTokens: usage.cachedInputTokens }
+            : {}),
+        ...(usage.cacheCreationInputTokens !== undefined
+            ? { cacheCreationInputTokens: usage.cacheCreationInputTokens }
+            : {}),
+        ...(usage.cacheReadInputTokens !== undefined
+            ? { cacheReadInputTokens: usage.cacheReadInputTokens }
+            : {}),
+        ...(usage.reasoningTokens !== undefined ? { reasoningTokens: usage.reasoningTokens } : {}),
+        ...(usage.billableInputTokens !== undefined
+            ? { billableInputTokens: usage.billableInputTokens }
+            : {}),
+        ...(usage.billableOutputTokens !== undefined
+            ? { billableOutputTokens: usage.billableOutputTokens }
+            : {}),
+        ...(usage.costUsd !== undefined ? { costUsd: usage.costUsd } : {}),
+        ...(usage.providerInputCostUsd !== undefined
+            ? { providerInputCostUsd: usage.providerInputCostUsd }
+            : {}),
+        ...(usage.providerOutputCostUsd !== undefined
+            ? { providerOutputCostUsd: usage.providerOutputCostUsd }
+            : {}),
+        ...(usage.providerCostUsd !== undefined ? { providerCostUsd: usage.providerCostUsd } : {}),
+        ...(usage.veryfrontInputChargeUsd !== undefined
+            ? { veryfrontInputChargeUsd: usage.veryfrontInputChargeUsd }
+            : {}),
+        ...(usage.veryfrontOutputChargeUsd !== undefined
+            ? { veryfrontOutputChargeUsd: usage.veryfrontOutputChargeUsd }
+            : {}),
+        ...(usage.veryfrontChargeUsd !== undefined
+            ? { veryfrontChargeUsd: usage.veryfrontChargeUsd }
+            : {}),
+        ...(usage.veryfrontBilledUsd !== undefined
+            ? { veryfrontBilledUsd: usage.veryfrontBilledUsd }
+            : {}),
+        ...(usage.costCredits !== undefined ? { costCredits: usage.costCredits } : {}),
+        ...(usage.costSource !== undefined ? { costSource: usage.costSource } : {}),
+        ...(usage.billingMode !== undefined ? { billingMode: usage.billingMode } : {}),
+        ...(usage.usageCaptureStatus !== undefined
+            ? { usageCaptureStatus: usage.usageCaptureStatus }
+            : {}),
+    };
+}
+function getResponseFinishReason(response) {
+    const finishReason = response.metadata?.finishReason;
+    return typeof finishReason === "string" && finishReason.length > 0 ? finishReason : undefined;
+}
 function shouldHideProjectToolAfterAgentWriteSuccess(toolName) {
     return toolName === "create_agent" || toolName === "update_agent";
 }
@@ -266,7 +322,13 @@ export class AgentRuntime {
                     throwIfAborted(streamAbortSignal);
                     callbacks?.onFinish?.(response);
                     throwIfAborted(streamAbortSignal);
-                    sendSSE(controller, encoder, { type: "message-finish" });
+                    const finishUsage = buildStreamFinishUsage(response.usage);
+                    const finishReason = getResponseFinishReason(response);
+                    sendSSE(controller, encoder, {
+                        type: "message-finish",
+                        ...(finishReason ? { finishReason } : {}),
+                        ...(finishUsage ? { totalUsage: finishUsage } : {}),
+                    });
                     closeSSEStream(controller);
                 }
                 catch (error) {

package/esm/src/agent/streaming/chat-ui-message-stream.d.ts

@@ -18,6 +18,20 @@ export type ChatUiMessageStreamFinishPart = {
             textTokens?: number;
             reasoningTokens?: number;
         };
+        billableInputTokens?: number;
+        billableOutputTokens?: number;
+        costUsd?: number;
+        providerInputCostUsd?: number;
+        providerOutputCostUsd?: number;
+        providerCostUsd?: number;
+        veryfrontInputChargeUsd?: number;
+        veryfrontOutputChargeUsd?: number;
+        veryfrontChargeUsd?: number;
+        veryfrontBilledUsd?: number;
+        costCredits?: number;
+        costSource?: "gateway" | "missing" | "partial";
+        billingMode?: "direct" | "deferred";
+        usageCaptureStatus?: "complete" | "partial" | "missing";
     };
 };
 /** Public API contract for chat UI message stream finish. */

package/esm/src/agent/streaming/chat-ui-message-stream.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chat-ui-message-stream.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/streaming/chat-ui-message-stream.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAmB,MAAM,wBAAwB,CAAC;AAChF,OAAO,KAAK,EAEV,aAAa,EACb,kBAAkB,EAClB,eAAe,EAChB,MAAM,qBAAqB,CAAC;AAa7B,kEAAkE;AAClE,MAAM,MAAM,6BAA6B,GAAG;IAC1C,IAAI,EAAE,QAAQ,CAAC;IACf,YAAY,EAAE,gBAAgB,CAAC;IAC/B,eAAe,EAAE,gBAAgB,CAAC;IAClC,UAAU,EAAE;QACV,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;QACpB,iBAAiB,EAAE;YACjB,aAAa,CAAC,EAAE,MAAM,CAAC;YACvB,eAAe,CAAC,EAAE,MAAM,CAAC;YACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;SAC3B,CAAC;QACF,kBAAkB,EAAE;YAClB,UAAU,CAAC,EAAE,MAAM,CAAC;YACpB,eAAe,CAAC,EAAE,MAAM,CAAC;SAC1B,CAAC;KACH,CAAC;CACH,CAAC;AAEF,6DAA6D;AAC7D,MAAM,MAAM,yBAAyB,CAAC,gBAAgB,GAAG,eAAe,IAAI;IAC1E,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC,CAAC;IACjD,cAAc,EAAE,KAAK,CAAC;IACtB,eAAe,EAAE,aAAa,CAAC,gBAAgB,CAAC,CAAC;IACjD,SAAS,EAAE,KAAK,CAAC;IACjB,YAAY,EAAE,gBAAgB,CAAC;CAChC,CAAC;AAEF,kDAAkD;AAClD,MAAM,MAAM,0BAA0B,CAAC,gBAAgB,GAAG,eAAe,IAAI;IAC3E,iBAAiB,CAAC,EAAE,MAAM,MAAM,CAAC;IACjC,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,MAAM,CAAC;IACrC,eAAe,CAAC,EAAE,CAChB,KAAK,EAAE;QAAE,IAAI,EAAE,6BAA6B,CAAA;KAAE,KAC3C,gBAAgB,GAAG,SAAS,CAAC;IAClC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,yBAAyB,CAAC,gBAAgB,CAAC,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzF,mBAAmB,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,KAAK,IAAI,CAAC;CAClF,CAAC;AAsbF,sDAAsD;AACtD,wBAAgB,uCAAuC,CAAC,gBAAgB,GAAG,eAAe,EACxF,KAAK,EAAE;IAAE,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAA;CAAE,EAC7C,OAAO,GAAE,0BAA0B,CAAC,gBAAgB,CAAM,GACzD,aAAa,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,CAiGpD"}
+{"version":3,"file":"chat-ui-message-stream.d.ts","sourceRoot":"","sources":["../../../../src/src/agent/streaming/chat-ui-message-stream.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAmB,MAAM,wBAAwB,CAAC;AAChF,OAAO,KAAK,EAEV,aAAa,EACb,kBAAkB,EAClB,eAAe,EAChB,MAAM,qBAAqB,CAAC;AAgB7B,kEAAkE;AAClE,MAAM,MAAM,6BAA6B,GAAG;IAC1C,IAAI,EAAE,QAAQ,CAAC;IACf,YAAY,EAAE,gBAAgB,CAAC;IAC/B,eAAe,EAAE,gBAAgB,CAAC;IAClC,UAAU,EAAE;QACV,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,WAAW,EAAE,MAAM,CAAC;QACpB,iBAAiB,EAAE;YACjB,aAAa,CAAC,EAAE,MAAM,CAAC;YACvB,eAAe,CAAC,EAAE,MAAM,CAAC;YACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;SAC3B,CAAC;QACF,kBAAkB,EAAE;YAClB,UAAU,CAAC,EAAE,MAAM,CAAC;YACpB,eAAe,CAAC,EAAE,MAAM,CAAC;SAC1B,CAAC;QACF,mBAAmB,CAAC,EAAE,MAAM,CAAC;QAC7B,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,oBAAoB,CAAC,EAAE,MAAM,CAAC;QAC9B,qBAAqB,CAAC,EAAE,MAAM,CAAC;QAC/B,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,uBAAuB,CAAC,EAAE,MAAM,CAAC;QACjC,wBAAwB,CAAC,EAAE,MAAM,CAAC;QAClC,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,UAAU,CAAC,EAAE,SAAS,GAAG,SAAS,GAAG,SAAS,CAAC;QAC/C,WAAW,CAAC,EAAE,QAAQ,GAAG,UAAU,CAAC;QACpC,kBAAkB,CAAC,EAAE,UAAU,GAAG,SAAS,GAAG,SAAS,CAAC;KACzD,CAAC;CACH,CAAC;AAEF,6DAA6D;AAC7D,MAAM,MAAM,yBAAyB,CAAC,gBAAgB,GAAG,eAAe,IAAI;IAC1E,QAAQ,EAAE,KAAK,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC,CAAC;IACjD,cAAc,EAAE,KAAK,CAAC;IACtB,eAAe,EAAE,aAAa,CAAC,gBAAgB,CAAC,CAAC;IACjD,SAAS,EAAE,KAAK,CAAC;IACjB,YAAY,EAAE,gBAAgB,CAAC;CAChC,CAAC;AAEF,kDAAkD;AAClD,MAAM,MAAM,0BAA0B,CAAC,gBAAgB,GAAG,eAAe,IAAI;IAC3E,iBAAiB,CAAC,EAAE,MAAM,MAAM,CAAC;IACjC,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,MAAM,CAAC;IACrC,eAAe,CAAC,EAAE,CAChB,KAAK,EAAE;QAAE,IAAI,EAAE,6BAA6B,CAAA;KAAE,KAC3C,gBAAgB,GAAG,SAAS,CAAC;IAClC,QAAQ,CAAC,EAAE,CAAC,MAAM,EAAE,yBAAyB,CAAC,gBAAgB,CAAC,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACzF,mBAAmB,CAAC,EAAE,CAAC,KAAK,EAAE;QAAE,UAAU,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,KAAK,IAAI,CAAC;CAClF,CAAC;AA4bF,sDAAsD;AACtD,wBAAgB,uCAAuC,CAAC,gBAAgB,GAAG,eAAe,EACxF,KAAK,EAAE;IAAE,MAAM,EAAE,cAAc,CAAC,UAAU,CAAC,CAAA;CAAE,EAC7C,OAAO,GAAE,0BAA0B,CAAC,gBAAgB,CAAM,GACzD,aAAa,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,CAoGpD"}