@probelabs/probe 0.6.0-rc306 → 0.6.0-rc308

package/cjs/agent/ProbeAgent.cjs

@@ -17714,10 +17714,10 @@ var init_dist2 = __esm({
       "ETIMEDOUT",
       "EPIPE"
     ];
-    VERSION = true ? "4.0.19" : "0.0.0-test";
+    VERSION = true ? "4.0.21" : "0.0.0-test";
     suspectProtoRx = /"(?:_|\\u005[Ff])(?:_|\\u005[Ff])(?:p|\\u0070)(?:r|\\u0072)(?:o|\\u006[Ff])(?:t|\\u0074)(?:o|\\u006[Ff])(?:_|\\u005[Ff])(?:_|\\u005[Ff])"\s*:/;
     suspectConstructorRx = /"(?:c|\\u0063)(?:o|\\u006[Ff])(?:n|\\u006[Ee])(?:s|\\u0073)(?:t|\\u0074)(?:r|\\u0072)(?:u|\\u0075)(?:c|\\u0063)(?:t|\\u0074)(?:o|\\u006[Ff])(?:r|\\u0072)"\s*:/;
-    ignoreOverride = Symbol(
+    ignoreOverride = /* @__PURE__ */ Symbol(
       "Let zodToJsonSchema decide on which parser to use"
     );
     defaultOptions = {
@@ -18053,7 +18053,7 @@ var init_dist2 = __esm({
       combined.$schema = "http://json-schema.org/draft-07/schema#";
       return combined;
     };
-    schemaSymbol = Symbol.for("vercel.ai.schema");
+    schemaSymbol = /* @__PURE__ */ Symbol.for("vercel.ai.schema");
     getOriginalFetch2 = () => globalThis.fetch;
     postJsonToApi = async ({
       url: url2,
@@ -20145,7 +20145,8 @@ async function prepareTools({
     } = await (0, import_internal2.prepareTools)({
       tools: ProviderTools,
       toolChoice,
-      supportsStructuredOutput: false
+      supportsStructuredOutput: false,
+      supportsStrictTools: false
     });
     toolWarnings.push(...anthropicToolWarnings);
     anthropicBetas.forEach((beta) => betas.add(beta));
@@ -22165,7 +22166,7 @@ var init_dist3 = __esm({
       details: external_exports.record(external_exports.string(), external_exports.unknown()).optional(),
       preview: external_exports.unknown().optional()
     });
-    VERSION2 = true ? "4.0.77" : "0.0.0-test";
+    VERSION2 = true ? "4.0.82" : "0.0.0-test";
     bedrockRerankingResponseSchema = lazySchema(
       () => zodSchema(
         external_exports.object({
@@ -24897,7 +24898,10 @@ var init_HookManager = __esm({
   }
 });
 
-// src/agent/imageConfig.js
+// src/agent/mediaConfig.js
+function isDocumentExtension(extension) {
+  return SUPPORTED_DOCUMENT_EXTENSIONS.includes(extension?.toLowerCase());
+}
 function isFormatSupportedByProvider(extension, provider) {
   if (!extension || typeof extension !== "string") {
     return false;
@@ -24906,7 +24910,7 @@ function isFormatSupportedByProvider(extension, provider) {
     return false;
   }
   const ext2 = extension.toLowerCase();
-  if (!SUPPORTED_IMAGE_EXTENSIONS.includes(ext2)) {
+  if (!SUPPORTED_MEDIA_EXTENSIONS.includes(ext2)) {
     return false;
   }
   if (!provider || typeof provider !== "string") {
@@ -24918,18 +24922,21 @@ function isFormatSupportedByProvider(extension, provider) {
   }
   return true;
 }
-var SUPPORTED_IMAGE_EXTENSIONS, IMAGE_MIME_TYPES, PROVIDER_UNSUPPORTED_FORMATS;
-var init_imageConfig = __esm({
-  "src/agent/imageConfig.js"() {
+var SUPPORTED_IMAGE_EXTENSIONS, SUPPORTED_DOCUMENT_EXTENSIONS, SUPPORTED_MEDIA_EXTENSIONS, MEDIA_MIME_TYPES, PROVIDER_UNSUPPORTED_FORMATS;
+var init_mediaConfig = __esm({
+  "src/agent/mediaConfig.js"() {
     "use strict";
     SUPPORTED_IMAGE_EXTENSIONS = ["png", "jpg", "jpeg", "webp", "bmp", "svg"];
-    IMAGE_MIME_TYPES = {
+    SUPPORTED_DOCUMENT_EXTENSIONS = ["pdf"];
+    SUPPORTED_MEDIA_EXTENSIONS = [...SUPPORTED_IMAGE_EXTENSIONS, ...SUPPORTED_DOCUMENT_EXTENSIONS];
+    MEDIA_MIME_TYPES = {
       "png": "image/png",
       "jpg": "image/jpeg",
       "jpeg": "image/jpeg",
       "webp": "image/webp",
       "bmp": "image/bmp",
-      "svg": "image/svg+xml"
+      "svg": "image/svg+xml",
+      "pdf": "application/pdf"
     };
     PROVIDER_UNSUPPORTED_FORMATS = {
       "google": ["svg"]
@@ -27899,7 +27906,7 @@ function resolveTargetPath(target, cwd) {
   }
   return filePart + suffix;
 }
-var import_path6, searchDelegateSchema, searchSchema, searchAllSchema, querySchema, extractSchema, delegateSchema, listSkillsSchema, useSkillSchema, listFilesSchema, searchFilesSchema, readImageSchema, bashSchema, analyzeAllSchema, executePlanSchema, cleanupExecutePlanSchema, searchDescription, searchDelegateDescription, queryDescription, extractDescription, delegateDescription, analyzeAllDescription;
+var import_path6, searchDelegateSchema, searchSchema, searchAllSchema, querySchema, extractSchema, delegateSchema, listSkillsSchema, useSkillSchema, listFilesSchema, searchFilesSchema, readImageSchema, readMediaSchema, bashSchema, analyzeAllSchema, executePlanSchema, cleanupExecutePlanSchema, searchDescription, searchDelegateDescription, queryDescription, extractDescription, delegateDescription, analyzeAllDescription;
 var init_common = __esm({
   "src/tools/common.js"() {
     "use strict";
@@ -27955,6 +27962,9 @@ var init_common = __esm({
     readImageSchema = external_exports2.object({
       path: external_exports2.string().describe("Path to the image file to read. Supports png, jpg, jpeg, webp, bmp, and svg formats.")
     });
+    readMediaSchema = external_exports2.object({
+      path: external_exports2.string().describe("Path to the media file to read. Supports images (png, jpg, jpeg, webp, bmp, svg) and documents (pdf).")
+    });
     bashSchema = external_exports2.object({
       command: external_exports2.string().describe("The bash command to execute"),
       workingDirectory: external_exports2.string().optional().describe("Directory to execute the command in (optional)"),
@@ -100906,7 +100916,7 @@ function debugLogToolResults(toolResults) {
     console.log(`[DEBUG]   tool: ${tr.toolName} | args: ${debugTruncate(argsStr)} | result: ${debugTruncate(resultStr)}`);
   }
 }
-var import_dotenv2, import_ai6, import_crypto9, import_events4, import_fs15, import_promises6, import_path18, ENGINE_ACTIVITY_TIMEOUT_DEFAULT, ENGINE_ACTIVITY_TIMEOUT_MIN, ENGINE_ACTIVITY_TIMEOUT_MAX, MAX_TOOL_ITERATIONS, MAX_HISTORY_MESSAGES, MAX_IMAGE_FILE_SIZE, ProbeAgent;
+var import_dotenv2, import_ai6, import_crypto9, import_events4, import_fs15, import_promises6, import_path18, ENGINE_ACTIVITY_TIMEOUT_DEFAULT, ENGINE_ACTIVITY_TIMEOUT_MIN, ENGINE_ACTIVITY_TIMEOUT_MAX, MAX_TOOL_ITERATIONS, MAX_HISTORY_MESSAGES, MAX_IMAGE_FILE_SIZE, MAX_DOCUMENT_FILE_SIZE, ProbeAgent;
 var init_ProbeAgent = __esm({
   "src/agent/ProbeAgent.js"() {
     import_dotenv2 = __toESM(require_main(), 1);
@@ -100921,7 +100931,7 @@ var init_ProbeAgent = __esm({
     init_simpleTelemetry();
     init_InMemoryStorageAdapter();
     init_HookManager();
-    init_imageConfig();
+    init_mediaConfig();
     init_tools2();
     init_common();
     init_taskTool();
@@ -100959,6 +100969,7 @@ var init_ProbeAgent = __esm({
     })();
     MAX_HISTORY_MESSAGES = 100;
     MAX_IMAGE_FILE_SIZE = 20 * 1024 * 1024;
+    MAX_DOCUMENT_FILE_SIZE = 32 * 1024 * 1024;
     ProbeAgent = class _ProbeAgent {
       /**
        * Create a new ProbeAgent instance
@@ -101587,28 +101598,29 @@ var init_ProbeAgent = __esm({
             this.toolImplementations.useSkill = useSkillToolInstance;
           }
         }
-        if (isToolAllowed("readImage")) {
-          this.toolImplementations.readImage = {
-            execute: async (params) => {
-              const imagePath = params.path;
-              if (!imagePath) {
-                throw new Error("Image path is required");
-              }
-              const filename = (0, import_path18.basename)(imagePath);
-              const extension = filename.toLowerCase().split(".").pop();
-              if (!extension || !SUPPORTED_IMAGE_EXTENSIONS.includes(extension)) {
-                throw new Error(`Invalid or unsupported image extension: ${extension}. Supported formats: ${SUPPORTED_IMAGE_EXTENSIONS.join(", ")}`);
-              }
-              if (this.apiType && !isFormatSupportedByProvider(extension, this.apiType)) {
-                throw new Error(`Image format '${extension}' is not supported by the current AI provider (${this.apiType}). Try using a different image format like PNG or JPEG.`);
-              }
-              const loaded = await this.loadImageIfValid(imagePath);
-              if (!loaded) {
-                throw new Error(`Failed to load image: ${imagePath}. The file may not exist, be too large, have an unsupported format, or be outside allowed directories.`);
-              }
-              return `Image loaded successfully: ${imagePath}. The image is now available for analysis in the conversation.`;
-            }
-          };
+        const readMediaExecute = async (params) => {
+          const mediaPath = params.path;
+          if (!mediaPath) {
+            throw new Error("File path is required");
+          }
+          const filename = (0, import_path18.basename)(mediaPath);
+          const extension = filename.toLowerCase().split(".").pop();
+          if (!extension || !SUPPORTED_MEDIA_EXTENSIONS.includes(extension)) {
+            throw new Error(`Unsupported file format: ${extension}. Supported formats: ${SUPPORTED_MEDIA_EXTENSIONS.join(", ")}`);
+          }
+          if (this.apiType && !isFormatSupportedByProvider(extension, this.apiType)) {
+            throw new Error(`File format '${extension}' is not supported by the current AI provider (${this.apiType}). Try converting to a different format.`);
+          }
+          const loaded = await this.loadMediaIfValid(mediaPath);
+          if (!loaded) {
+            throw new Error(`Failed to load file: ${mediaPath}. The file may not exist, be too large, have an unsupported format, or be outside allowed directories.`);
+          }
+          const mediaType = isDocumentExtension(extension) ? "Document" : "Image";
+          return `${mediaType} loaded successfully: ${mediaPath}. The file is now available for analysis in the conversation.`;
+        };
+        if (isToolAllowed("readMedia") || isToolAllowed("readImage")) {
+          this.toolImplementations.readMedia = { execute: readMediaExecute };
+          this.toolImplementations.readImage = { execute: readMediaExecute };
         }
         if (this.enableBash && wrappedTools.bashToolInstance && isToolAllowed("bash")) {
           this.toolImplementations.bash = wrappedTools.bashToolInstance;
@@ -102604,9 +102616,9 @@ var init_ProbeAgent = __esm({
             schema: searchFilesSchema,
             description: "Find files matching a glob pattern with recursive search capability."
           },
-          readImage: {
-            schema: readImageSchema,
-            description: "Read and load an image file for AI analysis."
+          readMedia: {
+            schema: readMediaSchema,
+            description: "Read and load a media file (image or PDF document) for AI analysis. Supports: png, jpg, jpeg, webp, bmp, svg, pdf."
           },
           listSkills: {
             schema: listSkillsSchema,
@@ -102751,7 +102763,7 @@ var init_ProbeAgent = __esm({
       async processImageReferences(content) {
         if (!content) return;
         const listFilesDirectories = this.extractListFilesDirectories(content);
-        const extensionsPattern = `(?:${SUPPORTED_IMAGE_EXTENSIONS.join("|")})`;
+        const extensionsPattern = `(?:${SUPPORTED_MEDIA_EXTENSIONS.join("|")})`;
         const imagePatterns = [
           // Direct file path mentions: "./screenshot.png", "/path/to/image.jpg", etc.
           new RegExp(`(?:\\.?\\.\\/)?[^\\s"'<>\\[\\]]+\\.${extensionsPattern}(?!\\w)`, "gi"),
@@ -102833,23 +102845,23 @@ var init_ProbeAgent = __esm({
         return directories;
       }
       /**
-       * Load and cache an image if it's valid and accessible
-       * @param {string} imagePath - Path to the image file
-       * @returns {Promise<boolean>} - True if image was loaded successfully
+       * Load and cache a media file (image or PDF) if it's valid and accessible
+       * @param {string} mediaPath - Path to the media file
+       * @returns {Promise<boolean>} - True if file was loaded successfully
        */
-      async loadImageIfValid(imagePath) {
+      async loadMediaIfValid(mediaPath) {
         try {
-          if (this.pendingImages.has(imagePath)) {
+          if (this.pendingImages.has(mediaPath)) {
             if (this.debug) {
-              console.log(`[DEBUG] Image already loaded: ${imagePath}`);
+              console.log(`[DEBUG] Media already loaded: ${mediaPath}`);
             }
             return true;
           }
           const allowedDirs = this.allowedFolders && this.allowedFolders.length > 0 ? this.allowedFolders : [process.cwd()];
           let absolutePath;
           let isPathAllowed2 = false;
-          if ((0, import_path18.isAbsolute)(imagePath)) {
-            absolutePath = safeRealpath((0, import_path18.resolve)(imagePath));
+          if ((0, import_path18.isAbsolute)(mediaPath)) {
+            absolutePath = safeRealpath((0, import_path18.resolve)(mediaPath));
             isPathAllowed2 = allowedDirs.some((dir) => {
               const resolvedDir = safeRealpath(dir);
               return absolutePath === resolvedDir || absolutePath.startsWith(resolvedDir + import_path18.sep);
@@ -102857,7 +102869,7 @@ var init_ProbeAgent = __esm({
           } else {
             for (const dir of allowedDirs) {
               const resolvedDir = safeRealpath(dir);
-              const resolvedPath = safeRealpath((0, import_path18.resolve)(dir, imagePath));
+              const resolvedPath = safeRealpath((0, import_path18.resolve)(dir, mediaPath));
               if (resolvedPath === resolvedDir || resolvedPath.startsWith(resolvedDir + import_path18.sep)) {
                 absolutePath = resolvedPath;
                 isPathAllowed2 = true;
@@ -102867,7 +102879,7 @@ var init_ProbeAgent = __esm({
           }
           if (!isPathAllowed2) {
             if (this.debug) {
-              console.log(`[DEBUG] Image path outside allowed directories: ${imagePath}`);
+              console.log(`[DEBUG] Media path outside allowed directories: ${mediaPath}`);
             }
             return false;
           }
@@ -102876,91 +102888,126 @@ var init_ProbeAgent = __esm({
             fileStats = await (0, import_promises6.stat)(absolutePath);
           } catch (error40) {
             if (this.debug) {
-              console.log(`[DEBUG] Image file not found: ${absolutePath}`);
+              console.log(`[DEBUG] Media file not found: ${absolutePath}`);
             }
             return false;
           }
-          if (fileStats.size > MAX_IMAGE_FILE_SIZE) {
+          const extension = absolutePath.toLowerCase().split(".").pop();
+          if (!SUPPORTED_MEDIA_EXTENSIONS.includes(extension)) {
             if (this.debug) {
-              console.log(`[DEBUG] Image file too large: ${absolutePath} (${fileStats.size} bytes, max: ${MAX_IMAGE_FILE_SIZE})`);
+              console.log(`[DEBUG] Unsupported media format: ${extension}`);
             }
             return false;
           }
-          const extension = absolutePath.toLowerCase().split(".").pop();
-          if (!SUPPORTED_IMAGE_EXTENSIONS.includes(extension)) {
+          const maxSize = isDocumentExtension(extension) ? MAX_DOCUMENT_FILE_SIZE : MAX_IMAGE_FILE_SIZE;
+          if (fileStats.size > maxSize) {
             if (this.debug) {
-              console.log(`[DEBUG] Unsupported image format: ${extension}`);
+              console.log(`[DEBUG] Media file too large: ${absolutePath} (${fileStats.size} bytes, max: ${maxSize})`);
             }
             return false;
           }
-          const mimeType = IMAGE_MIME_TYPES[extension];
+          const mimeType = MEDIA_MIME_TYPES[extension];
           const fileBuffer = await (0, import_promises6.readFile)(absolutePath);
           const base64Data = fileBuffer.toString("base64");
-          const dataUrl = `data:${mimeType};base64,${base64Data}`;
-          this.pendingImages.set(imagePath, dataUrl);
+          if (isDocumentExtension(extension)) {
+            this.pendingImages.set(mediaPath, {
+              type: "document",
+              mimeType,
+              data: base64Data,
+              filename: (0, import_path18.basename)(mediaPath)
+            });
+          } else {
+            const dataUrl = `data:${mimeType};base64,${base64Data}`;
+            this.pendingImages.set(mediaPath, dataUrl);
+          }
           if (this.debug) {
-            console.log(`[DEBUG] Successfully loaded image: ${imagePath} (${fileBuffer.length} bytes)`);
+            console.log(`[DEBUG] Successfully loaded media: ${mediaPath} (${fileBuffer.length} bytes, ${mimeType})`);
           }
           return true;
         } catch (error40) {
           if (this.debug) {
-            console.log(`[DEBUG] Failed to load image ${imagePath}: ${error40.message}`);
+            console.log(`[DEBUG] Failed to load media ${mediaPath}: ${error40.message}`);
           }
           return false;
         }
       }
+      /**
+       * Backward-compatible alias for loadMediaIfValid
+       * @param {string} imagePath - Path to the image file
+       * @returns {Promise<boolean>}
+       */
+      async loadImageIfValid(imagePath) {
+        return this.loadMediaIfValid(imagePath);
+      }
       /**
        * Get all currently loaded images as an array for AI model consumption
-       * @returns {Array<string>} - Array of base64 data URLs
+       * @returns {Array<string>} - Array of base64 data URLs (images only, for backward compat)
        */
       getCurrentImages() {
-        return Array.from(this.pendingImages.values());
+        return Array.from(this.pendingImages.values()).filter((v) => typeof v === "string");
       }
       /**
-       * Clear loaded images (useful for new conversations)
+       * Get all currently loaded media as an array of content parts
+       * @returns {Array<Object>} - Array of Vercel AI SDK content parts
+       */
+      getCurrentMedia() {
+        const parts = [];
+        for (const entry of this.pendingImages.values()) {
+          if (typeof entry === "string") {
+            parts.push({ type: "image", image: entry });
+          } else if (entry && entry.type === "document") {
+            parts.push({
+              type: "file",
+              mediaType: entry.mimeType,
+              data: entry.data,
+              filename: entry.filename
+            });
+          }
+        }
+        return parts;
+      }
+      /**
+       * Clear loaded media (useful for new conversations)
        */
       clearLoadedImages() {
         this.pendingImages.clear();
         this.currentImages = [];
         if (this.debug) {
-          console.log("[DEBUG] Cleared all loaded images");
+          console.log("[DEBUG] Cleared all loaded media");
         }
       }
       /**
-       * Prepare messages for AI consumption, adding images to the latest user message if available
+       * Prepare messages for AI consumption, adding media to the latest user message if available
        * @param {Array} messages - Current conversation messages
-       * @returns {Array} - Messages formatted for AI SDK with potential image content
+       * @returns {Array} - Messages formatted for AI SDK with potential media content
        */
       prepareMessagesWithImages(messages) {
-        const loadedImages = this.getCurrentImages();
-        if (loadedImages.length === 0) {
+        const mediaParts = this.getCurrentMedia();
+        if (mediaParts.length === 0) {
           return messages;
         }
-        const messagesWithImages = [...messages];
-        const lastUserMessageIndex = messagesWithImages.map((m) => m.role).lastIndexOf("user");
+        const messagesWithMedia = [...messages];
+        const lastUserMessageIndex = messagesWithMedia.map((m) => m.role).lastIndexOf("user");
         if (lastUserMessageIndex === -1) {
           if (this.debug) {
-            console.log("[DEBUG] No user messages found to attach images to");
+            console.log("[DEBUG] No user messages found to attach media to");
           }
           return messages;
         }
-        const lastUserMessage = messagesWithImages[lastUserMessageIndex];
+        const lastUserMessage = messagesWithMedia[lastUserMessageIndex];
         if (typeof lastUserMessage.content === "string") {
-          messagesWithImages[lastUserMessageIndex] = {
+          messagesWithMedia[lastUserMessageIndex] = {
             ...lastUserMessage,
             content: [
               { type: "text", text: lastUserMessage.content },
-              ...loadedImages.map((imageData) => ({
-                type: "image",
-                image: imageData
-              }))
+              ...mediaParts
             ]
           };
           if (this.debug) {
-            console.log(`[DEBUG] Added ${loadedImages.length} images to the latest user message`);
+            console.log(`[DEBUG] Added ${mediaParts.length} media items to the latest user message`);
           }
         }
-        return messagesWithImages;
+        return messagesWithMedia;
       }
       /**
        * Initialize mock model for testing
@@ -103858,40 +103905,42 @@ or
               }
               const jsonStr = responseText.replace(/^```(?:json)?\s*/, "").replace(/\s*```$/, "");
               const decision = JSON.parse(jsonStr);
+              let grantedMs = 0;
+              let grantedMin = 0;
               if (decision.extend && decision.minutes > 0) {
                 const requestedMs = Math.min(decision.minutes, maxPerReqMin) * 6e4;
-                const grantedMs2 = Math.min(requestedMs, remainingBudgetMs, negotiatedTimeoutState.maxPerRequestMs);
-                const grantedMin2 = Math.round(grantedMs2 / 6e4 * 10) / 10;
+                grantedMs = Math.min(requestedMs, remainingBudgetMs, negotiatedTimeoutState.maxPerRequestMs);
+                grantedMin = Math.round(grantedMs / 6e4 * 10) / 10;
                 negotiatedTimeoutState.extensionsUsed++;
-                negotiatedTimeoutState.totalExtraTimeMs += grantedMs2;
-                negotiatedTimeoutState.extensionMessage = `\u23F0 Time limit was reached. The timeout observer granted ${grantedMin2} more minute(s) (reason: ${decision.reason || "work in progress"}). Extensions remaining: ${negotiatedTimeoutState.maxRequests - negotiatedTimeoutState.extensionsUsed}. Continue your work efficiently.`;
+                negotiatedTimeoutState.totalExtraTimeMs += grantedMs;
+                negotiatedTimeoutState.extensionMessage = `\u23F0 Time limit was reached. The timeout observer granted ${grantedMin} more minute(s) (reason: ${decision.reason || "work in progress"}). Extensions remaining: ${negotiatedTimeoutState.maxRequests - negotiatedTimeoutState.extensionsUsed}. Continue your work efficiently.`;
                 negotiatedTimeoutState.softTimeoutId = setTimeout(() => {
                   runTimeoutObserver();
-                }, grantedMs2);
+                }, grantedMs);
                 if (this.debug) {
-                  console.log(`[DEBUG] Timeout observer: granted ${grantedMin2} min (reason: ${decision.reason}). Extensions: ${negotiatedTimeoutState.extensionsUsed}/${negotiatedTimeoutState.maxRequests}`);
+                  console.log(`[DEBUG] Timeout observer: granted ${grantedMin} min (reason: ${decision.reason}). Extensions: ${negotiatedTimeoutState.extensionsUsed}/${negotiatedTimeoutState.maxRequests}`);
                 }
                 if (this.tracer) {
                   this.tracer.addEvent("negotiated_timeout.observer_extended", {
                     decision_reason: decision.reason,
                     requested_minutes: decision.minutes,
-                    granted_ms: grantedMs2,
-                    granted_min: grantedMin2,
+                    granted_ms: grantedMs,
+                    granted_min: grantedMin,
                     extensions_used: negotiatedTimeoutState.extensionsUsed,
                     max_requests: negotiatedTimeoutState.maxRequests,
                     total_extra_time_ms: negotiatedTimeoutState.totalExtraTimeMs,
-                    budget_remaining_ms: remainingBudgetMs - grantedMs2,
+                    budget_remaining_ms: remainingBudgetMs - grantedMs,
                     active_tools: activeToolsList.map((t) => t.name),
                     active_tools_count: activeToolsList.length
                   });
                 }
                 this.events.emit("timeout.extended", {
-                  grantedMs: grantedMs2,
+                  grantedMs,
                   reason: decision.reason || "work in progress",
                   extensionsUsed: negotiatedTimeoutState.extensionsUsed,
                   extensionsRemaining: negotiatedTimeoutState.maxRequests - negotiatedTimeoutState.extensionsUsed,
                   totalExtraTimeMs: negotiatedTimeoutState.totalExtraTimeMs,
-                  budgetRemainingMs: remainingBudgetMs - grantedMs2
+                  budgetRemainingMs: remainingBudgetMs - grantedMs
                 });
               } else {
                 if (this.debug) {