o ^۷i< @sUddlmZddlZddlmZmZddlmZmZm Z m Z m Z m Z m Z mZmZddlmZddlmZmZmZddlZddlZddlmZdd lmZdd lmZdd lmZdd lmZdd l m!Z!m"Z"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z*e+e,Z-ee.e e*fZ/de0d<Gddde1Z2e de/fZ3e de3dZ4e de eee/ffZ5e de5dZ6GdddeZ7GdddeZ8Gddde e%Z9Gdd d e9e4Z:Gd!d"d"e9e6Z;edGd&d'Zee;ee8fZ?dS)P) annotationsN)ABCabstractmethod) AnyUnionGenericTypeVarCallableIterable Coroutinecastoverload)iscoroutinefunction)Literal TypeAliasoverride) BaseModel)_compat)is_dict)cached_property) TypeAdapter) BetaToolParamBetaToolUnionParamBetaCacheControlEphemeralParam) CallableT) InputSchema)ContentrBetaFunctionToolResultTypecs,eZdZUdZded<dfdd ZZS) 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"}}, ]) rcontentreturnNonecst|tr|}n.g}|D] }|d}|dur|t|q |d|dddq |r4d|nd}t|||_dS)Ntext[typeunknown] z Tool error) isinstancestrgetappendjoinsuper__init__r )selfr messagepartsblockr# __class__Y/home/ubuntu/vllm_env/lib/python3.10/site-packages/anthropic/lib/tools/_beta_functions.pyr/+s    zToolError.__init__)r rr!r")__name__ __module__ __qualname____doc____annotations__r/ __classcell__r6r6r4r7rs  r. FunctionT)boundAsyncFunctionTc@6eZdZedddZeddd Zedd d Zd S)BetaBuiltinFunctionToolr!rcCdSNr6r0r6r6r7to_dictBzBetaBuiltinFunctionTool.to_dictinputobjectrcCrCrDr6r0rHr6r6r7callErGzBetaBuiltinFunctionTool.callr*cC |}d|vr |dS|dSNmcp_server_namenamerFr0rawr6r6r7rOHzBetaBuiltinFunctionTool.nameNr!rrHrIr!rr!r*r8r9r:rrFrKpropertyrOr6r6r6r7rBA  rBc@rA)BetaAsyncBuiltinFunctionToolr!rcCrCrDr6rEr6r6r7rFQrGz$BetaAsyncBuiltinFunctionTool.to_dictrHrIrcsdSrDr6rJr6r6r7rKTsz!BetaAsyncBuiltinFunctionTool.callr*cCrLrMrPrQr6r6r7rOWrSz!BetaAsyncBuiltinFunctionTool.nameNrTrUrVrWr6r6r6r7rZPrYrZc @seZdZUded< ded< ded<ded<dddddddddd d)ddZed*ddZd+ddZed,d d!Z d-d"d#Z d.d$d%Z ed/d'd(Z dS)0BaseFunctionToolrfuncr*rO descriptionr input_schemaN rOr]r^ defer_loading cache_controlallowed_callerseager_input_streaminginput_examplesstrict str | None$InputSchema | type[BaseModel] | Noner` bool | Nonera%BetaCacheControlEphemeralParam | NonerbTlist[Literal['direct', 'code_execution_20250825', 'code_execution_20260120']] | Nonercrd"Iterable[dict[str, object]] | Nonerer!r"c Cstjrtd||_t||_|p|j|_||_ ||_ ||_ ||_ | |_ | |_|p-||_|durDt|tr?||_dS||_dS||_dS)N2Tool functions are only supported with Pydantic v2)r PYDANTIC_V1 RuntimeErrorr\pydantic validate_call_func_with_validater8rO_defer_loading_cache_control_allowed_callers_eager_input_streaming_input_examples_strict_get_description_from_docstringr]r)r%model_json_schemar^_create_schema_from_function) r0r\rOr]r^r`rarbrcrdrer6r6r7r/js"    zBaseFunctionTool.__init__cCs|jSrDr\rEr6r6r7__call__szBaseFunctionTool.__call__rcCs|j|j|jd}|jdur|j|d<|jdur|j|d<|jdur'|j|d<|jdur1|j|d<|jdur;|j|d<|jdurE|j|d<|S)N)rOr]r^r`rarbrcrdre) rOr]r^rrrsrtrurvrw)r0defnr6r6r7rFs"            zBaseFunctionTool.to_dictdocstring_parser.DocstringcCst|jjpdS)N)docstring_parserparser\r;rEr6r6r7_parsed_docstringsz"BaseFunctionTool._parsed_docstringcCs2|jjr|jj}|jjr|d|jj7}|SdS)z*Extract description from parsed docstring.z r)rshort_descriptionlong_description)r0r]r6r6r7rxs z0BaseFunctionTool._get_description_from_docstringcCsVddlm}ddlm}m}ddlm}Gddd|}||j|jd}|j j |dS) z:Create JSON schema from function signature using pydantic.r) CoreSchema)JsonSchemaValueGenerateJsonSchema)ArgumentsParametercs:eZdZdfdd Zdd d Zedfdd ZZS)zOBaseFunctionTool._create_schema_from_function..CustomGenerateJsonSchemar\Callable[..., Any]parsed_docstringrr!r"cst||_||_dSrD)r.r/_funcr)r0r\rr4r6r7r/s  zXBaseFunctionTool._create_schema_from_function..CustomGenerateJsonSchema.__init___args_kwds'CustomGenerateJsonSchema'c_s|SrDr6)r0rrr6r6r7r|rGzXBaseFunctionTool._create_schema_from_function..CustomGenerateJsonSchema.__call__ arguments'list[ArgumentsParameter]'var_kwargs_schemaCoreSchema | Nonercs~t||}|ddkr|S|d}|rt|s|S|jjD]}||j}|r/t|s0q!|jr.CustomGenerateJsonSchema.kw_arguments_schema)r\rrrr!r")rrrrr!r)rrrrr!r)r8r9r:r/r|rrr=r6r6r4r7CustomGenerateJsonSchemas  r)r\r)schema_generator) pydantic_corerpydantic.json_schemarrpydantic_core.core_schemarr\r_adapter json_schema)r0rrrrrrr6r6r7rzs  "z-BaseFunctionTool._create_schema_from_functionTypeAdapter[Any]cCs t|jSrD)rrqrEr6r6r7rs zBaseFunctionTool._adapter)r\rrOrfr]rfr^rgr`rhrarirbrjrcrhrdrkrerhr!r")r!r)r!r)r!r~rV)r!r)r!r) r8r9r:r<r/rXr|rFrrrxrzrr6r6r6r7r[_s4 %     ,r[c@eZdZdddZdS) BetaFunctionToolrHrIr!rc Cstt|jr tdt|stdt|jz |jditt |WSt j y9}z t d|j |d}~ww)NzJCannot call a coroutine function synchronously. Use `@async_tool` instead. Input must be a dictionary, got Invalid arguments for function r6rr\rnr TypeErrorr%r8rqr rroValidationError ValueErrorrOr0rHer6r6r7rKs zBetaFunctionTool.callNrUr8r9r:rKr6r6r6r7rrc@r) BetaAsyncFunctionToolrHrIr!rc s|t|js tdt|stdt|jz|jditt |IdHWSt j y=}z t d|j |d}~ww)NzGCannot call a synchronous function asynchronously. Use `@tool` instead.rrr6rrr6r6r7rKs zBetaAsyncFunctionTool.callNrUrr6r6r6r7rrrr\r!BetaFunctionTool[FunctionT]cCrCrDr6r{r6r6r7 beta_toolrGrr_rOrfr]r^rgr`rhrarirbrjrcrdrkrec CrCrDr6 r\rOr]r^r`rarbrcrdrer6r6r7r 2Callable[[FunctionT], BetaFunctionTool[FunctionT]]c CrCrDr6r_r6r6r7r FunctionT | NonePBetaFunctionTool[FunctionT] | Callable[[FunctionT], BetaFunctionTool[FunctionT]]c @tjrtdd f dd } |dur| |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: ... rlfnr>r!rc t|d SNr_)rr rbrar`r]rcrdr^rOrer6r7_makeAzbeta_tool.._makeN)rr>r!rrrmrn r\rOr]r^r`rarbrcrdrerr6rr7r$ %BetaAsyncFunctionTool[AsyncFunctionT]cCrCrDr6r{r6r6r7beta_async_toolUrGrc CrCrDr6rr6r6r7rYrACallable[[AsyncFunctionT], BetaAsyncFunctionTool[AsyncFunctionT]]c CrCrDr6r_r6r6r7rirAsyncFunctionT | NoneiBetaAsyncFunctionTool[AsyncFunctionT] | Callable[[AsyncFunctionT], BetaAsyncFunctionTool[AsyncFunctionT]]c r) 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: ... rlrr@r!rc rr)rrrr6r7rrzbeta_async_tool.._makeN)rr@r!rrrr6rr7rxr)r\r>r!r)r\r>rOrfr]rfr^rgr`rhrarirbrjrcrhrdrkrerhr!r)rOrfr]rfr^rgr`rhrarirbrjrcrhrdrkrerhr!rrD)r\rrOrfr]rfr^rgr`rhrarirbrjrcrhrdrkrerhr!r)r\r@r!r)r\r@rOrfr]rfr^rgr`rhrarirbrjrcrhrdrkrerhr!r)rOrfr]rfr^rgr`rhrarirbrjrcrhrdrkrerhr!r)r\rrOrfr]rfr^rgr`rhrarirbrjrcrhrdrkrerhr!r)@ __future__rloggingabcrrtypingrrrrr r r r r inspectrtyping_extensionsrrrrorrrr_utilsrr_modelsr types.betarrr _utils._utilsrtypes.tool_paramr'types.beta.beta_tool_result_block_paramr BetaContent getLoggerr8logr*rr< ExceptionrFunctionr> AsyncFunctionr@rBrZr[rrrrBetaRunnableToolBetaAsyncRunnableToolr6r6r6r7s,              1 1