@probelabs/probe 0.6.0-rc207 → 0.6.0-rc208

package/src/agent/mcp/xmlBridge.js

@@ -127,6 +127,7 @@ export function parseXmlMcpToolCall(xmlString, mcpToolNames = []) {
 export class MCPXmlBridge {
   constructor(options = {}) {
     this.debug = options.debug || false;
+    this.tracer = options.tracer || null;
     this.mcpTools = {};
     this.mcpManager = null;
     this.xmlDefinitions = {};
@@ -179,8 +180,8 @@ export class MCPXmlBridge {
         console.error('[MCP DEBUG] Initializing MCP client manager...');
       }
 
-      // Initialize the MCP client manager
-      this.mcpManager = new MCPClientManager({ debug: this.debug });
+      // Initialize the MCP client manager with tracer support
+      this.mcpManager = new MCPClientManager({ debug: this.debug, tracer: this.tracer });
       const result = await this.mcpManager.initialize(mcpConfigs);
 
       // Get tools from the manager
@@ -211,11 +212,20 @@ export class MCPXmlBridge {
   }
 
   /**
-   * Get all XML tool definitions for inclusion in system prompt
+   * Get XML tool definitions for inclusion in system prompt
+   * @param {Array<string>|null} filterToolNames - Optional list of tool names to include (if null, include all)
    * @returns {string} Combined XML tool definitions
    */
-  getXmlToolDefinitions() {
-    return Object.values(this.xmlDefinitions).join('\n\n');
+  getXmlToolDefinitions(filterToolNames = null) {
+    if (filterToolNames === null) {
+      return Object.values(this.xmlDefinitions).join('\n\n');
+    }
+
+    // Filter definitions based on provided tool names
+    return Object.entries(this.xmlDefinitions)
+      .filter(([name]) => filterToolNames.includes(name))
+      .map(([, def]) => def)
+      .join('\n\n');
   }
 
   /**

package/src/agent/simpleTelemetry.js

@@ -231,6 +231,32 @@ export class SimpleAppTracer {
     });
   }
 
+  /**
+   * Record MCP (Model Context Protocol) events
+   * Tracks server connections, tool discovery, method filtering, and tool execution
+   */
+  recordMcpEvent(eventType, data = {}) {
+    if (!this.isEnabled()) return;
+
+    this.addEvent(`mcp.${eventType}`, {
+      'session.id': this.sessionId,
+      ...data
+    });
+  }
+
+  /**
+   * Record bash tool events
+   * Tracks command permission checks, allowed/denied commands, and execution
+   */
+  recordBashEvent(eventType, data = {}) {
+    if (!this.isEnabled()) return;
+
+    this.addEvent(`bash.${eventType}`, {
+      'session.id': this.sessionId,
+      ...data
+    });
+  }
+
   setAttributes(attributes) {
     // For simplicity, just log attributes when no active span
     if (this.telemetry && this.telemetry.enableConsole) {

package/src/tools/bash.js

@@ -31,16 +31,18 @@ export const bashTool = (options = {}) => {
     bashConfig = {},
     debug = false,
     cwd,
-    allowedFolders = []
+    allowedFolders = [],
+    tracer = null
   } = options;
 
-  // Create permission checker
+  // Create permission checker with tracer for telemetry
   const permissionChecker = new BashPermissionChecker({
     allow: bashConfig.allow,
     deny: bashConfig.deny,
     disableDefaultAllow: bashConfig.disableDefaultAllow,
     disableDefaultDeny: bashConfig.disableDefaultDeny,
-    debug
+    debug,
+    tracer
   });
 
   // Determine default working directory