diff --git a/frontend/packages/core/src/pages/aiSetting/AIFlowChart.tsx b/frontend/packages/core/src/pages/aiSetting/AIFlowChart.tsx
index a9a1334f..fc6a87a6 100644
--- a/frontend/packages/core/src/pages/aiSetting/AIFlowChart.tsx
+++ b/frontend/packages/core/src/pages/aiSetting/AIFlowChart.tsx
@@ -114,17 +114,17 @@ const AIFlowChart = () => {
           title: 'API Keys',
           keys: model.keys.map((key, index) => ({
             id: key.id,
-            status: key.status === 'normal' ? 'enabled' : 'disable',
+            status: key.status,
             priority: index + 1
           }))
         }
       }))
     ]
 
-    const newEdges = [
+    const newEdges: any = [
       ...modelData.map((model) => ({
         id: `service-${model.id}`,
-        source: 'service',
+        source: 'apiService',
         target: model.id,
         label: `apis(${model.api_count})`,
         style: { stroke: '#ddd', cursor: 'pointer' },
diff --git a/frontend/packages/core/src/pages/aiSetting/components/KeyStatusNode.tsx b/frontend/packages/core/src/pages/aiSetting/components/KeyStatusNode.tsx
index 9b2eecd1..23cfed1b 100644
--- a/frontend/packages/core/src/pages/aiSetting/components/KeyStatusNode.tsx
+++ b/frontend/packages/core/src/pages/aiSetting/components/KeyStatusNode.tsx
@@ -16,7 +16,6 @@ export const KeyStatusNode: React.FC<{ data: KeyStatusNodeData }> = ({ data }) =
   const { title, keys = [] } = data
   const totalKeys = keys.length
   const keyWidth = totalKeys > 5 ? `calc((100% - ${(totalKeys - 1) * 0.25}rem) / ${totalKeys})` : KEY_SIZE
-
   return (
     <div
       className="relative p-4 bg-white rounded-lg shadow-sm node-card nodrag"