investbuddy-mcp-server 1.0.0 → 1.0.2

package/index.js

@@ -87,30 +87,6 @@ const TOOLS = [
       required: ['holdings']
     }
   },
-  {
-    name: 'optimize_portfolio',
-    description: 'Get AI-powered portfolio optimization recommendations',
-    inputSchema: {
-      type: 'object',
-      properties: {
-        holdings: {
-          type: 'array',
-          items: {
-            type: 'object',
-            properties: {
-              symbol: { type: 'string' },
-              shares: { type: 'number' }
-            }
-          }
-        },
-        risk_tolerance: {
-          type: 'string',
-          enum: ['conservative', 'moderate', 'aggressive']
-        }
-      },
-      required: ['holdings', 'risk_tolerance']
-    }
-  },
   {
     name: 'batch_predict',
     description: 'Get predictions for multiple stocks in a single call',
@@ -129,27 +105,43 @@ const TOOLS = [
 ];
 
 /**
- * Call InvestBuddy MCP API
+ * Map MCP tool names to API endpoints and methods
+ */
+const TOOL_ENDPOINTS = {
+  'get_stock_prediction': { path: '/mcp/predict', method: 'POST' },
+  'get_market_regime': { path: '/mcp/market-regime', method: 'GET' },
+  'discover_stocks': { path: '/mcp/discover', method: 'POST' },
+  'analyze_portfolio': { path: '/mcp/analyze-portfolio', method: 'POST' },
+  'batch_predict': { path: '/mcp/batch-predict', method: 'POST' }
+};
+
+/**
+ * Call InvestBuddy API
  */
-function callMCPAPI(tool, args) {
+function callAPI(tool, args) {
   return new Promise((resolve, reject) => {
-    const data = JSON.stringify({
-      tool: tool,
-      arguments: args
-    });
+    const endpoint = TOOL_ENDPOINTS[tool];
+    if (!endpoint) {
+      return reject(new Error(`Unknown tool: ${tool}`));
+    }
+
+    const data = endpoint.method === 'POST' ? JSON.stringify(args) : null;
 
     const options = {
       hostname: 'www.investbuddy.ai',
       port: 443,
-      path: '/mcp/call',
-      method: 'POST',
+      path: endpoint.path,
+      method: endpoint.method,
       headers: {
         'Content-Type': 'application/json',
-        'Content-Length': data.length,
-        'Authorization': `Bearer ${API_KEY}`
+        'x-api-key': API_KEY
       }
     };
 
+    if (data) {
+      options.headers['Content-Length'] = data.length;
+    }
+
     const req = https.request(options, (res) => {
       let responseData = '';
 
@@ -160,13 +152,9 @@ function callMCPAPI(tool, args) {
       res.on('end', () => {
         try {
           const parsed = JSON.parse(responseData);
-          if (parsed.success) {
-            resolve(parsed.data);
-          } else {
-            reject(new Error(parsed.error || 'API call failed'));
-          }
+          resolve(parsed);
         } catch (error) {
-          reject(error);
+          reject(new Error(`Failed to parse response: ${responseData}`));
         }
       });
     });
@@ -175,7 +163,9 @@ function callMCPAPI(tool, args) {
       reject(error);
     });
 
-    req.write(data);
+    if (data) {
+      req.write(data);
+    }
     req.end();
   });
 }
@@ -209,7 +199,7 @@ async function handleMessage(message) {
 
       case 'tools/call':
         const { name, arguments: toolArgs } = params;
-        const apiResult = await callMCPAPI(name, toolArgs || {});
+        const apiResult = await callAPI(name, toolArgs || {});
         result = {
           content: [
             {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "investbuddy-mcp-server",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "Model Context Protocol server for InvestBuddy AI - Stock predictions with 79.86% accuracy",
   "main": "index.js",
   "bin": {