fix(runtime): release 2.7.1 with consistent ADB and failure diagnostics

artemis/sdk/agent.py

@@ -654,6 +654,13 @@ class Agent:
                 # publish a new active session while the current task is still
                 # finishing.
                 self._prepare_tracing(task=task, context=context)
+                from artemis.runtime.diagnostics import capture_runtime_diagnostics
+
+                await capture_runtime_diagnostics(
+                    context.data_engine.base_dir,
+                    "start",
+                    android=request.execution_target != "computer",
+                )
                 if device_lock is not None and os.environ.get("ARTEMIS_TASK_WORKER") == "1":
                     from artemis.runtime.task_control import TaskControl
 
@@ -1333,6 +1340,14 @@ class Agent:
 
     async def _finalize_tracing(self, task: Task, context: ArtemisContext):
         if context.data_engine:
+            if task.status == "failed":
+                from artemis.runtime.diagnostics import capture_runtime_diagnostics
+
+                await capture_runtime_diagnostics(
+                    context.data_engine.base_dir,
+                    "failure",
+                    android=context.execution_target != "computer",
+                )
             session_status = "completed"
             if task.status == "failed":
                 session_status = "failed"