From 670b425101cac1ce898569504ad8acd27b5f638f Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Mon, 24 Aug 2026 17:53:17 +0800
Subject: [PATCH] fix: Fix Canvas Console Warning and AI Node Model Selection
 Component Usage

---
 ui/src/api/admin/workspace/model/model.ts     |  2 -
 .../business/model-select/index.vue           | 43 ++++---------
 .../workflow/ApplicationWorkflowView.vue      |  4 ++
 ui/src/workflow-canvas/core/CustomLine.vue    |  9 +--
 ui/src/workflow-canvas/core/edge.ts           | 30 ++-------
 ui/src/workflow-canvas/index.vue              |  6 +-
 .../nodes/ai-chat-node/index.vue              | 35 ++++++++--
 .../workflow-canvas/store/api/shared/index.ts |  1 +
 .../workflow-canvas/store/api/system/index.ts |  4 ++
 .../store/api/workspace-shared/index.ts       |  1 +
 .../store/api/workspace/index.ts              |  6 ++
 ui/src/workflow-canvas/store/index.ts         | 64 +++++++++++++++++++
 12 files changed, 135 insertions(+), 70 deletions(-)
 create mode 100644 ui/src/workflow-canvas/store/api/shared/index.ts
 create mode 100644 ui/src/workflow-canvas/store/api/system/index.ts
 create mode 100644 ui/src/workflow-canvas/store/api/workspace-shared/index.ts
 create mode 100644 ui/src/workflow-canvas/store/api/workspace/index.ts
 create mode 100644 ui/src/workflow-canvas/store/index.ts

diff --git a/ui/src/api/admin/workspace/model/model.ts b/ui/src/api/admin/workspace/model/model.ts
index 6953223f958..3dc0c4cc2a6 100644
--- a/ui/src/api/admin/workspace/model/model.ts
+++ b/ui/src/api/admin/workspace/model/model.ts
@@ -42,8 +42,6 @@ const putPauseModelDownload = (modelId: string) => {
   return put<undefined, boolean>(`${getPrefix()}/${modelId}/pause_download`)
 }
 
-
-
 export default {
   deleteModel,
   getModelDetail,
diff --git a/ui/src/components/business/model-select/index.vue b/ui/src/components/business/model-select/index.vue
index 4bcb9944400..b6981541b1f 100644
--- a/ui/src/components/business/model-select/index.vue
+++ b/ui/src/components/business/model-select/index.vue
@@ -1,17 +1,14 @@
 <script setup lang="ts">
-import { computed, onMounted, ref } from 'vue'
+import { computed, ref } from 'vue'
 import { Check } from '@element-plus/icons-vue'
-import ProviderApi from '@/api/admin/workspace/model/provider'
 import { MODEL_STATUS } from '@/api/enums'
 import type { ModelProviderItem, ModelItem } from '@/api/types'
-
+import { groupBy } from 'lodash'
 defineOptions({
   name: 'ModelSelect',
   inheritAttrs: false,
 })
 
-type ModelOptions = Record<string, ModelItem[]>
-
 interface ModelOptionGroup {
   icon: string
   models: ModelItem[]
@@ -21,23 +18,24 @@ interface ModelOptionGroup {
 
 const props = withDefaults(
   defineProps<{
-    modelType?: string
     modelValue: string
-    options: ModelOptions | null
+    options: ModelItem[]
+    providerOptions: ModelProviderItem[]
   }>(),
   {
-    modelType: '',
     modelValue: '',
-    options: null,
+    options: () => [],
+    providerOptions: () => [],
   },
 )
-
+const _options = computed(() => {
+  return groupBy(props.options, 'provider')
+})
 const emit = defineEmits<{
   change: [modelId: string]
   'update:modelValue': [modelId: string]
 }>()
 
-const providerOptions = ref<ModelProviderItem[]>([])
 const loading = ref(false)
 
 const selectedModelId = computed({
@@ -50,10 +48,10 @@ const selectedModelId = computed({
 
 const modelOptionGroups = computed<ModelOptionGroup[]>(() => {
   const providerMap = new Map(
-    providerOptions.value.map((provider) => [provider.provider, provider]),
+    props.providerOptions.map((provider) => [provider.provider, provider]),
   )
 
-  return Object.entries(props.options ?? {}).map(([provider, models]) => {
+  return Object.entries(_options.value ?? {}).map(([provider, models]) => {
     const providerOption = providerMap.get(provider)
     return {
       icon: providerOption?.icon ?? '',
@@ -74,25 +72,6 @@ const selectedProviderIcon = computed(
       models.some(({ id }) => id === selectedModelId.value),
     )?.icon ?? '',
 )
-
-function loadProviderOptions() {
-  loading.value = true
-
-  return ProviderApi.getProviderList()
-    .then((providers) => {
-      providerOptions.value = providers
-    })
-    .catch(() => {
-      providerOptions.value = []
-    })
-    .finally(() => {
-      loading.value = false
-    })
-}
-
-onMounted(() => {
-  loadProviderOptions()
-})
 </script>
 
 <template>
diff --git a/ui/src/views/workflow/ApplicationWorkflowView.vue b/ui/src/views/workflow/ApplicationWorkflowView.vue
index 8cfba655266..c49daefab1c 100644
--- a/ui/src/views/workflow/ApplicationWorkflowView.vue
+++ b/ui/src/views/workflow/ApplicationWorkflowView.vue
@@ -46,6 +46,10 @@ function handleSave() {
       loading.value = false
     })
 }
+const defaultWorkflow = {
+  nodes: [],
+  deges: [],
+}
 
 function loadApplicationDetail() {
   loading.value = true
diff --git a/ui/src/workflow-canvas/core/CustomLine.vue b/ui/src/workflow-canvas/core/CustomLine.vue
index 951cbac867c..f0831fc762c 100644
--- a/ui/src/workflow-canvas/core/CustomLine.vue
+++ b/ui/src/workflow-canvas/core/CustomLine.vue
@@ -1,13 +1,14 @@
 <script setup lang="ts">
 import type { BaseEdgeModel } from '@logicflow/core'
-
-const props = defineProps<{ model: BaseEdgeModel }>()
+import { inject } from 'vue'
+const getModel = inject('getModel') as () => BaseEdgeModel
+const model = getModel()
 const deleteEdge = () => {
-  props.model.graphModel.deleteEdgeById(props.model.id)
+  model.graphModel.deleteEdgeById(model.id)
 }
 </script>
 <template>
-  <div class="custom-edge cursor" @mouseup.stop @click.stop v-show="props.model.isHovered">
+  <div class="custom-edge cursor" @mouseup.stop @click.stop v-show="model.isHovered">
     <svg
       @click="deleteEdge"
       width="22"
diff --git a/ui/src/workflow-canvas/core/edge.ts b/ui/src/workflow-canvas/core/edge.ts
index c3612202622..10b63f3ef60 100644
--- a/ui/src/workflow-canvas/core/edge.ts
+++ b/ui/src/workflow-canvas/core/edge.ts
@@ -6,8 +6,7 @@ import {
   type GraphModel,
 } from '@logicflow/core'
 import type { App } from 'vue'
-import { createApp, h as vh } from 'vue'
-import { isActive, connect, disconnect } from './teleport'
+import { connect, disconnect } from './teleport'
 import CustomLine from './CustomLine.vue'
 
 function isMouseInElement(element: Element, event: PointerEvent) {
@@ -47,25 +46,12 @@ class CustomEdge2 extends BezierEdge {
   protected renderVueComponent(root: HTMLDivElement) {
     this.unmountVueComponent()
     this.root = root
-    const { graphModel } = this.props
     if (root) {
-      if (isActive()) {
-        connect(
-          this.targetId(),
-          CustomLine,
-          root,
-          this.props.model,
-          graphModel,
-          (node: BaseEdgeModel, graph: GraphModel) => {
-            return { model: node, graph }
-          },
-        )
-      } else {
-        this.customLineApp = createApp({
-          render: () => vh(CustomLine, { model: this.props.model }),
-        })
-        this.customLineApp?.mount(root)
-      }
+      connect(this.targetId(), CustomLine, root, () => {
+        return {
+          getModel: () => this.props.model,
+        }
+      })
     }
   }
   protected targetId() {
@@ -78,9 +64,7 @@ class CustomEdge2 extends BezierEdge {
     if (super.componentWillUnmount) {
       super.componentWillUnmount()
     }
-    if (isActive()) {
-      disconnect(this.targetId())
-    }
+    disconnect(this.targetId())
     this.unmountVueComponent()
   }
   /**
diff --git a/ui/src/workflow-canvas/index.vue b/ui/src/workflow-canvas/index.vue
index 4cd5de10458..28b91153ec4 100644
--- a/ui/src/workflow-canvas/index.vue
+++ b/ui/src/workflow-canvas/index.vue
@@ -28,8 +28,8 @@ const props = withDefaults(
 const nodeModules = import.meta.glob<{ default: LogicFlow.RegisterConfig }>('./nodes/**/index.ts', {
   eager: true,
 })
-const workflow_mode = inject('workflowMode') || WorkflowMode.Application
-const loop_workflow_mode = inject('loopWorkflowMode') || WorkflowMode.ApplicationLoop
+const workflow_mode = inject('workflowMode', WorkflowMode.Application)
+const loop_workflow_mode = inject('loopWorkflowMode', WorkflowMode.ApplicationLoop)
 
 const TeleportContainer = getTeleport()
 const lf = shallowRef<LogicFlow>()
@@ -69,7 +69,6 @@ function renderGraphData(data: LogicFlow.GraphConfigData = props.data ?? {}) {
   })
 
   initDefaultShortcut(lf.value, lf.value.graphModel)
-  console.log(Object.values(nodeModules).map(({ default: node }) => node))
   lf.value.batchRegister([...Object.values(nodeModules).map(({ default: node }) => node), AppEdge])
   lf.value.setDefaultEdgeType('app-edge')
   lf.value.render(data ? data : {})
@@ -82,6 +81,7 @@ function renderGraphData(data: LogicFlow.GraphConfigData = props.data ?? {}) {
     getGraph: () => graph,
     workflowMode: workflow_mode,
     loopWorkflowMode: loop_workflow_mode,
+    apiType: 'workspace',
   })
   lf.value.graphModel.eventCenter.on('delete_edge', (edgeIds: string[]) => {
     edgeIds.forEach((edgeId) => lf.value?.deleteEdge(edgeId))
diff --git a/ui/src/workflow-canvas/nodes/ai-chat-node/index.vue b/ui/src/workflow-canvas/nodes/ai-chat-node/index.vue
index b0c7c97ae1e..dbdc0f4090f 100644
--- a/ui/src/workflow-canvas/nodes/ai-chat-node/index.vue
+++ b/ui/src/workflow-canvas/nodes/ai-chat-node/index.vue
@@ -1,14 +1,17 @@
 <script setup lang="ts">
-import { computed, inject, onMounted, useTemplateRef } from 'vue'
-
-import { set } from 'lodash'
+import { computed, inject, onMounted, ref, useTemplateRef } from 'vue'
+import ModelSelect from '@/components/business/model-select/index.vue'
+import { groupBy, set } from 'lodash'
 import type { FormInstance } from 'element-plus'
 import NodeCascader from '@/workflow-canvas/core/NodeCascader.vue'
 import NodeContainer from '@/workflow-canvas/core/NodeContainer.vue'
 import type { BaseNodeModel } from '@logicflow/core'
+import { useWorkflowStore } from '@/workflow-canvas/store'
+import type { ModelItem } from '@/api/types'
 
 defineOptions({ name: 'WorkflowAiChatNode' })
 const getModel = inject('getModel') as () => BaseNodeModel
+const apiType = (inject('apiType') as string) || 'workspace'
 const model = getModel()
 interface AiChatNodeForm {
   model_id: string
@@ -47,8 +50,22 @@ function validate() {
     formRef.value?.validate(),
   ]).catch((error) => Promise.reject({ node: model, errMessage: error }))
 }
-
-onMounted(() => set(model, 'validate', validate))
+const store = useWorkflowStore(apiType)
+const modelList = ref<Array<ModelItem>>([])
+const providerOptions = ref<Array<any>>([])
+onMounted(() => {
+  set(model, 'validate', validate)
+  store
+    .getModelList({
+      model_type: 'LLM',
+    })
+    .then((data) => {
+      modelList.value = data
+    })
+  store.getProviderList().then((data) => {
+    providerOptions.value = data
+  })
+})
 </script>
 
 <template>
@@ -88,7 +105,13 @@ onMounted(() => set(model, 'validate', validate))
           class="w-full"
           placeholder="请选择变量"
         />
-        <el-input v-else v-model="formData.model_id" placeholder="请输入 AI 模型 ID" />
+        <ModelSelect
+          v-else
+          placeholder="请输入 AI 模型 ID"
+          :options="modelList"
+          :provider-options="providerOptions"
+          v-model="formData.model_id"
+        ></ModelSelect>
       </el-form-item>
 
       <el-form-item label="系统提示词">
diff --git a/ui/src/workflow-canvas/store/api/shared/index.ts b/ui/src/workflow-canvas/store/api/shared/index.ts
new file mode 100644
index 00000000000..b1c6ea436a5
--- /dev/null
+++ b/ui/src/workflow-canvas/store/api/shared/index.ts
@@ -0,0 +1 @@
+export default {}
diff --git a/ui/src/workflow-canvas/store/api/system/index.ts b/ui/src/workflow-canvas/store/api/system/index.ts
new file mode 100644
index 00000000000..948c1b07c41
--- /dev/null
+++ b/ui/src/workflow-canvas/store/api/system/index.ts
@@ -0,0 +1,4 @@
+import modelAPI from '@/api/admin/workspace/model/model'
+export default {
+  getModelList: modelAPI.getModelList,
+}
diff --git a/ui/src/workflow-canvas/store/api/workspace-shared/index.ts b/ui/src/workflow-canvas/store/api/workspace-shared/index.ts
new file mode 100644
index 00000000000..b1c6ea436a5
--- /dev/null
+++ b/ui/src/workflow-canvas/store/api/workspace-shared/index.ts
@@ -0,0 +1 @@
+export default {}
diff --git a/ui/src/workflow-canvas/store/api/workspace/index.ts b/ui/src/workflow-canvas/store/api/workspace/index.ts
new file mode 100644
index 00000000000..b0cc75fd125
--- /dev/null
+++ b/ui/src/workflow-canvas/store/api/workspace/index.ts
@@ -0,0 +1,6 @@
+import modelAPI from '@/api/admin/workspace/model/model'
+import providerAPI from '@/api/admin/workspace/model/provider'
+export default {
+  getModelList: modelAPI.getModelList,
+  getProviderList: providerAPI.getProviderList,
+}
diff --git a/ui/src/workflow-canvas/store/index.ts b/ui/src/workflow-canvas/store/index.ts
new file mode 100644
index 00000000000..58a3e0256db
--- /dev/null
+++ b/ui/src/workflow-canvas/store/index.ts
@@ -0,0 +1,64 @@
+import type { ModelItem, ModelProviderItem, RequestParams } from '@/api/types'
+
+type ApiModule = {
+  getModelList: (query?: RequestParams) => Promise<ModelItem[]>
+  getProviderList: () => Promise<ModelProviderItem[]>
+}
+
+const apiModules = import.meta.glob<{ default: ApiModule }>('./api/*/index.ts', {
+  eager: true,
+})
+
+const apiMap = Object.fromEntries(
+  Object.entries(apiModules).map(([path, mod]) => {
+    const key = path.match(/\.\/api\/(.+)\/index\.ts/)?.[1] ?? ''
+    return [key, mod.default]
+  }),
+) as Record<string, ApiModule>
+
+const cacheMap = new Map<string, Map<string, unknown>>()
+const pendingMap = new Map<string, Map<string, Promise<unknown>>>()
+
+export function useWorkflowStore(apiType: string): ApiModule {
+  const api = apiMap[apiType]
+  if (!api) {
+    throw new Error(`[useWorkflowStore] unknown apiType: "${apiType}"`)
+  }
+
+  const cache = cacheMap.get(apiType) ?? new Map<string, unknown>()
+  cacheMap.set(apiType, cache)
+
+  const pending = pendingMap.get(apiType) ?? new Map<string, Promise<unknown>>()
+  pendingMap.set(apiType, pending)
+
+  return {
+    getModelList(query?: RequestParams): Promise<ModelItem[]> {
+      const key = `model:${JSON.stringify(query ?? {})}`
+      if (cache.has(key)) return Promise.resolve(cache.get(key) as ModelItem[])
+      if (pending.has(key)) return pending.get(key) as Promise<ModelItem[]>
+
+      const promise = api.getModelList(query).then((list) => {
+        cache.set(key, list)
+        pending.delete(key)
+        return list
+      })
+
+      pending.set(key, promise)
+      return promise
+    },
+    getProviderList(): Promise<ModelProviderItem[]> {
+      const key = 'provider'
+      if (cache.has(key)) return Promise.resolve(cache.get(key) as ModelProviderItem[])
+      if (pending.has(key)) return pending.get(key) as Promise<ModelProviderItem[]>
+
+      const promise = api.getProviderList().then((list) => {
+        cache.set(key, list)
+        pending.delete(key)
+        return list
+      })
+
+      pending.set(key, promise)
+      return promise
+    },
+  }
+}