_j %SrSSKJr SSKrSSKJrJrJrJrJ r J r J r J r J r Jr SSKJrJr SSKJr SSKJrJrJrJr SS KJrJrJr \ "S \S 9r"S S \\5r"SS\5rg)z!Asynchronous Research API client.) annotationsN) AnyAsyncGeneratorDictGenericLiteralOptionalTypeTypeVarUnionoverload) BaseModel TypeAdapter)AsyncResearchBaseClient) ResearchDto ResearchEventListResearchResponseDto ResearchModel)async_stream_sse_eventsis_pydantic_modelpydantic_to_json_schemaT)boundc"\rSrSrSrSSjrSrg)AsyncResearchTyped%z6Wrapper for typed research responses in async context.cXlX lURUlURUlURUlUR UlUR UlURUl[US5(aURUl [US5(aURUl [US5(aURUl [US5(aURUl gg)Neventsoutput cost_dollarserror) research parsed_output research_idstatus created_atmodel instructions output_schemahasattrrr r!r")selfr#r$s Q/home/rurouni/.local/lib/python3.13/site-packages/exa_py/research/async_client.py__init__AsyncResearchTyped.__init__(s *#//oo "--^^ $11%33 8X & &"//DK 8X & &"//DK 8^ , , ( 5 5D  8W % %!DJ &) r!r'r"rr)r(r r*r$r#r%r&N)r#rr$r)__name__ __module__ __qualname____firstlineno____doc__r.__static_attributes__r0r-rr%s @(r0rc2\rSrSrSr\SS.SSjj5r\SS.S Sjj5r\SS.S!Sjj5rSS S .S"S jjr\S#S j5r\S S S.S$Sjj5r\S S.S%Sjj5r\S S S.S&Sjj5rS S S S.S'SjjrS S S.S(Sjjr\SSS S.S)Sjj5r \SSS S.S*Sjj5r SSS S S.S+Sjjr Sr g ),AsyncResearchClient<z)Asynchronous client for the Research API.zexa-research-fast)r(c # g7fNr7)r,r)r(s r-createAsyncResearchClient.create?s z exa-researchc # g7fr<r7r,r)r(r*s r-r=r>G r?c # g7fr<r7rAs r-r=r>PrBr?N)r(r*c# UUS.nUb#[U5(a[U5US'OX4S'URSSUS9IShvN n[[5nUR U5$N$7f)zCreate a new research request. Args: instructions: The research instructions. model: The model to use for research. output_schema: Optional JSON schema or Pydantic model for structured output. Returns: The created research object. )r)r(N outputSchemaPOST)methoddata)rrrequestrrvalidate_python)r,r)r(r*payloadresponseadapters r-r=r>Ysv$)   $ //*A-*P'*7'bgFFk*&&x00GsAA)A'%A)c # g7fr<r7)r,r%s r-getAsyncResearchClient.getys r?F)streamrc # g7fr<r7r,r%rRrs r-rPrQrBr?rc # g7fr<r7rTs r-rPrQs /2r?c # g7fr<r7)r,r%rRrr*s r-rPrQs !$r?)rRrr*c# 0nU(dSUS'U(aSUS'O$SUS'Ub[U5R5US'U(a(URSU3SUSS 9IShvN n[U5$URSU3SUS 9IShvN n[ [ 5nUR U5nU(a[US 5(aUR(aURR(a&URURR5n O:S SK n U RURR5n URU 5n [X5$U$NN![a Us$f=f7f) a?Get a research request by ID. Args: research_id: The research ID. stream: Whether to stream events. events: Whether to include events in non-streaming response. output_schema: Optional Pydantic model for typed output validation. Returns: Research object, typed research, or async event generator. falserRtruerN/GETT)rHparamsrRrHr]r r)strlowerrJrrrrKr+r parsedmodel_validatejsonloadscontentr Exception) r,r%rRrr*r]rMrNr#rarc parsed_datas r-rPrQse&&F8 #)x %F8 !#&v;#4#4#6x !\\K=!%t*H+84 4!\\K=!%*H"+.G..x8H8!T>T!U#&*jj1H1H&I !.!=!=k!J-h?? O3  !$#O$sJAE5!E"&E5E! A E5BE#E5!E5# E2/E51E22E5)cursorlimitc# URX5nURSSUS9IShvN n[R"U5$N7f)zList research requests. Args: cursor: Pagination cursor. limit: Maximum number of results. Returns: List of research objects with pagination info. rFr\r^N)build_pagination_paramsrJrrb)r,rhrir]rMs r-listAsyncResearchClient.listsE--f<bvFF&55h??Gs&AAAi' ) poll_interval timeout_msrc # g7fr<r7)r,r%rorprs r-poll_until_finished'AsyncResearchClient.poll_until_finisheds r?c # g7fr<r7)r,r%rorprr*s r-rrrss !$r?)rorprr*c `# US- nUS- nSn[R"5R5n Sn U(aURXUS9IShvN n OURXS9IShvN n Sn [ U S5(a U R OSn U S;aU $[R"5R5U - U:a[S USUS35e[R"U5IShvN MNN![ a(n U S - n X:a[S US US U 35eSn A NSn A ff=fN?7f)aPoll until research is finished. Args: research_id: The research ID. poll_interval: Milliseconds between polls (default 1000). timeout_ms: Maximum time to wait in milliseconds (default 600000). events: Whether to include events in the response. output_schema: Optional Pydantic model for typed output validation. Returns: Completed research object or typed research. Raises: TimeoutError: If research doesn't complete within timeout. RuntimeError: If polling fails too many times. rnr)rr*NrUr&) completedfailedcanceledrzPolling failed z times in a row for research z: z Research z did not complete within ms) asyncioget_event_looptimerPr+r&rf RuntimeError TimeoutErrorsleep)r,r%rorprr*poll_interval_sec timeout_secmax_consecutive_failures start_timeconsecutive_failuresresultr&es r-rrrssp2*D0 4' #$ ++-224    #'88#-$,$F$(88K8#GGF'($+2&(*C*C@@!MA%%',,.;kI" },Ej\QST-- 12 2 2;H $)$'C&)*B)CD((3}Bqc;D  3s_3D.C7C3C7(C5),C7AD.,D,-D.3C75C77 D)D$D.$D))D.r7)r)r_r(rreturnr)r)r_r(+Literal['exa-research', 'exa-research-pro']r*zDict[str, Any]rr)r)r_r(rr*Type[T]rr)r)r_r(rr*z0Optional[Union[Dict[str, Any], Type[BaseModel]]]rr)r%r_rr)r%r_rRLiteral[False]rboolrr)r%r_rRz Literal[True]rzOptional[bool]rz#AsyncGenerator[ResearchEvent, None]) r%r_rRrrrr*rrAsyncResearchTyped[T]) r%r_rRrrrr*Optional[Type[BaseModel]]rzKUnion[ResearchDto, AsyncResearchTyped, AsyncGenerator[ResearchEvent, None]])rhz Optional[str]riz Optional[int]rr) r%r_rointrprrrrr) r%r_rorrprrrr*rrr) r%r_rorrprrrr*rrz&Union[ResearchDto, AsyncResearchTyped]) r1r2r3r4r5r r=rPrlrrr6r7r0r-r9r9<sT3 3     >L ;  &    >L ;     >LJN 11; 1 H 1  1@   "'       "& 22 2  2 - 22 "' $$ $  $  $ $$37 77 7  7 1 7 U7x!%# @@ @ ! @& "         "  $$ $  $  $$ $$" 37<3<3 <3  <3  <31<3 0<3<3r0r9) r5 __future__rr{typingrrrrrr r r r r pydanticrrbasermodelsrrrrutilsrrrrrr9r7r0r-rsf'"   ,)   Cy!((.|31|3r0