hedgequantx 2.6.73 → 2.6.75

package/package.json

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

package/src/menus/ai-agent.js

@@ -286,7 +286,8 @@ const selectActiveAgent = async () => {
   
   const choice = await prompts.textInput(chalk.cyan('SELECT AGENT:'));
   
-  if (choice === '<' || choice?.toLowerCase() === 'b') {
+  // Empty input or < = go back
+  if (!choice || choice.trim() === '' || choice === '<' || choice?.toLowerCase() === 'b') {
     return await aiAgentMenu();
   }
   
@@ -340,7 +341,8 @@ const selectAgentForModelChange = async () => {
   
   const choice = await prompts.textInput(chalk.cyan('SELECT AGENT:'));
   
-  if (choice === '<' || choice?.toLowerCase() === 'b') {
+  // Empty input or < = go back
+  if (!choice || choice.trim() === '' || choice === '<' || choice?.toLowerCase() === 'b') {
     return await aiAgentMenu();
   }
   
@@ -393,7 +395,8 @@ const selectAgentToRemove = async () => {
   
   const choice = await prompts.textInput(chalk.cyan('SELECT AGENT TO REMOVE:'));
   
-  if (choice === '<' || choice?.toLowerCase() === 'b') {
+  // Empty input or < = go back
+  if (!choice || choice.trim() === '' || choice === '<' || choice?.toLowerCase() === 'b') {
     return await aiAgentMenu();
   }
   
@@ -898,7 +901,7 @@ const setupBrowserOAuth = async (provider, config) => {
     return await selectProviderOption(provider);
   }
   
-  spinner.succeed('AUTHENTICATION SUCCESSFUL');
+  spinner.text = 'FETCHING AVAILABLE MODELS...';
   
   // Store OAuth credentials
   const credentials = {
@@ -916,15 +919,35 @@ const setupBrowserOAuth = async (provider, config) => {
     credentials.apiKey = result.apiKey;
   }
   
-  // Use default model for OAuth providers (they typically have limited model access)
+  // Fetch available models for the provider
+  let models = [];
+  try {
+    const { fetchModelsWithOAuth } = require('../services/ai/client');
+    models = await fetchModelsWithOAuth(provider.id, result.access);
+  } catch (e) {
+    // Fallback to default models if fetch fails
+  }
+  
+  // Fallback default models if API doesn't return list
   const defaultModels = {
-    anthropic: 'claude-sonnet-4-20250514',
-    openai: 'gpt-4o',
-    gemini: 'gemini-2.5-pro',
-    iflow: 'deepseek-v3'
+    anthropic: ['claude-sonnet-4-20250514', 'claude-opus-4-20250514', 'claude-3-5-sonnet-20241022', 'claude-3-5-haiku-20241022'],
+    openai: ['gpt-4o', 'gpt-4o-mini', 'gpt-4-turbo', 'o1', 'o1-mini', 'o3-mini'],
+    gemini: ['gemini-2.5-pro', 'gemini-2.5-flash', 'gemini-2.0-flash', 'gemini-1.5-pro'],
+    iflow: ['deepseek-v3', 'deepseek-chat', 'kimi', 'glm-4']
   };
   
-  const selectedModel = defaultModels[provider.id] || 'default';
+  if (!models || models.length === 0) {
+    models = defaultModels[provider.id] || ['default'];
+    spinner.warn('USING DEFAULT MODEL LIST');
+  } else {
+    spinner.succeed(`FOUND ${models.length} MODELS`);
+  }
+  
+  // Let user select model
+  const selectedModel = await selectModelFromList(models, config.name);
+  if (!selectedModel) {
+    return await selectProviderOption(provider);
+  }
   
   // Add agent with OAuth credentials
   try {
@@ -1037,7 +1060,7 @@ const setupDeviceFlowOAuth = async (provider, config) => {
     return await selectProviderOption(provider);
   }
   
-  pollSpinner.succeed('AUTHENTICATION SUCCESSFUL');
+  pollSpinner.text = 'FETCHING AVAILABLE MODELS...';
   
   // Store OAuth credentials
   const credentials = {
@@ -1049,7 +1072,16 @@ const setupDeviceFlowOAuth = async (provider, config) => {
     }
   };
   
-  const selectedModel = 'qwen3-coder-plus';
+  // Default models for Qwen
+  const defaultModels = ['qwen3-coder-plus', 'qwen3-235b', 'qwen-max', 'qwen-plus', 'qwen-turbo'];
+  
+  pollSpinner.succeed('AUTHENTICATION SUCCESSFUL');
+  
+  // Let user select model
+  const selectedModel = await selectModelFromList(defaultModels, config.name);
+  if (!selectedModel) {
+    return await selectProviderOption(provider);
+  }
   
   // Add agent with OAuth credentials
   try {
@@ -1292,7 +1324,8 @@ const selectModelFromList = async (models, providerName) => {
   
   const choice = await prompts.textInput(chalk.cyan('SELECT MODEL:'));
   
-  if (choice === '<' || choice?.toLowerCase() === 'b') {
+  // Empty input or < = go back
+  if (!choice || choice.trim() === '' || choice === '<' || choice?.toLowerCase() === 'b') {
     return null;
   }
   
@@ -1417,7 +1450,8 @@ const selectModel = async (agent) => {
   
   const choice = await prompts.textInput(chalk.cyan('SELECT MODEL:'));
   
-  if (choice === '<' || choice?.toLowerCase() === 'b') {
+  // Empty input or < = go back
+  if (!choice || choice.trim() === '' || choice === '<' || choice?.toLowerCase() === 'b') {
     return await aiAgentMenu();
   }
   

package/src/services/ai/client.js

@@ -565,6 +565,47 @@ const fetchOpenAIModels = async (endpoint, apiKey) => {
   }
 };
 
+/**
+ * Fetch available models for OAuth-authenticated providers
+ * @param {string} providerId - Provider ID (anthropic, openai, gemini, etc.)
+ * @param {string} accessToken - OAuth access token
+ * @returns {Promise<Array|null>} Array of model IDs or null on error
+ */
+const fetchModelsWithOAuth = async (providerId, accessToken) => {
+  if (!accessToken) return null;
+  
+  try {
+    switch (providerId) {
+      case 'anthropic':
+        return await fetchAnthropicModelsOAuth(accessToken);
+      
+      case 'openai':
+        // OpenAI OAuth uses the same endpoint as API key
+        return await fetchOpenAIModels('https://api.openai.com/v1', accessToken);
+      
+      case 'gemini':
+        // Gemini OAuth - try to fetch from API
+        const geminiUrl = 'https://generativelanguage.googleapis.com/v1/models';
+        const geminiHeaders = {
+          'Authorization': `Bearer ${accessToken}`
+        };
+        const geminiResponse = await makeRequest(geminiUrl, { method: 'GET', headers: geminiHeaders, timeout: 10000 });
+        if (geminiResponse.models && Array.isArray(geminiResponse.models)) {
+          return geminiResponse.models
+            .filter(m => m.supportedGenerationMethods?.includes('generateContent'))
+            .map(m => m.name.replace('models/', ''))
+            .filter(Boolean);
+        }
+        return null;
+      
+      default:
+        return null;
+    }
+  } catch (error) {
+    return null;
+  }
+};
+
 module.exports = {
   callAI,
   analyzeTrading,
@@ -577,5 +618,6 @@ module.exports = {
   fetchAnthropicModelsOAuth,
   fetchGeminiModels,
   fetchOpenAIModels,
+  fetchModelsWithOAuth,
   getValidOAuthToken
 };