diff --git a/frontend/src/pages/configForm/configUi/components/llm-container.tsx b/frontend/src/pages/configForm/configUi/components/llm-container.tsx index b73bd9cfc..67b2e747e 100644 --- a/frontend/src/pages/configForm/configUi/components/llm-container.tsx +++ b/frontend/src/pages/configForm/configUi/components/llm-container.tsx @@ -39,7 +39,7 @@ const LLMContainer = (props) => { return; } const modelData = (config.from === 'graph' ? graphOperator.getConfig(config.defaultValue) : config.defaultValue); - form.setFieldsValue(pick(modelData, ['model', 'temperature', 'systemPrompt'])); + form.setFieldsValue(pick(modelData, ['accessInfo', 'model', 'temperature', 'systemPrompt'])); llmRef.current.setPromptValue(modelData.systemPrompt); }, [config, appConfig]); diff --git a/frontend/src/pages/configForm/configUi/components/llm.tsx b/frontend/src/pages/configForm/configUi/components/llm.tsx index 5423eb460..b029aabce 100644 --- a/frontend/src/pages/configForm/configUi/components/llm.tsx +++ b/frontend/src/pages/configForm/configUi/components/llm.tsx @@ -49,13 +49,7 @@ const LLM = (props) => { const handleGetModels = (open) => { if (!open) return; getModels().then((res) => { - const models = res.models.map((model) => { - return { - ...model, - id: model.serviceName + '***' + model.tag - } - }) - setModels(models); + setModels(res.models); }) } @@ -105,8 +99,9 @@ const LLM = (props) => { }; // 校验模型是否存在 - const checkExist = (rules, value) => { - if (!models.find(item => item.id === value)) { + const checkExist = () => { + const accessInfo = form.getFieldValue('accessInfo'); + if (!models.find(item => item.serviceName === accessInfo.serviceName && item.tag === accessInfo.tag)) { return Promise.reject(new Error(`${t('LLM')}${value}${t('modelNotExistTip')}`)); } return Promise.resolve(); @@ -118,7 +113,8 @@ const LLM = (props) => { dispatch(setValidateInfo(appValidateInfo.filter(item => (item.configCheckId !== validateItem.configCheckId) && item.configName !== 'accessInfo'))); }) } - const updateModelInfo = models.find(item => item.id === value); + const updateModelInfo = models.find(item => item.serviceName === value); + form.setFieldValue('accessInfo', { serviceName: updateModelInfo.serviceName, tag: updateModelInfo.tag }); updateData({ model: value, accessInfo: { serviceName: updateModelInfo.serviceName, tag: updateModelInfo.tag } }); }; @@ -160,7 +156,7 @@ const LLM = (props) => { onChange={(value) => handleUpdateModel(models, value)} fieldNames={{ label: 'serviceName', - value: 'id' + value: 'serviceName' }} > diff --git a/frontend/src/pages/configForm/configUi/components/recommend-container.tsx b/frontend/src/pages/configForm/configUi/components/recommend-container.tsx index 6ea3068fd..5dfe62ffa 100644 --- a/frontend/src/pages/configForm/configUi/components/recommend-container.tsx +++ b/frontend/src/pages/configForm/configUi/components/recommend-container.tsx @@ -31,6 +31,7 @@ const RecommendContainer = (props) => { dispatch(setConfigItem({ key: config.name, value })); } updateData(); + setRecommendValues(value); }; // 新增猜你想问