feat:add BaseHookEvent for multiagent use

Author: JackYPCOnline

src/strands/hooks/__init__.py

@@ -39,7 +39,7 @@ def log_end(self, event: AfterInvocationEvent) -> None:
     BeforeToolCallEvent,
     MessageAddedEvent,
 )
-from .registry import HookCallback, HookEvent, HookProvider, HookRegistry,MultiAgentHookEvent
+from .registry import BaseHookEvent, HookCallback, HookEvent, HookProvider, HookRegistry
 
 __all__ = [
     "AgentInitializedEvent",
@@ -54,5 +54,6 @@ def log_end(self, event: AfterInvocationEvent) -> None:
     "HookProvider",
     "HookCallback",
     "HookRegistry",
-    "MultiAgentHookEvent"
+    "HookEvent",
+    "BaseHookEvent",
 ]

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:
+class BaseHookEvent:
+    """Base class for all hook events."""
+
     @property
     def should_reverse_callbacks(self) -> bool:
         """Determine if callbacks for this event should be invoked in reverse order.
@@ -56,27 +59,22 @@ 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.
+class HookEvent(BaseHookEvent):
+    """Base class for single agent 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)
+TEvent = TypeVar("TEvent", bound=BaseHookEvent, contravariant=True)
 """Generic for adding callback handlers - contravariant to allow adding handlers which take in base classes."""
 
-TInvokeEvent = TypeVar("TInvokeEvent", bound=HookEventBase)
+TInvokeEvent = TypeVar("TInvokeEvent", bound=BaseHookEvent)
 """Generic for invoking events - non-contravariant to enable returning events."""