11"""Tests for accepting callable hook callbacks in Agent constructor's hooks parameter."""
22
3- from unittest .mock import MagicMock
4-
53import pytest
64
75from strands import Agent
@@ -46,7 +44,7 @@ def on_init(self, event: AgentInitializedEvent) -> None:
4644 self .events .append (event )
4745
4846 provider = MyProvider ()
49- agent = Agent (hooks = [provider ], callback_handler = None )
47+ Agent (hooks = [provider ], callback_handler = None )
5048
5149 assert len (provider .events ) == 1
5250 assert isinstance (provider .events [0 ], AgentInitializedEvent )
@@ -81,9 +79,7 @@ def on_before(event: BeforeInvocationEvent) -> None:
8179 def on_after (event : AfterInvocationEvent ) -> None :
8280 after_events .append (event )
8381
84- mock_model = MockedModelProvider (
85- [{"role" : "assistant" , "content" : [{"text" : "Hello!" }]}]
86- )
82+ mock_model = MockedModelProvider ([{"role" : "assistant" , "content" : [{"text" : "Hello!" }]}])
8783
8884 agent = Agent (
8985 model = mock_model ,
@@ -119,9 +115,7 @@ def test_hooks_param_callable_with_explicit_type_hint(self):
119115 def on_model_call (event : BeforeModelCallEvent ) -> None :
120116 model_call_events .append (event )
121117
122- mock_model = MockedModelProvider (
123- [{"role" : "assistant" , "content" : [{"text" : "result" }]}]
124- )
118+ mock_model = MockedModelProvider ([{"role" : "assistant" , "content" : [{"text" : "result" }]}])
125119
126120 agent = Agent (
127121 model = mock_model ,
@@ -152,7 +146,7 @@ def callback_a(event: AgentInitializedEvent) -> None:
152146 def callback_b (event : AgentInitializedEvent ) -> None :
153147 events_b .append (event )
154148
155- agent = Agent (hooks = [callback_a , callback_b ], callback_handler = None )
149+ Agent (hooks = [callback_a , callback_b ], callback_handler = None )
156150
157151 assert len (events_a ) == 1
158152 assert len (events_b ) == 1
@@ -168,9 +162,7 @@ def test_hooks_param_accepts_async_before_invocation_callback(self):
168162 async def my_async_callback (event : BeforeInvocationEvent ) -> None :
169163 events_received .append (event )
170164
171- mock_model = MockedModelProvider (
172- [{"role" : "assistant" , "content" : [{"text" : "Hello!" }]}]
173- )
165+ mock_model = MockedModelProvider ([{"role" : "assistant" , "content" : [{"text" : "Hello!" }]}])
174166
175167 agent = Agent (
176168 model = mock_model ,
@@ -188,5 +180,7 @@ def test_hooks_param_rejects_async_agent_initialized_callback(self):
188180 async def my_async_callback (event : AgentInitializedEvent ) -> None :
189181 pass
190182
191- with pytest .raises (ValueError , match = "AgentInitializedEvent can only be registered with a synchronous callback" ):
183+ with pytest .raises (
184+ ValueError , match = "AgentInitializedEvent can only be registered with a synchronous callback"
185+ ):
192186 Agent (hooks = [my_async_callback ], callback_handler = None )
0 commit comments