estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix: dataset retrieve npe when dataset desc is null (#669)

Browse Source
This commit is contained in:
John Wang
2023-07-28 17:40:36 +08:00
committed by GitHub
parent 23e52f14e3
commit aae2fb8a30
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/tool/dataset_retriever_tool.py
View File

@@ -32,10 +32,11 @@ class DatasetRetrieverTool(BaseTool):
@classmethod @classmethod
def from_dataset(cls, dataset: Dataset, **kwargs): def from_dataset(cls, dataset: Dataset, **kwargs):
description = dataset.description.replace('\n', '').replace('\r', '') description = dataset.description
if not description: if not description:
description = 'useful for when you want to answer queries about the ' + dataset.name 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 description += '\nID of dataset MUST be ' + dataset.id
return cls( return cls(
tenant_id=dataset.tenant_id, tenant_id=dataset.tenant_id,