From 95ce224df0f4fd659a6e00805bddcad8816ecdb9 Mon Sep 17 00:00:00 2001
From: Wu Tianwei <30284043+WTW0313@users.noreply.github.com>
Date: Thu, 23 Oct 2025 11:33:58 +0800
Subject: [PATCH] fix: enhance checklist functionality with embedding and
 rerank model lists (#27312)

---
 web/app/components/workflow/hooks/use-checklist.ts | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/web/app/components/workflow/hooks/use-checklist.ts b/web/app/components/workflow/hooks/use-checklist.ts
index 1f474a699a..32945a8927 100644
--- a/web/app/components/workflow/hooks/use-checklist.ts
+++ b/web/app/components/workflow/hooks/use-checklist.ts
@@ -192,6 +192,8 @@ export const useChecklistBeforePublish = () => {
   const { getStartNodes } = useWorkflow()
   const workflowStore = useWorkflowStore()
   const { getNodesAvailableVarList } = useGetNodesAvailableVarList()
+  const { data: embeddingModelList } = useModelList(ModelTypeEnum.textEmbedding)
+  const { data: rerankModelList } = useModelList(ModelTypeEnum.rerank)
 
   const getCheckData = useCallback((data: CommonNodeType<{}>, datasets: DataSet[]) => {
     let checkData = data
@@ -211,6 +213,13 @@ export const useChecklistBeforePublish = () => {
         _datasets,
       } as CommonNodeType<KnowledgeRetrievalNodeType>
     }
+    else if (data.type === BlockEnum.KnowledgeBase) {
+      checkData = {
+        ...data,
+        _embeddingModelList: embeddingModelList,
+        _rerankModelList: rerankModelList,
+      } as CommonNodeType<KnowledgeBaseNodeType>
+    }
     return checkData
   }, [])