diff --git a/vision_agent/agent/agent.py b/vision_agent/agent/agent.py
index b2fccb01..10c7255f 100644
--- a/vision_agent/agent/agent.py
+++ b/vision_agent/agent/agent.py
@@ -9,7 +9,7 @@ def __call__(
         self,
         input: Union[List[Dict[str, str]], str],
         image: Optional[Union[str, Path]] = None,
-    ) -> str:
+    ) -> Union[str, Dict[str, Any]]:
         pass
 
     @abstractmethod