Allow for FragmentType

apollographql · Dec 20, 2024 · bbeb330 · bbeb330
1 parent b727b1d
commit bbeb330
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/react/hooks/useSuspenseFragment.ts b/src/react/hooks/useSuspenseFragment.ts
@@ -1,6 +1,5 @@
 import type {
   ApolloClient,
-  MaybeMasked,
   OperationVariables,
   Reference,
   StoreObject,
@@ -13,6 +12,7 @@ import React, { useMemo } from "rehackt";
 import type { FragmentKey } from "../internal/cache/types.js";
 import { __use } from "./internal/__use.js";
 import { wrapHook } from "./internal/index.js";
+import type { FragmentType, MaybeMasked } from "../../masking/index.js";
 
 export interface UseSuspenseFragmentOptions<TData, TVars>
   extends Omit<
@@ -23,7 +23,7 @@ export interface UseSuspenseFragmentOptions<TData, TVars>
       Cache.ReadFragmentOptions<TData, TVars>,
       "id" | "variables" | "returnPartialData"
     > {
-  from: StoreObject | Reference | string | null;
+  from: StoreObject | Reference | FragmentType<NoInfer<TData>> | string | null;
   // Override this field to make it optional (default: true).
   optimistic?: boolean;
   /**