update

src/components/AiBox.vue

@@ -58,7 +58,7 @@
 				</div>
 				<div class="text">概括页面内容</div>
 			</div>
-			<div class="submit">
+			<div class="submit" @click="sendMessage">
 				<img src="@/assets/icons/aiBox/submit.png" alt="" />
 			</div>
 		</div>
@@ -126,78 +126,78 @@ const updateLastAIMessage = content => {
 };
 
 // 使用 fetchEventSource 连接
-// const connectSSE = async question => {
-// 	// 添加用户消息
-// 	addMessage("user", question);
-
-// 	// 添加空的 AI 消息用于流式更新
-// 	addMessage("ai", "");
-
-// 	isLoading.value = true;
-
-// 	// 创建 AbortController 用于取消请求
-// 	abortController.value = new AbortController();
-
-// 	try {
-// 		await fetchEventSource("/api/chat-stream", {
-// 			method: "POST",
-// 			headers: {
-// 				"Content-Type": "application/json"
-// 			},
-// 			body: JSON.stringify({
-// 				question: question
-// 				// 其他参数...
-// 			}),
-// 			signal: abortController.value.signal,
-
-// 			onopen: async response => {
-// 				console.log("SSE 连接已建立", response.status);
-// 				if (response.status !== 200) {
-// 					throw new Error(`请求失败: ${response.status}`);
-// 				}
-// 			},
-
-// 			onmessage: event => {
-// 				try {
-// 					if (event.data === "[DONE]") {
-// 						// 流式输出结束
-// 						isLoading.value = false;
-// 						return;
-// 					}
-
-// 					const data = JSON.parse(event.data);
-
-// 					if (data.type === "content" && data.content) {
-// 						// 流式更新内容
-// 						updateLastAIMessage(prev => prev + data.content);
-// 					} else if (data.type === "error") {
-// 						throw new Error(data.message || "请求失败");
-// 					}
-// 				} catch (error) {
-// 					console.error("解析 SSE 数据错误:", error);
-// 				}
-// 			},
-
-// 			onclose: () => {
-// 				console.log("SSE 连接已关闭");
-// 				isLoading.value = false;
-// 			},
-
-// 			onerror: error => {
-// 				console.error("SSE 连接错误:", error);
-// 				ElMessage.error("连接失败，请重试");
-// 				isLoading.value = false;
-// 				// 不要抛出错误，否则会重试
-// 			}
-// 		});
-// 	} catch (error) {
-// 		console.error("SSE 请求失败:", error);
-// 		if (error.name !== "AbortError") {
-// 			ElMessage.error(error.message || "请求失败");
-// 		}
-// 		isLoading.value = false;
-// 	}
-// };
+const connectSSE = async question => {
+	// 添加用户消息
+	addMessage("user", question);
+
+	// 添加空的 AI 消息用于流式更新
+	addMessage("ai", "");
+
+	isLoading.value = true;
+
+	// 创建 AbortController 用于取消请求
+	abortController.value = new AbortController();
+
+	try {
+		await fetchEventSource("/api/chat-stream", {
+			method: "POST",
+			headers: {
+				"Content-Type": "application/json"
+			},
+			body: JSON.stringify({
+				question: question
+				// 其他参数...
+			}),
+			signal: abortController.value.signal,
+
+			onopen: async response => {
+				console.log("SSE 连接已建立", response.status);
+				if (response.status !== 200) {
+					throw new Error(`请求失败: ${response.status}`);
+				}
+			},
+
+			onmessage: event => {
+				try {
+					if (event.data === "[DONE]") {
+						// 流式输出结束
+						isLoading.value = false;
+						return;
+					}
+
+					const data = JSON.parse(event.data);
+
+					if (data.type === "content" && data.content) {
+						// 流式更新内容
+						updateLastAIMessage(prev => prev + data.content);
+					} else if (data.type === "error") {
+						throw new Error(data.message || "请求失败");
+					}
+				} catch (error) {
+					console.error("解析 SSE 数据错误:", error);
+				}
+			},
+
+			onclose: () => {
+				console.log("SSE 连接已关闭");
+				isLoading.value = false;
+			},
+
+			onerror: error => {
+				console.error("SSE 连接错误:", error);
+				ElMessage.error("连接失败，请重试");
+				isLoading.value = false;
+				// 不要抛出错误，否则会重试
+			}
+		});
+	} catch (error) {
+		console.error("SSE 请求失败:", error);
+		if (error.name !== "AbortError") {
+			ElMessage.error(error.message || "请求失败");
+		}
+		isLoading.value = false;
+	}
+};
 
 // 发送消息
 const sendMessage = async () => {

src/views/writtingAsstaint/index.vue

@@ -182,28 +182,29 @@ const getStreamChat = async (search, inputValue) => {
 		},
 		async onmessage(res) {
 			let msgData = JSON.parse(res.data);
-
-			console.log("ress", msgData.content);
-
+			console.log("resss", msgData.content);
 			let str = msgData.content;
-
-			reportContent.value += str;
-
-			updateContent(reportContent.value, scrollContainer.value);
+			if (str !== "[DONE]") {
+				reportContent.value += str;
+				updateContent(reportContent.value, scrollContainer.value);
+			} else {
+				ElMessage.success("报文生成结束");
+				abortController.value.abort();
+				abortController.value = new AbortController();
+			}
 		},
 		onerror(error) {
 			ElMessage({
-				message: "问题回答失败请重试",
+				message: "写报生成报错!",
 				type: "warning"
 			});
 			abortController.value.abort();
 			abortController.value = new AbortController();
-
 			throw new Error(error);
 		}
 	}).catch(error => {
 		ElMessage({
-			message: "问题回答失败请重试",
+			message: "写报生成报错!",
 			type: "warning"
 		});
 		abortController.value.abort();