diff --git a/main.py b/main.py index 7e59d3a..f9d2ded 100644 --- a/main.py +++ b/main.py @@ -10,8 +10,10 @@ def main(): persist_dir = "./chroma_db" vectorstore = create_vectorstore(persist_directory=persist_dir) - docs_dir = "./documents" - load_documents(docs_dir, vectorstore) + # Загружаем документы только если база пустая + if vectorstore._collection.count() == 0: + docs_dir = "./documents" + load_documents(docs_dir, vectorstore) agent_executor = create_agent(vectorstore)