Доработка: Добавить в файл vectorstore.py реализацию инструмента `web
This commit is contained in:
+3
-3
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user