Update index.tsx

indexnetwork · Jun 25, 2024 · a77d855 · a77d855
1 parent 4fb8d54
commit a77d855
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/web-app/src/components/site/indexes/AskIndexes/index.tsx b/web-app/src/components/site/indexes/AskIndexes/index.tsx
@@ -224,6 +224,23 @@ const AskIndexes: FC<AskIndexesProps> = ({ sources }) => {
       return;
     }
 
+    if (p.channel === "update") {
+      const newMessage: Message = {
+        id: p.data.messageId,
+        role: p.data.payload.role,
+        name: p.data.payload.name,
+        content: p.data.payload.content,
+      };
+      console.log(newMessage);
+      setStreamingMessages((prevMessages: any) => {
+        const updated = [...prevMessages, newMessage];
+        dispatch(setMessages(updated));
+        return updated;
+      });
+
+      return;
+    }
+
     if (p.channel === "chunk") {
       setStreamingMessages((prevMessages: any) => {
         let streamingMessage = prevMessages.find(