From 75aba6f51a9f8a75e6c5efd08fd3ebc5c38f9aa3 Mon Sep 17 00:00:00 2001 From: wuyiqunLu Date: Tue, 24 Sep 2024 22:25:08 +0800 Subject: [PATCH] remove extra code --- vision_agent/agent/vision_agent.py | 19 +++++-------------- 1 file changed, 5 insertions(+), 14 deletions(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index b54d08b8..bf35e5e9 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -30,12 +30,6 @@ if str(WORKSPACE) != "": os.environ["PYTHONPATH"] = f"{WORKSPACE}:{os.getenv('PYTHONPATH', '')}" -STUCK_IN_LOOP_ERROR_MESSAGE = { - "name": "Error when running conversation agent", - "value": "Agent is stuck in conversation loop, exited", - "traceback_raw": [], -} - class BoilerplateCode: pre_code = [ @@ -298,13 +292,6 @@ def chat_with_code( # sometimes it gets stuck in a loop, so we force it to exit if last_response == response: response["let_user_respond"] = True - self.streaming_message( - { - "role": "assistant", - "content": "{}", - "error": STUCK_IN_LOOP_ERROR_MESSAGE, - } - ) finished = response["let_user_respond"] @@ -317,7 +304,11 @@ def chat_with_code( { "role": "assistant", "content": "{}", - "error": STUCK_IN_LOOP_ERROR_MESSAGE, + "error": { + "name": "Error when running conversation agent", + "value": "Agent is stuck in conversation loop, exited", + "traceback_raw": [], + }, "finished": finished and code_action is None, } )