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

fix: remove - in dataset retriever tool name (#3381)

Browse Source
This commit is contained in:
Yeuoly
2024-04-12 11:12:52 +08:00
committed by GitHub
parent f7a417fdb4
commit 44448ba68d
3 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/tools/tool/dataset_retriever/dataset_multi_retriever_tool.py
View File

@@ -31,7 +31,7 @@ class DatasetMultiRetrieverToolInput(BaseModel):
class DatasetMultiRetrieverTool(BaseTool):
"""Tool for querying multi dataset."""
name: str = "dataset-"
name: str = "dataset_"
args_schema: type[BaseModel] = DatasetMultiRetrieverToolInput
description: str = "dataset multi retriever and rerank. "
tenant_id: str
@@ -47,7 +47,7 @@ class DatasetMultiRetrieverTool(BaseTool):
@classmethod
def from_dataset(cls, dataset_ids: list[str], tenant_id: str, **kwargs):
return cls(
name=f'dataset-{tenant_id}',
name=f"dataset_{tenant_id.replace('-', '_')}",
tenant_id=tenant_id,
dataset_ids=dataset_ids,
**kwargs

2
api/core/tools/tool/dataset_retriever/dataset_retriever_tool.py
View File

@@ -46,7 +46,7 @@ class DatasetRetrieverTool(BaseTool):
description = description.replace('\n', '').replace('\r', '')
return cls(
name=f'dataset-{dataset.id}',
name=f"dataset_{dataset.id.replace('-', '_')}",
tenant_id=dataset.tenant_id,
dataset_id=dataset.id,
description=description,