feat: init_db script for documents/ folder
This commit is contained in:
+20
@@ -0,0 +1,20 @@
|
||||
"""Инициализация 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})")
|
||||
Reference in New Issue
Block a user