@probelabs/visor 0.1.138 → 0.1.139

package/dist/sdk/{chunk-QSB2P6VY.mjs → chunk-UYMFCE4F.mjs}

@@ -3922,7 +3922,7 @@ async function executeWorkflowAsTool(workflowId, args, context2, argsOverrides)
     ...args,
     ...argsOverrides
   };
-  const { WorkflowCheckProvider: WorkflowCheckProvider2 } = await import("./workflow-check-provider-I3SB5RG7.mjs");
+  const { WorkflowCheckProvider: WorkflowCheckProvider2 } = await import("./workflow-check-provider-O2NXWYDY.mjs");
   const provider = new WorkflowCheckProvider2();
   const checkConfig = {
     type: "workflow",
@@ -11408,7 +11408,7 @@ var init_state_machine_execution_engine = __esm({
           try {
             const map = options?.webhookContext?.webhookData;
             if (map) {
-              const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-JX7FK3MS.mjs");
+              const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-QDSFKMQN.mjs");
               const reg = CheckProviderRegistry2.getInstance();
               const p = reg.getProvider("http_input");
               if (p && typeof p.setWebhookContext === "function") p.setWebhookContext(map);
@@ -11531,7 +11531,7 @@ var init_state_machine_execution_engine = __esm({
           tag_filter: tagFilter
         } : config;
         try {
-          const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-JX7FK3MS.mjs");
+          const { CheckProviderRegistry: CheckProviderRegistry2 } = await import("./check-provider-registry-QDSFKMQN.mjs");
           const registry = CheckProviderRegistry2.getInstance();
           registry.setCustomTools(configWithTagFilter.tools || {});
         } catch (error) {
@@ -11595,7 +11595,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-LBSAGK6P.mjs");
+              const { extractSlackContext: extractSlackContext2 } = await import("./schedule-tool-handler-S3R5YY5S.mjs");
               const slackCtx = extractSlackContext2(webhookData);
               if (slackCtx) {
                 const payload = Array.from(webhookData.values())[0];
@@ -35704,7 +35704,13 @@ var init_mcp_check_provider = __esm({
               };
             }
           }
-          const result = await this.executeMcpMethod(cfg, methodArgs, prInfo, dependencyResults);
+          const result = await this.executeMcpMethod(
+            cfg,
+            methodArgs,
+            prInfo,
+            dependencyResults,
+            sessionInfo
+          );
           let finalOutput = result;
           if (cfg.transform) {
             try {
@@ -35805,22 +35811,26 @@ var init_mcp_check_provider = __esm({
       /**
        * Execute an MCP method using the configured transport
        */
-      async executeMcpMethod(config, methodArgs, prInfo, dependencyResults) {
+      async executeMcpMethod(config, methodArgs, prInfo, dependencyResults, sessionInfo) {
         const transport = config.transport || "stdio";
         const timeout = (config.timeout || 60) * 1e3;
         if (transport === "custom") {
-          if (!this.customToolExecutor) {
-            throw new Error(
-              'No custom tools available. Define tools in the "tools" section of your configuration.'
-            );
-          }
-          const hasTool = await this.customToolExecutor.hasTool(config.method);
-          if (!hasTool) {
-            const availableToolNames = await this.customToolExecutor.getToolNames();
+          const localTools = sessionInfo?._parentContext?.config?.tools || sessionInfo?.config?.tools || {};
+          let localExecutor;
+          if (Object.keys(localTools).length > 0) {
+            localExecutor = new CustomToolExecutor(localTools);
+          }
+          const hasLocalTool = localExecutor ? await localExecutor.hasTool(config.method) : false;
+          const hasGlobalTool = this.customToolExecutor ? await this.customToolExecutor.hasTool(config.method) : false;
+          if (!hasLocalTool && !hasGlobalTool) {
+            const localNames = localExecutor ? await localExecutor.getToolNames() : [];
+            const globalNames = this.customToolExecutor ? await this.customToolExecutor.getToolNames() : [];
+            const availableToolNames = [.../* @__PURE__ */ new Set([...localNames, ...globalNames])];
             throw new Error(
               `Custom tool not found: ${config.method}. Available tools: ${availableToolNames.join(", ")}`
             );
           }
+          const activeExecutor = hasLocalTool ? localExecutor : this.customToolExecutor;
           const context2 = {
             pr: prInfo ? {
               number: prInfo.number,
@@ -35833,7 +35843,7 @@ var init_mcp_check_provider = __esm({
             outputs: this.buildOutputContext(dependencyResults),
             env: this.getSafeEnvironmentVariables()
           };
-          return await this.customToolExecutor.execute(config.method, methodArgs, context2);
+          return await activeExecutor.execute(config.method, methodArgs, context2);
         } else if (transport === "stdio") {
           return await this.executeStdioMethod(config, methodArgs, timeout);
         } else if (transport === "sse") {
@@ -37280,7 +37290,7 @@ function buildBuiltinGlobals(opts) {
   const scheduleFn = async (args = {}) => {
     try {
       const { handleScheduleAction: handleScheduleAction2, buildScheduleToolContext: buildScheduleToolContext2 } = await import("./schedule-tool-2COUUTF7.mjs");
-      const { extractSlackContext: extractSlackContext2 } = await import("./schedule-tool-handler-LBSAGK6P.mjs");
+      const { extractSlackContext: extractSlackContext2 } = await import("./schedule-tool-handler-S3R5YY5S.mjs");
       const parentCtx = opts.sessionInfo?._parentContext;
       const webhookData = parentCtx?.prInfo?.eventContext?.webhookData;
       const visorCfg = parentCtx?.config;
@@ -39414,6 +39424,7 @@ function projectWorkflowToGraph(workflow, workflowInputs, _parentCheckId) {
   }
   const config = {
     checks,
+    tools: workflow.tools,
     version: "1.0",
     output: {
       pr_comment: {
@@ -40201,7 +40212,8 @@ ${errors}`);
           on: loaded.on,
           // Carry over optional inputs/outputs if present so callers can consume them
           inputs: loaded.inputs,
-          outputs: loaded.outputs
+          outputs: loaded.outputs,
+          tools: loaded.tools
         };
         return workflowDef;
       }
@@ -40232,4 +40244,4 @@ undici/lib/fetch/body.js:
 undici/lib/websocket/frame.js:
   (*! ws. MIT License. Einar Otto Stangvik <einaros@gmail.com> *)
 */
-//# sourceMappingURL=chunk-QSB2P6VY.mjs.map
+//# sourceMappingURL=chunk-UYMFCE4F.mjs.map