Se realiza una nueva implmentacion para la autenticacion con la api de Google.
This commit is contained in:
@@ -2,16 +2,18 @@ from langchain_chroma import Chroma
|
||||
import os
|
||||
|
||||
|
||||
def create_verctorstore(docs_split: list, embeddings, file_name: str):
|
||||
def create_vectorstore(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,
|
||||
)
|
||||
# Crear el directorio si no existe
|
||||
os.makedirs(persist_directory, exist_ok=True)
|
||||
|
||||
return vectordb
|
||||
# Siempre crear/actualizar el vectorstore
|
||||
vectordb = Chroma.from_documents(
|
||||
persist_directory=persist_directory,
|
||||
documents=docs_split,
|
||||
embedding=embeddings,
|
||||
)
|
||||
|
||||
return vectordb
|
||||
|
||||
Reference in New Issue
Block a user