17 lines
611 B
Python
17 lines
611 B
Python
"""
|
|
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}") |