܎jdZddlmZddlZddlmZmZmZmZddl m Z ddl m Z m Z dd lmZdd lmZdd lmZmZdd lmZgd ZGdde ZGdde ZedeZd2dZd3dZd4d#Zd5d%Zd6d)Z d7d-Z!d8d1Z"dS)9a=Shared tool-dispatch helpers for the tool runners. Both ``client.beta.messages.tool_runner`` (the Messages tool runner) and ``client.beta.sessions.events.tool_runner`` (the sessions-side :class:`~anthropic.lib.tools._beta_session_runner.SessionToolRunner`) do the same three small things: index the supplied tools by name, run a runnable tool over a JSON input, and turn an exception raised by a tool into tool-result content. Those steps are factored out here so the two runners stay consistent instead of each carrying its own copy. Consumed by the runner helpers only. ) annotationsN)UnionTypeVarIterable Awaitable)Protocol) ToolErrorBetaFunctionToolResultType)BetaMessageParam)BetaContentBlockParam)Tool BetaRequestToolRemovalBlockParam)!BetaRequestToolAdditionBlockParam) tool_registrytool_error_contentrun_runnable_toolavailable_tool_namesc*eZdZdZeddZdS) _NamedTooluHAnything with a ``name`` — the shape :func:`tool_registry` indexes on.returnstrcdSN)selfs PC:\Projects\DSCons\.venv\Lib\site-packages\anthropic\lib\tools\_tool_dispatch.pynamez_NamedTool.name!s3N)rr)__name__ __module__ __qualname____doc__propertyrrr rrrs,RR Xr rceZdZdZddZdS) _CallableToolzhA runnable tool: ``call`` may be sync or async (it returns either the result or an awaitable of it).inputobjectrHUnion[BetaFunctionToolResultType, Awaitable[BetaFunctionToolResultType]]cdSrr)rr(s rcallz_CallableTool.call)sr N)r(r)rr*)r!r"r#r$r,rr rr'r'%s)&&srrrrrr r' NamedToolT)boundtoolsIterable[NamedToolT]rdict[str, NamedToolT]cd|DS)zIndex ``tools`` by their ``name`` for O(1) dispatch lookup. On a duplicate name the later tool wins, matching a plain dict comprehension. ci|] }|j| Sr)r).0tools r z!tool_registry..4s . . .DIt . . .r r)r/s rrr/s / . . . ..r messagesIterable[BetaMessageParam] tool_names Iterable[str]set[str]ct|}|D]A}|d}|ddkst|tr,|D]}t||B|S)uvFold mid-conversation ``tool_removal`` / ``tool_addition`` blocks over the locally runnable ``tool_names``. Only ``role: "system"`` messages carry these blocks, and only a ``tool_reference`` can name a locally runnable tool — MCP references are executed server-side, so they (and any unknown block/reference type) are ignored rather than raising. contentrolesystem)set isinstancer_apply_tool_change)r7r9 availablemessager=blocks rrr7svJI11)$ 6?h & &*Wc*B*B &  1 1E ui 0 0 0 0 1 r rErrCNonec t|tsdS|ddks |ddkrt||dS|ddkr5|dD]*}|ddks |ddkrt||+dSdS)a6Apply a single ``tool_removal`` / ``tool_addition`` block to ``available``. A ``mid_conv_system`` block's ``content`` is limited by the API schema to ``text`` / ``tool_addition`` / ``tool_removal``, so exactly one level is walked (no deeper nesting exists); every other block type is a no-op. Ntype tool_removal tool_additionmid_conv_systemr=)rAdict_apply_tool_reference_change)rErCinners rrBrBJs eT " "  V}&&%-?*J*J$UI66666 v+ + +9% ? ?EV}..%-?2R2R,UI>>> ? ? r JUnion[BetaRequestToolRemovalBlockParam, BetaRequestToolAdditionBlockParam]ct|d}|dS|ddkr||dS||dS)zGFold one ``tool_removal`` / ``tool_addition`` block into ``available``.r5NrHrI)_referenced_tool_namediscardadd)rErCrs rrMrMas` !v / /D | V}&&$ dr ref_ToolChangeReference str | Nonec.|ddkr|dSdS)a1The locally runnable tool name a tool-change reference resolves to. Only ``tool_reference`` names a runnable tool; ``mcp_tool_reference`` / ``mcp_toolset_reference`` execute server-side and unknown reference types are ignored (forward compatibility), so all of those resolve to ``None``. rHtool_referencerNr)rTs rrQrQns# 6{&&&6{ 4r exc BaseExceptionr cXt|tr|jSt|S)aRender an exception raised by a tool as tool-result content. A :class:`ToolError` carries its own structured content; anything else is rendered with ``repr`` (which, unlike ``str``, keeps the exception type). The caller owns the ``is_error`` flag and any logging. )rAr r=repr)rYs rrrzs*#y!!{ 99r r5r(dict[str, object]clK||}tj|r|d{VS|S)a Call ``tool`` with ``input``, awaiting the result if the tool is async. Bridges the sync (:class:`~anthropic.lib.tools.BetaFunctionTool`) and async (:class:`~anthropic.lib.tools.BetaAsyncFunctionTool`) runnable-tool shapes behind a single ``await``. N)r,inspect isawaitable)r5r(results rrrsCYYu  F6""|||||| Mr )r/r0rr1)r7r8r9r:rr;)rErrCr;rrF)rErOrCr;rrF)rTrUrrV)rYrZrr )r5r'r(r]rr )#r$ __future__rr_typingrrrrtyping_extensionsr_beta_functionsr r types.beta.beta_message_paramr #types.beta.beta_content_block_paramr0types.beta.beta_request_tool_removal_block_paramrrUr1types.beta.beta_request_tool_addition_block_paramr__all__rr'r-rrrBrMrQrrrr rrks  #"""""666666666666&&&&&&BBBBBBBB======HHHHHHcbbbbb ^ ^ ^sssssHsssW\ 4 4 4 ////&    .                  r