🐛 fix: fix a bug that export a session without messages

lobehub · Nov 30, 2023 · 8e84f35 · 8e84f35
1 parent 5d2b750
commit 8e84f35
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/src/database/models/message.ts b/src/database/models/message.ts
@@ -163,6 +163,10 @@ class _MessageModel extends BaseModel {
     return count === 0;
   }
 
+  async queryBySessionId(sessionId: string) {
+    return this.table.where('sessionId').equals(sessionId).toArray();
+  }
+
   private mapChatMessageToDBMessage(message: ChatMessage): DB_Message {
     const { extra, ...messageData } = message;
 

diff --git a/src/services/config.ts b/src/services/config.ts
@@ -118,7 +118,7 @@ class ConfigService {
     const session = this.getSession(id);
     if (!session) return;
 
-    const messages = await messageService.getMessages(id);
+    const messages = await messageService.getAllMessagesInSession(id);
     const topics = await topicService.getTopics({ sessionId: id });
 
     const config = createConfigFile('singleSession', { messages, sessions: [session], topics });

diff --git a/src/services/message.ts b/src/services/message.ts
@@ -32,6 +32,10 @@ export class MessageService {
     return MessageModel.delete(id);
   }
 
+  async getAllMessagesInSession(sessionId: string) {
+    return MessageModel.queryBySessionId(sessionId);
+  }
+
   async updateMessageContent(id: string, content: string) {
     return MessageModel.update(id, { content });
   }