|
|
|
|
@ -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(
|
|
|
|
|
|