Merge pull request #11 from langchain-ai/dqbd/ui-optional

fix: mark ui state key optional

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 }) =>

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;