Доработка: 1) В agent.py реализовать функцию main() с циклом чтения пол
This commit is contained in:
@@ -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])
|
||||||
Reference in New Issue
Block a user