From f16ea43e9a98814ef6c05919c3306dcd636ffeb8 Mon Sep 17 00:00:00 2001
From: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
Date: Thu, 3 Jul 2025 12:58:17 +1000
Subject: [PATCH] feat(ui): enable RTK Query's refetchOnReconnect

---
 .../frontend/web/src/app/components/GlobalHookIsolator.tsx   | 5 +++++
 invokeai/frontend/web/src/services/api/index.ts              | 1 +
 2 files changed, 6 insertions(+)

diff --git a/invokeai/frontend/web/src/app/components/GlobalHookIsolator.tsx b/invokeai/frontend/web/src/app/components/GlobalHookIsolator.tsx
index 77d7eb2b9e..b017fc3893 100644
--- a/invokeai/frontend/web/src/app/components/GlobalHookIsolator.tsx
+++ b/invokeai/frontend/web/src/app/components/GlobalHookIsolator.tsx
@@ -1,4 +1,5 @@
 import { useGlobalModifiersInit } from '@invoke-ai/ui-library';
+import { setupListeners } from '@reduxjs/toolkit/query';
 import type { StudioInitAction } from 'app/hooks/useStudioInitAction';
 import { useStudioInitAction } from 'app/hooks/useStudioInitAction';
 import { useSyncQueueStatus } from 'app/hooks/useSyncQueueStatus';
@@ -62,6 +63,10 @@ export const GlobalHookIsolator = memo(
       dispatch(appStarted());
     }, [dispatch]);
 
+    useEffect(() => {
+      return setupListeners(dispatch);
+    }, [dispatch]);
+
     useStudioInitAction(studioInitAction);
     useStarterModelsToast();
     useSyncQueueStatus();
diff --git a/invokeai/frontend/web/src/services/api/index.ts b/invokeai/frontend/web/src/services/api/index.ts
index 02c4d77b6a..87070cd4e6 100644
--- a/invokeai/frontend/web/src/services/api/index.ts
+++ b/invokeai/frontend/web/src/services/api/index.ts
@@ -112,6 +112,7 @@ export const api = customCreateApi({
   endpoints: () => ({}),
   invalidationBehavior: 'immediately',
   serializeQueryArgs: stableHash,
+  refetchOnReconnect: true,
 });
 
 function getCircularReplacer() {