hedgequantx 2.5.12 → 2.5.14

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "hedgequantx",
-  "version": "2.5.12",
+  "version": "2.5.14",
   "description": "HedgeQuantX - Prop Futures Trading CLI",
   "main": "src/app.js",
   "bin": {

package/src/app.js

@@ -235,13 +235,14 @@ const mainMenu = async () => {
   console.log(chalk.cyan('╠' + '═'.repeat(innerWidth) + '╣'));
 
   menuRow(chalk.cyan('[1] ProjectX'), chalk.cyan('[2] Rithmic'));
-  menuRow(chalk.cyan('[3] Tradovate'), chalk.red('[X] Exit'));
+  menuRow(chalk.cyan('[3] Tradovate'), chalk.magenta('[I] AI Agent'));
+  menuRow(chalk.red('[X] Exit'), '');
 
   console.log(chalk.cyan('╚' + '═'.repeat(innerWidth) + '╝'));
 
-  const input = await prompts.textInput(chalk.cyan('Select (1/2/3/X)'));
+  const input = await prompts.textInput(chalk.cyan('Select (1/2/3/I/X)'));
 
-  const actions = { '1': 'projectx', '2': 'rithmic', '3': 'tradovate', 'x': 'exit' };
+  const actions = { '1': 'projectx', '2': 'rithmic', '3': 'tradovate', 'i': 'ai_agent', 'x': 'exit' };
   return actions[(input || '').toLowerCase()] || 'exit';
 };
 
@@ -282,6 +283,11 @@ const run = async () => {
             process.exit(0);
           }
 
+          if (choice === 'ai_agent') {
+            await aiAgentMenu();
+            continue;
+          }
+
           let service = null;
           if (choice === 'projectx') service = await projectXMenu();
           else if (choice === 'rithmic') service = await rithmicMenu();

package/src/menus/ai-agent.js

@@ -1,6 +1,6 @@
 /**
  * AI Agent Menu
- * Configure AI provider connection
+ * Configure multiple AI provider connections
  */
 
 const chalk = require('chalk');
@@ -31,62 +31,311 @@ const aiAgentMenu = async () => {
   
   console.clear();
   displayBanner();
-  drawBoxHeaderContinue('AI AGENT', boxWidth);
+  drawBoxHeaderContinue('AI AGENTS', boxWidth);
   
-  // Show current status
-  const connection = aiService.getConnection();
+  // Get all connected agents
+  const agents = aiService.getAgents();
+  const agentCount = agents.length;
   
-  if (connection) {
-    console.log(makeLine(chalk.green('STATUS: ● CONNECTED'), 'left'));
-    console.log(makeLine(chalk.white(`PROVIDER: ${connection.provider.name}`), 'left'));
-    console.log(makeLine(chalk.white(`MODEL: ${connection.model}`), 'left'));
+  if (agentCount === 0) {
+    console.log(makeLine(chalk.gray('STATUS: NO AGENTS CONNECTED'), 'left'));
   } else {
-    console.log(makeLine(chalk.gray('STATUS: ○ NOT CONNECTED'), 'left'));
+    console.log(makeLine(chalk.green(`STATUS: ${agentCount} AGENT${agentCount > 1 ? 'S' : ''} CONNECTED`), 'left'));
+    console.log(chalk.cyan('╠' + '═'.repeat(W) + '╣'));
+    
+    // List all agents
+    for (let i = 0; i < agents.length; i++) {
+      const agent = agents[i];
+      const activeMarker = agent.isActive ? chalk.yellow(' *') : '';
+      const providerColor = agent.providerId === 'anthropic' ? chalk.magenta :
+                           agent.providerId === 'openai' ? chalk.green :
+                           agent.providerId === 'openrouter' ? chalk.yellow : chalk.cyan;
+      
+      console.log(makeLine(
+        chalk.white(`[${i + 1}] `) + 
+        providerColor(agent.name) + 
+        activeMarker + 
+        chalk.gray(` - ${agent.model}`)
+      ));
+    }
   }
   
   console.log(chalk.cyan('╠' + '═'.repeat(W) + '╣'));
   
   // Menu options
-  const options = [];
-  
-  if (!connection) {
-    options.push({ label: chalk.green('[1] CONNECT PROVIDER'), value: 'connect' });
-  } else {
-    options.push({ label: chalk.cyan('[1] CHANGE PROVIDER'), value: 'connect' });
-    options.push({ label: chalk.yellow('[2] CHANGE MODEL'), value: 'model' });
-    options.push({ label: chalk.red('[3] DISCONNECT'), value: 'disconnect' });
+  console.log(makeLine(chalk.green('[+] ADD NEW AGENT')));
+  
+  if (agentCount > 0) {
+    console.log(makeLine(chalk.cyan('[S] SET ACTIVE AGENT')));
+    console.log(makeLine(chalk.yellow('[M] CHANGE MODEL')));
+    console.log(makeLine(chalk.red('[R] REMOVE AGENT')));
+    if (agentCount > 1) {
+      console.log(makeLine(chalk.red('[X] REMOVE ALL')));
+    }
   }
-  options.push({ label: chalk.gray('[<] BACK'), value: 'back' });
   
-  for (const opt of options) {
-    console.log(makeLine(opt.label, 'left'));
-  }
+  console.log(makeLine(chalk.gray('[<] BACK')));
   
   drawBoxFooter(boxWidth);
   
   const choice = await prompts.textInput(chalk.cyan('SELECT:'));
+  const input = (choice || '').toLowerCase();
   
-  switch (choice?.toLowerCase()) {
-    case '1':
+  // Handle number input (select agent for details)
+  const num = parseInt(choice);
+  if (!isNaN(num) && num >= 1 && num <= agentCount) {
+    return await showAgentDetails(agents[num - 1]);
+  }
+  
+  switch (input) {
+    case '+':
       return await showExistingTokens();
-    case '2':
-      if (connection) {
-        return await selectModel(connection.provider);
+    case 's':
+      if (agentCount > 1) {
+        return await selectActiveAgent();
       }
-      return;
-    case '3':
-      if (connection) {
-        aiService.disconnect();
-        console.log(chalk.yellow('\n  AI AGENT DISCONNECTED'));
+      return await aiAgentMenu();
+    case 'm':
+      if (agentCount > 0) {
+        return await selectAgentForModelChange();
+      }
+      return await aiAgentMenu();
+    case 'r':
+      if (agentCount > 0) {
+        return await selectAgentToRemove();
+      }
+      return await aiAgentMenu();
+    case 'x':
+      if (agentCount > 1) {
+        aiService.disconnectAll();
+        console.log(chalk.yellow('\n  ALL AGENTS REMOVED'));
         await prompts.waitForEnter();
       }
-      return;
+      return await aiAgentMenu();
     case '<':
     case 'b':
       return;
     default:
-      return;
+      return await aiAgentMenu();
+  }
+};
+
+/**
+ * Show agent details
+ */
+const showAgentDetails = async (agent) => {
+  const boxWidth = getLogoWidth();
+  const W = boxWidth - 2;
+  
+  const makeLine = (content) => {
+    const plainLen = content.replace(/\x1b\[[0-9;]*m/g, '').length;
+    const padding = W - plainLen;
+    return chalk.cyan('║') + ' ' + content + ' '.repeat(Math.max(0, padding - 1)) + chalk.cyan('║');
+  };
+  
+  console.clear();
+  displayBanner();
+  drawBoxHeaderContinue('AGENT DETAILS', boxWidth);
+  
+  const providerColor = agent.providerId === 'anthropic' ? chalk.magenta :
+                       agent.providerId === 'openai' ? chalk.green :
+                       agent.providerId === 'openrouter' ? chalk.yellow : chalk.cyan;
+  
+  console.log(makeLine(chalk.white('NAME: ') + providerColor(agent.name)));
+  console.log(makeLine(chalk.white('PROVIDER: ') + chalk.gray(agent.provider?.name || agent.providerId)));
+  console.log(makeLine(chalk.white('MODEL: ') + chalk.gray(agent.model)));
+  console.log(makeLine(chalk.white('STATUS: ') + (agent.isActive ? chalk.green('ACTIVE') : chalk.gray('STANDBY'))));
+  
+  console.log(chalk.cyan('╠' + '═'.repeat(W) + '╣'));
+  
+  if (!agent.isActive) {
+    console.log(makeLine(chalk.cyan('[A] SET AS ACTIVE')));
+  }
+  console.log(makeLine(chalk.yellow('[M] CHANGE MODEL')));
+  console.log(makeLine(chalk.red('[R] REMOVE')));
+  console.log(makeLine(chalk.gray('[<] BACK')));
+  
+  drawBoxFooter(boxWidth);
+  
+  const choice = await prompts.textInput(chalk.cyan('SELECT:'));
+  
+  switch ((choice || '').toLowerCase()) {
+    case 'a':
+      if (!agent.isActive) {
+        aiService.setActiveAgent(agent.id);
+        console.log(chalk.green(`\n  ${agent.name} IS NOW ACTIVE`));
+        await prompts.waitForEnter();
+      }
+      return await aiAgentMenu();
+    case 'm':
+      return await selectModel(agent);
+    case 'r':
+      aiService.removeAgent(agent.id);
+      console.log(chalk.yellow(`\n  ${agent.name} REMOVED`));
+      await prompts.waitForEnter();
+      return await aiAgentMenu();
+    case '<':
+    case 'b':
+      return await aiAgentMenu();
+    default:
+      return await aiAgentMenu();
+  }
+};
+
+/**
+ * Select active agent
+ */
+const selectActiveAgent = async () => {
+  const boxWidth = getLogoWidth();
+  const W = boxWidth - 2;
+  
+  const makeLine = (content) => {
+    const plainLen = content.replace(/\x1b\[[0-9;]*m/g, '').length;
+    const padding = W - plainLen;
+    return chalk.cyan('║') + ' ' + content + ' '.repeat(Math.max(0, padding - 1)) + chalk.cyan('║');
+  };
+  
+  console.clear();
+  displayBanner();
+  drawBoxHeaderContinue('SET ACTIVE AGENT', boxWidth);
+  
+  const agents = aiService.getAgents();
+  
+  for (let i = 0; i < agents.length; i++) {
+    const agent = agents[i];
+    const activeMarker = agent.isActive ? chalk.yellow(' (CURRENT)') : '';
+    const providerColor = agent.providerId === 'anthropic' ? chalk.magenta :
+                         agent.providerId === 'openai' ? chalk.green : chalk.cyan;
+    
+    console.log(makeLine(
+      chalk.white(`[${i + 1}] `) + providerColor(agent.name) + activeMarker
+    ));
+  }
+  
+  console.log(makeLine(''));
+  console.log(makeLine(chalk.gray('[<] BACK')));
+  
+  drawBoxFooter(boxWidth);
+  
+  const choice = await prompts.textInput(chalk.cyan('SELECT AGENT:'));
+  
+  if (choice === '<' || choice?.toLowerCase() === 'b') {
+    return await aiAgentMenu();
+  }
+  
+  const index = parseInt(choice) - 1;
+  if (isNaN(index) || index < 0 || index >= agents.length) {
+    return await aiAgentMenu();
+  }
+  
+  aiService.setActiveAgent(agents[index].id);
+  console.log(chalk.green(`\n  ${agents[index].name} IS NOW ACTIVE`));
+  await prompts.waitForEnter();
+  return await aiAgentMenu();
+};
+
+/**
+ * Select agent to change model
+ */
+const selectAgentForModelChange = async () => {
+  const agents = aiService.getAgents();
+  
+  if (agents.length === 1) {
+    return await selectModel(agents[0]);
+  }
+  
+  const boxWidth = getLogoWidth();
+  const W = boxWidth - 2;
+  
+  const makeLine = (content) => {
+    const plainLen = content.replace(/\x1b\[[0-9;]*m/g, '').length;
+    const padding = W - plainLen;
+    return chalk.cyan('║') + ' ' + content + ' '.repeat(Math.max(0, padding - 1)) + chalk.cyan('║');
+  };
+  
+  console.clear();
+  displayBanner();
+  drawBoxHeaderContinue('SELECT AGENT TO CHANGE MODEL', boxWidth);
+  
+  for (let i = 0; i < agents.length; i++) {
+    const agent = agents[i];
+    console.log(makeLine(
+      chalk.white(`[${i + 1}] `) + chalk.cyan(agent.name) + chalk.gray(` - ${agent.model}`)
+    ));
+  }
+  
+  console.log(makeLine(''));
+  console.log(makeLine(chalk.gray('[<] BACK')));
+  
+  drawBoxFooter(boxWidth);
+  
+  const choice = await prompts.textInput(chalk.cyan('SELECT AGENT:'));
+  
+  if (choice === '<' || choice?.toLowerCase() === 'b') {
+    return await aiAgentMenu();
   }
+  
+  const index = parseInt(choice) - 1;
+  if (isNaN(index) || index < 0 || index >= agents.length) {
+    return await aiAgentMenu();
+  }
+  
+  return await selectModel(agents[index]);
+};
+
+/**
+ * Select agent to remove
+ */
+const selectAgentToRemove = async () => {
+  const agents = aiService.getAgents();
+  
+  if (agents.length === 1) {
+    aiService.removeAgent(agents[0].id);
+    console.log(chalk.yellow(`\n  ${agents[0].name} REMOVED`));
+    await prompts.waitForEnter();
+    return await aiAgentMenu();
+  }
+  
+  const boxWidth = getLogoWidth();
+  const W = boxWidth - 2;
+  
+  const makeLine = (content) => {
+    const plainLen = content.replace(/\x1b\[[0-9;]*m/g, '').length;
+    const padding = W - plainLen;
+    return chalk.cyan('║') + ' ' + content + ' '.repeat(Math.max(0, padding - 1)) + chalk.cyan('║');
+  };
+  
+  console.clear();
+  displayBanner();
+  drawBoxHeaderContinue('SELECT AGENT TO REMOVE', boxWidth);
+  
+  for (let i = 0; i < agents.length; i++) {
+    const agent = agents[i];
+    console.log(makeLine(
+      chalk.white(`[${i + 1}] `) + chalk.red(agent.name)
+    ));
+  }
+  
+  console.log(makeLine(''));
+  console.log(makeLine(chalk.gray('[<] BACK')));
+  
+  drawBoxFooter(boxWidth);
+  
+  const choice = await prompts.textInput(chalk.cyan('SELECT AGENT TO REMOVE:'));
+  
+  if (choice === '<' || choice?.toLowerCase() === 'b') {
+    return await aiAgentMenu();
+  }
+  
+  const index = parseInt(choice) - 1;
+  if (isNaN(index) || index < 0 || index >= agents.length) {
+    return await aiAgentMenu();
+  }
+  
+  aiService.removeAgent(agents[index].id);
+  console.log(chalk.yellow(`\n  ${agents[index].name} REMOVED`));
+  await prompts.waitForEnter();
+  return await aiAgentMenu();
 };
 
 // Cache for scanned tokens (avoid multiple Keychain prompts)
@@ -217,11 +466,12 @@ const showExistingTokens = async () => {
       return await showExistingTokens();
     }
     
-    // Save connection
+    // Add as new agent
     const model = provider.defaultModel;
-    await aiService.connect(selectedToken.provider, 'api_key', credentials, model);
+    const agentName = `${provider.name} (${selectedToken.source})`;
+    await aiService.addAgent(selectedToken.provider, 'api_key', credentials, model, agentName);
     
-    spinner.succeed(`CONNECTED TO ${provider.name}`);
+    spinner.succeed(`AGENT ADDED: ${provider.name}`);
     console.log(chalk.gray(`  SOURCE: ${selectedToken.source}`));
     console.log(chalk.gray(`  MODEL: ${model}`));
     
@@ -644,11 +894,11 @@ const setupConnection = async (provider, option) => {
     return await selectProviderOption(provider);
   }
   
-  // Save connection
+  // Add as new agent
   try {
     const model = credentials.model || provider.defaultModel;
-    await aiService.connect(provider.id, option.id, credentials, model);
-    spinner.succeed(`CONNECTED TO ${provider.name}`);
+    await aiService.addAgent(provider.id, option.id, credentials, model, provider.name);
+    spinner.succeed(`AGENT ADDED: ${provider.name}`);
     
     // Show available models for local providers
     if (validation.models && validation.models.length > 0) {
@@ -665,9 +915,9 @@ const setupConnection = async (provider, option) => {
 };
 
 /**
- * Select/change model for current provider
+ * Select/change model for an agent
  */
-const selectModel = async (provider) => {
+const selectModel = async (agent) => {
   const boxWidth = getLogoWidth();
   const W = boxWidth - 2;
   
@@ -679,9 +929,9 @@ const selectModel = async (provider) => {
   
   console.clear();
   displayBanner();
-  drawBoxHeaderContinue('SELECT MODEL', boxWidth);
+  drawBoxHeaderContinue(`SELECT MODEL - ${agent.name}`, boxWidth);
   
-  const models = provider.models || [];
+  const models = agent.provider?.models || [];
   
   if (models.length === 0) {
     console.log(makeLine(chalk.gray('NO PREDEFINED MODELS. ENTER MODEL NAME MANUALLY.')));
@@ -693,9 +943,7 @@ const selectModel = async (provider) => {
     if (!model || model === '<') {
       return await aiAgentMenu();
     }
-    const settings = aiService.getAISettings();
-    settings.model = model;
-    aiService.saveAISettings(settings);
+    aiService.updateAgent(agent.id, { model });
     console.log(chalk.green(`\n  MODEL CHANGED TO: ${model}`));
     await prompts.waitForEnter();
     return await aiAgentMenu();
@@ -704,7 +952,8 @@ const selectModel = async (provider) => {
   models.forEach((model, index) => {
     // Truncate long model names
     const displayModel = model.length > W - 10 ? model.substring(0, W - 13) + '...' : model;
-    console.log(makeLine(chalk.cyan(`[${index + 1}] ${displayModel}`)));
+    const currentMarker = model === agent.model ? chalk.yellow(' (CURRENT)') : '';
+    console.log(makeLine(chalk.cyan(`[${index + 1}] ${displayModel}`) + currentMarker));
   });
   
   console.log(makeLine(''));
@@ -724,9 +973,7 @@ const selectModel = async (provider) => {
   }
   
   const selectedModel = models[index];
-  const settings = aiService.getAISettings();
-  settings.model = selectedModel;
-  aiService.saveAISettings(settings);
+  aiService.updateAgent(agent.id, { model: selectedModel });
   
   console.log(chalk.green(`\n  MODEL CHANGED TO: ${selectedModel}`));
   await prompts.waitForEnter();

package/src/menus/dashboard.js

@@ -12,6 +12,7 @@ const { getCachedStats } = require('../services/stats-cache');
 const { prompts } = require('../utils');
 const aiService = require('../services/ai');
 
+
 /**
  * Dashboard menu after login
  */
@@ -49,32 +50,27 @@ const dashboardMenu = async (service) => {
   if (statsInfo) {
     console.log(chalk.cyan('╠' + '═'.repeat(W) + '╣'));
     
-    const balStr = statsInfo.balance !== null ? `$${statsInfo.balance.toLocaleString()}` : '--';
+const balStr = statsInfo.balance !== null ? `$${statsInfo.balance.toLocaleString()}` : '--';
     const balColor = statsInfo.balance !== null ? chalk.green : chalk.gray;
     
-    // AI status
-    const aiConnection = aiService.getConnection();
-    const aiStatus = aiConnection 
-      ? aiConnection.provider.name.split(' ')[0]  // Just "CLAUDE" or "OPENAI"
-      : null;
-    
-    // Build plain text for length calculation (unicode ✔ and ○ are 1 char width each)
-    // Format: "✔ CONNECTIONS: X    ✔ ACCOUNTS: X    ✔ BALANCE: $X    ○ AI: NONE"
-    const plainText = `* CONNECTIONS: ${statsInfo.connections}    * ACCOUNTS: ${statsInfo.accounts}    * BALANCE: ${balStr}    * AI: ${aiStatus || 'NONE'}`;
+    // Build plain text for length calculation
+    // Format: "✔ CONNECTIONS: X    ✔ ACCOUNTS: X    ✔ BALANCE: $X    ✔ AI: CONNECTED"
+    const aiText = aiConnected ? 'CONNECTED' : 'NONE';
+    const plainText = `* CONNECTIONS: ${statsInfo.connections}    * ACCOUNTS: ${statsInfo.accounts}    * BALANCE: ${balStr}    * AI: ${aiText}`;
     const statsLen = plainText.length;
     const statsLeftPad = Math.max(0, Math.floor((W - statsLen) / 2));
     const statsRightPad = Math.max(0, W - statsLen - statsLeftPad);
     
     // Build with unicode icons and colors
     const checkIcon = chalk.yellow('✔ ');
-    const aiIcon = aiStatus ? chalk.magenta('✔ ') : chalk.gray('○ ');
-    const aiText = aiStatus ? chalk.magenta(aiStatus) : chalk.gray('NONE');
+    const aiIcon = aiConnected ? chalk.magenta('✔ ') : chalk.gray('○ ');
+    const aiTextColored = aiConnected ? chalk.magenta('CONNECTED') : chalk.gray('NONE');
     
     console.log(chalk.cyan('║') + ' '.repeat(statsLeftPad) +
       checkIcon + chalk.white(`CONNECTIONS: ${statsInfo.connections}`) + '    ' +
       checkIcon + chalk.white(`ACCOUNTS: ${statsInfo.accounts}`) + '    ' +
       checkIcon + chalk.white('BALANCE: ') + balColor(balStr) + '    ' +
-      aiIcon + chalk.white('AI: ') + aiText +
+      aiIcon + chalk.white('AI: ') + aiTextColored +
       ' '.repeat(statsRightPad) + chalk.cyan('║'));
   }