fix: init message is still needed for reload (#90)


lib/hooks/useVisionAgent.ts

@@ -44,6 +44,7 @@ const useVisionAgent = (chat: ChatWithMessages) => {
     onError: err => {
       err && toast.error(err.message);
     },
+    initialMessages: convertDBMessageToAPIMessage(dbMessages),
   });
 
   /**
@@ -51,16 +52,11 @@ const useVisionAgent = (chat: ChatWithMessages) => {
    */
   const once = useRef(true);
   useEffect(() => {
-    if (
-      !isLoading &&
-      messages.length === 1 &&
-      messages[0].role === 'user' &&
-      once.current
-    ) {
+    if (!isLoading && !latestDbMessage.result && once.current) {
       once.current = false;
       reload();
     }
-  }, [isLoading, messages, reload]);
+  }, [isLoading, latestDbMessage.result, messages, reload]);
 
   return {
     messages: messages as MessageUI[],