estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: dataset_tool npe (#274)

Browse Source
This commit is contained in:
John Wang
2023-05-31 17:16:27 +08:00
committed by GitHub
parent a616bf3129
commit 44a1aa5e44
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/chain/multi_dataset_router_chain.py
View File

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