diff --git a/vision_agent/agent/agent.py b/vision_agent/agent/agent.py
index 05421e41..5054f170 100644
--- a/vision_agent/agent/agent.py
+++ b/vision_agent/agent/agent.py
@@ -6,6 +6,8 @@
 class Agent(ABC):
     @abstractmethod
     def __call__(
-        self, input: Union[List[Dict[str, str]], str], image: Optional[Union[str, Path]] = None
+        self,
+        input: Union[List[Dict[str, str]], str],
+        image: Optional[Union[str, Path]] = None,
     ) -> str:
         pass