"j%SrSSKJr SSKJrJr SSKJrJr SSK J r J r J r J r Jr SSKrSSKJrJr SS KJr SS KJrJrJrJrJrJr \ (aSS KJr "S S \5r"SS\5r\"SS9"SS\55r\"SS9"SS\55r \RB"\ \"\S45r#S\$S'\RB"\5r%S\$S'S$Sjr&S%Sjr'\&\S'\'\S'S\S'S\S 'S\\'S\\ 'S&S!jr(S'S"jr)S&S#jr*g)(z4Typed message parts for deferred capability loading.) annotations) CollectionSequence)KW_ONLY dataclass) TYPE_CHECKING AnnotatedLiteralUnioncastN) NotRequired TypedDict)copy_dataclass_fields)_TOOL_CALL_NARROWERS_TOOL_RETURN_NARROWERS_TYPED_PART_TAGS_TYPED_PART_TAGS_BY_TYPE ToolCallPartToolReturnPart) ModelMessagec$\rSrSr%SrS\S'Srg)LoadCapabilityArgsz2Typed arguments for a `load_capability` tool call.zOAnnotated[str, pydantic.Field(description='The id of the capability to load.')]idN__name__ __module__ __qualname____firstlineno____doc____annotations____static_attributes__r_/home/mande/repo/quber/.venv/lib/python3.13/site-packages/pydantic_ai/_deferred_capabilities.pyrrs<  (r%rc$\rSrSr%SrS\S'Srg)LoadCapabilityReturn(z2Typed return value for the `load_capability` tool.zNotRequired[str] instructionsrNrrr%r&r(r((s<""1r%r(F)reprc|\rSrSr%SrS\S'SrS\S'SrS \S 'S rS \S '\ SSj5r \ SSj5r Sr g)LoadCapabilityCallPart/z4Typed `ToolCallPart` for the `load_capability` tool.r_load_capabilityLiteral['load_capability'] tool_nameNzstr | LoadCapabilityArgs | Noneargscapability-loadLiteral['capability-load'] tool_kindc~URcg[SURSS95$![[4a gf=f)zJParsed load-capability arguments, or `None` for incomplete streaming args.NrT)raise_if_invalid)r3r args_as_dict ValueErrorAssertionErrorselfs r& typed_args!LoadCapabilityCallPart.typed_args>sJ 99  ,d.?.?QU.?.VW WN+  s )<<cDURnUcgURS5$)z1Capability id from the parsed args, if available.Nr)r>get)r=typeds r& capability_id$LoadCapabilityCallPart.capability_idHs# =yyr%r)returnzLoadCapabilityArgs | NonerEz str | None) rrr r!r"r#r2r3r6propertyr>rCr$rr%r&r-r-/sZ>J,=I)=+,0D )0',=I)=/ r%r-cd\rSrSr%SrS\S'S\S'SrS\S 'S rS \S '\SS j5r Sr g)LoadCapabilityReturnPartQz6Typed `ToolReturnPart` for the `load_capability` tool.rr/r(contentr0r1r2r4r5r6c8URRS5$)z'Loaded capability instructions, if any.r*)rKrAr<s r&r*%LoadCapabilityReturnPart.instructionscs||//r%rNrF) rrr r!r"r#r2r6rGr*r$rr%r&rIrIQsE@J !! ->I)=+,=I)=/ 00r%rIz5pydantic.TypeAdapter[str | LoadCapabilityArgs | None]_LOAD_CAPABILITY_CALL_ARGS_TAz*pydantic.TypeAdapter[LoadCapabilityReturn]"_LOAD_CAPABILITY_RETURN_CONTENT_TAc[U[5(aU$[RUR5n[ U[USS9$)Nr4)r3r6) isinstancer-rNvalidate_pythonr3r)partvalidated_argss r&_narrow_load_capability_callrUqs<$.// 2BB499MN '=N^o ppr%c[U[5(aU$[RUR5n[ U[USS9$)Nr4)rKr6)rQrIrOrRrKr)rSvalidated_contents r&_narrow_load_capability_returnrXxs>$011 :JJ4<<X '?IZfw xxr%r4zcapability-load-call)z tool-callr4zcapability-load-return)z tool-returnr4c0SSKJn [U"U55$)u Parse visible history to find capabilities loaded via `load_capability`. Every [`CompactionPart`][pydantic_ai.messages.CompactionPart] resets the derived state at its exact position in a response. This is deliberately provider-agnostic: over-counting can expose tools whose load evidence is no longer visible, while under-counting once only permitted a redundant, idempotent load. Now that availability gates execution, an under-count also *refuses* the call — see [`post_compaction_window`][pydantic_ai.messages.post_compaction_window] for when that is wrong and what is tracked to fix it. Only the [`post_compaction_window`][pydantic_ai.messages.post_compaction_window] is scanned — the one definition of the boundary — so only pairs entirely after the boundary count. r)post_compaction_window)messagesrZ_parse_loaded_capabilities)r[rZs r&parse_loaded_capabilitiesr]s 1 %&2D":;;044T5F5FG%JJv& Mr%)rSrrEr-)rSrrErI)r[Sequence[ModelMessage]rEset[str])r[rjr`zCollection[str]rErk)+r" __future__rcollections.abcrr dataclassesrrtypingrr r r r pydantictyping_extensionsr r_utilsrr[rrrrrrrrr(r-rI TypeAdapterstrrNr#rOrUrXr]rar\rr%r&rusB:"0*AA4)& ( (292 \B 0~00.X`WkWk #!4 '(XTRZQeQeR"$N qy+G&',J()5K127O343I/05M12H* E r%