@sebastianandreasson/pi-autonomous-agents 0.5.2 → 0.7.0

package/src/pi-supervisor.mjs

@@ -49,6 +49,7 @@ import {
   shouldPersistLatestTesterFeedback,
 } from './pi-flow.mjs'
 import { runStartupPreflight } from './pi-preflight.mjs'
+import { startVisualizerServer } from './pi-visualizer-server.mjs'
 
 let stopRequested = false
 let shutdownEscalationTimer = null
@@ -337,6 +338,15 @@ async function runAgentInvocation({
   sessionId,
   sessionFile,
 }) {
+  await updateRunOwnership(config, {
+    status: 'agent_running',
+    iteration,
+    phase,
+    activeKind: kind,
+    activeRole: role,
+    activeReason: reason,
+  })
+
   const beforeSnapshot = getRepoSnapshot(config.cwd)
   const resolvedModel = resolveRoleModel(config, role)
   const promptSnapshot = [
@@ -513,6 +523,15 @@ async function runHarnessGitFinalize({
   phase,
   commitPlan,
 }) {
+  await updateRunOwnership(config, {
+    status: 'git_finalize_running',
+    iteration,
+    phase,
+    activeKind: 'git_finalize',
+    activeRole: '',
+    activeReason: '',
+  })
+
   const beforeSnapshot = getRepoSnapshot(config.cwd)
   const dirtyFiles = new Set(listChangedFiles(config.cwd))
   const requestedFiles = Array.isArray(commitPlan.files) ? commitPlan.files.filter(Boolean) : []
@@ -624,6 +643,15 @@ async function runHarnessGitFinalize({
 }
 
 async function runVerificationStep({ config, iteration, phase, kind }) {
+  await updateRunOwnership(config, {
+    status: 'verification_running',
+    iteration,
+    phase,
+    activeKind: kind,
+    activeRole: '',
+    activeReason: '',
+  })
+
   const beforeSnapshot = getRepoSnapshot(config.cwd)
   const verification = await runVerification(config)
   const afterSnapshot = getRepoSnapshot(config.cwd)
@@ -988,6 +1016,15 @@ async function runTesterCommitTurn({
 }
 
 async function runVisualReview({ config, iteration, phase, task, changedFiles }) {
+  await updateRunOwnership(config, {
+    status: 'visual_capture_running',
+    iteration,
+    phase,
+    activeKind: 'visual_capture',
+    activeRole: '',
+    activeReason: '',
+  })
+
   const capture = await runVisualCapture(config, {
     iteration,
     phase,
@@ -1035,6 +1072,15 @@ async function runVisualReview({ config, iteration, phase, task, changedFiles })
     }
   }
 
+  await updateRunOwnership(config, {
+    status: 'visual_review_running',
+    iteration,
+    phase,
+    activeKind: 'visual_review',
+    activeRole: 'visualReview',
+    activeReason: '',
+  })
+
   const visualReviewModel = resolveRoleModel(config, 'visualReview')
   const reviewRequest = {
     iteration,
@@ -1196,6 +1242,9 @@ async function runIteration({ config, state, iteration }) {
     iteration,
     phase,
     task,
+    activeKind: '',
+    activeRole: '',
+    activeReason: '',
   })
   const canResumePriorSession = (
     state.lastTransport === config.transport
@@ -1576,6 +1625,9 @@ async function runIteration({ config, state, iteration }) {
     task,
     lastCompletedIteration: iteration,
     lastStatus: finalStatus,
+    activeKind: '',
+    activeRole: '',
+    activeReason: '',
   })
 
   await appendLog(
@@ -1681,11 +1733,13 @@ async function main() {
   config.runTelemetryCsv = path.join(runDir, 'pi_telemetry.csv')
   config.runStateFile = path.join(runDir, 'state.json')
   config.runLastIterationSummaryFile = path.join(runDir, 'last-iteration.json')
+  config.runLastAgentOutputFile = path.join(runDir, 'last-output.txt')
 
   ensureRepo(config.cwd)
   await ensureFileExists(config.taskFile, 'task file')
   await ensureFileExists(config.developerInstructionsFile, 'developer instructions file')
   await ensureFileExists(config.testerInstructionsFile, 'tester instructions file')
+  let visualizer = null
   const lockResult = await acquireRunLock(config.activeRunFile, {
     runId,
     pid: process.pid,
@@ -1704,6 +1758,16 @@ async function main() {
     process.env.PI_RUN_LOG_FILE = config.runLogFile
     await ensureTelemetryFiles(config)
     await appendLog(config.logFile, `Run started pid=${process.pid} mode=${config.mode}`)
+    if (config.mode === 'run' && process.env.PI_VISUALIZER !== '0' && process.env.PI_VISUALIZER !== 'false') {
+      try {
+        visualizer = await startVisualizerServer(config)
+        await appendLog(config.logFile, `Visualizer started at ${visualizer.url}`)
+        process.stderr.write(`[PI visualizer] ${visualizer.url}\n`)
+      } catch (error) {
+        await appendLog(config.logFile, `Visualizer failed to start: ${error instanceof Error ? error.message : String(error)}`)
+        process.stderr.write(`[PI visualizer] failed to start: ${error instanceof Error ? error.message : String(error)}\n`)
+      }
+    }
     if (lockResult.staleLock) {
       await appendLog(
         config.logFile,
@@ -1728,6 +1792,9 @@ async function main() {
       await updateRunOwnership(config, {
         status: 'starting_iteration',
         iteration,
+        activeKind: '',
+        activeRole: '',
+        activeReason: '',
       })
       const result = await runIteration({ config, state, iteration })
       await writeIterationSummary(config, result.iterationSummary ?? result.summary)
@@ -1754,7 +1821,13 @@ async function main() {
     await updateRunOwnership(config, {
       status: stopRequested ? 'stopped' : 'finished',
       heartbeatAt: timestamp(),
+      activeKind: '',
+      activeRole: '',
+      activeReason: '',
     })
+    if (visualizer) {
+      await visualizer.close().catch(() => {})
+    }
     await releaseRunLock(config.activeRunFile, runId)
     delete process.env.PI_RUN_ID
     delete process.env.PI_RUN_LOG_FILE

package/src/pi-telemetry.mjs

@@ -10,6 +10,10 @@ function csvEscape(value) {
 
 export async function ensureTelemetryFiles(config) {
   await fs.writeFile(config.lastAgentOutputFile, '', 'utf8')
+  if (config.runLastAgentOutputFile && config.runLastAgentOutputFile !== config.lastAgentOutputFile) {
+    await fs.mkdir(path.dirname(config.runLastAgentOutputFile), { recursive: true })
+    await fs.writeFile(config.runLastAgentOutputFile, '', 'utf8')
+  }
   await fs.writeFile(config.lastVerificationOutputFile, '', 'utf8')
   await fs.writeFile(config.changedFilesFile, '', 'utf8')
   await fs.writeFile(config.lastPromptFile, '', 'utf8')