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

feature:add markdown embedding

Browse Source
This commit is contained in:
chenketing 2023-05-14 23:27:17 +08:00
parent 293ba94d22
commit 645530a31d
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/knowledge_embedding/pdf_embedding_test.py
View File

@ -1,10 +1,11 @@
from pilot.source_embedding.pdf_embedding import PDFEmbedding from pilot.source_embedding.pdf_embedding import PDFEmbedding
path = "xxx.pdf" path = "xxx.pdf"
model_name = "/Users/chenketing/Desktop/project/all-MiniLM-L6-v2" path = "your_path/OceanBase-数据库-V4.1.0-应用开发.pdf"
vector_store_path = "/pilot/source_embedding/" model_name = "your_path/all-MiniLM-L6-v2"
vector_store_path = "your_path/"
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 = PDFEmbedding(file_path=path, model_name=model_name, vector_store_config={"vector_store_name": "ob-pdf", "vector_store_path": vector_store_path})
pdf_embedding.source_embedding() pdf_embedding.source_embedding()
print("success") print("success")