优化聊天页面代码，刷新页面之后自动加载当前对话的 role_id 和 model_id

api/handler/chatimpl/chat_handler.go

@@ -543,9 +543,9 @@ func (h *ChatHandler) saveChatHistory(
 	}
 	historyUserMsg.CreatedAt = promptCreatedAt
 	historyUserMsg.UpdatedAt = promptCreatedAt
-	res := h.DB.Save(&historyUserMsg)
+	err = h.DB.Save(&historyUserMsg).Error
-	if res.Error != nil {
+	if err != nil {
-		logger.Error("failed to save prompt history message: ", res.Error)
+		logger.Error("failed to save prompt history message: ", err)
 	}
 
 	// for reply
@@ -565,9 +565,9 @@ func (h *ChatHandler) saveChatHistory(
 	}
 	historyReplyMsg.CreatedAt = replyCreatedAt
 	historyReplyMsg.UpdatedAt = replyCreatedAt
-	res = h.DB.Create(&historyReplyMsg)
+	err = h.DB.Create(&historyReplyMsg).Error
-	if res.Error != nil {
+	if err != nil {
-		logger.Error("failed to save reply history message: ", res.Error)
+		logger.Error("failed to save reply history message: ", err)
 	}
 
 	// 更新用户算力
@@ -576,8 +576,8 @@ func (h *ChatHandler) saveChatHistory(
 	}
 	// 保存当前会话
 	var chatItem model.ChatItem
-	res = h.DB.Where("chat_id = ?", session.ChatId).First(&chatItem)
+	err = h.DB.Where("chat_id = ?", session.ChatId).First(&chatItem).Error
-	if res.Error != nil {
+	if err != nil {
 		chatItem.ChatId = session.ChatId
 		chatItem.UserId = userVo.Id
 		chatItem.RoleId = role.Id
@@ -588,7 +588,10 @@ func (h *ChatHandler) saveChatHistory(
 			chatItem.Title = prompt
 		}
 		chatItem.Model = req.Model
-		h.DB.Create(&chatItem)
+		err = h.DB.Create(&chatItem).Error
+		if err != nil {
+			logger.Error("failed to save chat item: ", err)
+		}
 	}
 }
 

web/src/views/ChatPlus.vue

@@ -272,10 +272,22 @@ const tools = ref([])
 const toolSelected = ref([])
 const loadHistory = ref(false)
 
+// 初始化角色ID参数
+if (router.currentRoute.value.query.role_id) {
+  roleId.value = parseInt(router.currentRoute.value.query.role_id)
+}
+
 // 初始化 ChatID
 chatId.value = router.currentRoute.value.params.id
 if (!chatId.value) {
   chatId.value = UUID()
+}else { // 查询对话信息
+  httpGet("/api/chat/detail", {chat_id: chatId.value}).then(res => {
+    roleId.value = res.data.role_id
+    modelID.value = res.data.model_id
+  }).catch(e => {
+    console.error("获取对话信息失败："+e.message)
+  })
 }
 
 if (isMobile()) {
@@ -344,24 +356,13 @@ onUnmounted(() => {
 
 // 初始化数据
 const initData = () => {
-  // 检查会话
-  checkSession().then((user) => {
-    loginUser.value = user
-    isLogin.value = true
 
-    // 获取会话列表
-    httpGet("/api/chat/list").then((res) => {
-      if (res.data) {
-        chatList.value = res.data;
-        allChats.value = res.data;
-      }
-      if (router.currentRoute.value.query.role_id) {
-        roleId.value = parseInt(router.currentRoute.value.query.role_id)
-      }
   // 加载模型
   httpGet('/api/model/list').then(res => {
     models.value = res.data
+    if (!modelID.value) {
       modelID.value = models.value[0].id
+    }
     // 加载角色列表
     httpGet(`/api/role/list`,{id:roleId.value}).then((res) => {
       roles.value = res.data;
@@ -369,34 +370,32 @@ const initData = () => {
         roleId.value = roles.value[0]['id']
       }
 
+      // 如果登录状态就创建对话连接
+      checkSession().then((user) => {
+        loginUser.value = user
+        isLogin.value = true
+
         newChat();
+      })
+
     }).catch((e) => {
       ElMessage.error('获取聊天角色失败: ' + e.messages)
     })
   }).catch(e => {
     ElMessage.error("加载模型失败: " + e.message)
   })
+
+  // 获取会话列表
+  httpGet("/api/chat/list").then((res) => {
+    if (res.data) {
+      chatList.value = res.data;
+      allChats.value = res.data;
+    }
   }).catch(() => {
     ElMessage.error("加载会话列表失败！")
   })
-  }).catch(() => {
-    // 加载模型
-    httpGet('/api/model/list',{id:roleId.value}).then(res => {
-      models.value = res.data
-      modelID.value = models.value[0].id
-    }).catch(e => {
-      ElMessage.error("加载模型失败: " + e.message)
-    })
-
-    // 加载角色列表
-    httpGet(`/api/role/list`).then((res) => {
-      roles.value = res.data;
-      roleId.value = roles.value[0]['id'];
-    }).catch((e) => {
-      ElMessage.error('获取聊天角色失败: ' + e.messages)
-    })
-  })
 
+  // 允许在输入框粘贴文件
   inputRef.value.addEventListener('paste', (event) => {
     const items = (event.clipboardData || window.clipboardData).items;
     let fileFound = false;