diff --git a/frontend/packages/core/src/pages/aiSetting/AiSettingModal.tsx b/frontend/packages/core/src/pages/aiSetting/AiSettingModal.tsx
index 3cd3a0b7..fa53f3a0 100644
--- a/frontend/packages/core/src/pages/aiSetting/AiSettingModal.tsx
+++ b/frontend/packages/core/src/pages/aiSetting/AiSettingModal.tsx
@@ -283,8 +283,9 @@ const AiSettingModalContent = forwardRef<AiSettingModalContentHandle, AiSettingM
       ),
       onOk: () => {
         return addModelModalRef.current?.save().then((res) => {
-          if (res === true) {
+          if (res) {
             getLlmList(lastLlmID)
+            form.setFieldValue('defaultLlm', res)
           }
         })
       },
diff --git a/frontend/packages/core/src/pages/aiSetting/contexts/AddModels.tsx b/frontend/packages/core/src/pages/aiSetting/contexts/AddModels.tsx
index 572ec51f..12c31f59 100644
--- a/frontend/packages/core/src/pages/aiSetting/contexts/AddModels.tsx
+++ b/frontend/packages/core/src/pages/aiSetting/contexts/AddModels.tsx
@@ -102,7 +102,7 @@ const AddModels = forwardRef<addModelsContentHandle, addModelContentProps>((prop
               ...value,
               id: modelID
             }
-            fetchData<BasicResponse<null>>('ai/provider/model', {
+            fetchData<BasicResponse<{ model: { id: string, name: string } }>>('ai/provider/model', {
               method: type === 'edit' ? 'PUT' : 'POST',
               eoParams: { provider: providerID },
               eoBody: finalValue,
@@ -112,7 +112,8 @@ const AddModels = forwardRef<addModelsContentHandle, addModelContentProps>((prop
                 const { code, msg } = response
                 if (code === STATUS_CODE.SUCCESS) {
                   message.success($t(RESPONSE_TIPS.success) || msg)
-                  resolve(true)
+                  const llmId = response.data?.model?.id
+                  resolve(llmId)
                 } else {
                   message.error(msg || $t(RESPONSE_TIPS.error))
                   reject(msg || $t(RESPONSE_TIPS.error))