@probelabs/probe 0.6.0-rc323 → 0.6.0-rc325

package/cjs/index.cjs

@@ -19643,7 +19643,7 @@ var init_dist2 = __esm({
       "ETIMEDOUT",
       "EPIPE"
     ];
-    VERSION = true ? "4.0.27" : "0.0.0-test";
+    VERSION = true ? "4.0.28" : "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 = /* @__PURE__ */ Symbol(
@@ -27587,7 +27587,7 @@ var init_dist3 = __esm({
           schema: amazonBedrockEmbeddingModelOptionsSchema
         })) != null ? _a17 : {};
         const isNovaModel = this.modelId.startsWith("amazon.nova-") && this.modelId.includes("embed");
-        const isCohereModel = this.modelId.startsWith("cohere.embed-");
+        const isCohereModel = this.modelId.includes("cohere.embed-");
         const args = isNovaModel ? {
           taskType: "SINGLE_EMBEDDING",
           singleEmbeddingParams: {
@@ -27611,7 +27611,7 @@ var init_dist3 = __esm({
           normalize: bedrockOptions.normalize
         };
         const url2 = this.getUrl(this.modelId);
-        const { value: response } = await postJsonToApi({
+        const { value: response, responseHeaders } = await postJsonToApi({
           url: url2,
           headers: await resolve(
             combineHeaders(await resolve(this.config.headers), headers)
@@ -27640,7 +27640,10 @@ var init_dist3 = __esm({
         } else {
           embedding = response.embeddings.float[0];
         }
-        const tokens = "inputTextTokenCount" in response ? response.inputTextTokenCount : "inputTokenCount" in response ? (_f = response.inputTokenCount) != null ? _f : 0 : NaN;
+        const headerTokenCount = Number(
+          responseHeaders == null ? void 0 : responseHeaders["x-amzn-bedrock-input-token-count"]
+        );
+        const tokens = "inputTextTokenCount" in response ? response.inputTextTokenCount : "inputTokenCount" in response ? (_f = response.inputTokenCount) != null ? _f : 0 : headerTokenCount;
         return {
           embeddings: [embedding],
           usage: { tokens },
@@ -27863,7 +27866,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.112" : "0.0.0-test";
+    VERSION2 = true ? "4.0.116" : "0.0.0-test";
     bedrockRerankingResponseSchema = lazySchema(
       () => zodSchema(
         external_exports.object({
@@ -86980,12 +86983,13 @@ function isRetryableError(error40, retryableErrors = DEFAULT_RETRYABLE_ERRORS) {
   if (!error40) return false;
   const errorString = error40.toString().toLowerCase();
   const errorMessage = (error40.message || "").toLowerCase();
+  const errorName = (error40.name || error40.constructor?.name || "").toLowerCase();
   const errorCode = (error40.code || "").toLowerCase();
   const errorType = (error40.type || "").toLowerCase();
   const statusCode = error40.statusCode || error40.status;
   for (const pattern of retryableErrors) {
     const lowerPattern = pattern.toLowerCase();
-    if (errorString.includes(lowerPattern) || errorMessage.includes(lowerPattern) || errorCode.includes(lowerPattern) || errorType.includes(lowerPattern) || statusCode?.toString() === pattern) {
+    if (errorString.includes(lowerPattern) || errorMessage.includes(lowerPattern) || errorName.includes(lowerPattern) || errorCode.includes(lowerPattern) || errorType.includes(lowerPattern) || statusCode?.toString() === pattern) {
       return true;
     }
   }
@@ -87022,7 +87026,9 @@ var init_RetryManager = __esm({
       "ECONNRESET",
       "ETIMEDOUT",
       "ENOTFOUND",
-      "api_error"
+      "api_error",
+      "No output generated",
+      "AI_NoOutputGeneratedError"
     ];
     RetryManager = class {
       /**
@@ -87042,7 +87048,7 @@ var init_RetryManager = __esm({
         this.backoffFactor = this._validateNumber(options.backoffFactor, 2, "backoffFactor", 1, 10);
         this.retryableErrors = options.retryableErrors || DEFAULT_RETRYABLE_ERRORS;
         this.debug = options.debug ?? false;
-        this.jitter = options.jitter ?? true;
+        this.jitter = typeof options.jitter === "boolean" ? options.jitter : true;
         if (this.maxDelay < this.initialDelay) {
           throw new Error("maxDelay must be greater than or equal to initialDelay");
         }
@@ -91424,6 +91430,9 @@ var init_acorn = __esm({
       // Can be either `"script"`, `"module"` or `"commonjs"`. This influences global
       // strict mode and parsing of `import` and `export` declarations.
       sourceType: "script",
+      // When set to true, enable strict parsing mode even if `sourceType`
+      // is `"script"`.
+      strict: false,
       // `onInsertedSemicolon` can be a callback that will be called when
       // a semicolon is automatically inserted. It will be passed the
       // position of the inserted semicolon as an offset, and if
@@ -91562,7 +91571,7 @@ var init_acorn = __esm({
       this.context = this.initialContext();
       this.exprAllowed = true;
       this.inModule = options.sourceType === "module";
-      this.strict = this.inModule || this.strictDirective(this.pos);
+      this.strict = this.inModule || options.strict === true || this.strictDirective(this.pos);
       this.potentialArrowAt = -1;
       this.potentialArrowInForAwait = false;
       this.yieldPos = this.awaitPos = this.awaitIdentPos = 0;
@@ -91580,9 +91589,12 @@ var init_acorn = __esm({
     };
     prototypeAccessors = { inFunction: { configurable: true }, inGenerator: { configurable: true }, inAsync: { configurable: true }, canAwait: { configurable: true }, allowReturn: { configurable: true }, allowSuper: { configurable: true }, allowDirectSuper: { configurable: true }, treatFunctionsAsVar: { configurable: true }, allowNewDotTarget: { configurable: true }, allowUsing: { configurable: true }, inClassStaticBlock: { configurable: true } };
     Parser2.prototype.parse = function parse8() {
+      var this$1$1 = this;
       var node = this.options.program || this.startNode();
       this.nextToken();
-      return this.parseTopLevel(node);
+      return this.catchStackOverflow(function() {
+        return this$1$1.parseTopLevel(node);
+      });
     };
     prototypeAccessors.inFunction.get = function() {
       return (this.currentVarScope().flags & SCOPE_FUNCTION) > 0;
@@ -91716,6 +91728,17 @@ var init_acorn = __esm({
       this.next();
       return true;
     };
+    pp$9.catchStackOverflow = function(f) {
+      try {
+        return f();
+      } catch (e) {
+        if (e instanceof Error && (/\bstack\b.*\b(exceeded|overflow)\b/i.test(e.message) || /\btoo much recursion\b/i.test(e.message))) {
+          this.raise(this.start, "Not enough stack space to parse input");
+        } else {
+          throw e;
+        }
+      }
+    };
     pp$9.expectContextual = function(name15) {
       if (!this.eatContextual(name15)) {
         this.unexpected();
@@ -91801,12 +91824,12 @@ var init_acorn = __esm({
     };
     pp$8 = Parser2.prototype;
     pp$8.parseTopLevel = function(node) {
-      var exports2 = /* @__PURE__ */ Object.create(null);
+      var exports$1 = /* @__PURE__ */ Object.create(null);
       if (!node.body) {
         node.body = [];
       }
       while (this.type !== types$1.eof) {
-        var stmt = this.parseStatement(null, true, exports2);
+        var stmt = this.parseStatement(null, true, exports$1);
         node.body.push(stmt);
       }
       if (this.inModule) {
@@ -91896,9 +91919,18 @@ var init_acorn = __esm({
         return true;
       }
       var id = this.input.slice(idStart, next);
-      if (keywordRelationalOperator.test(id) || isFor && id === "of") {
+      if (keywordRelationalOperator.test(id)) {
         return false;
       }
+      if (isFor && !isAwaitUsing && id === "of") {
+        skipWhiteSpace.lastIndex = next;
+        var skipAfterOf = skipWhiteSpace.exec(this.input);
+        next = next + skipAfterOf[0].length;
+        if (this.input.charCodeAt(next) !== 61 || // Check for ==, === and => operators
+        (ch = this.input.charCodeAt(next + 1)) === 61 || ch === 62) {
+          return false;
+        }
+      }
       return true;
     };
     pp$8.isAwaitUsing = function(isFor) {
@@ -91907,7 +91939,7 @@ var init_acorn = __esm({
     pp$8.isUsing = function(isFor) {
       return this.isUsingKeyword(false, isFor);
     };
-    pp$8.parseStatement = function(context, topLevel, exports2) {
+    pp$8.parseStatement = function(context, topLevel, exports$1) {
       var starttype = this.type, node = this.startNode(), kind;
       if (this.isLet(context)) {
         starttype = types$1._var;
@@ -91976,7 +92008,7 @@ var init_acorn = __esm({
               this.raise(this.start, "'import' and 'export' may appear only with 'sourceType: module'");
             }
           }
-          return starttype === types$1._import ? this.parseImport(node) : this.parseExport(node, exports2);
+          return starttype === types$1._import ? this.parseImport(node) : this.parseExport(node, exports$1);
         // If the statement does not start with a statement keyword or a
         // brace, it's an ExpressionStatement or LabeledStatement. We
         // simply start parsing an expression, and afterwards, if the
@@ -91995,6 +92027,9 @@ var init_acorn = __esm({
             if (!this.allowUsing) {
               this.raise(this.start, "Using declaration cannot appear in the top level when source type is `script` or in the bare case statement");
             }
+            if (context) {
+              this.raise(this.start, "Using declaration is not allowed in single-statement positions");
+            }
             if (usingKind === "await using") {
               if (!this.canAwait) {
                 this.raise(this.start, "Await using cannot appear outside of async function");
@@ -92129,14 +92164,15 @@ var init_acorn = __esm({
     };
     pp$8.parseForAfterInit = function(node, init, awaitAt) {
       if ((this.type === types$1._in || this.options.ecmaVersion >= 6 && this.isContextual("of")) && init.declarations.length === 1) {
-        if (this.options.ecmaVersion >= 9) {
-          if (this.type === types$1._in) {
-            if (awaitAt > -1) {
-              this.unexpected(awaitAt);
-            }
-          } else {
-            node.await = awaitAt > -1;
+        if (this.type === types$1._in) {
+          if ((init.kind === "using" || init.kind === "await using") && !init.declarations[0].init) {
+            this.raise(this.start, "Using declaration is not allowed in for-in loops");
           }
+          if (this.options.ecmaVersion >= 9 && awaitAt > -1) {
+            this.unexpected(awaitAt);
+          }
+        } else if (this.options.ecmaVersion >= 9) {
+          node.await = awaitAt > -1;
         }
         return this.parseForIn(node, init);
       }
@@ -92632,11 +92668,11 @@ var init_acorn = __esm({
         }
       }
     };
-    pp$8.parseExportAllDeclaration = function(node, exports2) {
+    pp$8.parseExportAllDeclaration = function(node, exports$1) {
       if (this.options.ecmaVersion >= 11) {
         if (this.eatContextual("as")) {
           node.exported = this.parseModuleExportName();
-          this.checkExport(exports2, node.exported, this.lastTokStart);
+          this.checkExport(exports$1, node.exported, this.lastTokStart);
         } else {
           node.exported = null;
         }
@@ -92652,22 +92688,22 @@ var init_acorn = __esm({
       this.semicolon();
       return this.finishNode(node, "ExportAllDeclaration");
     };
-    pp$8.parseExport = function(node, exports2) {
+    pp$8.parseExport = function(node, exports$1) {
       this.next();
       if (this.eat(types$1.star)) {
-        return this.parseExportAllDeclaration(node, exports2);
+        return this.parseExportAllDeclaration(node, exports$1);
       }
       if (this.eat(types$1._default)) {
-        this.checkExport(exports2, "default", this.lastTokStart);
+        this.checkExport(exports$1, "default", this.lastTokStart);
         node.declaration = this.parseExportDefaultDeclaration();
         return this.finishNode(node, "ExportDefaultDeclaration");
       }
       if (this.shouldParseExportStatement()) {
         node.declaration = this.parseExportDeclaration(node);
         if (node.declaration.type === "VariableDeclaration") {
-          this.checkVariableExport(exports2, node.declaration.declarations);
+          this.checkVariableExport(exports$1, node.declaration.declarations);
         } else {
-          this.checkExport(exports2, node.declaration.id, node.declaration.id.start);
+          this.checkExport(exports$1, node.declaration.id, node.declaration.id.start);
         }
         node.specifiers = [];
         node.source = null;
@@ -92676,7 +92712,7 @@ var init_acorn = __esm({
         }
       } else {
         node.declaration = null;
-        node.specifiers = this.parseExportSpecifiers(exports2);
+        node.specifiers = this.parseExportSpecifiers(exports$1);
         if (this.eatContextual("from")) {
           if (this.type !== types$1.string) {
             this.unexpected();
@@ -92724,66 +92760,66 @@ var init_acorn = __esm({
         return declaration;
       }
     };
-    pp$8.checkExport = function(exports2, name15, pos) {
-      if (!exports2) {
+    pp$8.checkExport = function(exports$1, name15, pos) {
+      if (!exports$1) {
         return;
       }
       if (typeof name15 !== "string") {
         name15 = name15.type === "Identifier" ? name15.name : name15.value;
       }
-      if (hasOwn(exports2, name15)) {
+      if (hasOwn(exports$1, name15)) {
         this.raiseRecoverable(pos, "Duplicate export '" + name15 + "'");
       }
-      exports2[name15] = true;
+      exports$1[name15] = true;
     };
-    pp$8.checkPatternExport = function(exports2, pat) {
+    pp$8.checkPatternExport = function(exports$1, pat) {
       var type = pat.type;
       if (type === "Identifier") {
-        this.checkExport(exports2, pat, pat.start);
+        this.checkExport(exports$1, pat, pat.start);
       } else if (type === "ObjectPattern") {
         for (var i = 0, list = pat.properties; i < list.length; i += 1) {
           var prop = list[i];
-          this.checkPatternExport(exports2, prop);
+          this.checkPatternExport(exports$1, prop);
         }
       } else if (type === "ArrayPattern") {
         for (var i$1 = 0, list$1 = pat.elements; i$1 < list$1.length; i$1 += 1) {
           var elt = list$1[i$1];
           if (elt) {
-            this.checkPatternExport(exports2, elt);
+            this.checkPatternExport(exports$1, elt);
           }
         }
       } else if (type === "Property") {
-        this.checkPatternExport(exports2, pat.value);
+        this.checkPatternExport(exports$1, pat.value);
       } else if (type === "AssignmentPattern") {
-        this.checkPatternExport(exports2, pat.left);
+        this.checkPatternExport(exports$1, pat.left);
       } else if (type === "RestElement") {
-        this.checkPatternExport(exports2, pat.argument);
+        this.checkPatternExport(exports$1, pat.argument);
       }
     };
-    pp$8.checkVariableExport = function(exports2, decls) {
-      if (!exports2) {
+    pp$8.checkVariableExport = function(exports$1, decls) {
+      if (!exports$1) {
         return;
       }
       for (var i = 0, list = decls; i < list.length; i += 1) {
         var decl = list[i];
-        this.checkPatternExport(exports2, decl.id);
+        this.checkPatternExport(exports$1, decl.id);
       }
     };
     pp$8.shouldParseExportStatement = function() {
       return this.type.keyword === "var" || this.type.keyword === "const" || this.type.keyword === "class" || this.type.keyword === "function" || this.isLet() || this.isAsyncFunction();
     };
-    pp$8.parseExportSpecifier = function(exports2) {
+    pp$8.parseExportSpecifier = function(exports$1) {
       var node = this.startNode();
       node.local = this.parseModuleExportName();
       node.exported = this.eatContextual("as") ? this.parseModuleExportName() : node.local;
       this.checkExport(
-        exports2,
+        exports$1,
         node.exported,
         node.exported.start
       );
       return this.finishNode(node, "ExportSpecifier");
     };
-    pp$8.parseExportSpecifiers = function(exports2) {
+    pp$8.parseExportSpecifiers = function(exports$1) {
       var nodes = [], first2 = true;
       this.expect(types$1.braceL);
       while (!this.eat(types$1.braceR)) {
@@ -92795,7 +92831,7 @@ var init_acorn = __esm({
         } else {
           first2 = false;
         }
-        nodes.push(this.parseExportSpecifier(exports2));
+        nodes.push(this.parseExportSpecifier(exports$1));
       }
       return nodes;
     };
@@ -93363,17 +93399,20 @@ var init_acorn = __esm({
       other[kind] = true;
     };
     pp$5.parseExpression = function(forInit, refDestructuringErrors) {
-      var startPos = this.start, startLoc = this.startLoc;
-      var expr = this.parseMaybeAssign(forInit, refDestructuringErrors);
-      if (this.type === types$1.comma) {
-        var node = this.startNodeAt(startPos, startLoc);
-        node.expressions = [expr];
-        while (this.eat(types$1.comma)) {
-          node.expressions.push(this.parseMaybeAssign(forInit, refDestructuringErrors));
+      var this$1$1 = this;
+      return this.catchStackOverflow(function() {
+        var startPos = this$1$1.start, startLoc = this$1$1.startLoc;
+        var expr = this$1$1.parseMaybeAssign(forInit, refDestructuringErrors);
+        if (this$1$1.type === types$1.comma) {
+          var node = this$1$1.startNodeAt(startPos, startLoc);
+          node.expressions = [expr];
+          while (this$1$1.eat(types$1.comma)) {
+            node.expressions.push(this$1$1.parseMaybeAssign(forInit, refDestructuringErrors));
+          }
+          return this$1$1.finishNode(node, "SequenceExpression");
         }
-        return this.finishNode(node, "SequenceExpression");
-      }
-      return expr;
+        return expr;
+      });
     };
     pp$5.parseMaybeAssign = function(forInit, refDestructuringErrors, afterLeftParse) {
       if (this.isContextual("yield")) {
@@ -93445,7 +93484,7 @@ var init_acorn = __esm({
       if (this.checkExpressionErrors(refDestructuringErrors)) {
         return expr;
       }
-      if (this.eat(types$1.question)) {
+      if (!(expr.type === "ArrowFunctionExpression" && expr.start === startPos) && this.eat(types$1.question)) {
         var node = this.startNodeAt(startPos, startLoc);
         node.test = expr;
         node.consequent = this.parseMaybeAssign();
@@ -93938,6 +93977,9 @@ var init_acorn = __esm({
       }
       var startPos = this.start, startLoc = this.startLoc;
       node.callee = this.parseSubscripts(this.parseExprAtom(null, false, true), startPos, startLoc, true, false);
+      if (node.callee.type === "Super") {
+        this.raiseRecoverable(startPos, "Invalid use of 'super'");
+      }
       if (this.eat(types$1.parenL)) {
         node.arguments = this.parseExprList(types$1.parenR, this.options.ecmaVersion >= 8, false);
       } else {
@@ -96590,7 +96632,7 @@ var init_acorn = __esm({
       }
       return this.finishToken(type, word);
     };
-    version2 = "8.16.0";
+    version2 = "8.17.0";
     Parser2.acorn = {
       Parser: Parser2,
       version: version2,
@@ -102761,27 +102803,41 @@ var init_ProbeAgent = __esm({
           timeoutState
         );
       }
+      /**
+       * Create a RetryManager from the agent's retry configuration.
+       * @param {Object} [overrides={}] - Retry option overrides
+       * @returns {RetryManager}
+       * @private
+       */
+      _createRetryManager(overrides = {}) {
+        return new RetryManager({
+          maxRetries: overrides.maxRetries ?? this.retryConfig.maxRetries ?? 3,
+          initialDelay: overrides.initialDelay ?? this.retryConfig.initialDelay ?? 1e3,
+          maxDelay: overrides.maxDelay ?? this.retryConfig.maxDelay ?? 3e4,
+          backoffFactor: overrides.backoffFactor ?? this.retryConfig.backoffFactor ?? 2,
+          retryableErrors: overrides.retryableErrors ?? this.retryConfig.retryableErrors,
+          jitter: overrides.jitter ?? this.retryConfig.jitter,
+          debug: overrides.debug ?? this.debug
+        });
+      }
       /**
        * Execute streamText with Vercel AI SDK using retry/fallback logic
        * @param {Object} options - streamText options
        * @param {AbortController} controller - Abort controller for the operation
+       * @param {Function} [consumeResult] - Optional callback to consume the stream result inside the same retry budget
        * @returns {Promise<Object>} - Stream result
        * @private
        */
-      async _executeWithVercelProvider(options, controller) {
+      async _executeWithVercelProvider(options, controller, consumeResult) {
         if (!this.retryManager) {
-          this.retryManager = new RetryManager({
-            maxRetries: this.retryConfig.maxRetries ?? 3,
-            initialDelay: this.retryConfig.initialDelay ?? 1e3,
-            maxDelay: this.retryConfig.maxDelay ?? 3e4,
-            backoffFactor: this.retryConfig.backoffFactor ?? 2,
-            retryableErrors: this.retryConfig.retryableErrors,
-            debug: this.debug
-          });
+          this.retryManager = this._createRetryManager();
         }
         if (!this.fallbackManager) {
           return await this.retryManager.executeWithRetry(
-            () => (0, import_ai4.streamText)({ ...options, abortSignal: controller.signal }),
+            async () => {
+              const result = (0, import_ai4.streamText)({ ...options, abortSignal: controller.signal });
+              return consumeResult ? await consumeResult(result) : result;
+            },
             {
               provider: this.apiType,
               model: this.model,
@@ -102806,16 +102862,14 @@ var init_ProbeAgent = __esm({
                 console.error(`[DEBUG] Stripped Gemini built-in tools for fallback to ${config2.provider} provider`);
               }
             }
-            const providerRetryManager = new RetryManager({
-              maxRetries: config2.maxRetries ?? this.retryConfig.maxRetries ?? 3,
-              initialDelay: this.retryConfig.initialDelay ?? 1e3,
-              maxDelay: this.retryConfig.maxDelay ?? 3e4,
-              backoffFactor: this.retryConfig.backoffFactor ?? 2,
-              retryableErrors: this.retryConfig.retryableErrors,
-              debug: this.debug
+            const providerRetryManager = this._createRetryManager({
+              maxRetries: config2.maxRetries ?? this.retryConfig.maxRetries
             });
             return await providerRetryManager.executeWithRetry(
-              () => (0, import_ai4.streamText)(fallbackOptions),
+              async () => {
+                const result = (0, import_ai4.streamText)(fallbackOptions);
+                return consumeResult ? await consumeResult(result) : result;
+              },
               {
                 provider: config2.provider,
                 model,
@@ -102954,10 +103008,11 @@ var init_ProbeAgent = __esm({
       /**
        * Execute streamText with retry and fallback support
        * @param {Object} options - streamText options
-       * @returns {Promise<Object>} - streamText result
+       * @param {Function} [consumeResult] - Optional callback to consume the result inside the same retry attempt
+       * @returns {Promise<Object>} - streamText result or consumer result
        * @private
        */
-      async streamTextWithRetryAndFallback(options) {
+      async streamTextWithRetryAndFallback(options, consumeResult) {
         const limiter = this.concurrencyLimiter;
         if (limiter && options.model) {
           options = { ...options, model: _ProbeAgent._wrapModelWithLimiter(options.model, limiter, this.debug) };
@@ -102989,6 +103044,7 @@ var init_ProbeAgent = __esm({
           const useClaudeCode = this.clientApiProvider === "claude-code" || process.env.USE_CLAUDE_CODE === "true";
           const useCodex = this.clientApiProvider === "codex" || process.env.USE_CODEX === "true";
           let result;
+          let usedVercelProvider = false;
           if (useClaudeCode || useCodex) {
             try {
               result = await this._tryEngineStreamPath(options, controller, timeoutState);
@@ -103003,7 +103059,11 @@ var init_ProbeAgent = __esm({
             }
           }
           if (!result) {
-            result = await this._executeWithVercelProvider(options, controller);
+            usedVercelProvider = true;
+            result = await this._executeWithVercelProvider(options, controller, consumeResult);
+          }
+          if (!usedVercelProvider && result && consumeResult) {
+            return await consumeResult(result);
           }
           return result;
         } finally {
@@ -104647,7 +104707,6 @@ You are working with a workspace. Available paths: ${workspaceDesc}
               activeTools.delete(key);
             }
           };
-          this.events.on("toolCall", onToolCall);
           const runTimeoutObserver = async () => {
             if (negotiatedTimeoutState.observerRunning) return;
             negotiatedTimeoutState.observerRunning = true;
@@ -105208,60 +105267,68 @@ Double-check your response based on the criteria above. If everything looks good
                 streamOptions.providerOptions = providerOpts;
               }
               const executeAIRequest = async () => {
-                const result = await this.streamTextWithRetryAndFallback(streamOptions);
-                let gracefulTimeoutId = null;
-                let hardAbortTimeoutId = null;
-                if (this.timeoutBehavior === "graceful" && gracefulTimeoutState && this.maxOperationTimeout > 0) {
-                  gracefulTimeoutId = setTimeout(() => {
-                    gracefulTimeoutState.triggered = true;
-                    if (this.debug) {
-                      console.log(`[DEBUG] Soft timeout after ${this.maxOperationTimeout}ms \u2014 entering wind-down mode (${gracefulTimeoutState.bonusStepsMax} bonus steps)`);
-                    }
-                    hardAbortTimeoutId = setTimeout(() => {
-                      if (this._abortController) {
-                        this._abortController.abort();
+                return await this.streamTextWithRetryAndFallback(streamOptions, async (result) => {
+                  this.events.on("toolCall", onToolCall);
+                  let gracefulTimeoutId = null;
+                  let hardAbortTimeoutId = null;
+                  if (this.timeoutBehavior === "graceful" && gracefulTimeoutState && this.maxOperationTimeout > 0) {
+                    gracefulTimeoutId = setTimeout(() => {
+                      gracefulTimeoutState.triggered = true;
+                      if (this.debug) {
+                        console.log(`[DEBUG] Soft timeout after ${this.maxOperationTimeout}ms \u2014 entering wind-down mode (${gracefulTimeoutState.bonusStepsMax} bonus steps)`);
                       }
+                      hardAbortTimeoutId = setTimeout(() => {
+                        if (this._abortController) {
+                          this._abortController.abort();
+                        }
+                        if (this.debug) {
+                          console.log(`[DEBUG] Hard abort \u2014 wind-down safety net expired after 60s`);
+                        }
+                      }, 6e4);
+                    }, this.maxOperationTimeout);
+                  }
+                  if (this.timeoutBehavior === "negotiated" && this.maxOperationTimeout > 0) {
+                    negotiatedTimeoutState.softTimeoutId = setTimeout(() => {
                       if (this.debug) {
-                        console.log(`[DEBUG] Hard abort \u2014 wind-down safety net expired after 60s`);
+                        console.log(`[DEBUG] Soft timeout after ${this.maxOperationTimeout}ms \u2014 invoking timeout observer`);
                       }
-                    }, 6e4);
-                  }, this.maxOperationTimeout);
-                }
-                if (this.timeoutBehavior === "negotiated" && this.maxOperationTimeout > 0) {
-                  negotiatedTimeoutState.softTimeoutId = setTimeout(() => {
+                      runTimeoutObserver();
+                    }, this.maxOperationTimeout);
+                  }
+                  try {
+                    const steps = await result.steps;
+                    let finalText;
+                    if (steps && steps.length > 1) {
+                      const lastStepText = steps[steps.length - 1].text;
+                      finalText = lastStepText || await result.text;
+                    } else {
+                      finalText = await result.text;
+                    }
+                    if (!options.schema && (!steps || steps.length === 0) && (!finalText || !finalText.trim())) {
+                      throw Object.assign(
+                        new Error("No output generated. Check the stream for errors."),
+                        { name: "AI_NoOutputGeneratedError" }
+                      );
+                    }
                     if (this.debug) {
-                      console.log(`[DEBUG] Soft timeout after ${this.maxOperationTimeout}ms \u2014 invoking timeout observer`);
+                      console.log(`[DEBUG] streamText completed: ${steps?.length || 0} steps, finalText=${finalText?.length || 0} chars`);
                     }
-                    runTimeoutObserver();
-                  }, this.maxOperationTimeout);
-                }
-                try {
-                  const steps = await result.steps;
-                  let finalText;
-                  if (steps && steps.length > 1) {
-                    const lastStepText = steps[steps.length - 1].text;
-                    finalText = lastStepText || await result.text;
-                  } else {
-                    finalText = await result.text;
-                  }
-                  if (this.debug) {
-                    console.log(`[DEBUG] streamText completed: ${steps?.length || 0} steps, finalText=${finalText?.length || 0} chars`);
-                  }
-                  const usage = await result.usage;
-                  if (usage) {
-                    this.tokenCounter.recordUsage(usage, result.experimental_providerMetadata);
-                  }
-                  return { finalText, result };
-                } finally {
-                  if (gracefulTimeoutId) clearTimeout(gracefulTimeoutId);
-                  if (hardAbortTimeoutId) clearTimeout(hardAbortTimeoutId);
-                  if (negotiatedTimeoutState.softTimeoutId) clearTimeout(negotiatedTimeoutState.softTimeoutId);
-                  if (this._gracefulStopHardAbortId) {
-                    clearTimeout(this._gracefulStopHardAbortId);
-                    this._gracefulStopHardAbortId = null;
+                    const usage = await result.usage;
+                    if (usage) {
+                      this.tokenCounter.recordUsage(usage, result.experimental_providerMetadata);
+                    }
+                    return { finalText, result };
+                  } finally {
+                    if (gracefulTimeoutId) clearTimeout(gracefulTimeoutId);
+                    if (hardAbortTimeoutId) clearTimeout(hardAbortTimeoutId);
+                    if (negotiatedTimeoutState.softTimeoutId) clearTimeout(negotiatedTimeoutState.softTimeoutId);
+                    if (this._gracefulStopHardAbortId) {
+                      clearTimeout(this._gracefulStopHardAbortId);
+                      this._gracefulStopHardAbortId = null;
+                    }
+                    this.events.removeListener("toolCall", onToolCall);
                   }
-                  this.events.removeListener("toolCall", onToolCall);
-                }
+                });
               };
               let aiResult;
               if (this.tracer) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe",
-  "version": "0.6.0-rc323",
+  "version": "0.6.0-rc325",
   "description": "Node.js wrapper for the probe code search tool",
   "main": "src/index.js",
   "module": "src/index.js",