From 1789b8e315b28429c0f987b2d45dc1b2085181dc Mon Sep 17 00:00:00 2001
From: Adrien de Peretti <adrien.deperetti@gmail.com>
Date: Thu, 11 Jul 2024 10:36:07 +0200
Subject: [PATCH] chore: properly infer return type from model config (#8079)

---
 .../core/utils/src/modules-sdk/types/medusa-service.ts     | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/packages/core/utils/src/modules-sdk/types/medusa-service.ts b/packages/core/utils/src/modules-sdk/types/medusa-service.ts
index f32650e434..3b0451765f 100644
--- a/packages/core/utils/src/modules-sdk/types/medusa-service.ts
+++ b/packages/core/utils/src/modules-sdk/types/medusa-service.ts
@@ -3,6 +3,7 @@ import {
   Context,
   FindConfig,
   IDmlEntity,
+  InferEntityType,
   Pluralize,
   RestoreReturn,
   SoftDeleteReturn,
@@ -40,7 +41,11 @@ export type ModelsConfigTemplate = { [key: string]: ModelDTOConfig }
 
 export type ModelConfigurationsToConfigTemplate<T extends ModelEntries> = {
   [Key in keyof T]: {
-    dto: T[Key] extends Constructor<any> ? InstanceType<T[Key]> : any
+    dto: T[Key] extends IDmlEntity<any, any>
+      ? InferEntityType<T[Key]>
+      : T[Key] extends Constructor<any>
+      ? InstanceType<T[Key]>
+      : any
     model: T[Key] extends { model: infer MODEL }
       ? MODEL
       : T[Key] extends IDmlEntity<any, any>