github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-16 07:24:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/summary' into dev

Browse Source
This commit is contained in:
aries-ckt 2023-06-01 20:44:15 +08:00
commit 43924cbff2
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pilot/source_embedding/knowledge_embedding.py
View File

@ -82,6 +82,9 @@ class KnowledgeEmbedding:
def similar_search(self, text, topk): def similar_search(self, text, topk):
return self.knowledge_embedding_client.similar_search(text, topk) return self.knowledge_embedding_client.similar_search(text, topk)
def vector_exist(self):
return self.knowledge_embedding_client.vector_name_exist()
def knowledge_persist_initialization(self, append_mode): def knowledge_persist_initialization(self, append_mode):
documents = self._load_knownlege(self.file_path) documents = self._load_knownlege(self.file_path)
self.vector_client = VectorStoreConnector( self.vector_client = VectorStoreConnector(