From 44a1aa5e4458c8c9566935e7c5a8055cba21685b Mon Sep 17 00:00:00 2001 From: John Wang Date: Wed, 31 May 2023 17:16:27 +0800 Subject: [PATCH] fix: dataset_tool npe (#274) --- api/core/chain/multi_dataset_router_chain.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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,