f3fi3UdZddlZddlZddlmZmZddlmZmZmZddl m Z ddl m Z ddl mZer dd lmZmZmZej&eZiaeeeeeffed <iaeee fed <d d deeddddfdZd d deeddddfdZddZy)zHook-based tool tracing for Claude Agent SDK. This module provides hook handlers that traces tool calls by intercepting `PreToolUse` and `PostToolUse` events. N)datetimetimezone) TYPE_CHECKINGAnyOptional)get_current_run_tree)RunTree)get_parent_run_tree) HookContext HookInputHookJSONOutput_active_tool_runs_client_managed_runs input_datar tool_use_idcontextr returnrc K|stjdiS|tvriSt|j dd}|j di} t xs t }|stjd|iStj}|j|d|rd|initj|tj } |j||ft |<tjd |d|diS#t$r%}tjd |d |Yd }~Rd }~wwxYw#t$r(}tjd|d |dYd }~iSd }~wwxYww)aETrace tool execution before it starts. Args: input_data: Contains `tool_name`, `tool_input`, `session_id` tool_use_id: Unique identifier for this tool invocation context: Hook context (currently contains only signal) Returns: Hook output (empty dict allows execution to proceed) z:PreToolUse hook called without tool_use_id, skipping trace tool_name unknown_tool tool_inputz"No parent run tree found for tool toolinput)tz)namerun_typeinputs start_timezFailed to post tool run for : NzStarted tool trace for  (id=)zError in PreToolUse hook for Texc_info)loggerdebugrstrgetr rtime create_childr fromtimestamprutcpost Exceptionwarningr) rrrrrparentrtool_runes l/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langsmith/integrations/claude_agent_sdk/_hooks.pypre_tool_use_hookr4#sy  QR **  ^DEI b1JX$&@*>*@ LL=i[I JIYY[ &&,6GZ(B--jX\\J '  L MMO+3J)?+& .yk{m1MN I L NN9)BqcJ K K L X6ykA3GRVWW IXsgAE71EE7AED)'EE7 ED;6E;EE E4 E/)E7/E44E7c 0K|stjdiSt|jdd}|jd}tj |d}|r t |tr|}n&t |trd|i}n|r dt|ini}d}t |tr|jd d}|j||r|jdnd |jiS tj |d} | stjd |d |diS| \} } t |tr|}n&t |trd|i}n|r dt|ini}d}t |tr|jd d}| j||r|jdnd  | jtj| z dz} tjd|d |d| ddiS#t$r#}tjd |Yd}~iSd}~wwxYw#t$r%}tjd|d|Yd}~d}~wwxYw#t$r(}tjd|d|dYd}~iSd}~wwxYww)aOTrace tool execution after it completes. Args: input_data: Contains `tool_name`, `tool_input`, `tool_response`, `session_id`, etc. tool_use_id: Unique identifier for this tool invocation context: Hook context (currently contains only signal) Returns: Hook output (empty `dict` by default) z;PostToolUse hook called without tool_use_id, skipping tracerr tool_responseNcontentoutputFis_error)outputserrorz%Failed to update client-managed run: z!No matching PreToolUse found for r!r"zFailed to patch tool run for r izCompleted tool trace for z , duration=z.2fzms)zError in PostToolUse hook for Tr#)r%r&r'r(rpop isinstancedictlistendpatchr.r/rr)) rrrrr6run_treer:r9r2run_infor1r duration_mss r3post_tool_use_hookrEYs  RS  ^DEINN?3M$'' T:H H-.'M40$m4r'tuplefloat__annotations__rr4rErJr3rYs  '//6''   8 $354U3:../4,.d3<(-33#33 3lXX#XX Xv!rX