Skip to content

Commit ad63880

Browse files
committed
update
1 parent d4719ca commit ad63880

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

libs/agno/tests/integration/teams/test_hooks.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,7 @@ def __init__(self, model_id: str, model_response_content: Optional[str] = None):
136136
self._mock_response.updated_session_state = None
137137
# Set event to assistant_response by default (matches ModelResponse default)
138138
from agno.models.response import ModelResponseEvent
139+
139140
self._mock_response.event = ModelResponseEvent.assistant_response.value
140141

141142
# Create Mock objects for response methods
@@ -205,7 +206,8 @@ def create_test_team(pre_hooks=None, post_hooks=None, model_response_content=Non
205206

206207
# Create a test model that inherits from Model
207208
mock_model = MockTestModel(
208-
model_id="test-team-model", model_response_content=model_response_content or "Test team response from mock model"
209+
model_id="test-team-model",
210+
model_response_content=model_response_content or "Test team response from mock model",
209211
)
210212

211213
return Team(

0 commit comments

Comments
 (0)