Pārlūkot izejas kodu

fix: dataset_tool npe (#274)

John Wang 1 gadu atpakaļ
vecāks
revīzija
44a1aa5e44
1 mainītis faili ar 4 papildinājumiem un 1 dzēšanām
  1. 4 1
      api/core/chain/multi_dataset_router_chain.py

+ 4 - 1
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,