From aae2481070877f9844a165c2363c9cf4ec56d980 Mon Sep 17 00:00:00 2001 From: shankar-landing-ai Date: Wed, 29 May 2024 13:36:07 -0700 Subject: [PATCH] returning dictionary instead of code --- vision_agent/agent/vision_agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index 355fc214..d5e65169 100644 --- a/vision_agent/agent/vision_agent.py +++ b/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,