From 92d2ea48d343fc00d81905167d033c40200ea716 Mon Sep 17 00:00:00 2001
From: wujingjing <gersonwu@qq.com>
Date: 星期二, 04 三月 2025 15:56:19 +0800
Subject: [PATCH] 快速引用;去重

---
 src/components/chat/smallChat/index.vue |  168 ++++++++++++++++++++++++++++++++++++++++++++++++++------
 1 files changed, 150 insertions(+), 18 deletions(-)

diff --git a/src/components/chat/smallChat/index.vue b/src/components/chat/smallChat/index.vue
index 6e1a57f..d19dd9c 100644
--- a/src/components/chat/smallChat/index.vue
+++ b/src/components/chat/smallChat/index.vue
@@ -134,7 +134,14 @@
 
 			<!-- 搴曢儴杈撳叆妗� -->
 			<div class="p-2 border-t">
-				<ChatInput v-model="inputText" @sendClick="sendClick" @toggleHistory="toggleHistory" :showHistory="showHistory" />
+				<ChatInput
+					:isTalking="lastIsLoading"
+					v-model="inputText"
+					@sendClick="sendClick"
+					@toggleHistory="toggleHistory"
+					@stopGenClick="stopGenClick"
+					:showHistory="showHistory"
+				/>
 			</div>
 		</div>
 		<Teleport to="body">
@@ -144,18 +151,23 @@
 </template>
 
 <script setup lang="ts" name="smallChat">
+import type { CancelTokenSource } from 'axios';
+import axios from 'axios';
+import { cloneDeep, defaults } from 'lodash-es';
+import { fromLonLat } from 'ol/proj';
 import { computed, nextTick, onMounted, ref } from 'vue';
 import ChatInput from './ChatInput.vue';
 import type { ChatMessage } from './types';
 import { AssistantContent } from './types';
+import WorkOrderDlg from './WorkOrderDlg.vue';
 import { agentStreamByPost } from '/@/api/ai/chat';
+import { getSearchMapElement } from '/@/api/map';
+import { useDrag } from '/@/hooks/useDrag';
 import { Logger } from '/@/model/logger/Logger';
 import { GaoDeSourceType, gaoDeSourceTypeMap, type OLMap } from '/@/model/map/OLMap';
-import assistantPic from '/static/images/role/assistant-200x192.png';
+import { systemGlobalConfig } from '/@/stores/global';
+import { formatDate } from '/@/utils/formatTime';
 import userPic from '/static/images/role/user-200x206.png';
-import { useDrag } from '/@/hooks/useDrag';
-import { cloneDeep, defaults } from 'lodash-es';
-import WorkOrderDlg from './WorkOrderDlg.vue';
 const props = defineProps<{
 	olMap?: OLMap;
 }>();
@@ -196,12 +208,17 @@
 	{ title: '鍒涘缓宸ュ崟', question: '鏉剧澶ч亾DN800鏉惧矖鑱旈�氱洃娴嬭澶囨病鏈夋暟鎹紝鍒涘缓涓�涓澶囩淮淇伐鍗曪紝璇峰強鏃舵淳浜虹淮淇��' },
 ]);
 const chatContentRef = ref<HTMLDivElement>(null);
-
 const getLastAssistantMessage = () => {
 	const last = historyMessages.value[historyMessages.value.length - 1];
-	const result = last.role === 'assistant' ? last : null;
+	const result = last?.role === 'assistant' ? last : null;
 	return result as ChatMessage<AssistantContent>;
 };
+
+const lastIsLoading = computed(() => {
+	const last = getLastAssistantMessage();
+	const loading = last?.content?.isLoading ?? false;
+	return loading;
+});
 
 //#region ====================== 娣诲姞宸ュ崟 ======================
 const optDlgIsShow = ref(false);
@@ -241,10 +258,81 @@
 	openOptDlg(formData ?? {});
 };
 
+const handleSwitchLayer = (formData: { layerId: string; visible: boolean }) => {
+	props.olMap.setLayerVisible(formData.layerId, formData.visible);
+	refreshAssistantMessage({ value: `鎴愬姛`, isError: false });
+};
+
+const changeTheme = (formData: { themeId: string }) => {
+	props.olMap.setThemeById(formData.themeId);
+	refreshAssistantMessage({ value: `鎴愬姛`, isError: false });
+};
+
+const handleQueryObject = async (formData: { objectName: string }) => {
+	const res = await getSearchMapElement({
+		search_text: formData.objectName,
+		max_count: 10,
+		time: formatDate(new Date()),
+	});
+	const result = res?.values ?? [];
+	props.olMap.clearObjectSearch();
+	const features = [];
+	for (const item of result) {
+		if (!item.WKT) continue;
+		const feature = props.olMap.readWKT(item.WKT);
+		features.push(feature);
+	}
+	props.olMap.zoomToFeatures(features);
+	props.olMap.highlightSearch(features);
+	refreshAssistantMessage({ value: `鎴愬姛`, isError: false });
+};
+
+const handleSearchAddress = async (formData: { address: string }) => {
+	const result = await props.olMap.gaodeAddressSearch(formData.address, systemGlobalConfig.value['ui.project_city']);
+	const pois = result?.pois ?? [];
+	const searchResultList =
+		pois
+			.map((item) => {
+				return {
+					name: item.name,
+					// address: item.address,
+					cityname: item.cityname,
+					adname: item.adname,
+					model: item,
+					isSearchObj: false,
+					location: item.location,
+				};
+			})
+			?.slice(0, 5) ?? [];
+
+	const features = [];
+	for (const item of searchResultList) {
+		if (!item.location) continue;
+		const [lon, lat] = item.location.split(',').map(Number);
+		const point = fromLonLat([lon, lat]);
+		const WKT = `SRID=3857;POINT(${point[0]} ${point[1]})`;
+		const feature = props.olMap.readWKT(WKT);
+		features.push(feature);
+	}
+	props.olMap.highlightSearch(features);
+	props.olMap.zoomToFeatures(features);
+	refreshAssistantMessage({ value: `鎴愬姛`, isError: false });
+};
+
+const handleSetBackgroundLayer = (formData: { LayerId: string }) => {
+	if (!formData.LayerId) return;
+	props.olMap.setSourceType(formData.LayerId as GaoDeSourceType);
+	refreshAssistantMessage({ value: `鎴愬姛`, isError: false });
+};
+let lastAxiosSource: CancelTokenSource = null;
+
 const startStream = (question: string) => {
 	if (lastIsInit) {
 		showHistory.value = false;
 	}
+
+	const currentSource = axios.CancelToken.source();
+	lastAxiosSource = currentSource;
 	// if (question === '鏉剧澶ч亾DN800鏉惧矖鑱旈�氱洃娴嬭澶囨病鏈夋暟鎹紝鍒涘缓涓�涓澶囩淮淇伐鍗曪紝璇峰強鏃舵淳浜虹淮淇��') {
 	// 	setTimeout(() => {
 	// 		openOptDlg();
@@ -263,24 +351,57 @@
 		(chunkRes) => {
 			Logger.info('agent stream response锛歕n\n' + JSON.stringify(chunkRes));
 
-			if (chunkRes.mode === 'map') {
-				haveMapOperate = true;
-				if (chunkRes.type === 'string') {
-					const jsonData = JSON.parse(chunkRes.value);
-					handleMapCommand(jsonData);
-				}
-			} else if (chunkRes.mode === 'create_work_order') {
-				haveMapOperate = true;
-				if(chunkRes.type==='string'){
-					const jsonData = JSON.parse(chunkRes.value);
-					handleCreateWorkOrder(jsonData)
+			if (
+				chunkRes.type === 'string' &&
+				['create_work_order', 'switch_layers', 'switch_topic', 'query_address', 'query_object', 'map'].includes(chunkRes.mode)
+			) {
+				const jsonData = JSON.parse(chunkRes.value);
+
+				switch (chunkRes.mode) {
+					case 'create_work_order':
+						haveMapOperate = true;
+						handleCreateWorkOrder(jsonData);
+						break;
+
+					case 'switch_layers':
+						haveMapOperate = true;
+						handleSwitchLayer(jsonData);
+						break;
+
+					case 'switch_topic':
+						haveMapOperate = true;
+						changeTheme(jsonData);
+						break;
+
+					case 'switch_background_layers':
+						haveMapOperate = true;
+						handleSetBackgroundLayer(jsonData);
+						break;
+
+					case 'query_address':
+						haveMapOperate = true;
+						handleSearchAddress(jsonData);
+						break;
+
+					case 'query_object':
+						haveMapOperate = true;
+						handleQueryObject(jsonData);
+						break;
+					case 'map':
+						haveMapOperate = true;
+						handleMapCommand(jsonData);
+						break;
 				}
 			}
+
 			if (chunkRes.mode === 'finish') {
 				if (!haveMapOperate) {
 					refreshAssistantMessage({ reason: `鏈瘑鍒埌鎿嶄綔锛�"${question}"` });
 				}
 			}
+		},
+		{
+			cancelToken: currentSource.token,
 		}
 	).catch((error) => {
 		Logger.error('agent stream error锛歕n\n' + error);
@@ -417,6 +538,17 @@
 	});
 };
 //#endregion
+
+//#region ====================== 娴佸仠姝� ======================
+const stopGenClick = () => {
+	lastAxiosSource?.cancel();
+	const last = getLastAssistantMessage();
+	if (!last) return;
+	last.content.isLoading = false;
+	last.content.isError = true;
+	last.content.reason = '鐢ㄦ埛鍋滄鎿嶄綔';
+};
+//#endregion
 onMounted(() => {});
 </script>
 

--
Gitblit v1.9.3