fix: type errors

Author: zastrowm

src/strands/hooks/registry.py

@@ -11,7 +11,7 @@
 from typing import TYPE_CHECKING, Any, Callable, Generator, Protocol, Type, TypeVar
 
 if TYPE_CHECKING:
-    from strands import Agent
+    from ..agent import Agent
 
 
 @dataclass
@@ -100,9 +100,9 @@ class HookRegistry:
     cleanup events, and provides type-safe event dispatching.
     """
 
-    def __init__(self):
+    def __init__(self) -> None:
         """Initialize an empty hook registry."""
-        self._registered_callbacks = {}
+        self._registered_callbacks: dict[Type, list[HookCallback]] = {}
 
     def add_callback(self, event_type: Type, callback: HookCallback) -> None:
         """Register a callback function for a specific event type.

tests/strands/agent/test_agent.py

@@ -6,7 +6,6 @@
 from unittest.mock import call
 
 import pytest
-from mocks.mock_hook_provider import MockHookProvider
 from pydantic import BaseModel
 
 import strands
@@ -20,6 +19,8 @@
 from strands.types.content import Messages
 from strands.types.exceptions import ContextWindowOverflowException, EventLoopException
 
+from ..mocks.mock_hook_provider import MockHookProvider
+
 
 @pytest.fixture
 def mock_randint():