mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-10-04 00:27:40 +00:00
13 lines
516 B
Python
13 lines
516 B
Python
|
|
|
|
from pilot.source_embedding.csv_embedding import CSVEmbedding
|
|
|
|
# path = "/Users/chenketing/Downloads/share_ireserve双写数据异常2.xlsx"
|
|
path = "/Users/chenketing/Downloads/vectors.csv"
|
|
model_name = "/Users/chenketing/Desktop/project/all-MiniLM-L6-v2"
|
|
vector_store_path = "/pilot/source_embedding/"
|
|
|
|
|
|
pdf_embedding = CSVEmbedding(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") |