18 lines
489 B
Python
18 lines
489 B
Python
from langchain_chroma import Chroma
|
|
import os
|
|
|
|
|
|
def create_verctorstore(docs_split: list, embeddings, file_name: str):
|
|
db_name: str = file_name.replace(".pdf", "").replace(" ", "_").lower()
|
|
|
|
persist_directory: str = f"embeddings/{db_name}"
|
|
|
|
if not os.path.exists(persist_directory):
|
|
vectordb = Chroma.from_documents(
|
|
persist_directory=persist_directory,
|
|
documents=docs_split,
|
|
embedding=embeddings,
|
|
)
|
|
|
|
return vectordb
|