@sebastianandreasson/pi-autonomous-agents 0.15.0 → 0.15.1

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@sebastianandreasson/pi-autonomous-agents",
   "private": false,
-  "version": "0.15.0",
+  "version": "0.15.1",
   "type": "module",
   "description": "Portable unattended PI harness for developer/tester/visual-review loops.",
   "license": "MIT",

package/pi-extensions/request-telemetry/index.mjs

@@ -118,6 +118,55 @@ function mergeSpanSummary(requestState) {
   }
 }
 
+function getSpanSignal(requestState) {
+  const spans = Array.isArray(requestState?.spans) ? requestState.spans : []
+  let toolSpanCount = 0
+
+  for (const span of spans) {
+    const kind = String(span?.spanKind ?? '').trim()
+    if (kind === 'tool_call' || kind === 'tool_result') {
+      toolSpanCount += 1
+    }
+  }
+
+  return {
+    spanCount: spans.length,
+    toolSpanCount,
+    toolNameCount: requestState?.toolNames?.size ?? 0,
+    fileCount: requestState?.files?.size ?? 0,
+  }
+}
+
+function shouldPreferSnapshot(currentState, candidateState) {
+  const current = getSpanSignal(currentState)
+  const candidate = getSpanSignal(candidateState)
+
+  const currentHasToolContext = current.toolSpanCount > 0 || current.toolNameCount > 0 || current.fileCount > 0
+  const candidateHasToolContext = candidate.toolSpanCount > 0 || candidate.toolNameCount > 0 || candidate.fileCount > 0
+
+  if (candidateHasToolContext && !currentHasToolContext) {
+    return true
+  }
+
+  if (candidate.fileCount !== current.fileCount) {
+    return candidate.fileCount > current.fileCount
+  }
+
+  if (candidate.toolSpanCount !== current.toolSpanCount) {
+    return candidate.toolSpanCount > current.toolSpanCount
+  }
+
+  if (candidate.toolNameCount !== current.toolNameCount) {
+    return candidate.toolNameCount > current.toolNameCount
+  }
+
+  if (current.spanCount === 0 && candidate.spanCount > 0) {
+    return true
+  }
+
+  return false
+}
+
 function applyAssistantMessage(requestState, message) {
   requestState.provider = String(message?.provider ?? requestState.provider ?? '').trim()
   requestState.model = String(message?.model ?? requestState.model ?? '').trim()
@@ -289,6 +338,23 @@ export function createRequestTelemetryExtension({ cwd = process.cwd() } = {}) {
     })
   }
 
+  function createComparableRequestState(baseState) {
+    return createRequestState({
+      sessionId: baseState.sessionId,
+      turnIndex: baseState.turnIndex,
+      startedAt: baseState.startedAt,
+      model: baseState.model,
+      metadata: {
+        runId: baseState.runId,
+        iteration: baseState.iteration,
+        phase: baseState.phase,
+        role: baseState.role,
+        kind: baseState.kind,
+        task: baseState.task,
+      },
+    })
+  }
+
   function createFallbackRequest() {
     const requestState = createProviderRequest()
 
@@ -417,7 +483,37 @@ export function createRequestTelemetryExtension({ cwd = process.cwd() } = {}) {
 
     pi.on('before_provider_request', (event) => {
       const requestState = createProviderRequest()
-      if (!applyProviderPayloadSnapshot(requestState, event?.payload) && !applyContextSnapshot(requestState)) {
+      const providerApplied = applyProviderPayloadSnapshot(requestState, event?.payload)
+
+      const contextCandidate = createComparableRequestState(requestState)
+      const contextApplied = applyContextSnapshot(contextCandidate)
+      if (contextApplied && shouldPreferSnapshot(requestState, contextCandidate)) {
+        requestState.contextMessages = contextCandidate.contextMessages
+        requestState.spans = contextCandidate.spans
+        requestState.spanSource = contextCandidate.spanSource
+        requestState.contextMessageCount = contextCandidate.contextMessageCount
+        requestState.spanCount = contextCandidate.spanCount
+        requestState.textChars = contextCandidate.textChars
+        requestState.textBytes = contextCandidate.textBytes
+        requestState.toolNames = contextCandidate.toolNames
+        requestState.files = contextCandidate.files
+      }
+
+      const sessionHistoryCandidate = createComparableRequestState(requestState)
+      const sessionHistoryApplied = applySessionHistorySnapshot(sessionHistoryCandidate)
+      if (sessionHistoryApplied && shouldPreferSnapshot(requestState, sessionHistoryCandidate)) {
+        requestState.contextMessages = sessionHistoryCandidate.contextMessages
+        requestState.spans = sessionHistoryCandidate.spans
+        requestState.spanSource = sessionHistoryCandidate.spanSource
+        requestState.contextMessageCount = sessionHistoryCandidate.contextMessageCount
+        requestState.spanCount = sessionHistoryCandidate.spanCount
+        requestState.textChars = sessionHistoryCandidate.textChars
+        requestState.textBytes = sessionHistoryCandidate.textBytes
+        requestState.toolNames = sessionHistoryCandidate.toolNames
+        requestState.files = sessionHistoryCandidate.files
+      }
+
+      if (!providerApplied && requestState.spans.length === 0 && !contextApplied && !sessionHistoryApplied) {
         applySessionHistorySnapshot(requestState)
       }
       state.pendingRequests.push(requestState)

package/src/pi-request-telemetry.mjs

@@ -427,6 +427,84 @@ function parseStructuredTextArray(items) {
   return parts
 }
 
+function normalizeProviderToolCalls(toolCalls) {
+  if (!Array.isArray(toolCalls)) {
+    return []
+  }
+
+  const parts = []
+  for (const item of toolCalls) {
+    const object = asObject(item)
+    const functionObject = asObject(object.function)
+    const id = normalizeString(object.id ?? object.call_id, '')
+    const name = normalizeString(
+      functionObject.name
+      ?? object.name,
+      ''
+    )
+    const argumentsValue = parseJsonLikeString(
+      functionObject.arguments
+      ?? object.arguments
+    )
+
+    if (name === '' && id === '' && argumentsValue === undefined) {
+      continue
+    }
+
+    parts.push({
+      type: 'toolCall',
+      id,
+      name,
+      arguments: argumentsValue,
+    })
+  }
+
+  return parts
+}
+
+function normalizeMessageContent(content) {
+  if (typeof content === 'string') {
+    return [{ type: 'text', text: content }]
+  }
+
+  if (Array.isArray(content)) {
+    return parseStructuredTextArray(content)
+  }
+
+  return []
+}
+
+function normalizeProviderRoleMessage(object) {
+  const role = normalizeString(object.role, '')
+  if (role === '') {
+    return null
+  }
+
+  if (role === 'tool') {
+    return {
+      role: 'toolResult',
+      toolCallId: normalizeString(object.tool_call_id ?? object.call_id ?? object.id, ''),
+      toolName: normalizeString(object.name, ''),
+      details: object,
+      content: [{
+        type: 'text',
+        text: typeof object.content === 'string' ? object.content : safeJson(object.content),
+      }],
+    }
+  }
+
+  const content = normalizeMessageContent(object.content)
+  const toolCalls = normalizeProviderToolCalls(object.tool_calls)
+
+  return {
+    role,
+    content: [...content, ...toolCalls],
+    toolCallId: normalizeString(object.toolCallId ?? object.tool_call_id ?? object.call_id, ''),
+    toolName: normalizeString(object.toolName ?? object.name, ''),
+    details: object.details,
+  }
+}
+
 function convertProviderInputItemToMessage(item) {
   const object = asObject(item)
 
@@ -470,20 +548,7 @@ function convertProviderInputItemToMessage(item) {
   }
 
   if (object.role) {
-    if (typeof object.content === 'string') {
-      return {
-        role: normalizeString(object.role, ''),
-        content: [{ type: 'text', text: object.content }],
-      }
-    }
-
-    return {
-      role: normalizeString(object.role, ''),
-      content: parseStructuredTextArray(object.content),
-      toolCallId: normalizeString(object.toolCallId, ''),
-      toolName: normalizeString(object.toolName, ''),
-      details: object.details,
-    }
+    return normalizeProviderRoleMessage(object)
   }
 
   const normalized = normalizeTextPart(object)
@@ -505,6 +570,22 @@ export function extractMessagesFromProviderPayload(payload) {
 
   if (Array.isArray(object.messages)) {
     return object.messages
+      .map((item) => {
+        if (typeof item === 'string') {
+          return {
+            role: 'user',
+            content: [{ type: 'text', text: item }],
+          }
+        }
+        if (!item || typeof item !== 'object') {
+          return null
+        }
+        if (item.role) {
+          return normalizeProviderRoleMessage(asObject(item))
+        }
+        return convertProviderInputItemToMessage(item)
+      })
+      .filter((message) => normalizeString(message?.role, '') !== '')
   }
 
   if (typeof object.input === 'string') {

package/src/pi-sdk-turn.mjs

@@ -1,5 +1,6 @@
 import path from 'node:path'
 import process from 'node:process'
+import { setMaxListeners } from 'node:events'
 import { pathToFileURL } from 'node:url'
 import {
   formatHeartbeatReason,
@@ -154,6 +155,16 @@ function applyRequestTelemetryEnv(request) {
   }
 }
 
+function relaxAbortSignalListenerLimit(signal) {
+  if (!signal || typeof signal !== 'object') {
+    return
+  }
+
+  try {
+    setMaxListeners(0, signal)
+  } catch {}
+}
+
 function deriveTokenAttributionKind({ activeToolName, pendingToolNames, pendingFiles, lastAssistantActivity }) {
   if (String(activeToolName ?? '').trim() !== '') {
     return 'tool_running'
@@ -484,6 +495,7 @@ export async function runSdkTurnWithPi(pi, request) {
   let tokenUsageEvents = 0
   let tokenUsage = createEmptyTokenUsage()
   let lastAssistantActivity = ''
+  let abortSignalLimitRelaxed = false
   const pendingToolNames = new Set()
   const pendingFiles = new Set()
   const events = []
@@ -569,6 +581,10 @@ export async function runSdkTurnWithPi(pi, request) {
       lastEventAt = Date.now()
 
       if (event.type === 'agent_start') {
+        if (!abortSignalLimitRelaxed) {
+          relaxAbortSignalListenerLimit(session?.agent?.signal)
+          abortSignalLimitRelaxed = true
+        }
         agentStarted = true
         emitLiveFeed(request, {
           type: 'agent_start',