From 39fb0faff7408e7127da1ab9fef48f066673afdc Mon Sep 17 00:00:00 2001
From: Tat Dat Duong <david@duong.cz>
Date: Tue, 4 Mar 2025 19:29:31 +0100
Subject: [PATCH] fix: mark ui state key optional

---
 src/components/thread/messages/ai.tsx | 2 +-
 src/providers/Stream.tsx              | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/components/thread/messages/ai.tsx b/src/components/thread/messages/ai.tsx
index 7680460..70579d4 100644
--- a/src/components/thread/messages/ai.tsx
+++ b/src/components/thread/messages/ai.tsx
@@ -17,7 +17,7 @@ function CustomComponent({
   const meta = thread.getMessagesMetadata(message);
   const seenState = meta?.firstSeenState;
   const customComponent = seenState?.values.ui
-    .slice()
+    ?.slice()
     .reverse()
     .find(
       ({ additional_kwargs }) =>
diff --git a/src/providers/Stream.tsx b/src/providers/Stream.tsx
index eac16b5..7b5093d 100644
--- a/src/providers/Stream.tsx
+++ b/src/providers/Stream.tsx
@@ -14,7 +14,7 @@ import { ArrowRight } from "lucide-react";
 import { PasswordInput } from "@/components/ui/password-input";
 
 const useTypedStream = useStream<
-  { messages: Message[]; ui: UIMessage[] },
+  { messages: Message[]; ui?: UIMessage[] },
   {
     UpdateType: {
       messages?: Message[] | Message | string;