i ddlmZddlZddlZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZddlmZddlZddlZddlmZdd lmZdd l m!Z!dd l"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+dd lm,Z,ddl-m.Z.ddl/m0Z0m1Z1ddl2m3Z3m4Z4ej5ddkZ6dZ7ddZ8gfddZ9ddZ:ddZ;dd#Z Gd&d'e#d()Z?Gd*d+e#d()Z@ee?e@fZA id,eBgd-eCgd.eDgd/eDgd0eBgd1eBgd2eBgd3eBgd4eBgd5eDgd6eDgd7e>eBgd8e=gd9eDgd:eEgd;eFeEgdZHed?ZIeDgeFeEgeFeEgeFeEgeFeEgeFeEgeCgeIgeCgeEgeDgd@ ZJeCgeBeDgeFgeDgdAZKddFZLdGZMddJZNedKZO edLZP GdMdNe#d()ZQGdOdPe#d()ZRGdQdRe#d()ZSGdSdTe#d()ZTGdUdVe#d()ZUGdWdXe#d()ZVGdYdZe#d()ZWGd[d\e#d()ZXGd]d^e#d()ZYeGd_d`ZZeGdadbZ[eGdcddZ\eGdedfZ]eGdgdhZ^eGdidjZ_eGdkdlZ`eGdmdnZaeGdodpZbeGdqdrZceGdsdtZdeGdudvZeeGdwdxZfeeeeffZgeGdydzZheGd{d|ehZieGd}d~ehZjeGddehZkeGddehZleGddZmeGddZneGddZoGddZpGddZqedZreGddZseGdde erZteGddZuedZv eGddZweGddZxddZyddZzGddZ{Gdde{Z|dS)) annotationsN) dataclass)wraps)AnyCallableDictGenericIterableIteratorListLiteralOptionalTypeVarUnionget_args get_originoverload)AsyncWebsetsClient)OpenAI)ChatCompletionMessageParam) ChatModel) TypedDict)ExaOpenAICompletion_convert_schema_input_get_package_versionadd_message_to_messagesformat_exa_resultmaybe_get_queryJSONSchemaInput) WebsetsClient)ExaJSONEncoder)SearchMonitorsClientAsyncSearchMonitorsClient)ResearchClientAsyncResearchClientIS_BETATruei' snake_strstrreturnc|dkrdS|dkrdS|d}|ddd|d d DzS) zConvert snake_case string to camelCase. Args: snake_str (str): The string in snake_case format. Returns: str: The string converted to camelCase format. schema_z$schemanot_not_rc3>K|]}|VdSN)title).0xs Q/home/ubuntu/.hermes/hermes-agent/venv/lib/python3.11/site-packages/exa_py/api.py z!snake_to_camel..Gs*"E"E17799"E"E"E"E"E"ErN)splitjoin)r) componentss r7snake_to_camelr=7sgIyFu%%J a=277"E"Ejn"E"E"EEE EEr9datadict skip_keys list[str]crt|tr fd|DS|S)z Convert keys in a dictionary from snake_case to camelCase recursively. Args: data (dict): The dictionary with keys in snake_case format. Returns: dict: The dictionary with keys converted to camelCase format. ci|]@\}}|t|t|tr|vrt|n|ASr3)r= isinstancer? to_camel_case)r5kvr@s r7 z!to_camel_case..Usf   1} 1  !T"" '( '9'9 -Q :::}}r9rDr?items)r>r@s `r7rErEJsS$           Kr9 camel_strc~tjdd|}tjdd|S)zConvert camelCase string to snake_case. Args: camel_str (str): The string in camelCase format. Returns: str: The string converted to snake_case format. z(.)([A-Z][a-z]+)z\1_\2z([a-z0-9])([A-Z]))resublower)rKr)s r7camel_to_snakerP_s9)8Y??I 6%x ; ; A A C CCr9clt|trd|DS|S)z Convert keys in a dictionary from camelCase to snake_case recursively. Args: data (dict): The dictionary with keys in camelCase format. Returns: dict: The dictionary with keys converted to snake_case format. c|i|]9\}}t|t|trt|n|:S)rPrDr? to_snake_caser5rFrGs r7rHz!to_snake_case..wsR   1 1  :a3F3FM}Q///A   r9rI)r>s r7rTrTlsF$         Kr9 entities_dataOptional[List[dict]]Optional[List]c\ |sdSddlm}m}mmm m m m m m m m dfd d fd d!fd d" fd d# fd d$ fd d% fd }d&fd }g}|D]}| d}|dkrL|||dd|d|| dii|dkrK|||dd|d|| di|r|ndS)'z Parse entity data from API response into Entity dataclasses. Handles field mapping for reserved Python keywords: - 'from' -> 'from_date' - 'to' -> 'to_date' Nr) CompanyEntity PersonEntityEntityCompanyPropertiesEntityPersonProperties EntityCompanyPropertiesWorkforce#EntityCompanyPropertiesHeadquarters!EntityCompanyPropertiesFinancials#EntityCompanyPropertiesFundingRound!EntityCompanyPropertiesWebTrafficEntityDateRange EntityPersonPropertiesCompanyRef&EntityPersonPropertiesWorkHistoryEntryr>Optional[dict]r+Optional[EntityDateRange]cr|sdS|d|dS)Nfromto) from_dateto_dateget)r>rcs r7parse_date_rangez)_parse_entities..parse_date_rangesE 4hhv&&HHTNN    r9*Optional[EntityPersonPropertiesCompanyRef]cr|sdS|d|dS)Nidname)rrrsrm)r>rds r7parse_company_refz*_parse_entities..parse_company_refsF 4//xx~~&!!    r9r?rec |d|d|d|dS)Nr4locationdatescompany)r4rvrwrxrm)r>rertros r7parse_work_history_entryz1_parse_entities..parse_work_history_entryso55((7##XXj))""488G#4#455%%dhhy&9&9::     r9-Optional[EntityCompanyPropertiesFundingRound]c|sdS|d|d|dS)Nrsdateamount)rsr|r}rm)r>ras r7parse_funding_roundz,_parse_entities..parse_funding_roundsW 422&!!&!!88H%%    r9+Optional[EntityCompanyPropertiesFinancials]c |sdS|d|d|dS)N revenueAnnual fundingTotalfundingLatestRound)revenue_annual funding_totalfunding_latest_roundrm)r>r`r~s r7parse_financialsz)_parse_entities..parse_financialssc 40088O44((>22!4!4TXX>R5S5S!T!T    r9+Optional[EntityCompanyPropertiesWebTraffic]cJ|sdS|dS)N visitsMonthly)visits_monthlyrm)r>rbs r7parse_web_trafficz*_parse_entities..parse_web_traffics9 40088O44    r9r\cZ|d}|d}|d|d|d|r|dnd|r[|d|d |d |d  nd|d |dS)N workforce headquartersrs foundedYear descriptiontotalraddresscity postalCodecountry)rr postal_coder financials webTraffic)rs founded_yearrrrr web_trafficrm)r>workforce_dataheadquarters_datar\r_r^rrs r7parse_company_propertiesz1_parse_entities..parse_company_propertiess:+.. HH^44&&&!!-00//]ku66^=O=OPW=X=XYYYYqu # -<<)--i88&**622-11,??)--i88  )-''(>(>??))$((<*@*@AA    r9r]c|d}|d|d|rfd|DndS)N workHistoryrsrvc&g|] }|SrSrS)r5whrys r7 zD_parse_entities..parse_person_properties..s%NNN222266NNNr9)rsrv work_historyrm)r>rr]rys r7parse_person_propertiesz0_parse_entities..parse_person_propertiessoxx .. %%&!!XXj))R^hNNNNNNNNdh    r9typerxrrversion properties)rrrrrperson)r>rfr+rg)r>rfr+rp)r>r?r+re)r>rfr+rz)r>rfr+r)r>rfr+r)r>r?r+r\)r>r?r+r]) exa_py.apirZr[r\r]r^r_r`rarbrcrdrernappend)rVrZr[rrentities entity_data entity_typer\r`rar_rbr^rcr]rdrertrorr~rrys @@@@@@@@@@@@@@@@r7_parse_entitiesr~s t                                                                             $       H$ !oof-- ) # # OOMMt$#I.33KOOLRT4U4UVV      H $ $ OOLLt$#I.22;??.s-CCq=**CCCCCCr9zInvalid value for option 'z': z. Expected one of )rJ ValueErrorany)rrkeyexpected_typesrs @r7validate_search_optionsrwsmmoo   U h  777788 8 = !#CCCCNCCCCC ^S^^U^^n^^    r9ct|tur|t|vSt|trt||SdS)NF)rr rrDr)r expected_types r7rrsO-  G++////-&&0%/// 5r9rawOptional[CostDollars]c|sdS|d}|dS|d}|d}t|||S)zb Parse the costDollars JSON into a CostDollars object, or return None if missing/invalid. Nrsearchr)rrr)rn CostDollars)rr search_part contents_parts r7parse_cost_dollarsrse t GGG  E }t''(##KGGJ''M U; O O OOr9)compactstandardfull) unspecifiedheader navigationbannerbodysidebarfooterrcFeZdZUdZded<ded<ded<ded <ded <d S) TextContentsOptionsa7A class representing the options that you can specify when requesting text Attributes: max_characters (int): The maximum number of characters to return. Default: None (no limit). include_html_tags (bool): If true, include HTML tags in the returned text. Default false. verbosity (VERBOSITY_OPTIONS): Controls verbosity level of returned content. "compact" (default): main content only; "standard": balanced; "full": all sections. Requires max_age_hours=0 to take effect. include_sections (List[SECTION_TAG]): Only include content from these semantic sections. Requires max_age_hours=0 to take effect. exclude_sections (List[SECTION_TAG]): Exclude content from these semantic sections. Requires max_age_hours=0 to take effect. intmax_charactersboolinclude_html_tagsVERBOSITY_OPTIONS verbosityzList[SECTION_TAG]include_sectionsexclude_sectionsNrrSr9r7rrsZ      ''''''''''r9rceZdZUdZded<ded<ded<ded<ded <d ed <d ed <ded<ded<ded<ded<ded<ded<ded<ded<dS) JSONSchemaa(Represents a JSON Schema definition used for structured summary output. .. deprecated:: 1.15.0 Use Pydantic models or dict[str, Any] directly instead. This will be removed in a future version. To learn more visit https://json-schema.org/overview/what-is-jsonschema. r*r-r4rzLLiteral['object', 'array', 'string', 'number', 'boolean', 'null', 'integer']rzDict[str, JSONSchema]rz#Union[JSONSchema, List[JSONSchema]]rJrrr enumzUnion[bool, JSONSchema]additionalProperties definitionspatternPropertieszList[JSONSchema]allOfanyOfoneOfr.NrrSr9r7rrsLLLJJJVVVV%%%%....JJJ1111&&&&,,,,r9rc(eZdZUdZded<ded<dS)SummaryContentsOptionsaA class representing the options that you can specify when requesting summary Attributes: query (str): The query string for the summary. Summary will bias towards answering the query. schema (Union[BaseModel, dict[str, Any]]): JSON schema for structured output from summary. Can be a Pydantic model (automatically converted) or a dict containing JSON Schema. r*rr schemaNrrSr9r7r!r!s3JJJr9r!c<eZdZUdZded<ded<ded<ded<dS) HighlightsContentsOptionsa0A class representing the options that you can specify when requesting highlights. Attributes: query (str): The query string for highlight generation. Highlights will be biased towards this query. max_characters (int): The maximum number of characters to return for highlights. Default: None (server default). num_sentences (int): Deprecated. Use max_characters instead. The number of sentences per highlight. highlights_per_url (int): Deprecated. Use max_characters instead. The number of highlights to return per URL. r*rrr num_sentenceshighlights_per_urlNrrSr9r7r$r$sKJJJr9r$ceZdZUdZded<dS)ContextContentsOptionsaYOptions for retrieving aggregated context from a set of search results. .. deprecated:: Use ``highlights`` or ``text`` instead. The ``context`` option is deprecated and will be removed in a future version. Attributes: max_characters (int): The maximum number of characters to include in the context string. rrNrrSr9r7r(r(s*r9r(c(eZdZUdZded<ded<dS) ExtrasOptionszFA class representing additional extraction fields (e.g. links, images)rlinks image_linksNrrSr9r7r*r*s-PPJJJr9r*cxeZdZUdZded<ded<ded<ded <d ed <d ed <d ed<d ed<ded<ded<dS)ContentsOptionsaOptions for retrieving page contents in search and find_similar methods. All fields are optional. If no content options are specified, text with max_characters=10000 is returned by default. Attributes: text (TextContentsOptions | True): Options for text extraction, or True for defaults. highlights (HighlightsContentsOptions | True): Options for highlight extraction, or True for defaults. summary (SummaryContentsOptions | True): Options for summary generation, or True for defaults. context (ContextContentsOptions | True): Deprecated. Use ``highlights`` or ``text`` instead. Will be removed in a future version. max_age_hours (int): Maximum age of cached content in hours. If content is older, it will be fetched fresh. Special values: 0 = always fetch fresh content, -1 = never fetch fresh (use cached content only). Example: 168 = fetch fresh for pages older than 7 days. subpages (int): Number of subpages to crawl. subpage_target (str | List[str]): Target subpage path(s) to crawl. extras (ExtrasOptions): Additional extraction options (links, images). )Union[TextContentsOptions, Literal[True]]rz/Union[HighlightsContentsOptions, Literal[True]]r,Union[SummaryContentsOptions, Literal[True]]rz,Union[ContextContentsOptions, Literal[True]]rLIVECRAWL_OPTIONSrrrrrzUnion[str, List[str]]rr*rNrrSr9r7r.r.s$4333????99999999    MMM))))r9r.c(eZdZUdZded<ded<dS)CostDollarsSearchz)Represents the cost breakdown for search.floatrkeywordNrrSr9r7r3r39s(33MMMNNNNNr9r3c(eZdZUdZded<ded<dS)CostDollarsContentsz+Represents the cost breakdown for contents.r4rrNrrSr9r7r7r7@s(55KKKNNNNNr9r7c:eZdZUdZded<dZded<dZded<dS) rz1Represents costDollars field in the API response.r4rNr3rr7r)rrrrrrrrSr9r7rrGsC;;LLL $F$$$$$(H((((((r9rc"eZdZUdZdZded<dS)r^zCompany workforce information.N Optional[int]r)rrrrrrrSr9r7r^r^Ts)((Er9r^cLeZdZUdZdZded<dZded<dZded<dZded<dS)r_z!Company headquarters information.N Optional[str]rrrr) rrrrrrrrrrSr9r7r_r_[s\++!G!!!!D!%K%%%%!G!!!!!!r9r_c>eZdZUdZdZded<dZded<dZded<dS)razFunding round information.Nr<rsr|r:r})rrrrrsrr|r}rSr9r7raraesK$$DD F      r9rac>eZdZUdZdZded<dZded<dZded<dS)r`zCompany financial information.Nr:rrrzr)rrrrrrrrrSr9r7r`r`nsL(($(N((((#'M''''JNNNNNNNr9r`c"eZdZUdZdZded<dS)rbz Company web traffic information.Nr:r)rrrrrrrSr9r7rbrbws)**$(N((((((r9rbcveZdZUdZdZded<dZded<dZded<dZded <dZ d ed <dZ d ed <dZ ded<dS)r\z+Structured properties for a company entity.Nr<rsr:rrz*Optional[EntityCompanyPropertiesWorkforce]rz-Optional[EntityCompanyPropertiesHeadquarters]rrrrr) rrrrrsrrrrrrrrSr9r7r\r\~s55D"&L&&&&!%K%%%%<@I@@@@BFLFFFF>BJBBBB?CKCCCCCCr9r\c0eZdZUdZdZded<dZded<dS)rcz$Date range for work history entries.Nr<rkrl)rrrrrkrrlrSr9r7rcrcs:..#I####!G!!!!!!r9rcc0eZdZUdZdZded<dZded<dS)rdz'Reference to a company in work history.Nr<rrrs)rrrrrrrrsrSr9r7rdrds:11BDr9rdcLeZdZUdZdZded<dZded<dZded<dZded <dS) rez)A single work history entry for a person.Nr<r4rvrgrwrprx) rrrrr4rrvrwrxrSr9r7reres\33E"H""""'+E++++:>G>>>>>>r9rec>eZdZUdZdZded<dZded<dZded<dS)r]z*Structured properties for a person entity.Nr<rsrvz6Optional[List[EntityPersonPropertiesWorkHistoryEntry]]r)rrrrrsrrvrrSr9r7r]r]sK44D"H""""KOLOOOOOOr9r]c<eZdZUdZded<ded<ded<ded <d S) rZz%Structured entity data for a company.r*rrzLiteral['company']rrrr\rNrrSr9r7rZrZsB// GGGLLL''''''r9rZc<eZdZUdZded<ded<ded<ded <d S) r[z$Structured entity data for a person.r*rrzLiteral['person']rrrr]rNrrSr9r7r[r[sB.. GGGLLL&&&&&&r9r[ceZdZUdZded<ded<dZded<dZded <dZded <dZded <dZ ded <dZ ded <dZ ded<dZ ded<dZ ded< ddZdZdS)_ResultaxA class representing the base fields of a search result. Attributes: title (str): The title of the search result. url (str): The URL of the search result. id (str): The temporary ID for the document. score (float, optional): A number from 0 to 1 representing similarity. published_date (str, optional): An estimate of the creation date, from parsing HTML content. author (str, optional): The author of the content (if available). image (str, optional): A URL to an image associated with the content (if available). favicon (str, optional): A URL to the favicon (if available). subpages (List[_Result], optional): Subpages of main page extras (Dict, optional): Additional metadata; e.g. links, images. entities (List[Entity], optional): Structured entity data for company or person searches. r*rrrNr<r4Optional[float]scorepublished_dateauthorimagefaviconzOptional[List[_Result]]rzOptional[Dict]rzOptional[List[Entity]]rc ||_||_||_||_||_||_||_||_| |_| |_ | |_ dSr3) rrrr4rJrKrLrMrNrrr) selfrrrr4rJrKrLrMrNrrrs r7__init__z_Result.__init__sV  ,        r9cd|jd|jd|jd|jd|jd|jd|jd|jd |jd |j d }|j r-d d |j D}|d |d z }|S)NTitle:  URL:  ID: z Score:  Published Date: Author: z Image: z Favicon: z Extras: z Subpages:  c3HK|]}d|jd|jjpdVdS)z - [z] UnknownN)rrrs)r5es r7r8z"_Result.__str__.. sS%%CBB!,"3"@yBB%%%%%%r9z Entities: ) r4rrrrJrKrLrMrNrrrr;)rPresult entities_strs r7__str__z_Result.__str__s +dj + +H + +7 + +j + + $2 + + { + + j + +  + +{ + + + + +  = 599%%%%%L 4L444 4F r9) NNNNNNNNN)rrrrrr4rJrKrLrMrNrrrrQr^rSr9r7rHrHs   HHH GGGE!E!!!!$(N(((( F    E!G!!!!(,H,,,,!F!!!!'+H++++ !!!!4r9rHceZdZUdZdZded<dZded<dZded<dZded < d fd Z fd Z xZ S) Resulta A class representing a search result with optional text, summary, and highlights. Attributes: text (str, optional): The text content of the page. summary (str, optional): A summary of the page content. highlights (List[str], optional): Relevant sentences from the page. highlight_scores (List[float], optional): Scores for each highlight. Nr<rrOptional[List[str]]rzOptional[List[float]]highlight_scoresc t||||||||| | | | |_| |_||_||_dSr3)superrQrrrrb)rPrrrr4rJrKrLrMrNrrrrrrrb __class__s r7rQzResult.__init__$sg$               $ 0r9ct}|d|jd|jdz}|jr|d|jdz }|jr|d|jdz }|S)NText: z Summary: rXz Highlights: zHighlight Scores: )rdr^rrrrb)rPbase_strr\res r7r^zResult.__str__Hs77??$$KTYKK4<KKKK ? 9 8T_888 8F   E D4+@DDD DF r9) NNNNNNNNNNNNN) rrrrrrrrrbrQr^ __classcell__res@r7r`r`sD!G!!!!&*J****.22222  !"1"1"1"1"1"1Hr9r`cpeZdZUdZejeZded< d fd Z fdZ xZ S) ResultWithTextz A class representing a search result with text present. Attributes: text (str): The text of the search result page. default_factoryr*rNr1c lt||||||||| | | | |_dSr3)rdrQr)rPrrrr4rJrKrLrMrNrrrrres r7rQzResultWithText.__init__]sO              r9c`t}|d|jdzS)NrgrX)rdr^rrPrhres r7r^zResultWithText.__str__{s/77??$$04900000r9 NNNNNNNNNr1) rrrr dataclassesfieldr*rrrQr^rirjs@r7rlrlRs" !#666D6666  <111111111r9rlcpeZdZUdZejeZded< d fd Z fdZ xZ S) ResultWithSummaryzk A class representing a search result with summary present. Attributes: summary (str) rmr*rNr1c lt||||||||| | | | |_dSr3)rdrQr)rPrrrr4rJrKrLrMrNrrrrres r7rQzResultWithSummary.__init__sO              r9c`t}|d|jdzS)N Summary: rX)rdr^rrqs r7r^zResultWithSummary.__str__s/77??$$6dl66666r9rr) rrrrrsrtr*rrrQr^rirjs@r7rvrvs%;$S999G9999 <777777777r9rvceZdZUdZejeZded<ejeZ ded< d fd Z fd Z xZ S) ResultWithTextAndSummaryz A class representing a search result with text and summary present. Attributes: text (str) summary (str) rmr*rrNr1c zt||||||||| | | | |_| |_dSr3)rdrQrr)rPrrrr4rJrKrLrMrNrrrrrres r7rQz!ResultWithTextAndSummary.__init__sV               r9cxt}|d|jdzd|jdzS)NrgrXry)rdr^rrrqs r7r^z ResultWithTextAndSummary.__str__sB77??$$04900003Ot|3O3O3OOOr9) NNNNNNNNNr1r1) rrrrrsrtr*rrrrQr^rirjs@r7r{r{s" !#666D6666$;$S999G9999  @PPPPPPPPPr9r{cpeZdZUdZded<ded<dZded<dZded<dZded <dZded < d d Z d Z dS) AnswerResultaA class representing a result for an answer. Attributes: title (str): The title of the search result. url (str): The URL of the search result. id (str): The temporary ID for the document. published_date (str, optional): An estimate of the creation date, from parsing HTML content. author (str, optional): If available, the author of the content. text (str, optional): The full page text from each search result. r*rrrNr<r4rKrLrcZ||_||_||_||_||_||_dSr3rrrr4rKrLr)rPrrrr4rKrLrs r7rQzAnswerResult.__init__s3 ,  r9c hd|jd|jd|jd|jd|jd|jd S)NrSrTrUrVrW Text:  )r4rrrrKrLrrPs r7r^zAnswerResult.__str__sq %dj % %H % %7 % % $2 % %{  % % Y  % % % r9)NNNN) rrrrrr4rKrLrrQr^rSr9r7rrs   GGG HHHE$(N(((( F    DKO     r9rc@eZdZUdZdZded<dZded<d d Zd d ZdS) StreamChunkzA class representing a single chunk of streaming data. Attributes: content (Optional[str]): The partial text content of the answer citations (Optional[List[AnswerResult]]): List of citations if provided in this chunk Nr<contentzOptional[List[AnswerResult]] citationsr+rc&|jdup|jduS)z&Check if this chunk contains any data.Nrrrs r7has_datazStreamChunk.has_datas|4'E4>+EEr9r*chd}|jr ||jz }|jr|dz }|jD] }|d|z } |S)z"Format the chunk data as a string.r1z Citations:rXrrPoutputsources r7r^zStreamChunk.__str__sY < # dl "F > ( n $F. ( (-v--' r9)r+r)r+r*) rrrrrrrrr^rSr9r7rr sp"G!!!!.2I2222FFFF      r9rc<eZdZUdZded<ded<dZded<d ZdS) AnswerResponsea(A class representing the response for an answer operation. Attributes: answer (str): The generated answer. citations (List[AnswerResult]): A list of citations used to generate the answer. cost_dollars (CostDollars, optional): The cost breakdown for this request. Union[str, dict[str, Any]]answerzList[AnswerResult]rNr cost_dollarscd|jd}|jD]U}|d|jz }|d|jz }|d|jz }|d|jz }|d|jz }|d|jz }|d z }V|S) NzAnswer: z Citations:z Title: rUrTrVrWrrX)rrr4rrrrKrLrrs r7r^zAnswerResponse.__str__3s7DK777n  F 0&,00 0F *vy** *F , ,, ,F B6+@BB BF 26=22 2F ... .F dNFF r9)rrrrrrr^rSr9r7rr%sZ'&&&!!!!*.L....     r9rc0eZdZdZd dZdZd dZdd Zd S)StreamAnswerResponse1A class representing a streaming answer response. raw_responserequests.Responsec<||_|dSr3 _raw_response_ensure_ok_statusrPrs r7rQzStreamAnswerResponse.__init__C")      r9cx|jjdkr)td|jjd|jjdSN Request failed with status code : r status_coderrrs r7rz&StreamAnswerResponse._ensure_ok_statusGM   )S 0 0n43E3QnnUYUgUlnn  1 0r9r+Iterator[StreamChunk]c#nK|jD]}|s|dd} t j|}n#tj$rYUwxYwd}d}d|vr?|dr7d|ddvr'|dddd}d|vr|dr|ddkrg}|dD]}t|}| t|d |d |d |d |d |dt||}| r|VdS)Nzutf-8data: choicesdeltarrrnullrrrr4rKrLrrr) r iter_linesdecode removeprefixjsonloadsJSONDecodeErrorrnrTrrrr) rPline decoded_linechunkrrssnake_s stream_chunks r7__iter__zStreamAnswerResponse.__iter__Ms&1133% #% #D ;;w//<>yIIGu$$+&%+&&00 {+  A+A..G$$$&{{400 ' E 2 2")++g"6"6+2;;7G+H+H#*;;x#8#8!(V!4!4     'w)LLLL$$&& #""""K% #% #s A  A21A2rc8|jdSz@Close the underlying raw response to release the network socket.Nrclosers r7rzStreamAnswerResponse.closeu   """""r9N)rr)r+rr+r)rrrrrQrrrrSr9r7rr@sf;;!!!! &#&#&#&#P######r9rc.eZdZdZd dZdZdZd d Zd S) AsyncStreamAnswerResponserrhttpx.Responsec<||_|dSr3rrs r7rQz"AsyncStreamAnswerResponse.__init__}rr9cx|jjdkr)td|jjd|jjdSrrrs r7rz+AsyncStreamAnswerResponse._ensure_ok_statusrr9c"fd}|S)NcVKj23d{V}|s |d} tj|}n#tj$rYFwxYwd}d}d|vr?|dr7d|ddvr'|dddd}d|vr|dr|ddkrg}|dD]}t|}|t|d|d |d |d |d |d t||}| r|WV6dS)Nrrrrrrrrrrr4rKrLrrr) r aiter_linesrrrrrnrTrrrr) rrrrrrrrrPs r7 generatorz6AsyncStreamAnswerResponse.__aiter__..generators "0<<>>% '% '% '% '% '% '% 'd#00::  J|44EE+H %%% *:%% "21"555"' "21"5g">"B"B9"M"M 5((k*)k*f44 "I";/  "/"2"2!(((#*;;t#4#4$+KK$6$6&-kk'&:&:/6{{;K/L/L'.{{8'<'<%,[[%8%8      +7iPPP ((**'&&&&&K?>>sF(AA$#A$rS)rPrs` r7 __aiter__z#AsyncStreamAnswerResponse.__aiter__s+& '& '& '& '& 'Py{{r9r+rc8|jdSrrrs r7rzAsyncStreamAnswerResponse.closerr9N)rrr)rrrrrQrrrrSr9r7rrzsa;;!!!! )))V######r9rTc2eZdZUdZded<ded<ded<dS) ContentStatuszAA class representing the status of a content retrieval operation.r*rrstatusrNrrSr9r7rrs1KK GGGKKKKKKKKr9rc~eZdZUdZded<ded<ded<dZded<dZd ed <dZd ed <dZd ed<dZ ded<dZ dS)SearchResponseaA class representing the response for a search operation. Attributes: results (List[Result]): A list of search results. resolved_search_type (str, optional): 'neural' or 'keyword' if auto. auto_date (str, optional): A date for filtering if autoprompt found one. context (str, optional): Deprecated. Combined context string when requested via contents.context. Use highlights or text instead. output (DeepSearchOutput, optional): Deep search synthesized output object containing content and field-level grounding. statuses (List[ContentStatus], optional): Status list from get_contents. cost_dollars (CostDollars, optional): Cost breakdown. search_time (float, optional): Time taken for the search in milliseconds. zList[T]resultsr<resolved_search_type auto_dateNrzOptional['DeepSearchOutput']rzOptional[List[ContentStatus]]statusesrrrI search_timecdd|jD}|jr |d|jz }|j |d|jz }|jr |d|jz }|j|d|jdz }|jrN|d|jjz }|jjr|d |jjz }|jj r|d |jj z }|j r |d |j z }|S) Nrc34K|]}t|VdSr3)r*)r5r\s r7r8z)SearchResponse.__str__..s(DDVS[[DDDDDDr9z Context: z Output: z Resolved Search Type: z Search Time: msz CostDollars: total=z - search: z - contents: z Statuses: ) r;rrrrrrrrrr)rPrs r7r^zSearchResponse.__str__s=DDt|DDDDD < 3 2DL22 2F ; " 04;00 0F  $ M L1JLL LF   ' <(8<<< -GIII = 5 4T]44 4F r9) rrrrrrrrrrr^rSr9r7rrs  ''''!G!!!!+/F////.2H2222*.L....#'K''''r9rc(eZdZUdZded<ded<dS)!DeepSearchOutputGroundingCitationz0Citation metadata for one grounded output field.r*rr4NrrSr9r7rrs(:: HHHJJJJJr9rlowmediumhighc2eZdZUdZded<ded<ded<dS) DeepSearchOutputGroundingzCGrounding metadata for one field in deep-search synthesized output.r*rtz'List[DeepSearchOutputGroundingCitation]r#DeepSearchOutputGroundingConfidence confidenceNrrSr9r7rrs9MMJJJ6666333333r9rc(eZdZUdZded<ded<dS)DeepSearchOutputz'Deep-search synthesized output payload.rrzList[DeepSearchOutputGrounding] groundingNrrSr9r7rrs011''''......r9rrOptional[DeepSearchOutput]c t|tsdS|d}t|tr|}nt|tr|}nd}g}|d}t|trJ|D]F}t|ts|d}t|tsDg}|d}t|tr|D]} t| ts| d} | d} t| tsX|t | t| tr| nd |d } | d vrd } |t||| Ht||S)zParse deep-search output into a typed object. Args: raw: Raw `output` field from API response. Returns: Parsed DeepSearchOutput when the payload is an object, otherwise None. Nrr1rrtrrr4)rr4rrr)rtrr)rr) rDr?rnr*listrrrr) r raw_contentrr raw_grounding grounding_rowrtr raw_citationscitationrr4rs r7parse_deep_search_outputr s c4 t'')$$K+s##.9 K & &13IGGK((M-&&$*# # MmT22 !%%g..EeS)) ACI)--k::M-..  -  H%h55! ",,u--C$LL11E%c3//! $$9 #+5eS+A+A"I%%r'**<88J!:::%   )')     Gy A A AAr9 original_dictrfields_to_nestrnew_keycVi}|D]}||vr||||<|||<|Sr3)pop)rrr nested_dictrts r7 nest_fieldsrMsNK :: M ! !!.!2!25!9!9K )M' r9ceZdZdZ dndod Z dpdqdZddddddddddddddddddddrd2Zdsd3Ze dtdudAZ edddddddddBdvdEZ edddddddddBdwdIZ edddddddddBdxdKZ dydLZ ddddddddddddddM dzdPZ eddddddddddddddddddddQd{dRZ eddddddddddddddddddddQd|dSZ eddddddddddddddddddddQd|dTZ eddddddddddddddddddddQd}dUZ eddddddddddddddddddddQd~dVZ ddWZ ddZZ dd`Z edadadddddbddhZdadadddddbddiZdadddddjddmZdS)Exaz&A client for interacting with Exa API.https://api.exa.aiNapi_keyr<base_urlr* user_agentc*|/ddl}|jd}|td|dt }||_||dd|_t||_t||_ t||_ dS)a~Initialize the Exa client with the provided API key and optional base URL and user agent. Args: api_key (str): The API key for authenticating with the Exa API. base_url (str, optional): The base URL for the Exa API. Defaults to "https://api.exa.ai". user_agent (str, optional): Custom user agent. Defaults to "exa-py {version}". Nr EXA_API_KEYzNAPI key must be provided as an argument or in EXA_API_KEY environment variablezexa-py/zapplication/json)z x-api-keyz User-Agentz Content-Type) osenvironrnrrrheadersr!websetsr%researchr#monitors)rPrrrrs r7rQz Exa.__init__as ? IIIjnn]33G d  ;#7#9#9;;J   $.   %T** &t,, ,T22 r9POSTendpointr>$Optional[Union[Dict[str, Any], str]]methodparamsOptional[Dict[str, Any]]rOptional[Dict[str, str]]r+(Union[Dict[str, Any], requests.Response]c,t|tr|}n|rtj|tnd}|r*t|t r|dp|o|ddk}i|j}|r||| dkrD|r"tj|j |z||d} | Stj|j |z||} n| d krD|r"tj |j |z||d } | Stj |j |z|| } n| d kr tj |j |z|| } nI| d krtj|j |z|} ntd|| jdkrtd| jd| j| S)a Send a request to the Exa API, optionally streaming if data['stream'] is True. Args: endpoint (str): The API endpoint (path). data (dict, optional): The JSON payload to send. Defaults to None. method (str, optional): The HTTP method to use. Defaults to "POST". params (Dict[str, Any], optional): Query parameters to include. Defaults to None. headers (Dict[str, str], optional): Additional headers to include in the request. Defaults to None. Returns: Union[dict, requests.Response]: If streaming, returns the Response object. Otherwise, returns the JSON-decoded response as a dict. Raises: ValueError: If the request fails (non-200 status code). )clsNstreamtrueGETT)rr r)rr r)r>rr)r>rPATCHDELETErUnsupported HTTP method: irr)rDr*rdumpsr"r?rnrupdateupperrequestsrpostpatchdeleterrr) rPrr>r r r json_dataneeds_streamingrequest_headersress r7requestz Exa.requests`2 dC  OIIAEN 4^<<<<$I QJtT$:$:Qtxx?Q?Q  5vzz(++v5  +T\*  ,  " "7 + + + <<>>U " " lMH,+!   lMH,of\\^^v % % mMH,"+   mMH,9o\\^^w & &. (y/CC\\^^x ' '/$-(":OTTTCCAAABB B ?c ! !P3?PPchPP xxzzr9rrrrrrrrrrrrrrrrrrrr0Optional[Union[ContentsOptions, Literal[False]]]rr:rrarrrrrrrr Optional[Union[SearchType, str]]rOptional[Category]rrOptional[bool]rrrrOptional[DeepOutputSchema]SearchResponse[Result]c\dtD}|dur|ddn|d|vrddtii|d<n|||d<t |t t |dg}|d |}t| d }g}|d D]m}t|}| t| d | d | d| d| d| d| d| d| d| d| d| d| d| dt| dot|d|vr|dndd|vr|dnd| dt| d|| d S)!a Perform a search. By default, returns text contents with 10,000 max characters. Use contents=False to opt-out. Args: query (str): The query string. contents (ContentsOptions | False, optional): Options for retrieving page contents. Defaults to {"text": {"maxCharacters": 10000}}. Use False to disable contents. See ContentsOptions for available options (text, highlights, summary, etc.). Note: The ``context`` option is deprecated; use ``highlights`` or ``text`` instead. num_results (int, optional): Number of search results to return. Default 10. include_domains (List[str], optional): Domains to include in the search. exclude_domains (List[str], optional): Domains to exclude from the search. start_crawl_date (str, optional): Only links crawled after this date. end_crawl_date (str, optional): Only links crawled before this date. start_published_date (str, optional): Only links published after this date. end_published_date (str, optional): Only links published before this date. include_text (List[str], optional): Strings that must appear in the page text. exclude_text (List[str], optional): Strings that must not appear in the page text. type (SearchType, optional): Search type - 'auto' (default), 'fast', 'deep', 'deep-reasoning', 'neural', or 'instant'. category (Category, optional): Data category to focus on (e.g. 'company', 'news', 'research paper'). flags (List[str], optional): Experimental flags for Exa usage. moderation (bool, optional): If True, the search results will be moderated for safety. user_location (str, optional): Two-letter ISO country code of the user (e.g. US). additional_queries (List[str], optional): Alternative query formulations for deep search to skip automatic LLM-based query expansion. Max 5 queries. Only applicable when type is 'deep' or 'deep-reasoning'. Example: ["machine learning", "ML algorithms", "neural networks"] system_prompt (str, optional): Deep-search-only instructions that guide both the search process and the final returned result. Use this to prefer certain sources, emphasize novelty, avoid duplicates, or constrain the response style. Only applicable when type is 'deep' or 'deep-reasoning'. output_schema (DeepOutputSchema, optional): Deep output schema for deep search. Use ``{"type": "text", "description": ...}`` for plain text output or ``{"type": "object", "properties": ..., "required": ...}`` for structured JSON. For object schemas, max nesting depth is 2 and max total properties is 10. Only applicable when type is 'deep' or 'deep-reasoning'. Returns: SearchResponse: The response containing search results, etc. Examples: # Basic search result = exa.search( "hottest AI startups", num_results=2, contents={"highlights": True} ) # Deep search with query variations deep_result = exa.search( "blog post about AI", type="deep", additional_queries=["AI blogpost", "machine learning blogs"], num_results=5 ) c*i|]\}}|dk | ||SrrSrUs r7rHzExa.search..*'VVVDAqV  1a r9FrNrrrr@/search costDollarsrrrrr4rJrKrLrMrNrrrrrbrrrrr4rJrKrLrMrNrrrrrrbrresolvedSearchTypeautoDaterr searchTimerrrr)localsrJrDEFAULT_MAX_CHARACTERSrSEARCH_OPTIONS_TYPESrEr#rrnrTrr`rrrrPrrrrrrrrrrrrrrrrrrrrr>rrr\ snake_results r7rz Exa.searchs`WVFHHNN$4$4VVV u   KK D ) ) ) )  *G";";#),<>T+U"VGJ    !"*GJ )=>>>O3DEEE||Iw//)$((=*A*ABB 9o  F(00L NN$((//#''--&**733&**733#/#3#34D#E#E'++H55&**733(,,Y77)--j99'++H55%))&11(,,Y77+// ==%1%5%56H%I%I,VZZ -C-CDD    & *>$*F*FD% & &D *d 2 2D  HHY''+DHHX,>,>??%..    r9c rd|i}|D] \}}||||< d|vrd|vrd|vr dti|d<i}|t|t|t t ||d|vr?t|dtr$|d}d|vrt|d|d<t|gdd }t|dg }| d |}t|d } g} |d D]m} t| } | t#| d| d| d| d| d| d| d| d| d| d| d| d| d| dt%| dot'| |d|d|dt)|d| |d S)!t DEPRECATED: Use search() instead. The search() method now returns text contents by default. Migration: - search_and_contents(query) → search(query) - search_and_contents(query, text=True) → search(query, contents={"text": True}) - search_and_contents(query, summary=True) → search(query, contents={"summary": True}) rNrrrrr" rrrrrrrrrrrr.r/r0rrrrr4rJrKrLrMrNrrrbrr1r2r3rrr4r5)rJr7rr8CONTENTS_OPTIONS_TYPESCONTENTS_ENDPOINT_OPTIONS_TYPESrrDr?rrrEr#rrnrTrr`rrr rPrkwargsrrFrGmerged_options summary_optsr>rrr\r:s r7search_and_contentszExa.search_and_contents[sAE"LLNN  DAq}  ' ! !((''/1GHGFO23334555=>>>888   Jwy/A4$H$H "9-L<'')>|H?U)V)V X&         H:>>>||Iw//)$((=*A*ABB 9o  F(00L NN$((//#''--&**733&**733#/#3#34D#E#E'++H55&**733(,,Y77)--j99'++H55%))&11(,,Y77+// ==%1%5%56H%I%I,VZZ -C-CDD    &  HH) * * HHZ HHY''+DHHX,>,>??%..    r9r$Union[str, List[str], List[_Result]]rrOptional[LIVECRAWL_OPTIONS]rrrrOptional[Union[str, List[str]]]rOptional[ExtrasOptions]SearchResponse[ResultWithText]c dSr3rS) rPrrrrrrrrrs r7 get_contentszExa.get_contentss *-r9)rrrrrrrrrr/c dSr3rS) rPrrrrrrrrrrs r7rKzExa.get_contentss *-r9rr0!SearchResponse[ResultWithSummary]c dSr3rS) rPrrrrrrrrrrs r7rKzExa.get_contentss -0Cr9(SearchResponse[ResultWithTextAndSummary]c dSr3rS) rPrrrrrrrrrrrs r7rKzExa.get_contentss 473r9c t|tr|g}nOt|tr:t|dkr't|dtr d|D}d|i}|D]\}}|dkr||||<d|vrd|vrd|vr d t i|d<i}|t|tt||d|vr?t|dtr$|d}d |vrt|d |d <t|d g}|d |}t|d } g} |d gD]`} | t%| d| d| dag} |dD]m} t'| }| t)|d|d|d|d|d|d|d|d|d|d|d|d|d|dt+| dot-| |d|d |d!| | |d"#S)$aRetrieve contents for a list of URLs. Args: urls (str | List[str] | List[Result]): A single URL, list of URLs, or list of Result objects. text (TextContentsOptions | True, optional): Options for text extraction. summary (SummaryContentsOptions | True, optional): Options for summary generation. max_age_hours (int, optional): Maximum age of cached content in hours. If content is older, it will be fetched fresh. Special values: 0 = always fetch fresh content, -1 = never fetch fresh (cache only). Example: 168 = fetch fresh for pages older than 7 days. filter_empty_results (bool, optional): Whether to filter out empty results. subpages (int, optional): Number of subpages to retrieve. subpage_target (str | List[str], optional): Target subpages to retrieve. extras (ExtrasOptions, optional): Options for extra content (links, image_links). flags (List[str], optional): Experimental flags. Returns: SearchResponse[Result]: The response containing the contents of the URLs. Examples: # Get contents for a single URL contents = exa.get_contents("https://example.com/article") # Get contents for multiple URLs contents = exa.get_contents([ "https://example.com/article1", "https://example.com/article2" ]) rcg|] }|j SrSrr5rs r7rz$Exa.get_contents..(((aAE(((r9rrPNrrrrr" /contentsr0rrrrrrrrrrrr4rJrKrLrMrNrrrbrr1r2r3rr4rrrr)rDr*rlenrHrJr7rr>r?rr?rrEr#rrnrrrTr`rrrPrrArrFrGrBrCr>rrrrr\r:s r7rKzExa.get_contentss< dC  )6DD d # # )D A *T!Wg:V:V ((4(((D4.LLNN  DAqF{{q}  ' ! !((''/1GHGFO4555=>>>888   Jwy/A4$H$H "9-L<'')>|H?U)V)V X&(44||K11)$((=*A*ABB hhz2..  F OOzz$''!::h//!::h//    9o  F(00L NN$((//#''--&**733&**733#/#3#34D#E#E'++H55&**733(,,Y77)--j99'++H55%))&11(,,Y77+// ==%1%5%56H%I%I,VZZ -C-CDD    &  HH) * * HHZ HHY''%..    r9 rrrrrrrrrrrrrrrc dtD}|dur|ddn|d|vrddtii|d<n|||d<t |t t |}|d|}t| d}g}|d D]m}t|}| t| d | d | d | d | d| d| d| d| d| d| d| d| d| dt| dot|| d| d|| dS)aAFinds similar pages to a given URL, potentially with domain filters and date filters. By default, returns text contents with 10,000 max characters. Use contents=False to opt-out. Args: url (str): The URL to find similar pages for. contents (ContentsOptions | False, optional): Options for retrieving page contents. Defaults to {"text": {"maxCharacters": 10000}}. Use False to disable contents. See ContentsOptions for available options (text, highlights, summary, etc.). num_results (int, optional): Number of results to return. Default is None (server default). include_domains (List[str], optional): Domains to include in the search. exclude_domains (List[str], optional): Domains to exclude from the search. start_crawl_date (str, optional): Only links crawled after this date. end_crawl_date (str, optional): Only links crawled before this date. start_published_date (str, optional): Only links published after this date. end_published_date (str, optional): Only links published before this date. include_text (List[str], optional): Strings that must appear in the page text. exclude_text (List[str], optional): Strings that must not appear in the page text. exclude_source_domain (bool, optional): Whether to exclude the source domain. category (Category, optional): Data category to focus on (e.g. 'company', 'news', 'research paper'). flags (List[str], optional): Experimental flags. Returns: SearchResponse[Result] Examples: similar_results = exa.find_similar( "miniclip.com", num_results=2, exclude_source_domain=True ) c*i|]\}}|dk | ||SrrSrUs r7rHz$Exa.find_similar..r-r9FrNrr /findSimilarr0rrrrr4rJrKrLrMrNrrrrrbrr1r2r3r4rr)r6rJrr7rFIND_SIMILAR_OPTIONS_TYPESrEr#rrnrTrr`rrrPrrrrrrrrrrrrrrrr>rrr\r:s r7 find_similarzExa.find_similarWstdWVFHHNN$4$4VVV u   KK D ) ) ) )  *G";";#),<>T+U"VGJ    !"*GJ )CDDD((||NG44)$((=*A*ABB 9o  F(00L NN$((//#''--&**733&**733#/#3#34D#E#E'++H55&**733(,,Y77)--j99'++H55%))&11(,,Y77+// ==%1%5%56H%I%I,VZZ -C-CDD    &  HH) * * HHZ %..     r9)rrrrrrrrrrrrrrrrrrrcdSr3rS)rPrrrrrrrrrrrrrrrrrrrrs r7find_similar_and_contentszExa.find_similar_and_contentss 0*-r9cdSr3rSrPrrrrrrrrrrrrrrrrrrrrrs r7rezExa.find_similar_and_contents 2*-r9cdSr3rSrgs r7rezExa.find_similar_and_contentsrhr9cdSr3rS)rPrrrrrrrrrrrrrrrrrrrrrs r7rezExa.find_similar_and_contentss 2-0Cr9cdSr3rS)rPrrrrrrrrrrrrrrrrrrrrrrs r7rezExa.find_similar_and_contents#s 4473r9c (d|i}|D] \}}||||< d|vrd|vr dti|d<i}|t|t|t t ||d|vr?t|dtr$|d}d|vrt|d|d<t|gdd}t|dg }| d |}t|d } g} |d D]m} t| } | t#| d| d | d| d| d| d| d| d| d| d| d| d| d| dt%| dot'| |d|d|d| |dSu DEPRECATED: Use find_similar() instead. The find_similar() method now returns text contents by default. Migration: - find_similar_and_contents(url) → find_similar(url) - find_similar_and_contents(url, text=True) → find_similar(url, contents={"text": True}) - find_similar_and_contents(url, summary=True) → find_similar(url, contents={"summary": True}) rNrrrr"r=rr.r_r0rrrr4rJrKrLrMrNrrrrbrr1r2r3rr4rrr)rJr7rrar>r?rrDr?rrrEr#rrnrTrr`rr rPrrArrFrGrBrCr>rrr\r:s r7rezExa.find_similar_and_contents?s#,LLNN  DAq}  Yg%=%=/1GHGFO89994555=>>>888   Jwy/A4$H$H "9-L<'')>|H?U)V)V X&         H:>>>||NG44)$((=*A*ABB 9o  F(00L NN$((//#''--&**733&**733#/#3#34D#E#E'++H55&**733(,,Y77)--j99'++H55%))&11(,,Y77+// ==%1%5%56H%I%I,VZZ -C-CDD    &  HH) * * HHZ HHY''%..     r9clientrc|jjjt d d!fd }t d||jj_|S)"aWrap an OpenAI client with Exa functionality. After wrapping, any call to `client.chat.completions.create` will be intercepted and enhanced with Exa RAG functionality. To disable Exa for a specific call, set `use_exa="none"` in the `create` method. Args: client (OpenAI): The OpenAI client to wrap. Returns: OpenAI: The wrapped OpenAI client. rNmessages$Iterable[ChatCompletionMessageParam]modelUnion[str, ChatModel]use_exa-Optional[Literal['required', 'none', 'auto']]rr:rrarrr<rrrrrrr&rr'result_max_lenrrc ||||||| | | | | |d }d|i}||t||||S)N) rrrrrrrrrrrrrv) create_fnrtmax_len create_kwargs exa_kwargs)r_create_with_toolr)rtrvrxrrrrrrrrrrrrzr openai_kwargsrr~funcrPs r7create_with_ragz!Exa.wrap..create_with_rags0 +#2#2$4"0(<&8 , ,$  J%e,M   / / /))h&+% * r9z.Wrapping OpenAI client with Exa functionality.)rrrNNNNNNNNNNrsN) rtrurvrwrxryrr:rrarrarr<rr<rr<rr<rrarrarr&rr'rzrrra)chat completionscreaterprint)rPrprrs` @r7wrapzExa.wraps{&- t FL)*3737.2,02604040459+/"&)-%. . . . . . .  . ` >???)8 & r9r|rrt List[ChatCompletionMessageParam]rc*ddddddddidgd d d g}||d <|dd |i|}t|}|stj|dS|||d|d|d|d|d|d|d|d|d|d|d|d } t | |} t ||| } |dd | i|}tj|| } | S)Nfunctionrz(Search the web for relevant information.objectrstringzThe query to search for.)rr)rrr)rsr parameters)rrtoolsrt) completion exa_resultrrrrrrrrrrrr) rrrrrrrrrrrrr)r}rS)rrfrom_completionrDrnrr) rPr|rtr}r~rrrrrexa_str new_messagesexa_completions r7rzExa._create_with_tools#$#M (#(0/I&&' &-I # #     ("' gYBBBMBB  ++ &6%$  --"}55&NN+<==&NN+<=='^^,>??%>>*:;;!+0F!G!G)~~.BCC#77#77''^^J//..)).  $J@@@.z8WMM YFF F FF ,<!j   r9Frrrrvrrrrv#Optional[Literal['exa', 'exa-pro']]Optional[JSONSchemaInput]+Union[AnswerResponse, StreamAnswerResponse]cdSr3rS)rPrrrrrvrrs r7rz Exa.answer s 7:cr9c |rtddtD}d|vr |dt|d|d<t |dg}|d|} g} | dD]} t | } | t| d| d| d | d | d | d  t| d} t| d| | S)aGenerate an answer to a query using Exa's search and LLM capabilities. Args: query (str): The query to answer. text (bool, optional): Whether to include full text in the results. Defaults to False. system_prompt (str, optional): A system prompt to guide the LLM's behavior when generating the answer. model (str, optional): The model to use for answering. Defaults to None. output_schema (dict[str, Any], optional): JSON schema describing the desired answer structure. Returns: AnswerResponse: An object containing the answer and citations. Raises: ValueError: If stream=True is provided. Use stream_answer() instead for streaming responses. Examples: response = exa.answer("What is the capital of France?") print(response.answer) # e.g. "Paris" print(response.citations) # list of citations used # If you want the full text of the citations in the response: response_with_text = exa.answer( "What is the capital of France?", text=True ) print(response_with_text.citations[0].text) # Full page text Zstream=True is not supported in `answer()`. Please use `stream_answer(...)` for streaming.c*i|]\}}|dk | ||SrrSrUs r7rHzExa.answer..P r-r9rN/answerrrrrr4rKrLrrr0r) rr6rJrrEr#rTrrrnrrrPrrrrrvrrrresponserr\r:rs r7rz Exa.answer# sN  A  WVFHHNN$4$4VVV g % %'/*B*N'. r-r9rNr.Trr)r6rJrrEr#r rPrrrrvrrrrs r7 stream_answerzExa.stream_answeri sBWVFHHNN$4$4VVV g % %'/*B*N'r r r*r r rr r+r(rr*rr%rr:rrarrarr<rr<rr<rr<rrarrarr&rr'rrarr(rr<rrarr<rr)r+r*rr*)NNNNNNNN)rrErr:rrFrr:rr(rr:rrGrrHrrar+rI)rrErr/rr:rrFrr:rr(rr:rrGrrHrrar+rI)rrErr0rr:rrFrr:rr(rr:rrGrrHrrar+rM)rrErr/rr0rr:rrFrr:rr(rr:rrGrrHrrar+rOrrErr*rr%rr:rrarrarr<rr<rr<rr<rrarrarr(rr'rrar+r*)*rr*rr:rrarrarr<rr<rr<rr<rrarrarr(rr'rrarr:rrFrr:rr(rr:rrGrrHr+rI),rr*rr/rr:rrarrarr<rr<rr<rr<rrarrarr(rr'rrarr:rrFrr:rr(rr:rrGrrHr+rI),rr*rr0rr:rrarrarr<rr<rr<rr<rrarrarr(rr'rrarr:rrFrr:rr(rr:rrGrrHr+rM).rr*rr/rr0rr:rrarrarr<rr<rr<rr<rrarrarr(rr'rrarr:rrFrr:rr(rr:rrGrrHr+rOrr*)rpr)r|rrtrr+rrr*rr(rr(rr<rvrrrrr<r+r)rr*rrrr<rvrrrrr<r+r)rrrrrQr#rrDrrKrcrerrrrrSr9r7rr^sA00 -$( $3$3$3$3$3R6:+/,0 QQQQQnFJ%)/3/3*.(,.2,0,0,015'+%)%)'+26'+48+      BT T T T l,015'+/3"&:>*.%) - - - -X - ,015'+/3"&:>*.%) - - - - -X - ,015'+/3"&:>*.%) 0 0 0 0 0X 0,015'+/3"&:>*.%)77777X7 e e e e VFJ%)/3/3*.(,.2,0,0,004'+%)!_ _ _ _ _ _ B &*/3/3*.(,.2,0,0,004'+%)+/15'+/3"&:>*.------X-2 &*/3/3*.(,.2,0,0,004'+%)+/15'+/3"&:>*./-----X-4 &*/3/3*.(,.2,0,0,004'+%)+/15'+/3"&:>*./-----X-4 &*/3/3*.(,.2,0,0,004'+%)+/15'+/3"&:>*./00000X04&*/3/3*.(,.2,0,0,004'+%)+/15'+/3"&:>*.177777X76O O O O bDDDDL>>>>@ "'$'+5937'+ : : : : :X : "'$'+5937'+DKDKDKDKDKDKT'+5937'+*2*2*2*2*2*2*2*2r9rceZdZdDdEfd ZedFdZ dGdHdZd d d d d d d d d d d d d d d d d d ddId-ZdJd.ZdKd1Z d d d d d d d d d d d d d d2 dLd5Z dMd6Z d7d7d d d d d8dNd?Z d7d d d d d@dOdCZ xZS)PAsyncExarrr*api_basect||t||_t ||_t ||_d|_dSr3) rdrQr&rrrr$r_client)rPrrres r7rQzAsyncExa.__init__ sU (++++D11 )$// 1$77  r9r+httpx.AsyncClientcj|j&tj|j|jd|_|jS)NiX)rrtimeout)rhttpx AsyncClientrrrs r7rpzAsyncExa.client s: <  , cDL|r9Nrrr rr cK|r*t|tr|dp|o|ddk}i|j}|r|||dkrs|rDt jd|j|z||}|j |dd{V} | S|j |j|z||d{V} n#|dkrr|rDt jd|j|z|| }|j |dd{V} | S|j |j|z|| d{V} n|d kr,|j |j|z|| d{V} nU|d kr+|j |j|z| d{V} ntd || jdkr*| jdkrtd| jd| j| S)aSend a request to the Exa API, optionally streaming if data['stream'] is True. Args: endpoint (str): The API endpoint (path). data (dict, optional): The JSON payload to send. method (str, optional): The HTTP method to use. Defaults to "POST". params (dict, optional): Query parameters. headers (Dict[str, str], optional): Additional headers to include in the request. Defaults to None. Returns: Union[dict, httpx.Response]: If streaming, returns the Response object. Otherwise, returns the JSON-decoded response as a dict. Raises: ValueError: If the request fails (non-200 status code). rrr)r rT)rNr)rrrrrrrrr)rDr?rnrrrrRequestrrpsendrrrrrrr) rPrr>r r rr r!r#r"s r7 async_requestzAsyncExa.async_request s2 QJtT$:$:Qtxx?Q?Q  5vzz(++v5  +T\*  ,  " "7 + + + <<>>U " " -MH,!+  !K,,WT,BBBBBBBB  KOOMH,V_,\\^^v % % -DMH44!K,,WT,BBBBBBBB  K,,MH,4-\\^^w & & )) (t_*CC\\^^x ' ' ** (/+CCAAABB B ?c ! !co&<&<P3?PPchPP xxzzr9r$rrr%rr:rrarrr<rrrrrrr&rr'rrr(rrrrr)r*cKdtD}|dur|ddn|d|vrddtii|d<n|||d<t |t t |dg}|d |d{V}t| d }g}|d D]m}t|}| t| d | d | d| d| d| d| d| d| d| d| d| d| d| dt| dot|| d| d| dt| d|| d S)!aDPerform a search with a prompt-engineered query to retrieve relevant results. By default, returns text contents with 10,000 max characters. Use contents=False to opt-out. Args: query (str): The query string. contents (ContentsOptions | False, optional): Options for retrieving page contents. Defaults to {"text": {"maxCharacters": 10000}}. Use False to disable contents. See ContentsOptions for available options (text, highlights, summary, etc.). Note: The ``context`` option is deprecated; use ``highlights`` or ``text`` instead. num_results (int, optional): Number of search results to return. Default 10. include_domains (List[str], optional): Domains to include in the search. exclude_domains (List[str], optional): Domains to exclude from the search. start_crawl_date (str, optional): Only links crawled after this date. end_crawl_date (str, optional): Only links crawled before this date. start_published_date (str, optional): Only links published after this date. end_published_date (str, optional): Only links published before this date. include_text (List[str], optional): Strings that must appear in the page text. exclude_text (List[str], optional): Strings that must not appear in the page text. type (SearchType, optional): Search type - 'auto' (default), 'fast', 'deep', 'deep-reasoning', 'neural', or 'instant'. category (Category, optional): Data category to focus on (e.g. 'company', 'news', 'research paper'). flags (List[str], optional): Experimental flags for Exa usage. moderation (bool, optional): If True, the search results will be moderated for safety. user_location (str, optional): Two-letter ISO country code of the user (e.g. US). additional_queries (List[str], optional): Alternative query formulations for deep search to skip automatic LLM-based query expansion. Max 5 queries. Only applicable when type is 'deep' or 'deep-reasoning'. Example: ["machine learning", "ML algorithms", "neural networks"] system_prompt (str, optional): Deep-search-only instructions that guide both the search process and the final returned result. Use this to prefer certain sources, emphasize novelty, avoid duplicates, or constrain the response style. Only applicable when type is 'deep' or 'deep-reasoning'. output_schema (DeepOutputSchema, optional): Deep output schema for deep search. Use ``{"type": "text", "description": ...}`` for plain text output or ``{"type": "object", "properties": ..., "required": ...}`` for structured JSON. For object schemas, max nesting depth is 2 and max total properties is 10. Only applicable when type is 'deep' or 'deep-reasoning'. Returns: SearchResponse: The response containing search results, etc. Examples: Basic async search: >>> async_exa = AsyncExa(api_key="your-api-key") >>> results = await async_exa.search("latest AI research papers") >>> print(results.results[0].title) Async search with filters: >>> results = await async_exa.search( ... "climate change research", ... include_domains=["nature.com", "science.org"], ... start_published_date="2024-01-01" ... ) c*i|]\}}|dk | ||SrrSrUs r7rHz#AsyncExa.search..B r-r9FrNrrrr.r/r0rrrrr4rJrKrLrMrNrrrrrbrr1r2r3rrr4r5)r6rJrr7rr8rErrrnrTrr`rrrr9s r7rzAsyncExa.search sZWVFHHNN$4$4VVV u   KK D ) ) ) )  *G";";#),<>T+U"VGJ    !"*GJ )=>>>O3DEEE'' 7;;;;;;;;)$((=*A*ABB 9o  F(00L NN$((//#''--&**733&**733#/#3#34D#E#E'++H55&**733(,,Y77)--j99'++H55%))&11(,,Y77+// ==%1%5%56H%I%I,VZZ -C-CDD    &  HH) * * HHZ HHY''+DHHX,>,>??%..    r9c@Kd|i}|D] \}}||||< d|vrd|vrd|vr dti|d<i}|t|t|t t ||d|vr?t|dtr$|d}d|vrt|d|d<t|gdd }t|dg }| d |d{V}t|d } g} |d D]m} t| } | t#| d| d| d| d| d| d| d| d| d| d| d| d| d| dt%| dot'| |d|d|d| |dS) r<rNrrrrr"r=rr.r/r0rrrrr4rJrKrLrMrNrrrbrr1r2r3rr4rn)rJr7rr8r>r?rrDr?rrrErrrnrTrr`rrr@s r7rDzAsyncExa.search_and_contentss sBE"LLNN  DAq}  ' ! !((''/1GHGFO23334555=>>>888   Jwy/A4$H$H "9-L<'')>|H?U)V)V X&         H:>>>'' 7;;;;;;;;)$((=*A*ABB 9o  F(00L NN$((//#''--&**733&**733#/#3#34D#E#E'++H55&**733(,,Y77)--j99'++H55%))&11(,,Y77+// ==%1%5%56H%I%I,VZZ -C-CDD    &  HH) * * HHZ HHY''%..     r9rrEcKt|tr|g}nOt|tr:t|dkr't|dtr d|D}d|i}|D]\}}|dkr||||<d|vrd|vrd|vr d t i|d<i}|t|tt||d|vr?t|dtr$|d}d |vrt|d |d <t|d g}|d |d{V}t|d } g} |d gD]`} | t%| d| d| dag} |dD]m} t'| }| t)|d|d|d|d|d|d|d|d|d|d|d|d|d|dt+| dot-| |d|d |d!| | |d"#S)$asRetrieve contents for a list of URLs asynchronously. Args: urls (str | List[str] | List[Result]): A single URL, list of URLs, or list of Result objects. text (TextContentsOptions | True, optional): Options for text extraction. summary (SummaryContentsOptions | True, optional): Options for summary generation. max_age_hours (int, optional): Maximum age of cached content in hours. If content is older, it will be fetched fresh. Special values: 0 = always fetch fresh content, -1 = never fetch fresh (cache only). Example: 168 = fetch fresh for pages older than 7 days. filter_empty_results (bool, optional): Whether to filter out empty results. subpages (int, optional): Number of subpages to retrieve. subpage_target (str | List[str], optional): Target subpages to retrieve. extras (ExtrasOptions, optional): Options for extra content (links, image_links). flags (List[str], optional): Experimental flags. Returns: SearchResponse[Result]: The response containing the contents of the URLs. Examples: Get contents for URLs asynchronously: >>> async_exa = AsyncExa(api_key="your-api-key") >>> results = await async_exa.get_contents("https://example.com/article") >>> print(results.results[0].text) Get contents from async search results: >>> search_results = await async_exa.search("AI news", contents=False) >>> contents = await async_exa.get_contents(search_results.results[:5]) rcg|] }|j SrSrSrTs r7rz)AsyncExa.get_contents.. rVr9rrPNrrrrr"rWr0rrrrrrXrrr4rJrKrLrMrNrrrbrr1r2r3rr4rY)rDr*rrZrHrJr7rr>r?rr?rrErrrnrrrTr`rrr[s r7rKzAsyncExa.get_contents s< dC  )6DD d # # )D A *T!Wg:V:V ((4(((D4.LLNN  DAqF{{q}  ' ! !((''/1GHGFO4555=>>>888   Jwy/A4$H$H "9-L<'')>|H?U)V)V X&(44'' W========)$((=*A*ABB hhz2..  F OOzz$''!::h//!::h//    9o  F(00L NN$((//#''--&**733&**733#/#3#34D#E#E'++H55&**733(,,Y77)--j99'++H55%))&11(,,Y77+// ==%1%5%56H%I%I,VZZ -C-CDD    &  HH) * * HHZ HHY''%..    r9r\rrc KdtD}|dur|ddn|d|vrddtii|d<n|||d<t |t t |}|d|d{V}t| d}g}|d D]m}t|}| t| d | d | d | d | d| d| d| d| d| d| d| d| d| dt| dot|| d| d|| dS)aFinds similar pages to a given URL, potentially with domain filters and date filters. By default, returns text contents with 10,000 max characters. Use contents=False to opt-out. Args: url (str): The URL to find similar pages for. contents (ContentsOptions | False, optional): Options for retrieving page contents. Defaults to {"text": {"maxCharacters": 10000}}. Use False to disable contents. See ContentsOptions for available options (text, highlights, summary, etc.). num_results (int, optional): Number of results to return. Default is None (server default). include_domains (List[str], optional): Domains to include in the search. exclude_domains (List[str], optional): Domains to exclude from the search. start_crawl_date (str, optional): Only links crawled after this date. end_crawl_date (str, optional): Only links crawled before this date. start_published_date (str, optional): Only links published after this date. end_published_date (str, optional): Only links published before this date. include_text (List[str], optional): Strings that must appear in the page text. exclude_text (List[str], optional): Strings that must not appear in the page text. exclude_source_domain (bool, optional): Whether to exclude the source domain. category (Category, optional): Data category to focus on (e.g. 'company', 'news', 'research paper'). flags (List[str], optional): Experimental flags. Returns: SearchResponse[Result] Examples: Find similar pages asynchronously: >>> async_exa = AsyncExa(api_key="your-api-key") >>> results = await async_exa.find_similar("https://www.nature.com/articles/s41586-021-03819-2") >>> print(results.results[0].title) Find similar with domain filters: >>> results = await async_exa.find_similar( ... "https://arxiv.org/abs/2301.00001", ... include_domains=["arxiv.org", "openreview.net"], ... exclude_source_domain=True ... ) c*i|]\}}|dk | ||SrrSrUs r7rHz)AsyncExa.find_similar..g r-r9FrNrrr_r0rrrrr4rJrKrLrMrNrrrrrbrr1r2r3r4r`)r6rJrr7rrarErrrnrTrr`rrrbs r7rczAsyncExa.find_similar/ spWVFHHNN$4$4VVV u   KK D ) ) ) )  *G";";#),<>T+U"VGJ    !"*GJ )CDDD((''@@@@@@@@)$((=*A*ABB 9o  F(00L NN$((//#''--&**733&**733#/#3#34D#E#E'++H55&**733(,,Y77)--j99'++H55%))&11(,,Y77+// ==%1%5%56H%I%I,VZZ -C-CDD    &  HH) * * HHZ %..     r9c8Kd|i}|D] \}}||||< d|vrd|vr dti|d<i}|t|t|t t ||d|vr?t|dtr$|d}d|vrt|d|d<t|gdd}t|dg }| d |d{V}t|d } g} |d D]m} t| } | t#| d| d | d| d| d| d| d| d| d| d| d| d| d| dt%| dot'| |d|d|d| |dSrm)rJr7rrar>r?rrDr?rrrErrrnrTrr`rrros r7rez"AsyncExa.find_similar_and_contents s3#,LLNN  DAq}  Yg%=%=/1GHGFO89994555=>>>888   Jwy/A4$H$H "9-L<'')>|H?U)V)V X&         H:>>>''@@@@@@@@)$((=*A*ABB 9o  F(00L NN$((//#''--&**733&**733#/#3#34D#E#E'++H55&**733(,,Y77)--j99'++H55%))&11(,,Y77+// ==%1%5%56H%I%I,VZZ -C-CDD    &  HH) * * HHZ HHY''%..     r9Frrrrvrrrc K|rtddtD}d|vr |dt|d|d<t |dg}|d|d{V} g} | dD]} t | } | t| d| d | d | d | d | d t| d} t| d| | S)aGenerate an answer to a query using Exa's search and LLM capabilities. Args: query (str): The query to answer. text (bool, optional): Whether to include full text in the results. Defaults to False. system_prompt (str, optional): A system prompt to guide the LLM's behavior when generating the answer. model (str, optional): The model to use for answering. Defaults to None. output_schema (dict[str, Any], optional): JSON schema describing the desired answer structure. Returns: AnswerResponse: An object containing the answer and citations. Raises: ValueError: If stream=True is provided. Use stream_answer() instead for streaming responses. Examples: Basic async question answering: >>> async_exa = AsyncExa(api_key="your-api-key") >>> response = await async_exa.answer("What are the latest developments in quantum computing?") >>> print(response.answer) Async answer with citations: >>> response = await async_exa.answer("Explain renewable energy benefits", text=True) >>> for citation in response.citations: ... print(f"{citation.title}: {citation.url}") rc*i|]\}}|dk | ||SrrSrUs r7rHz#AsyncExa.answer.. r-r9rNr.rrrrrr4rKrLrrr0r) rr6rJrrErrTrrrnrrrs r7rzAsyncExa.answer sJ  A  WVFHHNN$4$4VVV g % %'/*B*N'>> async_exa = AsyncExa(api_key="your-api-key") >>> async for chunk in async_exa.stream_answer("What is quantum computing?"): ... if chunk.text: ... print(chunk.text, end="", flush=True) Async stream with citations: >>> async for chunk in async_exa.stream_answer("Explain climate change", text=True): ... if chunk.text: ... print(chunk.text, end="") ... if chunk.citations: ... print(f"\nCitations: {[c.url for c in chunk.citations]}") c*i|]\}}|dk | ||SrrSrUs r7rHz*AsyncExa.stream_answer..P r-r9rNr.Trr)r6rJrrErrrs r7rzAsyncExa.stream_answer* sLWVFHHNN$4$4VVV g % %'/*B*N'r?r@rAr+r?)rKr*r+r*)r>r?r+r?)rVrWr+rX)rrrr?r+r)rr?r+r)rrr+r)rrrrrr*)} __future__rrsrrrMr functoolsrtypingrrrr r r r r rrrrrrrrrropenair/openai.types.chat.chat_completion_message_paramropenai.types.chat_modelrtyping_extensionsr exa_py.utilsrrrrrrr r!websets.core.baser"rr#r$rr%r&getenvis_betar7r=rErPrTrCategory SearchTyperrDeepOutputSchemar*rrrr?r8rar1r>r?rrrr SECTION_TAGrrr!r$r(r*r.r3r7rr^r_rar`rbr\rcrdrer]rZr[EntityrHr`rlrvr{rrrrrrrrrrrrrrrrrSr9r7rs/"""""" !!!!!!"(''''' VVVVVV------''''''#"""""------EEEEEEEE99999999 ")I  & ( FFFF&68* D D D D$w*w*w*w*v  q l9EYe-/EEF# cU#C5# # v ##se# # #cU#  !#& '#,   -#4 5#6 dV7#84&9#:t ;#< =#BUVE###L 55vve E%FF"V Vt $NOF 4Ld|,d|t #$U!FV  DkfV ##0PPPP&9: \ 3((((()5((((,%6     Ye                   Ye    IUiu> )5 ))))))) )           """"""" " !!!!!!! ! OOOOOOO O ))))))) )   D D D D D D D  D """"""" "   ??????? ? PPPPPPP P ((((((( ( ''''''' ' }l* + JJJJJJJ JZ ;;;;;W;; ;| *1*1*1*1*1W*1*1 *1Z *7*7*7*7*7*7*7 *7Z .P.P.P.P.Pw.P.P .Pb % % % % % % %  % P  6  47#7#7#7#7#7#7#7#t:#:#:#:#:#:#:#:#z GCLL   )))))WQZ)) )X  '..E&F#4 4444444 4 /////// /=B=B=B=B@"u2u2u2u2u2u2u2u2p!C 7C 7C 7C 7C 7sC 7C 7C 7C 7C 7r9