diff --git a/init_knowledge_base.py b/init_knowledge_base.py new file mode 100644 index 0000000..955824c --- /dev/null +++ b/init_knowledge_base.py @@ -0,0 +1,17 @@ +""" +init_knowledge_base.py — скрипт инициализации базы знаний из директории. + +Использование: + python init_knowledge_base.py [директория] + +По умолчанию — папка ./docs +""" + +import sys +from rag_agent import load_documents_from_directory + +if __name__ == "__main__": + directory = sys.argv[1] if len(sys.argv) > 1 else "./docs" + print(f"Загрузка документов из «{directory}»...") + count = load_documents_from_directory(directory) + print(f"\nГотово. Загружено файлов: {count}") \ No newline at end of file