zastrowm
diff --git a/‎src/strands/agent/agent.py
Lines changed: 9 additions & 0 deletions b/‎src/strands/agent/agent.py
Lines changed: 9 additions & 0 deletions
diff --git a/‎src/strands/hooks/__init__.py b/‎src/strands/hooks/__init__.py
diff --git a/‎src/strands/hooks/agent_hook.py
Lines changed: 56 additions & 0 deletions b/‎src/strands/hooks/agent_hook.py
Lines changed: 56 additions & 0 deletions
@@ -24,6 +24,7 @@
 from ..event_loop.event_loop import event_loop_cycle
 from ..handlers.callback_handler import CompositeCallbackHandler, PrintingCallbackHandler, null_callback_handler
 from ..handlers.tool_handler import AgentToolHandler
+from ..hooks.agent_hook import AgentHook, AgentHookManager, AgentInitialized
 from ..models.bedrock import BedrockModel
 from ..telemetry.metrics import EventLoopMetrics
 from ..telemetry.tracer import get_tracer
@@ -186,6 +187,7 @@ def __init__(
         model: Union[Model, str, None] = None,
         messages: Optional[Messages] = None,
         tools: Optional[List[Union[str, Dict[str, str], Any]]] = None,
+        hooks: Optional[List[AgentHook]] = None,
         system_prompt: Optional[str] = None,
         callback_handler: Optional[
             Union[Callable[..., Any], _DefaultCallbackHandlerSentinel]
@@ -289,6 +291,13 @@ def __init__(
         self.trace_span: Optional[trace.Span] = None
 
         self.tool_caller = Agent.ToolCaller(self)
+        self.hooks = AgentHookManager(agent=self)
+
+        if hooks is not None:
+            for hook in hooks:
+                self.hooks.add(hook)
+
+        self.hooks.get_hook(AgentInitialized)(agent=self)
 
     @property
     def tool(self) -> ToolCaller:
 
@@ -0,0 +1,56 @@
+from typing import Optional, Protocol, TYPE_CHECKING, List, Type, Dict, Any, TypeVar, ParamSpec, Callable, Generic
+
+from strands.types.tools import AgentTool
+
+
+if TYPE_CHECKING:
+    from strands import Agent
+
+T = TypeVar('T')
+
+class Reference(Generic[T]):
+    def __init__(self, value: T) -> None:
+        self.value = value
+
+class AgentInitialized(Protocol):
+    def __call__(self, *, agent: "Agent") -> None: ...
+
+class ToolTransformer(Protocol):
+    def __call__(self, agent: "Agent", tool: Reference[AgentTool]) -> None: ...
+
+class AgentHook(Protocol):
+    def register_hooks(self, hooks: "AgentHookManager", agent: "Agent") -> None: ...
+
+
+class AgentHookManager:
+
+    registered_hooks: Dict[Type, List[Any]] = {}
+
+    def __init__(self, agent: "Agent",  hooks: Optional[List[AgentHook]] = None) -> None:
+        self.agent = agent
+        self.hooks = hooks
+
+    def add(self, hook: AgentHook):
+        hook.register_hooks(hooks=self, agent=self.agent)
+
+    def add_hook(self, hook_type: T, callback: T):
+        if hook_type not in self.registered_hooks:
+            self.registered_hooks[hook_type] = []
+
+        self.registered_hooks[hook_type].append(callback)
+
+    def get_hook(self, hook_type: Type[T]) -> T:
+        return lambda *args, **kwargs: self._invoke_hook(hook_type, *args, **kwargs)
+
+    def _invoke_hook(self, hook_type: Type[Callable], *args: Any, **kwargs: Any) -> None:
+        if hook_type not in self.registered_hooks:
+            return
+
+        for hook in self.registered_hooks[hook_type]:
+            hook(*args, **kwargs)
+
+        return
+
+
+
+