o ȳg@spdZddlZddlmZmZmZmZddlZddlZddl m Z ddl m Z m Z mZmZdZGddde ZdS) zUtil that calls Tavily Search API. In order to set this up, follow instructions at: https://docs.tavily.com/docs/tavily-api/introduction N)AnyDictListOptionalget_from_dict_or_env) BaseModel ConfigDict SecretStrmodel_validatorzhttps://api.tavily.comc@seZdZUdZeed<eddZedde de de fd d Z d d ggd d d fde deedee deee deee deedeedeede fddZd d ggd d d fde deedee deee deee deedeedeedee fddZd d ggd d d fde deedee deee deee deedeedeede fddZd d ggd d d fde deedee deee deee deedeedeedee fddZdee dee fdd Zd!S)"TavilySearchAPIWrapperzWrapper for Tavily Search API.tavily_api_keyforbid)extrabefore)modevaluesreturncCst|dd}||d<|S)z9Validate that api key and endpoint exists in environment.r TAVILY_API_KEYr)clsrr rg/var/www/html/chatdoc2/venv/lib/python3.10/site-packages/langchain_community/utilities/tavily_search.pyvalidate_environments z+TavilySearchAPIWrapper.validate_environmentadvancedFquery max_results search_depthinclude_domainsexclude_domainsinclude_answerinclude_raw_contentinclude_imagesc CsB|j||||||||d } tjtd| d} | | S)N api_keyrrrrrr r!r"/searchjson)r get_secret_valuerequestspostTAVILY_API_URLraise_for_statusr') selfrrrrrr r!r"paramsresponserrr raw_results&s   z"TavilySearchAPIWrapper.raw_resultsc Cs(|j||||||||d} || dS)aRun query through Tavily Search and return metadata. Args: query: The query to search for. max_results: The maximum number of results to return. search_depth: The depth of the search. Can be "basic" or "advanced". include_domains: A list of domains to include in the search. exclude_domains: A list of domains to exclude from the search. include_answer: Whether to include the answer in the results. include_raw_content: Whether to include the raw content in the results. include_images: Whether to include images in the results. Returns: query: The query that was searched for. follow_up_questions: A list of follow up questions. response_time: The response time of the query. answer: The answer to the query. images: A list of images. results: A list of dictionaries containing the results: title: The title of the result. url: The url of the result. content: The content of the result. score: The score of the result. raw_content: The raw content of the result. )rrrrr r!r"results)r0 clean_results) r-rrrrrr r!r"raw_search_resultsrrrr1Ds# zTavilySearchAPIWrapper.resultsc s:dtff dd } | IdH} t| S)z6Get results from the Tavily Search API asynchronously.rc sjd }t4IdHT}|jtd|d4IdH/}|jdkrJ|IdH}|WdIdHWdIdHStd|jd|j 1IdHs\wYWdIdHdS1IdHsrwYdS)Nr#r%r&zError z: ) r r(aiohttp ClientSessionr*r+statustext Exceptionreason)r.sessionresdata rr rr"r!rrrr-rrfetchs*  .z7TavilySearchAPIWrapper.raw_results_async..fetchN)strr'loads) r-rrrrrr r!r"r?results_json_strrr>rraw_results_asyncss"  z(TavilySearchAPIWrapper.raw_results_asyncc s0|j||||||||dIdH} || dS)N)rrrrrr r!r"r1)rCr2) r-rrrrrr r!r" results_jsonrrr results_asyncs   z$TavilySearchAPIWrapper.results_asyncr1cCs*g}|D]}||d|ddq|S)z%Clean results from Tavily Search API.urlcontent)rFrG)append)r-r1r2resultrrrr2sz$TavilySearchAPIWrapper.clean_resultsN)__name__ __module__ __qualname____doc__r __annotations__r model_configr classmethodrrrr@rintrboolr0r1rCrEr2rrrrr s      !    2    (    r )rMr'typingrrrrr5r)langchain_core.utilsrpydanticrr r r r+r rrrrs