hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-09 19:06:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: move hardcoded text to language settings (#16990) (#17133)

Browse Source
This commit is contained in:
Perfecto
2025-04-02 22:35:51 +08:00
committed by GitHub
parent 5cffcd6336
commit 16c722d1d8
12 changed files with 40 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/rag/retrieval/dataset_retrieval.py
View File

@@ -52,6 +52,7 @@ from core.rag.retrieval.template_prompts import (
METADATA_FILTER_USER_PROMPT_2,
METADATA_FILTER_USER_PROMPT_3,
)
from core.tools.entities.common_entities import I18nObject
from core.tools.utils.dataset_retriever.dataset_retriever_base_tool import DatasetRetrieverBaseTool
from extensions.ext_database import db
from libs.json_in_md_parser import parse_and_check_json_markdown
@@ -294,7 +295,10 @@ class DatasetRetrieval:
for dataset in available_datasets:
description = dataset.description
if not description:
description = "useful for when you want to answer queries about the " + dataset.name
description = I18nObject(
en_US="useful for when you want to answer queries about the " + dataset.name,
zh_Hans="用于回答关于 " + dataset.name + " 的查询",
)
description = description.replace("\n", "").replace("\r", "")
message_tool = PromptMessageTool(

6
api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py
View File

@@ -6,6 +6,7 @@ from core.rag.datasource.retrieval_service import RetrievalService
from core.rag.entities.context_entities import DocumentContext
from core.rag.models.document import Document as RetrievalDocument
from core.rag.retrieval.retrieval_methods import RetrievalMethod
from core.tools.entities.common_entities import I18nObject
from core.tools.utils.dataset_retriever.dataset_retriever_base_tool import DatasetRetrieverBaseTool
from extensions.ext_database import db
from models.dataset import Dataset
@@ -38,7 +39,10 @@ class DatasetRetrieverTool(DatasetRetrieverBaseTool):
def from_dataset(cls, dataset: Dataset, **kwargs):
description = dataset.description
if not description:
description = "useful for when you want to answer queries about the " + dataset.name
description = I18nObject(
en_US="useful for when you want to answer queries about the " + dataset.name,
zh_Hans="用于回答关于 " + dataset.name + " 的查询",
)
description = description.replace("\n", "").replace("\r", "")
return cls(