| @@ -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, | |||