returning dictionary instead of code

vision_agent/agent/vision_agent.py

@@ -329,7 +329,7 @@ def __call__(
  input = [{"role": "user", "content": input}]
  results = self.chat_with_workflow(input, media)
  results.pop("working_memory")
- return results["code"] # type: ignore
+ return results # type: ignore
 
  def chat_with_workflow(
  self,