diff --git a/vision_agent/agent/vision_agent_v2.py b/vision_agent/agent/vision_agent_v2.py
index a83f8b27..d7bf1372 100644
--- a/vision_agent/agent/vision_agent_v2.py
+++ b/vision_agent/agent/vision_agent_v2.py
@@ -39,7 +39,7 @@ def build_working_memory(working_memory: Mapping[str, List[str]]) -> Sim:
     for key, value in working_memory.items():
         data["desc"].append(key)
         data["doc"].append("\n".join(value))
-    df = pd.DataFrame(data)
+    df = pd.DataFrame(data)  # type: ignore
     return Sim(df, sim_key="desc")
 
 
diff --git a/vision_agent/agent/vision_agent_v3.py b/vision_agent/agent/vision_agent_v3.py
index a5484992..2f7edfad 100644
--- a/vision_agent/agent/vision_agent_v3.py
+++ b/vision_agent/agent/vision_agent_v3.py
@@ -3,7 +3,7 @@
 import logging
 import sys
 from pathlib import Path
-from typing import Any, Dict, List, Optional, Union, cast, Callable
+from typing import Any, Dict, List, Optional, Union, cast, Callable, no_type_check
 
 from rich.console import Console
 from rich.syntax import Syntax
@@ -273,16 +273,17 @@ def __init__(
         self.max_retries = 2
         self.report_progress_callback = report_progress_callback
 
+    @no_type_check
     def __call__(
         self,
         input: Union[List[Dict[str, str]], str],
         image: Optional[Union[str, Path]] = None,
-    ) -> Dict[str, Any]:  # type: ignore
+    ) -> Dict[str, Any]:
         if isinstance(input, str):
             input = [{"role": "user", "content": input}]
         results = self.chat_with_workflow(input, image)
         results.pop("working_memory")
-        return results  # type: ignore
+        return results
 
     def chat_with_workflow(
         self,