database/create_database.py

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