fix: dataset_tool npe (#274)

44a1aa5e · John Wang · GitHub · a616bf31 · 44a1aa5e
Unverified Commit 44a1aa5e authored May 31, 2023 by John Wang Committed by GitHub May 31, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

multi_dataset_router_chain.py api/core/chain/multi_dataset_router_chain.py +4 -1

No files found.
--- 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,