andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-19 21:08:13 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Enhance task execution timeout handling with better error messages and resource cleanup

Browse Source
This commit is contained in:
hafsatariq18
2025-02-04 14:27:31 +05:30
parent 447fbec6f9
commit 80a5018f6a
2 changed files with 58 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/task_test.py
View File

@@ -1356,7 +1356,7 @@ def test_task_max_execution_time_zero():
role="Researcher",
goal="Test goal",
backstory="Test backstory",
max_execution_time=0 # Immediate timeout
max_execution_time=1 # Set to minimum valid value
)
task = Task(
@@ -1365,7 +1365,7 @@ def test_task_max_execution_time_zero():
agent=researcher
)
# Simulate immediate timeout
# Simulate immediate timeout using FuturesTimeoutError
with patch('concurrent.futures.ThreadPoolExecutor') as mock_executor:
mock_future = MagicMock()
mock_future.result.side_effect = FuturesTimeoutError()
@@ -1373,4 +1373,4 @@ def test_task_max_execution_time_zero():
with pytest.raises(TimeoutError) as excinfo:
task.execute_sync(agent=researcher)
assert "timed out after 0 seconds" in str(excinfo.value)
assert "timed out after 1 seconds" in str(excinfo.value)