fix: dataset_tool npe (#274)

2023-05-31 17:16:27 +08:00 · 2023-05-31 17:16:27 +08:00 · 44a1aa5e44
parent a616bf3129
commit 44a1aa5e44
1 changed files with 4 additions and 1 deletions
--- 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,