diff --git a/vectorstore.py b/vectorstore.py index bcc74c4..b1b41de 100644 --- a/vectorstore.py +++ b/vectorstore.py @@ -64,13 +64,13 @@ def load_documents(directory: str, vectorstore: Chroma) -> None: # ------------------------------------------------------------------ # Инструмент web_search, использующий TavilySearchResults -def web_search(query: str) -> List[dict]: +def web_search(query: str) -> str: """ - Выполняет поиск по интернету через Tavily и возвращает список результатов. + Выполняет поиск по интернету через Tavily и возвращает строку с результатами. """ tavily = TavilySearchResults(api_key=TAVILY_API_KEY) results = tavily.run(query) - return results + return "\n".join([f"{r['title']}: {r['url']}" for r in results]) # ------------------------------------------------------------------ # Инструмент search_local_kb, использующий локальный Chroma