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

feat: knowledge management backend api

Browse Source 
1.create knowledge space
2.list knowledge space
3.create knowledge document
4.list knowledge document
5.save document chunks
6.sync embedding document
This commit is contained in:
aries_ckt
2023-06-26 15:24:25 +08:00
parent 364f251a12
commit db28894443
13 changed files with 648 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pilot/embedding_engine/knowledge_embedding.py
View File

@@ -48,9 +48,10 @@ class KnowledgeEmbedding:
def knowledge_embedding_batch(self, docs):
# docs = self.knowledge_embedding_client.read_batch()
self.knowledge_embedding_client.index_to_store(docs)
return self.knowledge_embedding_client.index_to_store(docs)
def read(self):
self.knowledge_embedding_client = self.init_knowledge_embedding()
return self.knowledge_embedding_client.read_batch()
def init_knowledge_embedding(self):
@@ -66,7 +67,7 @@ class KnowledgeEmbedding:
embedding = knowledge_class(
self.file_path,
vector_store_config=self.vector_store_config,
**knowledge_args,
**knowledge_args
)
return embedding
raise ValueError(f"Unsupported knowledge file type '{extension}'")