From e52ddd3e5eb606112608f4750f5e52ed79fed700 Mon Sep 17 00:00:00 2001 From: Steve Androulakis Date: Sun, 1 Jun 2025 08:58:19 -0700 Subject: [PATCH] straggler linted file --- tests/test_tool_activities.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/tests/test_tool_activities.py b/tests/test_tool_activities.py index 39fe636..d74d0b6 100644 --- a/tests/test_tool_activities.py +++ b/tests/test_tool_activities.py @@ -98,7 +98,9 @@ class TestToolActivities: # Mock the completion function mock_response = MagicMock() mock_response.choices = [MagicMock()] - mock_response.choices[0].message.content = ( + mock_response.choices[ + 0 + ].message.content = ( '{"next": "confirm", "tool": "TestTool", "response": "Test response"}' ) @@ -136,9 +138,9 @@ class TestToolActivities: mock_response = MagicMock() mock_response.choices = [MagicMock()] - mock_response.choices[0].message.content = ( - '{"next": "done", "response": "Test"}' - ) + mock_response.choices[ + 0 + ].message.content = '{"next": "done", "response": "Test"}' with patch("activities.tool_activities.completion") as mock_completion: mock_completion.return_value = mock_response @@ -362,9 +364,9 @@ class TestEdgeCases: # Mock the completion response mock_response = MagicMock() mock_response.choices = [MagicMock()] - mock_response.choices[0].message.content = ( - '{"next": "done", "response": "Processed long prompt"}' - ) + mock_response.choices[ + 0 + ].message.content = '{"next": "done", "response": "Processed long prompt"}' with patch("activities.tool_activities.completion", return_value=mock_response): activity_env = ActivityEnvironment()