diff --git a/api/src/nlp/lib/BaseNlpHelper.ts b/api/src/nlp/lib/BaseNlpHelper.ts index 3c1a39f7..e6c2a855 100644 --- a/api/src/nlp/lib/BaseNlpHelper.ts +++ b/api/src/nlp/lib/BaseNlpHelper.ts @@ -20,9 +20,17 @@ import { v4 as uuidv4 } from 'uuid'; import { LoggerService } from '@/logger/logger.service'; -import { NlpEntity, NlpEntityFull } from '@/nlp/schemas/nlp-entity.schema'; +import { + NlpEntity, + NlpEntityDocument, + NlpEntityFull, +} from '@/nlp/schemas/nlp-entity.schema'; import { NlpSample, NlpSampleFull } from '@/nlp/schemas/nlp-sample.schema'; -import { NlpValue, NlpValueFull } from '@/nlp/schemas/nlp-value.schema'; +import { + NlpValue, + NlpValueDocument, + NlpValueFull, +} from '@/nlp/schemas/nlp-value.schema'; import { Settings } from '@/setting/schemas/types'; import { Nlp } from './types'; @@ -68,7 +76,7 @@ export default abstract class BaseNlpHelper { * @param entity - The entity to add * @returns The added entity otherwise an error */ - addEntity(_entity: NlpEntity): Promise { + addEntity(_entity: NlpEntityDocument): Promise { return new Promise((resolve, _reject) => { return resolve(uuidv4()); }); @@ -103,7 +111,7 @@ export default abstract class BaseNlpHelper { * * @returns The added value otherwise it should throw an error */ - addValue(_value: NlpValue): Promise { + addValue(_value: NlpValueDocument): Promise { return new Promise((resolve, _reject) => { return resolve(uuidv4()); }); diff --git a/api/src/nlp/repositories/nlp-entity.repository.ts b/api/src/nlp/repositories/nlp-entity.repository.ts index 7afbb716..bcd75fbe 100644 --- a/api/src/nlp/repositories/nlp-entity.repository.ts +++ b/api/src/nlp/repositories/nlp-entity.repository.ts @@ -9,7 +9,7 @@ import { Injectable } from '@nestjs/common'; import { EventEmitter2 } from '@nestjs/event-emitter'; import { InjectModel } from '@nestjs/mongoose'; -import { Document, Model, Query, TFilterQuery, Types } from 'mongoose'; +import { Document, Model, Query, TFilterQuery } from 'mongoose'; import { BaseRepository, DeleteResult } from '@/utils/generics/base-repository'; @@ -18,6 +18,7 @@ import { NlpValueRepository } from './nlp-value.repository'; import { NLP_ENTITY_POPULATE, NlpEntity, + NlpEntityDocument, NlpEntityFull, NlpEntityPopulate, } from '../schemas/nlp-entity.schema'; @@ -44,11 +45,8 @@ export class NlpEntityRepository extends BaseRepository< * * @param created - The newly created NLP entity document. */ - async postCreate( - _created: Document & - NlpEntity & { _id: Types.ObjectId }, - ): Promise { - if (!_created.builtin) { + async postCreate(_created: NlpEntityDocument): Promise { + if (!_created) { // Bypass builtin entities (probably fixtures) this.eventEmitter.emit('hook:nlpEntity:create', _created); } diff --git a/api/src/nlp/repositories/nlp-value.repository.ts b/api/src/nlp/repositories/nlp-value.repository.ts index 6727c578..983b7ef2 100644 --- a/api/src/nlp/repositories/nlp-value.repository.ts +++ b/api/src/nlp/repositories/nlp-value.repository.ts @@ -17,6 +17,7 @@ import { NlpSampleEntityRepository } from './nlp-sample-entity.repository'; import { NLP_VALUE_POPULATE, NlpValue, + NlpValueDocument, NlpValueFull, NlpValuePopulate, } from '../schemas/nlp-value.schema'; @@ -40,7 +41,7 @@ export class NlpValueRepository extends BaseRepository< * * @param created - The newly created NLP value document. */ - async postCreate(created: NlpValue): Promise { + async postCreate(created: NlpValueDocument): Promise { if (!created.builtin) { // Bypass builtin entities (probably fixtures) this.eventEmitter.emit('hook:nlpValue:create', created);