hedgequantx 2.6.75 → 2.6.76

package/package.json

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

package/src/menus/ai-agent.js

@@ -858,13 +858,22 @@ const setupBrowserOAuth = async (provider, config) => {
   } else {
     console.log(makeLine(chalk.yellow('COULD NOT OPEN BROWSER (VPS/SSH?)')));
     console.log(makeLine(''));
-    console.log(makeLine(chalk.white('AFTER LOGGING IN, YOU WILL SEE A CODE')));
-    console.log(makeLine(chalk.white('COPY THE ENTIRE CODE AND PASTE IT BELOW')));
-    console.log(makeLine(''));
-    if (config.codeFormat) {
-      console.log(makeLine(chalk.white(`THE CODE LOOKS LIKE: ${config.codeFormat}`)));
+    
+    // Different instructions for different providers
+    if (provider.id === 'anthropic') {
+      console.log(makeLine(chalk.white('AFTER LOGGING IN, YOU WILL SEE A CODE')));
+      console.log(makeLine(chalk.white('COPY THE ENTIRE CODE AND PASTE IT BELOW')));
       console.log(makeLine(''));
+      console.log(makeLine(chalk.white('THE CODE LOOKS LIKE: abc123...#xyz789...')));
+    } else {
+      // Gemini, OpenAI, iFlow redirect to localhost - need full URL
+      console.log(makeLine(chalk.white('AFTER LOGGING IN, YOU WILL SEE A BLANK PAGE')));
+      console.log(makeLine(chalk.white('THE URL WILL START WITH: localhost:...')));
+      console.log(makeLine(''));
+      console.log(makeLine(chalk.green('COPY THE ENTIRE URL FROM THE ADDRESS BAR')));
+      console.log(makeLine(chalk.white('AND PASTE IT BELOW')));
     }
+    console.log(makeLine(''));
     console.log(makeLine(chalk.white('TYPE < TO CANCEL')));
     drawBoxFooter(boxWidth);
     // Display URL outside the box for easy copy-paste (no line breaks)
@@ -875,24 +884,41 @@ const setupBrowserOAuth = async (provider, config) => {
     console.log();
   }
   
-  const code = await prompts.textInput(chalk.cyan('PASTE AUTHORIZATION CODE:'));
+  const promptText = provider.id === 'anthropic' 
+    ? 'PASTE AUTHORIZATION CODE:' 
+    : 'PASTE FULL CALLBACK URL:';
+  
+  const input = await prompts.textInput(chalk.cyan(promptText));
   
-  if (!code || code === '<') {
+  if (!input || input === '<') {
     return await selectProviderOption(provider);
   }
   
+  // Extract code from URL if user pasted full callback URL
+  let code = input.trim();
+  if (code.includes('code=')) {
+    try {
+      const urlObj = new URL(code.replace('localhost', 'http://localhost'));
+      code = urlObj.searchParams.get('code') || code;
+    } catch (e) {
+      // Try regex extraction as fallback
+      const match = code.match(/[?&]code=([^&]+)/);
+      if (match) code = match[1];
+    }
+  }
+  
   // Exchange code for tokens
   const spinner = ora({ text: 'EXCHANGING CODE FOR TOKENS...', color: 'cyan' }).start();
   
   let result;
   if (provider.id === 'anthropic') {
-    result = await config.oauthModule.exchange(code.trim(), verifier);
+    result = await config.oauthModule.exchange(code, verifier);
   } else if (provider.id === 'gemini') {
-    result = await config.oauthModule.exchange(code.trim());
+    result = await config.oauthModule.exchange(code);
   } else if (provider.id === 'iflow') {
-    result = await config.oauthModule.exchange(code.trim(), redirectUri);
+    result = await config.oauthModule.exchange(code, redirectUri);
   } else {
-    result = await config.oauthModule.exchange(code.trim(), verifier);
+    result = await config.oauthModule.exchange(code, verifier);
   }
   
   if (result.type === 'failed') {