From bb7f454ecd99be258488545cfe468f7c555e3eaa Mon Sep 17 00:00:00 2001 From: John Wang Date: Mon, 29 May 2023 19:56:36 +0800 Subject: [PATCH] fix: dataset desc npe (#246) --- api/core/chain/multi_dataset_router_chain.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api/core/chain/multi_dataset_router_chain.py b/api/core/chain/multi_dataset_router_chain.py index 365f68eaf6..b11e3271ab 100644 --- a/api/core/chain/multi_dataset_router_chain.py +++ b/api/core/chain/multi_dataset_router_chain.py @@ -90,7 +90,9 @@ class MultiDatasetRouterChain(Chain): callback_manager=llm_callback_manager ) - destinations = ["{}: {}".format(d.id, d.description.replace('\n', ' ')) for d in datasets] + destinations = ["{}: {}".format(d.id, d.description.replace('\n', ' ') if d.description + else ('useful for when you want to answer queries about the ' + d.name)) + for d in datasets] destinations_str = "\n".join(destinations) router_template = MULTI_PROMPT_ROUTER_TEMPLATE.format( destinations=destinations_str