Skip to content

Commit e3f9481

Browse files
fix: address hatch run prepare lint and formatting issues
- Remove unused variable assignments (F841 lint errors) - Apply auto-formatting from hatch run format
1 parent 0d197ef commit e3f9481

1 file changed

Lines changed: 8 additions & 14 deletions

File tree

tests/strands/agent/test_agent_hooks_callable.py

Lines changed: 8 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
11
"""Tests for accepting callable hook callbacks in Agent constructor's hooks parameter."""
22

3-
from unittest.mock import MagicMock
4-
53
import pytest
64

75
from 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

Comments
 (0)