feat: 增加AI回复手动重试

ai_knowledge_base_v2/apps/user-client/src/App.vue

@@ -137,7 +137,8 @@ async function selectSession(sessionId: number, force = false) {
 }
 
 async function send(message: string, complete: (success: boolean) => void) {
-  if (!activeSessionId.value || sending.value) {
+  const sessionId = activeSessionId.value;
+  if (!sessionId || sending.value) {
     complete(false);
     return;
   }
@@ -152,15 +153,35 @@ async function send(message: string, complete: (success: boolean) => void) {
   };
   messages.value.push(userMessage, assistantMessage);
   const assistantIndex = messages.value.length - 1;
+  complete(true);
+  await messageList.value?.scrollToMessage(userMessage.id);
+  await runGeneration(sessionId, message, assistantIndex, false);
+}
+
+async function retryMessage(messageId: string) {
+  const sessionId = activeSessionId.value;
+  if (!sessionId || sending.value) return;
+  const assistantIndex = messages.value.findIndex((item) => item.id === messageId);
+  const assistantMessage = messages.value[assistantIndex];
+  if (!assistantMessage?.retryQuestion) return;
+  await messageList.value?.scrollToMessage(messageId);
+  await runGeneration(sessionId, assistantMessage.retryQuestion, assistantIndex, true);
+}
+
+async function runGeneration(sessionId: number, message: string, assistantIndex: number, retry: boolean) {
   const currentAssistant = () => messages.value[assistantIndex];
+  currentAssistant().content = "";
+  currentAssistant().reasoning = "";
+  currentAssistant().showReasoning = false;
+  currentAssistant().errorMessage = undefined;
+  currentAssistant().streaming = true;
   sending.value = true;
   followingOutput.value = true;
   activeAbortController.value = new AbortController();
   let hasContent = false;
-  await messageList.value?.scrollToMessage(userMessage.id);
   try {
     await streamChat(
-      activeSessionId.value,
+      sessionId,
       message,
       async (chunk) => {
         if (!hasContent) {
@@ -183,21 +204,22 @@ async function send(message: string, complete: (success: boolean) => void) {
         if (followingOutput.value) await messageList.value?.scrollToBottom();
       },
       activeAbortController.value.signal,
+      retry,
     );
     currentAssistant().streaming = false;
+    currentAssistant().retryQuestion = undefined;
     currentAssistant().createdAt = new Date().toISOString();
-    complete(true);
     await refreshSessionList();
     await refreshProfile();
   } catch (error) {
     currentAssistant().streaming = false;
     if (error instanceof DOMException && error.name === "AbortError") {
       currentAssistant().content = currentAssistant().content || "已停止生成";
-      complete(true);
+      currentAssistant().retryQuestion = undefined;
     } else {
-      currentAssistant().content = currentAssistant().content || "回答生成失败，请稍后重试。";
-      complete(false);
-      handleError(error, "AI 回复失败");
+      currentAssistant().errorMessage = chatErrorMessage(error);
+      currentAssistant().retryQuestion = message;
+      showToast("本次回答未完成，可以点击重新生成");
     }
   } finally {
     sending.value = false;
@@ -205,6 +227,14 @@ async function send(message: string, complete: (success: boolean) => void) {
   }
 }
 
+function chatErrorMessage(error: unknown) {
+  const message = error instanceof Error ? error.message : "";
+  if (/429|too many|请求过多|排队/i.test(message)) return "当前请求较多，AI 服务暂时繁忙。";
+  if (/timeout|超时/i.test(message)) return "等待 AI 回复超时，请稍后重新生成。";
+  if (/network|网络|连接/i.test(message)) return "网络连接不稳定，本次回答没有完成。";
+  return "AI 回答生成失败，本次不会扣除成功回答额度。";
+}
+
 async function stop() {
   if (!activeSessionId.value || !sending.value) return;
   activeAbortController.value?.abort();
@@ -551,6 +581,7 @@ async function copyText(text: string) {
           :messages="messages"
           :loading-session="loadingSession"
           @follow-change="followingOutput = $event"
+          @retry="retryMessage"
         />
         <ChatComposer :loading="sending" :disabled="!activeSessionId || loadingSession" @send="send" @stop="stop" />
         <SessionDrawer