From 0247de107c0772f154a626b7b290749d1be38cdb Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Tue, 15 Oct 2024 17:15:40 -0700 Subject: [PATCH] remove verbosity arg --- vision_agent/tools/meta_tools.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py index fd3f306e..b7352583 100644 --- a/vision_agent/tools/meta_tools.py +++ b/vision_agent/tools/meta_tools.py @@ -28,7 +28,6 @@ CURRENT_LINE = 0 DEFAULT_WINDOW_SIZE = 100 ZMQ_PORT = os.environ.get("ZMQ_PORT", None) -VERBOSITY = os.environ.get("VERBOSITY", 0) def report_progress_callback(port: int, inp: Dict[str, Any]) -> None: @@ -443,14 +442,16 @@ def detect_dogs(image_path: str): dogs = owl_v2("dog", image) return dogs """ + # verbosity is set to 0 to avoid adding extra content to the VisionAgent conversation if ZMQ_PORT is not None: agent = va.agent.VisionAgentCoder( report_progress_callback=lambda inp: report_progress_callback( int(ZMQ_PORT), inp - ) + ), + verbosity=0, ) else: - agent = va.agent.VisionAgentCoder(verbosity=int(VERBOSITY)) + agent = va.agent.VisionAgentCoder(verbosity=0) fixed_chat: List[Message] = [{"role": "user", "content": chat, "media": media}] response = agent.generate_code( @@ -514,7 +515,8 @@ def detect_dogs(image_path: str): return dogs """ - agent = va.agent.VisionAgentCoder(verbosity=int(VERBOSITY)) + # verbosity is set to 0 to avoid adding extra content to the VisionAgent conversation + agent = va.agent.VisionAgentCoder(verbosity=0) if name not in artifacts: print(f"[Artifact {name} does not exist]") return f"[Artifact {name} does not exist]"