From cabf2d5fa44a7bf52c5745450cda5bb77b0b127c Mon Sep 17 00:00:00 2001
From: duanfuxiang <duanfuxiang@haizhi.com>
Date: Wed, 12 Mar 2025 21:38:56 +0800
Subject: [PATCH] update vector db, add new field.

---
 src/database/modules/conversation/conversation-manager.ts  | 4 ++++
 .../modules/conversation/conversation-repository.ts        | 7 ++++---
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/database/modules/conversation/conversation-manager.ts b/src/database/modules/conversation/conversation-manager.ts
index 4270b8b..cdb8930 100644
--- a/src/database/modules/conversation/conversation-manager.ts
+++ b/src/database/modules/conversation/conversation-manager.ts
@@ -108,6 +108,7 @@ export class ConversationManager {
 			const userMessage: ChatUserMessage = message
 			return {
 				...base,
+				apply_status: message.applyStatus,
 				content: userMessage.content ? JSON.stringify(userMessage.content) : null,
 				promptContent: userMessage.promptContent
 					? typeof userMessage.promptContent === 'string'
@@ -123,6 +124,7 @@ export class ConversationManager {
 			const assistantMessage: ChatAssistantMessage = message
 			return {
 				...base,
+				apply_status: message.applyStatus,
 				content: assistantMessage.content,
 				reasoningContent: assistantMessage.reasoningContent,
 				metadata: assistantMessage.metadata ? JSON.stringify(assistantMessage.metadata) : null,
@@ -136,6 +138,7 @@ export class ConversationManager {
 			return {
 				id: message.id,
 				role: 'user',
+				applyStatus: message.apply_status,
 				content: message.content ? JSON.parse(message.content) : null,
 				promptContent: message.prompt_content
 					? message.prompt_content.startsWith('{')
@@ -154,6 +157,7 @@ export class ConversationManager {
 		} else {
 			return {
 				id: message.id,
+				applyStatus: message.apply_status,
 				role: 'assistant',
 				content: message.content || '',
 				reasoningContent: message.reasoning_content || '',
diff --git a/src/database/modules/conversation/conversation-repository.ts b/src/database/modules/conversation/conversation-repository.ts
index 4106ec8..8e820ec 100644
--- a/src/database/modules/conversation/conversation-repository.ts
+++ b/src/database/modules/conversation/conversation-repository.ts
@@ -35,15 +35,16 @@ export class ConversationRepository {
   async createMessage(message: InsertMessage): Promise<SelectMessage> {
 		const result = await this.db.query<SelectMessage>(
       `INSERT INTO messages (
-        id, conversation_id, role, content, reasoning_content,
+        id, conversation_id, apply_status, role, content, reasoning_content,
         prompt_content, metadata, mentionables, 
         similarity_search_results, created_at
       )
-      VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
+      VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11)
       RETURNING *`,
       [
         message.id,
         message.conversationId,
+        message.apply_status,
         message.role,
         message.content,
         message.reasoningContent,
@@ -53,7 +54,7 @@ export class ConversationRepository {
         message.similaritySearchResults,
         message.createdAt || new Date()
       ]
-    )
+		)
     return result.rows[0]
   }