hedgequantx 2.5.26 → 2.5.27

package/package.json

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

package/src/menus/ai-agent.js

@@ -1042,7 +1042,7 @@ const selectModel = async (agent) => {
   drawBoxFooter(boxWidth);
   
   // Fetch models from real API
-  const { fetchAnthropicModels, fetchAnthropicModelsOAuth, fetchOpenAIModels } = require('../services/ai/client');
+  const { fetchAnthropicModels, fetchAnthropicModelsOAuth, fetchGeminiModels, fetchOpenAIModels } = require('../services/ai/client');
   
   let models = null;
   const agentCredentials = aiService.getAgentCredentials(agent.id);
@@ -1060,6 +1060,9 @@ const selectModel = async (agent) => {
       // Standard API key
       models = await fetchAnthropicModels(token);
     }
+  } else if (agent.providerId === 'gemini') {
+    // Google Gemini API
+    models = await fetchGeminiModels(agentCredentials?.apiKey);
   } else {
     // OpenAI-compatible providers
     const endpoint = agentCredentials?.endpoint || agent.provider?.endpoint;

package/src/services/ai/client.js

@@ -367,6 +367,33 @@ const fetchAnthropicModelsOAuth = async (accessToken) => {
   }
 };
 
+/**
+ * Fetch available models from Google Gemini API
+ * @param {string} apiKey - API key
+ * @returns {Promise<Array|null>} Array of model IDs or null on error
+ * 
+ * Data source: https://generativelanguage.googleapis.com/v1/models (GET)
+ */
+const fetchGeminiModels = async (apiKey) => {
+  if (!apiKey) return null;
+  
+  const url = `https://generativelanguage.googleapis.com/v1/models?key=${apiKey}`;
+  
+  try {
+    const response = await makeRequest(url, { method: 'GET', timeout: 10000 });
+    if (response.models && Array.isArray(response.models)) {
+      // Filter only generative models and extract the model name
+      return response.models
+        .filter(m => m.supportedGenerationMethods?.includes('generateContent'))
+        .map(m => m.name.replace('models/', ''))
+        .filter(Boolean);
+    }
+    return null;
+  } catch (error) {
+    return null;
+  }
+};
+
 /**
  * Fetch available models from OpenAI-compatible API
  * @param {string} endpoint - API endpoint
@@ -407,6 +434,7 @@ module.exports = {
   callGemini,
   fetchAnthropicModels,
   fetchAnthropicModelsOAuth,
+  fetchGeminiModels,
   fetchOpenAIModels,
   getValidOAuthToken
 };

package/src/services/ai/providers/index.js

@@ -88,8 +88,8 @@ const PROVIDERS = {
     name: 'GEMINI (GOOGLE)',
     description: 'Direct connection to Gemini',
     category: 'direct',
-    models: ['gemini-1.5-pro', 'gemini-1.5-flash', 'gemini-pro'],
-    defaultModel: 'gemini-1.5-flash',
+    models: [], // Fetched from API at runtime
+    defaultModel: null, // Will use first model from API
     options: [
       {
         id: 'api_key',