@sebastianandreasson/pi-autonomous-agents 0.2.0 → 0.4.0

package/src/pi-report.mjs

@@ -35,6 +35,17 @@ async function main() {
     console.log(`- ${kind}: ${count}`)
   }
 
+  const iterationSummaries = recent.filter((event) => event.kind === 'iteration_summary')
+  const warningsByIteration = iterationSummaries
+    .filter((event) => String(event.riskWarnings ?? '').trim() !== '')
+
+  if (warningsByIteration.length > 0) {
+    console.log('\nLarge file warnings:')
+    for (const event of warningsByIteration.slice(-5)) {
+      console.log(`- iteration ${event.iteration}: ${event.riskWarnings}`)
+    }
+  }
+
   const last = recent.at(-1)
   if (!last) {
     return

package/src/pi-rpc-adapter.mjs

@@ -54,6 +54,44 @@ function extractToolTarget(toolName, args) {
   return ''
 }
 
+function extractShellCommand(args) {
+  if (!args || typeof args !== 'object') {
+    return ''
+  }
+
+  if (typeof args.command === 'string') {
+    return args.command
+  }
+
+  if (typeof args.cmd === 'string') {
+    return args.cmd
+  }
+
+  return ''
+}
+
+function isLargeShellRead(command) {
+  const text = String(command ?? '').trim()
+  if (text === '') {
+    return false
+  }
+
+  if (/^\s*cat\s+\S+/.test(text)) {
+    return true
+  }
+
+  const sedMatch = text.match(/sed\s+-n\s+['"]?(\d+)\s*,\s*(\d+)p['"]?/)
+  if (sedMatch) {
+    const start = Number.parseInt(sedMatch[1], 10)
+    const end = Number.parseInt(sedMatch[2], 10)
+    if (Number.isFinite(start) && Number.isFinite(end) && end >= start) {
+      return (end - start) >= 120
+    }
+  }
+
+  return false
+}
+
 function extractAssistantText(message) {
   if (!message || message.role !== 'assistant' || !Array.isArray(message.content)) {
     return ''
@@ -121,6 +159,7 @@ async function run() {
   const pending = new Map()
   let requestCounter = 0
   const streamTerminal = request.streamTerminal === true
+  const requestedModel = typeof request.model === 'string' ? request.model : ''
   const loopRepeatThreshold = Number.isFinite(Number(request.loopRepeatThreshold))
     ? Number(request.loopRepeatThreshold)
     : 12
@@ -294,6 +333,7 @@ async function run() {
       activeToolName = String(data.toolName ?? '')
       activeToolStartedAt = Date.now()
       const target = extractToolTarget(data.toolName, data.args)
+      const shellCommand = data.toolName === 'bash' ? extractShellCommand(data.args) : ''
       if (signature === lastToolSignature) {
         repeatedToolCount += 1
       } else {
@@ -324,6 +364,9 @@ async function run() {
       }
 
       writeLive(`[PI tool:start] ${data.toolName}${suffix}\n`)
+      if (data.toolName === 'bash' && isLargeShellRead(shellCommand)) {
+        writeLive('[PI warning] large bash file read detected; prefer read or a smaller exact window to avoid truncated context.\n')
+      }
     }
 
     if (data.type === 'tool_execution_end') {
@@ -416,6 +459,9 @@ async function run() {
     await waitForAgentEnd()
 
     if (heartbeatTimedOut) {
+      const toolCalls = events.filter((event) => event.type === 'tool_execution_start').length
+      const toolErrors = events.filter((event) => event.type === 'tool_execution_end' && event.isError).length
+      const messageUpdates = events.filter((event) => event.type === 'message_update').length
       console.log(JSON.stringify({
         sessionId: request.sessionId ?? '',
         sessionFile: request.sessionFile ?? '',
@@ -438,6 +484,15 @@ async function run() {
           continueAccepted ? 'continue_accepted=true' : '',
           continueRejected ? 'continue_rejected=true' : '',
         ].join(' '),
+        role: '',
+        model: requestedModel,
+        toolCalls,
+        toolErrors,
+        messageUpdates,
+        stopReason: '',
+        loopDetected: false,
+        loopSignature: '',
+        terminalReason: 'heartbeat_timeout',
       }))
       return
     }
@@ -464,6 +519,15 @@ async function run() {
       : assistantError !== '' || (assistantText === '' && toolCalls === 0 && messageUpdates === 0)
         ? 'failed'
         : 'success'
+    const terminalReason = loopDetected
+      ? 'loop_detected'
+      : assistantError !== ''
+        ? 'assistant_error'
+        : assistantStopReason === 'length'
+          ? 'assistant_stop_length'
+          : status === 'failed'
+            ? 'empty_agent_turn'
+            : 'agent_completed'
     const notes = [
       `PI session ${state.data.sessionId}`,
       `pi_pid=${child.pid ?? 'unknown'}`,
@@ -494,6 +558,15 @@ async function run() {
       status,
       output,
       notes,
+      role: '',
+      model: requestedModel,
+      toolCalls,
+      toolErrors,
+      messageUpdates,
+      stopReason: assistantStopReason,
+      loopDetected,
+      loopSignature,
+      terminalReason,
     }))
   } finally {
     if (heartbeatInterval) {