File tree Expand file tree Collapse file tree 1 file changed +9
-0
lines changed
web/app/components/workflow/hooks Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments