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

10
dbgpt/serve/rag/assembler/embedding.py
View File

@@ -29,7 +29,7 @@ class EmbeddingAssembler(BaseAssembler):
def __init__(
self,
knowledge: Knowledge = None,
knowledge: Knowledge,
chunk_parameters: Optional[ChunkParameters] = None,
embedding_model: Optional[str] = None,
embedding_factory: Optional[EmbeddingFactory] = None,
@@ -69,7 +69,7 @@ class EmbeddingAssembler(BaseAssembler):
@classmethod
def load_from_knowledge(
cls,
knowledge: Knowledge = None,
knowledge: Knowledge,
chunk_parameters: Optional[ChunkParameters] = None,
embedding_model: Optional[str] = None,
embedding_factory: Optional[EmbeddingFactory] = None,
@@ -99,7 +99,11 @@ class EmbeddingAssembler(BaseAssembler):
)
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]: