From 4e0bfd996727277a5c31897bb58b0fd69809e25e Mon Sep 17 00:00:00 2001 From: wuyiqunLu Date: Wed, 11 Sep 2024 00:17:51 +0800 Subject: [PATCH] fix lint --- vision_agent/agent/vision_agent.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index d3450088..cc8b4f53 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -272,7 +272,7 @@ def chat_with_code( artifacts_loaded = artifacts.show() int_chat.append({"role": "observation", "content": artifacts_loaded}) orig_chat.append({"role": "observation", "content": artifacts_loaded}) - self.callback_message({"role": "observation", "content": artifacts_loaded}) + self.streaming_message({"role": "observation", "content": artifacts_loaded}) while not finished and iterations < self.max_iterations: response = run_conversation(self.agent, int_chat) @@ -285,7 +285,7 @@ def chat_with_code( if last_response == response: response["let_user_respond"] = True - self.callback_message({"role": "assistant", "content": response}) + self.streaming_message({"role": "assistant", "content": response}) if response["let_user_respond"]: break @@ -306,7 +306,7 @@ def chat_with_code( orig_chat.append( {"role": "observation", "content": obs, "execution": result} ) - self.callback_message( + self.streaming_message( { "role": "observation", "content": obs,