@probelabs/visor 0.1.182-ee → 0.1.183-ee

package/dist/sdk/{chunk-DHETLQIX.mjs → chunk-GA2TYKSR.mjs}

@@ -3,11 +3,11 @@ import {
   fallback_ndjson_exports,
   init_fallback_ndjson,
   init_trace_helpers
-} from "./chunk-7ERVRLDV.mjs";
+} from "./chunk-XOAEKFKB.mjs";
 import {
   addFailIfTriggered,
   init_metrics
-} from "./chunk-34QX63WK.mjs";
+} from "./chunk-FWWLD555.mjs";
 import {
   createPermissionHelpers,
   detectLocalMode,
@@ -20,11 +20,11 @@ import {
 import {
   MemoryStore,
   init_memory_store
-} from "./chunk-UFHOIB3R.mjs";
+} from "./chunk-RI4ONH5X.mjs";
 import {
   init_logger,
   logger_exports
-} from "./chunk-FT3I25QV.mjs";
+} from "./chunk-ZPYODGYA.mjs";
 import {
   __esm,
   __export,
@@ -736,4 +736,4 @@ export {
   failure_condition_evaluator_exports,
   init_failure_condition_evaluator
 };
-//# sourceMappingURL=chunk-DHETLQIX.mjs.map
+//# sourceMappingURL=chunk-GA2TYKSR.mjs.map

package/dist/sdk/{chunk-6DPPP7LD.mjs → chunk-IDL3AA3G.mjs}

@@ -58,7 +58,7 @@ import {
 import {
   config_exports,
   init_config
-} from "./chunk-4I3TJ7UJ.mjs";
+} from "./chunk-ZOF5QT6U.mjs";
 import {
   ExecutionJournal,
   checkLoopBudget,
@@ -67,14 +67,15 @@ import {
   init_routing,
   init_snapshot_store,
   snapshot_store_exports
-} from "./chunk-KBGQJKIZ.mjs";
+} from "./chunk-NPSLGKXB.mjs";
 import {
   FailureConditionEvaluator,
   init_failure_condition_evaluator
-} from "./chunk-U6K5SK7X.mjs";
+} from "./chunk-V45TITKX.mjs";
 import {
   addEvent,
   emitNdjsonFallback,
+  emitNdjsonFullSpan,
   emitNdjsonSpanWithEvents,
   fallback_ndjson_exports,
   init_fallback_ndjson,
@@ -82,11 +83,11 @@ import {
   setSpanAttributes,
   trace_helpers_exports,
   withActiveSpan
-} from "./chunk-3JFK6KCD.mjs";
+} from "./chunk-TFUQ2D5L.mjs";
 import {
   generateHumanId,
   init_human_id
-} from "./chunk-6VVXKXTI.mjs";
+} from "./chunk-MFXPJUUE.mjs";
 import {
   addDiagramBlock,
   init_metrics,
@@ -265,8 +266,13 @@ function createProbeTracerAdapter(fallbackTracer) {
     "delegation.start",
     "delegation.complete",
     "graceful_stop.invoked",
+    "graceful_stop.initiated",
     "probe.timeout_configured",
-    "negotiated_timeout.observer"
+    "negotiated_timeout.observer",
+    "negotiated_timeout.observer_extended",
+    "negotiated_timeout.observer_declined",
+    "negotiated_timeout.observer_exhausted",
+    "negotiated_timeout.abort_summary"
   ]);
   const emitEvent = (name, attrs) => {
     try {
@@ -289,12 +295,64 @@ function createProbeTracerAdapter(fallbackTracer) {
     }
   };
   return {
-    withSpan: async (name, fn, attrs) => withActiveSpan(name, attrs, async (span) => {
-      if (fallback && typeof fallback.withSpan === "function") {
-        return await fallback.withSpan(name, async () => fn(span), attrs);
-      }
-      return await fn(span);
-    }),
+    withSpan: async (name, fn, attrs, onResult) => {
+      const startHr = process.hrtime();
+      return withActiveSpan(name, attrs, async (span) => {
+        let result;
+        if (fallback && typeof fallback.withSpan === "function") {
+          result = await fallback.withSpan(name, async () => fn(span), attrs);
+        } else {
+          result = await fn(span);
+        }
+        const extraAttrs = {};
+        if (onResult) {
+          try {
+            const capturingSpan = {
+              setAttributes: (a) => {
+                Object.assign(extraAttrs, a);
+                try {
+                  span.setAttributes(a);
+                } catch {
+                }
+              },
+              setAttribute: (k, v) => {
+                extraAttrs[k] = v;
+                try {
+                  span.setAttribute(k, v);
+                } catch {
+                }
+              },
+              addEvent: (...args) => {
+                try {
+                  span.addEvent(...args);
+                } catch {
+                }
+              }
+            };
+            onResult(capturingSpan, result);
+          } catch {
+          }
+        }
+        try {
+          const endHr = process.hrtime();
+          const ctx = span.spanContext?.() || {};
+          const parentCtx = span.parentSpanContext || span._parentSpanContext;
+          emitNdjsonFullSpan({
+            name,
+            traceId: ctx.traceId,
+            spanId: ctx.spanId,
+            parentSpanId: parentCtx?.spanId,
+            startTime: [startHr[0], startHr[1]],
+            endTime: [endHr[0], endHr[1]],
+            attributes: { ...flattenAttrs(attrs), ...flattenAttrs(extraAttrs) },
+            events: [],
+            status: { code: 1 }
+          });
+        } catch {
+        }
+        return result;
+      });
+    },
     recordEvent: (name, attrs) => {
       emitEvent(name, attrs);
       if (fallback && typeof fallback.recordEvent === "function") {
@@ -398,10 +456,19 @@ function createProbeTracerAdapter(fallbackTracer) {
       }
       if (!span && fallbackSpan) return fallbackSpan;
       if (!span) return null;
+      const accumulated = {
+        "delegation.session_id": sessionId,
+        "delegation.task": task
+      };
+      const delegStartHr = process.hrtime();
+      let delegStatus = 1;
       return {
         setAttributes: (attrs) => {
           try {
-            if (attrs) span.setAttributes(attrs);
+            if (attrs) {
+              Object.assign(accumulated, attrs);
+              span.setAttributes(attrs);
+            }
           } catch {
           }
           if (fallbackSpan && typeof fallbackSpan.setAttributes === "function") {
@@ -413,6 +480,8 @@ function createProbeTracerAdapter(fallbackTracer) {
         },
         setStatus: (status) => {
           try {
+            const code = status?.code;
+            if (code === 2) delegStatus = 2;
             span.setStatus(status);
           } catch {
           }
@@ -434,6 +503,23 @@ function createProbeTracerAdapter(fallbackTracer) {
             } catch {
             }
           }
+          try {
+            const delegEndHr = process.hrtime();
+            const ctx = span.spanContext?.() || {};
+            emitNdjsonFullSpan({
+              name: "probe.delegation",
+              traceId: ctx.traceId,
+              spanId: ctx.spanId,
+              parentSpanId: void 0,
+              // delegation spans are roots within probe
+              startTime: [delegStartHr[0], delegStartHr[1]],
+              endTime: [delegEndHr[0], delegEndHr[1]],
+              attributes: flattenAttrs(accumulated) || {},
+              events: [],
+              status: { code: delegStatus }
+            });
+          } catch {
+          }
         }
       };
     },
@@ -481,6 +567,7 @@ var init_ai_review_service = __esm({
     init_logger();
     init_lazy_otel();
     init_trace_helpers();
+    init_fallback_ndjson();
     init_tracer_init();
     init_diff_processor();
     init_comment_metadata();
@@ -1817,6 +1904,15 @@ ${"=".repeat(60)}
           if (!systemPrompt && schema !== "code-review" && !this.config.promptType) {
             systemPrompt = "You are general assistant, follow user instructions.";
           }
+          if (systemPrompt && (this.config.aiTimeout || this.config.timeoutBehavior)) {
+            systemPrompt += `
+
+IMPORTANT \u2014 Time-limit behavior:
+If you receive a message that the time limit has been reached or your operation is being interrupted, this means YOUR ALLOCATED TIME BUDGET for this task has been exhausted. The system is NOT shutting down or going offline.
+- Frame it as: "I ran out of the allocated time for this task."
+- NEVER say "the system is shutting down", "try again when the system is back online", or similar misleading language.
+- Summarize what you accomplished, what you didn't finish, and any actionable recommendations the user can follow up on.`;
+          }
           log(
             `\u{1F527} AIReviewService config: allowEdit=${this.config.allowEdit}, allowBash=${this.config.allowBash}, promptType=${this.config.promptType}`
           );
@@ -4295,7 +4391,7 @@ async function executeWorkflowAsTool(workflowId, args, context2, argsOverrides)
     ...args,
     ...argsOverrides
   };
-  const { WorkflowCheckProvider: WorkflowCheckProvider2 } = await import("./workflow-check-provider-VKYGI5GK.mjs");
+  const { WorkflowCheckProvider: WorkflowCheckProvider2 } = await import("./workflow-check-provider-YXALZNAQ.mjs");
   const provider = new WorkflowCheckProvider2();
   const checkConfig = {
     type: "workflow",
@@ -10890,7 +10986,7 @@ async function executeCheckWithForEachItems2(checkId, forEachParent, forEachItem
           }
         }
         try {
-          const { evaluateTransitions } = await import("./routing-XALEDC2G.mjs");
+          const { evaluateTransitions } = await import("./routing-JMZ7HDCC.mjs");
           const transTarget = await evaluateTransitions(
             onFinish.transitions,
             forEachParent,
@@ -10950,7 +11046,7 @@ async function executeCheckWithForEachItems2(checkId, forEachParent, forEachItem
             `[LevelDispatch] Error evaluating on_finish transitions for ${forEachParent}: ${e instanceof Error ? e.message : String(e)}`
           );
         }
-        const { evaluateGoto: evaluateGoto2 } = await import("./routing-XALEDC2G.mjs");
+        const { evaluateGoto: evaluateGoto2 } = await import("./routing-JMZ7HDCC.mjs");
         if (context2.debug) {
           logger.info(
             `[LevelDispatch] Evaluating on_finish.goto_js for forEach parent: ${forEachParent}`
@@ -14041,7 +14137,7 @@ var init_state_machine_execution_engine = __esm({
           try {
             const map = options?.webhookContext?.webhookData;
             if (map) {
-              const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-RRWCXSTG.mjs");
+              const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-YRADEEQY.mjs");
               const reg = CheckProviderRegistry2.getInstance();
               const p = reg.getProvider("http_input");
               if (p && typeof p.setWebhookContext === "function") p.setWebhookContext(map);
@@ -14154,7 +14250,7 @@ var init_state_machine_execution_engine = __esm({
           logger.info("[StateMachine] Using state machine engine");
         }
         if (!config) {
-          const { ConfigManager } = await import("./config-2STD74CJ.mjs");
+          const { ConfigManager } = await import("./config-JE4HKTWW.mjs");
           const configManager = new ConfigManager();
           config = await configManager.getDefaultConfig();
           logger.debug("[StateMachine] Using default configuration (no config provided)");
@@ -14164,7 +14260,7 @@ var init_state_machine_execution_engine = __esm({
           tag_filter: tagFilter
         } : config;
         try {
-          const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-RRWCXSTG.mjs");
+          const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-YRADEEQY.mjs");
           const registry = CheckProviderRegistry2.getInstance();
           registry.setCustomTools(configWithTagFilter.tools || {});
         } catch (error) {
@@ -14228,7 +14324,7 @@ var init_state_machine_execution_engine = __esm({
           try {
             const webhookData = this.executionContext?.webhookContext?.webhookData;
             if (webhookData instanceof Map) {
-              const { extractSlackContext: extractSlackContext2 } = await import("./schedule-tool-handler-J4NUETJ6.mjs");
+              const { extractSlackContext: extractSlackContext2 } = await import("./schedule-tool-handler-KLHE2SOW.mjs");
               const slackCtx = extractSlackContext2(webhookData);
               if (slackCtx) {
                 const payload = Array.from(webhookData.values())[0];
@@ -14257,7 +14353,7 @@ var init_state_machine_execution_engine = __esm({
         if (Array.isArray(configWithTagFilter.frontends) && configWithTagFilter.frontends.length > 0) {
           try {
             const { EventBus } = await import("./event-bus-5K3Y2FCS.mjs");
-            const { FrontendsHost } = await import("./host-X5ZZCEWN.mjs");
+            const { FrontendsHost } = await import("./host-T4LNVU2H.mjs");
             const bus = new EventBus();
             context2.eventBus = bus;
             frontendsHost = new FrontendsHost(bus, logger);
@@ -14609,9 +14705,9 @@ var init_state_machine_execution_engine = __esm({
        * @returns Array of failure condition evaluation results
        */
       async evaluateFailureConditions(checkName, reviewSummary, config, previousOutputs, authorAssociation) {
-        const { FailureConditionEvaluator: FailureConditionEvaluator2 } = await import("./failure-condition-evaluator-FFWJRAEQ.mjs");
+        const { FailureConditionEvaluator: FailureConditionEvaluator2 } = await import("./failure-condition-evaluator-H3PBFBYT.mjs");
         const evaluator = new FailureConditionEvaluator2();
-        const { addEvent: addEvent3 } = await import("./trace-helpers-3FFAI7X3.mjs");
+        const { addEvent: addEvent3 } = await import("./trace-helpers-XV5GAX5L.mjs");
         const { addFailIfTriggered } = await import("./metrics-JTOG2HNO.mjs");
         const checkConfig = config.checks?.[checkName];
         if (!checkConfig) {
@@ -15447,7 +15543,7 @@ var init_scheduler = __esm({
           inputs: schedule.workflowInputs
         });
         if (this.taskStore) {
-          const { trackExecution } = await import("./track-execution-AMQQNXKE.mjs");
+          const { trackExecution } = await import("./track-execution-66RLL6QT.mjs");
           await trackExecution(
             {
               taskStore: this.taskStore,
@@ -15576,7 +15672,7 @@ Please provide an updated response based on the reminder above. You may referenc
             debug: process.env.VISOR_DEBUG === "true"
           });
           if (this.taskStore) {
-            const { trackExecution } = await import("./track-execution-AMQQNXKE.mjs");
+            const { trackExecution } = await import("./track-execution-66RLL6QT.mjs");
             await trackExecution(
               {
                 taskStore: this.taskStore,
@@ -15846,6 +15942,13 @@ async function handleScheduleAction(args, context2) {
   }
 }
 async function handleCreate(args, context2, store) {
+  if (context2.executionSource === "scheduler") {
+    return {
+      success: false,
+      message: "Scheduled runs cannot create new schedules",
+      error: "Creating schedules from a scheduler-triggered execution is not allowed to prevent recursive scheduling."
+    };
+  }
   if (!args.reminder_text && !args.workflow) {
     return {
       success: false,
@@ -16095,6 +16198,13 @@ function formatTrigger(trigger) {
   return `\`${trigger.id.substring(0, 8)}\` - channels: ${channels} \u2192 workflow: "${trigger.workflow}"${filterStr}${status}`;
 }
 async function handleCreateTrigger(args, context2, store) {
+  if (context2.executionSource === "scheduler") {
+    return {
+      success: false,
+      message: "Scheduled runs cannot create new triggers",
+      error: "Creating message triggers from a scheduler-triggered execution is not allowed to prevent recursive scheduling."
+    };
+  }
   const workflow = args.workflow || "default";
   if (context2.availableWorkflows && !context2.availableWorkflows.includes(workflow)) {
     return {
@@ -16532,6 +16642,7 @@ function slackChannelTypeToScheduleType(channelType) {
   }
 }
 function buildScheduleToolContext(sources, availableWorkflows, permissions, outputInfo) {
+  const executionSource = sources.schedulerContext ? "scheduler" : "user";
   if (sources.slackContext) {
     const contextType = `slack:${sources.slackContext.userId}`;
     const scheduleType = determineScheduleType(
@@ -16555,7 +16666,8 @@ function buildScheduleToolContext(sources, availableWorkflows, permissions, outp
       availableWorkflows,
       scheduleType: finalScheduleType,
       permissions,
-      allowedScheduleType
+      allowedScheduleType,
+      executionSource
     };
   }
   if (sources.githubContext) {
@@ -16567,8 +16679,9 @@ function buildScheduleToolContext(sources, availableWorkflows, permissions, outp
       availableWorkflows,
       scheduleType: "personal",
       permissions,
-      allowedScheduleType: "personal"
+      allowedScheduleType: "personal",
       // GitHub context only allows personal schedules
+      executionSource
     };
   }
   return {
@@ -16578,8 +16691,9 @@ function buildScheduleToolContext(sources, availableWorkflows, permissions, outp
     availableWorkflows,
     scheduleType: "personal",
     permissions,
-    allowedScheduleType: "personal"
+    allowedScheduleType: "personal",
     // CLI context only allows personal schedules
+    executionSource
   };
 }
 var init_schedule_tool = __esm({
@@ -16795,6 +16909,7 @@ var init_mcp_custom_sse_server = __esm({
       workflowContext;
       keepaliveInterval = null;
       activeToolCalls = 0;
+      inFlightToolCalls = /* @__PURE__ */ new Set();
       lastActivityAt = Date.now();
       gracefulStopRequested = false;
       static KEEPALIVE_INTERVAL_MS = 3e4;
@@ -16925,6 +17040,53 @@ var init_mcp_custom_sse_server = __esm({
           this.keepaliveInterval = null;
         }
       }
+      beginToolCall(toolName, workspace) {
+        if (workspace) {
+          workspace.acquire();
+        }
+        const toolCall = {
+          toolName,
+          startedAt: Date.now(),
+          workspace,
+          counted: true,
+          released: false
+        };
+        this.inFlightToolCalls.add(toolCall);
+        this.activeToolCalls++;
+        this.lastActivityAt = Date.now();
+        return toolCall;
+      }
+      finalizeToolCall(toolCall, reason) {
+        if (toolCall.counted) {
+          this.activeToolCalls = Math.max(0, this.activeToolCalls - 1);
+          toolCall.counted = false;
+        }
+        if (toolCall.workspace && !toolCall.released) {
+          toolCall.workspace.release();
+          toolCall.released = true;
+        }
+        this.inFlightToolCalls.delete(toolCall);
+        this.lastActivityAt = Date.now();
+        if (reason === "forced-stop") {
+          logger.warn(
+            `[CustomToolsSSEServer:${this.sessionId}] Force-released stranded tool call '${toolCall.toolName}' after ${Date.now() - toolCall.startedAt}ms`
+          );
+        }
+      }
+      forceDrainToolCalls() {
+        if (this.inFlightToolCalls.size === 0) {
+          if (this.activeToolCalls > 0) {
+            logger.warn(
+              `[CustomToolsSSEServer:${this.sessionId}] Resetting active tool counter with no in-flight records (${this.activeToolCalls})`
+            );
+            this.activeToolCalls = 0;
+          }
+          return;
+        }
+        for (const toolCall of Array.from(this.inFlightToolCalls)) {
+          this.finalizeToolCall(toolCall, "forced-stop");
+        }
+      }
       /**
        * Stop the server and cleanup resources
        */
@@ -16955,6 +17117,7 @@ var init_mcp_custom_sse_server = __esm({
             logger.warn(
               `[CustomToolsSSEServer:${this.sessionId}] Drain timeout reached; stopping with ${this.activeToolCalls} active tool call(s)`
             );
+            this.forceDrainToolCalls();
           }
         }
         if (this.debug) {
@@ -17339,11 +17502,7 @@ var init_mcp_custom_sse_server = __esm({
        */
       async handleToolCall(id, toolName, args) {
         const workspace = this.workflowContext?.workspace;
-        if (workspace) {
-          workspace.acquire();
-        }
-        this.activeToolCalls++;
-        this.lastActivityAt = Date.now();
+        const toolCall = this.beginToolCall(toolName, workspace);
         try {
           if (this.debug) {
             logger.debug(
@@ -17436,6 +17595,7 @@ var init_mcp_custom_sse_server = __esm({
             try {
               if (isScheduleTool(toolName)) {
                 const webhookData = this.workflowContext?.executionContext?.webhookContext?.webhookData;
+                const webhookEventType = this.workflowContext?.executionContext?.webhookContext?.eventType;
                 const slackContext = webhookData ? extractSlackContext(webhookData) : null;
                 const visorCfg = this.workflowContext?.visorConfig;
                 const availableWorkflows = visorCfg?.checks ? Object.keys(visorCfg.checks) : void 0;
@@ -17454,6 +17614,9 @@ var init_mcp_custom_sse_server = __esm({
                       userName: slackContext.userName,
                       timezone: slackContext.timezone,
                       channelType: slackContext.channelType
+                    } : void 0,
+                    schedulerContext: webhookEventType === "schedule" ? {
+                      scheduleId: this.workflowContext?.executionContext?.inputs?.schedule?.id
                     } : void 0
                   },
                   availableWorkflows,
@@ -17630,11 +17793,7 @@ var init_mcp_custom_sse_server = __esm({
             }
           };
         } finally {
-          this.activeToolCalls = Math.max(0, this.activeToolCalls - 1);
-          this.lastActivityAt = Date.now();
-          if (workspace) {
-            workspace.release();
-          }
+          this.finalizeToolCall(toolCall, "completed");
         }
       }
       /**
@@ -42425,10 +42584,11 @@ function buildBuiltinGlobals(opts) {
   const asyncFunctionNames = /* @__PURE__ */ new Set();
   const scheduleFn = async (args = {}) => {
     try {
-      const { handleScheduleAction: handleScheduleAction2, buildScheduleToolContext: buildScheduleToolContext2 } = await import("./schedule-tool-Z6QYL2B3.mjs");
-      const { extractSlackContext: extractSlackContext2 } = await import("./schedule-tool-handler-J4NUETJ6.mjs");
+      const { handleScheduleAction: handleScheduleAction2, buildScheduleToolContext: buildScheduleToolContext2 } = await import("./schedule-tool-4M45RK3E.mjs");
+      const { extractSlackContext: extractSlackContext2 } = await import("./schedule-tool-handler-KLHE2SOW.mjs");
       const parentCtx = opts.sessionInfo?._parentContext;
       const webhookData = parentCtx?.prInfo?.eventContext?.webhookData;
+      const webhookEventType = parentCtx?.webhookContext?.eventType || parentCtx?.prInfo?.eventContext?.eventType;
       const visorCfg = parentCtx?.config;
       const slackContext = webhookData ? extractSlackContext2(webhookData) : null;
       const availableWorkflows = visorCfg?.checks ? Object.keys(visorCfg.checks) : void 0;
@@ -42436,7 +42596,8 @@ function buildBuiltinGlobals(opts) {
       const context2 = buildScheduleToolContext2(
         {
           slackContext: slackContext || void 0,
-          cliContext: slackContext ? void 0 : { userId: "script" }
+          cliContext: slackContext ? void 0 : { userId: "script" },
+          schedulerContext: webhookEventType === "schedule" ? { scheduleId: parentCtx?.inputs?.schedule?.id } : void 0
         },
         availableWorkflows,
         permissions
@@ -46150,4 +46311,4 @@ undici/lib/fetch/body.js:
 undici/lib/websocket/frame.js:
   (*! ws. MIT License. Einar Otto Stangvik <einaros@gmail.com> *)
 */
-//# sourceMappingURL=chunk-6DPPP7LD.mjs.map
+//# sourceMappingURL=chunk-IDL3AA3G.mjs.map