fix mypy errors

landing-ai · May 29, 2024 · ad1bc52 · ad1bc52
1 parent b770584
commit ad1bc52
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/vision_agent/agent/agent_coder.py b/vision_agent/agent/agent_coder.py
@@ -67,7 +67,7 @@ def parse_file_name(s: str) -> str:
  return "".join([p for p in s.split(" ") if p.endswith(".png")])
 
 
-def write_program(question: str, feedback: str, model: LLM, image=None) -> str:
+def write_program(question: str, feedback: str, model: LLM, image: Optional[Union[str, Path]]=None) -> str:
  prompt = PROGRAM.format(
  docstring=TOOL_DOCSTRING, question=question, feedback=feedback
  )

diff --git a/vision_agent/agent/vision_agent_v3.py b/vision_agent/agent/vision_agent_v3.py
@@ -315,7 +315,7 @@ def chat_with_workflow(
 
  while not success and retries < self.max_retries:
  plan_i = write_plan(
- chat, TOOL_DESCRIPTIONS, format_memory(working_memory), self.planner, images=[image]
+ chat, TOOL_DESCRIPTIONS, format_memory(working_memory), self.planner, images=[image] if image else None
  )
  plan_i_str = "\n-".join([e["instructions"] for e in plan_i])
  if self.verbosity >= 1: