Refactor query invalidation to use async/await for improved consistency in useChat, useSignIn, ProcessingPage, and ReportDetailPage

Author: adamrefaey

frontend/src/common/hooks/useChat.ts

@@ -21,8 +21,8 @@ export function useChat(sessionId?: string) {
   // Mutation for creating a new session
   const createSession = useMutation({
     mutationFn: () => bedrockService.createChatSession(),
-    onSuccess: (newSessionId) => {
-      queryClient.invalidateQueries({ queryKey: [QueryKey.Chat, 'sessions'] });
+    onSuccess: async (newSessionId) => {
+      await queryClient.invalidateQueries({ queryKey: [QueryKey.Chat, 'sessions'] });
       return newSessionId;
     },
   });
@@ -33,9 +33,9 @@ export function useChat(sessionId?: string) {
       if (!sessionId) throw new Error('No active session');
       return bedrockService.sendMessage(sessionId, message);
     },
-    onSuccess: () => {
-      queryClient.invalidateQueries({ queryKey: [QueryKey.Chat, sessionId] });
-      queryClient.invalidateQueries({ queryKey: [QueryKey.Chat, 'sessions'] });
+    onSuccess: async () => {
+      await queryClient.invalidateQueries({ queryKey: [QueryKey.Chat, sessionId] });
+      await queryClient.invalidateQueries({ queryKey: [QueryKey.Chat, 'sessions'] });
     },
   });
 

frontend/src/pages/Auth/SignIn/api/useSignIn.ts

@@ -60,9 +60,9 @@ export const useSignIn = () => {
 
   return useMutation({
     mutationFn: signIn,
-    onSuccess: () => {
-      queryClient.invalidateQueries({ queryKey: [QueryKey.UserTokens] });
-      queryClient.invalidateQueries({ queryKey: [QueryKey.Users, 'current'] });
+    onSuccess: async () => {
+      await queryClient.invalidateQueries({ queryKey: [QueryKey.UserTokens] });
+      await queryClient.invalidateQueries({ queryKey: [QueryKey.Users, 'current'] });
     },
   });
 };

frontend/src/pages/Processing/ProcessingPage.tsx

@@ -73,8 +73,8 @@ const ProcessingPage: React.FC = () => {
 
         console.log('Processing complete');
 
-        queryClient.invalidateQueries({ queryKey: [QueryKey.Reports] });
-        queryClient.invalidateQueries({ queryKey: [QueryKey.LatestReports] });
+        await queryClient.invalidateQueries({ queryKey: [QueryKey.Reports] });
+        await queryClient.invalidateQueries({ queryKey: [QueryKey.LatestReports] });
 
         history.push(`/tabs/reports/${reportId}`);
       } else if (data.status === 'failed') {

frontend/src/pages/ReportDetail/ReportDetailPage.tsx

@@ -109,9 +109,9 @@ const ReportDetailPage: React.FC = () => {
         duration: 2000,
       });
 
-      queryClient.invalidateQueries({ queryKey: [QueryKey.Reports] });
-      queryClient.invalidateQueries({ queryKey: [QueryKey.LatestReports] });
-      queryClient.invalidateQueries({ queryKey: [QueryKey.ReportDetail, reportId] });
+      await queryClient.invalidateQueries({ queryKey: [QueryKey.Reports] });
+      await queryClient.invalidateQueries({ queryKey: [QueryKey.LatestReports] });
+      await queryClient.invalidateQueries({ queryKey: [QueryKey.ReportDetail, reportId] });
 
       // Navigate back
       history.push('/tabs/home');
@@ -136,9 +136,9 @@ const ReportDetailPage: React.FC = () => {
       await axios.delete(`${API_URL}/api/reports/${reportId}`, await getAuthConfig());
       setIsProcessing(false);
 
-      queryClient.invalidateQueries({ queryKey: [QueryKey.Reports] });
-      queryClient.invalidateQueries({ queryKey: [QueryKey.LatestReports] });
-      queryClient.invalidateQueries({ queryKey: [QueryKey.ReportDetail, reportId] });
+      await queryClient.invalidateQueries({ queryKey: [QueryKey.Reports] });
+      await queryClient.invalidateQueries({ queryKey: [QueryKey.LatestReports] });
+      await queryClient.invalidateQueries({ queryKey: [QueryKey.ReportDetail, reportId] });
 
       setIsUploadModalOpen(true);
     } catch (error) {