From 43d63cfa82e5293418c848689302300891ddbeef Mon Sep 17 00:00:00 2001
From: Mohamed Marrouchi <marrouchi.mohamed@gmail.com>
Date: Thu, 23 Jan 2025 18:21:35 +0100
Subject: [PATCH] fix: return type

---
 api/src/helper/lib/base-nlp-helper.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/api/src/helper/lib/base-nlp-helper.ts b/api/src/helper/lib/base-nlp-helper.ts
index c88ed6cf..5b0e8310 100644
--- a/api/src/helper/lib/base-nlp-helper.ts
+++ b/api/src/helper/lib/base-nlp-helper.ts
@@ -123,7 +123,7 @@ export default abstract class BaseNlpHelper<
   async format(
     samples: NlpSampleFull[],
     entities: NlpEntityFull[],
-  ): Promise<Record<string, any>[]> {
+  ): Promise<Record<string, any>[] | Record<string, any>> {
     const entityMap = NlpEntity.getEntityMap(entities);
     const valueMap = NlpValue.getValueMap(
       NlpValue.getValuesFromEntities(entities),