diff --git a/src/google/adk/agents/__init__.py b/src/google/adk/agents/__init__.py
index e1f773c47..e48844397 100644
--- a/src/google/adk/agents/__init__.py
+++ b/src/google/adk/agents/__init__.py
@@ -21,6 +21,8 @@
 from .parallel_agent import ParallelAgent
 from .run_config import RunConfig
 from .sequential_agent import SequentialAgent
+from .callback_context import CallbackContext
+
 
 __all__ = [
     'Agent',
@@ -29,4 +31,5 @@
     'LoopAgent',
     'ParallelAgent',
     'SequentialAgent',
+    'CallbackContext',
 ]