investbuddy-mcp-server 1.0.0 → 1.0.1

package/index.js

@@ -129,24 +129,38 @@ const TOOLS = [
 ];
 
 /**
- * Call InvestBuddy MCP API
+ * Map MCP tool names to API endpoints
  */
-function callMCPAPI(tool, args) {
+const TOOL_ENDPOINTS = {
+  'get_stock_prediction': '/mcp/predict',
+  'get_market_regime': '/mcp/market-regime',
+  'discover_stocks': '/mcp/discover',
+  'analyze_portfolio': '/mcp/analyze-portfolio',
+  'optimize_portfolio': '/mcp/optimize-portfolio',
+  'batch_predict': '/mcp/batch-predict'
+};
+
+/**
+ * Call InvestBuddy API
+ */
+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 = JSON.stringify(args);
 
     const options = {
       hostname: 'www.investbuddy.ai',
       port: 443,
-      path: '/mcp/call',
+      path: endpoint,
       method: 'POST',
       headers: {
         'Content-Type': 'application/json',
         'Content-Length': data.length,
-        'Authorization': `Bearer ${API_KEY}`
+        'x-api-key': API_KEY
       }
     };
 
@@ -160,13 +174,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}`));
         }
       });
     });
@@ -209,7 +219,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.1",
   "description": "Model Context Protocol server for InvestBuddy AI - Stock predictions with 79.86% accuracy",
   "main": "index.js",
   "bin": {