From 798c625a02c55c52c9c4f68ae5ec3d1cd9e52f3a Mon Sep 17 00:00:00 2001
From: creatorrr <creatorrr@users.noreply.github.com>
Date: Thu, 26 Dec 2024 14:27:32 +0000
Subject: [PATCH] refactor: Lint agents-api (CI)

---
 agents-api/agents_api/common/protocol/tasks.py | 4 +++-
 agents-api/tests/test_execution_workflow.py    | 6 +++---
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/agents-api/agents_api/common/protocol/tasks.py b/agents-api/agents_api/common/protocol/tasks.py
index 67757925a..8226486de 100644
--- a/agents-api/agents_api/common/protocol/tasks.py
+++ b/agents-api/agents_api/common/protocol/tasks.py
@@ -239,7 +239,9 @@ def model_dump(self, *args, **kwargs) -> dict[str, Any]:
 
         return dump | execution_input
 
-    async def prepare_for_step(self, *args, include_remote=False, **kwargs) -> dict[str, Any]:
+    async def prepare_for_step(
+        self, *args, include_remote=False, **kwargs
+    ) -> dict[str, Any]:
         # FIXME: include_remote is deprecated
 
         current_input = self.current_input
diff --git a/agents-api/tests/test_execution_workflow.py b/agents-api/tests/test_execution_workflow.py
index a6b9ccf19..4419e1b59 100644
--- a/agents-api/tests/test_execution_workflow.py
+++ b/agents-api/tests/test_execution_workflow.py
@@ -12,11 +12,11 @@
 from agents_api.routers.tasks.create_task_execution import start_execution
 
 from .fixtures import (
-    test_agent,
-    test_developer_id,
-    pg_dsn,
     client,
+    pg_dsn,
     s3_client,
+    test_agent,
+    test_developer_id,
 )
 from .utils import patch_testing_temporal