feat: add delete message api

Author: namchuai

cortex-js/src/infrastructure/controllers/threads.controller.ts

@@ -27,6 +27,7 @@ import { TransformInterceptor } from '../interceptors/transform.interceptor';
 import { ListMessagesResponseDto } from '../dtos/messages/list-message.dto';
 import { CreateMessageDto } from '../dtos/threads/create-message.dto';
 import { UpdateMessageDto } from '../dtos/threads/update-message.dto';
+import DeleteMessageDto from '../dtos/threads/delete-message.dto';
 
 @ApiTags('Threads')
 @Controller('threads')
@@ -173,6 +174,29 @@ export class ThreadsController {
     );
   }
 
+  @ApiResponse({
+    status: 200,
+    description: 'Deletion status.',
+    type: DeleteMessageDto,
+  })
+  @ApiParam({
+    name: 'thread_id',
+    required: true,
+    description: 'The ID of the thread to which this message belongs.',
+  })
+  @ApiParam({
+    name: 'message_id',
+    required: true,
+    description: 'The ID of the message to delete.',
+  })
+  @Delete(':thread_id/messages/:message_id')
+  deleteMessage(
+    @Param('thread_id') threadId: string,
+    @Param('message_id') messageId: string,
+  ) {
+    return this.threadsUsecases.deleteMessage(threadId, messageId);
+  }
+
   @ApiResponse({
     status: 200,
     description: 'Ok',

cortex-js/src/infrastructure/dtos/threads/delete-message.dto.ts

@@ -0,0 +1,12 @@
+import { ApiProperty } from '@nestjs/swagger';
+
+export default class DeleteMessageDto {
+  @ApiProperty()
+  id: string;
+
+  @ApiProperty()
+  object: string;
+
+  @ApiProperty()
+  deleted: boolean;
+}

cortex-js/src/usecases/threads/threads.usecases.ts

@@ -11,6 +11,7 @@ import { ulid } from 'ulid';
 import { ContentType, MessageStatus } from '@/domain/models/message.interface';
 import { UpdateMessageDto } from '@/infrastructure/dtos/threads/update-message.dto';
 import { Thread } from '@/domain/models/thread.interface';
+import DeleteMessageDto from '@/infrastructure/dtos/threads/delete-message.dto';
 
 @Injectable()
 export class ThreadsUsecases {
@@ -145,4 +146,26 @@ export class ThreadsUsecases {
   remove(id: string) {
     this.threadRepository.delete(id);
   }
+
+  async deleteMessage(
+    _threadId: string,
+    messageId: string,
+  ): Promise<DeleteMessageDto> {
+    // we still allow user to delete message even if the thread is not there
+    const message = await this.messageRepository.findOne({
+      where: {
+        id: messageId,
+      },
+    });
+    if (!message) {
+      throw new NotFoundException(`Message with id ${messageId} not found`);
+    }
+    await this.messageRepository.delete(messageId);
+
+    return {
+      id: messageId,
+      object: 'thread.message.deleted',
+      deleted: true,
+    };
+  }
 }