@probelabs/probe 0.6.0-rc295 → 0.6.0-rc297

package/index.d.ts

@@ -61,6 +61,32 @@ export interface ProbeAgentOptions {
   skillDirs?: string[];
   /** Custom prompt to run after attempt_completion for validation/review (runs before mermaid/JSON validation) */
   completionPrompt?: string;
+  /** Enable task management system for tracking multi-step progress */
+  enableTasks?: boolean;
+  /** Timeout in ms for AI requests (default: 120000 or REQUEST_TIMEOUT env var). Used to abort hung requests. */
+  requestTimeout?: number;
+  /** Maximum timeout in ms for the entire operation including all retries and fallbacks (default: 300000 or MAX_OPERATION_TIMEOUT env var). */
+  maxOperationTimeout?: number;
+  /** Timeout behavior: 'graceful' winds down with bonus steps, 'hard' aborts immediately, 'negotiated' uses a timeout observer (separate LLM call) to decide extensions (default: 'graceful'). Env var: TIMEOUT_BEHAVIOR */
+  timeoutBehavior?: 'graceful' | 'hard' | 'negotiated';
+  /** Number of bonus steps during graceful timeout wind-down (default: 4, range: 1-20). Env var: GRACEFUL_TIMEOUT_BONUS_STEPS */
+  gracefulTimeoutBonusSteps?: number;
+  /** Total extra time budget in ms for negotiated timeout (default: 1800000 = 30 min). Env var: NEGOTIATED_TIMEOUT_BUDGET */
+  negotiatedTimeoutBudget?: number;
+  /** Max time extension requests for negotiated timeout (default: 3, range: 1-10). Env var: NEGOTIATED_TIMEOUT_MAX_REQUESTS */
+  negotiatedTimeoutMaxRequests?: number;
+  /** Max ms per extension request for negotiated timeout (default: 600000 = 10 min). Env var: NEGOTIATED_TIMEOUT_MAX_PER_REQUEST */
+  negotiatedTimeoutMaxPerRequest?: number;
+  /** How long to wait for subagents/MCP to wind down after observer declines (default: 45000 = 45s). Env var: GRACEFUL_STOP_DEADLINE */
+  gracefulStopDeadline?: number;
+  /** Retry configuration for handling transient API failures */
+  retry?: any;
+  /** Fallback configuration for multi-provider support */
+  fallback?: any;
+  /** Enable the delegate tool for task distribution to subagents */
+  enableDelegate?: boolean;
+  /** Enable bash tool for command execution */
+  enableBash?: boolean;
 }
 
 /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe",
-  "version": "0.6.0-rc295",
+  "version": "0.6.0-rc297",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",
@@ -81,7 +81,7 @@
     "@anthropic-ai/claude-agent-sdk": "^0.1.46",
     "@modelcontextprotocol/sdk": "^1.0.0",
     "@nyariv/sandboxjs": "github:probelabs/SandboxJS",
-    "@probelabs/maid": "^0.0.27",
+    "@probelabs/maid": "^0.0.28",
     "acorn": "^8.15.0",
     "acorn-walk": "^8.3.4",
     "adm-zip": "^0.5.16",

package/src/agent/ProbeAgent.d.ts

@@ -106,10 +106,48 @@ export interface ProbeAgentOptions {
   requestTimeout?: number;
   /** Maximum timeout in ms for the entire operation including all retries and fallbacks (default: 300000 or MAX_OPERATION_TIMEOUT env var). This is the absolute maximum time for streamTextWithRetryAndFallback. */
   maxOperationTimeout?: number;
-  /** Timeout behavior: 'graceful' winds down with bonus steps giving the agent a chance to respond, 'hard' aborts immediately (default: 'graceful'). Env var: TIMEOUT_BEHAVIOR */
-  timeoutBehavior?: 'graceful' | 'hard';
+  /** Timeout behavior: 'graceful' winds down with bonus steps, 'hard' aborts immediately, 'negotiated' uses a timeout observer (separate LLM call) to decide extensions (default: 'graceful'). Env var: TIMEOUT_BEHAVIOR */
+  timeoutBehavior?: 'graceful' | 'hard' | 'negotiated';
   /** Number of bonus steps during graceful timeout wind-down (default: 4, range: 1-20). Env var: GRACEFUL_TIMEOUT_BONUS_STEPS */
   gracefulTimeoutBonusSteps?: number;
+  /** Total extra time budget in ms for negotiated timeout (default: 1800000 = 30 min). Env var: NEGOTIATED_TIMEOUT_BUDGET */
+  negotiatedTimeoutBudget?: number;
+  /** Max time extension requests for negotiated timeout (default: 3, range: 1-10). Env var: NEGOTIATED_TIMEOUT_MAX_REQUESTS */
+  negotiatedTimeoutMaxRequests?: number;
+  /** Max ms per extension request for negotiated timeout (default: 600000 = 10 min). Env var: NEGOTIATED_TIMEOUT_MAX_PER_REQUEST */
+  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;
 }
 
 /**