From aae2fb8a3093817bfe453d5a7cb1f4f5037b8312 Mon Sep 17 00:00:00 2001 From: John Wang Date: Fri, 28 Jul 2023 17:40:36 +0800 Subject: [PATCH] fix: dataset retrieve npe when dataset desc is null (#669) --- api/core/tool/dataset_retriever_tool.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/api/core/tool/dataset_retriever_tool.py b/api/core/tool/dataset_retriever_tool.py index bf881159fa..d728528267 100644 --- a/api/core/tool/dataset_retriever_tool.py +++ b/api/core/tool/dataset_retriever_tool.py @@ -32,10 +32,11 @@ class DatasetRetrieverTool(BaseTool): @classmethod def from_dataset(cls, dataset: Dataset, **kwargs): - description = dataset.description.replace('\n', '').replace('\r', '') + description = dataset.description if not description: description = 'useful for when you want to answer queries about the ' + dataset.name + description = description.replace('\n', '').replace('\r', '') description += '\nID of dataset MUST be ' + dataset.id return cls( tenant_id=dataset.tenant_id,