fraim-framework 2.0.95 → 2.0.96

package/dist/src/local-mcp-server/stdio-server.js

@@ -1411,15 +1411,6 @@ class FraimLocalMCPServer {
         }
         const response = await this._doProxyToRemote(injectedRequest, requestId);
         const processedResponse = await this.finalizeToolResponse(injectedRequest, response, requestSessionId, requestId);
-        // Single point for usage tracking - log all tool calls
-        if (injectedRequest.method === 'tools/call' && requestSessionId && toolName) {
-            const success = !processedResponse.error;
-            this.log(`📊 Collecting usage: ${toolName} (session: ${requestSessionId}, success: ${success})`);
-            this.usageCollector.collectMCPCall(toolName, args, requestSessionId, success);
-        }
-        else if (injectedRequest.method === 'tools/call') {
-            this.log(`⚠️ Skipping usage collection: toolName=${toolName}, sessionId=${requestSessionId}`);
-        }
         this.log(`📤 ${injectedRequest.method} → ${processedResponse.error ? 'ERROR' : 'OK'}`);
         return processedResponse;
     }
@@ -1540,6 +1531,8 @@ class FraimLocalMCPServer {
                             const response = await this.handleRequest(message);
                             // Only send response if we got one (null means we handled it internally)
                             if (response) {
+                                // Collect usage for all tools/call requests before sending response
+                                this.collectUsageForResponse(message, response);
                                 process.stdout.write(JSON.stringify(response) + '\n');
                             }
                         }
@@ -1593,6 +1586,26 @@ class FraimLocalMCPServer {
         });
         this.log('✅ FRAIM Local MCP Server ready');
     }
+    /**
+     * Collect usage analytics for tools/call requests
+     */
+    collectUsageForResponse(request, response) {
+        // Only collect usage for tools/call requests
+        if (request.method !== 'tools/call') {
+            return;
+        }
+        const toolName = request.params?.name;
+        const args = request.params?.arguments || {};
+        const requestSessionId = this.extractSessionIdFromRequest(request);
+        if (toolName && requestSessionId) {
+            const success = !response.error;
+            this.log(`📊 Collecting usage: ${toolName} (session: ${requestSessionId}, success: ${success})`);
+            this.usageCollector.collectMCPCall(toolName, args, requestSessionId, success);
+        }
+        else if (request.method === 'tools/call') {
+            this.log(`⚠️ Skipping usage collection: toolName=${toolName}, sessionId=${requestSessionId}`);
+        }
+    }
     /**
      * Flush collected usage data to the local database
      */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "fraim-framework",
-  "version": "2.0.95",
+  "version": "2.0.96",
   "description": "FRAIM v2: Framework for Rigor-based AI Management - Transform from solo developer to AI manager orchestrating production-ready code with enterprise-grade discipline",
   "main": "index.js",
   "bin": {