fix type errors

landing-ai · May 23, 2024 · 2afce10 · 2afce10
1 parent e0c52e2
commit 2afce10
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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