opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: enhance checklist functionality with embedding and rerank model lists (#27312)

This commit is contained in:
Wu Tianwei 2025-10-23 11:33:58 +08:00 committed by GitHub
parent 8555635967
commit 95ce224df0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
web/app/components/workflow/hooks/use-checklist.ts
View File

@ -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
}, [])