| @@ -52,7 +52,6 @@ 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 | |||
| @@ -295,10 +294,7 @@ class DatasetRetrieval: | |||
| for dataset in available_datasets: | |||
| description = dataset.description | |||
| if not description: | |||
| description = I18nObject( | |||
| en_US="useful for when you want to answer queries about the " + dataset.name, | |||
| zh_Hans="用于回答关于 " + dataset.name + " 的查询", | |||
| ) | |||
| description = "useful for when you want to answer queries about the " + dataset.name | |||
| description = description.replace("\n", "").replace("\r", "") | |||
| message_tool = PromptMessageTool( | |||
| @@ -6,7 +6,6 @@ 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 | |||
| @@ -39,10 +38,7 @@ class DatasetRetrieverTool(DatasetRetrieverBaseTool): | |||
| def from_dataset(cls, dataset: Dataset, **kwargs): | |||
| description = dataset.description | |||
| if not description: | |||
| description = I18nObject( | |||
| en_US="useful for when you want to answer queries about the " + dataset.name, | |||
| zh_Hans="用于回答关于 " + dataset.name + " 的查询", | |||
| ) | |||
| description = "useful for when you want to answer queries about the " + dataset.name | |||
| description = description.replace("\n", "").replace("\r", "") | |||
| return cls( | |||
| @@ -21,7 +21,6 @@ from core.plugin.entities.plugin import ModelProviderID | |||
| from core.rag.index_processor.constant.built_in_field import BuiltInField | |||
| from core.rag.index_processor.constant.index_type import IndexType | |||
| from core.rag.retrieval.retrieval_methods import RetrievalMethod | |||
| from core.tools.entities.common_entities import I18nObject | |||
| from events.dataset_event import dataset_was_deleted | |||
| from events.document_event import document_was_deleted | |||
| from extensions.ext_database import db | |||
| @@ -1379,10 +1378,7 @@ class DocumentService: | |||
| cut_length = 18 | |||
| cut_name = documents[0].name[:cut_length] | |||
| dataset.name = cut_name + "..." | |||
| dataset.description = I18nObject( | |||
| en_US="useful for when you want to answer queries about the " + documents[0].name, | |||
| zh_Hans="用于回答关于 " + documents[0].name + " 的查询", | |||
| ) | |||
| dataset.description = "useful for when you want to answer queries about the " + documents[0].name | |||
| db.session.commit() | |||
| return dataset, documents, batch | |||