github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-10 13:29:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor:rename knowledge embedding api

Browse Source 
1.replace knowledge_embedding of embedding_engine
This commit is contained in:
aries_ckt
2023-07-10 15:50:34 +08:00
parent 877db265ef
commit 56b32ab094
15 changed files with 28 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tools/knowledge_init.py
View File

@@ -16,7 +16,7 @@ from pilot.configs.model_config import (
DATASETS_DIR,
LLM_MODEL_CONFIG,
)
from pilot.embedding_engine.knowledge_embedding import KnowledgeEmbedding
from pilot.embedding_engine.embedding_engine import EmbeddingEngine
knowledge_space_service = KnowledgeService()
@@ -37,7 +37,7 @@ class LocalKnowledgeInit:
for root, _, files in os.walk(file_path, topdown=False):
for file in files:
filename = os.path.join(root, file)
ke = KnowledgeEmbedding(
ke = EmbeddingEngine(
knowledge_source=filename,
knowledge_type=KnowledgeType.DOCUMENT.value,
model_name=self.model_name,