@elizaos/sweagent-root 2.0.0-alpha

package/python/sweagent/inspector/style.css

@@ -0,0 +1,454 @@
+body {
+  font-family: Arial, sans-serif;
+  margin: 0;
+  padding: 20px;
+  background-color: #1e1e1e;
+  color: #d4d4d4;
+}
+h1,
+h2 {
+  color: #d4d4d4;
+}
+#fileList {
+  list-style-type: none;
+  padding: 10px;
+  max-height: 400px;
+  overflow-y: auto;
+  margin: 0;
+  background-color: #252526;
+  border: 3px solid #1697e2;
+  border-radius: 10px;
+}
+#fileList li {
+  cursor: pointer;
+  padding: 10px;
+  background-color: #2d2d2d;
+  margin-bottom: 5px;
+  border: 2px solid #454545;
+  border-radius: 5px;
+  transition: background-color 0.3s;
+  color: #d4d4d4;
+}
+#fileList li:hover {
+  background-color: #3e3e3e;
+}
+#fileList li.selected {
+  border-color: #fabb00;
+}
+#fileContent {
+  background-color: #1e1e1e;
+  border: 1px solid #454545;
+  padding: 10px;
+  margin-top: 20px;
+  white-space: pre-wrap;
+  color: #d4d4d4;
+}
+.button-container {
+  display: flex;
+  justify-content: center;
+  align-items: center;
+}
+#refreshButton {
+  padding: 4px 10px;
+  min-width: 80px;
+  border: none;
+  font: inherit;
+  color: #373030;
+  border-radius: 10px;
+  outline: none;
+  text-decoration: none;
+  cursor: default;
+  font-weight: 400;
+  background: #fff;
+  box-shadow:
+    0px 0px 1px #0000004d,
+    0px 1px 1px #00000066;
+}
+#refreshButton:hover {
+  /* hover MUST come before active */
+  background: linear-gradient(#0000004d, #00000066);
+  color: #fff;
+  position: relative;
+}
+#refreshButton:active {
+  background: linear-gradient(#4faefc, #006bff);
+  color: #fff;
+  position: relative;
+}
+.history-item {
+  border: 3px solid black;
+  border-radius: 5px;
+  padding: 0px;
+  /* padding-bottom: 5%; */
+  margin-bottom: 15px;
+  overflow-x: hidden;
+  white-space: normal;
+  /* overflow-x: auto; Enables horizontal scrolling */
+  /* white-space: nowrap; Keeps content in a single line */
+  max-height: 450px; /* Adjust as needed for 25 lines */
+  overflow: hidden;
+  position: relative;
+}
+.shadow {
+  height: 30px; /* Height of the shadow */
+  background: linear-gradient(to bottom, transparent, rgba(0, 0, 0, 0.4));
+  position: absolute;
+  bottom: 0;
+  left: 0;
+  right: 0;
+  pointer-events: none; /* Ensures the shadow doesn't interfere with interaction */
+  display: none; /* Initially hidden */
+}
+.has-shadow .shadow {
+  display: block;
+}
+.content-container {
+  max-height: 400px; /* Adjust as needed */
+  overflow-y: auto;
+  position: relative;
+  padding: 10px;
+}
+.content-container pre {
+  white-space: pre-wrap; /* Wrap lines and preserve whitespace */
+  overflow-wrap: break-word; /* Handle long words */
+}
+.container {
+  max-width: 1000px;
+  margin: 0 auto; /* Centers the container */
+  padding: 20px; /* Optional: for some inner spacing */
+}
+.history-item.user {
+  border-color: #1697e2;
+}
+.history-item.tool {
+  border-color: #1483c3;
+}
+.history-item.system {
+  border-color: #004b80;
+}
+.history-item.subroutine {
+  border-color: #006b00;
+}
+.history-item.gold-patch {
+  border-color: #fabb00;
+}
+.history-item.assistant {
+  border-color: rgb(0, 0, 0);
+}
+.history-item.test-patch {
+  border-color: #7373d9;
+}
+.history-item.evaluation-report {
+  border-color: #35614b;
+}
+
+/* filepath-tree stuff */
+.filepath {
+  display: flex;
+  flex-direction: column; /* Changes layout to one part per line */
+  align-items: flex-start; /* Aligns parts to the start of the container */
+  font-size: 16px;
+  gap: 10px;
+  padding: 5px;
+  background-color: #2d2d2d;
+}
+.part {
+  border: 1px solid #454545;
+  white-space: nowrap; /* Prevents wrapping within parts */
+  padding: 5px;
+  background-color: #3e3e3e;
+  border-radius: 5px;
+  color: #d4d4d4;
+}
+
+@keyframes fadeIn {
+  from {
+    opacity: 0;
+  }
+  to {
+    opacity: 1;
+  }
+}
+
+.fade-in {
+  animation: fadeIn 1s ease-out;
+}
+
+.trajectory-container {
+  position: relative;
+  width: 100%;
+}
+
+.trajectory-item {
+  display: flex;
+  flex-direction: column;
+  margin-bottom: 0px;
+  border: 3px solid #1697e2;
+  border-radius: 5px;
+  position: relative;
+  background: #2d2d2d;
+  min-height: fit-content; /* Added to ensure proper expansion */
+  height: auto; /* Added to ensure proper expansion */
+  overflow: visible; /* Changed from hidden/auto to visible */
+}
+
+.trajectory-main {
+  flex: 1;
+  min-width: 0;
+  display: flex;
+  flex-direction: column;
+  gap: 0;
+}
+
+.response-section,
+.observation-section,
+.messages-section {
+  padding: 0 4px;
+  position: relative;
+  min-height: 30px;
+  height: 400px;
+  max-height: min-content;
+  display: flex;
+  flex-direction: column;
+  resize: vertical;
+  overflow: hidden;
+  width: calc(100% - 16px);
+}
+
+.response-section {
+  background-color: #252526;
+  border-bottom: 1px solid #454545;
+  color: #d4d4d4;
+}
+
+.observation-section {
+  background-color: #2d2d2d;
+  color: #d4d4d4;
+}
+
+.observation-images-section {
+  padding: 4px;
+  position: relative;
+  min-height: 30px;
+  height: 200px;
+  max-height: 200px;
+  background: #2d2d30;
+  border-top: 1px solid #454545;
+  display: flex;
+  flex-direction: column;
+  resize: vertical;
+  overflow: hidden;
+  width: calc(100% - 16px);
+  color: #d4d4d4;
+}
+
+/* Add section headers */
+.response-section::before,
+.observation-section::before,
+.observation-images-section::before,
+.messages-section::before {
+  content: attr(data-title);
+  font-weight: bold;
+  padding: 8px 12px; /* Keep this comfortable padding */
+  background: rgba(255, 255, 255, 0.05);
+  margin: 0 0 8px 0; /* Add some bottom margin to separate from content */
+  border-bottom: 1px solid #454545;
+  position: sticky;
+  top: 0;
+  width: 100%;
+  box-sizing: border-box;
+  color: #d4d4d4;
+}
+
+/* Scrollable content containers */
+.content-wrapper {
+  overflow-y: auto;
+  flex: 1;
+  padding: 0;
+  margin: 0;
+  display: flex; /* Add flex display */
+  flex-direction: column; /* Stack children vertically */
+}
+
+.content-wrapper pre {
+  margin: 0;
+  padding: 0;
+  flex: 1; /* Allow pre to fill the space */
+  display: flex; /* Make pre a flex container */
+  flex-direction: column; /* Stack children vertically */
+}
+
+.content-wrapper pre code {
+  margin: 0;
+  padding: 0;
+  flex: 1; /* Allow code to fill the space */
+}
+
+.messages-section {
+  padding: 4px;
+  position: relative;
+  min-height: 30px;
+  height: 30px;
+  max-height: min-content; /* Only expand to fit content */
+  background: #252526;
+  border-top: 1px solid #454545;
+  display: flex;
+  flex-direction: column;
+  resize: vertical;
+  overflow: hidden;
+  width: calc(100% - 16px);
+  color: #d4d4d4;
+}
+
+.messages-section::before {
+  content: attr(data-title);
+  font-weight: bold;
+  padding: 8px 12px; /* Increased padding around title text */
+  background: rgba(255, 255, 255, 0.05);
+  margin: -8px -8px 5px -8px; /* Reduced bottom margin */
+  border-bottom: 1px solid #454545;
+  position: sticky;
+  top: -8px;
+  width: calc(100% + 16px);
+  box-sizing: border-box;
+  color: #d4d4d4;
+}
+
+.messages-toggle {
+  display: none;
+}
+
+.execution-time {
+  font-size: 0.8em;
+  color: #666;
+  text-align: right;
+  padding: 5px;
+  border-top: 1px solid #ddd;
+}
+
+/* Add a visual indicator for resizable areas */
+.response-section:hover,
+.observation-section:hover,
+.observation-images-section:hover,
+.messages-section.expanded:hover {
+  outline: 1px dashed #999;
+}
+
+.demo-message {
+  color: #2ecc71;
+  font-weight: bold;
+}
+
+/* Syntax highlighting overrides for dark theme */
+.hljs {
+  background: #1e1e1e !important;
+  color: #d4d4d4 !important;
+}
+
+.hljs-keyword,
+.hljs-selector-tag,
+.hljs-built_in,
+.hljs-name,
+.hljs-tag {
+  color: #569cd6 !important; /* Blue */
+}
+
+.hljs-string,
+.hljs-title,
+.hljs-section,
+.hljs-attribute,
+.hljs-literal,
+.hljs-template-tag,
+.hljs-template-variable,
+.hljs-type {
+  color: #ce9178 !important; /* Orange/Pink */
+}
+
+.hljs-comment,
+.hljs-quote {
+  color: #6a9955 !important; /* Green */
+}
+
+.hljs-number,
+.hljs-regexp,
+.hljs-symbol,
+.hljs-variable,
+.hljs-template-variable,
+.hljs-link,
+.hljs-selector-attr,
+.hljs-selector-pseudo {
+  color: #b5cea8 !important; /* Light green */
+}
+
+label {
+  color: #d4d4d4; /* Matches the theme's text color */
+  display: flex;
+  align-items: center;
+  gap: 5px;
+}
+
+input[type="checkbox"] {
+  margin: 0;
+  cursor: pointer;
+}
+
+/* observation images styles */
+.observation-images {
+  margin-top: 10px;
+  padding: 10px 0;
+}
+
+.observation-image-container {
+  margin-bottom: 15px;
+  text-align: center;
+}
+
+.observation-image {
+  max-width: 100%;
+  max-height: 400px;
+  border: 1px solid #454545;
+  border-radius: 5px;
+  background-color: #1e1e1e;
+  box-shadow: 0 2px 8px rgba(0, 0, 0, 0.3);
+  cursor: pointer;
+  transition: transform 0.2s ease;
+}
+
+.observation-image:hover {
+  transform: scale(1.02);
+  border-color: #1697e2;
+}
+
+.image-caption {
+  margin-top: 5px;
+  font-size: 0.9em;
+  color: #b3b3b3;
+  font-style: italic;
+}
+
+.observation-image.expanded {
+  position: fixed;
+  top: 50%;
+  left: 50%;
+  transform: translate(-50%, -50%) scale(1);
+  max-width: 90vw;
+  max-height: 90vh;
+  z-index: 1000;
+  border: 2px solid #1697e2;
+  box-shadow: 0 0 20px rgba(0, 0, 0, 0.8);
+}
+
+.image-overlay {
+  position: fixed;
+  top: 0;
+  left: 0;
+  width: 100%;
+  height: 100%;
+  background-color: rgba(0, 0, 0, 0.8);
+  z-index: 999;
+  display: none;
+}
+
+.image-overlay.active {
+  display: block;
+}

package/python/sweagent/run/_progress.py

@@ -0,0 +1,158 @@
+"""This module contains an auxiliary class for rendering progress of the batch run."""
+
+import collections
+from pathlib import Path
+from threading import Lock
+
+import yaml
+from rich.console import Group
+from rich.progress import (
+    BarColumn,
+    MofNCompleteColumn,
+    Progress,
+    SpinnerColumn,
+    TaskID,
+    TaskProgressColumn,
+    TextColumn,
+    TimeElapsedColumn,
+    TimeRemainingColumn,
+)
+from rich.table import Table
+
+from sweagent.agent.models import GLOBAL_STATS
+
+
+def _shorten_str(s: str, max_len: int, shorten_left=False) -> str:
+    if not shorten_left:
+        s = s[: max_len - 3] + "..." if len(s) > max_len else s
+    else:
+        s = "..." + s[-max_len + 3 :] if len(s) > max_len else s
+    return f"{s:<{max_len}}"
+
+
+class RunBatchProgressManager:
+    def __init__(
+        self,
+        num_instances: int,
+        yaml_report_path: Path | None = None,
+    ):
+        """This class manages a progress bar/UI for run-batch
+
+        Args:
+            num_instances: Number of task instances
+            yaml_report_path: Path to save a yaml report of the instances and their exit statuses
+        """
+
+        self._spinner_tasks: dict[str, TaskID] = {}
+        """We need to map instance ID to the task ID that is used by the rich progress bar."""
+
+        self._lock = Lock()
+
+        self._instances_by_exit_status = collections.defaultdict(list)
+        self._main_progress_bar = Progress(
+            SpinnerColumn(spinner_name="dots2"),
+            TextColumn("[progress.description]{task.description} (${task.fields[total_cost]})"),
+            BarColumn(),
+            MofNCompleteColumn(),
+            TaskProgressColumn(),
+            TimeElapsedColumn(),
+            TextColumn("[cyan]eta:[/cyan]"),
+            TimeRemainingColumn(),
+            # Wait 5 min before estimating speed
+            speed_estimate_period=60 * 5,
+        )
+        self._task_progress_bar = Progress(
+            SpinnerColumn(spinner_name="dots2"),
+            TextColumn("{task.fields[instance_id]}"),
+            TextColumn("{task.fields[status]}"),
+            TimeElapsedColumn(),
+        )
+        """Task progress bar for individual instances. There's only one progress bar
+        with one task for each instance.
+        """
+
+        self._main_task_id = self._main_progress_bar.add_task(
+            "[cyan]Overall Progress", total=num_instances, total_cost=0
+        )
+
+        self.render_group = Group(Table(), self._task_progress_bar, self._main_progress_bar)
+        self._yaml_report_path = yaml_report_path
+
+    @property
+    def n_completed(self) -> int:
+        return sum(len(instances) for instances in self._instances_by_exit_status.values())
+
+    def update_exit_status_table(self):
+        # We cannot update the existing table, so we need to create a new one and
+        # assign it back to the render group.
+        t = Table()
+        t.add_column("Exit Status")
+        t.add_column("Count", justify="right", style="bold cyan")
+        t.add_column("Most recent instances")
+        t.show_header = False
+        with self._lock:
+            t.show_header = True
+            # Sort by number of instances in descending order
+            sorted_items = sorted(self._instances_by_exit_status.items(), key=lambda x: len(x[1]), reverse=True)
+            for status, instances in sorted_items:
+                instances_str = _shorten_str(", ".join(reversed(instances)), 55)
+                t.add_row(status, str(len(instances)), instances_str)
+        assert self.render_group is not None
+        self.render_group.renderables[0] = t
+
+    def _update_total_costs(self) -> None:
+        with self._lock:
+            self._main_progress_bar.update(self._main_task_id, total_cost=f"{GLOBAL_STATS.total_cost:.2f}")
+
+    def update_instance_status(self, instance_id: str, message: str):
+        assert self._task_progress_bar is not None
+        assert self._main_progress_bar is not None
+        with self._lock:
+            self._task_progress_bar.update(
+                self._spinner_tasks[instance_id],
+                status=_shorten_str(message, 30),
+                instance_id=_shorten_str(instance_id, 25, shorten_left=True),
+            )
+        self._update_total_costs()
+
+    def on_instance_start(self, instance_id: str):
+        with self._lock:
+            self._spinner_tasks[instance_id] = self._task_progress_bar.add_task(
+                description=f"Task {instance_id}",
+                status="Task initialized",
+                total=None,
+                instance_id=instance_id,
+            )
+
+    def on_instance_end(self, instance_id: str, exit_status: str | None) -> None:
+        self._instances_by_exit_status[exit_status].append(instance_id)
+        with self._lock:
+            self._task_progress_bar.remove_task(self._spinner_tasks[instance_id])
+            self._main_progress_bar.update(TaskID(0), advance=1)
+        self.update_exit_status_table()
+        self._update_total_costs()
+        if self._yaml_report_path is not None:
+            self._save_overview_data_yaml(self._yaml_report_path)
+
+    def on_uncaught_exception(self, instance_id: str, exception: Exception) -> None:
+        self.on_instance_end(instance_id, f"Uncaught {type(exception).__name__}")
+
+    def print_report(self) -> None:
+        """Print complete list of instances and their exit statuses."""
+        for status, instances in self._instances_by_exit_status.items():
+            print(f"{status}: {len(instances)}")
+            for instance in instances:
+                print(f"  {instance}")
+
+    def _get_overview_data(self) -> dict:
+        """Get data like exit statuses, total costs, etc."""
+        return {
+            # convert defaultdict to dict because of serialization
+            "instances_by_exit_status": dict(self._instances_by_exit_status),
+            "total_cost": GLOBAL_STATS.total_cost,
+        }
+
+    def _save_overview_data_yaml(self, path: Path) -> None:
+        """Save a yaml report of the instances and their exit statuses."""
+        with self._lock:
+            path.write_text(yaml.dump(self._get_overview_data(), indent=4))