瀏覽代碼

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

John Wang 1 年之前
父節點
當前提交
aae2fb8a30
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      api/core/tool/dataset_retriever_tool.py

+ 2 - 1
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,