fix: do not require content in ChatModelRunResult (#696)

Yonom · Aug 21, 2024 · 53cf707 · 53cf707
1 parent 16777c9
commit 53cf707
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 7 deletions.
diff --git a/.changeset/warm-vans-grin.md b/.changeset/warm-vans-grin.md
@@ -0,0 +1,5 @@
+---
+"@assistant-ui/react": patch
+---
+
+fix: do not require content in ChatModelRunResult
diff --git a/packages/react/src/runtimes/edge/streams/runResultStream.ts b/packages/react/src/runtimes/edge/streams/runResultStream.ts
@@ -78,8 +78,8 @@ export function runResultStream() {
 }
 
 const appendOrUpdateText = (message: ChatModelRunResult, textDelta: string) => {
-  let contentParts = message.content;
-  let contentPart = message.content.at(-1);
+  let contentParts = message.content ?? [];
+  let contentPart = message.content?.at(-1);
   if (contentPart?.type !== "text") {
     contentPart = { type: "text", text: textDelta };
   } else {
@@ -98,8 +98,8 @@ const appendOrUpdateToolCall = (
   toolName: string,
   argsText: string,
 ) => {
-  let contentParts = message.content;
-  let contentPart = message.content.at(-1);
+  let contentParts = message.content ?? [];
+  let contentPart = message.content?.at(-1);
   if (
     contentPart?.type !== "tool-call" ||
     contentPart.toolCallId !== toolCallId
@@ -132,7 +132,7 @@ const appendOrUpdateToolResult = (
   result: any,
 ) => {
   let found = false;
-  const newContentParts = message.content.map((part) => {
+  const newContentParts = message.content?.map((part) => {
     if (part.type !== "tool-call" || part.toolCallId !== toolCallId)
       return part;
     found = true;
@@ -154,7 +154,7 @@ const appendOrUpdateToolResult = (
 
   return {
     ...message,
-    content: newContentParts,
+    content: newContentParts!,
   };
 };
 

diff --git a/packages/react/src/runtimes/local/ChatModelAdapter.tsx b/packages/react/src/runtimes/local/ChatModelAdapter.tsx
@@ -13,7 +13,7 @@ export type ChatModelRunUpdate = {
 };
 
 export type ChatModelRunResult = {
-  content: ThreadAssistantContentPart[];
+  content?: ThreadAssistantContentPart[];
   status?: MessageStatus;
   metadata?: {
     roundtrips?: ThreadRoundtrip[];