add base hook event to separate single/multi agent

Author: JackYPCOnline

src/strands/hooks/__init__.py

@@ -35,7 +35,7 @@ def log_end(self, event: AfterInvocationEvent) -> None:
     BeforeInvocationEvent,
     MessageAddedEvent,
 )
-from .registry import HookCallback, HookEvent, HookProvider, HookRegistry,MultiAgentHookEvent
+from .registry import HookCallback, HookEvent, HookProvider, HookRegistry
 
 __all__ = [
     "AgentInitializedEvent",
@@ -46,5 +46,4 @@ def log_end(self, event: AfterInvocationEvent) -> None:
     "HookProvider",
     "HookCallback",
     "HookRegistry",
-    "MultiAgentHookEvent"
 ]

src/strands/hooks/registry.py

@@ -7,14 +7,17 @@
 via hook provider objects.
 """
 
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 from typing import TYPE_CHECKING, Any, Generator, Generic, Protocol, Type, TypeVar
 
 if TYPE_CHECKING:
     from ..agent import Agent
 
+
 @dataclass
 class HookEventBase:
+    """Base class for single hook events."""
+
     @property
     def should_reverse_callbacks(self) -> bool:
         """Determine if callbacks for this event should be invoked in reverse order.
@@ -56,21 +59,16 @@ def __setattr__(self, name: str, value: Any) -> None:
 
         raise AttributeError(f"Property {name} is not writable")
 
-@dataclass
-class MultiAgentHookEvent(HookEventBase):
-    pass
-
 
 @dataclass
 class HookEvent(HookEventBase):
-    """Base class for all hook events.
+    """Base class for single hook events.
 
     Attributes:
         agent: The agent instance that triggered this event.
     """
 
-    # agent: "Agent | None" = field(default=None, kw_only=True)
-    agent : "Agent"
+    agent: "Agent"
 
 
 TEvent = TypeVar("TEvent", bound=HookEventBase, contravariant=True)