Доработка: 1) В agent.py реализовать функцию main() с циклом чтения пол

This commit is contained in:
2026-06-04 16:20:06 +00:00
parent a55a5e039e
commit b680e4edcb
+9 -1
View File
@@ -2,6 +2,11 @@ from langchain.tools import tool
from langchain_chroma import Chroma from langchain_chroma import Chroma
from langchain_ollama import OllamaEmbeddings from langchain_ollama import OllamaEmbeddings
from langchain_tavily import TavilySearchResults from langchain_tavily import TavilySearchResults
import os
from dotenv import load_dotenv
# Загружаем переменные окружения из .env
load_dotenv()
@tool @tool
def search_local_kb(query: str, top_k: int = 3) -> str: 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 @tool
def web_search(query: str) -> str: def web_search(query: str) -> str:
"""Поиск в интернете через Tavily.""" """Поиск в интернете через 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) results = search.run(query)
return "\n\n".join([result.get("content", "") for result in results]) return "\n\n".join([result.get("content", "") for result in results])