hedgequantx 2.6.98 → 2.6.99

package/package.json

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

package/src/menus/ai-agent.js

@@ -782,9 +782,7 @@ const getOAuthConfig = (providerId) => {
 
 /**
  * Setup OAuth connection for any provider with OAuth support
- * Automatically detects environment and uses the appropriate method:
- * - Local (PC/Mac): Uses CLIProxyAPI for automatic token management
- * - Remote (VPS/Server): Uses cli.hedgequantx.com as OAuth relay
+ * Unified flow: copy code from URL (works on local and VPS)
  */
 const setupOAuthConnection = async (provider) => {
   const config = getOAuthConfig(provider.id);
@@ -794,47 +792,8 @@ const setupOAuthConnection = async (provider) => {
     return await selectProviderOption(provider);
   }
   
-  // Check if we can open a browser locally
-  const canUseBrowser = proxyManager.canOpenBrowser();
-  const isServer = proxyManager.isServerEnvironment();
-  
-  // If on server or can't open browser, use remote OAuth
-  if (isServer || !canUseBrowser) {
-    // Only anthropic is supported for remote OAuth currently
-    if (provider.id === 'anthropic') {
-      return await setupRemoteOAuth(provider, config);
-    } else {
-      // For other providers on VPS, show message
-      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('SERVER DETECTED', boxWidth);
-      
-      console.log(makeLine(chalk.yellow('VPS/SERVER ENVIRONMENT DETECTED')));
-      console.log(makeLine(''));
-      console.log(makeLine(chalk.white('OAuth for this provider requires a browser.')));
-      console.log(makeLine(''));
-      console.log(makeLine(chalk.white('OPTIONS:')));
-      console.log(makeLine(chalk.cyan('1. Use Claude (supports remote OAuth)')));
-      console.log(makeLine(chalk.cyan('2. Use API Key instead of OAuth')));
-      console.log(makeLine(chalk.cyan('3. Run this on a local machine first')));
-      console.log(makeLine(''));
-      
-      drawBoxFooter(boxWidth);
-      await prompts.waitForEnter();
-      return await selectProviderOption(provider);
-    }
-  }
-  
-  // Local machine - use CLIProxyAPI
-  return await setupProxyOAuth(provider, config);
+  // Use unified manual flow for all providers and environments
+  return await setupRemoteOAuth(provider, config);
 };
 
 /**
@@ -851,8 +810,24 @@ const setupRemoteOAuth = async (provider, config) => {
     return chalk.cyan('║') + ' ' + content + ' '.repeat(Math.max(0, padding - 1)) + chalk.cyan('║');
   };
   
-  // Generate OAuth URL using the existing oauth module
-  const authResult = oauthAnthropic.authorize('max');
+  // Get the right OAuth module for this provider
+  const oauthModules = {
+    anthropic: oauthAnthropic,
+    openai: oauthOpenai,
+    gemini: oauthGemini,
+    qwen: oauthQwen,
+    iflow: oauthIflow
+  };
+  
+  const oauthModule = oauthModules[provider.id];
+  if (!oauthModule) {
+    console.log(chalk.red(`OAuth not supported for ${provider.id}`));
+    await prompts.waitForEnter();
+    return await selectProviderOption(provider);
+  }
+  
+  // Generate OAuth URL using the provider's oauth module
+  const authResult = oauthModule.authorize(config.optionId || 'max');
   const url = authResult.url;
   const verifier = authResult.verifier;
   
@@ -895,7 +870,7 @@ const setupRemoteOAuth = async (provider, config) => {
   // Exchange code for tokens
   const spinner = ora({ text: 'Exchanging code for tokens...', color: 'cyan' }).start();
   
-  const result = await oauthAnthropic.exchange(code.trim(), verifier);
+  const result = await oauthModule.exchange(code.trim(), verifier);
   
   if (result.type === 'failed') {
     spinner.fail(`Authentication failed: ${result.error || 'Invalid code'}`);
@@ -922,8 +897,8 @@ const setupRemoteOAuth = async (provider, config) => {
   
   let models = [];
   try {
-    const { fetchAnthropicModelsOAuth } = require('../services/ai/client');
-    models = await fetchAnthropicModelsOAuth(result.access);
+    const { fetchModelsWithOAuth } = require('../services/ai/client');
+    models = await fetchModelsWithOAuth(provider.id, result.access);
   } catch (e) {
     // Ignore - will use defaults
   }