github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-10-24 19:08:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

5
dbgpt/serve/rag/assembler/summary.py
View File

@@ -32,7 +32,7 @@ class SummaryAssembler(BaseAssembler):
def __init__(
self,
knowledge: Knowledge = None,
knowledge: Knowledge,
chunk_parameters: Optional[ChunkParameters] = None,
model_name: Optional[str] = None,
llm_client: Optional[LLMClient] = None,
@@ -69,7 +69,7 @@ class SummaryAssembler(BaseAssembler):
@classmethod
def load_from_knowledge(
cls,
knowledge: Knowledge = None,
knowledge: Knowledge,
chunk_parameters: Optional[ChunkParameters] = None,
model_name: Optional[str] = None,
llm_client: Optional[LLMClient] = None,
@@ -104,6 +104,7 @@ class SummaryAssembler(BaseAssembler):
def persist(self) -> List[str]:
"""Persist chunks into store."""
raise NotImplementedError
def _extract_info(self, chunks) -> List[Chunk]:
"""Extract info from chunks."""