Pārlūkot izejas kodu

fix: dataset_tool npe (#274)

tags/0.3.2
John Wang pirms 2 gadiem
vecāks
revīzija
44a1aa5e44
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
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 Parādīt failu

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

Notiek ielāde…
Atcelt
Saglabāt