@probelabs/probe 0.6.0-rc296 → 0.6.0-rc298

package/build/agent/ProbeAgent.d.ts

@@ -118,6 +118,38 @@ export interface ProbeAgentOptions {
   negotiatedTimeoutMaxPerRequest?: number;
 }
 
+/**
+ * Emitted when the negotiated timeout observer grants a time extension.
+ * Parent processes should listen to this event and extend their own deadlines accordingly.
+ */
+export interface TimeoutExtendedEvent {
+  /** Duration of the granted extension in milliseconds */
+  grantedMs: number;
+  /** Reason the observer granted the extension */
+  reason: string;
+  /** Number of extensions used so far */
+  extensionsUsed: number;
+  /** Number of extensions remaining */
+  extensionsRemaining: number;
+  /** Total extra time granted across all extensions in ms */
+  totalExtraTimeMs: number;
+  /** Remaining budget for future extensions in ms */
+  budgetRemainingMs: number;
+}
+
+/**
+ * Emitted when the negotiated timeout observer declines an extension and begins wind-down.
+ * After this event, the agent will produce its final answer and no more extensions will be granted.
+ */
+export interface TimeoutWindingDownEvent {
+  /** Reason the observer declined the extension */
+  reason: string;
+  /** Number of extensions used before declining */
+  extensionsUsed: number;
+  /** Total extra time granted across all extensions in ms */
+  totalExtraTimeMs: number;
+}
+
 /**
  * Tool execution event data
  */

package/build/agent/ProbeAgent.js

@@ -2759,7 +2759,7 @@ export class ProbeAgent {
       }
 
       // Initialize the MCP XML bridge
-      this.mcpBridge = new MCPXmlBridge({ debug: this.debug });
+      this.mcpBridge = new MCPXmlBridge({ debug: this.debug, agentEvents: this.events });
       await this.mcpBridge.initialize(mcpConfig);
 
       const mcpToolNames = this.mcpBridge.getToolNames();
@@ -3847,6 +3847,18 @@ or
                 active_tools_count: activeToolsList.length,
               });
             }
+
+            // Notify the parent that the agent extended its timeout (#522).
+            // The parent can listen to this event and extend its own deadline
+            // (e.g., adjust Promise.race timeout) instead of killing the agent.
+            this.events.emit('timeout.extended', {
+              grantedMs,
+              reason: decision.reason || 'work in progress',
+              extensionsUsed: negotiatedTimeoutState.extensionsUsed,
+              extensionsRemaining: negotiatedTimeoutState.maxRequests - negotiatedTimeoutState.extensionsUsed,
+              totalExtraTimeMs: negotiatedTimeoutState.totalExtraTimeMs,
+              budgetRemainingMs: remainingBudgetMs - grantedMs,
+            });
           } else {
             // Observer decided not to extend — two-phase graceful stop
             if (this.debug) {
@@ -3863,6 +3875,13 @@ or
               });
             }
 
+            // Notify the parent that the agent is winding down — no more extensions (#522)
+            this.events.emit('timeout.windingDown', {
+              reason: decision.reason || 'observer declined',
+              extensionsUsed: negotiatedTimeoutState.extensionsUsed,
+              totalExtraTimeMs: negotiatedTimeoutState.totalExtraTimeMs,
+            });
+
             await this._initiateGracefulStop(gracefulTimeoutState, `observer declined: ${decision.reason}`);
           }
         };

package/build/agent/mcp/client.js

@@ -164,6 +164,8 @@ export class MCPClientManager {
     this.debug = options.debug || process.env.DEBUG_MCP === '1';
     this.config = null;
     this.tracer = options.tracer || null;
+    // Optional event emitter for broadcasting tool call lifecycle to the agent (#522)
+    this.agentEvents = options.agentEvents || null;
   }
 
   /**
@@ -452,6 +454,7 @@ export class MCPClientManager {
     }
 
     const startTime = Date.now();
+    const toolCallId = `mcp-${toolName}-${startTime}`;
 
     // Record tool call start
     this.recordMcpEvent('tool.call_started', {
@@ -460,6 +463,17 @@ export class MCPClientManager {
       originalToolName: tool.originalName
     });
 
+    // Emit toolCall event so the agent's activeTools map tracks MCP tool calls (#522)
+    if (this.agentEvents) {
+      this.agentEvents.emit('toolCall', {
+        toolCallId,
+        name: toolName,
+        args,
+        status: 'started',
+        timestamp: new Date().toISOString(),
+      });
+    }
+
     try {
       if (this.debug) {
         console.error(`[MCP DEBUG] Calling ${toolName} with args:`, JSON.stringify(args, null, 2));
@@ -502,6 +516,16 @@ export class MCPClientManager {
         durationMs
       });
 
+      // Emit toolCall completion so agent's activeTools removes this entry (#522)
+      if (this.agentEvents) {
+        this.agentEvents.emit('toolCall', {
+          toolCallId,
+          name: toolName,
+          status: 'completed',
+          timestamp: new Date().toISOString(),
+        });
+      }
+
       return result;
     } catch (error) {
       const durationMs = Date.now() - startTime;
@@ -521,6 +545,16 @@ export class MCPClientManager {
         isTimeout: error.message.includes('timeout')
       });
 
+      // Emit toolCall error so agent's activeTools removes this entry (#522)
+      if (this.agentEvents) {
+        this.agentEvents.emit('toolCall', {
+          toolCallId,
+          name: toolName,
+          status: 'error',
+          timestamp: new Date().toISOString(),
+        });
+      }
+
       throw error;
     }
   }

package/build/agent/mcp/xmlBridge.js

@@ -36,6 +36,7 @@ export class MCPXmlBridge {
   constructor(options = {}) {
     this.debug = options.debug || false;
     this.tracer = options.tracer || null;
+    this.agentEvents = options.agentEvents || null;
     this.mcpTools = {};
     this.mcpManager = null;
     this.toolDescriptions = {};
@@ -84,7 +85,7 @@ export class MCPXmlBridge {
         console.error('[MCP DEBUG] Initializing MCP client manager...');
       }
 
-      this.mcpManager = new MCPClientManager({ debug: this.debug, tracer: this.tracer });
+      this.mcpManager = new MCPClientManager({ debug: this.debug, tracer: this.tracer, agentEvents: this.agentEvents });
       const result = await this.mcpManager.initialize(mcpConfigs);
 
       // Get tools from the manager (already in Vercel format)