From 7b275c40f4eeaa9385703e518baa64f3df53a496 Mon Sep 17 00:00:00 2001
From: Mohamed Marrouchi <marrouchi.mohamed@gmail.com>
Date: Tue, 22 Oct 2024 15:10:39 +0100
Subject: [PATCH] fix: ts issue

---
 api/src/chat/schemas/message.schema.ts     | 2 +-
 api/src/nlp/services/nlp-sample.service.ts | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/api/src/chat/schemas/message.schema.ts b/api/src/chat/schemas/message.schema.ts
index 19e38cf6..302db508 100644
--- a/api/src/chat/schemas/message.schema.ts
+++ b/api/src/chat/schemas/message.schema.ts
@@ -6,7 +6,7 @@
  * 2. All derivative works must include clear attribution to the original creator and software, Hexastack and Hexabot, in a prominent location (e.g., in the software's "About" section, documentation, and README file).
  */
 
-import { Prop, Schema, SchemaFactory, ModelDefinition } from '@nestjs/mongoose';
+import { ModelDefinition, Prop, Schema, SchemaFactory } from '@nestjs/mongoose';
 import { Transform, Type } from 'class-transformer';
 import { Schema as MongooseSchema } from 'mongoose';
 
diff --git a/api/src/nlp/services/nlp-sample.service.ts b/api/src/nlp/services/nlp-sample.service.ts
index 8d5366f8..d66f8942 100644
--- a/api/src/nlp/services/nlp-sample.service.ts
+++ b/api/src/nlp/services/nlp-sample.service.ts
@@ -8,8 +8,9 @@
 
 import { Injectable } from '@nestjs/common';
 import { OnEvent } from '@nestjs/event-emitter';
+import { THydratedDocument } from 'mongoose';
 
-import { AnyMessage } from '@/chat/schemas/types/message';
+import { Message } from '@/chat/schemas/message.schema';
 import { Language } from '@/i18n/schemas/language.schema';
 import { LanguageService } from '@/i18n/services/language.service';
 import { LoggerService } from '@/logger/logger.service';
@@ -67,7 +68,7 @@ export class NlpSampleService extends BaseService<
   }
 
   @OnEvent('hook:message:preCreate')
-  async handleNewMessage(doc: AnyMessage) {
+  async handleNewMessage(doc: THydratedDocument<Message>) {
     // If message is sent by the user then add it as an inbox sample
     if (
       'sender' in doc &&