diff --git a/rag_tools.py b/rag_tools.py index ae7a062..995c18b 100644 --- a/rag_tools.py +++ b/rag_tools.py @@ -3,10 +3,7 @@ from __future__ import annotations from langchain.tools import tool -try: - from knowledge_base import KnowledgeBase, build_knowledge_base -except ModuleNotFoundError: - from .knowledge_base import KnowledgeBase, build_knowledge_base +from knowledge_base import KnowledgeBase, build_knowledge_base _kb: KnowledgeBase | None = None @@ -24,8 +21,8 @@ def search_knowledge_base(query: str, max_results: int = 5) -> str: """Семантический поиск в базе знаний. Args: - query: поисковый запрос - max_results: максимум фрагментов в ответе + query: Поисковый запрос на естественном языке. + max_results: Максимум результатов (по умолчанию 5). """ results = get_knowledge_base().search(query, max_results=max_results) if not results: @@ -40,11 +37,11 @@ def search_knowledge_base(query: str, max_results: int = 5) -> str: @tool def add_to_knowledge_base(content: str, title: str) -> str: - """Добавляет документ в векторную базу знаний. + """Добавление документа в базу знаний. Args: - content: текст документа - title: заголовок / имя документа + content: Текст документа. + title: Заголовок / имя документа. """ if not content.strip(): return "Ошибка: пустой content."