tj dZddlmZmZmZmZmZmZmZm Z ddl m Z ddl m Z mZddlmZddlmZmZddlmZGdd eZe d d d GddeZe d d d GddeZdS)zTool for the Tavily search API.)AnyDictListLiteralOptionalTupleTypeUnion) deprecated)AsyncCallbackManagerForToolRunCallbackManagerForToolRun)BaseTool) BaseModelField)TavilySearchAPIWrapperc6eZdZUdZedZeed<dS) TavilyInputzInput for the Tavily tool.zsearch query to look up) descriptionqueryN)__name__ __module__ __qualname____doc__rrstr__annotations__j/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain_community/tools/tavily_search/tool.pyrrs6$$#<===E3=====rrz0.3.25z1.0zlangchain_tavily.TavilySearch)sinceremovalalternative_importc eZdZUdZdZeed<dZeed<eZ e e ed< dZ e ed< d Zeed < gZeeed < gZeeed < d Zeed< d Zeed< d Zeed< eeZeed<dZeded<deddffd Z ddedeedee ee!eefefe!ffdZ" ddedee#dee ee!eefefe!ffdZ$xZ%S)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_formatkwargsreturnNc td|vrt|d|d<tjdi|dS)Ntavily_api_key)r8r2r)rsuper__init__)selfr5 __class__s rr:zTavilySearchResults.__init__sU v % %$:%&67%%%F= ! ""6"""""rr run_managerc " |j||j|j|j|j|j|j|j}n(#t$r}t|ifcYd}~Sd}~wwxYw|j |d|fS) Use the tool.Nresults) r2 raw_resultsr(r*r+r,r-r.r/ Exceptionrepr clean_resultsr;rr=rAes r_runzTavilySearchResults._runs *66 !$$#(#  KK   77B;       --k).DEE{RRsAA A,A'!A,'A,c 2K |j||j|j|j|j|j|j|jd{V}n(#t$r}t|ifcYd}~Sd}~wwxYw|j |d|fS)Use the tool asynchronously.Nr@) r2raw_results_asyncr(r*r+r,r-r.r/rBrCrDrEs r_arunzTavilySearchResults._aruns  $ 0 B B !$$#(# ! !      KK   77B;       --k).DEE{RRsA A A4A/)A4/A4N)&rrrrr%rrrrr&r rr(intr*r+rr,r-boolr.r/rrr2r4rrr:rr rr rrGr rK __classcell__)r<s@rr#r#sT IIV-D#,,, * $/Ki...#K4"L#""""$OT#Y### "$OT#Y### !ND    !&%%% !ND   +0%@V*W*W*WK'WWW7MOW34MMM########<@SSS78S uT$sCx.)3./5 6 SSSS2AESSS<=S uT$sCx.)3./5 6 SSSSSSSSrr#c eZdZUdZdZeed<dZeed<ee Z e ed<e Z e eed< dd ed eed eeeeffd Z dd ed eed eeeeffdZd S) 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.rr0r2r&Nrr=r6c |j|ddddS#t$r}t|cYd}~Sd}~wwxYw)r?r'Tbasicr(r-r*answerN)r2rArBrC)r;rr=rFs rrGzTavilyAnswer._runss  #//#$ 0       77NNNNNN s#& A AA A cK |j|dddd{V}|dS#t$r}t|cYd}~Sd}~wwxYw)rIr'TrTrUNrV)r2rJrBrC)r;rr=resultrFs rrKzTavilyAnswer._aruns +==#$ >F (# #   77NNNNNN s+0 AAAArL)rrrrr%rrrrrr2rr&r rrr r rrrGr rKrrrrQrQs KJD# G +0%@V*W*W*WK'WWW#.Ki... <@78 tDz3  &AE<= tDz3  rrQN)rtypingrrrrrrr r langchain_core._apir langchain_core.callbacksr r langchain_core.toolsrpydanticrr+langchain_community.utilities.tavily_searchrrr#rQrrrr_s%%IIIIIIIIIIIIIIIIIIII*******)))))%%%%%%%%NNNNNN>>>>>)>>>    6 uSuSuSuSuS(uSuS  uSp   6 ,,,,,8,,  ,,,r