优化聊天页面代码，刷新页面之后自动加载当前对话的 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,59 +356,46 @@ onUnmounted(() => {
 
 // 初始化数据
 const initData = () => {
-  // 检查会话
-  checkSession().then((user) => {
-    loginUser.value = user
-    isLogin.value = true
 
-    // 获取会话列表
+  // 加载模型
-    httpGet("/api/chat/list").then((res) => {
+  httpGet('/api/model/list').then(res => {
-      if (res.data) {
+    models.value = res.data
-        chatList.value = res.data;
+    if (!modelID.value) {
-        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
-        modelID.value = models.value[0].id
-        // 加载角色列表
-        httpGet(`/api/role/list`,{id:roleId.value}).then((res) => {
-          roles.value = res.data;
-          if (!roleId.value) {
-            roleId.value = roles.value[0]['id']
-          }
-
-          newChat();
-        }).catch((e) => {
-          ElMessage.error('获取聊天角色失败: ' + e.messages)
-        })
-      }).catch(e => {
-        ElMessage.error("加载模型失败: " + e.message)
-      })
-    }).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) => {
+    httpGet(`/api/role/list`,{id:roleId.value}).then((res) => {
       roles.value = res.data;
-      roleId.value = roles.value[0]['id'];
+      if (!roleId.value) {
+        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("加载会话列表失败！")
+  })
+
+  // 允许在输入框粘贴文件
   inputRef.value.addEventListener('paste', (event) => {
     const items = (event.clipboardData || window.clipboardData).items;
     let fileFound = false;