From 88c5a0dd0173f75e59abae185c80444a12cd31d5 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Sun, 16 Mar 2025 22:37:48 +0000 Subject: [PATCH] Fix linting and type checking issues Co-Authored-By: Joe Moura --- .../utilities/exceptions/agent_execution_timeout_error.py | 2 +- tests/test_timeout/test_agent_timeout.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/crewai/utilities/exceptions/agent_execution_timeout_error.py b/src/crewai/utilities/exceptions/agent_execution_timeout_error.py index f2c0fbba5..8435dbc6f 100644 --- a/src/crewai/utilities/exceptions/agent_execution_timeout_error.py +++ b/src/crewai/utilities/exceptions/agent_execution_timeout_error.py @@ -1,7 +1,7 @@ class AgentExecutionTimeoutError(Exception): """Exception raised when an agent execution exceeds the maximum allowed time.""" - def __init__(self, max_execution_time: int, message: str = None): + def __init__(self, max_execution_time: int, message: str | None = None): self.max_execution_time = max_execution_time self.message = message or f"Agent execution exceeded maximum allowed time of {max_execution_time} seconds" super().__init__(self.message) diff --git a/tests/test_timeout/test_agent_timeout.py b/tests/test_timeout/test_agent_timeout.py index 26cfba5ab..71ab86d9c 100644 --- a/tests/test_timeout/test_agent_timeout.py +++ b/tests/test_timeout/test_agent_timeout.py @@ -1,6 +1,7 @@ -import time import pytest -from unittest.mock import patch, MagicMock +import time +from unittest.mock import MagicMock, patch + from crewai import Agent, Task def test_agent_max_execution_time():