From 1d9dbd6b308f0576f34beac3b7e5d4a498e6107d 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?= <mirzagitov03@mail.ru>
Date: Wed, 17 Jun 2026 13:13:22 +0000
Subject: [PATCH] =?UTF-8?q?=D0=9F=D1=83=D0=B1=D0=BB=D0=B8=D0=BA=D0=B0?=
 =?UTF-8?q?=D1=86=D0=B8=D1=8F=20=D0=BF=D0=BE=D0=BB=D0=BD=D0=BE=D0=B3=D0=BE?=
 =?UTF-8?q?=20=D0=BD=D0=B0=D0=B1=D0=BE=D1=80=D0=B0=20=D1=84=D0=B0=D0=B9?=
 =?UTF-8?q?=D0=BB=D0=BE=D0=B2:=20add=20tools.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tools.py | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 tools.py

diff --git a/tools.py b/tools.py
new file mode 100644
index 0000000..7289501
--- /dev/null
+++ b/tools.py
@@ -0,0 +1,42 @@
+import os
+from typing import Any
+
+from langchain.tools import tool
+from langchain_tavily import TavilySearch
+
+from vectorstore import create_vectorstore
+
+
+def _format_local_docs(results: list[Any]) -> str:
+    if not results:
+        return "Source: qdrant\nNo relevant local documents found."
+
+    parts = ["Source: qdrant"]
+    for index, doc in enumerate(results, start=1):
+        source = doc.metadata.get("source", "unknown")
+        parts.append(f"{index}. ({source}) {doc.page_content}")
+    return "\n".join(parts)
+
+
+@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})
+    results = retriever.invoke(query)
+    return _format_local_docs(results)
+
+
+@tool
+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."
+
+    search_tool = TavilySearch(
+        max_results=5,
+        topic="general",
+    )
+    result = search_tool.invoke({"query": query})
+    return f"Source: tavily\n{result}"