from setup.easy_imports import Chroma
from _utils.langchain_utils.embeddings import embedding


class VectorStoreClass:
    def chroma(self, lista_de_documents):
        return Chroma.from_documents(lista_de_documents, embedding.open_ai())


vector_store = VectorStoreClass()