File size: 950 Bytes
4022a3e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
"""Tools to search the web and extract information."""
import os
from tavily import AsyncTavilyClient
from llama_index.core.tools import FunctionTool
async def tavily_web_tool(query: str) -> str:
"""
Search the web with a given query and return the first two results.
If results do not return what you are looking for, rephrasing the query can help.
"""
client = AsyncTavilyClient(api_key=os.getenv("TAVILY"))
res = await client.search(query)
urls = [r.get("url") for r in res.get("results", []) if "url" in r][:2]
page_contents = {
url: await client.extract(url)
for url in urls
}
return "\n\n".join(
f"RESULT {i}:\n{content}"
for i, content in enumerate(page_contents.values())
)
def get_search_web_tool():
return FunctionTool.from_defaults(
fn=tavily_web_tool,
description="Search the web with a given query and return the first two results."
) |