@assistant-ui/react 0.8.6 → 0.8.8

package/dist/utils/json/is-json.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/utils/json/is-json.ts"],"sourcesContent":["import {\n  ReadonlyJSONArray,\n  ReadonlyJSONObject,\n  ReadonlyJSONValue,\n} from \"./json-value\";\n\nexport function isJSONValue(value: unknown): value is ReadonlyJSONValue {\n  if (\n    value === null ||\n    typeof value === \"string\" ||\n    typeof value === \"number\" ||\n    typeof value === \"boolean\"\n  ) {\n    return true;\n  }\n\n  if (Array.isArray(value)) {\n    return value.every(isJSONValue);\n  }\n\n  if (typeof value === \"object\") {\n    return Object.entries(value).every(\n      ([key, val]) => typeof key === \"string\" && isJSONValue(val),\n    );\n  }\n\n  return false;\n}\n\nexport function isJSONArray(value: unknown): value is ReadonlyJSONArray {\n  return Array.isArray(value) && value.every(isJSONValue);\n}\n\nexport function isJSONObject(value: unknown): value is ReadonlyJSONObject {\n  return (\n    value != null &&\n    typeof value === \"object\" &&\n    Object.entries(value).every(\n      ([key, val]) => typeof key === \"string\" && isJSONValue(val),\n    )\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAMO,SAAS,YAAY,OAA4C;AACtE,MACE,UAAU,QACV,OAAO,UAAU,YACjB,OAAO,UAAU,YACjB,OAAO,UAAU,WACjB;AACA,WAAO;AAAA,EACT;AAEA,MAAI,MAAM,QAAQ,KAAK,GAAG;AACxB,WAAO,MAAM,MAAM,WAAW;AAAA,EAChC;AAEA,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO,OAAO,QAAQ,KAAK,EAAE;AAAA,MAC3B,CAAC,CAAC,KAAK,GAAG,MAAM,OAAO,QAAQ,YAAY,YAAY,GAAG;AAAA,IAC5D;AAAA,EACF;AAEA,SAAO;AACT;AAEO,SAAS,YAAY,OAA4C;AACtE,SAAO,MAAM,QAAQ,KAAK,KAAK,MAAM,MAAM,WAAW;AACxD;AAEO,SAAS,aAAa,OAA6C;AACxE,SACE,SAAS,QACT,OAAO,UAAU,YACjB,OAAO,QAAQ,KAAK,EAAE;AAAA,IACpB,CAAC,CAAC,KAAK,GAAG,MAAM,OAAO,QAAQ,YAAY,YAAY,GAAG;AAAA,EAC5D;AAEJ;","names":[]}
+{"version":3,"sources":["../../../src/utils/json/is-json.ts"],"sourcesContent":["import {\n  ReadonlyJSONArray,\n  ReadonlyJSONObject,\n  ReadonlyJSONValue,\n} from \"./json-value\";\n\nexport function isJSONValue(\n  value: unknown,\n  currentDepth: number = 0,\n): value is ReadonlyJSONValue {\n  // Protect against too deep recursion\n  if (currentDepth > 100) {\n    return false;\n  }\n\n  if (\n    value === null ||\n    typeof value === \"string\" ||\n    typeof value === \"boolean\"\n  ) {\n    return true;\n  }\n\n  // Handle special number cases\n  if (typeof value === \"number\") {\n    return !Number.isNaN(value) && Number.isFinite(value);\n  }\n\n  if (Array.isArray(value)) {\n    return value.every((item) => isJSONValue(item, currentDepth + 1));\n  }\n\n  if (typeof value === \"object\") {\n    return Object.entries(value).every(\n      ([key, val]) =>\n        typeof key === \"string\" && isJSONValue(val, currentDepth + 1),\n    );\n  }\n\n  return false;\n}\n\nexport function isJSONArray(value: unknown): value is ReadonlyJSONArray {\n  return Array.isArray(value) && value.every(isJSONValue);\n}\n\nexport function isJSONObject(value: unknown): value is ReadonlyJSONObject {\n  return (\n    value != null &&\n    typeof value === \"object\" &&\n    Object.entries(value).every(\n      ([key, val]) => typeof key === \"string\" && isJSONValue(val),\n    )\n  );\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAMO,SAAS,YACd,OACA,eAAuB,GACK;AAE5B,MAAI,eAAe,KAAK;AACtB,WAAO;AAAA,EACT;AAEA,MACE,UAAU,QACV,OAAO,UAAU,YACjB,OAAO,UAAU,WACjB;AACA,WAAO;AAAA,EACT;AAGA,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO,CAAC,OAAO,MAAM,KAAK,KAAK,OAAO,SAAS,KAAK;AAAA,EACtD;AAEA,MAAI,MAAM,QAAQ,KAAK,GAAG;AACxB,WAAO,MAAM,MAAM,CAAC,SAAS,YAAY,MAAM,eAAe,CAAC,CAAC;AAAA,EAClE;AAEA,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO,OAAO,QAAQ,KAAK,EAAE;AAAA,MAC3B,CAAC,CAAC,KAAK,GAAG,MACR,OAAO,QAAQ,YAAY,YAAY,KAAK,eAAe,CAAC;AAAA,IAChE;AAAA,EACF;AAEA,SAAO;AACT;AAEO,SAAS,YAAY,OAA4C;AACtE,SAAO,MAAM,QAAQ,KAAK,KAAK,MAAM,MAAM,WAAW;AACxD;AAEO,SAAS,aAAa,OAA6C;AACxE,SACE,SAAS,QACT,OAAO,UAAU,YACjB,OAAO,QAAQ,KAAK,EAAE;AAAA,IACpB,CAAC,CAAC,KAAK,GAAG,MAAM,OAAO,QAAQ,YAAY,YAAY,GAAG;AAAA,EAC5D;AAEJ;","names":[]}

package/dist/utils/json/is-json.mjs

@@ -1,14 +1,20 @@
 // src/utils/json/is-json.ts
-function isJSONValue(value) {
-  if (value === null || typeof value === "string" || typeof value === "number" || typeof value === "boolean") {
+function isJSONValue(value, currentDepth = 0) {
+  if (currentDepth > 100) {
+    return false;
+  }
+  if (value === null || typeof value === "string" || typeof value === "boolean") {
     return true;
   }
+  if (typeof value === "number") {
+    return !Number.isNaN(value) && Number.isFinite(value);
+  }
   if (Array.isArray(value)) {
-    return value.every(isJSONValue);
+    return value.every((item) => isJSONValue(item, currentDepth + 1));
   }
   if (typeof value === "object") {
     return Object.entries(value).every(
-      ([key, val]) => typeof key === "string" && isJSONValue(val)
+      ([key, val]) => typeof key === "string" && isJSONValue(val, currentDepth + 1)
     );
   }
   return false;

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/utils/json/is-json.ts"],"sourcesContent":["import {\n  ReadonlyJSONArray,\n  ReadonlyJSONObject,\n  ReadonlyJSONValue,\n} from \"./json-value\";\n\nexport function isJSONValue(value: unknown): value is ReadonlyJSONValue {\n  if (\n    value === null ||\n    typeof value === \"string\" ||\n    typeof value === \"number\" ||\n    typeof value === \"boolean\"\n  ) {\n    return true;\n  }\n\n  if (Array.isArray(value)) {\n    return value.every(isJSONValue);\n  }\n\n  if (typeof value === \"object\") {\n    return Object.entries(value).every(\n      ([key, val]) => typeof key === \"string\" && isJSONValue(val),\n    );\n  }\n\n  return false;\n}\n\nexport function isJSONArray(value: unknown): value is ReadonlyJSONArray {\n  return Array.isArray(value) && value.every(isJSONValue);\n}\n\nexport function isJSONObject(value: unknown): value is ReadonlyJSONObject {\n  return (\n    value != null &&\n    typeof value === \"object\" &&\n    Object.entries(value).every(\n      ([key, val]) => typeof key === \"string\" && isJSONValue(val),\n    )\n  );\n}\n"],"mappings":";AAMO,SAAS,YAAY,OAA4C;AACtE,MACE,UAAU,QACV,OAAO,UAAU,YACjB,OAAO,UAAU,YACjB,OAAO,UAAU,WACjB;AACA,WAAO;AAAA,EACT;AAEA,MAAI,MAAM,QAAQ,KAAK,GAAG;AACxB,WAAO,MAAM,MAAM,WAAW;AAAA,EAChC;AAEA,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO,OAAO,QAAQ,KAAK,EAAE;AAAA,MAC3B,CAAC,CAAC,KAAK,GAAG,MAAM,OAAO,QAAQ,YAAY,YAAY,GAAG;AAAA,IAC5D;AAAA,EACF;AAEA,SAAO;AACT;AAEO,SAAS,YAAY,OAA4C;AACtE,SAAO,MAAM,QAAQ,KAAK,KAAK,MAAM,MAAM,WAAW;AACxD;AAEO,SAAS,aAAa,OAA6C;AACxE,SACE,SAAS,QACT,OAAO,UAAU,YACjB,OAAO,QAAQ,KAAK,EAAE;AAAA,IACpB,CAAC,CAAC,KAAK,GAAG,MAAM,OAAO,QAAQ,YAAY,YAAY,GAAG;AAAA,EAC5D;AAEJ;","names":[]}
+{"version":3,"sources":["../../../src/utils/json/is-json.ts"],"sourcesContent":["import {\n  ReadonlyJSONArray,\n  ReadonlyJSONObject,\n  ReadonlyJSONValue,\n} from \"./json-value\";\n\nexport function isJSONValue(\n  value: unknown,\n  currentDepth: number = 0,\n): value is ReadonlyJSONValue {\n  // Protect against too deep recursion\n  if (currentDepth > 100) {\n    return false;\n  }\n\n  if (\n    value === null ||\n    typeof value === \"string\" ||\n    typeof value === \"boolean\"\n  ) {\n    return true;\n  }\n\n  // Handle special number cases\n  if (typeof value === \"number\") {\n    return !Number.isNaN(value) && Number.isFinite(value);\n  }\n\n  if (Array.isArray(value)) {\n    return value.every((item) => isJSONValue(item, currentDepth + 1));\n  }\n\n  if (typeof value === \"object\") {\n    return Object.entries(value).every(\n      ([key, val]) =>\n        typeof key === \"string\" && isJSONValue(val, currentDepth + 1),\n    );\n  }\n\n  return false;\n}\n\nexport function isJSONArray(value: unknown): value is ReadonlyJSONArray {\n  return Array.isArray(value) && value.every(isJSONValue);\n}\n\nexport function isJSONObject(value: unknown): value is ReadonlyJSONObject {\n  return (\n    value != null &&\n    typeof value === \"object\" &&\n    Object.entries(value).every(\n      ([key, val]) => typeof key === \"string\" && isJSONValue(val),\n    )\n  );\n}\n"],"mappings":";AAMO,SAAS,YACd,OACA,eAAuB,GACK;AAE5B,MAAI,eAAe,KAAK;AACtB,WAAO;AAAA,EACT;AAEA,MACE,UAAU,QACV,OAAO,UAAU,YACjB,OAAO,UAAU,WACjB;AACA,WAAO;AAAA,EACT;AAGA,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO,CAAC,OAAO,MAAM,KAAK,KAAK,OAAO,SAAS,KAAK;AAAA,EACtD;AAEA,MAAI,MAAM,QAAQ,KAAK,GAAG;AACxB,WAAO,MAAM,MAAM,CAAC,SAAS,YAAY,MAAM,eAAe,CAAC,CAAC;AAAA,EAClE;AAEA,MAAI,OAAO,UAAU,UAAU;AAC7B,WAAO,OAAO,QAAQ,KAAK,EAAE;AAAA,MAC3B,CAAC,CAAC,KAAK,GAAG,MACR,OAAO,QAAQ,YAAY,YAAY,KAAK,eAAe,CAAC;AAAA,IAChE;AAAA,EACF;AAEA,SAAO;AACT;AAEO,SAAS,YAAY,OAA4C;AACtE,SAAO,MAAM,QAAQ,KAAK,KAAK,MAAM,MAAM,WAAW;AACxD;AAEO,SAAS,aAAa,OAA6C;AACxE,SACE,SAAS,QACT,OAAO,UAAU,YACjB,OAAO,QAAQ,KAAK,EAAE;AAAA,IACpB,CAAC,CAAC,KAAK,GAAG,MAAM,OAAO,QAAQ,YAAY,YAAY,GAAG;AAAA,EAC5D;AAEJ;","names":[]}

package/package.json

@@ -29,7 +29,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.8.6",
+  "version": "0.8.8",
   "license": "MIT",
   "exports": {
     ".": {
@@ -56,7 +56,7 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@ai-sdk/provider": "^1.0.7",
+    "@ai-sdk/provider": "^1.1.0",
     "@radix-ui/primitive": "^1.1.1",
     "@radix-ui/react-compose-refs": "^1.1.1",
     "@radix-ui/react-context": "^1.1.1",
@@ -72,7 +72,7 @@
     "zod": "^3.24.1",
     "zod-to-json-schema": "^3.24.1",
     "zustand": "^5.0.3",
-    "assistant-stream": "^0.0.21"
+    "assistant-stream": "^0.0.23"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -94,8 +94,8 @@
     "eslint": "^9",
     "eslint-config-next": "15.1.6",
     "tsx": "^4.19.2",
-    "@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/cloud/auiV0.ts

@@ -82,8 +82,15 @@ export const auiV0Encode = (message: ThreadMessage): AuiV0Message => {
           };
         }
 
-        case "source":
-          return part;
+        case "source": {
+          return {
+            type: "source",
+            sourceType: part.sourceType,
+            id: part.id,
+            url: part.url,
+            ...(part.title ? { title: part.title } : undefined),
+          };
+        }
 
         case "tool-call": {
           if (!isJSONValue(part.result)) {

package/src/primitives/contentPart/useContentPartFile.tsx

@@ -0,0 +1,18 @@
+"use client";
+
+import { ContentPartState } from "../../api/ContentPartRuntime";
+import { useContentPart } from "../../context/react/ContentPartContext";
+import { FileContentPart } from "../../types";
+
+export const useContentPartFile = () => {
+  const file = useContentPart((c) => {
+    if (c.type !== "file")
+      throw new Error(
+        "ContentPartFile can only be used inside file content parts.",
+      );
+
+    return c as ContentPartState & FileContentPart;
+  });
+
+  return file;
+};

package/src/primitives/contentPart/useContentPartSource.tsx

@@ -0,0 +1,18 @@
+"use client";
+
+import { ContentPartState } from "../../api/ContentPartRuntime";
+import { useContentPart } from "../../context/react/ContentPartContext";
+import { SourceContentPart } from "../../types";
+
+export const useContentPartSource = () => {
+  const source = useContentPart((c) => {
+    if (c.type !== "source")
+      throw new Error(
+        "ContentPartSource can only be used inside source content parts.",
+      );
+
+    return c as ContentPartState & SourceContentPart;
+  });
+
+  return source;
+};

package/src/primitives/index.ts

@@ -9,7 +9,9 @@ export * as ThreadPrimitive from "./thread";
 export * as ThreadListPrimitive from "./threadList";
 export * as ThreadListItemPrimitive from "./threadListItem";
 
-export { useContentPartImage } from "./contentPart/useContentPartImage";
 export { useContentPartText } from "./contentPart/useContentPartText";
 export { useContentPartReasoning } from "./contentPart/useContentPartReasoning";
+export { useContentPartSource } from "./contentPart/useContentPartSource";
+export { useContentPartFile } from "./contentPart/useContentPartFile";
+export { useContentPartImage } from "./contentPart/useContentPartImage";
 export { useThreadViewportAutoScroll } from "./thread/useThreadViewportAutoScroll";

package/src/runtimes/adapters/thread-history/ThreadHistoryAdapter.ts

@@ -1,9 +1,13 @@
+import { ChatModelRunOptions, ChatModelRunResult } from "../../local";
 import {
   ExportedMessageRepository,
   ExportedMessageRepositoryItem,
 } from "../../utils/MessageRepository";
 
 export type ThreadHistoryAdapter = {
-  load(): Promise<ExportedMessageRepository>;
+  load(): Promise<ExportedMessageRepository & { unstable_resume?: boolean }>;
+  resume?(
+    options: ChatModelRunOptions,
+  ): AsyncGenerator<ChatModelRunResult, void, unknown>;
   append(item: ExportedMessageRepositoryItem): Promise<void>;
 };

package/src/runtimes/dangerous-in-browser/DangerousInBrowserAdapter.ts

@@ -5,13 +5,13 @@ import {
 import { toCoreMessages } from "../edge/converters/toCoreMessages";
 import { toLanguageModelTools } from "../edge/converters/toLanguageModelTools";
 import { EdgeRuntimeRequestOptions } from "../edge/EdgeRuntimeRequestOptions";
-import { runResultStream } from "../edge/streams/runResultStream";
 import { toolResultStream } from "../edge/streams/toolResultStream";
 import { asAsyncIterable } from "../edge/EdgeChatAdapter";
 import {
   CreateEdgeRuntimeAPIOptions,
   getEdgeRuntimeStream,
 } from "../edge/createEdgeRuntimeAPI";
+import { AssistantMessageAccumulator } from "assistant-stream";
 
 export type DangerousInBrowserAdapterOptions = CreateEdgeRuntimeAPIOptions;
 
@@ -33,7 +33,7 @@ export class DangerousInBrowserAdapter implements ChatModelAdapter {
 
     const stream = res
       .pipeThrough(toolResultStream(context.tools, abortSignal))
-      .pipeThrough(runResultStream());
+      .pipeThrough(new AssistantMessageAccumulator());
 
     for await (const update of asAsyncIterable(stream)) {
       yield update;

package/src/runtimes/edge/EdgeChatAdapter.ts

@@ -2,13 +2,9 @@ import {
   ChatModelAdapter,
   ChatModelRunOptions,
 } from "../local/ChatModelAdapter";
-import { ChatModelRunResult } from "../local/ChatModelAdapter";
 import { toCoreMessages } from "./converters/toCoreMessages";
 import { toLanguageModelTools } from "./converters/toLanguageModelTools";
 import { EdgeRuntimeRequestOptions } from "./EdgeRuntimeRequestOptions";
-import { assistantDecoderStream } from "./streams/assistantDecoderStream";
-import { streamPartDecoderStream } from "./streams/utils/streamPartDecoderStream";
-import { runResultStream } from "./streams/runResultStream";
 import { toolResultStream } from "./streams/toolResultStream";
 import { toLanguageModelMessages } from "./converters";
 import { ThreadMessage } from "../../types";
@@ -16,6 +12,10 @@ import { Tool } from "../../model-context";
 import { z } from "zod";
 import zodToJsonSchema from "zod-to-json-schema";
 import { JSONSchema7 } from "json-schema";
+import {
+  AssistantMessageAccumulator,
+  DataStreamDecoder,
+} from "assistant-stream";
 
 export function asAsyncIterable<T>(
   source: ReadableStream<T>,
@@ -67,13 +67,13 @@ export type EdgeChatAdapterOptions = {
   onError?: (error: Error) => void;
 
   credentials?: RequestCredentials;
-  
+
   /**
    * Headers to be sent with the request.
    * Can be a static headers object or a function that returns a Promise of headers.
    */
   headers?: HeadersValue | (() => Promise<HeadersValue>);
-  
+
   body?: object;
 
   /**
@@ -120,10 +120,11 @@ export class EdgeChatAdapter implements ChatModelAdapter {
     unstable_assistantMessageId,
     unstable_getMessage,
   }: ChatModelRunOptions) {
-    const headersValue = typeof this.options.headers === 'function' 
-      ? await this.options.headers() 
-      : this.options.headers;
-    
+    const headersValue =
+      typeof this.options.headers === "function"
+        ? await this.options.headers()
+        : this.options.headers;
+
     const headers = new Headers(headersValue);
     headers.set("Content-Type", "application/json");
 
@@ -165,21 +166,19 @@ export class EdgeChatAdapter implements ChatModelAdapter {
       if (!result.ok) {
         throw new Error(`Status ${result.status}: ${await result.text()}`);
       }
+      if (!result.body) {
+        throw new Error("Response body is null");
+      }
 
-      const stream = result
-        .body!.pipeThrough(streamPartDecoderStream())
-        .pipeThrough(assistantDecoderStream())
+      const stream = result.body
+        .pipeThrough(new DataStreamDecoder())
         .pipeThrough(toolResultStream(context.tools, abortSignal))
-        .pipeThrough(runResultStream());
+        .pipeThrough(new AssistantMessageAccumulator());
 
-      let update: ChatModelRunResult | undefined;
-      for await (update of asAsyncIterable(stream)) {
+      for await (const update of asAsyncIterable(stream)) {
         yield update;
       }
 
-      if (update === undefined)
-        throw new Error("No data received from Edge Runtime");
-
       this.options.onFinish?.(unstable_getMessage());
     } catch (error: unknown) {
       this.options.onError?.(error as Error);

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

@@ -4,12 +4,12 @@ import { Writable } from "stream";
 import { ReadonlyJSONObject } from "../../../utils/json/json-value";
 
 type fromLanguageModelMessagesOptions = {
-  mergeSteps: boolean;
+  mergeSteps?: boolean;
 };
 
 export const fromLanguageModelMessages = (
   lm: LanguageModelV1Message[],
-  { mergeSteps }: fromLanguageModelMessagesOptions,
+  { mergeSteps }: fromLanguageModelMessagesOptions = {},
 ): CoreMessage[] => {
   const messages: CoreMessage[] = [];
 
@@ -70,18 +70,28 @@ export const fromLanguageModelMessages = (
         break;
       }
       case "assistant": {
-        const newContent = lmMessage.content.map((part) => {
-          if (part.type === "tool-call") {
-            return {
-              type: "tool-call",
-              toolCallId: part.toolCallId,
-              toolName: part.toolName,
-              argsText: JSON.stringify(part.args),
-              args: part.args as ReadonlyJSONObject,
-            } satisfies ToolCallContentPart;
-          }
-          return part;
-        });
+        const newContent = lmMessage.content
+          .map((part) => {
+            if (part.type === "tool-call") {
+              return {
+                type: "tool-call",
+                toolCallId: part.toolCallId,
+                toolName: part.toolName,
+                argsText: JSON.stringify(part.args),
+                args: part.args as ReadonlyJSONObject,
+              } satisfies ToolCallContentPart;
+            }
+            // TODO handle these
+            if (
+              part.type === "redacted-reasoning" ||
+              part.type === "file" ||
+              part.type === "reasoning"
+            ) {
+              return null;
+            }
+            return part;
+          })
+          .filter((p) => !!p);
 
         if (mergeSteps) {
           const previousMessage = messages[messages.length - 1];

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

@@ -23,8 +23,12 @@ export const toCoreMessage = <T extends boolean = false>(
         role,
         content: message.content
           .map((part) => {
-            if (part.type === "reasoning" || part.type === "source")
-              return null; // reasoning and source parts are omitted
+            if (
+              part.type === "reasoning" ||
+              part.type === "source" ||
+              part.type === "file"
+            )
+              return null; // reasoning, source, and file parts are omitted
             if (part.type === "tool-call") {
               const { argsText, ...rest } = part;
               return rest;

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

@@ -151,8 +151,9 @@ export function toLanguageModelMessages(
           const type = part.type;
           switch (type) {
             case "reasoning":
-            case "source": {
-              break; // reasoning and source parts are omitted
+            case "source":
+            case "file": {
+              break; // reasoning, source, and file parts are omitted
             }
 
             case "text": {

package/src/runtimes/edge/createEdgeRuntimeAPI.ts

@@ -10,24 +10,24 @@ import {
   ThreadMessage,
   ThreadStep,
 } from "../../types/AssistantTypes";
-import { assistantEncoderStream } from "./streams/assistantEncoderStream";
 import { EdgeRuntimeRequestOptionsSchema } from "./EdgeRuntimeRequestOptions";
 import { toLanguageModelMessages } from "./converters/toLanguageModelMessages";
 import { toLanguageModelTools } from "./converters/toLanguageModelTools";
-import {
-  toolResultStream,
-  ToolResultStreamPart,
-} from "./streams/toolResultStream";
-import { runResultStream } from "./streams/runResultStream";
+import { toolResultStream } from "./streams/toolResultStream";
 import {
   LanguageModelConfig,
   LanguageModelV1CallSettings,
   LanguageModelV1CallSettingsSchema,
   Tool,
 } from "../../model-context/ModelContextTypes";
-import { CoreChatModelRunResult } from "../local/ChatModelAdapter";
-import { streamPartEncoderStream } from "./streams/utils/streamPartEncoderStream";
 import { z } from "zod";
+import {
+  AssistantMessage,
+  AssistantMessageAccumulator,
+  AssistantStreamChunk,
+  DataStreamEncoder,
+} from "assistant-stream";
+import { LanguageModelV1StreamDecoder } from "assistant-stream/ai-sdk";
 
 type FinishResult = {
   messages: readonly (CoreMessage | ThreadMessage)[];
@@ -48,14 +48,6 @@ export type CreateEdgeRuntimeAPIOptions = LanguageModelV1CallSettings & {
   onFinish?: (result: FinishResult) => void;
 };
 
-const voidStream = () => {
-  return new WritableStream({
-    abort(reason) {
-      console.error("Server stream processing aborted:", reason);
-    },
-  });
-};
-
 type GetEdgeRuntimeStreamOptions = {
   abortSignal: AbortSignal;
   requestData: z.infer<typeof EdgeRuntimeRequestOptionsSchema>;
@@ -106,7 +98,7 @@ export const getEdgeRuntimeStream = async ({
       ? await modelOrCreator({ apiKey, baseUrl, modelName })
       : modelOrCreator;
 
-  let stream: ReadableStream<ToolResultStreamPart>;
+  let stream: ReadableStream<AssistantStreamChunk>;
   const streamResult = await streamMessage({
     ...(settings as Partial<StreamMessageOptions>),
     ...callSettings,
@@ -119,7 +111,7 @@ export const getEdgeRuntimeStream = async ({
     tools: lmServerTools.concat(clientTools as LanguageModelV1FunctionTool[]),
     ...(toolChoice ? { toolChoice } : undefined),
   });
-  stream = streamResult.stream;
+  stream = streamResult.stream.pipeThrough(new LanguageModelV1StreamDecoder());
 
   // add tool results if we have server tools
   const canExecuteTools = hasServerTools && toolChoice?.type !== "none";
@@ -134,13 +126,13 @@ export const getEdgeRuntimeStream = async ({
     let serverStream = tees[1];
 
     if (onFinish) {
-      let lastChunk: CoreChatModelRunResult | undefined;
-      serverStream = serverStream.pipeThrough(runResultStream()).pipeThrough(
-        new TransformStream({
-          transform(chunk) {
+      let lastChunk: AssistantMessage | undefined;
+      serverStream.pipeThrough(new AssistantMessageAccumulator()).pipeTo(
+        new WritableStream({
+          write(chunk) {
             lastChunk = chunk;
           },
-          flush() {
+          close() {
             if (!lastChunk?.status || lastChunk.status.type === "running")
               return;
 
@@ -152,32 +144,22 @@ export const getEdgeRuntimeStream = async ({
                 role: "assistant",
                 content: lastChunk.content,
                 status: lastChunk.status,
-                metadata: {
-                  unstable_data: lastChunk.metadata?.unstable_data ?? [],
-                  unstable_annotations:
-                    lastChunk.metadata?.unstable_annotations ?? [],
-                  steps: lastChunk.metadata?.steps ?? [],
-                  custom: lastChunk.metadata?.custom ?? {},
-                },
+                metadata: lastChunk.metadata,
               } satisfies ThreadMessage,
             ];
             onFinish({
               messages: resultingMessages,
               metadata: {
-                // TODO
-                // eslint-disable-next-line @typescript-eslint/no-non-null-asserted-optional-chain
-                steps: lastChunk.metadata?.steps!,
+                steps: lastChunk.metadata.steps,
               },
             });
           },
+          abort(e) {
+            console.error("Server stream processing error:", e);
+          },
         }),
       );
     }
-
-    // drain the server stream
-    serverStream.pipeTo(voidStream()).catch((e) => {
-      console.error("Server stream processing error:", e);
-    });
   }
 
   return stream;
@@ -191,16 +173,12 @@ export const getEdgeRuntimeResponse = async (
   options: getEdgeRuntimeResponse.Options,
 ) => {
   const stream = await getEdgeRuntimeStream(options);
-  return new Response(
-    stream
-      .pipeThrough(assistantEncoderStream())
-      .pipeThrough(streamPartEncoderStream()),
-    {
-      headers: {
-        "Content-Type": "text/plain; charset=utf-8",
-      },
+  return new Response(stream.pipeThrough(new DataStreamEncoder()), {
+    headers: {
+      "Content-Type": "text/plain; charset=utf-8",
+      "x-vercel-ai-data-stream": "v1",
     },
-  );
+  });
 };
 
 export const createEdgeRuntimeAPI = (options: CreateEdgeRuntimeAPIOptions) => ({

package/src/runtimes/edge/index.ts

@@ -1,7 +1,5 @@
 export * from "./converters";
 
-export * from "./streams/utils";
-
 export { useEdgeRuntime, type EdgeRuntimeOptions } from "./useEdgeRuntime";
 export { EdgeChatAdapter } from "./EdgeChatAdapter";
 export type { EdgeRuntimeRequestOptions } from "./EdgeRuntimeRequestOptions";