UNPKG

@tanstack/ai

Version:

Core TanStack AI library - Open source AI SDK

45 lines (44 loc) 2.12 kB
import { UIMessage } from '../../../types.js'; import { ToolCallState, ToolResultState } from './types.js'; /** * Update or add a text part to a message. * * If the last part is a text part, update it (continuing the same text segment). * Otherwise, create a new text part (starting a new text segment after tool calls). */ export declare function updateTextPart(messages: Array<UIMessage>, messageId: string, content: string): Array<UIMessage>; /** * Update or add a tool call part to a message. */ export declare function updateToolCallPart(messages: Array<UIMessage>, messageId: string, toolCall: { id: string; name: string; arguments: string; state: ToolCallState; }): Array<UIMessage>; /** * Update or add a tool result part to a message. */ export declare function updateToolResultPart(messages: Array<UIMessage>, messageId: string, toolCallId: string, content: string, state: ToolResultState, error?: string): Array<UIMessage>; /** * Update a tool call part with approval request metadata. */ export declare function updateToolCallApproval(messages: Array<UIMessage>, messageId: string, toolCallId: string, approvalId: string): Array<UIMessage>; /** * Update a tool call part's state (e.g., to "input-complete"). */ export declare function updateToolCallState(messages: Array<UIMessage>, messageId: string, toolCallId: string, state: ToolCallState): Array<UIMessage>; /** * Update a tool call part with output. * Searches all messages to find the tool call by ID. */ export declare function updateToolCallWithOutput(messages: Array<UIMessage>, toolCallId: string, output: any, state?: ToolCallState, errorText?: string): Array<UIMessage>; /** * Update a tool call part with approval response. * Searches all messages to find the tool call by approval ID. */ export declare function updateToolCallApprovalResponse(messages: Array<UIMessage>, approvalId: string, approved: boolean): Array<UIMessage>; /** * Update or add a thinking part to a message. */ export declare function updateThinkingPart(messages: Array<UIMessage>, messageId: string, content: string): Array<UIMessage>;