Se realiza una nueva implmentacion para la autenticacion con la api de Google.

This commit is contained in:
Mongar28
2024-11-27 02:13:30 -05:00
parent 86b3fc8d0e
commit 283da64cd3
16 changed files with 624 additions and 4640 deletions

View File

@@ -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