Доработка: Добавить в файл vectorstore.py реализацию инструмента `web

This commit is contained in:
2026-06-02 16:56:13 +00:00
parent 7caaee8ad9
commit f46fb9afbf
+3 -3
View File
@@ -64,13 +64,13 @@ def load_documents(directory: str, vectorstore: Chroma) -> None:
# ------------------------------------------------------------------ # ------------------------------------------------------------------
# Инструмент web_search, использующий TavilySearchResults # Инструмент 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) tavily = TavilySearchResults(api_key=TAVILY_API_KEY)
results = tavily.run(query) results = tavily.run(query)
return results return "\n".join([f"{r['title']}: {r['url']}" for r in results])
# ------------------------------------------------------------------ # ------------------------------------------------------------------
# Инструмент search_local_kb, использующий локальный Chroma # Инструмент search_local_kb, использующий локальный Chroma