{hdZddlmZmZmZmZmZmZmZddl m Z m Z ddl m Z ddlmZmZddlmZGddeZGd d e ZGd d e Zy )zTool for the Tavily search API.)DictListLiteralOptionalTupleTypeUnion)AsyncCallbackManagerForToolRunCallbackManagerForToolRun)BaseTool) BaseModelField)TavilySearchAPIWrapperc.eZdZUdZedZeed<y) TavilyInputzInput for the Tavily tool.zsearch query to look up) descriptionqueryN)__name__ __module__ __qualname____doc__rrstr__annotations__f/var/www/html/engine/venv/lib/python3.12/site-packages/langchain_community/tools/tavily_search/tool.pyrrs$#<=E3=rrc teZdZUdZdZeed<dZeed<eZ e e ed< dZ e ed< d Zeed < gZeeed < gZeeed < d Zeed< d Zeed< d Zeed< eeZeed<dZeded< ddedeedeeeeeefefeffdZ ddedee!deeeeeefefeffdZ"y)TavilySearchResultsa 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:: json { "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', ) tavily_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_resultsadvanced search_depthinclude_domainsexclude_domainsFinclude_answerinclude_raw_contentinclude_imagesdefault_factory api_wrappercontent_and_artifactresponse_formatNr run_managerreturnc Z |jj||j|j|j|j |j |j|j}|jj|d|fS#t$r}t|ifcYd}~Sd}~wwxYw) Use the tool.Nresults) r- raw_resultsr#r%r&r'r(r)r* Exceptionrepr clean_resultsselfrr0r5es r_runzTavilySearchResults._runs **66  !!$$$$##((## K--k).DE{RR 7B;  sA(B B* B%B*%B*c vK |jj||j|j|j|j |j |j|jd{}|jj|d|fS7$#t$r}t|ifcYd}~Sd}~wwxYww)Use the tool asynchronously.Nr4) r-raw_results_asyncr#r%r&r'r(r)r*r6r7r8r9s r_arunzTavilySearchResults._aruns  $ 0 0 B B  !!$$$$##((## ! K--k).DE{RR  7B;  sGB9A+B/B0B4 B9B B6 B1+B6,B91B66B9N)#rrrrr rrrrr!rr r#intr%r&rr'r(boolr)r*rrr-r/rrr rr rr<r r@rrrrrs{IV-D#, * $/Ki.#K4"L#""$OT#Y# "$OT#Y# !ND  !&% !ND +0@V*WK'W7MOW34M <@SS78S uT$sCx.)3./5 6 S2AESS<=S uT$sCx.)3./5 6 Srrc eZdZUdZdZeed<dZeed<ee Z e ed<e Z e eed< dd ed eed eeeeffd Z dd ed eed eeeeffdZy ) TavilyAnswerz@Tool that queries the Tavily Search API and gets back an answer. tavily_answerr zA 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-r!Nrr0r1c |jj|ddddS#t$r}t|cYd}~Sd}~wwxYw)r3r"Tbasicr#r(r%answerN)r-r5r6r7)r:rr0r;s rr<zTavilyAnswer._runsV  ##//#$ 0     7N s!$ A =AAcK |jj|dddd{}|dS7 #t$r}t|cYd}~Sd}~wwxYww)r>r"TrHrINrJ)r-r?r6r7)r:rr0resultr;s rr@zTavilyAnswer._arunsc ++==#$ >F (# #  7N s<A"202A2 A A AA AArA)rrrrr rrrrrr-rr!rr rr r rrr<r r@rrrrErEsJD# G +0@V*WK'W#.Ki. <@78 tDz3  &AE<= tDz3  rrEN)rtypingrrrrrrr langchain_core.callbacksr r langchain_core.toolsr pydanticr r+langchain_community.utilities.tavily_searchrrrrErrrrRsJ%DDD*%N>)> lS(lS^,8,r