Better format for execution outputs

landing-ai · Jun 4, 2024 · 87bb602 · 87bb602
1 parent b9fdaa5
commit 87bb602
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/vision_agent/utils/execute.py b/vision_agent/utils/execute.py
@@ -266,15 +266,17 @@ def text(self, include_logs: bool = True) -> str:
  """
  prefix = str(self.logs) if include_logs else ""
  if self.error:
- return prefix + "\n" + self.error.traceback
- return next(
+ return prefix + "\nError:" + self.error.traceback
+
+ result_str = [
  (
- prefix + "\n" + (res.text or "")
- for res in self.results
+ f"Final output:{res.text}"
  if res.is_main_result
- ),
- prefix,
- )
+ else f"Intermediate output:{res.text}"
+ )
+ for res in self.results
+ ]
+ return prefix + "\n" + "\n".join(result_str)
 
  @property
  def success(self) -> bool: