github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-10-26 12:20:39 +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

6
dbgpt/serve/rag/assembler/db_schema.py
View File

@@ -129,7 +129,11 @@ class DBSchemaAssembler(BaseAssembler):
return self._chunks
def persist(self) -> List[str]:
"""Persist chunks into vector store."""
"""Persist chunks into vector store.
Returns:
List[str]: List of chunk ids.
"""
return self._vector_store_connector.load_document(self._chunks)
def _extract_info(self, chunks) -> List[Chunk]: