last try on fixing mypy errors

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")
 
 

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,