veryfront 0.1.149 → 0.1.151

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

@@ -1,4 +1,4 @@
-import type { ToolUIPart, UIMessagePart } from "../types.js";
+import type { ChatMessagePart, ChatToolPart } from "../types.js";
 import { createAssistantMessage, generateClientId } from "../utils.js";
 import { buildCurrentParts } from "./parts-builder.js";
 import type {
@@ -14,7 +14,7 @@ interface StreamingState {
   toolCalls: Map<string, OrderedToolCall>;
   reasoningBlocks: Map<string, OrderedReasoning>;
   steps: Map<number, OrderedStep>;
-  messageParts: UIMessagePart[];
+  messageParts: ChatMessagePart[];
   currentTextId: string;
   messageId: string;
   partOrderCounter: number;
@@ -43,7 +43,7 @@ export async function handleStreamingResponse(
   const decoder = new TextDecoder();
   const state = createStreamingState();
 
-  const getBuildParts = (): UIMessagePart[] =>
+  const getBuildParts = (): ChatMessagePart[] =>
     buildCurrentParts(state.textBlocks, state.reasoningBlocks, state.toolCalls, state.steps);
 
   let buffer = "";
@@ -89,7 +89,7 @@ function processEvent(
   parsed: Record<string, unknown>,
   state: StreamingState,
   callbacks: StreamingCallbacks,
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const { onMessage, onData, onUpdate, onToolCall } = callbacks;
 
@@ -181,7 +181,7 @@ function handleTextDelta(
   parsed: Record<string, unknown>,
   state: StreamingState,
   onUpdate: StreamingCallbacks["onUpdate"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const textId = (parsed.id as string) || state.currentTextId || "default";
   const delta = (parsed.textDelta ?? parsed.delta ?? "") as string;
@@ -217,7 +217,7 @@ function handleToolInputStart(
   parsed: Record<string, unknown>,
   state: StreamingState,
   onUpdate: StreamingCallbacks["onUpdate"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const toolCallId = (parsed.toolCallId as string) || generateClientId("tool");
   const toolCall: OrderedToolCall = {
@@ -237,7 +237,7 @@ function handleToolInputDelta(
   parsed: Record<string, unknown>,
   state: StreamingState,
   onUpdate: StreamingCallbacks["onUpdate"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const toolCallId = parsed.toolCallId as string;
   const toolCall = state.toolCalls.get(toolCallId);
@@ -252,7 +252,7 @@ function handleToolInputAvailable(
   state: StreamingState,
   onUpdate: StreamingCallbacks["onUpdate"],
   onToolCall: StreamingCallbacks["onToolCall"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const toolCallId = parsed.toolCallId as string;
   const toolCall = state.toolCalls.get(toolCallId);
@@ -287,7 +287,7 @@ function handleToolInputAvailable(
       toolName: toolCall.toolName,
       state: "input-available",
       input: toolCall.input,
-    } as ToolUIPart);
+    } as ChatToolPart);
   }
 
   onUpdate?.(getBuildParts(), state.messageId);
@@ -297,7 +297,7 @@ function handleToolOutputAvailable(
   parsed: Record<string, unknown>,
   state: StreamingState,
   onUpdate: StreamingCallbacks["onUpdate"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const toolCallId = parsed.toolCallId as string;
   const toolCall = state.toolCalls.get(toolCallId);
@@ -320,7 +320,7 @@ function handleToolError(
   parsed: Record<string, unknown>,
   state: StreamingState,
   onUpdate: StreamingCallbacks["onUpdate"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const toolCallId = parsed.toolCallId as string;
   const toolCall = state.toolCalls.get(toolCallId);
@@ -337,7 +337,7 @@ function handleReasoningStart(
   parsed: Record<string, unknown>,
   state: StreamingState,
   onUpdate: StreamingCallbacks["onUpdate"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const reasoningId = (parsed.id as string) || generateClientId("reasoning");
   const reasoning: OrderedReasoning = {
@@ -355,7 +355,7 @@ function handleReasoningDelta(
   parsed: Record<string, unknown>,
   state: StreamingState,
   onUpdate: StreamingCallbacks["onUpdate"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const reasoningId = parsed.id as string;
   const reasoning = state.reasoningBlocks.get(reasoningId);
@@ -369,7 +369,7 @@ function handleReasoningEnd(
   parsed: Record<string, unknown>,
   state: StreamingState,
   onUpdate: StreamingCallbacks["onUpdate"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const reasoningId = parsed.id as string;
   const reasoning = state.reasoningBlocks.get(reasoningId);
@@ -389,7 +389,7 @@ function handleStepStart(
   _parsed: Record<string, unknown>,
   state: StreamingState,
   onUpdate: StreamingCallbacks["onUpdate"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const stepIndex = state.currentStep;
   const step: OrderedStep = {
@@ -405,7 +405,7 @@ function handleStepEnd(
   _parsed: Record<string, unknown>,
   state: StreamingState,
   onUpdate: StreamingCallbacks["onUpdate"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const step = state.steps.get(state.currentStep);
   if (step) {
@@ -420,7 +420,7 @@ function handleStepEnd(
 function handleFinish(
   state: StreamingState,
   onMessage: StreamingCallbacks["onMessage"],
-  getBuildParts: () => UIMessagePart[],
+  getBuildParts: () => ChatMessagePart[],
 ): void {
   const finalParts = getBuildParts();
   if (finalParts.length > 0) {

package/src/src/agent/react/use-chat/streaming/parts-builder.ts

@@ -1,9 +1,9 @@
-import type { ToolUIPart, UIMessagePart } from "../types.js";
+import type { ChatMessagePart, ChatToolPart } from "../types.js";
 import type { OrderedReasoning, OrderedStep, OrderedToolCall, TextBlock } from "./types.js";
 
 interface OrderedPart {
   order: number;
-  part: UIMessagePart;
+  part: ChatMessagePart;
 }
 
 export function buildCurrentParts(
@@ -11,7 +11,7 @@ export function buildCurrentParts(
   reasoningBlocks: Map<string, OrderedReasoning>,
   toolCalls: Map<string, OrderedToolCall>,
   steps?: Map<number, OrderedStep>,
-): UIMessagePart[] {
+): ChatMessagePart[] {
   const orderedParts: OrderedPart[] = [];
 
   addTextParts(orderedParts, textBlocks);
@@ -67,9 +67,9 @@ function addToolParts(
       errorText: tool.error,
     };
 
-    const part: UIMessagePart = tool.dynamic
+    const part: ChatMessagePart = tool.dynamic
       ? { type: "dynamic-tool", ...base }
-      : ({ type: `tool-${tool.toolName}`, ...base } as ToolUIPart);
+      : ({ type: `tool-${tool.toolName}`, ...base } as ChatToolPart);
 
     orderedParts.push({ order: tool.order, part });
   }

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

@@ -1,9 +1,9 @@
-import type { OnToolCallArg, ToolState, UIMessage, UIMessagePart } from "../types.js";
+import type { ChatMessage, ChatMessagePart, ChatToolState, OnToolCallArg } from "../types.js";
 
 export interface StreamingCallbacks {
-  onMessage: (message: UIMessage) => void;
+  onMessage: (message: ChatMessage) => void;
   onData: (data: unknown) => void;
-  onUpdate?: (parts: UIMessagePart[], messageId: string) => void;
+  onUpdate?: (parts: ChatMessagePart[], messageId: string) => void;
   onToolCall?: (arg: OnToolCallArg) => void;
 }
 
@@ -14,7 +14,7 @@ export interface StreamingToolCall {
   input?: unknown;
   output?: unknown;
   error?: string;
-  state: ToolState;
+  state: ChatToolState;
   /** Whether this is a dynamic tool (MCP, user-defined, etc.) */
   dynamic?: boolean;
 }

package/src/src/agent/react/use-chat/types.ts

@@ -1,5 +1,15 @@
 import * as dntShim from "../../../../_dnt.shims.js";
-type StreamState = "streaming" | "done";
+import type {
+  ChatDynamicToolPart,
+  ChatMessage,
+  ChatMessagePart,
+  ChatReasoningPart,
+  ChatStepPart,
+  ChatTextPart,
+  ChatToolPart,
+  ChatToolResultPart,
+  ChatToolState,
+} from "../../../chat/protocol.js";
 
 /** Where inference is happening */
 export type InferenceMode = "cloud" | "server-local" | "browser";
@@ -13,73 +23,17 @@ export type BrowserInferenceStatus =
   | "generating"
   | "error";
 
-export interface TextUIPart {
-  type: "text";
-  text: string;
-  state?: StreamState;
-}
-
-export interface ReasoningUIPart {
-  type: "reasoning";
-  text: string;
-  state?: StreamState;
-}
-
-export type ToolState =
-  | "input-streaming"
-  | "input-available"
-  | "output-streaming"
-  | "output-available"
-  | "output-error";
-
-export interface ToolUIPart<NAME extends string = string, INPUT = unknown, OUTPUT = unknown> {
-  type: `tool-${NAME}`;
-  toolCallId: string;
-  toolName: NAME;
-  state: ToolState;
-  input?: INPUT;
-  output?: OUTPUT;
-  errorText?: string;
-}
-
-export interface ToolResultUIPart<RESULT = unknown> {
-  type: "tool-result";
-  toolCallId: string;
-  toolName: string;
-  result: RESULT;
-  isError?: boolean;
-}
-
-export interface DynamicToolUIPart {
-  type: "dynamic-tool";
-  toolCallId: string;
-  toolName: string;
-  state: ToolState;
-  input?: unknown;
-  output?: unknown;
-  errorText?: string;
-}
-
-export interface StepUIPart {
-  type: "step-start" | "step-end";
-  stepIndex: number;
-}
-
-export type UIMessagePart =
-  | TextUIPart
-  | ReasoningUIPart
-  | ToolUIPart
-  | ToolResultUIPart
-  | DynamicToolUIPart
-  | StepUIPart;
-
-export interface UIMessage {
-  id: string;
-  role: "system" | "user" | "assistant";
-  parts: UIMessagePart[];
-  metadata?: Record<string, unknown>;
-  createdAt?: Date | string;
-}
+export type {
+  ChatDynamicToolPart,
+  ChatMessage,
+  ChatMessagePart,
+  ChatReasoningPart,
+  ChatStepPart,
+  ChatTextPart,
+  ChatToolPart,
+  ChatToolResultPart,
+  ChatToolState,
+};
 
 type ToolOutputState = "output-available" | "output-error";
 
@@ -102,7 +56,7 @@ export interface OnToolCallArg {
 
 export interface UseChatOptions {
   api: string;
-  initialMessages?: UIMessage[];
+  initialMessages?: ChatMessage[];
   body?: Record<string, unknown>;
   headers?: Record<string, string>;
   credentials?: RequestCredentials;
@@ -110,10 +64,10 @@ export interface UseChatOptions {
   model?: string;
   /** System prompt for browser-side inference (server uses agent config) */
   systemPrompt?: string;
-  /** Enable/disable browser fallback when server can't provide AI. Default: true */
+  /** Enable/disable browser fallback when server can't provide a runtime. Default: true */
   browserFallback?: boolean;
   onResponse?: (response: dntShim.Response) => void;
-  onFinish?: (message: UIMessage) => void;
+  onFinish?: (message: ChatMessage) => void;
   onError?: (error: Error) => void;
   onToolCall?: (arg: OnToolCallArg) => void | Promise<void>;
 }
@@ -124,7 +78,7 @@ export interface BranchInfo {
 }
 
 export interface UseChatResult {
-  messages: UIMessage[];
+  messages: ChatMessage[];
   input: string;
   isLoading: boolean;
   error: Error | null;
@@ -148,7 +102,7 @@ export interface UseChatResult {
   switchBranch: (messageId: string, branchIndex: number) => void;
   reload: () => Promise<void>;
   stop: () => void;
-  setMessages: (messages: UIMessage[]) => void;
+  setMessages: (messages: ChatMessage[]) => void;
   addToolOutput: (output: ToolOutput) => void;
   data?: unknown;
   handleInputChange: (e: React.ChangeEvent<HTMLInputElement | HTMLTextAreaElement>) => void;

package/src/src/agent/react/use-chat/use-chat.ts

@@ -20,10 +20,10 @@ import { handleStreamingResponse } from "./streaming/index.js";
 import type {
   BranchInfo,
   BrowserInferenceStatus,
+  ChatMessage,
+  ChatMessagePart,
   InferenceMode,
   ToolOutput,
-  UIMessage,
-  UIMessagePart,
   UseChatOptions,
   UseChatResult,
 } from "./types.js";
@@ -31,14 +31,14 @@ import { generateClientId } from "./utils.js";
 
 /** A snapshot of messages from a branch point onward */
 interface Branch {
-  messages: UIMessage[];
+  messages: ChatMessage[];
 }
 
 /** Tracks branches keyed by the message ID where the edit occurred */
 interface BranchState {
   branches: Branch[];
   currentIndex: number;
-  baseMessages: UIMessage[];
+  baseMessages: ChatMessage[];
 }
 
 export function isLatestRequest(activeRequestId: number, requestId: number): boolean {
@@ -55,7 +55,7 @@ export function resolveBranchKey(
 }
 
 export function findBranchUserMessageIndex(
-  messages: UIMessage[],
+  messages: ChatMessage[],
   branchKey: string,
   branchKeyByMessageId: Map<string, string>,
 ): number {
@@ -68,7 +68,7 @@ export function findBranchUserMessageIndex(
  * useChat hook for managing chat state with veryfront stream events.
  */
 export function useChat(options: UseChatOptions): UseChatResult {
-  const [messages, setMessages] = useState<UIMessage[]>(options.initialMessages ?? []);
+  const [messages, setMessages] = useState<ChatMessage[]>(options.initialMessages ?? []);
   const messagesRef = useRef(messages);
   messagesRef.current = messages;
   const [input, setInput] = useState("");
@@ -90,7 +90,7 @@ export function useChat(options: UseChatOptions): UseChatResult {
 
   // System prompt for browser fallback (from 503 response or options)
   const systemPromptRef = useRef<string>(
-    options.systemPrompt ?? "You are a helpful AI assistant.",
+    options.systemPrompt ?? "You are a helpful assistant.",
   );
 
   // Track pending tool outputs for addToolOutput
@@ -136,7 +136,7 @@ export function useChat(options: UseChatOptions): UseChatResult {
    * when server-side inference works fine.
    */
   const doBrowserInference = useCallback(
-    async (allMessages: UIMessage[]) => {
+    async (allMessages: ChatMessage[]) => {
       browserInferenceActiveRef.current = true;
 
       try {
@@ -149,7 +149,7 @@ export function useChat(options: UseChatOptions): UseChatResult {
           let hasAddedMessage = false;
 
           runBrowserInference(allMessages, systemPromptRef.current, {
-            onUpdate: (parts: UIMessagePart[], messageId: string) => {
+            onUpdate: (parts: ChatMessagePart[], messageId: string) => {
               if (!hasAddedMessage) {
                 hasAddedMessage = true;
                 setMessages((prev) => [
@@ -165,7 +165,7 @@ export function useChat(options: UseChatOptions): UseChatResult {
               }
               setMessages((prev) => prev.map((m) => (m.id === messageId ? { ...m, parts } : m)));
             },
-            onMessage: (assistantMessage: UIMessage) => {
+            onMessage: (assistantMessage: ChatMessage) => {
               const withMeta = {
                 ...assistantMessage,
                 metadata: { ...assistantMessage.metadata, model: model ?? "browser" },
@@ -205,13 +205,13 @@ export function useChat(options: UseChatOptions): UseChatResult {
    * Send a message and stream assistant updates.
    */
   const sendMessage = useCallback(
-    async (message: { text: string; baseMessages?: UIMessage[]; userMessageId?: string }) => {
+    async (message: { text: string; baseMessages?: ChatMessage[]; userMessageId?: string }) => {
       // Abort any in-flight request before starting a new one
       abortControllerRef.current?.abort();
       abortControllerRef.current = null;
       const requestId = ++requestIdRef.current;
 
-      const userMessage: UIMessage = {
+      const userMessage: ChatMessage = {
         id: message.userMessageId ?? generateClientId("msg"),
         role: "user",
         parts: [{ type: "text", text: message.text }],
@@ -249,7 +249,7 @@ export function useChat(options: UseChatOptions): UseChatResult {
           signal: abortController.signal,
         });
 
-        // Handle 503 — server can't provide AI, fall back to browser
+        // Handle 503 — server can't provide a runtime, fall back to browser
         if (response.status === 503 && (options.browserFallback ?? true)) {
           try {
             const body = await response.json();

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

@@ -1,10 +1,10 @@
-import type { UIMessage, UIMessagePart } from "./types.js";
+import type { ChatMessage, ChatMessagePart } from "./types.js";
 
 export function generateClientId(prefix: string): string {
   return `${prefix}-${Date.now().toString(36)}-${Math.random().toString(36).slice(2, 8)}`;
 }
 
-export function createAssistantMessage(messageId: string, parts: UIMessagePart[]): UIMessage {
+export function createAssistantMessage(messageId: string, parts: ChatMessagePart[]): ChatMessage {
   return {
     id: messageId || generateClientId("msg"),
     role: "assistant",