github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-14 06:26:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:bind knowledge_init with space

Browse Source 
bind knowledge_init with space
This commit is contained in:
aries_ckt 2023-06-30 11:31:47 +08:00
parent b074c487e9
commit 9a32ac7e3d
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tools/knowlege_init.py
View File

@ -5,6 +5,8 @@ import os
import sys import sys
from pilot.embedding_engine.knowledge_type import KnowledgeType from pilot.embedding_engine.knowledge_type import KnowledgeType
from pilot.openapi.knowledge.knowledge_service import KnowledgeService
from pilot.openapi.knowledge.request.knowledge_request import KnowledgeSpaceRequest
sys.path.append(os.path.dirname(os.path.dirname(os.path.realpath(__file__)))) sys.path.append(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))
@ -14,6 +16,7 @@ from pilot.configs.model_config import (
LLM_MODEL_CONFIG, LLM_MODEL_CONFIG,
) )
from pilot.embedding_engine.knowledge_embedding import KnowledgeEmbedding from pilot.embedding_engine.knowledge_embedding import KnowledgeEmbedding
knowledge_space_service = KnowledgeService()
CFG = Config() CFG = Config()
@ -23,7 +26,7 @@ class LocalKnowledgeInit:
def __init__(self, vector_store_config) -> None: def __init__(self, vector_store_config) -> None:
self.vector_store_config = vector_store_config self.vector_store_config = vector_store_config
self.model_name = LLM_MODEL_CONFIG["text2vec"] self.model_name = LLM_MODEL_CONFIG[CFG.EMBEDDING_MODEL]
def knowledge_persist(self, file_path): def knowledge_persist(self, file_path):
"""knowledge persist""" """knowledge persist"""
@ -39,6 +42,12 @@ class LocalKnowledgeInit:
) )
client = ke.init_knowledge_embedding() client = ke.init_knowledge_embedding()
client.source_embedding() client.source_embedding()
print(f"""begin create {self.vector_store_config["vector_store_name"]} space""")
space = KnowledgeSpaceRequest
space.name = self.vector_store_config["vector_store_name"]
space.desc = ""
space.owner = "knowledge_init.py"
knowledge_space_service.create_knowledge_space(space)
if __name__ == "__main__": if __name__ == "__main__":