Unverified Commit 44a1aa5e authored by John Wang's avatar John Wang Committed by GitHub

fix: dataset_tool npe (#274)

parent a616bf31
...@@ -110,7 +110,10 @@ class MultiDatasetRouterChain(Chain): ...@@ -110,7 +110,10 @@ class MultiDatasetRouterChain(Chain):
response_mode='no_synthesizer', # "compact" response_mode='no_synthesizer', # "compact"
callback_handler=DatasetToolCallbackHandler(conversation_message_task) callback_handler=DatasetToolCallbackHandler(conversation_message_task)
) )
dataset_tools[dataset.id] = dataset_tool
if dataset_tool:
dataset_tools[dataset.id] = dataset_tool
return cls( return cls(
router_chain=router_chain, router_chain=router_chain,
dataset_tools=dataset_tools, dataset_tools=dataset_tools,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment