hedgequantx 2.5.28 → 2.5.30

package/package.json

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

package/src/menus/ai-agent.js

@@ -181,17 +181,39 @@ const showAgentDetails = async (agent) => {
   
   console.log(makeLine(chalk.white('NAME: ') + providerColor(agent.name)));
   console.log(makeLine(chalk.white('PROVIDER: ') + chalk.white(agent.provider?.name || agent.providerId)));
-  console.log(makeLine(chalk.white('MODEL: ') + chalk.white(agent.model)));
+  console.log(makeLine(chalk.white('MODEL: ') + chalk.white(agent.model || 'N/A')));
   console.log(makeLine(chalk.white('STATUS: ') + (agent.isActive ? chalk.green('ACTIVE') : chalk.white('STANDBY'))));
   
   console.log(chalk.cyan('╠' + '═'.repeat(W) + '╣'));
   
+  // Menu in 2 columns
+  const colWidth = Math.floor(W / 2);
+  
+  const menuRow = (col1, col2 = '') => {
+    const c1Plain = col1.replace(/\x1b\[[0-9;]*m/g, '');
+    const c2Plain = col2.replace(/\x1b\[[0-9;]*m/g, '');
+    
+    const pad1Left = Math.floor((colWidth - c1Plain.length) / 2);
+    const pad1Right = colWidth - c1Plain.length - pad1Left;
+    
+    const col2Width = W - colWidth;
+    const pad2Left = Math.floor((col2Width - c2Plain.length) / 2);
+    const pad2Right = col2Width - c2Plain.length - pad2Left;
+    
+    const line = 
+      ' '.repeat(pad1Left) + col1 + ' '.repeat(pad1Right) +
+      ' '.repeat(pad2Left) + col2 + ' '.repeat(pad2Right);
+    
+    console.log(chalk.cyan('║') + line + chalk.cyan('║'));
+  };
+  
   if (!agent.isActive) {
-    console.log(makeLine(chalk.cyan('[A] SET AS ACTIVE')));
+    menuRow(chalk.cyan('[A] SET AS ACTIVE'), chalk.yellow('[M] CHANGE MODEL'));
+    menuRow(chalk.red('[R] REMOVE'), chalk.white('[<] BACK'));
+  } else {
+    menuRow(chalk.yellow('[M] CHANGE MODEL'), chalk.red('[R] REMOVE'));
+    menuRow(chalk.white('[<] BACK'), '');
   }
-  console.log(makeLine(chalk.yellow('[M] CHANGE MODEL')));
-  console.log(makeLine(chalk.red('[R] REMOVE')));
-  console.log(makeLine(chalk.white('[<] BACK')));
   
   drawBoxFooter(boxWidth);
   
@@ -912,20 +934,44 @@ const setupConnection = async (provider, option) => {
     return await selectProviderOption(provider);
   }
   
-  // Add as new agent
+  spinner.text = 'FETCHING AVAILABLE MODELS...';
+  
+  // Fetch models from real API
+  const { fetchAnthropicModels, fetchGeminiModels, fetchOpenAIModels } = require('../services/ai/client');
+  
+  let models = null;
+  if (provider.id === 'anthropic') {
+    models = await fetchAnthropicModels(credentials.apiKey);
+  } else if (provider.id === 'gemini') {
+    models = await fetchGeminiModels(credentials.apiKey);
+  } else {
+    const endpoint = credentials.endpoint || provider.endpoint;
+    models = await fetchOpenAIModels(endpoint, credentials.apiKey);
+  }
+  
+  if (!models || models.length === 0) {
+    spinner.fail('COULD NOT FETCH MODELS FROM API');
+    console.log(chalk.white('  Check your API key or network connection.'));
+    await prompts.waitForEnter();
+    return await selectProviderOption(provider);
+  }
+  
+  spinner.succeed(`FOUND ${models.length} MODELS`);
+  
+  // Let user select a model
+  const selectedModel = await selectModelFromList(models, provider.name);
+  if (!selectedModel) {
+    return await selectProviderOption(provider);
+  }
+  
+  // Add as new agent with selected model
   try {
-    const model = credentials.model || provider.defaultModel;
-    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) {
-      console.log(chalk.white(`  AVAILABLE MODELS: ${validation.models.slice(0, 5).join(', ')}`));
-    }
+    await aiService.addAgent(provider.id, option.id, credentials, selectedModel, provider.name);
     
-    console.log(chalk.white(`  USING MODEL: ${model}`));
+    console.log(chalk.green(`\n  AGENT ADDED: ${provider.name}`));
+    console.log(chalk.white(`  MODEL: ${selectedModel}`));
   } catch (error) {
-    spinner.fail(`FAILED TO SAVE: ${error.message}`);
+    console.log(chalk.red(`\n  FAILED TO SAVE: ${error.message}`));
   }
   
   await prompts.waitForEnter();