AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-11 01:18:58 +08:00
Code Issues 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats 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,