fix: dataset_tool npe (#274)

This commit is contained in:
John Wang 2023-05-31 17:16:27 +08:00 committed by GitHub
parent a616bf3129
commit 44a1aa5e44
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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