_jMSrSSKJr SSKrSSKJrJrJrJrJ r J r SSK r SSK r SSK JrJr SSKJrJrJrJrJrJrJrJrJr SSjrSS jrSS jrSS jrSS jrSS jrSSjr g)zUtilities for the Research API.) annotationsN)AnyAsyncGeneratorDict GeneratorOptionalType) BaseModelValidationError) ResearchEventResearchDefinitionEventResearchOutputEventResearchPlanDefinitionEventResearchPlanOperationEventResearchPlanOutputEventResearchTaskDefinitionEventResearchTaskOperationEventResearchTaskOutputEventc~[U[5=(a [U[5$![[ 4a gf=f)zCheck if the given schema is a Pydantic model class. Args: schema: The schema to check. Returns: True if schema is a Pydantic model class, False otherwise. F) isinstancetype issubclassr TypeErrorAttributeError)schemas J/home/rurouni/.local/lib/python3.13/site-packages/exa_py/research/utils.pyis_pydantic_modelr s7&$'IJvy,II ~ &s &)<<cSSKJn U"U5$)zConvert a Pydantic model to JSON Schema. Args: model: The Pydantic model class. Returns: JSON Schema dictionary with all references inlined. r)_convert_schema_input) exa_py.utilsr )modelr s rpydantic_to_json_schemar#/s3 ! ''cU(aUR5(dgSU;agURS5upnUR5UR54$)zParse a single SSE line. Args: line: The SSE line to parse. Returns: Tuple of (field, value) or None if not a valid SSE line. N:)strip partition)linefield_values rparse_sse_liner-?sG tzz|| $nnS)OEe ;;=%++- ''r$c,SnSnUHDn[U5nU(dMUupVUS:XaUnM%US:XdM-[R"U5nMF U(a"U(a[ U[ 5(aXS'U$g![Ra UnMf=f)zParse SSE event lines into a raw event dictionary. Args: event_lines: List of lines that make up an SSE event. Returns: Parsed event data or None if invalid. Neventdata eventType)r-jsonloadsJSONDecodeErrorrdict) event_lines event_name event_datar)parsedr*r,s rparse_sse_event_rawr:RsJJ%   G J f_ #!ZZ. j j$ ' '&0{ # '' #"  #sA99BBc URS5nU(dg[[[[[ [ [[S.nURU5nU(dgURU5$![a gf=f)zParse a raw event dictionary into a typed ResearchEvent. Args: raw_event: Raw event dictionary with eventType field. Returns: Typed ResearchEvent or None if parsing fails. r1N)zresearch-definitionzresearch-outputzplan-definitionzplan-operationz plan-outputztask-definitionztask-operationz task-output) getrrrrrrrrmodel_validater ) raw_event event_type event_models model_classs rparse_research_eventrBvsy{+J  7.64.64. L"":.K )))44 sA00 A=<A=c## /nUR5HxnU(d5U(a,[U5nU(a[U5nU(aUv /nM=M?[U[5(aUR S5OUnUR U5 Mz U(a+[U5nU(a[U5nU(aUv gggg7f)zStream SSE events from a requests Response. Args: response: The streaming response object. Yields: Parsed ResearchEvent objects. zutf-8N) iter_linesr:rBrbytesdecodeappend)responser6r)r>r/ decoded_lines rstream_sse_eventsrJsK##%/ < 0;E#  4>dE3J3J4;;w/PTL   | ,&' 4 (3E  sCCcN# /nUR5ShvN nU(d6U(a-[U5nU(a[U5nU(aU7v /nMDMFURU5 MYNT U(a,[U5nU(a[U5nU(aU7v gggg7f)zStream SSE events from an httpx Response. Args: response: The async streaming response object. Yields: Parsed ResearchEvent objects. N) aiter_linesr:rBrG)rHr6r)r>r/s rasync_stream_sse_eventsrMsK**, %d/ < 0;E#     t $ %,' 4 (3E  s&B%A0A.A0AB%.A005B%)rrreturnbool)r"zType[BaseModel]rNDict[str, Any])r)strrNzOptional[tuple[str, str]])r6z list[str]rNzOptional[Dict[str, Any]])r>rPrNzOptional[ResearchEvent])rHzrequests.ResponserNz$Generator[ResearchEvent, None, None])rHzhttpx.ResponserNz#AsyncGenerator[ResearchEvent, None])!__doc__ __future__rr2typingrrrrrr httpxrequestspydanticr r modelsr rrrrrrrrrr#r-r:rBrJrMr$rrZsx%"  /     ( (&!H!H!!)!H  ( r$