From d243c2be5df84a1155ba8283fa4bec7d4b3f6d7a Mon Sep 17 00:00:00 2001
From: Samuel Amoah <samuel.amoah@shopify.com>
Date: Mon, 16 Dec 2024 22:04:21 -0500
Subject: [PATCH] Improve ResponsePickerApi type

---
 .../admin/api/resource-picker/resource-picker.ts       | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/packages/ui-extensions/src/surfaces/admin/api/resource-picker/resource-picker.ts b/packages/ui-extensions/src/surfaces/admin/api/resource-picker/resource-picker.ts
index 971ede13ff..45875c110c 100644
--- a/packages/ui-extensions/src/surfaces/admin/api/resource-picker/resource-picker.ts
+++ b/packages/ui-extensions/src/surfaces/admin/api/resource-picker/resource-picker.ts
@@ -202,11 +202,11 @@ interface Filters {
   query?: string;
 }
 
-export interface ResourcePickerOptions {
+export interface ResourcePickerOptions<Type extends keyof ResourceTypes> {
   /**
    * The type of resource you want to pick.
    */
-  type: 'product' | 'variant' | 'collection';
+  type: Type;
   /**
    *  The action verb appears in the title and as the primary action of the Resource Picker.
    * @defaultValue 'add'
@@ -235,6 +235,6 @@ export interface ResourcePickerOptions {
   selectionIds?: BaseResource[];
 }
 
-export type ResourcePickerApi = (
-  options: ResourcePickerOptions,
-) => Promise<SelectPayload<ResourcePickerOptions['type']> | undefined>;
+export type ResourcePickerApi = <Type extends keyof ResourceTypes>(
+  options: ResourcePickerOptions<Type>,
+) => Promise<SelectPayload<Type> | undefined>;