Skip to content

feat: Add support for claude agent sdk hooks #71

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
Abhijeet Prasad (AbhiPrasad) wants to merge 6 commits into from
Closed

feat: Add support for claude agent sdk hooks #71

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
5f34430
feat: Add support for hooks instrumentation to claude agent sdk
AbhiPrasad Mar 12, 2026
ec74036
feat: Support module level query func
AbhiPrasad Mar 12, 2026
2afe06c
clean up
AbhiPrasad Mar 12, 2026
b3e122a
fix some regressions
AbhiPrasad Mar 17, 2026
4b112c8
refactor and cleanup
AbhiPrasad Mar 17, 2026
e9a15ad
chore: formatting
viadezo1er Mar 18, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 17 additions & 1 deletion py/src/braintrust/wrappers/claude_agent_sdk/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,13 @@

from braintrust.logger import NOOP_SPAN, current_span, init_logger

from ._wrapper import _create_client_wrapper_class, _create_tool_wrapper_class, _wrap_tool_factory
from ._wrapper import (
_create_client_wrapper_class,
_create_tool_wrapper_class,
_wrap_query_function,
_wrap_tool_factory,
)


logger = logging.getLogger(__name__)

Expand Down Expand Up @@ -67,9 +73,11 @@ def setup_claude_agent_sdk(
import claude_agent_sdk

original_client = claude_agent_sdk.ClaudeSDKClient if hasattr(claude_agent_sdk, "ClaudeSDKClient") else None
original_query_fn = claude_agent_sdk.query if hasattr(claude_agent_sdk, "query") else None
original_tool_class = claude_agent_sdk.SdkMcpTool if hasattr(claude_agent_sdk, "SdkMcpTool") else None
original_tool_fn = claude_agent_sdk.tool if hasattr(claude_agent_sdk, "tool") else None

wrapped_client = None
if original_client:
wrapped_client = _create_client_wrapper_class(original_client)
claude_agent_sdk.ClaudeSDKClient = wrapped_client
Expand All @@ -79,6 +87,14 @@ def setup_claude_agent_sdk(
if getattr(module, "ClaudeSDKClient", None) is original_client:
setattr(module, "ClaudeSDKClient", wrapped_client)

if original_query_fn and wrapped_client:
wrapped_query_fn = _wrap_query_function(original_query_fn, wrapped_client)
claude_agent_sdk.query = wrapped_query_fn

for module in list(sys.modules.values()):
if module and getattr(module, "query", None) is original_query_fn:
setattr(module, "query", wrapped_query_fn)

if original_tool_class:
wrapped_tool_class = _create_tool_wrapper_class(original_tool_class)
claude_agent_sdk.SdkMcpTool = wrapped_tool_class
Expand Down
Loading