"""
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}")