a bg@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 e e ddd Z d d ggd d d fe eeee eee eee eeeeeee d ddZd d ggd d d fe eeee eee eee eeeeeeee d ddZd d ggd d d fe eeee eee eee eeeeeee d ddZd d ggd d d fe eeee eee eee eeeeeeee d ddZee ee dddZdS)TavilySearchAPIWrapperzWrapper for Tavily Search API.tavily_api_keyZforbid)extrabefore)mode)valuesreturncCst|dd}||d<|S)z9Validate that api key and endpoint exists in environment.r ZTAVILY_API_KEYr)clsrr ry/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/utilities/tavily_search.pyvalidate_environments z+TavilySearchAPIWrapper.validate_environmentZadvancedF) query max_results search_depthinclude_domainsexclude_domainsinclude_answerinclude_raw_contentinclude_imagesrc CsB|j||||||||d } tjtd| d} | | S)N Zapi_keyrrrrrrrr/searchjson)r get_secret_valuerequestspostTAVILY_API_URLraise_for_statusr#) selfrrrrrrrrparamsresponserrr 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. )rrrrrrrresults)r, clean_results) r)rrrrrrrrZraw_search_resultsrrrr-Ds# zTavilySearchAPIWrapper.resultsc s8tdf dd } | IdH} t| S)z6Get results from the Tavily Search API asynchronously.)rc sjd }t4IdH}|jtd|d4IdHr}|jdkr|IdH}|WdIdHWdIdHStd|jd|j WdIdHq1IdHs0YWdIdHq1IdHs0YdS)Nr r!r"zError z: ) r r$aiohttpZ ClientSessionr&r'statustext Exceptionreason)r*sessionresdata rrrrrrrrr)rrfetchs   ,z7TavilySearchAPIWrapper.raw_results_async..fetchN)strr#loads) r)rrrrrrrrr9Zresults_json_strrr8rraw_results_asyncss" z(TavilySearchAPIWrapper.raw_results_asyncc s.|j||||||||dIdH} || dS)N)rrrrrrrrr-)r<r.) r)rrrrrrrrZ results_jsonrrr results_asyncs  z$TavilySearchAPIWrapper.results_async)r-rcCs*g}|D]}||d|ddq|S)z%Clean results from Tavily Search API.urlcontent)r>r?)append)r)r-r.resultrrrr.sz$TavilySearchAPIWrapper.clean_resultsN)__name__ __module__ __qualname____doc__r __annotations__r Z model_configr classmethodrrrr:rintrboolr,r-r<r=r.rrrrr s     !   2   (   r )rEr#typingrrrrr0r%Zlangchain_core.utilsrZpydanticrr r r r'r rrrrs