From 2a2c3c9af35e1ef7db3bcafd055b878e47297950 Mon Sep 17 00:00:00 2001
From: zhangshaohu <shaohu.zhang@fit2cloud.com>
Date: Fri, 23 Aug 2024 19:52:14 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?=E4=BF=9D=E5=AD=98=E6=97=A0=E6=B3=95=E5=9B=9E=E6=98=BE=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../serializers/application_serializers.py    |  2 +-
 ui/src/workflow/nodes/ai-chat-node/index.vue  |  3 ++-
 ui/src/workflow/nodes/question-node/index.vue | 25 ++++++++++++++++++-
 3 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/apps/application/serializers/application_serializers.py b/apps/application/serializers/application_serializers.py
index 47e8b95e4..b71bb6801 100644
--- a/apps/application/serializers/application_serializers.py
+++ b/apps/application/serializers/application_serializers.py
@@ -694,7 +694,7 @@ class ApplicationSerializer(serializers.Serializer):
 
             update_keys = ['name', 'desc', 'model_id', 'multiple_rounds_dialogue', 'prologue', 'status',
                            'dataset_setting', 'model_setting', 'problem_optimization',
-                           'api_key_is_active', 'icon', 'work_flow']
+                           'api_key_is_active', 'icon', 'work_flow', 'model_params_setting']
             for update_key in update_keys:
                 if update_key in instance and instance.get(update_key) is not None:
                     if update_key == 'multiple_rounds_dialogue':
diff --git a/ui/src/workflow/nodes/ai-chat-node/index.vue b/ui/src/workflow/nodes/ai-chat-node/index.vue
index b46e11707..ad0496427 100644
--- a/ui/src/workflow/nodes/ai-chat-node/index.vue
+++ b/ui/src/workflow/nodes/ai-chat-node/index.vue
@@ -31,6 +31,7 @@
                 <span>AI 模型<span class="danger">*</span></span>
               </div>
               <el-button
+                :disabled="!chat_data.model_id"
                 type="primary"
                 link
                 @click="openAIParamSettingDialog(chat_data.model_id)"
@@ -313,7 +314,7 @@ const openCreateModel = (provider?: Provider) => {
 
 const openAIParamSettingDialog = (modelId: string) => {
   if (modelId) {
-    AIModeParamSettingDialogRef.value?.open(modelId)
+    AIModeParamSettingDialogRef.value?.open(modelId,chat_data.value.model_params_setting)
   }
 }
 
diff --git a/ui/src/workflow/nodes/question-node/index.vue b/ui/src/workflow/nodes/question-node/index.vue
index f84221db1..f9d1a0685 100644
--- a/ui/src/workflow/nodes/question-node/index.vue
+++ b/ui/src/workflow/nodes/question-node/index.vue
@@ -14,8 +14,9 @@
         class="mb-24"
         label-width="auto"
         ref="questionNodeFormRef"
+        hide-required-asterisk
       >
-        <el-form-item
+      <el-form-item
           label="AI 模型"
           prop="model_id"
           :rules="{
@@ -24,6 +25,22 @@
             trigger: 'change'
           }"
         >
+          <template #label>
+            <div class="flex-between">
+              <div>
+                <span>AI 模型<span class="danger">*</span></span>
+              </div>
+              <el-button
+                type="primary"
+                link
+                :disabled="!form_data.model_id"
+                @click="openAIParamSettingDialog(form_data.model_id)"
+                @refreshForm="refreshParam"
+              >
+                {{ $t('views.application.applicationForm.form.paramSetting') }}
+              </el-button>
+            </div>
+          </template>
           <el-select
             @wheel="wheel"
             @keydown="isKeyDown = true"
@@ -195,6 +212,7 @@ import useStore from '@/stores'
 import { relatedObject } from '@/utils/utils'
 import type { Provider } from '@/api/type/model'
 import { isLastNode } from '@/workflow/common/data'
+const AIModeParamSettingDialogRef = ref<InstanceType<typeof AIModeParamSettingDialog>>()
 
 const { model } = useStore()
 const isKeyDown = ref(false)
@@ -235,6 +253,11 @@ function refreshParam(data: any) {
   set(props.nodeModel.properties.node_data, 'model_params_setting', data)
 }
 
+const openAIParamSettingDialog = (modelId: string) => {
+  if (modelId) {
+    AIModeParamSettingDialogRef.value?.open(modelId,form_data.value.model_params_setting)
+  }
+}
 const form_data = computed({
   get: () => {
     if (props.nodeModel.properties.node_data) {