Spaces:
Sleeping
Sleeping
| from langchain_chroma import Chroma | |
| from utils.load_llm import TEXT_SPLITTER, EMBEDDING_MODEL | |
| from config.nodes import PERSIST_DIRECTORY | |
| def create_or_add_to_collection(collection_name: str, docs): | |
| texts = TEXT_SPLITTER.split_documents(docs) | |
| database_chroma = Chroma( | |
| collection_name=collection_name, | |
| persist_directory=PERSIST_DIRECTORY, | |
| embedding_function=EMBEDDING_MODEL | |
| ) | |
| database_chroma.add_documents(texts) | |
| return database_chroma |