ai 7.0.141 → 7.0.143

package/src/ui/process-ui-message-stream.ts

@@ -39,6 +39,7 @@ export type StreamingUIMessageState<UI_MESSAGE extends UIMessage> = {
     string,
     {
       text: string;
+      inputEncoding?: 'json' | 'text';
       index: number;
       toolName: string;
       dynamic?: boolean;
@@ -95,6 +96,7 @@ export function createStreamingUIMessageState<UI_MESSAGE extends UIMessage>({
 
     partialToolCalls[part.toolCallId] = {
       text: part.rawInput ?? '',
+      inputEncoding: part.inputEncoding,
       index,
       toolName:
         part.type === 'dynamic-tool' ? part.toolName : getStaticToolName(part),
@@ -227,6 +229,7 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
                   state: 'input-streaming';
                   input: unknown;
                   rawInput?: string;
+                  inputEncoding?: 'json' | 'text';
                   providerExecuted?: boolean;
                   providerMetadata?: ProviderMetadata;
                 }
@@ -272,6 +275,11 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
               anyPart.output = anyOptions.output;
               anyPart.errorText = anyOptions.errorText;
               anyPart.rawInput = anyOptions.rawInput;
+              if (anyOptions.inputEncoding != null) {
+                anyPart.inputEncoding = anyOptions.inputEncoding;
+              } else {
+                delete anyPart.inputEncoding;
+              }
               anyPart.preliminary = anyOptions.preliminary;
               if (options.title !== undefined) {
                 anyPart.title = options.title;
@@ -310,6 +318,9 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
                   ? { toolMetadata: options.toolMetadata }
                   : {}),
                 input: anyOptions.input,
+                ...(anyOptions.inputEncoding != null
+                  ? { inputEncoding: anyOptions.inputEncoding }
+                  : {}),
                 output: anyOptions.output,
                 rawInput: anyOptions.rawInput,
                 errorText: anyOptions.errorText,
@@ -343,6 +354,7 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
                   state: 'input-streaming';
                   input: unknown;
                   rawInput?: string;
+                  inputEncoding?: 'json' | 'text';
                   providerMetadata?: ProviderMetadata;
                 }
               | {
@@ -384,6 +396,11 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
               anyPart.output = anyOptions.output;
               anyPart.errorText = anyOptions.errorText;
               anyPart.rawInput = anyOptions.rawInput;
+              if (anyOptions.inputEncoding != null) {
+                anyPart.inputEncoding = anyOptions.inputEncoding;
+              } else {
+                delete anyPart.inputEncoding;
+              }
               anyPart.preliminary = anyOptions.preliminary;
               if (options.title !== undefined) {
                 anyPart.title = options.title;
@@ -419,6 +436,9 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
                 toolCallId: options.toolCallId,
                 state: options.state,
                 input: anyOptions.input,
+                ...(anyOptions.inputEncoding != null
+                  ? { inputEncoding: anyOptions.inputEncoding }
+                  : {}),
                 output: anyOptions.output,
                 errorText: anyOptions.errorText,
                 preliminary: anyOptions.preliminary,
@@ -632,6 +652,7 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
               // add the partial tool call to the map
               state.partialToolCalls[chunk.toolCallId] = {
                 text: '',
+                inputEncoding: chunk.inputEncoding,
                 toolName: chunk.toolName,
                 index: toolInvocations.length,
                 dynamic: chunk.dynamic,
@@ -644,6 +665,7 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
                   toolCallId: chunk.toolCallId,
                   toolName: chunk.toolName,
                   state: 'input-streaming',
+                  inputEncoding: chunk.inputEncoding,
                   input: undefined,
                   providerExecuted: chunk.providerExecuted,
                   title: chunk.title,
@@ -655,6 +677,7 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
                   toolCallId: chunk.toolCallId,
                   toolName: chunk.toolName,
                   state: 'input-streaming',
+                  inputEncoding: chunk.inputEncoding,
                   input: undefined,
                   providerExecuted: chunk.providerExecuted,
                   title: chunk.title,
@@ -681,15 +704,17 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
 
               partialToolCall.text += chunk.inputTextDelta;
 
-              const { value: partialArgs } = await parsePartialJson(
-                partialToolCall.text,
-              );
+              const partialArgs =
+                partialToolCall.inputEncoding === 'text'
+                  ? partialToolCall.text
+                  : (await parsePartialJson(partialToolCall.text)).value;
 
               if (partialToolCall.dynamic) {
                 updateDynamicToolPart({
                   toolCallId: chunk.toolCallId,
                   toolName: partialToolCall.toolName,
                   state: 'input-streaming',
+                  inputEncoding: partialToolCall.inputEncoding,
                   input: partialArgs,
                   rawInput: partialToolCall.text,
                   title: partialToolCall.title,
@@ -700,6 +725,7 @@ export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
                   toolCallId: chunk.toolCallId,
                   toolName: partialToolCall.toolName,
                   state: 'input-streaming',
+                  inputEncoding: partialToolCall.inputEncoding,
                   input: partialArgs,
                   rawInput: partialToolCall.text,
                   title: partialToolCall.title,

package/src/ui/ui-messages.ts

@@ -308,6 +308,8 @@ export type UIToolInvocation<TOOL extends UITool | Tool> = {
        * and later resumed.
        */
       rawInput?: string;
+      /** Encoding of `rawInput`. Defaults to JSON when omitted. */
+      inputEncoding?: 'json' | 'text';
       output?: never;
       errorText?: never;
       callProviderMetadata?: ProviderMetadata;
@@ -460,6 +462,8 @@ export type DynamicToolUIPart = {
        * and later resumed.
        */
       rawInput?: string;
+      /** Encoding of `rawInput`. Defaults to JSON when omitted. */
+      inputEncoding?: 'json' | 'text';
       output?: never;
       errorText?: never;
       callProviderMetadata?: ProviderMetadata;

package/src/ui/validate-ui-messages.ts

@@ -159,6 +159,7 @@ const uiMessagesSchema = lazySchema(() => {
                   state: z.literal('input-streaming'),
                   input: z.unknown().optional(),
                   rawInput: z.string().optional(),
+                  inputEncoding: z.enum(['json', 'text']).optional(),
                   providerExecuted: z.boolean().optional(),
                   callProviderMetadata: providerMetadataSchema.optional(),
                   output: z.never().optional(),
@@ -269,6 +270,7 @@ const uiMessagesSchema = lazySchema(() => {
                   callProviderMetadata: providerMetadataSchema.optional(),
                   input: z.unknown().optional(),
                   rawInput: z.string().optional(),
+                  inputEncoding: z.enum(['json', 'text']).optional(),
                   output: z.never().optional(),
                   errorText: z.never().optional(),
                   approval: z.never().optional(),

package/src/ui-message-stream/to-ui-message-chunk.ts

@@ -192,6 +192,9 @@ export function toUIMessageChunk<
         type: 'tool-input-start',
         toolCallId: part.id,
         toolName: part.toolName,
+        ...(part.inputEncoding != null
+          ? { inputEncoding: part.inputEncoding }
+          : {}),
         ...(part.providerExecuted != null
           ? { providerExecuted: part.providerExecuted }
           : {}),

package/src/ui-message-stream/ui-message-chunks.ts

@@ -49,6 +49,7 @@ export const uiMessageChunkSchema = lazySchema(() =>
         type: z.literal('tool-input-start'),
         toolCallId: z.string(),
         toolName: z.string(),
+        inputEncoding: z.enum(['json', 'text']).optional(),
         providerExecuted: z.boolean().optional(),
         providerMetadata: providerMetadataSchema.optional(),
         toolMetadata: toolMetadataSchema.optional(),
@@ -345,6 +346,8 @@ export type UIMessageChunk<
       type: 'tool-input-start';
       toolCallId: string;
       toolName: string;
+      /** Encoding of the input deltas. Defaults to JSON when omitted. */
+      inputEncoding?: 'json' | 'text';
       providerExecuted?: boolean;
       providerMetadata?: ProviderMetadata;
       toolMetadata?: JSONObject;