@assistant-ui/react 0.8.1 → 0.8.4

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

@@ -56,6 +56,12 @@ export function runResultStream() {
           break;
         }
 
+        case "source": {
+          message = appendSource(message, chunk);
+          controller.enqueue(message);
+          break;
+        }
+
         case "tool-result": {
           message = appendOrUpdateToolResult(
             message,
@@ -111,7 +117,7 @@ export function runResultStream() {
   });
 }
 
-const appendOrUpdateReasoning= (
+const appendOrUpdateReasoning = (
   message: CoreChatModelRunResult,
   textDelta: string,
 ) => {
@@ -168,7 +174,7 @@ const appendOrUpdateToolCall = (
       toolCallId,
       toolName,
       argsText: argsTextDelta,
-      args: argsTextDelta ? parsePartialJson(argsTextDelta) : {}, 
+      args: argsTextDelta ? parsePartialJson(argsTextDelta) : {},
     };
     contentParts = [...contentParts, contentPart];
   } else {
@@ -176,7 +182,7 @@ const appendOrUpdateToolCall = (
     contentPart = {
       ...contentPart,
       argsText,
-      args: argsTextDelta ? parsePartialJson(argsText) : {}, 
+      args: argsTextDelta ? parsePartialJson(argsText) : {},
     };
     contentParts = [
       ...contentParts.slice(0, contentPartIdx),
@@ -299,6 +305,22 @@ const appendOrUpdateFinish = (
   };
 };
 
+const appendSource = (
+  message: CoreChatModelRunResult,
+  chunk: ToolResultStreamPart & { type: "source" },
+): CoreChatModelRunResult => {
+  return {
+    ...message,
+    content: [
+      ...message.content,
+      {
+        type: "source",
+        ...chunk.source,
+      },
+    ],
+  };
+};
+
 const getStatus = (
   chunk:
     | (LanguageModelV1StreamPart & { type: "finish" })

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

@@ -14,6 +14,15 @@ export type ToolResultStreamPart =
       type: "data";
       data: ReadonlyJSONValue[];
     }
+  | {
+      type: "source";
+      source: {
+        readonly sourceType: "url";
+        readonly id: string;
+        readonly url: string;
+        readonly title?: string;
+      };
+    }
   | {
       type: "tool-result";
       toolCallType: "function";
@@ -129,6 +138,7 @@ export function toolResultStream(
         // ignore other parts
         case "text-delta":
         case "reasoning":
+        case "source":
         case "tool-call-delta":
         case "tool-result":
         case "step-finish":

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

@@ -13,7 +13,11 @@ import {
   FileContentPart,
   Unstable_AudioContentPart,
 } from "../../types";
-import { ReasoningContentPart, ThreadStep } from "../../types/AssistantTypes";
+import {
+  ReasoningContentPart,
+  SourceContentPart,
+  ThreadStep,
+} from "../../types/AssistantTypes";
 import {
   ReadonlyJSONObject,
   ReadonlyJSONValue,
@@ -27,6 +31,7 @@ export type ThreadMessageLike = {
     | readonly (
         | TextContentPart
         | ReasoningContentPart
+        | SourceContentPart
         | ImageContentPart
         | FileContentPart
         | Unstable_AudioContentPart
@@ -95,6 +100,9 @@ export const fromThreadMessageLike = (
                 if (part.text.trim().length === 0) return null;
                 return part;
 
+              case "source":
+                return part;
+
               case "tool-call": {
                 if (part.args) {
                   return {
@@ -144,7 +152,7 @@ export const fromThreadMessageLike = (
               return part;
 
             default: {
-              const unhandledType: "tool-call" | "reasoning" = type;
+              const unhandledType: "tool-call" | "reasoning" | "source" = type;
               throw new Error(
                 `Unsupported user content part type: ${unhandledType}`,
               );

package/src/runtimes/local/ChatModelAdapter.tsx

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

package/src/types/AssistantTypes.ts

@@ -1,4 +1,3 @@
-
 import { CompleteAttachment } from "./AttachmentTypes";
 import {
   ReadonlyJSONObject,
@@ -17,6 +16,14 @@ export type ReasoningContentPart = {
   readonly text: string;
 };
 
+export type SourceContentPart = {
+  readonly type: "source";
+  readonly sourceType: "url";
+  readonly id: string;
+  readonly url: string;
+  readonly title?: string;
+};
+
 export type ImageContentPart = {
   readonly type: "image";
   readonly image: string;
@@ -103,7 +110,8 @@ export type ThreadUserContentPart =
 export type ThreadAssistantContentPart =
   | TextContentPart
   | ReasoningContentPart
-  | ToolCallContentPart;
+  | ToolCallContentPart
+  | SourceContentPart;
 
 type MessageCommonProps = {
   readonly id: string;
@@ -111,6 +119,7 @@ type MessageCommonProps = {
 };
 
 export type ThreadStep = {
+  readonly messageId?: string;
   readonly usage?:
     | {
         readonly promptTokens: number;

package/src/types/ContentPartComponentTypes.tsx

@@ -4,6 +4,7 @@ import type {
   FileContentPart,
   ImageContentPart,
   ReasoningContentPart,
+  SourceContentPart,
   TextContentPart,
   ToolCallContentPart,
   Unstable_AudioContentPart,
@@ -22,6 +23,9 @@ export type ReasoningContentPartProps = ContentPartState & ReasoningContentPart;
 export type ReasoningContentPartComponent =
   ComponentType<ReasoningContentPartProps>;
 
+export type SourceContentPartProps = ContentPartState & SourceContentPart;
+export type SourceContentPartComponent = ComponentType<SourceContentPartProps>;
+
 export type ImageContentPartProps = ContentPartState & ImageContentPart;
 export type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
 

package/src/types/index.ts

@@ -9,6 +9,7 @@ export type {
   AppendMessage,
   TextContentPart,
   ReasoningContentPart,
+  SourceContentPart,
   ImageContentPart,
   FileContentPart,
   Unstable_AudioContentPart,
@@ -40,6 +41,8 @@ export type {
   TextContentPartComponent,
   ReasoningContentPartProps,
   ReasoningContentPartComponent,
+  SourceContentPartProps,
+  SourceContentPartComponent,
   ImageContentPartProps,
   ImageContentPartComponent,
   FileContentPartProps,