Browse Source

fix: dataset desc npe (#246)

John Wang 1 năm trước cách đây
mục cha
commit
bb7f454ecd

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

@@ -90,7 +90,9 @@ class MultiDatasetRouterChain(Chain):
             callback_manager=llm_callback_manager
             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)
         destinations_str = "\n".join(destinations)
         router_template = MULTI_PROMPT_ROUTER_TEMPLATE.format(
         router_template = MULTI_PROMPT_ROUTER_TEMPLATE.format(
             destinations=destinations_str
             destinations=destinations_str