From 645530a31de4efd54448bddff0b6656a436caf7e Mon Sep 17 00:00:00 2001 From: chenketing Date: Sun, 14 May 2023 23:27:17 +0800 Subject: [PATCH] feature:add markdown embedding --- examples/knowledge_embedding/pdf_embedding_test.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/examples/knowledge_embedding/pdf_embedding_test.py b/examples/knowledge_embedding/pdf_embedding_test.py index 03aa35d23..6c3f3588e 100644 --- a/examples/knowledge_embedding/pdf_embedding_test.py +++ b/examples/knowledge_embedding/pdf_embedding_test.py @@ -1,10 +1,11 @@ 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/" +path = "your_path/OceanBase-数据库-V4.1.0-应用开发.pdf" +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() print("success") \ No newline at end of file