"jSrSSKJr SSKJrJr SSKJrJ r J r SSK J r J r Jr SSKJr SSKJrJrJrJrJrJrJrJrJrJrJrJrJr SS KJ r! \ (aSS K"J#r# \r$\\-r%\"S \%S 9r&\"S S55r'g)aThis module provides functionality to manage and update parts of a model's streamed response. The manager tracks which parts (in particular, text and tool calls) correspond to which vendor-specific identifiers (e.g., `index`, `tool_call_id`, etc., as appropriate for a given model), and produces Pydantic AI-format events as appropriate for consumers of the streaming APIs. The "vendor-specific identifiers" to use depend on the semantics of the responses of the responses from the vendor, and are tightly coupled to the specific model being used, and the Pydantic AI Model subclass implementation. This `ModelResponsePartsManager` is used in each of the subclasses of `StreamedResponse` as a way to consolidate event-emitting logic. ) annotations)HashableIterator) dataclassfieldreplace) TYPE_CHECKINGAnyTypeVar)UnexpectedModelBehavior) ModelResponsePartModelResponseStreamEventNativeToolCallPartPartDeltaEventPartStartEventProviderDetailsDeltaTextPart TextPartDelta ThinkingPartThinkingPartDelta ToolCallPartToolCallPartDelta ToolPartKind)generate_tool_call_id)ModelRequestParametersPartT)boundc\rSrSr%SrS\S'\"\\SS9r S\S'\"\ \ \ 4SS9r S \S '\"\ \ \\4SSSS 9rS \S '\"\ \\4SSS9rS\S'S4SjrS5SjrS6SjrS7SjrS8SjrS9SjrS:SjrSSSSSS.S;SjjrSSSSSS.SS jjrS?S!jrS@S"jrSASBS#jjrSCS$jr SDS%jr!SES&jr"SFS'jr#SGS(jr$SHS)jr%SHS*jr&SIS+jr'SJS,jr(SKS-jr)SLS.jr*SMS/jr+SNS0jr,SOS1jr-SPS2jr.S3r/g)QModelResponsePartsManager9zManages a sequence of parts that make up a model's streamed response. Parts are generally added and/or updated by providing deltas, which are tracked by vendor-specific IDs. rmodel_request_parametersF)default_factoryinitlist[ManagedPart]_partszdict[VendorId, int]_vendor_id_to_part_index)r#r$reprcomparezdict[int, list[str]]_string_buffers)r#r$r(zdict[str, ToolPartKind]_tool_kind_by_namecURRVs0sH(oRcMURUR_M* snUlgs snfN)r"function_tools tool_kindnamer+)selftds W/home/mande/repo/quber/.venv/lib/python3.13/site-packages/pydantic_ai/_parts_manager.py __post_init__'ModelResponsePartsManager.__post_init__RsF,0,I,I,X,X# ,Xb\h\h !BGGR\\ !,X# # s AAc[U5RSUR<SUR5<SUR<S3$)Nz(model_request_parameters=z , _parts=z, _vendor_id_to_part_index=))type __qualname__r"_materialized_partsr')r1s r3__repr__"ModelResponsePartsManager.__repr__WsVDz&&'(((,(E(E'HI..034((,(E(E'H K c8URURLa|[U[5(deURUR 5UR UR 4URUR 5UR UR 4:H$[$r-) __class__ isinstancer r"r:r'r+NotImplemented)r1others r3__eq__ ModelResponsePartsManager.__eq___s ??dnn ,e%>?? ??--((*--''  ..))+..((   r=c8URRU5$r-)r+get)r1 tool_names r3_tool_kind_for(ModelResponsePartsManager._tool_kind_foros&&**955r=cURbU$URUR5nUcU$[R"XS9$)zPromote a base `ToolCallPart` to a typed subclass via `ToolDefinition.tool_kind`. Safe no-op for unknown tool names (model hallucinations) and for tool defs without a `tool_kind`. )r/)r/rHrGr narrow_type)r1partkinds r3_typed_call_part*ModelResponsePartsManager._typed_call_partrsB >> %K""4>>2 <K''==r=c[UR5H8n[URU[5(aM'UR U5 M: URVs/sHn[U[5(aMUPM sn$s snf)zReturn only model response parts that are complete (i.e., not ToolCallPartDelta's). Returns: A list of ModelResponsePart objects. ToolCallPartDelta objects are excluded. )tupler*r@r&r_materialize_and_cache_part)r1 part_indexps r3 get_parts#ModelResponsePartsManager.get_partssf   4 45Jdkk*57HII00<6 ;;O;ajJJ% $ J 7  Z |%L%L"22::>J#EE%7K[  x883@3L0$($D$DZ$PM14`R_Qa2bcc Wa(88;;N[kl l l  ' /)c'la.?7??CTCTG-sD!..tDN ~A A.J * &%"99:L\!1J ((4=..d:=%66"<   HH"~IL))*6P6PRYZ&2DKK # zD D[ ms&BHG;A"H:G>;DH>Hrar[ signaturer\r]c## SnUcUR[5nOeURRU5nUbGURUn [ U [5(d UR U5n [SU <35eX4nUcaUcUcUbM[U5(aU"S5OUn [U=(d SUUUU S9n URX5n [XS9v g[S5eUupUc UcUcUcg[UUURX5US9nUSL=(d/ URSL=(d USL=(d U R0:HnU(a[U5(aURRU5nUb [!U5OSnU"U R5n [#USU S 9nUR%U 5nUcURR'US5 O%UUS2 UURU'OUR)XUS 9nU(aUR+UUR,U5 UURU'[/XS 9v g7f) aHandle incoming thinking content, creating or updating a ThinkingPart in the manager as appropriate. When `vendor_part_id` is None, the latest part is updated if it exists and is a ThinkingPart; otherwise, a new ThinkingPart is created. When a non-None ID is specified, the ThinkingPart corresponding to that vendor ID is either created or updated. Args: vendor_part_id: The ID the vendor uses to identify this piece of thinking. If None, a new part will be created unless the latest part is already a ThinkingPart. content: The thinking content to append to the appropriate ThinkingPart. id: An optional id for the thinking part. signature: An optional signature for the thinking content. provider_name: An optional provider name for the thinking part. provider_details: Either a dict of provider-specific details, or a callable that takes the existing part's `provider_details` and returns the updated details. Callables allow provider-specific update logic without the parts manager knowing the details. Yields: A `PartStartEvent` if a new part was created, or a `PartDeltaEvent` if an existing part was updated. Raises: UnexpectedModelBehavior: If attempting to apply a thinking delta to a part that is not a ThinkingPart. Nz/Cannot apply a thinking delta to existing_part=r}rbzLCannot create a ThinkingPart with no content, signature, or provider_details)resignature_deltar\r]r)rer]rfrh)rjrr'rFr&r@rRr callablerprrrqr\r]r*rnrapplypoprrrsrar)r1rtrar[r~r\r] existing_thinking_part_and_indexrSrvresolved_detailsrLrwexisting_thinking_partryrgbuffer buffer_lengthmetadata_deltarzs r3handle_thinking_delta/ModelResponsePartsManager.handle_thinking_deltasDMQ(  !/3/K/KL/Y ,66::>JJ% $ J 7 !->>$($D$DZ$PM14dVcUe2fgg3@3L0 + 3"i&;?O?[>FFV=W=W#3D#9]m ##Mr'"/%5 "&!2!24!H$>EE-b2R . "9#49NScSk*% )"99:P`!1 J%A++47A#4/A*::b@  (+;"<"<--11*=/5/AF q #34J4[4[#\ !(4Zj!k-334JK >((,,Z>}~.7=D((4#LL*~ M  ))*l6J6JGT&2DKK # zD DsH?I)rGargs tool_call_idr\r]c SnUc#UcUR[[[5nOsURR U5nUbUUR Un [U [[-[-5(d URU5n [SU <35eX4nUc[UUUUUS9n U R5=(d U n [U [5(aURU 5n URX5n [U [[-5(a [XS9$gUup[UUUURX5US9n URR U5n U b [!U 5OSnUR#XU 5n[U[5(aURU5nXR U'[U[[-5(a\[U [5(a [XS9$UR((a$U R((d[+XR(S9n [-XS9$g![$aB XR U'U cURR'US5 eXS2 XRU'ef=f)aHandle or update a tool call, creating or updating a `ToolCallPart`, `NativeToolCallPart`, or `ToolCallPartDelta`. Managed items remain as `ToolCallPartDelta`s until they have at least a tool_name, at which point they are upgraded to `ToolCallPart`s. If `vendor_part_id` is None, updates the latest matching ToolCallPart (or ToolCallPartDelta) if any. Otherwise, a new part (or delta) may be created. Args: vendor_part_id: The ID the vendor uses for this tool call. If None, the latest matching tool call may be updated. tool_name: The name of the tool. If None, the manager does not enforce a name match when `vendor_part_id` is None. args: Arguments for the tool call, either as a string, a dictionary of key-value pairs, or None. tool_call_id: An optional string representing an identifier for this tool call. provider_name: An optional provider name for the tool call part. provider_details: An optional dictionary of provider-specific details for the tool call part. Returns: - A `PartStartEvent` if a new ToolCallPart or NativeToolCallPart is created. - A `PartDeltaEvent` if an existing part is updated. - `None` if no new event is emitted (e.g., the part is still incomplete). Raises: UnexpectedModelBehavior: If attempting to apply a tool call delta to a part that is not a ToolCallPart, NativeToolCallPart, or ToolCallPartDelta. Nz0Cannot apply a tool call delta to existing_part=)tool_name_delta args_deltarr\r]rbr)rrh)rjrrrr'rFr&r@rRr as_partrNrprrqr*rn_apply_tool_call_delta Exceptionrrrr)r1rtrGrrr\r] existing_matching_part_and_indexrSrvrirLrwrrrzs r3handle_tool_call_delta0ModelResponsePartsManager.handle_tool_call_deltafsL  )  ! 373O3O "46G40 66::>JJ% $ J 7 !-1B\1QTf1fgg$($D$DZ$PM14eWdVf2ghh3@3L0 + 3% ))+!1 E==?+eD$ --,,T2!..tDN$ /A ABB%NFFC)I %M% ))"99-W!1 E))--j9F+1+=CK1M #:::V[\ lL99#'#8#8#FL'3KK #, 7I(IJJm->??) NN$009K9K '((,,Z>~.7=((4 s .8H--A I9)rr[r\r]c [UUU=(d [5UUUS9nURU5nUcURU5n OzURR U5n U bUR U OSn U b)[U [5(aU n URX5 OURU5n XRU'[XS9$)a{Immediately create or fully-overwrite a ToolCallPart with the given information. This does not apply a delta; it directly sets the tool call part contents. Args: vendor_part_id: The vendor's ID for this tool call part. If not None and an existing part is found, that part is overwritten. tool_name: The name of the tool being invoked. args: The arguments for the tool call, either as a string, a dictionary, or None. tool_call_id: An optional string identifier for this tool call. id: An optional identifier for this tool call part. provider_name: An optional provider name for the tool call part. provider_details: An optional dictionary of provider-specific details for the tool call part. Returns: ModelResponseStreamEvent: A `PartStartEvent` indicating that a new tool call part has been added to the manager, or replaced an existing part. )rGrrr[r\r]Nrb) r_generate_tool_call_idrNrpr'rFr&r@ _replace_partr) r1rtrGrrr[r\r]new_partrwmaybe_part_indexrvs r3handle_tool_call_part/ModelResponsePartsManager.handle_tool_call_parts: %A)?)A'-  ((2  !!..x8N $<<@@P =M=YDKK(89_cM+ =,0W0W!1"">8!%!2!24!8>r=cDUbURRUS5 gg)z>Stop tracking a vendor_part_id (no-op if None or not tracked).N)r'rr1rts r3_stop_tracking_vendor_id2ModelResponsePartsManager._stop_tracking_vendor_id)s#  %  ) ) - -nd C &r=c[UR5nURRU5 UbX0RU'U$)zAAppend a part, optionally track vendor_part_id, return new index.)rnr&appendr')r1rLrt new_indexs r3rp&ModelResponsePartsManager._append_part.s< $  4  %*>>;;}- -$   !4!48N8N$UtLl,=>> 88HI > * * 6"*>>;;}- -$  E$6$6%:M:MQVQgQg$UtL  * * 6"9F9L"8M MF))*5 er=cXURUnURRU5nUcU$SRU5n[ U[ [ -5(a [X$S9$[ U[5(a/[ UR[5(aS5e[X$S9$[ U[[-5(aW[ UR[5(aS5e[X$S9n[ U[5(aURU5nU$[SU<35e)zLCompute the materialized form of a part without mutating any buffered state.r)raz=Cannot materialize string arguments onto dictionary argumentsr)rz%Cannot materialize string deltas for )r&r*rFjoinr@rrrrrrrrrrNAssertionError)r1rSrLrvalue materializeds r3_materialized_part,ModelResponsePartsManager._materialized_parts{{:&%%))*5 >K dH|3 4 44/ / d- . .!$//488 O 842 2 dL+== > >!$))T22 s4s s2"44L, 55#44\B  DTHMNNr=c|URU5nXR;aURU X RU'U$)zPMaterialize buffered string deltas for one part, caching the result in `_parts`.)rr*r&r1rSrLs r3rR5ModelResponsePartsManager._materialize_and_cache_parts=&&z2 -- -$$Z0&*KK # r=cUR(d UR$[[UR55Vs/sHoR U5PM sn$s snf)zReturn the fully materialized parts without mutating buffered state. Used by `__eq__`/`__repr__` so that reading the manager never flushes its buffers as a side effect. )r*r&rangernr)r1rcs r3r:-ModelResponsePartsManager._materialized_partssH ##;;  tag - create new ThinkingPart.r)rar\r]rbN)rrrpr)r1rtr\r]rLrs r3rm9ModelResponsePartsManager._handle_embedded_thinking_starts< %%n5Bmg%%d; 988s9;c#J# [UURX5US9nURSL=(d$ URSL=(d UR0:HnUR XUS9nU(aUR X!R U5 XRU'[X&S9v g7f)z)Handle content inside ....rdNrfrh) rrqr\r]rrrsrar&r) r1rvrSrar\r]ryrgrzs r3rl;ModelResponsePartsManager._handle_embedded_thinking_contents'!55mS-  $ $D 0 4**$6 4--3  DD nE    % %j2G2G Q". J:@@sB!B#c&URU5 g)zCHandle tag - stop tracking so next delta creates new part.N)rrs r3rk7ModelResponsePartsManager._handle_embedded_thinking_ends %%n5r=c@URbX!R:waU$g)zBReturn the provider name if it has not been set on previous parts.N)r\)r1rvr\s r3rq0ModelResponsePartsManager._resolve_provider_names$  & & .-C^C^2^ r=cx[U[5(a%URURURS9 g[U[ 5(akUR UR5nUce[U[5(aeURURURRU5S9 gg)zQApply a replayed stream event to the managed parts, so `get_parts()` reflects it.)rtrLN) r@rrrcrLrrYrrir)r1eventrLs r3 apply_event%ModelResponsePartsManager.apply_events e^ , ,   EKKejj  I ~ . .--ekk:D# ##!$(9:: ::   EKKekk>O>OPT>U  V /r=)r+)returnNone)rr)rBobjectrbool)rGrrzToolPartKind | None)rLrrr)rzlist[ModelResponsePart])rXVendorIdrzManagedPart | None)rtVendorId | Nonerarr[ str | Noner\rr]dict[str, Any] | Noner^ztuple[str, str] | Noner_rr"Iterator[ModelResponseStreamEvent])rtHashable | Nonerarr[rr~rr\rr]rrr)rtrrGrrstr | dict[str, Any] | Nonerrr\rr]rrzModelResponseStreamEvent | None)rtrrGrrrrrr[rr\rr]rrr)rtrrLr rr)rtrrrr-)rL ManagedPartrtrrint)rSrrv5ToolCallPartDelta | ToolCallPart | NativeToolCallPartrirrr) rSrrvrrirrrrr) rSrrv!ToolCallPart | NativeToolCallPartrirrrrr)rvTextPart | ThinkingPartryz!TextPartDelta | ThinkingPartDeltargrrr)rSrrrrirrr)rSrrr)rr%)rSrrLrrr)rz type[PartT]rztuple[PartT, int] | None)rtrr\rr]rrr) rvrrSrrarr\rr]rrr)rtrrr)rvz%ModelResponsePart | ToolCallPartDeltar\rrr)rrrr)0__name__ __module__r9__firstlineno____doc____annotations__rlistrr&rrrr'rr*rr+r4r;rCrHrNrUrYr{rrrrrrprrrrrrsrrRr:rrjrmrlrkrqr__static_attributes__r=r3r r 9s 54 !&d;6Ge TF Tf49$xY\}J]di4j1jc,1S$s)^,5ue-O)=27SR^M^H_fkrw2x/x[   6 > P &$(2604*/aE(aE aE  aE " aE0aE.aE$(aE ,aEN# $$(15iE(iE iE  iE  iE"iE/iE ,iE^!%,0#'$(26lI(lI lI * lI ! lI"lI0lI )lIh$($(263C(3C 3C * 3C ! 3C 3C"3C03C "3Cj?(? ? " ?@D ]]M]! ] ? ]0.?HYad ><.OXiqt *(.6   ! *$O.U' 9&97A9Uj9 +9A#AA A " A 0 A ,A66BS] Wr=r N)(r __future__r _annotationscollections.abcrr dataclassesrrrtypingr r r pydantic_ai.exceptionsr pydantic_ai.messagesr rrrrrrrrrrrr_utilsrrmodelsrrrrr rr=r3rs 3.11..: D.  "33  {+ \ W\ W \ Wr=