21 lines
1004 B
Python
21 lines
1004 B
Python
"""Инициализация ChromaDB: загрузка документов из папки documents/."""
|
|
from pathlib import Path
|
|
from vectorstore import create_vectorstore, load_documents
|
|
|
|
CHROMA_DIR = "./chroma_db"
|
|
DOCS_DIR = "./documents"
|
|
|
|
if not Path(DOCS_DIR).exists():
|
|
Path(DOCS_DIR).mkdir(parents=True)
|
|
sample = Path(DOCS_DIR) / "intro.txt"
|
|
sample.write_text("LangChain — фреймворк для создания AI-приложений с LLM.
|
|
"
|
|
"RAG (Retrieval Augmented Generation) позволяет давать LLM доступ к внешним документам.
|
|
"
|
|
"ChromaDB — векторная база данных с открытым исходным кодом.", encoding="utf-8")
|
|
print(f"Создан тестовый файл: {sample}")
|
|
|
|
vs = create_vectorstore(CHROMA_DIR)
|
|
count = load_documents(DOCS_DIR, vs)
|
|
print(f"Загружено {count} чанков из {DOCS_DIR}/ в ChromaDB ({CHROMA_DIR})")
|