From 05e63f745d89c9a16280428208c28a98258f9391 Mon Sep 17 00:00:00 2001
From: wujingjing <gersonwu@qq.com>
Date: 星期五, 07 三月 2025 17:53:48 +0800
Subject: [PATCH] 选择列记忆

---
 src/components/chat/Chat.vue |  104 +++++++++++++++++++++++++++++++++++++++-------------
 1 files changed, 78 insertions(+), 26 deletions(-)

diff --git a/src/components/chat/Chat.vue b/src/components/chat/Chat.vue
index a1a2058..60c0d18 100644
--- a/src/components/chat/Chat.vue
+++ b/src/components/chat/Chat.vue
@@ -76,16 +76,26 @@
 } from '/@/stores/chatRoom';
 import emitter from '/@/utils/mitt';
 import { useCompRef } from '/@/utils/types';
-import { toMyFixed } from '/@/utils/util';
+import { toFormData, toMyFixed } from '/@/utils/util';
 import { useLoadData } from './hooks/useLoadData';
 import { useSyncMsg } from './hooks/useSyncMsg';
 import { getCurrentPosition } from '/@/utils/brower';
+import { deepClone } from '/@/utils/other';
 const containerRef = useCompRef(ChatContainer);
 const chatListDom = computed(() => containerRef.value?.chatListDom);
 const scrollToBottom = () => {
 	containerRef.value?.scrollToBottom();
 };
-const { loadReplyData, parseContent, parseExtraContent, convertProcessItem, convertProcessToStep, formatShowTimeYear } = useLoadData();
+const {
+	loadReplyData,
+	parseContent,
+	parseExtraContent,
+	convertProcessItem,
+	convertProcessToStep,
+	convertAttach,
+	formatShowTimeYear,
+	getStepGroupList,
+} = useLoadData();
 const voicePageIsShow = ref(false);
 let isTalking = ref(false);
 const chatWidth = computed(() => containerRef.value?.chatWidth);
@@ -114,6 +124,8 @@
 		isTalking.value = !isTalking.value;
 	});
 };
+
+let streamOutputIsStart = false;
 let position: Position = null;
 const questionAi = async (text) => {
 	let judgeParams = null;
@@ -131,6 +143,10 @@
 		raw_mode: roomConfig.value?.[currentRouteId]?.isAnswerByLLM ?? false,
 		...judgeParams,
 	} as any;
+	const tableList = attachList.value.filter((item) => item.type === 'table').map((item) => item.model);
+	if (tableList?.length > 0) {
+		params.tables = JSON.stringify(tableList);
+	}
 
 	// if (!position) {
 	// 	const loadingInstance = ElLoadingService({
@@ -156,7 +172,12 @@
 		params.sample_id = currentSampleId;
 		currentSampleId = '';
 	}
-
+	const formDataParams = toFormData(params);
+	const fileList = attachList.value.filter((item) => item.type === 'file').map((item) => item.model);
+	for (const item of fileList) {
+		formDataParams.append('files', item.file);
+	}
+	// clearAttach();
 	let lastTimestamp = new Date().getTime();
 	questionRes = {};
 	let lastIsResult = false;
@@ -176,10 +197,9 @@
 			return isEmpty;
 		};
 		questionStreamByPost(
-			params,
+			formDataParams,
 			(chunkRes) => {
 				Logger.info('chunk response锛歕n\n' + JSON.stringify(chunkRes));
-
 				if (chunkRes.mode === 'result') {
 					lastIsResult = true;
 					const res = chunkRes.value;
@@ -206,6 +226,13 @@
 					triggerRefresh();
 					return;
 					// chunkRes.value = '鍑嗗鏁版嵁鍒嗘瀽';
+				}
+
+				if (chunkRes.mode === 'create_work_order') {
+					const lastMsg = computedMessageList.value.at(-1);
+					lastMsg.modeContent = chunkRes;
+					triggerRefresh();
+					return;
 				}
 
 				if (chunkRes.mode === 'summary') {
@@ -257,10 +284,12 @@
 
 				if (chunkRes.mode === 'conclusion') {
 					const lastReport = computedMessageList.value.at(-1)?.content?.values?.at(-1);
+
 					if (lastReport) {
 						lastReport.conclusion = chunkRes.value;
-						chunkRes.value = '鍒嗘瀽缁撴潫';
 					}
+					chunkRes.value = '鍒嗘瀽缁撴潫';
+
 				}
 				const getLastGroup = () => {
 					const lastGroup = computedMessageList.value.at(-1).stepGroup[0];
@@ -307,29 +336,44 @@
 					});
 					lastIsResult = false;
 				}
-				const lastGroup = computedMessageList.value.at(-1).stepGroup[0];
+				const lastGroup = getLastGroup();
 				const stepList = lastGroup?.value ?? [];
 				const currentTimeStamp = new Date().getTime();
-				const ms = toMyFixed(currentTimeStamp - lastTimestamp, 2) + ' ms';
 				if (chunkRes.mode === 'finish') {
+					const ms = toMyFixed(currentTimeStamp - lastTimestamp, 2) + ' ms';
 					stepList.at(-1).ms = ms;
 					isTalking.value = false;
+					streamOutputIsStart = false;
 					return;
 				}
 
-				if (stepList?.length >= 1) {
+				if (stepList?.length >= 1 && !streamOutputIsStart) {
+					const ms = toMyFixed(currentTimeStamp - lastTimestamp, 2) + ' ms';
+
 					stepList.at(-1).ms = ms;
+				}
+
+				if (!streamOutputIsStart) {
+					lastTimestamp = currentTimeStamp;
+				}
+				if (!streamOutputIsStart) {
+					const stepItem = convertProcessItem(chunkRes);
+					stepList.push(stepItem);
 				} else {
-					const stepGroup = computedMessageList.value.at(-1).stepGroup;
-					if (stepGroup.length > 1) {
-						const lastStepList = stepGroup.at(-2).value;
-						lastStepList.at(-1).ms = ms;
+					const lastItem = stepList.at(-1);
+					if (lastItem) {
+						lastItem.title += chunkRes.value ?? '';
 					}
 				}
-				lastTimestamp = currentTimeStamp;
-				const stepItem = convertProcessItem(chunkRes);
 
-				stepList.push(stepItem);
+				if (chunkRes.mode === 'begin_stream') {
+					streamOutputIsStart = true;
+					lastTimestamp = currentTimeStamp;
+				}
+				if (chunkRes.mode === 'end_stream') {
+					streamOutputIsStart = false;
+				}
+
 				// 寮哄埗瑙﹀彂鏇存柊
 
 				scrollToBottom();
@@ -354,12 +398,14 @@
 	const content = parseContent(questionRes, true);
 	return content;
 };
-
-const clearMessageContent = () =>
-	(messageContent.value = {
+const playBarRef = useCompRef(PlayBar);
+const attachList = computed(() => playBarRef.value?.attachList ?? []);
+const clearMessageContent = () => {
+	messageContent.value = {
 		type: AnswerType.Text,
 		values: '',
-	});
+	};
+};
 
 let currentSampleId = '';
 
@@ -386,7 +432,7 @@
 
 const addChatItem = (content: ChatContent) => {
 	isTalking.value = true;
-	const userItem: ChatMessage = { role: RoleEnum.user, content, isChecked: false } as any;
+	const userItem: ChatMessage = { role: RoleEnum.user, content, isChecked: false, attachList: deepClone(attachList.value) } as any;
 	const assistantItem: ChatMessage = {
 		role: RoleEnum.assistant,
 		content: {
@@ -399,6 +445,7 @@
 				isShow: true,
 			},
 		],
+
 		isStopMsg: false,
 		isChecked: false,
 	} as any;
@@ -408,6 +455,13 @@
 	messageList.value.push(assistantItem);
 	scrollToBottom();
 	return [userItem, assistantItem];
+};
+
+/**
+ * 娓呴櫎闄勪欢
+ */
+const clearAttach = () => {
+	playBarRef.value?.clearAttach();
 };
 
 const sendChatMessage = async (content: ChatContent = messageContent.value) => {
@@ -512,6 +566,7 @@
 			type: AnswerType.Text,
 			values: msgValue.question,
 		},
+		attachList: convertAttach(msgValue),
 		isChecked: false,
 	};
 
@@ -519,10 +574,8 @@
 		historyId: msgValue.history_id,
 		role: RoleEnum.assistant,
 		content: parseContent(msgValue),
-		stepGroup: (msgValue?.reports ?? []).map((item) => ({
-			value: convertProcessToStep(item?.exec_process),
-			isShow: false,
-		})),
+		stepGroup: getStepGroupList(msgValue?.reports ?? []),
+
 		isStopMsg: false,
 
 		conclusion: msgValue.conclusion ?? [],
@@ -571,7 +624,6 @@
 	messageContent.value.values = content;
 };
 //#endregion
-const playBarRef = useCompRef(PlayBar);
 //鐢ㄦ埛闂璁剧疆涓哄父鐢ㄨ
 const setCommonPhraseClick = (item) => {
 	playBarRef.value.addPhrase(item);

--
Gitblit v1.9.3