From eecbf6b5bb50361969b4746a2cd021c30bb5c561 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=A0=D0=B8=D0=BD=D0=B0=D1=80=20=D0=9C=D0=B8=D1=80=D0=B7?= =?UTF-8?q?=D0=B0=D0=B3=D0=B8=D1=82=D0=BE=D0=B2?= Date: Thu, 18 Jun 2026 09:07:36 +0000 Subject: [PATCH] =?UTF-8?q?=D0=A0=D0=B5=D1=88=D0=B5=D0=BD=D0=B8=D0=B5=20?= =?UTF-8?q?=D0=B3=D0=BE=D1=82=D0=BE=D0=B2=D0=BE=20=D0=BA=20=D0=BF=D1=83?= =?UTF-8?q?=D0=B1=D0=BB=D0=B8=D0=BA=D0=B0=D1=86=D0=B8=D0=B8:=20update=20to?= =?UTF-8?q?ols.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tools.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/tools.py b/tools.py index 7289501..71b8954 100644 --- a/tools.py +++ b/tools.py @@ -9,11 +9,11 @@ from vectorstore import create_vectorstore def _format_local_docs(results: list[Any]) -> str: if not results: - return "Source: qdrant\nNo relevant local documents found." + return "Источник: chromadb\nНет релевантных локальных документов." - parts = ["Source: qdrant"] + parts = ["Источник: chromadb"] for index, doc in enumerate(results, start=1): - source = doc.metadata.get("source", "unknown") + source = doc.metadata.get("source", "неизвестно") parts.append(f"{index}. ({source}) {doc.page_content}") return "\n".join(parts) @@ -21,8 +21,8 @@ def _format_local_docs(results: list[Any]) -> str: @tool def search_local_kb(query: str, top_k: int = 3) -> str: """Search the local ChromaDB knowledge base for internal course notes and local documents.""" - vectorstore = create_vectorstore() - retriever = vectorstore.as_retriever(search_kwargs={"k": top_k}) + collection = create_vectorstore() + retriever = collection.as_retriever(search_kwargs={"k": top_k}) results = retriever.invoke(query) return _format_local_docs(results) @@ -32,11 +32,11 @@ def web_search(query: str) -> str: """Search the web with Tavily for fresh facts, current news, or anything not covered by local documents.""" api_key = os.getenv("TAVILY_API_KEY") if not api_key: - return "Source: tavily\nTAVILY_API_KEY is not set, so web search is unavailable." + return "Источник: tavily\nTAVILY_API_KEY не задан, поиск в интернете недоступен." search_tool = TavilySearch( max_results=5, topic="general", ) result = search_tool.invoke({"query": query}) - return f"Source: tavily\n{result}" + return f"Источник: tavily\n{result}"