From b680e4edcb334f730da4d792a28f43bff7fd7665 Mon Sep 17 00:00:00 2001
From: lonpatovaadelina <lonpatovaadelina@mail.ru>
Date: Thu, 4 Jun 2026 16:20:06 +0000
Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D1=80=D0=B0=D0=B1=D0=BE=D1=82?=
 =?UTF-8?q?=D0=BA=D0=B0:=201)=20=D0=92=20agent.py=20=D1=80=D0=B5=D0=B0?=
 =?UTF-8?q?=D0=BB=D0=B8=D0=B7=D0=BE=D0=B2=D0=B0=D1=82=D1=8C=20=D1=84=D1=83?=
 =?UTF-8?q?=D0=BD=D0=BA=D1=86=D0=B8=D1=8E=20main()=20=D1=81=20=D1=86=D0=B8?=
 =?UTF-8?q?=D0=BA=D0=BB=D0=BE=D0=BC=20=D1=87=D1=82=D0=B5=D0=BD=D0=B8=D1=8F?=
 =?UTF-8?q?=20=D0=BF=D0=BE=D0=BB?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tools.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/tools.py b/tools.py
index c4373f9..3c21352 100644
--- a/tools.py
+++ b/tools.py
@@ -2,6 +2,11 @@ from langchain.tools import tool
 from langchain_chroma import Chroma
 from langchain_ollama import OllamaEmbeddings
 from langchain_tavily import TavilySearchResults
+import os
+from dotenv import load_dotenv
+
+# Загружаем переменные окружения из .env
+load_dotenv()
 
 @tool
 def search_local_kb(query: str, top_k: int = 3) -> str:
@@ -16,6 +21,9 @@ def search_local_kb(query: str, top_k: int = 3) -> str:
 @tool
 def web_search(query: str) -> str:
     """Поиск в интернете через Tavily."""
-    search = TavilySearchResults()
+    tavily_api_key = os.getenv("TAVILY_API_KEY")
+    if not tavily_api_key:
+        raise ValueError("Требуется переменная окружения TAVILY_API_KEY")
+    search = TavilySearchResults(tavily_api_key=tavily_api_key)
     results = search.run(query)
     return "\n\n".join([result.get("content", "") for result in results])
\ No newline at end of file