github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-03 18:17:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

feature:pdf embedding test

Browse Source
This commit is contained in:
chenketing
2023-05-11 20:28:15 +08:00
parent 0c241dfaad
commit ed855df01d
5 changed files with 11 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
examples/knowledge_embedding/pdf_embedding_test.py Normal file
View File

@@ -0,0 +1,10 @@
from pilot.source_embedding.pdf_embedding import PDFEmbedding
path = "xxx.pdf"
model_name = "/Users/chenketing/Desktop/project/all-MiniLM-L6-v2"
vector_store_path = "/pilot/source_embedding/"
pdf_embedding = PDFEmbedding(file_path=path, model_name=model_name, vector_store_config={"vector_store_name": "ob", "vector_store_path": "vector_store_path"})
pdf_embedding.source_embedding()
print("success")