Skip to content

Commit 95ce224

Browse files
authored
fix: enhance checklist functionality with embedding and rerank model lists (langgenius#27312)
1 parent 8555635 commit 95ce224

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

‎web/app/components/workflow/hooks/use-checklist.ts‎

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,8 @@ export const useChecklistBeforePublish = () => {
192192
const { getStartNodes } = useWorkflow()
193193
const workflowStore = useWorkflowStore()
194194
const { getNodesAvailableVarList } = useGetNodesAvailableVarList()
195+
const { data: embeddingModelList } = useModelList(ModelTypeEnum.textEmbedding)
196+
const { data: rerankModelList } = useModelList(ModelTypeEnum.rerank)
195197

196198
const getCheckData = useCallback((data: CommonNodeType<{}>, datasets: DataSet[]) => {
197199
let checkData = data
@@ -211,6 +213,13 @@ export const useChecklistBeforePublish = () => {
211213
_datasets,
212214
} as CommonNodeType<KnowledgeRetrievalNodeType>
213215
}
216+
else if (data.type === BlockEnum.KnowledgeBase) {
217+
checkData = {
218+
...data,
219+
_embeddingModelList: embeddingModelList,
220+
_rerankModelList: rerankModelList,
221+
} as CommonNodeType<KnowledgeBaseNodeType>
222+
}
214223
return checkData
215224
}, [])
216225

0 commit comments

Comments
 (0)