From e47ef16a47bff39148edbd86b4d591b1092a8fb1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=AD=D0=BC=D0=B8=D0=BB=D1=8C=20=D0=90=D0=BC=D0=B8=D1=80?= =?UTF-8?q?=D0=BE=D0=B2?= Date: Wed, 27 May 2026 12:28:37 +0000 Subject: [PATCH] fix: align with accepted structure (init_knowledge.py) --- init_knowledge.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/init_knowledge.py b/init_knowledge.py index 262f202..4fc332a 100644 --- a/init_knowledge.py +++ b/init_knowledge.py @@ -4,7 +4,10 @@ from __future__ import annotations import argparse from pathlib import Path -from vector_store import add_document_to_store +try: + from knowledge_base import build_knowledge_base +except ModuleNotFoundError: + from .knowledge_base import build_knowledge_base SUPPORTED_SUFFIXES = {".txt", ".md", ".markdown"} @@ -14,13 +17,14 @@ def load_directory(directory: str | Path) -> int: if not root.exists(): raise FileNotFoundError(f"Директория не найдена: {root}") + kb = build_knowledge_base() total_chunks = 0 for path in sorted(root.rglob("*")): if not path.is_file() or path.suffix.lower() not in SUPPORTED_SUFFIXES: continue content = path.read_text(encoding="utf-8") title = str(path.relative_to(root)) - total_chunks += add_document_to_store(content, title) + total_chunks += kb.add_document(content, title) print(f"Загружен: {title} ({len(content)} символов)") return total_chunks