@probelabs/visor 0.1.80 → 0.1.81

package/dist/sdk/{chunk-GNA5GPF7.mjs → chunk-V5PUTL5N.mjs}

@@ -20,7 +20,9 @@ var init_session_registry = __esm({
     SessionRegistry = class _SessionRegistry {
       static instance;
       sessions = /* @__PURE__ */ new Map();
+      exitHandlerRegistered = false;
       constructor() {
+        this.registerExitHandlers();
       }
       /**
        * Get the singleton instance of SessionRegistry
@@ -54,7 +56,15 @@ var init_session_registry = __esm({
       unregisterSession(sessionId) {
         if (this.sessions.has(sessionId)) {
           console.error(`\u{1F5D1}\uFE0F  Unregistering AI session: ${sessionId}`);
+          const agent = this.sessions.get(sessionId);
           this.sessions.delete(sessionId);
+          if (agent && typeof agent.cleanup === "function") {
+            try {
+              agent.cleanup();
+            } catch (error) {
+              console.error(`\u26A0\uFE0F  Warning: Failed to cleanup ProbeAgent: ${error}`);
+            }
+          }
         }
       }
       /**
@@ -62,6 +72,14 @@ var init_session_registry = __esm({
        */
       clearAllSessions() {
         console.error(`\u{1F9F9} Clearing all AI sessions (${this.sessions.size} sessions)`);
+        for (const [, agent] of this.sessions.entries()) {
+          if (agent && typeof agent.cleanup === "function") {
+            try {
+              agent.cleanup();
+            } catch {
+            }
+          }
+        }
         this.sessions.clear();
       }
       /**
@@ -76,6 +94,44 @@ var init_session_registry = __esm({
       hasSession(sessionId) {
         return this.sessions.has(sessionId);
       }
+      /**
+       * Register process exit handlers to cleanup sessions on exit
+       */
+      registerExitHandlers() {
+        if (this.exitHandlerRegistered) {
+          return;
+        }
+        const cleanupAndExit = (signal) => {
+          if (this.sessions.size > 0) {
+            console.error(`
+\u{1F9F9} [${signal}] Cleaning up ${this.sessions.size} active AI sessions...`);
+            this.clearAllSessions();
+          }
+        };
+        process.on("exit", () => {
+          if (this.sessions.size > 0) {
+            console.error(`\u{1F9F9} [exit] Cleaning up ${this.sessions.size} active AI sessions...`);
+            for (const [, agent] of this.sessions.entries()) {
+              if (agent && typeof agent.cleanup === "function") {
+                try {
+                  agent.cleanup();
+                } catch {
+                }
+              }
+            }
+            this.sessions.clear();
+          }
+        });
+        process.on("SIGINT", () => {
+          cleanupAndExit("SIGINT");
+          process.exit(0);
+        });
+        process.on("SIGTERM", () => {
+          cleanupAndExit("SIGTERM");
+          process.exit(0);
+        });
+        this.exitHandlerRegistered = true;
+      }
     };
   }
 });
@@ -1493,7 +1549,7 @@ var PRReviewer = class {
   async reviewPR(owner, repo, prNumber, prInfo, options = {}) {
     const { debug = false, config, checks } = options;
     if (config && checks && checks.length > 0) {
-      const { CheckExecutionEngine: CheckExecutionEngine2 } = await import("./check-execution-engine-2EQWRXWI.mjs");
+      const { CheckExecutionEngine: CheckExecutionEngine2 } = await import("./check-execution-engine-47OZJFUV.mjs");
       const engine = new CheckExecutionEngine2();
       const { results } = await engine.executeGroupedChecks(
         prInfo,
@@ -2141,17 +2197,6 @@ var IssueFilter = class {
 // src/providers/ai-check-provider.ts
 import fs3 from "fs/promises";
 import path3 from "path";
-
-// src/providers/claude-code-types.ts
-async function safeImport(moduleName) {
-  try {
-    return await import(moduleName);
-  } catch {
-    return null;
-  }
-}
-
-// src/providers/ai-check-provider.ts
 var AICheckProvider = class extends CheckProvider {
   aiReviewService;
   liquidEngine;
@@ -2429,54 +2474,6 @@ var AICheckProvider = class extends CheckProvider {
       );
     }
   }
-  /**
-   * Setup MCP tools based on AI configuration
-   */
-  async setupMcpTools(aiConfig) {
-    const tools = [];
-    if (aiConfig.mcpServers) {
-      try {
-        const mcpModule = await safeImport("@modelcontextprotocol/sdk");
-        if (!mcpModule) {
-          console.warn("@modelcontextprotocol/sdk package not found. MCP servers disabled.");
-          return tools;
-        }
-        const createSdkMcpServer = mcpModule.createSdkMcpServer || mcpModule.default?.createSdkMcpServer;
-        if (typeof createSdkMcpServer === "function") {
-          for (const [serverName, serverConfig] of Object.entries(aiConfig.mcpServers)) {
-            try {
-              const server = await createSdkMcpServer({
-                name: serverName,
-                command: serverConfig.command,
-                args: serverConfig.args || [],
-                env: { ...process.env, ...serverConfig.env }
-              });
-              const serverTools = await server.listTools();
-              tools.push(
-                ...serverTools.map((tool) => ({
-                  name: tool.name,
-                  server: serverName
-                }))
-              );
-            } catch (serverError) {
-              console.warn(
-                `Failed to setup MCP server ${serverName}: ${serverError instanceof Error ? serverError.message : "Unknown error"}`
-              );
-            }
-          }
-        } else {
-          console.warn(
-            "createSdkMcpServer function not found in @modelcontextprotocol/sdk. MCP servers disabled."
-          );
-        }
-      } catch (error) {
-        console.warn(
-          `Failed to import MCP SDK: ${error instanceof Error ? error.message : "Unknown error"}. MCP servers disabled.`
-        );
-      }
-    }
-    return tools;
-  }
   async execute(prInfo, config, _dependencyResults, sessionInfo) {
     if (config.env) {
       const result = EnvironmentResolver.withTemporaryEnv(config.env, () => {
@@ -2533,11 +2530,9 @@ var AICheckProvider = class extends CheckProvider {
     }
     if (Object.keys(mcpServers).length > 0) {
       aiConfig.mcpServers = mcpServers;
-      const mcpConfig = { mcpServers };
-      const mcpTools = await this.setupMcpTools(mcpConfig);
       if (aiConfig.debug) {
         console.error(
-          `\u{1F527} Debug: AI check MCP configured with ${Object.keys(mcpServers).length} servers; discovered ${mcpTools.length} tools`
+          `\u{1F527} Debug: AI check MCP configured with ${Object.keys(mcpServers).length} servers`
         );
       }
     }
@@ -3367,6 +3362,17 @@ var LogCheckProvider = class extends CheckProvider {
 // src/providers/claude-code-check-provider.ts
 import fs4 from "fs/promises";
 import path4 from "path";
+
+// src/providers/claude-code-types.ts
+async function safeImport(moduleName) {
+  try {
+    return await import(moduleName);
+  } catch {
+    return null;
+  }
+}
+
+// src/providers/claude-code-check-provider.ts
 function isClaudeCodeConstructor(value) {
   return typeof value === "function";
 }
@@ -3468,59 +3474,6 @@ var ClaudeCodeCheckProvider = class extends CheckProvider {
       );
     }
   }
-  /**
-   * Setup MCP tools based on configuration
-   */
-  async setupMcpTools(config) {
-    const tools = [];
-    if (config.allowedTools) {
-      for (const toolName of config.allowedTools) {
-        tools.push({ name: toolName });
-      }
-    }
-    if (config.mcpServers) {
-      try {
-        const mcpModule = await safeImport("@modelcontextprotocol/sdk");
-        if (!mcpModule) {
-          console.warn("@modelcontextprotocol/sdk package not found. MCP servers disabled.");
-          return tools;
-        }
-        const createSdkMcpServer = mcpModule.createSdkMcpServer || mcpModule.default?.createSdkMcpServer;
-        if (typeof createSdkMcpServer === "function") {
-          for (const [serverName, serverConfig] of Object.entries(config.mcpServers)) {
-            try {
-              const server = await createSdkMcpServer({
-                name: serverName,
-                command: serverConfig.command,
-                args: serverConfig.args || [],
-                env: { ...process.env, ...serverConfig.env }
-              });
-              const serverTools = await server.listTools();
-              tools.push(
-                ...serverTools.map((tool) => ({
-                  name: tool.name,
-                  server: serverName
-                }))
-              );
-            } catch (serverError) {
-              console.warn(
-                `Failed to setup MCP server ${serverName}: ${serverError instanceof Error ? serverError.message : "Unknown error"}`
-              );
-            }
-          }
-        } else {
-          console.warn(
-            "createSdkMcpServer function not found in @modelcontextprotocol/sdk. MCP servers disabled."
-          );
-        }
-      } catch (error) {
-        console.warn(
-          `Failed to import MCP SDK: ${error instanceof Error ? error.message : "Unknown error"}. MCP servers disabled.`
-        );
-      }
-    }
-    return tools;
-  }
   /**
    * Group files by their file extension for template context
    */
@@ -3751,14 +3704,18 @@ var ClaudeCodeCheckProvider = class extends CheckProvider {
     const startTime = Date.now();
     try {
       const client = await this.initializeClaudeCodeClient();
-      const tools = await this.setupMcpTools(claudeCodeConfig);
       const query = {
         query: processedPrompt,
-        tools: tools.length > 0 ? tools : void 0,
         maxTurns: claudeCodeConfig.maxTurns || 5,
         systemPrompt: claudeCodeConfig.systemPrompt,
         subagent: claudeCodeConfig.subagent
       };
+      if (claudeCodeConfig.allowedTools && claudeCodeConfig.allowedTools.length > 0) {
+        query.tools = claudeCodeConfig.allowedTools.map((name) => ({ name }));
+      }
+      if (claudeCodeConfig.mcpServers && Object.keys(claudeCodeConfig.mcpServers).length > 0) {
+        query.mcpServers = claudeCodeConfig.mcpServers;
+      }
       let response;
       if (sessionInfo?.reuseSession && sessionInfo.parentSessionId) {
         response = await client.query({
@@ -3786,8 +3743,7 @@ var ClaudeCodeCheckProvider = class extends CheckProvider {
         // Claude Code specific debug info
         sessionId: response.session_id,
         turnCount: response.turn_count,
-        usage: response.usage,
-        toolsUsed: tools.map((t) => t.name)
+        usage: response.usage
       };
       const suppressionEnabled = config.suppressionEnabled !== false;
       const issueFilter = new IssueFilter(suppressionEnabled);
@@ -7493,14 +7449,6 @@ ${expr}
         }
       }
     }
-    const executionStatistics = this.buildExecutionStatistics();
-    if (logFn === console.log) {
-      this.logExecutionSummary(executionStatistics);
-    }
-    if (shouldStopExecution) {
-      logger.info("");
-      logger.warn(`\u26A0\uFE0F  Execution stopped early due to fail-fast`);
-    }
     if (debug) {
       if (shouldStopExecution) {
         log2(
@@ -7521,6 +7469,14 @@ ${expr}
         }
       }
     }
+    const executionStatistics = this.buildExecutionStatistics();
+    if (logFn === console.log) {
+      this.logExecutionSummary(executionStatistics);
+    }
+    if (shouldStopExecution) {
+      logger.info("");
+      logger.warn(`\u26A0\uFE0F  Execution stopped early due to fail-fast`);
+    }
     return this.aggregateDependencyAwareResults(
       results,
       dependencyGraph,
@@ -8677,4 +8633,4 @@ export {
   logger,
   CheckExecutionEngine
 };
-//# sourceMappingURL=chunk-GNA5GPF7.mjs.map
+//# sourceMappingURL=chunk-V5PUTL5N.mjs.map