@assistant-ui/react 0.7.70 → 0.7.72

package/dist/types/ContentPartComponentTypes.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/types/ContentPartComponentTypes.tsx"],"sourcesContent":["import type { ComponentType } from \"react\";\nimport type {\n  ContentPartStatus,\n  FileContentPart,\n  ImageContentPart,\n  TextContentPart,\n  ToolCallContentPart,\n  UIContentPart,\n  Unstable_AudioContentPart,\n} from \"./AssistantTypes\";\nimport { ContentPartState } from \"../api/ContentPartRuntime\";\nimport { ReadonlyJSONObject } from \"../utils/json/json-value\";\n\nexport type EmptyContentPartProps = {\n  status: ContentPartStatus;\n};\nexport type EmptyContentPartComponent = ComponentType<EmptyContentPartProps>;\n\nexport type TextContentPartProps = ContentPartState & TextContentPart;\nexport type TextContentPartComponent = ComponentType<TextContentPartProps>;\n\nexport type ImageContentPartProps = ContentPartState & ImageContentPart;\nexport type ImageContentPartComponent = ComponentType<ImageContentPartProps>;\n\nexport type FileContentPartProps = ContentPartState & FileContentPart;\nexport type FileContentPartComponent = ComponentType<FileContentPartProps>;\n\nexport type Unstable_AudioContentPartProps = ContentPartState &\n  Unstable_AudioContentPart;\nexport type Unstable_AudioContentPartComponent =\n  ComponentType<Unstable_AudioContentPartProps>;\n\nexport type UIContentPartProps = ContentPartState & UIContentPart;\nexport type UIContentPartComponent = ComponentType<UIContentPartProps>;\n\nexport type ToolCallContentPartProps<\n  TArgs extends ReadonlyJSONObject = any,\n  TResult = unknown,\n> = ContentPartState &\n  ToolCallContentPart<TArgs, TResult> & {\n    addResult: (result: any) => void;\n  };\n\nexport type ToolCallContentPartComponent<\n  TArgs extends ReadonlyJSONObject = any,\n  TResult = any,\n> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;\n"],"mappings":";;;;;;;;;;;;;;;;AAAA;AAAA;","names":[]}
+{"version":3,"sources":["../../src/types/ContentPartComponentTypes.tsx"],"sourcesContent":["import type { ComponentType } from \"react\";\nimport type {\n  ContentPartStatus,\n  FileContentPart,\n  ImageContentPart,\n  ReasoningContentPart,\n  TextContentPart,\n  ToolCallContentPart,\n  UIContentPart,\n  Unstable_AudioContentPart,\n} from \"./AssistantTypes\";\nimport { ContentPartState } from \"../api/ContentPartRuntime\";\nimport { ReadonlyJSONObject } from \"../utils/json/json-value\";\n\nexport type EmptyContentPartProps = {\n  status: ContentPartStatus;\n};\nexport type EmptyContentPartComponent = ComponentType<EmptyContentPartProps>;\n\nexport type TextContentPartProps = ContentPartState & TextContentPart;\nexport type TextContentPartComponent = ComponentType<TextContentPartProps>;\n\nexport type ReasoningContentPartProps = ContentPartState & ReasoningContentPart;\nexport type ReasoningContentPartComponent = ComponentType<ReasoningContentPartProps>;\n\nexport type ImageContentPartProps = ContentPartState & ImageContentPart;\nexport type ImageContentPartComponent = ComponentType<ImageContentPartProps>;\n\nexport type FileContentPartProps = ContentPartState & FileContentPart;\nexport type FileContentPartComponent = ComponentType<FileContentPartProps>;\n\nexport type Unstable_AudioContentPartProps = ContentPartState &\n  Unstable_AudioContentPart;\nexport type Unstable_AudioContentPartComponent =\n  ComponentType<Unstable_AudioContentPartProps>;\n\nexport type UIContentPartProps = ContentPartState & UIContentPart;\nexport type UIContentPartComponent = ComponentType<UIContentPartProps>;\n\nexport type ToolCallContentPartProps<\n  TArgs extends ReadonlyJSONObject = any,\n  TResult = unknown,\n> = ContentPartState &\n  ToolCallContentPart<TArgs, TResult> & {\n    addResult: (result: any) => void;\n  };\n\nexport type ToolCallContentPartComponent<\n  TArgs extends ReadonlyJSONObject = any,\n  TResult = any,\n> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;\n"],"mappings":";;;;;;;;;;;;;;;;AAAA;AAAA;","names":[]}

package/dist/types/index.d.ts

@@ -1,5 +1,5 @@
 export type { Attachment, PendingAttachment, CompleteAttachment, AttachmentStatus, } from "./AttachmentTypes";
-export type { AppendMessage, TextContentPart, ImageContentPart, FileContentPart, Unstable_AudioContentPart, ToolCallContentPart, UIContentPart, MessageStatus, ThreadUserContentPart, ThreadAssistantContentPart, ThreadSystemMessage, ThreadAssistantMessage, ThreadUserMessage, ThreadMessage, CoreUserContentPart, CoreAssistantContentPart, CoreSystemMessage, CoreUserMessage, CoreAssistantMessage, CoreMessage, } from "./AssistantTypes";
-export type { EmptyContentPartProps, EmptyContentPartComponent, TextContentPartProps, TextContentPartComponent, ImageContentPartProps, ImageContentPartComponent, FileContentPartProps, FileContentPartComponent, Unstable_AudioContentPartProps, Unstable_AudioContentPartComponent, UIContentPartProps, UIContentPartComponent, ToolCallContentPartProps, ToolCallContentPartComponent, } from "./ContentPartComponentTypes";
+export type { AppendMessage, TextContentPart, ReasoningContentPart, ImageContentPart, FileContentPart, Unstable_AudioContentPart, ToolCallContentPart, UIContentPart, MessageStatus, ThreadUserContentPart, ThreadAssistantContentPart, ThreadSystemMessage, ThreadAssistantMessage, ThreadUserMessage, ThreadMessage, CoreUserContentPart, CoreAssistantContentPart, CoreSystemMessage, CoreUserMessage, CoreAssistantMessage, CoreMessage, } from "./AssistantTypes";
+export type { EmptyContentPartProps, EmptyContentPartComponent, TextContentPartProps, TextContentPartComponent, ReasoningContentPartProps, ReasoningContentPartComponent, ImageContentPartProps, ImageContentPartComponent, FileContentPartProps, FileContentPartComponent, Unstable_AudioContentPartProps, Unstable_AudioContentPartComponent, UIContentPartProps, UIContentPartComponent, ToolCallContentPartProps, ToolCallContentPartComponent, } from "./ContentPartComponentTypes";
 export type { Unsubscribe } from "./Unsubscribe";
 //# sourceMappingURL=index.d.ts.map

package/dist/types/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,UAAU,EACV,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,GACjB,MAAM,mBAAmB,CAAC;AAE3B,YAAY,EACV,aAAa,EACb,eAAe,EACf,gBAAgB,EAChB,eAAe,EACf,yBAAyB,EACzB,mBAAmB,EACnB,aAAa,EACb,aAAa,EAGb,qBAAqB,EACrB,0BAA0B,EAC1B,mBAAmB,EACnB,sBAAsB,EACtB,iBAAiB,EACjB,aAAa,EAGb,mBAAmB,EACnB,wBAAwB,EACxB,iBAAiB,EACjB,eAAe,EACf,oBAAoB,EACpB,WAAW,GACZ,MAAM,kBAAkB,CAAC;AAE1B,YAAY,EACV,qBAAqB,EACrB,yBAAyB,EACzB,oBAAoB,EACpB,wBAAwB,EACxB,qBAAqB,EACrB,yBAAyB,EACzB,oBAAoB,EACpB,wBAAwB,EACxB,8BAA8B,EAC9B,kCAAkC,EAClC,kBAAkB,EAClB,sBAAsB,EACtB,wBAAwB,EACxB,4BAA4B,GAC7B,MAAM,6BAA6B,CAAC;AAErC,YAAY,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,YAAY,EACV,UAAU,EACV,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,GACjB,MAAM,mBAAmB,CAAC;AAE3B,YAAY,EACV,aAAa,EACb,eAAe,EACf,oBAAoB,EACpB,gBAAgB,EAChB,eAAe,EACf,yBAAyB,EACzB,mBAAmB,EACnB,aAAa,EACb,aAAa,EAGb,qBAAqB,EACrB,0BAA0B,EAC1B,mBAAmB,EACnB,sBAAsB,EACtB,iBAAiB,EACjB,aAAa,EAGb,mBAAmB,EACnB,wBAAwB,EACxB,iBAAiB,EACjB,eAAe,EACf,oBAAoB,EACpB,WAAW,GACZ,MAAM,kBAAkB,CAAC;AAE1B,YAAY,EACV,qBAAqB,EACrB,yBAAyB,EACzB,oBAAoB,EACpB,wBAAwB,EACxB,yBAAyB,EACzB,6BAA6B,EAC7B,qBAAqB,EACrB,yBAAyB,EACzB,oBAAoB,EACpB,wBAAwB,EACxB,8BAA8B,EAC9B,kCAAkC,EAClC,kBAAkB,EAClB,sBAAsB,EACtB,wBAAwB,EACxB,4BAA4B,GAC7B,MAAM,6BAA6B,CAAC;AAErC,YAAY,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC"}

package/dist/types/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/types/index.ts"],"sourcesContent":["export type {\n  Attachment,\n  PendingAttachment,\n  CompleteAttachment,\n  AttachmentStatus,\n} from \"./AttachmentTypes\";\n\nexport type {\n  AppendMessage,\n  TextContentPart,\n  ImageContentPart,\n  FileContentPart,\n  Unstable_AudioContentPart,\n  ToolCallContentPart,\n  UIContentPart,\n  MessageStatus,\n\n  // thread message types\n  ThreadUserContentPart,\n  ThreadAssistantContentPart,\n  ThreadSystemMessage,\n  ThreadAssistantMessage,\n  ThreadUserMessage,\n  ThreadMessage,\n\n  // core message types\n  CoreUserContentPart,\n  CoreAssistantContentPart,\n  CoreSystemMessage,\n  CoreUserMessage,\n  CoreAssistantMessage,\n  CoreMessage,\n} from \"./AssistantTypes\";\n\nexport type {\n  EmptyContentPartProps,\n  EmptyContentPartComponent,\n  TextContentPartProps,\n  TextContentPartComponent,\n  ImageContentPartProps,\n  ImageContentPartComponent,\n  FileContentPartProps,\n  FileContentPartComponent,\n  Unstable_AudioContentPartProps,\n  Unstable_AudioContentPartComponent,\n  UIContentPartProps,\n  UIContentPartComponent,\n  ToolCallContentPartProps,\n  ToolCallContentPartComponent,\n} from \"./ContentPartComponentTypes\";\n\nexport type { Unsubscribe } from \"./Unsubscribe\";\n"],"mappings":";;;;;;;;;;;;;;;;AAAA;AAAA;","names":[]}
+{"version":3,"sources":["../../src/types/index.ts"],"sourcesContent":["export type {\n  Attachment,\n  PendingAttachment,\n  CompleteAttachment,\n  AttachmentStatus,\n} from \"./AttachmentTypes\";\n\nexport type {\n  AppendMessage,\n  TextContentPart,\n  ReasoningContentPart,\n  ImageContentPart,\n  FileContentPart,\n  Unstable_AudioContentPart,\n  ToolCallContentPart,\n  UIContentPart,\n  MessageStatus,\n\n  // thread message types\n  ThreadUserContentPart,\n  ThreadAssistantContentPart,\n  ThreadSystemMessage,\n  ThreadAssistantMessage,\n  ThreadUserMessage,\n  ThreadMessage,\n\n  // core message types\n  CoreUserContentPart,\n  CoreAssistantContentPart,\n  CoreSystemMessage,\n  CoreUserMessage,\n  CoreAssistantMessage,\n  CoreMessage,\n} from \"./AssistantTypes\";\n\nexport type {\n  EmptyContentPartProps,\n  EmptyContentPartComponent,\n  TextContentPartProps,\n  TextContentPartComponent,\n  ReasoningContentPartProps,\n  ReasoningContentPartComponent,\n  ImageContentPartProps,\n  ImageContentPartComponent,\n  FileContentPartProps,\n  FileContentPartComponent,\n  Unstable_AudioContentPartProps,\n  Unstable_AudioContentPartComponent,\n  UIContentPartProps,\n  UIContentPartComponent,\n  ToolCallContentPartProps,\n  ToolCallContentPartComponent,\n} from \"./ContentPartComponentTypes\";\n\nexport type { Unsubscribe } from \"./Unsubscribe\";\n"],"mappings":";;;;;;;;;;;;;;;;AAAA;AAAA;","names":[]}

package/package.json

@@ -29,7 +29,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.7.70",
+  "version": "0.7.72",
   "license": "MIT",
   "exports": {
     ".": {
@@ -65,29 +65,29 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "@ai-sdk/provider": "^1.0.4",
+    "@ai-sdk/provider": "^1.0.7",
     "@radix-ui/primitive": "^1.1.1",
-    "@radix-ui/react-avatar": "^1.1.2",
+    "@radix-ui/react-avatar": "^1.1.3",
     "@radix-ui/react-compose-refs": "^1.1.1",
     "@radix-ui/react-context": "^1.1.1",
-    "@radix-ui/react-dialog": "^1.1.4",
-    "@radix-ui/react-popover": "^1.1.4",
-    "@radix-ui/react-primitive": "^2.0.1",
-    "@radix-ui/react-slot": "^1.1.1",
-    "@radix-ui/react-tooltip": "^1.1.6",
+    "@radix-ui/react-dialog": "^1.1.6",
+    "@radix-ui/react-popover": "^1.1.6",
+    "@radix-ui/react-primitive": "^2.0.2",
+    "@radix-ui/react-slot": "^1.1.2",
+    "@radix-ui/react-tooltip": "^1.1.8",
     "@radix-ui/react-use-callback-ref": "^1.1.0",
     "@radix-ui/react-use-escape-keydown": "^1.1.0",
     "class-variance-authority": "^0.7.1",
     "classnames": "^2.5.1",
     "json-schema": "^0.4.0",
-    "lucide-react": "^0.473.0",
+    "lucide-react": "^0.475.0",
     "nanoid": "^5.0.9",
     "react-textarea-autosize": "^8.5.7",
     "secure-json-parse": "^3.0.2",
     "zod": "^3.24.1",
     "zod-to-json-schema": "^3.24.1",
     "zustand": "^5.0.3",
-    "assistant-stream": "^0.0.18"
+    "assistant-stream": "^0.0.19"
   },
   "peerDependencies": {
     "@types/react": "*",
@@ -109,17 +109,17 @@
   },
   "devDependencies": {
     "@types/json-schema": "^7.0.15",
-    "@types/node": "^22.10.7",
+    "@types/node": "^22.13.1",
     "autoprefixer": "^10.4.20",
     "eslint": "^9",
-    "eslint-config-next": "15.1.5",
+    "eslint-config-next": "15.1.6",
     "postcss": "^8.5.1",
     "tailwindcss": "^3.4.17",
     "tailwindcss-animate": "^1.0.7",
     "tsx": "^4.19.2",
+    "@assistant-ui/tailwindcss-transformer": "0.1.0",
     "@assistant-ui/tsbuildutils": "^0.0.1",
-    "@assistant-ui/tsconfig": "0.0.0",
-    "@assistant-ui/tailwindcss-transformer": "0.1.0"
+    "@assistant-ui/tsconfig": "0.0.0"
   },
   "publishConfig": {
     "access": "public",

package/src/cloud/auiV0.ts

@@ -15,6 +15,10 @@ type AuiV0MessageContentPart =
       readonly type: "text";
       readonly text: string;
     }
+  | {
+      readonly type: "reasoning";
+      readonly text: string;
+    }
   | {
       readonly type: "tool-call";
       readonly toolCallId: string;
@@ -64,6 +68,13 @@ export const auiV0Encode = (message: ThreadMessage): AuiV0Message => {
           };
         }
 
+        case "reasoning": {
+          return {
+            type: "reasoning",
+            text: part.text,
+          };
+        }
+
         case "tool-call": {
           if (!isJSONValue(part.result)) {
             console.warn(

package/src/primitives/contentPart/useContentPartReasoning.tsx

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

package/src/primitives/index.ts

@@ -12,4 +12,5 @@ export * as ThreadListItemPrimitive from "./threadListItem";
 export { useContentPartDisplay } from "./contentPart/useContentPartDisplay";
 export { useContentPartImage } from "./contentPart/useContentPartImage";
 export { useContentPartText } from "./contentPart/useContentPartText";
+export { useContentPartReasoning } from "./contentPart/useContentPartReasoning";
 export { useThreadViewportAutoScroll } from "./thread/useThreadViewportAutoScroll";

package/src/primitives/message/MessageContent.tsx

@@ -24,6 +24,7 @@ import type {
   ToolCallContentPartProps,
   UIContentPartComponent,
   FileContentPartComponent,
+  ReasoningContentPartComponent,
 } from "../../types/ContentPartComponentTypes";
 import { ContentPartPrimitiveInProgress } from "../contentPart/ContentPartInProgress";
 import { ContentPartStatus } from "../../types/AssistantTypes";
@@ -33,6 +34,7 @@ export namespace MessagePrimitiveContent {
     components?:
       | {
           Empty?: EmptyContentPartComponent | undefined;
+          Reasoning?: ReasoningContentPartComponent | undefined;
           Text?: TextContentPartComponent | undefined;
           Image?: ImageContentPartComponent | undefined;
           File?: FileContentPartComponent | undefined;
@@ -74,6 +76,7 @@ const defaultComponents = {
       </ContentPartPrimitiveInProgress>
     </p>
   ),
+  Reasoning: () => null,
   Image: () => <ContentPartPrimitiveImage />,
   File: () => null,
   Unstable_Audio: () => null,
@@ -87,6 +90,7 @@ type MessageContentPartComponentProps = {
 const MessageContentPartComponent: FC<MessageContentPartComponentProps> = ({
   components: {
     Text = defaultComponents.Text,
+    Reasoning = defaultComponents.Reasoning,
     Image = defaultComponents.Image,
     File = defaultComponents.File,
     Unstable_Audio: Audio = defaultComponents.Unstable_Audio,
@@ -114,6 +118,9 @@ const MessageContentPartComponent: FC<MessageContentPartComponentProps> = ({
     case "text":
       return <Text {...part} />;
 
+    case "reasoning":
+      return <Reasoning {...part} />;
+
     case "image":
       // eslint-disable-next-line jsx-a11y/alt-text
       return <Image {...part} />;
@@ -160,6 +167,7 @@ const MessageContentPart = memo(
   (prev, next) =>
     prev.partIndex === next.partIndex &&
     prev.components?.Text === next.components?.Text &&
+    prev.components?.Reasoning === next.components?.Reasoning &&
     prev.components?.Image === next.components?.Image &&
     prev.components?.File === next.components?.File &&
     prev.components?.Unstable_Audio === next.components?.Unstable_Audio &&

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

@@ -22,13 +22,14 @@ export const toCoreMessage = <T extends boolean = false>(
       return {
         role,
         content: message.content.map((part) => {
+          if (part.type === "reasoning") return null; // reasoning parts are omitted
           if (part.type === "ui") throw new Error("UI parts are not supported");
           if (part.type === "tool-call") {
             const { argsText, ...rest } = part;
             return rest;
           }
           return part;
-        }),
+        }).filter(c => !!c),
         ...(includeId ? { unstable_id: message.id } : {}),
       };
 

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

@@ -147,6 +147,10 @@ export function toLanguageModelMessages(
         for (const part of message.content) {
           const type = part.type;
           switch (type) {
+            case "reasoning": {
+              break; // reasoning parts are omitted
+            }
+
             case "text": {
               splitter.addTextContentPart(part);
               break;

package/src/runtimes/edge/createEdgeRuntimeAPI.ts

@@ -5,7 +5,7 @@ import {
   LanguageModelV1Prompt,
   LanguageModelV1CallOptions,
 } from "@ai-sdk/provider";
-import { CoreMessage, ThreadStep } from "../../types/AssistantTypes";
+import { CoreMessage, ThreadMessage, ThreadStep } from "../../types/AssistantTypes";
 import { assistantEncoderStream } from "./streams/assistantEncoderStream";
 import { EdgeRuntimeRequestOptionsSchema } from "./EdgeRuntimeRequestOptions";
 import { toLanguageModelMessages } from "./converters/toLanguageModelMessages";
@@ -26,7 +26,7 @@ import { streamPartEncoderStream } from "./streams/utils/streamPartEncoderStream
 import { z } from "zod";
 
 type FinishResult = {
-  messages: readonly CoreMessage[];
+  messages: readonly (CoreMessage | ThreadMessage)[];
   metadata: {
     steps: readonly ThreadStep[];
   };
@@ -143,9 +143,18 @@ export const getEdgeRuntimeStream = async ({
             const resultingMessages = [
               ...messages,
               {
+                id: "DEFAULT",
+                createdAt: new Date(),
                 role: "assistant",
                 content: lastChunk.content,
-              } satisfies CoreMessage,
+                status: lastChunk.status,
+                metadata: {
+                  unstable_data: lastChunk.metadata?.unstable_data ?? [],
+                  unstable_annotations: lastChunk.metadata?.unstable_annotations ?? [],
+                  steps: lastChunk.metadata?.steps ?? [],
+                  custom: lastChunk.metadata?.custom ?? {},
+                }
+              } satisfies ThreadMessage,
             ];
             onFinish({
               messages: resultingMessages,

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

@@ -14,6 +14,14 @@ export function assistantEncoderStream() {
     transform(chunk, controller) {
       const chunkType = chunk.type;
       switch (chunkType) {
+        case "reasoning": {
+          controller.enqueue({
+            type: AssistantStreamChunkType.ReasoningDelta,
+            value: chunk.textDelta,
+          });
+          break;
+        }
+
         case "text-delta": {
           if (!chunk.textDelta) break; // ignore empty text deltas
           controller.enqueue({

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

@@ -14,6 +14,12 @@ export function runResultStream() {
     transform(chunk, controller) {
       const chunkType = chunk.type;
       switch (chunkType) {
+        case "reasoning": {
+          message = appendOrUpdateReasoning(message, chunk.textDelta);
+          controller.enqueue(message);
+          break;
+        }
+
         case "text-delta": {
           message = appendOrUpdateText(message, chunk.textDelta);
           controller.enqueue(message);
@@ -105,6 +111,24 @@ export function runResultStream() {
   });
 }
 
+const appendOrUpdateReasoning= (
+  message: CoreChatModelRunResult,
+  textDelta: string,
+) => {
+  let contentParts = message.content ?? [];
+  let contentPart = message.content?.at(-1);
+  if (contentPart?.type !== "reasoning") {
+    contentPart = { type: "reasoning", text: textDelta };
+  } else {
+    contentParts = contentParts.slice(0, -1);
+    contentPart = { type: "reasoning", text: contentPart.text + textDelta };
+  }
+  return {
+    ...message,
+    content: contentParts.concat([contentPart]),
+  };
+};
+
 const appendOrUpdateText = (
   message: CoreChatModelRunResult,
   textDelta: string,
@@ -144,7 +168,7 @@ const appendOrUpdateToolCall = (
       toolCallId,
       toolName,
       argsText: argsTextDelta,
-      args: parsePartialJson(argsTextDelta),
+      args: argsTextDelta ? parsePartialJson(argsTextDelta) : {}, 
     };
     contentParts = [...contentParts, contentPart];
   } else {
@@ -152,7 +176,7 @@ const appendOrUpdateToolCall = (
     contentPart = {
       ...contentPart,
       argsText,
-      args: parsePartialJson(argsText),
+      args: argsTextDelta ? parsePartialJson(argsTextDelta) : {}, 
     };
     contentParts = [
       ...contentParts.slice(0, contentPartIdx),

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

@@ -128,6 +128,7 @@ export function toolResultStream(
 
         // ignore other parts
         case "text-delta":
+        case "reasoning":
         case "tool-call-delta":
         case "tool-result":
         case "step-finish":

package/src/runtimes/external-store/ThreadMessageLike.tsx

@@ -14,7 +14,7 @@ import {
   FileContentPart,
   Unstable_AudioContentPart,
 } from "../../types";
-import { ThreadStep } from "../../types/AssistantTypes";
+import { ReasoningContentPart, ThreadStep } from "../../types/AssistantTypes";
 import {
   ReadonlyJSONObject,
   ReadonlyJSONValue,
@@ -27,6 +27,7 @@ export type ThreadMessageLike = {
     | string
     | readonly (
         | TextContentPart
+        | ReasoningContentPart
         | ImageContentPart
         | FileContentPart
         | Unstable_AudioContentPart
@@ -92,6 +93,7 @@ export const fromThreadMessageLike = (
             const type = part.type;
             switch (type) {
               case "text":
+              case "reasoning":
                 if (part.text.trim().length === 0) return null;
                 return part;
 
@@ -110,7 +112,7 @@ export const fromThreadMessageLike = (
                 return {
                   ...part,
                   toolCallId: part.toolCallId ?? "tool-" + generateId(),
-                  args: part.args ?? parsePartialJson(part.argsText ?? ""),
+                  args: part.args ?? parsePartialJson(part.argsText ?? "{}"),
                   argsText: part.argsText ?? "",
                 };
               }
@@ -148,7 +150,7 @@ export const fromThreadMessageLike = (
               return part;
 
             default: {
-              const unhandledType: "tool-call" = type;
+              const unhandledType: "tool-call" | "reasoning" = type;
               throw new Error(
                 `Unsupported user content part type: ${unhandledType}`,
               );

package/src/runtimes/local/ChatModelAdapter.tsx

@@ -1,5 +1,6 @@
 import type {
   MessageStatus,
+  ReasoningContentPart,
   RunConfig,
   TextContentPart,
   ThreadAssistantContentPart,
@@ -27,7 +28,7 @@ export type ChatModelRunResult = {
 };
 
 export type CoreChatModelRunResult = Omit<ChatModelRunResult, "content"> & {
-  readonly content: readonly (TextContentPart | ToolCallContentPart)[];
+  readonly content: readonly (TextContentPart | ReasoningContentPart | ToolCallContentPart)[];
 };
 
 export type ChatModelRunOptions = {

package/src/styles/tailwindcss/thread.css

@@ -1,11 +1,10 @@
 /* thread */
 .aui-thread-root {
   @apply bg-aui-background box-border h-full;
-  @apply [&>.aui-thread-viewport]:bg-inherit;
 }
 
 .aui-thread-viewport {
-  @apply bg-aui-background flex h-full flex-col items-center overflow-y-scroll scroll-smooth px-4 pt-8;
+  @apply flex h-full flex-col items-center overflow-y-scroll scroll-smooth bg-inherit px-4 pt-8;
 }
 
 .aui-thread-viewport-footer {
@@ -60,8 +59,12 @@
   @apply placeholder:text-aui-muted-foreground max-h-40 flex-grow resize-none border-none bg-transparent px-2 py-4 text-sm outline-none focus:ring-0 disabled:cursor-not-allowed;
 }
 
-.aui-composer-send,
-.aui-composer-cancel,
+.aui-composer-send {
+  @apply my-2.5 size-8 p-2 transition-opacity ease-in;
+}
+.aui-composer-cancel {
+  @apply my-2.5 size-8 p-2 transition-opacity ease-in;
+}
 .aui-composer-attach {
   @apply my-2.5 size-8 p-2 transition-opacity ease-in;
 }
@@ -111,36 +114,35 @@
   @apply max-w-[var(--aui-thread-max-width)] w-full py-4;
 }
 
-:where(.aui-user-message-root) > .aui-user-action-bar-root {
-  @apply col-start-1 row-start-2 mr-3 mt-2.5;
-}
-
-:where(.aui-user-message-root) > .aui-user-message-attachments {
-  @apply col-span-full col-start-1 row-start-1;
-  @apply justify-end;
-}
-
-:where(.aui-user-message-root) > .aui-user-message-content {
-  @apply col-start-2 row-start-2;
+:where(.aui-user-message-root) > .aui-branch-picker-root {
+  @apply col-span-full col-start-1 row-start-3;
+  @apply -mr-1 justify-end;
 }
 
-:where(.aui-user-message-root) > .aui-branch-picker-root {
+.aui-user-branch-picker {
   @apply col-span-full col-start-1 row-start-3;
   @apply -mr-1 justify-end;
 }
 
 .aui-user-message-content {
   @apply bg-aui-muted text-aui-foreground max-w-[calc(var(--aui-thread-max-width)*0.8)] break-words rounded-3xl px-5 py-2.5;
+  
+  @apply col-start-2 row-start-2;
 }
 
 .aui-user-message-attachments {
   @apply flex w-full flex-row gap-3;
+
+  @apply col-span-full col-start-1 row-start-1;
+  @apply justify-end;
 }
 
 /* user action bar */
 
 .aui-user-action-bar-root {
   @apply flex flex-col items-end;
+
+  @apply col-start-1 row-start-2 mr-3 mt-2.5;
 }
 
 /* edit composer */
@@ -168,28 +170,33 @@
   @apply col-start-1 row-span-full row-start-1 mr-4;
 }
 
+.aui-assistant-avatar {
+  @apply col-start-1 row-span-full row-start-1 mr-4;
+}
+
 :where(.aui-assistant-message-root) > .aui-branch-picker-root {
   @apply col-start-2 row-start-2;
   @apply -ml-2 mr-2;
 }
 
-:where(.aui-assistant-message-root) > .aui-assistant-action-bar-root {
-  @apply col-start-3 row-start-2;
-  @apply -ml-1;
-}
-
-:where(.aui-assistant-message-root) > .aui-assistant-message-content {
-  @apply col-span-2 col-start-2 row-start-1 my-1.5;
+.aui-assistant-branch-picker {
+  @apply col-start-2 row-start-2;
+  @apply -ml-2 mr-2;
 }
 
 .aui-assistant-message-content {
   @apply text-aui-foreground max-w-[calc(var(--aui-thread-max-width)*0.8)] break-words leading-7;
+
+  @apply col-span-2 col-start-2 row-start-1 my-1.5;
 }
 
 /* assistant action bar */
 
 .aui-assistant-action-bar-root {
   @apply text-aui-muted-foreground flex gap-1;
+
+  @apply col-start-3 row-start-2;
+  @apply -ml-1;
 }
 
 :where(.aui-assistant-action-bar-root)[data-floating] {

package/src/types/AssistantTypes.ts

@@ -12,6 +12,12 @@ export type TextContentPart = {
   readonly text: string;
 };
 
+export type ReasoningContentPart = {
+  readonly type: "reasoning";
+  readonly text: string;
+};
+
+
 export type ImageContentPart = {
   readonly type: "image";
   readonly image: string;
@@ -101,6 +107,7 @@ export type ThreadUserContentPart =
 
 export type ThreadAssistantContentPart =
   | TextContentPart
+  | ReasoningContentPart
   | ToolCallContentPart
   | UIContentPart;
 

package/src/types/ContentPartComponentTypes.tsx

@@ -3,6 +3,7 @@ import type {
   ContentPartStatus,
   FileContentPart,
   ImageContentPart,
+  ReasoningContentPart,
   TextContentPart,
   ToolCallContentPart,
   UIContentPart,
@@ -19,6 +20,9 @@ export type EmptyContentPartComponent = ComponentType<EmptyContentPartProps>;
 export type TextContentPartProps = ContentPartState & TextContentPart;
 export type TextContentPartComponent = ComponentType<TextContentPartProps>;
 
+export type ReasoningContentPartProps = ContentPartState & ReasoningContentPart;
+export type ReasoningContentPartComponent = ComponentType<ReasoningContentPartProps>;
+
 export type ImageContentPartProps = ContentPartState & ImageContentPart;
 export type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
 

package/src/types/index.ts

@@ -8,6 +8,7 @@ export type {
 export type {
   AppendMessage,
   TextContentPart,
+  ReasoningContentPart,
   ImageContentPart,
   FileContentPart,
   Unstable_AudioContentPart,
@@ -37,6 +38,8 @@ export type {
   EmptyContentPartComponent,
   TextContentPartProps,
   TextContentPartComponent,
+  ReasoningContentPartProps,
+  ReasoningContentPartComponent,
   ImageContentPartProps,
   ImageContentPartComponent,
   FileContentPartProps,