hedgequantx 2.6.119 → 2.6.120

package/package.json

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

package/src/menus/ai-agent.js

@@ -1104,71 +1104,82 @@ const setupProxyOAuth = async (provider, config) => {
     return await selectProviderOption(provider);
   }
   
-  // Step 5: Fetch models from CLIProxyAPI
-  const modelSpinner = ora({ text: 'Fetching available models from API...', color: 'cyan' }).start();
-  
-  let models = [];
+  // Wrap entire post-auth flow in try-catch to catch any uncaught errors
   try {
-    models = await proxyManager.getModels();
-    modelSpinner.succeed(`Found ${models.length} models`);
-  } catch (error) {
-    modelSpinner.fail(`Failed to fetch models: ${error.message}`);
+    // Step 5: Fetch models from CLIProxyAPI
+    const modelSpinner = ora({ text: 'Fetching available models from API...', color: 'cyan' }).start();
+    
+    let models = [];
+    try {
+      models = await proxyManager.getModels();
+      modelSpinner.succeed(`Found ${models.length} models`);
+    } catch (error) {
+      modelSpinner.fail(`Failed to fetch models: ${error.message}`);
+      await prompts.waitForEnter();
+      return await selectProviderOption(provider);
+    }
+    
+    // Filter models for this provider
+    const providerPrefixes = {
+      anthropic: ['claude'],
+      openai: ['gpt', 'o1', 'o3', 'o4'],
+      gemini: ['gemini'],
+      qwen: ['qwen'],
+      iflow: ['deepseek', 'kimi', 'glm']
+    };
+    
+    const prefixes = providerPrefixes[provider.id] || [];
+    const filteredModels = models.filter(m => {
+      const modelLower = m.toLowerCase();
+      return prefixes.some(p => modelLower.includes(p));
+    });
+    
+    // Use filtered models if available, otherwise use all models
+    const availableModels = filteredModels.length > 0 ? filteredModels : models;
+    
+    if (!availableModels || availableModels.length === 0) {
+      console.log();
+      console.log(chalk.red('  ERROR: No models found for this provider'));
+      console.log(chalk.gray('  Make sure your subscription is active.'));
+      console.log();
+      await prompts.waitForEnter();
+      return await selectProviderOption(provider);
+    }
+    
+    // Step 6: Let user select model
+    const selectedModel = await selectModelFromList(availableModels, config.name);
+    if (!selectedModel) {
+      return await selectProviderOption(provider);
+    }
+    
+    // Step 7: Save agent config (CLIProxyAPI stores the actual tokens)
+    // We just save a reference to use the proxy
+    const credentials = {
+      useProxy: true,
+      provider: provider.id
+    };
+    
+    try {
+      await aiService.addAgent(provider.id, config.optionId, credentials, selectedModel, config.agentName);
+      
+      console.log(chalk.green(`\n  CONNECTED TO ${config.name}`));
+      console.log(chalk.white(`  MODEL: ${selectedModel}`));
+      console.log(chalk.white('  UNLIMITED USAGE WITH YOUR SUBSCRIPTION'));
+    } catch (error) {
+      console.log(chalk.red(`\n  FAILED TO SAVE: ${error.message}`));
+    }
+    
     await prompts.waitForEnter();
-    return await selectProviderOption(provider);
-  }
-  
-  // Filter models for this provider
-  const providerPrefixes = {
-    anthropic: ['claude'],
-    openai: ['gpt', 'o1', 'o3', 'o4'],
-    gemini: ['gemini'],
-    qwen: ['qwen'],
-    iflow: ['deepseek', 'kimi', 'glm']
-  };
-  
-  const prefixes = providerPrefixes[provider.id] || [];
-  const filteredModels = models.filter(m => {
-    const modelLower = m.toLowerCase();
-    return prefixes.some(p => modelLower.includes(p));
-  });
-  
-  // Use filtered models if available, otherwise use all models
-  const availableModels = filteredModels.length > 0 ? filteredModels : models;
-  
-  if (!availableModels || availableModels.length === 0) {
+    return await aiAgentMenu();
+  } catch (unexpectedError) {
+    // Catch any uncaught errors in the post-auth flow
     console.log();
-    console.log(chalk.red('  ERROR: No models found for this provider'));
-    console.log(chalk.gray('  Make sure your subscription is active.'));
+    console.log(chalk.red(`  UNEXPECTED ERROR: ${unexpectedError.message}`));
+    console.log(chalk.gray(`  Stack: ${unexpectedError.stack}`));
     console.log();
     await prompts.waitForEnter();
     return await selectProviderOption(provider);
   }
-  
-  // Step 6: Let user select model
-  const selectedModel = await selectModelFromList(availableModels, config.name);
-  if (!selectedModel) {
-    return await selectProviderOption(provider);
-  }
-  
-  // Step 7: Save agent config (CLIProxyAPI stores the actual tokens)
-  // We just save a reference to use the proxy
-  const credentials = {
-    useProxy: true,
-    provider: provider.id
-  };
-  
-  try {
-    await aiService.addAgent(provider.id, config.optionId, credentials, selectedModel, config.agentName);
-    
-    console.log(chalk.green(`\n  CONNECTED TO ${config.name}`));
-    console.log(chalk.white(`  MODEL: ${selectedModel}`));
-    console.log(chalk.white('  UNLIMITED USAGE WITH YOUR SUBSCRIPTION'));
-  } catch (error) {
-    console.log(chalk.red(`\n  FAILED TO SAVE: ${error.message}`));
-  }
-  
-  await prompts.waitForEnter();
-  return await aiAgentMenu();
 };
 
 /**