update save message and get messages

Sma1lboy · Oct 26, 2024 · aaea83e · aaea83e
1 parent 1f2ab88
commit aaea83e
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/backend/src/chat/chat.resolver.ts b/backend/src/chat/chat.resolver.ts
@@ -27,6 +27,7 @@ export class ChatResolver {
         if (chunk) {
           await this.chatService.saveMessage(
             input.id,
+            chunk.id,
             chunk.choices[0].delta.content,
           );
           yield chunk;
@@ -38,6 +39,13 @@ export class ChatResolver {
     }
   }
 
+  @Query(() => Message, { nullable: true })
+  async getMessageDetail(
+    @Args('messageId') messageId: string,
+  ): Promise<Message> {
+    return this.chatService.getMessageById(messageId);
+  }
+
   @Query(() => [Message])
   async getChatHistory(@Args('chatId') chatId: string): Promise<Message[]> {
     return this.chatService.getChatHistory(chatId);

diff --git a/backend/src/chat/chat.service.ts b/backend/src/chat/chat.service.ts
@@ -158,6 +158,12 @@ export class ChatService {
     return chat ? chat.messages : [];
   }
 
+  async getMessageById(messageId: string): Promise<Message> {
+    return await this.messageRepository.findOne({
+      where: { id: messageId },
+    });
+  }
+
   async getChatDetails(chatId: string): Promise<Chat> {
     return this.chatRepository.findOne({
       where: { id: chatId },
@@ -208,13 +214,18 @@ export class ChatService {
     return null;
   }
 
-  async saveMessage(chatId: string, messageContent: string): Promise<Message> {
+  async saveMessage(
+    chatId: string,
+    chunkId: string,
+    messageContent: string,
+  ): Promise<Message> {
     // Find the chat instance
     const chat = await this.chatRepository.findOne({ where: { id: chatId } });
     if (!chat) throw new Error('Chat not found');
 
     // Create a new message associated with the chat
     const message = this.messageRepository.create({
+      id: chunkId,
       content: messageContent,
       role: Role.Model,
       chat,