DjEddlZddlZddlZddlmZmZmZmZmZm Z m Z ddl Z ddl m Z ddlmZmZmZmZmZmZmZmZdefdZd dZGd d ZdS) N)LiteralSequenceOptionalListUnionAsyncGenerator Awaitable)get_max_items_from_list)UsageLimitExceededErrorInvalidAPIKeyErrorMissingAPIKeyErrorBadRequestErrorForbiddenError TimeoutErrorTavilyKeylessLimitErrorKeylessUnsupportedEndpointErrorreturnct|toUt|dto-t|ddtS)zWReturn True when the response body matches the Tavily recoverable-error envelope shape.errorcode) isinstancedictgetstr)bodys P/home/lighthouse/health/venv/lib/python3.11/site-packages/tavily/async_tavily.py_is_keyless_envelopersX 4 7 txx(($ / / 7 tG}((00# 6 6c |d}t|dpd|d|d|d|dpg) zHRaise ``TavilyKeylessLimitError`` from an envelope-shaped response body.rmessagerwindowretry_after_seconds next_actions)r!rr#r$r%)rr)rerrs r_raise_keyless_enveloper'sv w-C ! ""(b WWV__wwx  GG$9::WW^,,2    rc)eZdZdZ d_deedeedeeeefdeedeed eed eed eed eed eej fdZ dZ dZ dZ d`dadZdeddfdZededeefdZ dbdededdeddedd ed!ed"ed#ed$eed%eed&eeed'fd(eeed)fd*ed+ed,ed-ed.ed/ed0edef(d1Z dbdededdeddedd ed!ed"ed#ed$eed%eed&eeed'fd(eeed)fd*ed+ed,ed-ed.ed/ed0edef(d2Z dcd4eeeefd*ed5ed'd6ed)d+ed.ed/eded7edefd8Z dcd4eeeefd*ed5ed'd6ed)d+ed.ed/eded7edefd9Z ddd;eded?ed@eedAeedBeed%eedCed*ed5ed'd6ed)d+ed.ed/ed7edef$dDZ ddd;eded?ed@eedAeedBeed%eedCed5ed'd*ed6ed)d+ed.ed/ed7edef$dEZ ded;eded?ed@eedAeedBeed%eedCed*ed+ed/edefdFZ ded;eded?ed@eedAeedBeed%eedCed*ed+ed/edefdGZ dfdededdedd"ed#ed$eed%eedMed+ed,ed.edefdNZ dgdededdedd"ed#ed$eed%eed+ed,ed.edefdPZ! dhdededd#ed+ed,edeef dQZ" didTedUedVdWedXedYedZd+eedee#e$dfe%effd[Z& didTedUedVdWedXedYedZd+eedeee#e$dfffd\Z'd]edefd^Z(dS)jAsyncTavilyClientz( Async Tavily API client class. Nnewsgeneralfinanceapi_keycompany_info_tagsproxies api_base_url client_source project_id session_idhuman_id client_nameclientc |tjd}|pd}|ptjd} |pd|_||_|duo| du|_|jr|pd} n|pd} ddi|rdd |ini|jrd d inid | i| rd | ini|rd|ini|rd|ini| rd| ini} | du|_| l| |_| D]"\}}||jjvr||jj|<#t|jj s|j|j_ dSdS|pi}| dtjd| dtjdd}d|D}|rd|Dnd}tj | |j||_dS)NTAVILY_API_KEYTAVILY_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-NamehttpTAVILY_HTTP_PROXYhttpsTAVILY_HTTPS_PROXY)zhttp://zhttps://ci|] \}}||| SrE).0keyvalues r z.AsyncTavilyClient.__init__..ks$[[[ZS%UZ[c5[[[rc@i|]\}}|tj|S))proxy)httpxAsyncHTTPTransport)rFschemerKs rrIz.AsyncTavilyClient.__init__..ns,kkk=651>>>kkkr)headersbase_urlmounts)osgetenv _api_base_url_company_info_tags_keyless_external_client_clientitemsrOrrPrrL AsyncClient)selfr.r/r0r1r2r3r4r5r6r7tavily_projectclient_source_headerdefault_headersrGrHmapped_proxies proxy_mountss r__init__zAsyncTavilyClient.__init__/s ?i 011G/T#Bry1A'B'B)E-E"34:FdN = D$1#K4K #0#CO  .  9@H!47!4!455b  7;mK& 22   3   4BI//r  0:A ++r  ,4; h''  2=D--"  !'d 2  !DL-3355 6 6 Udl22205DL(-t|,-- ;(,(: %%% ; ;mG#;;vry9L/M/MNN#KK;O1P1PQQN \[>;O;O;Q;Q[[[N"kkTbThThTjTjkkkk  !,'+#DLLLrcZK|js!|jd{VdSdS)z7Close the client and release connection pool resources.N)rWrXacloser[s rclosezAsyncTavilyClient.closeysG$ (,%%'' ' ' ' ' ' ' ' ' ' ( (rc K|SNrErds r __aenter__zAsyncTavilyClient.__aenter__~s  rc>K|d{VdSrg)re)r[exc_typeexc_valexc_tbs r __aexit__zAsyncTavilyClient.__aexit__s,jjllrrcv||}n( |}n#t$rd}YnwxYw|jrt|rt |d}t |t r@ |diddpd}n)#t$rd}YnwxYwt |tr|}|j dkrt||j dvrt||j dkrt||j dkrt||) a Raise an appropriate exception for a non-2xx httpx response. On keyless calls, checks the response body for the recoverable-error envelope shape and raises ``TavilyKeylessLimitError`` with the envelope's structured fields when present. Nr"detailri)iiiii)json ExceptionrVrr'rrrr status_coder rr rraise_for_status)r[response body_overriderros r_handle_error_responsez(AsyncTavilyClient._handle_error_responsesv  $ DD }}     = *1$77 * #D ) ) ) dD ! !  (B//33GTBBHb     c " " F  3 & &)&11 1  !_ 4 4 (( (  !S ( ($V,, ,  !S ( (!&)) )++-- -s +++,B B'&B'methodc2|jrt|dS)zGRaise ``KeylessUnsupportedEndpointError`` when running in keyless mode.N)rVr)r[rws r_check_keyless_supportedz*AsyncTavilyClient._check_keyless_supporteds$ = :1&99 9 : :rkwargscri}dD]/\}}||d}|t|||<0|pdS)zPop session_id, human_id, and client_name from kwargs and return them as headers. Returns None when no overrides are provided so callers can omit the headers kwarg. ))r4r=)r5r>)r6r?N)popr)rz overridesrG header_namerHs r_pop_request_headersz&AsyncTavilyClient._pop_request_headerssV  !  4 4 C JJsD))E ),U +& D r<query search_depth)basicadvancedfastz ultra-fasttopic)r,r+r- time_range)dayweekmonthyear start_dateend_datedays max_resultsinclude_domainsexclude_domainsinclude_answer)rrinclude_raw_content)markdowntextinclude_imagestimeoutcountryauto_parametersinclude_favicon include_usage exact_matchcFKid|d|d|d|d|d|d|d| d | d |d | d | d | d|d|d|d|d|i}d|D}||}|r||t|d} |jjdt j||d|rd|inid{V}n"#tj $rt|wxYw|j dkr|S| |dS)zH Internal search method to send the request to the API. rrrrrrrrrrrrrrrrrrci|] \}}||| SrgrErFkvs rrIz-AsyncTavilyClient._search..???A1rx/searchcontentrrON)r) rYrupdateminrXpostrpdumpsrLTimeoutExceptionrrrrv)r[rrrrrrrrrrrrrrrrrrrrzdataoverride_headersrts r_searchzAsyncTavilyClient._searchs74 U L  U  *  *     D  n  "#6  ;      n  w   ! " ]# $ ;%  *@????44V<<  KK   gs## (.T\.d$*TBRBR\cddJZibiruEiFiF`bddddddddHH% ( ( (w'' ' (  3 & &==?? "  ' ' 1 1 1 1 1s 3C C*cKt|d}|j|fid|d|d|d|d|d|d|d | d | d | d | d | d|d|d|d|d|d||d{V}|dg}||d<|S)zr Combined search method. Set search_depth to either "basic", "advanced", "fast", or "ultra-fast". rrrrrrrrrrrrrrrrrrrNresults)rrr)r[rrrrrrrrrrrrrrrrrrrrz response_dicttavily_resultss rsearchzAsyncTavilyClient.searchs`2gs##*dl5---8D -16-7Aj-7Aj - 5=H - 15 -8C{-.Trr/extractrrONr)r) rYrrrXrrprrLrrrrrv)r[rrrrrrrrrrzrrrts r_extractzAsyncTavilyClient._extract7sp$,*.*!2   @????44V<<  KK    (.T\.e4:dCSCS]deeK[jcjsvFjGjGaceeeeeeeeHH% ( ( (w'' ' (  3 & &==?? "  ' ' 1 1 1 1 1s 3BB-c K|j|||||f|||| d| d{V} | dg} | dg} | | d<| | d<| S)z{ Combined extract method. include_favicon: If True, include the favicon in the extraction results. )rrrrNrfailed_results)rr)r[rrrrrrrrrrzrrrs rextractzAsyncTavilyClient.extractds ,dmD,:,9,2,3 . =L:G27>O . ./5 . . . . . . . . '**9b99&**+;R@@#1 i *8 &'rurl max_depth max_breadthlimit instructions select_pathsselect_domains exclude_pathsallow_externalcKid|d|d|d|d|d|d|d|d | d | d | d | d | d|d|d|d|}||}|r||d|D} |jjdt j||d|rd|inid{V}n"#tj$rt|wxYw|j dkr|S| |dS)zG Internal crawl method to send the request to the API. rrrrrrrrrrrrrrrrrci|] \}}||| SrgrErs rrIz,AsyncTavilyClient._crawl..rr/crawlrrONr)r rrrYrXrrprrLrrrrrv)r[rrrrrrrrrrrrrrrrrrzrrrts r_crawlzAsyncTavilyClient._crawls. 3   ;  U  L  L   n  ]    n  n  ]  f  w   ]! " !2# ( 44V<<  KK   ????? (.T\.cDAQAQ[bccIYhahqtDhEhE_accccccccHH% ( ( (w'' ' (  3 & &==?? "  ' ' 1 1 1 1 1s 3B77CcK|d|j|fid|d|d|d|d|d|d|d | d | d | d | d | d|d|d|d||d{V}|S)z) Combined crawl method. crawlrrrrrrrrrrrrrrrrN)ryr)r[rrrrrrrrrrrrrrrrrrzrs rrzAsyncTavilyClient.crawls*0 %%g...)dk#....7i.0; .+0%.2> . 2> . 4B> .3@-.5DO.4B>.3@-.4B>.,26.-4G.5DO.3@-. 7H6G&,#........ &rcK||||||||| | | | | d }||}|r||d|D} |jjdt j|| d|rd|inid{V}n"#tj$rt| wxYw|j dkr|S| |dS) zE Internal map method to send the request to the API. ) rrrrrrrrrrrrrci|] \}}||| SrgrErs rrIz*AsyncTavilyClient._map..rr/maprrONr)rr)r[rrrrrrrrrrrrrrzrrrts r_mapzAsyncTavilyClient._maps{("&((,*.,,*    44V<<  KK   ????? (.T\.atz$?O?OY`aaGWf_forBfCfC]_aaaaaaaaHH% ( ( (w'' ' (  3 & &==?? "  ' ' 1 1 1 1 1s 3BB1cvK|d|j|f|||||||| | | | | d |d{V}|S)z' Combined map method. map) rrrrrrrrrrrrN)ryr)r[rrrrrrrrrrrrrrzrs rrzAsyncTavilyClient.map$s( %%e,,,'di ..70;*/1=1=3A2?4C3A3A,32? . .'- . . . . . . . . rrr, max_tokensc K|dt| d} |j|f||||||ddd| | | d | d{V} | 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_contextrF) rrrrrrrrrrrrNrc0g|]}|d|ddS)rr)rrrE)rFsources r z8AsyncTavilyClient.get_search_context..rs)^^^F6%=VI5FGG^^^r)ryrrrrprr )r[rrrrrrrrrrrrzrsourcescontexts rrz$AsyncTavilyClient.get_search_contextJs, %%&:;;;gs##*dl5-8D16047B;J;J:??D:?5<3:;J--.4--------  ##Ir22^^V]^^^z1':FFGGGrrc K|dt|d}|j|f||||||ddd|| | d | d{V} | ddS) z` Q&A search method. Search depth is advanced by default to get the best answer. qna_searchrFT) rrrrrrrrrrrrNanswerr")ryrrr) r[rrrrrrrrrrrzrs rrzAsyncTavilyClient.qna_searchus %%l333gs##*dl5-8D16047B;J;J?D:?:>5<3:;J--.4--------   2...rcN Kdtddtffd g}tj fdjDd{VD]!}d|vr||d"t|dd d}|S) z` Company information search method. Search depth is advanced by default to get the best answer. get_company_inforrc LK|dd{VS)NF)rrrrrr)r)rrrrrr[rs r_perform_searchz;AsyncTavilyClient.get_company_info.._perform_searchsTe3?,12=5:07.5 &77777777 7rc&g|] }|SrErE)rFrrs rrz6AsyncTavilyClient.get_company_info..s#*g*g*ge??5+A+A*g*g*grNrc|dS)NscorerE)xs rz4AsyncTavilyClient.get_company_info..s 1W:rT)rGreverse)ryrrasynciogatherrUextendsorted) r[rrrrr all_resultsrsorted_resultsrs `````` @rrz"AsyncTavilyClient.get_company_infos %%&8999gs## 7 7 7 7 7 7 7 7 7 7 7 7 !.*g*g*g*gtOf*g*g*ghhhhhhh 4 4DD  ""4 ?333 1E1EtTTTUaVaUabrFnumberedinputmodel)miniproauto output_schemastreamcitation_format)rmlaapachicagoc $ |||||d d D | |r ||r'dttdff fd }|S fd} | S)zJ Internal research method to send the request to the API. )rrrrrci|] \}}||| SrgrErs rrIz/AsyncTavilyClient._research..rrrNcK jj d tjdrdini4d{V }|jdkr |d{V}t |tr|dn|}n#t$rd}YnwxYwd} tj |}n#t$r|}YnwxYw ||| 23d{V}|r|WV6 dddd{VdS#1d{VswxYwYdS#tj$rtt$r$}td t!|d}~wwxYw) NPOST /researchrrOrzutf-8z Unknown error)ruzError during research stream: )rr)rXrrprrrareadrbytesdecoderqloadsrv aiter_bytesrLrrr) rt error_textruchunkerrr[rs rstream_generatorz5AsyncTavilyClient._research..stream_generatorsO2t|2# !% 4 0 0 '    =MTI'788RT   ,,,,,,,, "#/366=3;>>3C3C-C-C-C-C-C-C KUV`bgKhKh-xZ->->w-G-G-Gnx #,===-< =-1M;04 :0F0F #,;;;0: ;!77P]7^^^+3+?+?+A+A,,,,,,,%$,&+ ,B+A-,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,2-000&w/// OOO#$MSVV$M$MNNNOs5D- DABD BDBD$B98D9 CDC.D6D< DD- D$$D-'D$(D--'E8E33E8c"K jjdtjdrdinid{V}n"#tj$rt wxYw|jdkr|S|dS)NrrrOr)r) rXrrprrLrrrrrv)rtrrr[rs r _make_requestz2AsyncTavilyClient._research.._make_requests0%6T\%6&nDJW[L\L\fm&n&nTdsls|OsPsPjl&n&n n n n n n nHH-000&w///0'3..#==??*//99999s 39A)rYrrrr) r[rrrrrrrzrrrrs ` ` @@r _researchzAsyncTavilyClient._researchs*.   @????44V<<  KK    . # ON5$;,G O O O O O O O O O@$#%% % : : : : : : : :!=?? "rc pK|d|jd||||||d|}|r|S|d{VS)a Research method to create a research task. Args: input: The research task description (required). model: Research depth - 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: When stream=False: dict - the response dictionary. When stream=True: AsyncGenerator[bytes, None] - iterate over this to get streaming chunks. research)rrrrrrNrE)ryr) r[rrrrrrrzresults rrzAsyncTavilyClient.researchsv2 %%j111 + /       M<<<<<< r request_idc.K|d |jd|d{V}n$#t$r}td|d}~wwxYw|jdvr|}|S||dS)z 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/NzError getting research: )r)ryrXrrqrrrprv)r[rrtrrs rrzAsyncTavilyClient.get_research(s %%n555 <!\--.G:.G.GHHHHHHHHHH < < <:q::;; ; <  : - -==??DK  ' ' 1 1 1 1 1s#= AAA) Nr*NNNNNNNNrgrN)NNNNNNNNNNNNrNNNNN)NNNrNNNN)NNNNNNNNNNNNrNNN) NNNNNNNNNNrN) rr,rrNNrrNN) rr,rrNNrNN)rrrN)NNFrN))__name__ __module__ __qualname____doc__rrrrrLrZrarerhrmrvry staticmethodrrintrboolfloatrrrrrrrrrrrrrrr rrrrErrr)r)*s 154R59/304-1-1+/.27;HH H$,SMH"4S>2H (}H!) H &c] H &c] H$C=H'smH"%"34HHHHT((( $.$.$.$.$.L:s:t:::: !T!htn!!!\!&PT;?BF" #-1-1HLLP#'$($("& $)?2?2?2""KL?278 ?2  >? ?2  ?2?2?2?2&c]?2&c]?2"$0C(D"DE?2"'tW5G-H'H!I?2!?2?2 !?2""#?2$"%?2& '?2()?2, -?2?2?2?2FY]DHKO'+%)!%(,6:6:QUUY,0&($(-1-1+/)-'444#*+T#U4$$@A4"))G!H 4 "% 4 # 44#&4'/sm4'/sm4&+49L1M+M%N4+0g>P6Q0Q*R4&*4$4"4 '+!4"'+#4$%)%4&#''4*+4444r$(:>26$("&%)+2+2S 3'+2!+2##67 +2 ./ +2  +2"+2 +2+2 #+2 +2+2+2+2^.2DH<@').2,0#'/3""!$s)S.1"&*"&--@%A"&&89 " % " (, "&*"!"*-" """"L!%"& #'-1/3.204&*&*=A59 #'+%)(,#92929292 92 92 ! 92 &c] 92 (}92'sm92!) 92 $92 $92&&9:92129292!%92 #!92"# 2&'92929292z&*'+!%(,26483759+/BF+/:>%(,0*.-1#,,,","%, , #& , #+3- ,%-SM,$,C=,&.c],%),$++>#?,%),$$67,#,&*, $(!,"(+#,&',,,,`!%"& #'-1/3.204&*&* #%)12121212 12 12 ! 12 &c] 12 (}12'sm12!) 12 $12 $1212#1212121212j&*'+!%(,26483759+/+/%(*.$$$"$"%$ $ #& $ #+3- $%-SM$$,C=$&.c]$%)$%)$#$$($$$$$PelPY-.45BFBF3724049=)H)H(+)H/67`/a)H)00L(M)H(+ )H /2 )H 3;3- )H3;3-)H.1)H+0)H+.)H37)H'*)H)H)H)HZ]gHQ%&,-:>:>*,(,15!/!/ #!/'./X'Y!/!((D E!/ # !/ '* !/ +33- !/+33-!/#(!/#&!/+/!/"!/!/!/!/Jcm2302.2 &)-45^-_-0).  ),  %-TN @;?(,!&R\-1 H#H#H# !67H#"&H# H# $++N#O H# $E? H#^E4K8)D/IJH#H#H#H#X@D-1&+Wa26 & & !& %&;<& '+&  $ & )00S(T & !) & "$ud{(C"CD& & & & P2'*2 $222222rr)r)rrprRtypingrrrrrrr rLutilsr errorsr r rrrrrrr!rr'r)rErrr&s@ VVVVVVVVVVVVVVVVVV ******                    $    T 2T 2T 2T 2T 2T 2T 2T 2T 2T 2r