Add useSuspenseFragment to hooks

apollographql · Feb 10, 2025 · 9b0fbaa · 9b0fbaa
1 parent c01a77b
commit 9b0fbaa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/packages/client-react-streaming/src/DataTransportAbstraction/hooks.ts b/packages/client-react-streaming/src/DataTransportAbstraction/hooks.ts
@@ -33,6 +33,9 @@ export const hookWrappers: HookWrappers = {
       return orig_useQueryRefHandlers(useWrapTransportedQueryRef(queryRef));
     }, []);
   },
+  useSuspenseFragment(orig_useSuspenseFragment) {
+    return wrap(orig_useSuspenseFragment, ["data"]);
+  },
 };
 
 function wrap<T extends (...args: any[]) => any>(