Use mutate()

dust-tt · Sep 29, 2023 · b12941f · b12941f
1 parent 8a90c05
commit b12941f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/front/hooks/useEventSource.ts b/front/hooks/useEventSource.ts
@@ -35,10 +35,10 @@ export function useEventSource(
     };
 
     es.onerror = (event) => {
-      console.error('useEventSource.onerror()', event)
+      console.error("useEventSource.onerror()", event);
       errorCount.current += 1;
       if (errorCount.current >= 3) {
-        console.log('too many errors, not reconnecting..')
+        console.log("too many errors, not reconnecting..");
         setIsError(new Error("Too many errors, closing connection."));
         es.close();
         return;

diff --git a/front/pages/api/v1/w/[wId]/assistant/conversations/[cId]/events.ts b/front/pages/api/v1/w/[wId]/assistant/conversations/[cId]/events.ts
@@ -56,7 +56,7 @@ async function handler(
       }
       res.write("data: done\n\n");
       // @ts-expect-error - We need it for streaming but it does not exists in the types.
-      res.flush();      
+      res.flush();
 
       res.end();
       return;

diff --git a/front/pages/w/[wId]/assistant/[cId]/index.tsx b/front/pages/w/[wId]/assistant/[cId]/index.tsx
@@ -1,5 +1,6 @@
 import { GetServerSideProps, InferGetServerSidePropsType } from "next";
 import { useRouter } from "next/router";
+import { mutate } from "swr";
 
 import Conversation from "@app/components/assistant/conversation/Conversation";
 import { ConversationTitle } from "@app/components/assistant/conversation/ConversationTitle";
@@ -81,6 +82,9 @@ export default function AssistantConversation({
       window.alert(`Error creating message: ${data.error.message}`);
       return;
     }
+    await mutate(
+      `/api/w/${owner.sId}/assistant/conversations/${conversationId}/messages`
+    );
   };
 
   const handdleDeleteConversation = async () => {