From 44ede4d9111696451496e779a6a126ad2132d749 Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <xerrors@163.com>
Date: Wed, 24 Dec 2025 11:41:03 +0800
Subject: [PATCH] =?UTF-8?q?feat(feedback):=20=E5=AE=9E=E7=8E=B0=E6=B6=88?=
 =?UTF-8?q?=E6=81=AF=E5=8F=8D=E9=A6=88=E7=8A=B6=E6=80=81=E7=9A=84=E9=A2=84?=
 =?UTF-8?q?=E5=8A=A0=E8=BD=BD=E5=92=8C=E5=90=8C=E6=AD=A5=20Fixes=20#394?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

在消息管理中预加载tool_calls和feedbacks数据
前端组件添加反馈状态初始化和消息变化监听
后端接口返回消息时包含当前用户的反馈状态
---
 server/routers/chat_router.py        | 21 +++++++++++++++++++--
 src/storage/conversation/manager.py  |  7 +++++--
 web/src/components/RefsComponent.vue | 21 ++++++++++++++++++++-
 3 files changed, 44 insertions(+), 5 deletions(-)

diff --git a/server/routers/chat_router.py b/server/routers/chat_router.py
index 4c8add0e..5ac3b2de 100644
--- a/server/routers/chat_router.py
+++ b/server/routers/chat_router.py
@@ -873,7 +873,7 @@ async def save_agent_config(agent_id: str, config: dict = Body(...), current_use
 async def get_agent_history(
     agent_id: str, thread_id: str, current_user: User = Depends(get_required_user), db: AsyncSession = Depends(get_db)
 ):
-    """获取智能体历史消息（需要登录）- NEW STORAGE ONLY"""
+    """获取智能体历史消息（需要登录）- 包含用户反馈状态"""
     try:
         # 获取Agent实例验证
         if not agent_manager.get_agent(agent_id):
@@ -883,12 +883,28 @@ async def get_agent_history(
         conv_manager = ConversationManager(db)
         messages = await conv_manager.get_messages_by_thread_id(thread_id)
 
+        # 当前用户ID - 用于过滤反馈
+        current_user_id = str(current_user.id)
+
         # Convert to frontend-compatible format
         history = []
         for msg in messages:
             # Map role to type that frontend expects
             role_type_map = {"user": "human", "assistant": "ai", "tool": "tool", "system": "system"}
 
+            # 查找当前用户的反馈
+            user_feedback = None
+            if msg.feedbacks:
+                for feedback in msg.feedbacks:
+                    if feedback.user_id == current_user_id:
+                        user_feedback = {
+                            "id": feedback.id,
+                            "rating": feedback.rating,
+                            "reason": feedback.reason,
+                            "created_at": feedback.created_at.isoformat() if feedback.created_at else None,
+                        }
+                        break
+
             msg_dict = {
                 "id": msg.id,  # Include message ID for feedback
                 "type": role_type_map.get(msg.role, msg.role),  # human/ai/tool/system
@@ -899,6 +915,7 @@ async def get_agent_history(
                 "extra_metadata": msg.extra_metadata,  # 保留完整的metadata以备前端需要
                 "message_type": msg.message_type,  # 添加消息类型字段
                 "image_content": msg.image_content,  # 添加图片内容字段
+                "feedback": user_feedback,  # 添加当前用户反馈状态
             }
 
             # Add tool calls if present (for AI messages)
@@ -918,7 +935,7 @@ async def get_agent_history(
 
             history.append(msg_dict)
 
-        logger.info(f"Loaded {len(history)} messages from new storage for thread {thread_id}")
+        logger.info(f"Loaded {len(history)} messages with feedback for thread {thread_id}")
         return {"history": history}
 
     except Exception as e:
diff --git a/src/storage/conversation/manager.py b/src/storage/conversation/manager.py
index 61a33d09..d50c96ca 100644
--- a/src/storage/conversation/manager.py
+++ b/src/storage/conversation/manager.py
@@ -241,11 +241,14 @@ class ConversationManager:
             offset: Number of messages to skip
 
         Returns:
-            List of Message objects
+            List of Message objects with preloaded tool_calls and feedbacks
         """
         query = (
             select(Message)
-            .options(selectinload(Message.tool_calls))
+            .options(
+                selectinload(Message.tool_calls),  # Preload tool calls
+                selectinload(Message.feedbacks),   # Preload feedbacks for UI state
+            )
             .filter(Message.conversation_id == conversation_id)
             .order_by(Message.created_at.asc())
         )
diff --git a/web/src/components/RefsComponent.vue b/web/src/components/RefsComponent.vue
index e54f6b82..663905ad 100644
--- a/web/src/components/RefsComponent.vue
+++ b/web/src/components/RefsComponent.vue
@@ -59,7 +59,7 @@
 </template>
 
 <script setup>
-import { ref, computed, reactive, onMounted } from 'vue'
+import { ref, computed, reactive, watch } from 'vue'
 import { useClipboard } from '@vueuse/core'
 import { message } from 'ant-design-vue'
 import {
@@ -94,6 +94,25 @@ const feedbackState = reactive({
   reason: null,
 })
 
+// 初始化反馈状态 - 从 message.feedback 读取历史反馈
+const initFeedbackState = () => {
+  if (msg.value?.feedback) {
+    feedbackState.hasSubmitted = true
+    feedbackState.rating = msg.value.feedback.rating
+    feedbackState.reason = msg.value.feedback.reason
+  } else {
+    feedbackState.hasSubmitted = false
+    feedbackState.rating = null
+    feedbackState.reason = null
+  }
+}
+
+// 监听 message prop 变化 (用于切换对话时更新状态)
+watch(() => props.message, () => {
+  msg.value = props.message
+  initFeedbackState()
+}, { immediate: true })
+
 // Modal state for dislike
 const dislikeModalVisible = ref(false)
 const dislikeReason = ref('')