github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-10-31 14:56:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat:knowledge rag graph

Browse Source
This commit is contained in:
aries_ckt
2023-10-10 20:25:51 +05:00
parent 813e2260a6
commit fc656e1c61
7 changed files with 58 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/embedding_engine/source_embedding.py
View File

@@ -31,11 +31,11 @@ class SourceEmbedding(ABC):
):
"""Initialize with Loader url, model_name, vector_store_config"""
self.file_path = file_path
self.vector_store_config = vector_store_config
self.vector_store_config = vector_store_config or {}
self.source_reader = source_reader or None
self.text_splitter = text_splitter or None
self.embedding_args = embedding_args
self.embeddings = vector_store_config["embeddings"]
self.embeddings = self.vector_store_config.get("embeddings", None)
@abstractmethod
@register