diff --git a/api/core/chain/multi_dataset_router_chain.py b/api/core/chain/multi_dataset_router_chain.py index b11e3271ab..a66b5fdb11 100644 --- a/api/core/chain/multi_dataset_router_chain.py +++ b/api/core/chain/multi_dataset_router_chain.py @@ -110,7 +110,10 @@ class MultiDatasetRouterChain(Chain): response_mode='no_synthesizer', # "compact" callback_handler=DatasetToolCallbackHandler(conversation_message_task) ) - dataset_tools[dataset.id] = dataset_tool + + if dataset_tool: + dataset_tools[dataset.id] = dataset_tool + return cls( router_chain=router_chain, dataset_tools=dataset_tools,