@assistant-ui/react 0.7.57 → 0.7.59

package/dist/utils/json/json-value.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/utils/json/parse-partial-json.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parse-partial-json.d.ts","sourceRoot":"","sources":["../../../src/utils/json/parse-partial-json.ts"],"names":[],"mappings":"AAGA,eAAO,MAAM,gBAAgB,SAAU,MAAM,QAU5C,CAAC"}

package/dist/{runtimes/edge/partial-json → utils/json}/parse-partial-json.js

@@ -27,7 +27,7 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 ));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
-// src/runtimes/edge/partial-json/parse-partial-json.ts
+// src/utils/json/parse-partial-json.ts
 var parse_partial_json_exports = {};
 __export(parse_partial_json_exports, {
   parsePartialJson: () => parsePartialJson

package/dist/utils/json/parse-partial-json.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/utils/json/parse-partial-json.ts"],"sourcesContent":["import sjson from \"secure-json-parse\";\nimport { fixJson } from \"./fix-json\";\n\nexport const parsePartialJson = (json: string) => {\n  try {\n    return sjson.parse(json);\n  } catch {\n    try {\n      return sjson.parse(fixJson(json));\n    } catch {\n      return undefined;\n    }\n  }\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,+BAAkB;AAClB,sBAAwB;AAEjB,IAAM,mBAAmB,CAAC,SAAiB;AAChD,MAAI;AACF,WAAO,yBAAAA,QAAM,MAAM,IAAI;AAAA,EACzB,QAAQ;AACN,QAAI;AACF,aAAO,yBAAAA,QAAM,UAAM,yBAAQ,IAAI,CAAC;AAAA,IAClC,QAAQ;AACN,aAAO;AAAA,IACT;AAAA,EACF;AACF;","names":["sjson"]}

package/dist/{runtimes/edge/partial-json → utils/json}/parse-partial-json.mjs

@@ -1,4 +1,4 @@
-// src/runtimes/edge/partial-json/parse-partial-json.ts
+// src/utils/json/parse-partial-json.ts
 import sjson from "secure-json-parse";
 import { fixJson } from "./fix-json.mjs";
 var parsePartialJson = (json) => {

package/dist/utils/json/parse-partial-json.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/utils/json/parse-partial-json.ts"],"sourcesContent":["import sjson from \"secure-json-parse\";\nimport { fixJson } from \"./fix-json\";\n\nexport const parsePartialJson = (json: string) => {\n  try {\n    return sjson.parse(json);\n  } catch {\n    try {\n      return sjson.parse(fixJson(json));\n    } catch {\n      return undefined;\n    }\n  }\n};\n"],"mappings":";AAAA,OAAO,WAAW;AAClB,SAAS,eAAe;AAEjB,IAAM,mBAAmB,CAAC,SAAiB;AAChD,MAAI;AACF,WAAO,MAAM,MAAM,IAAI;AAAA,EACzB,QAAQ;AACN,QAAI;AACF,aAAO,MAAM,MAAM,QAAQ,IAAI,CAAC;AAAA,IAClC,QAAQ;AACN,aAAO;AAAA,IACT;AAAA,EACF;AACF;","names":[]}

package/package.json

@@ -29,7 +29,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.7.57",
+  "version": "0.7.59",
   "license": "MIT",
   "exports": {
     ".": {
@@ -118,8 +118,8 @@
     "tailwindcss-animate": "^1.0.7",
     "tsx": "^4.19.2",
     "@assistant-ui/tailwindcss-transformer": "0.1.0",
-    "@assistant-ui/tsconfig": "0.0.0",
-    "@assistant-ui/tsbuildutils": "^0.0.1"
+    "@assistant-ui/tsbuildutils": "^0.0.1",
+    "@assistant-ui/tsconfig": "0.0.0"
   },
   "publishConfig": {
     "access": "public",

package/src/{runtimes/remote-thread-list/cloud → cloud}/AssistantCloudRuns.tsx

@@ -1,4 +1,4 @@
-import { CoreMessage } from "../../../types";
+import { CoreMessage } from "../types";
 import { AssistantCloudAPI } from "./AssistantCloudAPI";
 import { AssistantStream, PlainTextDecoder } from "assistant-stream";
 

package/src/cloud/AssistantCloudThreadHistoryAdapter.tsx

@@ -0,0 +1,64 @@
+import { RefObject, useState } from "react";
+import { useThreadListItemRuntime } from "../context";
+import { ThreadHistoryAdapter } from "../runtimes/adapters/thread-history/ThreadHistoryAdapter";
+import { ExportedMessageRepositoryItem } from "../runtimes/utils/MessageRepository";
+import { AssistantCloud } from "./AssistantCloud";
+import { auiV0Decode, auiV0Encode } from "./auiV0";
+import { ThreadListItemRuntime } from "../api";
+
+class AssistantCloudThreadHistoryAdapter implements ThreadHistoryAdapter {
+  constructor(
+    private cloudRef: RefObject<AssistantCloud>,
+    private threadListItemRuntime: ThreadListItemRuntime,
+  ) {}
+
+  private _getIdForLocalId: Record<string, string | Promise<string>> = {};
+
+  async append({ parentId, message }: ExportedMessageRepositoryItem) {
+    const { remoteId } = await this.threadListItemRuntime.initialize();
+    const task = this.cloudRef.current.threads.messages
+      .create(remoteId, {
+        parent_id: parentId
+          ? ((await this._getIdForLocalId[parentId]) ?? parentId)
+          : null,
+        format: "aui/v0",
+        content: auiV0Encode(message),
+      })
+      .then(({ message_id }) => {
+        this._getIdForLocalId[message.id] = message_id;
+        return message_id;
+      });
+
+    this._getIdForLocalId[message.id] = task;
+
+    return task.then(() => {});
+  }
+
+  async load() {
+    const remoteId = this.threadListItemRuntime.getState().remoteId;
+    if (!remoteId) return { messages: [] };
+    const { messages } =
+      await this.cloudRef.current.threads.messages.list(remoteId);
+    const payload = {
+      messages: messages
+        .filter(
+          (m): m is typeof m & { format: "aui/v0" } => m.format === "aui/v0",
+        )
+        .map(auiV0Decode)
+        .reverse(),
+    };
+    return payload;
+  }
+}
+
+export const useAssistantCloudThreadHistoryAdapter = (
+  cloudRef: RefObject<AssistantCloud>,
+): ThreadHistoryAdapter => {
+  const threadListItemRuntime = useThreadListItemRuntime();
+  const [adapter] = useState(
+    () =>
+      new AssistantCloudThreadHistoryAdapter(cloudRef, threadListItemRuntime),
+  );
+
+  return adapter;
+};

package/src/cloud/AssistantCloudThreadMessages.tsx

@@ -0,0 +1,48 @@
+import { ReadonlyJSONObject } from "../utils/json/json-value";
+import { AssistantCloudAPI } from "./AssistantCloudAPI";
+
+export type CloudMessage = {
+  id: string;
+  parent_id: string | null;
+  height: number;
+  created_at: Date;
+  updated_at: Date;
+  format: "aui/v0" | string;
+  content: ReadonlyJSONObject;
+};
+
+type AssistantCloudThreadMessageListResponse = {
+  messages: CloudMessage[];
+};
+
+type AssistantCloudThreadMessageCreateBody = {
+  parent_id: string | null;
+  format: "aui/v0" | string;
+  content: ReadonlyJSONObject;
+};
+
+type AssistantCloudMessageCreateResponse = {
+  message_id: string;
+};
+
+export class AssistantCloudThreadMessages {
+  constructor(private cloud: AssistantCloudAPI) {}
+
+  public async list(
+    threadId: string,
+  ): Promise<AssistantCloudThreadMessageListResponse> {
+    return this.cloud.makeRequest(
+      `/threads/${encodeURIComponent(threadId)}/messages`,
+    );
+  }
+
+  public async create(
+    threadId: string,
+    body: AssistantCloudThreadMessageCreateBody,
+  ): Promise<AssistantCloudMessageCreateResponse> {
+    return this.cloud.makeRequest(
+      `/threads/${encodeURIComponent(threadId)}/messages`,
+      { method: "POST", body },
+    );
+  }
+}

package/src/{runtimes/remote-thread-list/cloud → cloud}/AssistantCloudThreads.tsx

@@ -1,4 +1,5 @@
 import { AssistantCloudAPI } from "./AssistantCloudAPI";
+import { AssistantCloudThreadMessages } from "./AssistantCloudThreadMessages";
 
 type AssistantCloudThreadsListQuery = {
   is_archived?: boolean;
@@ -42,7 +43,11 @@ type AssistantCloudThreadsUpdateBody = {
 };
 
 export class AssistantCloudThreads {
-  constructor(private cloud: AssistantCloudAPI) {}
+  public readonly messages: AssistantCloudThreadMessages;
+
+  constructor(private cloud: AssistantCloudAPI) {
+    this.messages = new AssistantCloudThreadMessages(cloud);
+  }
 
   public async list(
     query?: AssistantCloudThreadsListQuery,
@@ -60,13 +65,15 @@ export class AssistantCloudThreads {
     threadId: string,
     body: AssistantCloudThreadsUpdateBody,
   ): Promise<void> {
-    return this.cloud.makeRequest(`/threads/${threadId}`, {
+    return this.cloud.makeRequest(`/threads/${encodeURIComponent(threadId)}`, {
       method: "PUT",
       body,
     });
   }
 
   public async delete(threadId: string): Promise<void> {
-    return this.cloud.makeRequest(`/threads/${threadId}`, { method: "DELETE" });
+    return this.cloud.makeRequest(`/threads/${encodeURIComponent(threadId)}`, {
+      method: "DELETE",
+    });
   }
 }

package/src/cloud/auiV0.ts

@@ -0,0 +1,130 @@
+import { JSONObject } from "@ai-sdk/provider";
+import { ThreadMessage } from "../types";
+import { MessageStatus } from "../types/AssistantTypes";
+import { fromThreadMessageLike } from "../runtimes/external-store/ThreadMessageLike";
+import { CloudMessage } from "./AssistantCloudThreadMessages";
+import { isJSONValue } from "../utils/json/is-json";
+import { ReadonlyJSONValue } from "../utils/json/json-value";
+import { ExportedMessageRepositoryItem } from "../runtimes/utils/MessageRepository";
+
+type AuiV0MessageContentPart =
+  | {
+      readonly type: "text";
+      readonly text: string;
+    }
+  | {
+      readonly type: "tool-call";
+      readonly toolCallId: string;
+      readonly toolName: string;
+      readonly args: ReadonlyJSONValue;
+      readonly result?: ReadonlyJSONValue;
+      readonly isError?: true;
+    }
+  | {
+      readonly type: "tool-call";
+      readonly toolCallId: string;
+      readonly toolName: string;
+      readonly argsText: string;
+      readonly result?: ReadonlyJSONValue;
+      readonly isError?: true;
+    };
+
+type AuiV0Message = {
+  readonly role: "assistant" | "user" | "system";
+  readonly status?: MessageStatus;
+  readonly content: readonly AuiV0MessageContentPart[];
+  readonly metadata: {
+    readonly unstable_annotations: readonly ReadonlyJSONValue[];
+    readonly unstable_data: readonly ReadonlyJSONValue[];
+    readonly steps: readonly {
+      readonly usage?: {
+        readonly promptTokens: number;
+        readonly completionTokens: number;
+      };
+    }[];
+    readonly custom: Readonly<JSONObject>;
+  };
+};
+
+export const auiV0Encode = (message: ThreadMessage): AuiV0Message => {
+  // TODO attachments are currently intentionally ignored
+  // info: ID and createdAt are ignored (we use the server value instead)
+  return {
+    role: message.role,
+    content: message.content.map((part) => {
+      const type = part.type;
+      switch (type) {
+        case "text": {
+          return {
+            type: "text",
+            text: part.text,
+          };
+        }
+
+        case "tool-call": {
+          if (!isJSONValue(part.result)) {
+            console.warn(
+              "tool-call result is not JSON! " + JSON.stringify(part),
+            );
+          }
+          return {
+            type: "tool-call",
+            toolCallId: part.toolCallId,
+            toolName: part.toolName,
+            ...(JSON.stringify(part.args) === part.argsText
+              ? {
+                  args: part.args,
+                }
+              : { argsText: part.argsText }),
+            ...(part.result
+              ? { result: part.result as ReadonlyJSONValue }
+              : {}),
+            ...(part.isError ? { isError: true } : {}),
+          };
+        }
+
+        default: {
+          const unhandledType: "ui" | "image" | "file" | "audio" = type;
+          throw new Error(
+            `Content part type not supported by aui/v0: ${unhandledType}`,
+          );
+        }
+      }
+    }),
+    metadata: message.metadata as AuiV0Message["metadata"],
+    ...(message.status
+      ? {
+          status:
+            message.status?.type === "running"
+              ? {
+                  type: "incomplete",
+                  reason: "cancelled",
+                }
+              : message.status,
+        }
+      : undefined),
+  };
+};
+
+export const auiV0Decode = (
+  cloudMessage: CloudMessage & { format: "aui/v0" },
+): ExportedMessageRepositoryItem => {
+  const payload = cloudMessage.content as unknown as AuiV0Message;
+  const message = fromThreadMessageLike(
+    {
+      id: cloudMessage.id,
+      createdAt: cloudMessage.created_at,
+      ...payload,
+    },
+    cloudMessage.id,
+    {
+      type: "complete",
+      reason: "unknown",
+    },
+  );
+
+  return {
+    parentId: cloudMessage.parent_id,
+    message,
+  };
+};

package/src/{runtimes/remote-thread-list/cloud → cloud}/useCloudThreadListRuntime.tsx

@@ -1,11 +1,21 @@
 "use client";
 
-import { useEffect, useRef, useState } from "react";
-import { useRemoteThreadListRuntime } from "../useRemoteThreadListRuntime";
+import {
+  FC,
+  PropsWithChildren,
+  useCallback,
+  useEffect,
+  useMemo,
+  useRef,
+  useState,
+} from "react";
 import { AssistantCloud } from "./AssistantCloud";
-import { AssistantRuntime } from "../../../api";
-import { RemoteThreadListSubscriber } from "../types";
-import { toCoreMessages } from "../../edge";
+import { AssistantRuntime } from "../api";
+import { RemoteThreadListSubscriber } from "../runtimes/remote-thread-list/types";
+import { toCoreMessages } from "../runtimes/edge";
+import { useRemoteThreadListRuntime } from "../runtimes/remote-thread-list/useRemoteThreadListRuntime";
+import { RuntimeAdapterProvider } from "../runtimes/adapters/RuntimeAdapterProvider";
+import { useAssistantCloudThreadHistoryAdapter } from "./AssistantCloudThreadHistoryAdapter";
 
 type ThreadData = {
   externalId: string;
@@ -104,6 +114,20 @@ export const useCloudThreadListRuntime = (adapter: CloudThreadListAdapter) => {
         subscribers.delete(callback);
       };
     },
+    unstable_Provider: useCallback<FC<PropsWithChildren>>(({ children }) => {
+      const history = useAssistantCloudThreadHistoryAdapter({
+        get current() {
+          return adapterRef.current.cloud;
+        },
+      });
+      const adapters = useMemo(() => ({ history }), [history]);
+
+      return (
+        <RuntimeAdapterProvider adapters={adapters}>
+          {children}
+        </RuntimeAdapterProvider>
+      );
+    }, []),
   });
 
   return runtime;

package/src/index.ts

@@ -1,4 +1,5 @@
 export * from "./api";
+export * from "./cloud";
 export * from "./context";
 export * from "./model-context";
 export * from "./primitives";

package/src/model-context/makeAssistantTool.tsx

@@ -2,15 +2,13 @@
 
 import { FC } from "react";
 import { type AssistantToolProps, useAssistantTool } from "./useAssistantTool";
+import { ReadonlyJSONObject } from "../utils/json/json-value";
 
 export type AssistantTool = FC & {
   unstable_tool: AssistantToolProps<any, any>;
 };
 
-export const makeAssistantTool = <
-  TArgs extends Record<string, unknown>,
-  TResult,
->(
+export const makeAssistantTool = <TArgs extends ReadonlyJSONObject, TResult>(
   tool: AssistantToolProps<TArgs, TResult>,
 ) => {
   const Tool: AssistantTool = () => {

package/src/model-context/makeAssistantToolUI.tsx

@@ -5,15 +5,13 @@ import {
   type AssistantToolUIProps,
   useAssistantToolUI,
 } from "./useAssistantToolUI";
+import { ReadonlyJSONObject } from "../utils/json/json-value";
 
 export type AssistantToolUI = FC & {
   unstable_tool: AssistantToolUIProps<any, any>;
 };
 
-export const makeAssistantToolUI = <
-  TArgs extends Record<string, unknown>,
-  TResult,
->(
+export const makeAssistantToolUI = <TArgs extends ReadonlyJSONObject, TResult>(
   tool: AssistantToolUIProps<TArgs, TResult>,
 ) => {
   const ToolUI: AssistantToolUI = () => {

package/src/model-context/useAssistantTool.tsx

@@ -7,9 +7,10 @@ import {
 } from "../context/react/AssistantContext";
 import type { ToolCallContentPartComponent } from "../types/ContentPartComponentTypes";
 import type { Tool } from "./ModelContextTypes";
+import { ReadonlyJSONObject } from "../utils/json/json-value";
 
 export type AssistantToolProps<
-  TArgs extends Record<string, unknown>,
+  TArgs extends ReadonlyJSONObject,
   TResult,
 > = Tool<TArgs, TResult> & {
   toolName: string;
@@ -17,10 +18,7 @@ export type AssistantToolProps<
   disabled?: boolean | undefined;
 };
 
-export const useAssistantTool = <
-  TArgs extends Record<string, unknown>,
-  TResult,
->(
+export const useAssistantTool = <TArgs extends ReadonlyJSONObject, TResult>(
   tool: AssistantToolProps<TArgs, TResult>,
 ) => {
   const assistantRuntime = useAssistantRuntime();

package/src/model-context/useAssistantToolUI.tsx

@@ -3,11 +3,9 @@
 import { useEffect } from "react";
 import { useToolUIsStore } from "../context/react/AssistantContext";
 import type { ToolCallContentPartComponent } from "../types/ContentPartComponentTypes";
+import { ReadonlyJSONObject } from "../utils/json/json-value";
 
-export type AssistantToolUIProps<
-  TArgs extends Record<string, unknown>,
-  TResult,
-> = {
+export type AssistantToolUIProps<TArgs extends ReadonlyJSONObject, TResult> = {
   toolName: string;
   render: ToolCallContentPartComponent<TArgs, TResult>;
 };

package/src/model-context/useInlineRender.tsx

@@ -3,8 +3,9 @@
 import { FC, useCallback, useEffect, useState } from "react";
 import { ToolCallContentPartProps } from "../types";
 import { create } from "zustand";
+import { ReadonlyJSONObject } from "../utils/json/json-value";
 
-export const useInlineRender = <TArgs extends Record<string, unknown>, TResult>(
+export const useInlineRender = <TArgs extends ReadonlyJSONObject, TResult>(
   toolUI: FC<ToolCallContentPartProps<TArgs, TResult>>,
 ): FC<ToolCallContentPartProps<TArgs, TResult>> => {
   const [useToolUI] = useState(() => create(() => toolUI));

package/src/runtimes/edge/EdgeChatAdapter.ts

@@ -68,9 +68,9 @@ export class EdgeChatAdapter implements ChatModelAdapter {
       body: JSON.stringify({
         system: context.system,
         messages: this.options.unstable_AISDKInterop
-          ? (toLanguageModelMessages(
-              messages,
-            ) as EdgeRuntimeRequestOptions["messages"]) // TODO figure out a better way to do this
+          ? (toLanguageModelMessages(messages, {
+              unstable_includeId: this.options.unstable_sendMessageIds,
+            }) as EdgeRuntimeRequestOptions["messages"]) // TODO figure out a better way to do this
           : toCoreMessages(messages, {
               unstable_includeId: this.options.unstable_sendMessageIds,
             }),

package/src/runtimes/edge/EdgeRuntimeRequestOptions.ts

@@ -4,6 +4,7 @@ import {
   LanguageModelV1CallSettingsSchema,
 } from "../../model-context/ModelContextTypes";
 import { z } from "zod";
+import { ReadonlyJSONObject } from "../../utils/json/json-value";
 
 const LanguageModelV1FunctionToolSchema = z.object({
   type: z.literal("function"),
@@ -42,11 +43,12 @@ const CoreToolCallContentPartSchema = z.object({
   type: z.literal("tool-call"),
   toolCallId: z.string(),
   toolName: z.string(),
-  args: z.record(z.unknown()),
+  args: z
+    .record(z.unknown())
+    .refine((c): c is ReadonlyJSONObject => c !== undefined),
   result: z.unknown().optional(),
   isError: z.boolean().optional(),
 });
-// args is required but unknown;
 
 const CoreUserMessageSchema = z.object({
   role: z.literal("user"),

package/src/runtimes/edge/converters/fromLanguageModelMessages.ts

@@ -1,6 +1,7 @@
 import { LanguageModelV1Message } from "@ai-sdk/provider";
 import { CoreMessage, ToolCallContentPart } from "../../../types";
 import { Writable } from "stream";
+import { ReadonlyJSONObject } from "../../../utils/json/json-value";
 
 type fromLanguageModelMessagesOptions = {
   mergeSteps: boolean;
@@ -76,7 +77,7 @@ export const fromLanguageModelMessages = (
               toolCallId: part.toolCallId,
               toolName: part.toolName,
               argsText: JSON.stringify(part.args),
-              args: part.args as Record<string, unknown>,
+              args: part.args as ReadonlyJSONObject,
             } satisfies ToolCallContentPart;
           }
           return part;

package/src/runtimes/edge/converters/toLanguageModelMessages.ts

@@ -74,12 +74,22 @@ const assistantMessageSplitter = () => {
 
 export function toLanguageModelMessages(
   message: readonly CoreMessage[] | readonly ThreadMessage[],
+  options: { unstable_includeId?: boolean | undefined } = {},
 ): LanguageModelV1Message[] {
+  const includeId = options.unstable_includeId ?? false;
   return message.flatMap((message) => {
     const role = message.role;
     switch (role) {
       case "system": {
-        return [{ role: "system", content: message.content[0].text }];
+        return [
+          {
+            ...(includeId
+              ? { unstable_id: (message as ThreadMessage).id }
+              : {}),
+            role: "system",
+            content: message.content[0].text,
+          },
+        ];
       }
 
       case "user": {
@@ -89,6 +99,7 @@ export function toLanguageModelMessages(
           ...attachments.map((a) => a.content).flat(),
         ];
         const msg: LanguageModelV1Message = {
+          ...(includeId ? { unstable_id: (message as ThreadMessage).id } : {}),
           role: "user",
           content: content.map(
             (

package/src/runtimes/edge/createEdgeRuntimeAPI.ts

@@ -26,7 +26,7 @@ import { streamPartEncoderStream } from "./streams/utils/streamPartEncoderStream
 import { z } from "zod";
 
 type FinishResult = {
-  messages: CoreMessage[];
+  messages: readonly CoreMessage[];
   metadata: {
     steps: readonly ThreadStep[];
   };
@@ -111,7 +111,7 @@ export const getEdgeRuntimeStream = async ({
     abortSignal,
 
     ...(!!system ? { system } : undefined),
-    messages: [...messages],
+    messages,
     tools: lmServerTools.concat(clientTools as LanguageModelV1FunctionTool[]),
     ...(toolChoice ? { toolChoice } : undefined),
   });
@@ -201,7 +201,7 @@ export const createEdgeRuntimeAPI = (options: CreateEdgeRuntimeAPIOptions) => ({
 type StreamMessageOptions = LanguageModelV1CallSettings & {
   model: LanguageModelV1;
   system?: string;
-  messages: CoreMessage[];
+  messages: readonly CoreMessage[];
   tools?: LanguageModelV1FunctionTool[];
   toolChoice?: LanguageModelV1ToolChoice;
   abortSignal: AbortSignal;
@@ -229,7 +229,7 @@ async function streamMessage({
 
 export function convertToLanguageModelPrompt(
   system: string | undefined,
-  messages: CoreMessage[],
+  messages: readonly CoreMessage[],
 ): LanguageModelV1Prompt {
   const languageModelMessages: LanguageModelV1Prompt = [];
 

package/src/runtimes/edge/streams/AssistantStreamChunkType.ts

@@ -1,4 +1,5 @@
-import { JSONValue, LanguageModelV1StreamPart } from "@ai-sdk/provider";
+import { LanguageModelV1StreamPart } from "@ai-sdk/provider";
+import { ReadonlyJSONValue } from "../../../utils/json/json-value";
 
 export enum AssistantStreamChunkType {
   TextDelta = "0",
@@ -17,8 +18,8 @@ export enum AssistantStreamChunkType {
 
 export type AssistantStreamChunk = {
   [AssistantStreamChunkType.TextDelta]: string;
-  [AssistantStreamChunkType.Data]: JSONValue[];
-  [AssistantStreamChunkType.Annotation]: JSONValue[];
+  [AssistantStreamChunkType.Data]: ReadonlyJSONValue[];
+  [AssistantStreamChunkType.Annotation]: ReadonlyJSONValue[];
   [AssistantStreamChunkType.ToolCall]: {
     toolCallId: string;
     toolName: string;

package/src/runtimes/edge/streams/runResultStream.ts

@@ -1,5 +1,5 @@
 import { CoreChatModelRunResult } from "../../local/ChatModelAdapter";
-import { parsePartialJson } from "../partial-json/parse-partial-json";
+import { parsePartialJson } from "../../../utils/json/parse-partial-json";
 import { LanguageModelV1StreamPart } from "@ai-sdk/provider";
 import { ToolResultStreamPart } from "./toolResultStream";
 import { MessageStatus, ToolCallContentPart } from "../../../types";

package/src/runtimes/edge/streams/toolResultStream.ts

@@ -1,17 +1,18 @@
 import { Tool } from "../../../model-context/ModelContextTypes";
-import { JSONValue, LanguageModelV1StreamPart } from "@ai-sdk/provider";
+import { LanguageModelV1StreamPart } from "@ai-sdk/provider";
 import { z } from "zod";
 import sjson from "secure-json-parse";
+import { ReadonlyJSONValue } from "../../../utils/json/json-value";
 
 export type ToolResultStreamPart =
   | LanguageModelV1StreamPart
   | {
       type: "annotations";
-      annotations: JSONValue[];
+      annotations: ReadonlyJSONValue[];
     }
   | {
       type: "data";
-      data: JSONValue[];
+      data: ReadonlyJSONValue[];
     }
   | {
       type: "tool-result";