From cf521ad79e92c4ffdf66f957db754148041192d7 Mon Sep 17 00:00:00 2001
From: jialin <jialinkuang@126.com>
Date: Mon, 2 Mar 2026 10:20:08 +0800
Subject: [PATCH] fix:  default_env field not display in backend editing yaml

---
 src/pages/backends/components/add-modal.tsx     | 6 +-----
 src/pages/backends/config/index.ts              | 7 +++++--
 src/pages/backends/hooks/use-create-backend.tsx | 3 ++-
 3 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/pages/backends/components/add-modal.tsx b/src/pages/backends/components/add-modal.tsx
index 05155ed7..4a1bbe52 100644
--- a/src/pages/backends/components/add-modal.tsx
+++ b/src/pages/backends/components/add-modal.tsx
@@ -145,11 +145,7 @@ const AddModal: React.FC<AddModalProps> = (props) => {
         is_built_in:
           data.is_built_in &&
           data.backend_source === BackendSourceValueMap.BUILTIN,
-        ..._.pick(values.built_in_version_configs?.[key], [
-          'image_name',
-          'run_command',
-          'entrypoint'
-        ])
+        ..._.pick(values.built_in_version_configs?.[key], versionFields)
       }));
 
       return {
diff --git a/src/pages/backends/config/index.ts b/src/pages/backends/config/index.ts
index 571c2a66..63c585f6 100644
--- a/src/pages/backends/config/index.ts
+++ b/src/pages/backends/config/index.ts
@@ -192,7 +192,8 @@ export const customBackendFields = [
   'health_check_path',
   'default_run_command',
   'version_configs',
-  'default_backend_param'
+  'default_backend_param',
+  'default_env'
 ];
 
 /**
@@ -201,7 +202,8 @@ export const customBackendFields = [
 export const builtInBackendFields = [
   'description',
   'version_configs',
-  'default_backend_param'
+  'default_backend_param',
+  'default_env'
 ];
 
 export const frameworks = [
@@ -278,6 +280,7 @@ export const yamlTemplate = `# ----------------------------------------
 #   - custom_framework:
 #       - required
 #       - choose from: ${Object.values(GPUDriverMap).join(', ')}, cpu
+#   - env: optional, map of env key and value
 
 backend_name: vllm-custom
 description: this is my custom vllm backend
diff --git a/src/pages/backends/hooks/use-create-backend.tsx b/src/pages/backends/hooks/use-create-backend.tsx
index ed585b66..ff1a2a97 100644
--- a/src/pages/backends/hooks/use-create-backend.tsx
+++ b/src/pages/backends/hooks/use-create-backend.tsx
@@ -1,5 +1,6 @@
 import IconFont from '@/components/icon-font';
 import { PageAction } from '@/config';
+import { PageActionType } from '@/config/types';
 import useBodyScroll from '@/hooks/use-body-scroll';
 import { ListItem } from '../config/types';
 import useCommunityBackend from './use-community-backend';
@@ -30,7 +31,7 @@ const useCreateBackend = () => {
   ];
 
   const handleEditBackend = (
-    action: PageAction,
+    action: PageActionType,
     title: string,
     row: ListItem
   ) => {