feat(rag): Support rag retriever evaluation (#1291)

This commit is contained in:
Fangyin Cheng
2024-03-14 13:06:57 +08:00
committed by GitHub
parent cd2dcc253c
commit adaa68eb00
34 changed files with 1452 additions and 67 deletions

View File

@@ -154,7 +154,7 @@ class Knowledge(ABC):
self._type = knowledge_type
self._data_loader = data_loader
def load(self):
def load(self) -> List[Document]:
"""Load knowledge from data_loader."""
documents = self._load()
return self._postprocess(documents)
@@ -174,7 +174,7 @@ class Knowledge(ABC):
return docs
@abstractmethod
def _load(self):
def _load(self) -> List[Document]:
"""Preprocess knowledge from data_loader."""
@classmethod