fix: return type

Hexastack · Jan 23, 2025 · 43d63cf · 43d63cf
1 parent 0243c71
commit 43d63cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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),