@probelabs/visor 0.1.181 → 0.1.182-ee

package/dist/sdk/{chunk-4DVP6KVC.mjs → chunk-TQQNSHQV.mjs}

@@ -58,7 +58,7 @@ import {
 import {
   config_exports,
   init_config
-} from "./chunk-VMVIM4JB.mjs";
+} from "./chunk-ZOF5QT6U.mjs";
 import {
   ExecutionJournal,
   checkLoopBudget,
@@ -67,11 +67,11 @@ import {
   init_routing,
   init_snapshot_store,
   snapshot_store_exports
-} from "./chunk-KBGQJKIZ.mjs";
+} from "./chunk-ANUT54HW.mjs";
 import {
   FailureConditionEvaluator,
   init_failure_condition_evaluator
-} from "./chunk-U6K5SK7X.mjs";
+} from "./chunk-DHETLQIX.mjs";
 import {
   addEvent,
   emitNdjsonFallback,
@@ -82,11 +82,11 @@ import {
   setSpanAttributes,
   trace_helpers_exports,
   withActiveSpan
-} from "./chunk-3JFK6KCD.mjs";
+} from "./chunk-7ERVRLDV.mjs";
 import {
   generateHumanId,
   init_human_id
-} from "./chunk-6VVXKXTI.mjs";
+} from "./chunk-QXT47ZHR.mjs";
 import {
   addDiagramBlock,
   init_metrics,
@@ -4295,7 +4295,7 @@ async function executeWorkflowAsTool(workflowId, args, context2, argsOverrides)
     ...args,
     ...argsOverrides
   };
-  const { WorkflowCheckProvider: WorkflowCheckProvider2 } = await import("./workflow-check-provider-CJXW2Z4F.mjs");
+  const { WorkflowCheckProvider: WorkflowCheckProvider2 } = await import("./workflow-check-provider-EXMC6JIS.mjs");
   const provider = new WorkflowCheckProvider2();
   const checkConfig = {
     type: "workflow",
@@ -10890,7 +10890,7 @@ async function executeCheckWithForEachItems2(checkId, forEachParent, forEachItem
           }
         }
         try {
-          const { evaluateTransitions } = await import("./routing-XALEDC2G.mjs");
+          const { evaluateTransitions } = await import("./routing-CVQT4KHX.mjs");
           const transTarget = await evaluateTransitions(
             onFinish.transitions,
             forEachParent,
@@ -10950,7 +10950,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-CVQT4KHX.mjs");
         if (context2.debug) {
           logger.info(
             `[LevelDispatch] Evaluating on_finish.goto_js for forEach parent: ${forEachParent}`
@@ -14041,7 +14041,7 @@ var init_state_machine_execution_engine = __esm({
           try {
             const map = options?.webhookContext?.webhookData;
             if (map) {
-              const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-4YFVBGYU.mjs");
+              const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-I4BCWKRU.mjs");
               const reg = CheckProviderRegistry2.getInstance();
               const p = reg.getProvider("http_input");
               if (p && typeof p.setWebhookContext === "function") p.setWebhookContext(map);
@@ -14154,7 +14154,7 @@ var init_state_machine_execution_engine = __esm({
           logger.info("[StateMachine] Using state machine engine");
         }
         if (!config) {
-          const { ConfigManager } = await import("./config-TSA5FUOM.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 +14164,7 @@ var init_state_machine_execution_engine = __esm({
           tag_filter: tagFilter
         } : config;
         try {
-          const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-4YFVBGYU.mjs");
+          const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-I4BCWKRU.mjs");
           const registry = CheckProviderRegistry2.getInstance();
           registry.setCustomTools(configWithTagFilter.tools || {});
         } catch (error) {
@@ -14205,7 +14205,7 @@ var init_state_machine_execution_engine = __esm({
             logger.debug(
               `[PolicyEngine] Loading enterprise policy engine (engine=${configWithTagFilter.policy.engine})`
             );
-            const { loadEnterprisePolicyEngine } = await import("./enterprise/loader");
+            const { loadEnterprisePolicyEngine } = await import("./loader-ZNKKJEZ3.mjs");
             context2.policyEngine = await loadEnterprisePolicyEngine(configWithTagFilter.policy);
             logger.debug(
               `[PolicyEngine] Initialized: ${context2.policyEngine?.constructor?.name || "unknown"}`
@@ -14228,7 +14228,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-PCERK6ZZ.mjs");
+              const { extractSlackContext: extractSlackContext2 } = await import("./schedule-tool-handler-6QLZRTQA.mjs");
               const slackCtx = extractSlackContext2(webhookData);
               if (slackCtx) {
                 const payload = Array.from(webhookData.values())[0];
@@ -14257,7 +14257,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-QE4L7UXE.mjs");
+            const { FrontendsHost } = await import("./host-QBJ7TOWG.mjs");
             const bus = new EventBus();
             context2.eventBus = bus;
             frontendsHost = new FrontendsHost(bus, logger);
@@ -14609,9 +14609,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-5DZYMCGW.mjs");
         const evaluator = new FailureConditionEvaluator2();
-        const { addEvent: addEvent3 } = await import("./trace-helpers-3FFAI7X3.mjs");
+        const { addEvent: addEvent3 } = await import("./trace-helpers-WJXYVV4S.mjs");
         const { addFailIfTriggered } = await import("./metrics-JTOG2HNO.mjs");
         const checkConfig = config.checks?.[checkName];
         if (!checkConfig) {
@@ -15447,7 +15447,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-MKIQXP2C.mjs");
           await trackExecution(
             {
               taskStore: this.taskStore,
@@ -15576,7 +15576,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-MKIQXP2C.mjs");
             await trackExecution(
               {
                 taskStore: this.taskStore,
@@ -16795,6 +16795,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 +16926,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 +17003,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 +17388,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(
@@ -17630,11 +17675,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,8 +42466,8 @@ function buildBuiltinGlobals(opts) {
   const asyncFunctionNames = /* @__PURE__ */ new Set();
   const scheduleFn = async (args = {}) => {
     try {
-      const { handleScheduleAction: handleScheduleAction2, buildScheduleToolContext: buildScheduleToolContext2 } = await import("./schedule-tool-ZMX3Y7LF.mjs");
-      const { extractSlackContext: extractSlackContext2 } = await import("./schedule-tool-handler-PCERK6ZZ.mjs");
+      const { handleScheduleAction: handleScheduleAction2, buildScheduleToolContext: buildScheduleToolContext2 } = await import("./schedule-tool-AECLFHSY.mjs");
+      const { extractSlackContext: extractSlackContext2 } = await import("./schedule-tool-handler-6QLZRTQA.mjs");
       const parentCtx = opts.sessionInfo?._parentContext;
       const webhookData = parentCtx?.prInfo?.eventContext?.webhookData;
       const visorCfg = parentCtx?.config;
@@ -46150,4 +46191,4 @@ undici/lib/fetch/body.js:
 undici/lib/websocket/frame.js:
   (*! ws. MIT License. Einar Otto Stangvik <einaros@gmail.com> *)
 */
-//# sourceMappingURL=chunk-4DVP6KVC.mjs.map
+//# sourceMappingURL=chunk-TQQNSHQV.mjs.map