github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2026-01-29 21:49:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

update:knowledge load script

Browse Source
This commit is contained in:
aries-ckt
2023-05-19 21:17:39 +08:00
parent 03f7ed32e5
commit 336ba1e042
9 changed files with 49 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tools/knowlege_init.py
View File

@@ -41,5 +41,4 @@ if __name__ == "__main__":
append_mode = args.append
kv = LocalKnowledgeInit()
vector_store = kv.knowledge_persist(file_path=DATASETS_DIR, vector_name=vector_name, append_mode=append_mode)
docs = vector_store.similarity_search("小明",1)
print("your knowledge embedding success...")