github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-24 12:45:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: set flow variable default value

Browse Source
This commit is contained in:
谨欣 2024-09-04 23:46:05 +08:00
parent 00d9e51495
commit ef04432902
2 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
web/components/flow/canvas-modal/add-flow-variable-modal.tsx
View File

@ -1,5 +1,5 @@
import { apiInterceptors, getKeys, getVariablesByKey } from '@/client/api';
import { IGetKeysResponseData, IVariableItem } from '@/types/flow';
import { IFlowUpdateParam, IGetKeysResponseData, IVariableItem } from '@/types/flow';
import { buildVariableString } from '@/utils/flow';
import { MinusCircleOutlined, PlusOutlined } from '@ant-design/icons';
import { Button, Cascader, Form, Input, Modal, Select, Space } from 'antd';
@ -17,7 +17,11 @@ interface Option {
isLeaf?: boolean;
}
export const AddFlowVariableModal: React.FC = () => {
type Props = {
flowInfo?: IFlowUpdateParam;
};
export const AddFlowVariableModal: React.FC<Props> = ({ flowInfo }) => {
const { t } = useTranslation();
const [isModalOpen, setIsModalOpen] = useState(false);
const [form] = Form.useForm();
@ -155,7 +159,7 @@ export const AddFlowVariableModal: React.FC = () => {
autoComplete='off'
layout='vertical'
className='mt-8'
initialValues={{ parameters: [{}] }}
initialValues={{ parameters: flowInfo?.variables || [{}] }}
>
<Form.List name='parameters'>
{(fields, { add, remove }) => (

15
web/pages/construct/flow/canvas/index.tsx
View File

@ -26,14 +26,13 @@ import 'reactflow/dist/style.css';
const nodeTypes = { customNode: CanvasNode };
const edgeTypes = { buttonedge: ButtonEdge };
const Canvas: React.FC = () => {
const { t } = useTranslation();
const [messageApi, contextHolder] = message.useMessage();
const searchParams = useSearchParams();
const id = searchParams?.get('id') || '';
const reactFlow = useReactFlow();
const [messageApi, contextHolder] = message.useMessage();
const [loading, setLoading] = useState(false);
const [nodes, setNodes, onNodesChange] = useNodesState([]);
@ -43,10 +42,10 @@ const Canvas: React.FC = () => {
const [isSaveFlowModalOpen, setIsSaveFlowModalOpen] = useState(false);
const [isExportFlowModalOpen, setIsExportFlowModalOpen] = useState(false);
const [isImportModalOpen, setIsImportFlowModalOpen] = useState(false);
if (localStorage.getItem('importFlowData') ) {
const importFlowData = JSON.parse(localStorage.getItem('importFlowData') );
localStorage.removeItem('importFlowData')
if (localStorage.getItem('importFlowData')) {
const importFlowData = JSON.parse(localStorage.getItem('importFlowData') || '');
localStorage.removeItem('importFlowData');
setLoading(true);
const flowData = mapUnderlineToHump(importFlowData.flow_data);
setFlowInfo(importFlowData);
@ -260,7 +259,7 @@ const Canvas: React.FC = () => {
<Background color='#aaa' gap={16} />
<AddFlowVariableModal />
<AddFlowVariableModal flowInfo={flowInfo} />
</ReactFlow>
</div>
</div>