o i @sddlmZddlmZddlmZddlmZmZddl m Z zddl m Z Wne y9Zze dedZ[wwd ZGd d d eZeeeZeGd d d ZdefddZdS)) dataclass)Literal) TypeAdapter)Any TypedDict)Tool)AsyncTavilyClientuPlease install `tavily-python` to use the Tavily search tool, you can use the `tavily` optional group — `pip install "pydantic-ai-slim[tavily]"`N)tavily_search_toolc@s8eZdZUdZeed< eed< eed< eed<dS)TavilySearchResultzA Tavily search result. See [Tavily Search Endpoint documentation](https://docs.tavily.com/api-reference/endpoint/search) for more information. titleurlcontentscoreN)__name__ __module__ __qualname____doc__str__annotations__floatrre/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/pydantic_ai/common_tools/tavily.pyr s r c @sNeZdZUdZeed<   ddededd ed d ed dBfd dZdS)TavilySearchToolzThe Tavily search tool.clientbasicgeneralNquery search_deep)radvancedtopic)rnews time_range)dayweekmonthyeardwmycs*|jj||||dIdH}t|dS)axSearches Tavily for the given query and returns the results. Args: query: The search query to execute with Tavily. search_deep: The depth of the search. topic: The category of the search. time_range: The time range back from the current date to filter results. Returns: The search results. ) search_depthrr!Nresults)rsearchtavily_search_tavalidate_python)selfrrrr!r+rrr__call__/szTavilySearchTool.__call__)rrN) rrrrrrrrr0rrrrr(s  rapi_keycCstttt|djdddS)zCreates a Tavily search tool. Args: api_key: The Tavily API key. You can get one by signing up at [https://app.tavily.com/home](https://app.tavily.com/home). )r tavily_searchzs*