Djeed?eed@eed#eedAed(ed3ed%d4ed'd)ed,ed-ed5edef$dBZ d`d9ed:ed;edeed?eed@eed#eedAed(ed3ed%d4ed'd)ed,ed-ed5edef$dCZ dad9ed:ed;edeed?eed@eed#eedAed(ed)ed-edefdDZ dad9ed:ed;edeed?eed@eed#eedAed(ed)ed-edefdEZ dbdededdedd ed!ed"eed#eedKed)ed*ed,edefdLZ dcdededdedd ed!ed"eed#eed)ed*ed,edefdNZ! dddQedRedSdTedUedVedWd)eedeee"e#ddfffdXZ$ dddQedRedSdTedUedVedWd)eedeee"e#ddfffdYZ%dZedefd[Z&dS)e TavilyClientz" Tavily API client class. Napi_keyproxies api_base_url client_source project_id session_idhuman_id client_namesessionc |tjd}|pd}|r|dntjd|r|dntjdd} d| Dpd} |ptjd} |pd |_||_|duo| du|_| |_|jr|pd } n|pd } d d i|jr dd|jini|jrddinid| i| rd| ini|rd|ini|rd|ini|rd|ini|_| du|_ | | ntj |_ |jD]"\} }| |j jvr||j j| <#|jr<|jD]$\}}||j jvr||j j|<#dSdS)NTAVILY_API_KEYhttpTAVILY_HTTP_PROXYhttpsTAVILY_HTTPS_PROXY)r4r6ci|] \}}||| Sr9.0kvs r z)TavilyClient.__init__..Cs#KKKTQKAqKKKrTAVILY_PROJECTzhttps://api.tavily.comztavily-python-keylessz tavily-pythonz Content-Typezapplication/json AuthorizationzBearer zX-Tavily-Access-ModekeylesszX-Client-Sourcez X-Project-ID X-Session-Id X-Human-Id X-Client-Name) osgetenvritemsbase_urlr)_keylessr*headers_external_sessionrequestsSessionr1)selfr)r*r+r,r-r.r/r0r1resolved_proxiestavily_projectclient_source_headerkeyvalueprotocolurls r__init__zTavilyClient.__init__-s ?i 011G/T,3VGKK''' BU8V8V-4YW[[)))")DX:Y:Y   LK-=-C-C-E-EKKKSt#Bry1A'B'B$@(@  4;GtO ' = D#0#K4K #0#CO  .  >BlR!94z(TavilyClient._search..???A1rxz/searchdatarrJN)rGrrupdateminrHrddumpsr1postrL exceptionsTimeoutrrfri)rNrtruryr}rrrrrrrrrrrrrrrrmroverride_headersrUpayloadrhs r_searchzTavilyClient._searchs4 U L  U  *  *     D  n  "#6  ;      n  w   ! " ]# $ ;%  *@????44V<<  KK   gs##mi'*T"" ((t|(L7GLLrBQJQZ\lPmPmHJLLHH"* ( ( (w'' ' (  3 & &==?? "  ' ' 1 1 1 1 1s 3C$C4c |j|fid|d|d|d|d|d|d|d| d | d | d | d | d |d|d|d|d|d||}|dg|S)z) Combined search method. ruryr}rrrrrrrrrrrrrrrresults)r setdefault)rNrtruryr}rrrrrrrrrrrrrrrrm response_dicts rsearchzTavilyClient.searchs!4% U///2>,/+05/1; /1; / /7h / +/$ /2=/6E_/6E_/5CN/:M9L/5CN/.5W/.5W/6E_/ 6E_!/"4A=#/$2='-'// (   B///rurls extract_depthformatchunks_per_sourcec ||||||||| d } d| D} || } | r| |  |jj|jdzft j| |d| rd| ini} n'#tj j $rt|wxYw| j dkr| S| | dS)zJ Internal extract method to send the request to the API. ) rrrrrrrrtrci|] \}}||| Sr[r9r:s rr>z)TavilyClient._extract..4rrz/extractrrJrN)rGrrrr1rrHrdrrLrrrrfri)rNrrrrrrrrtrrmrrrhs r_extractzTavilyClient._extractsU ,*.*!2   @????44V<<  KK    ((t|()Cl$*UYJZJZdkllRbqjqz}MqNqNhjllHH"* ( ( (w'' ' (  3 & &==?? "  ' ' 1 1 1 1 1s 6B$B3c  |j|||||f|||| d| } | dg| dg| S)z* Combined extract method. )rrrtrrfailed_results)rr) rNrrrrrrrrtrrmrs rextractzTavilyClient.extractDs}& d&4&3&,&- 0 7F4A,18I 0 0)/ 0 0    B///  !12666rrU max_depth max_breadthlimit instructions select_pathsselect_domains exclude_pathsallow_externalc *id|d|d|d|d|d|d|d|d | d | d | d | d | d|d|d|d|}||}|r||d|D} |jj|jdzft j||d|rd|ini}n'#tj j $rt|wxYw|j dkr|S| |dS)z Internal crawl method to send the request to the API. include_favicon: If True, include the favicon in the crawl results. rUrrrrrrrrrrrrrrrrci|] \}}||| Sr[r9r:s rr>z'TavilyClient._crawl..rrz/crawlrrJrNrrrrGr1rrHrdrrLrrrrfri)rNrUrrrrrrrrrrrrrrrrrmrrrhs r_crawlzTavilyClient._crawlas0 3   ;  U  L  L   n  ]    n  n  ]  f  w   ]! " !2# ( 44V<<  KK   ????? ((t|()Aj SWHXHXbijjP`ohox{KoLoLfhjjHH"* ( ( (w'' ' (  3 & &==?? "  ' ' 1 1 1 1 1s 6B88$Cc |d|j|fid|d|d|d|d|d|d|d | d | d | d | d | d|d|d|d||S)zt Combined crawl method. include_favicon: If True, include the favicon in the crawl results. crawlrrrrrrrrrrrrrrrr)rlr)rNrUrrrrrrrrrrrrrrrrrms rrzTavilyClient.crawls0 %%g...t{3%%%%.Y%'2{%"'%)5 % )5 % +9. %*7%,;?%+9.%+9.%*7%#)&%$+7%,;?%*7% .?->##%% %rc ||||||||| | | | | d }||}|r||d|D} |jj|jdzft j|| d|rd|ini}n'#tj j $rt| wxYw|j dkr|S| |dS)zE Internal map method to send the request to the API. ) rUrrrrrrrrrrrrci|] \}}||| Sr[r9r:s rr>z%TavilyClient._map..rrz/maprrJrNr)rNrUrrrrrrrrrrrrrmrrrhs r_mapzTavilyClient._maps`("&((,*.,,*    44V<<  KK   ????? ((t|()?hdjQUFVFV`ghhN^mfmvyImJmJdfhhHH"* ( ( (w'' ' (  3 & &==?? "  ' ' 1 1 1 1 1s 6B$B7c b|d|j|f|||||||| | | | | d |S)z' Combined map method. map) rrrrrrrrrrrr)rlr)rNrUrrrrrrrrrrrrrms rrzTavilyClient.mapsh( %%e,,,ty ##,%0$&2&2(6'4)8(6(6!('4 # #" # # #rrvrz max_tokensc  "|dtjdtd|j|f||||||ddd| | | d | } | dg}d|D}t jt||S) a Get the search context for a query. Useful for getting only related content from retrieved websites without having to deal with context extraction and limitation yourself. max_tokens: The maximum number of tokens to return (based on openai token compute). Defaults to 4000. Returns a string of JSON containing the search context up to context limit. get_search_contextzHget_search_context is deprecated and will be removed in future versions. stacklevelF) ruryrrrrrrrrrrrc0g|]}|d|ddS)rUcontent)rUrr9)r;sources r z3TavilyClient.get_search_context..Js9***"%=VI5FGG***r) rlwarningswarnDeprecationWarningrrrdrr )rNrtruryrrrrrrrrrmrsourcescontexts rrzTavilyClient.get_search_context s, %%&:;;; `(Q 8 8 8 8% U'2>+0*.1<5D5D499>49-4-45D''(.''  ##Ir22**!(***z1':FFGGGrrwc  |dtjdtd|j|f||||||ddd|| | d | } | dd S) z` Q&A search method. Search depth is advanced by default to get the best answer. qna_searchz@qna_search is deprecated and will be removed in future versions.rrFT) ruryrrrrrrrrrranswerr!)rlrrrrr) rNrtruryrrrrrrrrmrs rrzTavilyClient.qna_searchNs %%l333 X(Q 8 8 8 8$ U'2>+0*.1<5D5D9>4948-4-45D''(.''   2...rFnumberedinputmodel)miniproauto output_schemastreamcitation_format)rmlaapachicagoc  |||||d}d|D}||} |r|||r |jj|jdzft j||dd| rd| ini n'#tj j $rt|wxYw j dkr| dttd d ff fd } | S |jj|jdzft j||d | rd| ini n'#tj j $rt|wxYw j dkr S| d S) zJ Internal research method to send the request to the API. )rrrrrci|] \}}||| Sr[r9r:s rr>z*TavilyClient._research..rrz /researchT)rrrrJrrNc3K dD]}|r|V dS#wxYw)N) chunk_size) iter_contentrX)chunkrhs rstream_generatorz0TavilyClient._research..stream_generatorsq%!)!6!6$!6!G!G(( ("'KKK(NN$$$$$HNN$$$$s ;Ar)rGrrrr1rrHrdrrLrrrrfrirbytes) rNrrrrrrrmrrrrhs @r _researchzTavilyClient._researchrs/*.   @????44V<<  KK    & 6 ,,4<,MK/D))#  9IP #344b &. , , ,"7+++ ,#s**++H555 %itT0A&B % % % % % %$#%% % ,,4<,MK/D))#9IP #344b  &. , , ,"7+++ ,#s**}}&++H55555s7B$B3<6D33$Ec T|d|jd||||||d|S)a Research method to create a research task. Args: input: The research task or question to investigate (required). model: The model used by the research agent - must be either 'mini', 'pro', or 'auto'. output_schema: Schema for the 'structured_output' response format (JSON Schema dict). stream: Whether to stream the research task. citation_format: Citation format - must be either 'numbered', 'mla', 'apa', or 'chicago'. timeout: Optional HTTP request timeout in seconds. **kwargs: Additional custom arguments. Returns: dict: Response containing request_id, created_at, status, input, and model. research)rrrrrrr9)rlr)rNrrrrrrrms rrzTavilyClient.researchsT0 %%j111t~ '+      r request_idc*|d |j|jd|z}n$#t$r}t d|d}~wwxYw|jdvr|S||dS)a  Get research results by request_id. Args: request_id: The research request ID. Returns: dict: Research response containing request_id, created_at, completed_at, status, content, and sources. get_researchz /research/zError getting research: N)r)rlr1rrHrerfrdri)rNrrhes rrzTavilyClient.get_researchs %%n555 <|'' 8QZ8Q8Q(QRRHH < < <:q::;; ; <  : - -==?? "  ' ' 1 1 1 1 1s%= AAA) NNNNNNNNNrN)NNNNNNNNNNNNrsNNNNN)NNNrNNNN)NNNNNNNNNNNNrNNN) NNNNNNNNNNrN) rvrzrrNNrrsNN) rwrzrrNNrsNN)NNFrN)'__name__ __module__ __qualname____doc__rrrrLrMrVrXr\rarirl staticmethodrrrintrrboolfloatrrrrrrrrrrrrrrrrr9rrr(r((sK  "&,0&*'+$($("&%).27979#79$sCx.)79sm 79  } 79 SM 79SM793-79c]79(*+79797979r!!! ....::s:t:::: !T!htn!!!\!$TX?CFJ"& $ #'1515LPPT'+!##(,(,&*$('A2A2A2%&OPA2;<A2$$BC A2  A2  A2A2!A2"*#A2"*#A2!&dG4G,H&H IA2&+49K1L+L%MA2!%A2A2A2 "&!A2""&#A2$ $%A2&"'A2*+A2A2A2A2LSW?CEI!%#"&0404KOOS&* ""'+'+%)#''///$%NO/:</##AB /  /  // /!) /!) / %T73F+G%GH/%*$8J0K*K$L/ $/// !%!/"!%#/$#%/&!'/*+////f)-?C7;"$)-'+"*.)2)2T#Y^,)2!%)2!((; <)2!!34 )2  )2 #' )2!%)2)2%()2)2)2)2)2Z(,>B6:!#(,&*!)-DIsN+ $ '':; 23    "&  $$'>"# $*.,0+/-1#'#':>26 $("&%)#:2:2:2:2:2 :2  :2 #3- :2%SM:2$C=:2&c]:2!:2!:2##67:2./:2:2":2  !:2" ##:2&':2:2:2:2| $!%"&,0.2-1/3%)%)<@48"&*$('+#*%*%*%*%*% *%  *% %SM *%'sm*%&c]*% (}*%#*%#*%%%89*%01*%*% $*% "!*%""%#*%&'*%*%*%*%\"# $*.,0+/-1#'#' "&1212121212 12  12 #3- 12%SM12$C=12&c]12!12!1212 1212121212j $!%"&,0.2-1/3%)%)"$("#"#"#"#"# "#  "# %SM "#'sm"#&c]"# (}"##"##"#"#""#"#"#"#"#L_fJS'(./<@<@-1,.*.37,H,H"%,H)01Z)[,H#**F"G,H"% ,H ), ,H -5SM ,H-5SM,H(+,H%*,H%(,H-1,H!$,H,H,H,H`WaBK &'4848$&"&+/"/"/"/!()R!S"/"">?"/ "/ !$ "/ %-SM "/%-SM"/""/ "/%)"/"/"/"/"/L;?(,!&R\-1 @6@6@6 !67@6"&@6 @6 $++N#O @6 $E? @6T9UD$->#??@@6@6@6@6H:>'+ %Q[,0 " " "  56" !%"  " #**M"N " #5/ " D)E4,=">>?" " " " H2!$2222222rr(c"eZdZdZfdZxZS)Clientzk Tavily API client class. WARNING! This class is deprecated. Please use TavilyClient instead. ctjdtdt|dS)Nz5Client is deprecated, please use TavilyClient insteadrr)rrrsuperrV)rNrm __class__s rrVzClient.__init__sB M(Q 8 8 8 8      r)rrrrrV __classcell__)rs@rrrsB !!!!!!!!!rrr)rLrdrErtypingrrrrrrutilsr errorsr r r rrrrrrrr&r(rr9rrr sc FFFFFFFFFFFFFFFF******                    $    E 2E 2E 2E 2E 2E 2E 2E 2P ! ! ! ! !\ ! ! ! ! !r