Invalidate queries on logout

All-Hands-AI · Jan 24, 2025 · d88994f · d88994f
1 parent c63ef7c
commit d88994f
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/frontend/src/hooks/mutation/use-logout.ts b/frontend/src/hooks/mutation/use-logout.ts
@@ -1,7 +1,13 @@
-import { useMutation } from "@tanstack/react-query";
+import { useMutation, useQueryClient } from "@tanstack/react-query";
 import OpenHands from "#/api/open-hands";
 
-export const useLogout = () =>
-  useMutation({
+export const useLogout = () => {
+  const queryClient = useQueryClient();
+
+  return useMutation({
     mutationFn: OpenHands.logout,
+    onSuccess: () => {
+      queryClient.invalidateQueries();
+    },
   });
+};