fix: handle zombie children (#581)

assistant-ui · Jul 26, 2024 · dbf8764 · dbf8764
1 parent 43fbf8d
commit dbf8764
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/react/src/context/providers/MessageProvider.tsx b/packages/react/src/context/providers/MessageProvider.tsx
@@ -30,7 +30,8 @@ const getMessageState = (
   messageIndex: number,
 ) => {
   const parentId = messages[messageIndex - 1]?.id ?? null;
-  const message = messages[messageIndex]!;
+  const message = messages[messageIndex];
+  if (!message) return null;
 
   const isLast = getIsLast(messages, message);
   const branches = getBranches(message.id);