Áj\ UddlmZddlZddlZddlmZmZddlmZm Z m Z m Z m Z m Z mZmZmZddlmZmZmZmZddlmZddlmZmZmZddlZddlZddlZddlmZd d l m!Z!d d l"m#Z#d d l!m$Z$d d l%m&Z&d dl'm(Z(m)Z)m*Z*d dl+m,Z,d dl-m.Z.d dl/m0Z1ejde3Z4e e5e e1fZ6de7d<Gdde8Z9e de6fZ:e de:Z;e deeee6ffZ<e de<Z=GddeZ>GddeZ?Gdde e,Z@Gd d!e@e;ZAGd"d#e@e=ZBd0d$ZCd0d%ZDd1d&ZEed2d'ZFedddddddddd( d3d)ZFedddddddddd( d4d*ZF d5dddddddddd( d6d+ZFed7d,ZGedddddddddd( d8d-ZGedddddddddd( d9d.ZG d5dddddddddd( d:d/ZGe eAee>fZHe eBee?fZIy);) annotationsN)ABCabstractmethod) AnyUnionGenericTypeVarCallableIterable Coroutinecastoverload) isawaitableisasyncgenfunctioniscoroutinefunctionisgeneratorfunction) Awaitable)Literal TypeAliasoverride) BaseModel)_compat)is_dict)cached_property) TypeAdapter) BetaToolParamBetaToolUnionParamBetaCacheControlEphemeralParam) CallableT) InputSchema)ContentrBetaFunctionToolResultTypec0eZdZUdZded<dfd ZxZS) ToolErroraError that can be raised from a tool to return structured content with ``is_error: True``. When the tool runner catches this error, it will use the :attr:`content` property as the tool result instead of ``repr(exc)``. Example:: raise ToolError( [ {"type": "text", "text": "Error details here"}, {"type": "image", "source": {"type": "base64", "data": "...", "media_type": "image/png"}}, ] ) r#contentc:t|tr|}nqg}|D]U}|jd}||jt|1|jd|jdddW|rdj |nd}t ||||_y)Ntext[typeunknown] z Tool error) isinstancestrgetappendjoinsuper__init__r&)selfr&messagepartsblockr( __class__s J/tmp/pip-target-k_scjv0h/lib/python/anthropic/lib/tools/_beta_functions.pyr4zToolError.__init__0s gs #G!E  Fyy(#LLT+LL1UYYvy%A$B!!DE  F */chhuoLG ! )r&r#returnNone)__name__ __module__ __qualname____doc____annotations__r4 __classcell__r9s@r:r%r%s ('  r;r%. FunctionT)boundAsyncFunctionTcBeZdZeddZeddZeddZy)BetaBuiltinFunctionToolcyNr5s r:to_dictzBetaBuiltinFunctionTool.to_dictH-0r;cyrKrLr5inputs r:callzBetaBuiltinFunctionTool.callKsADr;c>|j}d|vr|dS|dSNmcp_server_namenamerNr5raws r:rWzBetaBuiltinFunctionTool.nameN+lln  #() )6{r;Nr<rrRobjectr<r#r<r/r>r?r@rrNrSpropertyrWrLr;r:rIrIGs000DD r;rIcBeZdZeddZeddZeddZy)BetaAsyncBuiltinFunctionToolcyrKrLrMs r:rNz$BetaAsyncBuiltinFunctionTool.to_dictWrOr;c KywrKrLrQs r:rSz!BetaAsyncBuiltinFunctionTool.callZs GJsc>|j}d|vr|dS|dSrUrXrYs r:rWz!BetaAsyncBuiltinFunctionTool.name]r[r;Nr\r]r_r`rLr;r:rcrcVs000JJ r;rcc eZdZUded< ded< ded<ded<dZd ed < dZd ed < dddddddddd ddZeddZddZ e ddZ ddZ ddZ e ddZy)BaseFunctionToolr funcr/rW descriptionr! input_schemaNz+Callable[[], None | Awaitable[None]] | Noneclosez object | None_context_manager rWrjrk defer_loading cache_controlallowed_callerseager_input_streaminginput_examplesstrictc tjr td||_t j ||_|xs |j|_||_ ||_ ||_ ||_ | |_ | |_|xs|j|_|.t#|t$r|j'|_y||_y|j+|_y)N2Tool functions are only supported with Pydantic v2)r PYDANTIC_V1 RuntimeErrorripydantic validate_call_func_with_validater>rW_defer_loading_cache_control_allowed_callers_eager_input_streaming_input_examples_strict_get_description_from_docstringrjr.r*model_json_schemark_create_schema_from_function) r5rirWrjrkrorprqrrrsrts r:r4zBaseFunctionTool.__init__s"   ST T #+#9#9$#? )DMM ++ /&;#- &P$*N*N*P  #,-1=1O1O1Q!$0! $ A A CD r;c|jSrKrirMs r:__call__zBaseFunctionTool.__call__s yyr;c|j|j|jd}|j|j|d<|j|j|d<|j |j |d<|j |j |d<|j|j|d<|j|j|d<|S)N)rWrjrkrorprqrrrsrt) rWrjrkr|r}r~rrr)r5defns r:rNzBaseFunctionTool.to_dictsII++ --     *$($7$7D !    *$($7$7D !  ,&*&;&;D" #  & & 2,0,G,GD( )    +%)%9%9D! " << #!\\DN r;c\tj|jjxsdS)N)docstring_parserparserirArMs r:_parsed_docstringz"BaseFunctionTool._parsed_docstrings"%%dii&7&7&=2>>r;c|jjrJ|jj}|jjr|d|jjz }|Sy)z*Extract description from parsed docstring.z r)rshort_descriptionlong_description)r5rjs r:rz0BaseFunctionTool._get_description_from_docstringsX  ! ! 3 300BBK%%66d&<&<&M&M%NOO  r;cddlm}ddlm}m}ddlm}Gdd|}||j|j}|jj|S) z:Create JSON schema from function signature using pydantic.r) CoreSchema)JsonSchemaValueGenerateJsonSchema)ArgumentsParametercJeZdZdfd ZddZe dfd ZxZS)OBaseFunctionTool._create_schema_from_function..CustomGenerateJsonSchemac>t|||_||_yrK)r3r4_funcr)r5riparsed_docstringr9s r:r4zXBaseFunctionTool._create_schema_from_function..CustomGenerateJsonSchema.__init__s "! )9&r;c|SrKrL)r5_args_kwdss r:rzXBaseFunctionTool._create_schema_from_function..CustomGenerateJsonSchema.__call__s r;c^t|||}|jddk7r|S|jd}|r t|s|S|jj D]L}|j|j }|r t|s,|js9d|vs>|j|d<N|S)Nr*r^ propertiesrj)r3kw_arguments_schemar0rrparamsarg_namerj)r5 argumentsvar_kwargs_schemaschemarparam prop_schemar9s r:rzcBaseFunctionTool._create_schema_from_function..CustomGenerateJsonSchema.kw_arguments_schemas 4Y@QR::f%1!M#ZZ 5 !)<!M"33::GE",.."@K&gk.B ((]+-M5:5F5F M2 G r;)rizCallable[..., Any]rrr<r=)rrrrr<z'CustomGenerateJsonSchema')rz'list[ArgumentsParameter]'rzCoreSchema | Noner<r)r>r?r@r4rrrrCrDs@r:CustomGenerateJsonSchemars= :   5 $5 !   r;r)rir)schema_generator) pydantic_corerpydantic.json_schemarrpydantic_core.core_schemarrir_adapter json_schema)r5rrrrrrs r:rz-BaseFunctionTool._create_schema_from_functionsN -L@ '9 D4UYUkUkl}}((:J(KKr;c,t|jSrK)rr{rMs r:rzBaseFunctionTool._adapters43344r;)rir rW str | Nonerjrrk$InputSchema | type[BaseModel] | Nonero bool | Nonerp%BetaCacheControlEphemeralParam | Nonerqolist[Literal['direct', 'code_execution_20250825', 'code_execution_20260120', 'code_execution_20260521']] | Nonerrrrs"Iterable[dict[str, object]] | Nonertrr<r=)r<r )r<r)r<zdocstring_parser.Docstringr_)r<r!)r<zTypeAdapter[Any])r>r?r@rBrlrmr4rarrNrrrrrrLr;r:rhrhes* O, I;9=E 6=$'+m* "&=A%)?C-1=A"&D&D &D  &D ; &D#&D=&D&D +&D;&D&D !&DP(??*LX55r;rhceZdZddZy)BetaFunctionToolcDt|jr tdt|s!t dt |j  |jditt|S#tj$r}td|j|d}~wwxYw)NzJCannot call a coroutine function synchronously. Use `@async_tool` instead. Input must be a dictionary, got Invalid arguments for function rLrrirxr TypeErrorr*r>r{r rryValidationError ValueErrorrWr5rRes r:rSzBetaFunctionTool.call s tyy )kl lu~>tE{?S?S>TUV V S+4++?d3.>? ?'' S>tyykJKQR R SsA..BBBNr]r>r?r@rSrLr;r:rr  Sr;rceZdZddZy)BetaAsyncFunctionToolc`Kt|js tdt|s!t dt |j  |jditt|d{S7#tj$r}td|j|d}~wwxYww)NzGCannot call a synchronous function asynchronously. Use `@tool` instead.rrrLrrs r:rSzBetaAsyncFunctionTool.calls"499-hi iu~>tE{?S?S>TUV V S111EDe4DEE EE'' S>tyykJKQR R Ss<A B.#A:3A84A:7B.8A::B+ B&&B++B.Nr]rrLr;r:rrrr;rc>t|dd}|duxr t|S)uTrue when ``fn`` is a function produced by :func:`contextlib.contextmanager`. ``contextmanager`` wraps the generator function with ``functools.wraps``, so the original generator function is reachable as ``__wrapped__`` — the same signal :mod:`inspect` itself uses. We never call ``fn`` to find out, so a plain tool function is never accidentally invoked during detection. __wrapped__N)getattrrfnwrappeds r:_is_sync_cm_factoryr%s(b-.G $  ?#6w#??r;c>t|dd}|duxr t|S)zRTrue when ``fn`` is a function produced by :func:`contextlib.asynccontextmanager`.rN)rrrs r:_is_async_cm_factoryr1s&b-.G $  >#5g#>>r;c Kt|ddxs t|dd}| |}t|r |d{t|dd}|7 t|dd}||dddd{y|j dddyy7K#t$r,}tj dt|dd|Yd}~yd}~wwxYw7T#t$r,}tj d t|dd|Yd}~yd}~wwxYww) uRun a runnable tool's optional cleanup. Drives the legacy ``aclose`` / ``close`` attribute (awaited if it returns an awaitable) and, when the tool was defined as a context manager via :func:`beta_tool` / :func:`beta_async_tool`, its ``__exit__`` / ``__aexit__``. Both run when both are present — the context-manager support is purely additive to ``close``. Exceptions are logged, never raised, so one tool's bad cleanup can't abort another tool's. acloseNrlz"tool.close failed tool=%s error=%srW?rm __aexit__z4tool context-manager cleanup failed tool=%s error=%s)rr Exceptionlogwarning__exit__)toolcloserresultrcmaexits r:aclose_runnable_toolr7s T8T * JgdGT.JF  ]XF6"  )4 0B ~ oB T2E D$--- D$-   ] KKXZ[ \ \ ]. o KKNPWX\^dfiPjlm n n osC9BBBC9C*B?+C/C90CC9B B<"B72C97B<<C9?C C6 "C1,C91C66C9cyrKrLrs r: beta_toolrVs?Br;rnc yrKrL rirWrjrkrorprqrrrsrts r:rrZs #&r;c yrKrLrns r:rrms:=r;c ~ tjr tdd f d } || |S| S)aFCreate a FunctionTool from a function with automatic schema inference. Can be used as a decorator with or without parentheses: @function_tool def my_func(x: int) -> str: ... @function_tool() def my_func(x: int) -> str: ... @function_tool(name="custom_name") def my_func(x: int) -> str: ... rvc r t|r tdt|rRtt|}|j } t tt|    }||_ |St |    S#t$r#|jtjwxYw)Nzi@beta_tool was applied to an @asynccontextmanager; use @beta_async_tool for an async context-manager toolrn) rrrr r __enter__rrE BaseExceptionrsysexc_inform) rrinnerrrqrprorjrrrsrkrWrts r:_makezbeta_tool.._makes  #I  r " c2BLLNE 'E* +!-"/"/$3*?#1! "%'D !K #%''+"7)  !  S\\^,  s #B ,B6)rrEr<BetaFunctionTool[FunctionT]rrwrx rirWrjrkrorprqrrrsrtrs ````````` r:rrs=:OPP, , \ T{ Lr;cyrKrLrs r:beta_async_toolrsTWr;c yrKrLrs r:rrs -0r;c yrKrLrns r:rrs ILr;c ~ tjr tdd f d } || |S| S)aUCreate an AsyncFunctionTool from a function with automatic schema inference. Can be used as a decorator with or without parentheses: @async_tool async def my_func(x: int) -> str: ... @async_tool() async def my_func(x: int) -> str: ... @async_tool(name="custom_name") async def my_func(x: int) -> str: ... rvc  t|r tdt|r tdtt|dddt j gd fd d fd }xs t|dd|_  n t|d d|_ ttt|    }j||St|    S) Nza@beta_async_tool was applied to a @contextmanager; use @beta_tool for a sync context-manager toolzan @asynccontextmanager tool needs an explicit input_schema= (its parameters can't be inferred until the context manager is entered)F) validatedenteredc&Kdsd4d{ds?jd{}tj|d<dd<d_dddd{dSdS7b7G7#1d{7swYdSxYww)NrrTr) __aenter__ryrzrm)rr enter_lockstatetool_boxs r:_enteredz0beta_async_tool.._make.._entered1sY')>>$Y/*,--/$9E191G1G1NE+./3E),<>HQK8>>[))u[))>$9>>>>[))sT BA2BA8A4+A8 B#A6$B4A86B8B>B ?B BcnKd{}tt|di|d{S7"7 w)NrL)r r#)kwargsrrs r:_lazyz-beta_async_tool.._make.._lazy=s3"**, 6i>Q&>Q8QRR-8Qs 5153 55r>rrArn)r<r)rrr<r#) rrrr ranyioLockrr>rArrGr1)rrrrrrrrrqrprorjrrrsrkrWrts @@@@@r:rzbeta_async_tool.._makes r "s   # #^c2B265$IEJDFH * * S"DWRV%DEN+6+BKPRT]_cHdEM(^U+')++ /&;- D OOD !K$ #%''+"7)  r;)rrGr<%BetaAsyncFunctionTool[AsyncFunctionT]rrs ````````` r:rrs?:OPP@ @ D T{ Lr;)rr^r<bool)rr^r<r=)rirEr<r)rirErWrrjrrkrrorrprrqrrrrrsrrtrr<r)rWrrjrrkrrorrprrqrrrrrsrrtrr<z2Callable[[FunctionT], BetaFunctionTool[FunctionT]]rK)rizFunctionT | NonerWrrjrrkrrorrprrqrrrrrsrrtrr<zPBetaFunctionTool[FunctionT] | Callable[[FunctionT], BetaFunctionTool[FunctionT]])rirGr<r)rirGrWrrjrrkrrorrprrqrrrrrsrrtrr<r)rWrrjrrkrrorrprrqrrrrrsrrtrr<zACallable[[AsyncFunctionT], BetaAsyncFunctionTool[AsyncFunctionT]])rizAsyncFunctionT | NonerWrrjrrkrrorrprrqrrrrrsrrtrr<ziBetaAsyncFunctionTool[AsyncFunctionT] | Callable[[AsyncFunctionT], BetaAsyncFunctionTool[AsyncFunctionT]])J __future__rrloggingabcrrtypingrrrr r r r r rinspectrrrrcollections.abcrtyping_extensionsrrrrryrrrr_utilsrr_modelsr types.betarrr _utils._utilsr types.tool_paramr!'types.beta.beta_tool_result_block_paramr" BetaContent getLoggerr>rr/r#rBrr%FunctionrE AsyncFunctionrGrIrcrhrrrrrrrBetaRunnableToolBetaAsyncRunnableToolrLr;r:rsL" #^^^]]%:: &"[[&+Mg!(-c8K3H.H(III D C33 4 Kx 0 iS2L(LMMN )? c  3 a5wy)a5H S' 2 S S,^< S @? o> B B "9=!%;?)-9=& & & & 7 &  &9& &'&7& &!& &$ "9=!%;?)-9== ==7 =  = 9 = ='=7= =8= =$"Q"9=!%;?)-9=Q Q Q Q 7 Q  Q9Q Q'Q7Q QVQh W W "9=!%;?)-9=0 0 0 0 7 0  090 0'070 0+0 0$ "9=!%;?)-9=L LL7 L  L 9 L L'L7L LGL L$#'e"9=!%;?)-9=e e e e 7 e  e9e e'e7e eoeP)#.0GGH3C8:VVWr;