fix(usePrismicClient()): accept a type parameter to define which do…

…cuments are supported by the client (e.g. `Content.AllDocumentTypes`) (#197)
prismicio · Aug 31, 2023 · b16e968 · b16e968
1 parent ca3f851
commit b16e968
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/usePrismicClient.ts b/src/usePrismicClient.ts
@@ -11,9 +11,11 @@ import { usePrismicContext } from "./usePrismicContext";
  *
  * @returns The `@prismicio/client` instance provided to `<PrismicProvider>`.
  */
-export const usePrismicClient = (
-	explicitClient?: prismic.Client,
-): prismic.Client => {
+export const usePrismicClient = <
+	TDocuments extends prismic.PrismicDocument = prismic.PrismicDocument,
+>(
+	explicitClient?: prismic.Client<TDocuments>,
+): prismic.Client<TDocuments> => {
 	const context = usePrismicContext();
 
 	const client = explicitClient || context?.client;