diff --git a/src/crewai/agents/crew_agent_executor.py b/src/crewai/agents/crew_agent_executor.py index 813ac8a08..7ced433cb 100644 --- a/src/crewai/agents/crew_agent_executor.py +++ b/src/crewai/agents/crew_agent_executor.py @@ -358,9 +358,9 @@ class CrewAgentExecutor(CrewAgentExecutorMixin): if self.crew is not None and hasattr(self.crew, "_train_iteration"): train_iteration = self.crew._train_iteration if agent_id in training_data and isinstance(train_iteration, int): - training_data[agent_id][train_iteration][ - "improved_output" - ] = result.output + training_data[agent_id][train_iteration]["improved_output"] = ( + result.output + ) training_handler.save(training_data) else: self._printer.print( diff --git a/src/crewai/cli/tools/main.py b/src/crewai/cli/tools/main.py index b2d2cec99..23aeae2f2 100644 --- a/src/crewai/cli/tools/main.py +++ b/src/crewai/cli/tools/main.py @@ -153,8 +153,12 @@ class ToolCommand(BaseCommand, PlusAPIMixin): login_response_json = login_response.json() settings = Settings() - settings.tool_repository_username = login_response_json["credential"]["username"] - settings.tool_repository_password = login_response_json["credential"]["password"] + settings.tool_repository_username = login_response_json["credential"][ + "username" + ] + settings.tool_repository_password = login_response_json["credential"][ + "password" + ] settings.dump() console.print( @@ -179,7 +183,7 @@ class ToolCommand(BaseCommand, PlusAPIMixin): capture_output=False, env=self._build_env_with_credentials(repository_handle), text=True, - check=True + check=True, ) if add_package_result.stderr: @@ -204,7 +208,11 @@ class ToolCommand(BaseCommand, PlusAPIMixin): settings = Settings() env = os.environ.copy() - env[f"UV_INDEX_{repository_handle}_USERNAME"] = str(settings.tool_repository_username or "") - env[f"UV_INDEX_{repository_handle}_PASSWORD"] = str(settings.tool_repository_password or "") + env[f"UV_INDEX_{repository_handle}_USERNAME"] = str( + settings.tool_repository_username or "" + ) + env[f"UV_INDEX_{repository_handle}_PASSWORD"] = str( + settings.tool_repository_password or "" + ) return env diff --git a/src/crewai/project/annotations.py b/src/crewai/project/annotations.py index bf0051c4d..3a806981f 100644 --- a/src/crewai/project/annotations.py +++ b/src/crewai/project/annotations.py @@ -66,7 +66,6 @@ def cache_handler(func): def crew(func) -> Callable[..., Crew]: - @wraps(func) def wrapper(self, *args, **kwargs) -> Crew: instantiated_tasks = [] diff --git a/src/crewai/project/crew_base.py b/src/crewai/project/crew_base.py index 0b43882f2..9dead8a8a 100644 --- a/src/crewai/project/crew_base.py +++ b/src/crewai/project/crew_base.py @@ -216,5 +216,5 @@ def CrewBase(cls: T) -> T: # Include base class (qual)name in the wrapper class (qual)name. WrappedClass.__name__ = CrewBase.__name__ + "(" + cls.__name__ + ")" WrappedClass.__qualname__ = CrewBase.__qualname__ + "(" + cls.__name__ + ")" - + return cast(T, WrappedClass) diff --git a/src/crewai/task.py b/src/crewai/task.py index 30ab79c00..cfe2b3cea 100644 --- a/src/crewai/task.py +++ b/src/crewai/task.py @@ -373,7 +373,9 @@ class Task(BaseModel): content = ( json_output if json_output - else pydantic_output.model_dump_json() if pydantic_output else result + else pydantic_output.model_dump_json() + if pydantic_output + else result ) self._save_file(content) diff --git a/tests/cli/tools/test_main.py b/tests/cli/tools/test_main.py index 10c29b920..b06c0b28c 100644 --- a/tests/cli/tools/test_main.py +++ b/tests/cli/tools/test_main.py @@ -28,9 +28,10 @@ def test_create_success(mock_subprocess): with in_temp_dir(): tool_command = ToolCommand() - with patch.object(tool_command, "login") as mock_login, patch( - "sys.stdout", new=StringIO() - ) as fake_out: + with ( + patch.object(tool_command, "login") as mock_login, + patch("sys.stdout", new=StringIO()) as fake_out, + ): tool_command.create("test-tool") output = fake_out.getvalue() @@ -82,7 +83,7 @@ def test_install_success(mock_get, mock_subprocess_run): capture_output=False, text=True, check=True, - env=unittest.mock.ANY + env=unittest.mock.ANY, ) assert "Successfully installed sample-tool" in output