From bb977213642322c5d83f530f5ce9aa9c0b2e3d7f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=B0=A8=E6=AC=A3?= <echo.cmy@antgroup.com>
Date: Wed, 4 Sep 2024 18:12:08 +0800
Subject: [PATCH] Update AddFlowVariableModal component to include parameter
 management

---
 .../flow/canvas-modal/add-flow-variable-modal.tsx   | 13 +++++--------
 1 file changed, 5 insertions(+), 8 deletions(-)

diff --git a/web/components/flow/canvas-modal/add-flow-variable-modal.tsx b/web/components/flow/canvas-modal/add-flow-variable-modal.tsx
index 94c65aa2a..d48c6c731 100644
--- a/web/components/flow/canvas-modal/add-flow-variable-modal.tsx
+++ b/web/components/flow/canvas-modal/add-flow-variable-modal.tsx
@@ -122,7 +122,7 @@ export const AddFlowVariableModal: React.FC = () => {
   const onFinish = (values: any) => {
     console.log('Received values of form:', values);
 
-    // 将表单的值转换为 JSON 字符串
+    // save variables to localStorage
     const variables = JSON.stringify(values.parameters);
     localStorage.setItem('variables', variables);
     setIsModalOpen(false);
@@ -201,6 +201,7 @@ export const AddFlowVariableModal: React.FC = () => {
       const param = parameters?.[index];
       if (param) {
         param.value = variableStr;
+        param.category = selectedVariableData?.data?.category;
         form.setFieldsValue({
           parameters: [...parameters],
         });
@@ -316,13 +317,9 @@ export const AddFlowVariableModal: React.FC = () => {
                       <Input placeholder='Parameter Description' />
                     </Form.Item>
 
-                    <Form.Item name={[name, 'key']} hidden initialValue='dbgpt.core.flow.params'>
-                      <Input />
-                    </Form.Item>
-
-                    <Form.Item name={[name, 'scope']} hidden initialValue='flow_priv'>
-                      <Input />
-                    </Form.Item>
+                    <Form.Item name={[name, 'key']} hidden initialValue='dbgpt.core.flow.params' />
+                    <Form.Item name={[name, 'scope']} hidden initialValue='flow_priv' />
+                    <Form.Item name={[name, 'category']} hidden initialValue='common' />
 
                     <MinusCircleOutlined onClick={() => remove(name)} />
                   </Space>