add VERBOSITY env var

landing-ai · Oct 10, 2024 · 7f4fc3b · 7f4fc3b
1 parent 1e786ff
commit 7f4fc3b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/vision_agent/tools/meta_tools.py b/vision_agent/tools/meta_tools.py
@@ -28,6 +28,7 @@
 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:
@@ -375,7 +376,7 @@ def detect_dogs(image_path: str):
  )
  )
  else:
- agent = va.agent.VisionAgentCoder()
+ agent = va.agent.VisionAgentCoder(verbosity=int(VERBOSITY))
 
  fixed_chat: List[Message] = [{"role": "user", "content": chat, "media": media}]
  response = agent.chat_with_workflow(
@@ -438,7 +439,7 @@ def detect_dogs(image_path: str):
  return dogs
  """
 
- agent = va.agent.VisionAgentCoder()
+ agent = va.agent.VisionAgentCoder(verbosity=int(VERBOSITY))
  if name not in artifacts:
  print(f"[Artifact {name} does not exist]")
  return f"[Artifact {name} does not exist]"