a bg@sdZddlmZmZmZmZmZmZmZddl m Z m Z ddl m Z ddlmZmZddlmZGdddeZGd d d e ZGd d d e Zd S)zTool for the Tavily search API.)DictListLiteralOptionalTupleTypeUnion)AsyncCallbackManagerForToolRunCallbackManagerForToolRun)BaseTool) BaseModelField)TavilySearchAPIWrapperc@s$eZdZUdZeddZeed<dS) TavilyInputzInput for the Tavily tool.zsearch query to look up) descriptionqueryN)__name__ __module__ __qualname____doc__r rstr__annotations__rrz/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/tools/tavily_search/tool.pyrs rc @s eZdZUdZdZeed<dZeed<eZ e e ed<dZ e ed<d Zeed <gZeeed <gZeeed <d Zeed<d Zeed<d Zeed<eedZeed<dZeded<deeeeeeeeefefefdddZ deee!eeeeeefefefdddZ"dS)TavilySearchResultsan Tool that queries the Tavily Search API and gets back json. Setup: Install ``langchain-openai`` and ``tavily-python``, and set environment variable ``TAVILY_API_KEY``. .. code-block:: bash pip install -U langchain-community tavily-python export TAVILY_API_KEY="your-api-key" Instantiate: .. code-block:: python from langchain_community.tools import TavilySearchResults tool = TavilySearchResults( max_results=5, include_answer=True, include_raw_content=True, include_images=True, # search_depth="advanced", # include_domains = [] # exclude_domains = [] ) Invoke directly with args: .. code-block:: python tool.invoke({'query': 'who won the last french open'}) .. code-block:: python '{ "url": "https://www.nytimes.com...", "content": "Novak Djokovic won the last French Open by beating Casper Ruud ...' Invoke with tool call: .. code-block:: python tool.invoke({"args": {'query': 'who won the last french open'}, "type": "tool_call", "id": "foo", "name": "tavily"}) .. code-block:: python ToolMessage( content='{ "url": "https://www.nytimes.com...", "content": "Novak Djokovic won the last French Open by beating Casper Ruud ...', artifact={ 'query': 'who won the last french open', 'follow_up_questions': None, 'answer': 'Novak ...', 'images': [ 'https://www.amny.com/wp-content/uploads/2023/06/AP23162622181176-1200x800.jpg', ... ], 'results': [ { 'title': 'Djokovic ...', 'url': 'https://www.nytimes.com...', 'content': "Novak...", 'score': 0.99505633, 'raw_content': 'Tennis Novak ...' }, ... ], 'response_time': 2.92 }, tool_call_id='1', name='tavily_search_results_json', ) Ztavily_search_results_jsonnamezA search engine optimized for comprehensive, accurate, and trusted results. Useful for when you need to answer questions about current events. Input should be a search query.r args_schema max_resultsZadvanced search_depthinclude_domainsexclude_domainsFinclude_answerinclude_raw_contentinclude_imagesdefault_factory api_wrapperZcontent_and_artifactresponse_formatNr run_managerreturnc Csrz,|j||j|j|j|j|j|j|j}Wn0t y\}zt |ifWYd}~Sd}~00|j |d|fS) Use the tool.Nresults) r' raw_resultsrrr r!r"r#r$ Exceptionrepr clean_resultsselfrr*r.errr_runs "zTavilySearchResults._runc sxz2|j||j|j|j|j|j|j|jIdH}Wn0t yb}zt |ifWYd}~Sd}~00|j |d|fS)Use the tool asynchronously.Nr-) r'raw_results_asyncrrr r!r"r#r$r/r0r1r2rrr_aruns "zTavilySearchResults._arun)N)N)#rrrrrrrrrrrr rintrr rr!r"boolr#r$r rr'r(rrr rrrr5r r8rrrrrs4 H        rc@seZdZUdZdZeed<dZeed<ee dZ e ed<e Z e eed<deeeeeeefd d d Zdeeeeeeefd d dZd S) TavilyAnswerz@Tool that queries the Tavily Search API and gets back an answer.Z tavily_answerrzA search engine optimized for comprehensive, accurate, and trusted results. Useful for when you need to answer questions about current events. Input should be a search query. This returns only the answer - not the original source data.rr%r'rNr)c CsLz|jj|dddddWStyF}zt|WYd}~Sd}~00dS)r,rTbasicrr"ranswerN)r'r.r/r0)r3rr*r4rrrr5szTavilyAnswer._runc sVz$|jj|ddddIdH}|dWStyP}zt|WYd}~Sd}~00dS)r6rTr<r=Nr>)r'r7r/r0)r3rr*resultr4rrrr8s  zTavilyAnswer._arun)N)N)rrrrrrrrr rr'rrrr rr rrrr5r r8rrrrr;s$    r;N)rtypingrrrrrrrZlangchain_core.callbacksr r Zlangchain_core.toolsr Zpydanticr r Z+langchain_community.utilities.tavily_searchrrrr;rrrrs$  -