Files
DB-GPT/examples/knowledge_embedding/url_embedding_test.py
2023-05-15 00:18:06 +08:00

10 lines
457 B
Python

from pilot.source_embedding.url_embedding import URLEmbedding
path = "https://www.understandingwar.org/backgrounder/russian-offensive-campaign-assessment-february-8-2023"
model_name = "your_path/all-MiniLM-L6-v2"
vector_store_path = "your_path"
pdf_embedding = URLEmbedding(file_path=path, model_name=model_name, vector_store_config={"vector_store_name": "url", "vector_store_path": "vector_store_path"})
pdf_embedding.source_embedding()
print("success")