diff --git a/packages/base-data-service/src/BaseDataService.ts b/packages/base-data-service/src/BaseDataService.ts index 3b45961225..26bcd3d729 100644 --- a/packages/base-data-service/src/BaseDataService.ts +++ b/packages/base-data-service/src/BaseDataService.ts @@ -2,6 +2,8 @@ import { Messenger, ActionConstraint, EventConstraint, + MessengerActions, + MessengerEvents, } from '@metamask/messenger'; import type { StorageServiceGetItemAction, @@ -161,7 +163,13 @@ export class BaseDataService< persistenceConfig, }: { name: ServiceName; - messenger: ServiceMessenger; + messenger: DataServiceActions['type'] extends + | MessengerActions['type'] + | DataServiceAllowedActions['type'] + ? DataServiceEvents['type'] extends MessengerEvents['type'] + ? ServiceMessenger + : never + : never; queryClientConfig?: QueryClientConfig; policyOptions?: CreateServicePolicyOptions; persistenceConfig?: PersistenceConfiguration;