From 01b10d403a0f3cb720c79ecc6712a1b28ef6ea4a Mon Sep 17 00:00:00 2001
From: Lucas Oliveira <lucas.edu.oli@hotmail.com>
Date: Fri, 5 Sep 2025 17:00:52 -0300
Subject: [PATCH] Create cancelNudges, to remove the query result when sending
 a message

---
 frontend/src/app/api/queries/useGetNudgesQuery.ts | 8 ++++++--
 frontend/src/app/chat/page.tsx                    | 8 ++++----
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/frontend/src/app/api/queries/useGetNudgesQuery.ts b/frontend/src/app/api/queries/useGetNudgesQuery.ts
index 809ca71e..38cc9502 100644
--- a/frontend/src/app/api/queries/useGetNudgesQuery.ts
+++ b/frontend/src/app/api/queries/useGetNudgesQuery.ts
@@ -17,7 +17,11 @@ export const useGetNudgesQuery = (
   options?: Omit<UseQueryOptions, "queryKey" | "queryFn">,
 ) => {
   const queryClient = useQueryClient();
-  console.log(chatId);
+
+  function cancel() {
+    queryClient.removeQueries({ queryKey: ["nudges", chatId] });
+  }
+
   async function getNudges(): Promise<Nudge[]> {
     try {
       const response = await fetch(`/api/nudges${chatId ? `/${chatId}` : ""}`);
@@ -38,5 +42,5 @@ export const useGetNudgesQuery = (
     queryClient,
   );
 
-  return queryResult;
+  return { ...queryResult, cancel };
 };
diff --git a/frontend/src/app/chat/page.tsx b/frontend/src/app/chat/page.tsx
index 34b27647..987786d9 100644
--- a/frontend/src/app/chat/page.tsx
+++ b/frontend/src/app/chat/page.tsx
@@ -619,7 +619,7 @@ function ChatPage() {
     };
   }, [isFilterDropdownOpen]);
 
-  const { data: nudges = [], refetch: refetchNudges } = useGetNudgesQuery(
+  const { data: nudges = [], cancel: cancelNudges } = useGetNudgesQuery(
     previousResponseIds[endpoint],
   );
 
@@ -1275,7 +1275,7 @@ function ChatPage() {
         setMessages((prev) => [...prev, finalMessage]);
         setStreamingMessage(null);
         if (previousResponseIds[endpoint]) {
-          refetchNudges();
+          cancelNudges();
         }
       }
 
@@ -1394,7 +1394,7 @@ function ChatPage() {
           };
           setMessages((prev) => [...prev, assistantMessage]);
           if (result.response_id) {
-            refetchNudges();
+            cancelNudges();
           }
 
           // Store the response ID if present for this endpoint
@@ -1968,7 +1968,7 @@ function ChatPage() {
       {/* Suggestion chips - always show unless streaming */}
       {!streamingMessage && (
         <Nudges
-          nudges={nudges as string[]}
+          nudges={loading ? [] : (nudges as string[])}
           handleSuggestionClick={handleSuggestionClick}
         />
       )}