@assistant-ui/react 0.8.0 → 0.8.2

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/runtimes/remote-thread-list/adapter/cloud.tsx

@@ -16,6 +16,7 @@ import { InMemoryThreadListAdapter } from "./in-memory";
 type ThreadData = {
   externalId: string;
 };
+
 type CloudThreadListAdapterOptions = {
   cloud?: AssistantCloud | undefined;
 
@@ -23,19 +24,9 @@ type CloudThreadListAdapterOptions = {
   delete?(threadId: string): Promise<void>;
 };
 
-const getEnvVariable = (name: string): string | undefined => {
-  if (
-    typeof process !== "undefined" &&
-    process.env &&
-    typeof process.env[name] === "string"
-  ) {
-    return process.env[name];
-  }
-
-  return undefined;
-};
-
-const baseUrl = getEnvVariable("NEXT_PUBLIC_ASSISTANT_BASE_URL");
+const baseUrl =
+  typeof process !== "undefined" &&
+  process?.env?.["NEXT_PUBLIC_ASSISTANT_BASE_URL"];
 const autoCloud = baseUrl
   ? new AssistantCloud({ baseUrl, anonymous: true })
   : undefined;

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,