From 34bc30a2c5b1e2b1dc20402c171f6f8e363d038d Mon Sep 17 00:00:00 2001
From: 2910410219 <2910410219@qq.com>
Date: Tue, 12 May 2026 15:03:47 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=94=B9=E8=A1=A8=E5=8D=95=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E7=B1=BB=E5=9E=8B=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/digitalHuman/creation/index.ts    |  1 +
 src/components/model/ModelSelector.vue    |  6 ++--
 src/views/digitalHuman/creation/index.vue | 38 +++++++++++------------
 3 files changed, 23 insertions(+), 22 deletions(-)

diff --git a/src/api/digitalHuman/creation/index.ts b/src/api/digitalHuman/creation/index.ts
index 7400514..f76b33b 100644
--- a/src/api/digitalHuman/creation/index.ts
+++ b/src/api/digitalHuman/creation/index.ts
@@ -22,6 +22,7 @@ export interface NodeLibraryModelConfig {
 export interface NodeLibraryItem {
 	nodeCode: string;
 	nodeName: string;
+	modelType: number;
 	skillOption: boolean;
 	formConfig: NodeLibraryFormItem[];
 	modelConfig: NodeLibraryModelConfig[];
diff --git a/src/components/model/ModelSelector.vue b/src/components/model/ModelSelector.vue
index 1094e6b..178d2cc 100644
--- a/src/components/model/ModelSelector.vue
+++ b/src/components/model/ModelSelector.vue
@@ -1,4 +1,4 @@
-<template>
+<template>
 	<el-dialog v-model="visible" title="选择模型" width="1000px" :close-on-click-modal="false" @close="handleClose">
 		<div class="model-selector-header">
 			<div class="search-bar">
@@ -127,6 +127,7 @@ interface ModelItem {
 interface Props {
 	modelValue: boolean;
 	defaultModel?: ModelItem | null;
+	modelType?: number;
 }
 
 interface Emits {
@@ -137,6 +138,7 @@ interface Emits {
 const props = withDefaults(defineProps<Props>(), {
 	modelValue: false,
 	defaultModel: null,
+	modelType: 1,
 });
 
 const emit = defineEmits<Emits>();
@@ -207,7 +209,7 @@ const fetchModelList = async () => {
 			pageNum: pagination.pageNum,
 			pageSize: pagination.pageSize,
 			modelName: searchParams.modelName || undefined,
-			modelType: 1, // 只获取推理模型
+			modelType: props.modelType, // 使用传入的 modelType
 		};
 		const res: any = await getModelModuleList(params);
 		modelList.value = res.data?.list || [];
diff --git a/src/views/digitalHuman/creation/index.vue b/src/views/digitalHuman/creation/index.vue
index a9c649f..4f2f01e 100644
--- a/src/views/digitalHuman/creation/index.vue
+++ b/src/views/digitalHuman/creation/index.vue
@@ -342,22 +342,8 @@
 								<el-input v-model="userInput" placeholder="说点什么..." class="chat-input" @keydown.enter="sendMessage" />
 
 								<!-- 右侧发送/暂停按钮 -->
-								<el-button 
-									v-if="!isCreating"
-									type="primary" 
-									:icon="Promotion" 
-									@click="sendMessage" 
-									class="send-btn" 
-									circle 
-								/>
-								<el-button 
-									v-else
-									type="danger" 
-									:icon="VideoPause" 
-									@click="stopExecution" 
-									class="send-btn" 
-									circle 
-								/>
+								<el-button v-if="!isCreating" type="primary" :icon="Promotion" @click="sendMessage" class="send-btn" circle />
+								<el-button v-else type="danger" :icon="VideoPause" @click="stopExecution" class="send-btn" circle />
 							</div>
 
 							<!-- 已选技能标签 -->
@@ -513,7 +499,7 @@
 		<SkillSelector v-model="showCreationSkillSelector" :default-skill="selectedCreationSkill" @confirm="handleCreationSkillConfirm" />
 
 		<!-- 模型选择器 -->
-		<ModelSelector v-model="showModelSelector" :default-model="selectedModelData" @confirm="handleModelConfirm" />
+		<ModelSelector v-model="showModelSelector" :default-model="selectedModelData" :model-type="currentNodeModelType" @confirm="handleModelConfirm" />
 
 		<!-- 对话模型选择器 -->
 		<el-dialog v-model="showChatModelSelector" title="设置对话模型" width="900px" :close-on-click-modal="false">
@@ -758,6 +744,18 @@ const currentNodeSkillOption = computed(() => {
 	});
 	return skillOption;
 });
+// 获取当前节点的模型类型
+const currentNodeModelType = computed(() => {
+	let modelType = 1; // 默认为推理模型
+	nodeLibraryGroups.value.forEach((group) => {
+		(group.items || []).forEach((item) => {
+			if (item.nodeCode === formState.nodeCode) {
+				modelType = item.modelType || 1;
+			}
+		});
+	});
+	return modelType;
+});
 // 获取当前选中模型的表单字段
 const currentModelForm = computed<NodeLibraryFormItem[]>(() => {
 	// 不显示模型的表单字段，返回空数组
@@ -1314,9 +1312,9 @@ const stopExecution = async () => {
 	try {
 		// TODO: 调用终止执行的接口
 		// await stopExecutionApi({ sessionId: getSessionId() });
-		
+
 		ElMessage.warning('终止执行功能开发中...');
-		
+
 		// 暂时直接停止加载状态
 		isCreating.value = false;
 	} catch (error) {
@@ -1347,7 +1345,7 @@ const handleTreeNodeClick = async (data: TreeNode) => {
 				currentSessionId.value = data.sessionId || null;
 				// 标记为从工作空间进入
 				isFromWorkspace.value = true;
-				
+
 				// 切换到创作模式
 				isCreationMode.value = true;
 				currentWorkflowForCreation.value = res.data;