hedgequantx 2.6.161 → 2.6.163

package/src/services/ai/index.js

@@ -7,6 +7,7 @@ const { getProviders, getProvider } = require('./providers');
 const { storage } = require('../session');
 const AISupervisor = require('./supervisor');
 const StrategySupervisor = require('./strategy-supervisor');
+const { validateConnection } = require('./validation');
 
 // In-memory cache of connections
 let connectionsCache = null;
@@ -348,263 +349,7 @@ const getCredentials = () => {
   return agent?.credentials || null;
 };
 
-/**
- * Validate API key with provider
- */
-const validateConnection = async (providerId, optionId, credentials) => {
-  const provider = getProvider(providerId);
-  if (!provider) return { valid: false, error: 'Invalid provider' };
-  
-  try {
-    switch (providerId) {
-      case 'anthropic':
-        return await validateAnthropic(credentials);
-      case 'openai':
-        return await validateOpenAI(credentials);
-      case 'gemini':
-        return await validateGemini(credentials);
-      case 'deepseek':
-        return await validateDeepSeek(credentials);
-      case 'groq':
-        return await validateGroq(credentials);
-      case 'ollama':
-        return await validateOllama(credentials);
-      case 'lmstudio':
-        return await validateLMStudio(credentials);
-      case 'custom':
-        return await validateCustom(credentials);
-      // OpenAI-compatible providers (use same validation)
-      case 'openrouter':
-        return await validateOpenRouter(credentials);
-      case 'xai':
-      case 'mistral':
-      case 'perplexity':
-      case 'together':
-      case 'qwen':
-      case 'moonshot':
-      case 'yi':
-      case 'zhipu':
-      case 'baichuan':
-        return await validateOpenAICompatible(provider, credentials);
-      default:
-        return { valid: false, error: 'Unknown provider' };
-    }
-  } catch (error) {
-    return { valid: false, error: error.message };
-  }
-};
-
-// Validation functions for each provider
-const validateAnthropic = async (credentials) => {
-  try {
-    const token = credentials.apiKey || credentials.sessionKey || credentials.accessToken;
-    if (!token) return { valid: false, error: 'No API key provided' };
-    
-    // Validate by fetching models from API - this proves the token works
-    const response = await fetch('https://api.anthropic.com/v1/models', {
-      method: 'GET',
-      headers: {
-        'x-api-key': token,
-        'anthropic-version': '2023-06-01'
-      }
-    });
-    
-    if (response.ok) {
-      const data = await response.json();
-      if (data.data && Array.isArray(data.data) && data.data.length > 0) {
-        return { valid: true, tokenType: 'api_key' };
-      }
-      return { valid: false, error: 'API returned no models' };
-    }
-    
-    const error = await response.json();
-    return { valid: false, error: error.error?.message || 'Invalid API key' };
-  } catch (e) {
-    return { valid: false, error: e.message };
-  }
-};
-
-const validateOpenAI = async (credentials) => {
-  try {
-    const response = await fetch('https://api.openai.com/v1/models', {
-      headers: {
-        'Authorization': `Bearer ${credentials.apiKey || credentials.accessToken}`
-      }
-    });
-    
-    if (response.ok) {
-      return { valid: true };
-    }
-    
-    return { valid: false, error: 'Invalid API key' };
-  } catch (e) {
-    return { valid: false, error: e.message };
-  }
-};
-
-const validateGemini = async (credentials) => {
-  try {
-    const response = await fetch(
-      `https://generativelanguage.googleapis.com/v1/models?key=${credentials.apiKey}`
-    );
-    
-    if (response.ok) {
-      return { valid: true };
-    }
-    
-    return { valid: false, error: 'Invalid API key' };
-  } catch (e) {
-    return { valid: false, error: e.message };
-  }
-};
-
-const validateDeepSeek = async (credentials) => {
-  try {
-    const response = await fetch('https://api.deepseek.com/v1/models', {
-      headers: {
-        'Authorization': `Bearer ${credentials.apiKey}`
-      }
-    });
-    
-    if (response.ok) {
-      return { valid: true };
-    }
-    
-    return { valid: false, error: 'Invalid API key' };
-  } catch (e) {
-    return { valid: false, error: e.message };
-  }
-};
-
-const validateGroq = async (credentials) => {
-  try {
-    const response = await fetch('https://api.groq.com/openai/v1/models', {
-      headers: {
-        'Authorization': `Bearer ${credentials.apiKey}`
-      }
-    });
-    
-    if (response.ok) {
-      return { valid: true };
-    }
-    
-    return { valid: false, error: 'Invalid API key' };
-  } catch (e) {
-    return { valid: false, error: e.message };
-  }
-};
-
-const validateOllama = async (credentials) => {
-  try {
-    const endpoint = credentials.endpoint || 'http://localhost:11434';
-    const response = await fetch(`${endpoint}/api/tags`);
-    
-    if (response.ok) {
-      const data = await response.json();
-      return { 
-        valid: true, 
-        models: data.models?.map(m => m.name) || [] 
-      };
-    }
-    
-    return { valid: false, error: 'Cannot connect to Ollama' };
-  } catch (e) {
-    return { valid: false, error: 'Ollama not running. Start with: ollama serve' };
-  }
-};
-
-const validateCustom = async (credentials) => {
-  try {
-    const response = await fetch(`${credentials.endpoint}/models`, {
-      headers: credentials.apiKey ? {
-        'Authorization': `Bearer ${credentials.apiKey}`
-      } : {}
-    });
-    
-    if (response.ok) {
-      return { valid: true };
-    }
-    
-    return { valid: false, error: 'Cannot connect to endpoint' };
-  } catch (e) {
-    return { valid: false, error: e.message };
-  }
-};
-
-const validateOpenRouter = async (credentials) => {
-  try {
-    const response = await fetch('https://openrouter.ai/api/v1/models', {
-      headers: {
-        'Authorization': `Bearer ${credentials.apiKey}`
-      }
-    });
-    
-    if (response.ok) {
-      return { valid: true };
-    }
-    
-    return { valid: false, error: 'Invalid API key' };
-  } catch (e) {
-    return { valid: false, error: e.message };
-  }
-};
-
-const validateLMStudio = async (credentials) => {
-  try {
-    const endpoint = credentials.endpoint || 'http://localhost:1234/v1';
-    const response = await fetch(`${endpoint}/models`);
-    
-    if (response.ok) {
-      const data = await response.json();
-      return { 
-        valid: true, 
-        models: data.data?.map(m => m.id) || [] 
-      };
-    }
-    
-    return { valid: false, error: 'Cannot connect to LM Studio' };
-  } catch (e) {
-    return { valid: false, error: 'LM Studio not running. Start local server first.' };
-  }
-};
-
-const validateOpenAICompatible = async (provider, credentials) => {
-  try {
-    const endpoint = provider.endpoint;
-    const response = await fetch(`${endpoint}/models`, {
-      headers: {
-        'Authorization': `Bearer ${credentials.apiKey}`,
-        'Content-Type': 'application/json'
-      }
-    });
-    
-    if (response.ok) {
-      return { valid: true };
-    }
-    
-    // Some providers don't have /models endpoint, try a simple chat
-    const chatResponse = await fetch(`${endpoint}/chat/completions`, {
-      method: 'POST',
-      headers: {
-        'Authorization': `Bearer ${credentials.apiKey}`,
-        'Content-Type': 'application/json'
-      },
-      body: JSON.stringify({
-        model: provider.defaultModel,
-        messages: [{ role: 'user', content: 'hi' }],
-        max_tokens: 5
-      })
-    });
-    
-    if (chatResponse.ok) {
-      return { valid: true };
-    }
-    
-    return { valid: false, error: 'Invalid API key or endpoint' };
-  } catch (e) {
-    return { valid: false, error: e.message };
-  }
-};
+// validateConnection imported from ./validation
 
 module.exports = {
   // Provider info

package/src/services/ai/providers/direct-providers.js

@@ -0,0 +1,323 @@
+/**
+ * Direct AI Providers Configuration
+ * @module services/ai/providers/direct-providers
+ * 
+ * Provider configurations that connect directly to each provider's API
+ */
+
+const DIRECT_PROVIDERS = {
+  anthropic: {
+    id: 'anthropic',
+    name: 'CLAUDE (ANTHROPIC)',
+    description: 'Pro/Max or API Key',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'oauth_max',
+        label: 'CLAUDE PRO/MAX (OAUTH)',
+        description: ['Login with your Claude subscription', 'Unlimited usage with your plan'],
+        fields: ['oauth'],
+        authType: 'oauth'
+      },
+      {
+        id: 'api_key',
+        label: 'API KEY (PAY-PER-USE)',
+        description: ['Get key at console.anthropic.com', '~$0.10 per trading session'],
+        fields: ['apiKey'],
+        url: 'https://console.anthropic.com/settings/keys'
+      }
+    ],
+    endpoint: 'https://api.anthropic.com/v1'
+  },
+  
+  openai: {
+    id: 'openai',
+    name: 'OPENAI (GPT-4/5)',
+    description: 'Plus/Pro or API Key',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'oauth_plus',
+        label: 'PLUS/PRO SUBSCRIPTION (OAUTH)',
+        description: ['Login with your ChatGPT account', 'Unlimited with your plan'],
+        fields: ['oauth'],
+        authType: 'oauth'
+      },
+      {
+        id: 'api_key',
+        label: 'API KEY (PAY-PER-USE)',
+        description: ['Get key at platform.openai.com', '~$0.15 per trading session'],
+        fields: ['apiKey'],
+        url: 'https://platform.openai.com/api-keys'
+      }
+    ],
+    endpoint: 'https://api.openai.com/v1'
+  },
+  
+  gemini: {
+    id: 'gemini',
+    name: 'GEMINI (GOOGLE)',
+    description: 'Advanced or API Key',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'oauth_advanced',
+        label: 'ADVANCED SUBSCRIPTION (OAUTH)',
+        description: ['Login with your Google account', 'Unlimited with your plan'],
+        fields: ['oauth'],
+        authType: 'oauth'
+      },
+      {
+        id: 'api_key',
+        label: 'API KEY (FREE TIER)',
+        description: ['Get key at aistudio.google.com', 'Free tier: 60 requests/min'],
+        fields: ['apiKey'],
+        url: 'https://aistudio.google.com/apikey'
+      }
+    ],
+    endpoint: 'https://generativelanguage.googleapis.com/v1'
+  },
+  
+  deepseek: {
+    id: 'deepseek',
+    name: 'DEEPSEEK',
+    description: 'Very cheap & capable',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'api_key',
+        label: 'API KEY (VERY CHEAP)',
+        description: ['Get key at platform.deepseek.com', '~$0.02 per trading session'],
+        fields: ['apiKey'],
+        url: 'https://platform.deepseek.com'
+      }
+    ],
+    endpoint: 'https://api.deepseek.com/v1'
+  },
+  
+  groq: {
+    id: 'groq',
+    name: 'GROQ',
+    description: 'Ultra fast inference',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'api_key',
+        label: 'API KEY (FREE TIER)',
+        description: ['Get key at console.groq.com', 'Generous free tier', 'Ultra low latency'],
+        fields: ['apiKey'],
+        url: 'https://console.groq.com/keys'
+      }
+    ],
+    endpoint: 'https://api.groq.com/openai/v1'
+  },
+
+  xai: {
+    id: 'xai',
+    name: 'GROK (XAI)',
+    description: 'Elon Musk\'s Grok AI',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'api_key',
+        label: 'API KEY',
+        description: ['Get key at console.x.ai', 'Grok models from xAI'],
+        fields: ['apiKey'],
+        url: 'https://console.x.ai'
+      }
+    ],
+    endpoint: 'https://api.x.ai/v1'
+  },
+
+  mistral: {
+    id: 'mistral',
+    name: 'MISTRAL',
+    description: 'European AI leader',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'api_key',
+        label: 'API KEY',
+        description: ['Get key at console.mistral.ai', 'Fast European models'],
+        fields: ['apiKey'],
+        url: 'https://console.mistral.ai'
+      }
+    ],
+    endpoint: 'https://api.mistral.ai/v1'
+  },
+
+  perplexity: {
+    id: 'perplexity',
+    name: 'PERPLEXITY',
+    description: 'Real-time web search AI',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'api_key',
+        label: 'API KEY',
+        description: ['Get key at perplexity.ai/settings/api', 'Real-time market news & data', 'Web search integrated'],
+        fields: ['apiKey'],
+        url: 'https://www.perplexity.ai/settings/api'
+      }
+    ],
+    endpoint: 'https://api.perplexity.ai'
+  },
+
+  together: {
+    id: 'together',
+    name: 'TOGETHER AI',
+    description: 'Open source models, fast & cheap',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'api_key',
+        label: 'API KEY',
+        description: ['Get key at api.together.xyz', '100+ open source models', 'Fast inference, good pricing'],
+        fields: ['apiKey'],
+        url: 'https://api.together.xyz/settings/api-keys'
+      }
+    ],
+    endpoint: 'https://api.together.xyz/v1'
+  },
+
+  iflow: {
+    id: 'iflow',
+    name: 'IFLOW',
+    description: 'Subscription (OAuth)',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'oauth_sub',
+        label: 'SUBSCRIPTION (OAUTH)',
+        description: ['Login with your iFlow account', 'Access to DeepSeek, Kimi, GLM & more'],
+        fields: ['oauth'],
+        authType: 'oauth'
+      }
+    ],
+    endpoint: 'https://api.iflow.com/v1'
+  },
+
+  qwen: {
+    id: 'qwen',
+    name: 'QWEN (ALIBABA)',
+    description: 'Chat subscription or API Key',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'oauth_chat',
+        label: 'CHAT SUBSCRIPTION (OAUTH)',
+        description: ['Login with your Qwen account', 'Unlimited with your plan'],
+        fields: ['oauth'],
+        authType: 'oauth'
+      },
+      {
+        id: 'api_key',
+        label: 'API KEY (DASHSCOPE)',
+        description: ['Get key at dashscope.aliyun.com', 'Qwen2.5 models', 'Very competitive pricing'],
+        fields: ['apiKey'],
+        url: 'https://dashscope.console.aliyun.com/apiKey'
+      }
+    ],
+    endpoint: 'https://dashscope.aliyuncs.com/compatible-mode/v1'
+  },
+
+  moonshot: {
+    id: 'moonshot',
+    name: 'MOONSHOT (KIMI)',
+    description: '200K context window',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'api_key',
+        label: 'API KEY',
+        description: ['Get key at platform.moonshot.cn', 'Up to 200K context', 'Good for long documents'],
+        fields: ['apiKey'],
+        url: 'https://platform.moonshot.cn/console/api-keys'
+      }
+    ],
+    endpoint: 'https://api.moonshot.cn/v1'
+  },
+
+  yi: {
+    id: 'yi',
+    name: '01.AI (YI)',
+    description: 'Yi models by Kai-Fu Lee',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'api_key',
+        label: 'API KEY',
+        description: ['Get key at platform.01.ai', 'Yi-Large: GPT-4 level', 'Affordable pricing'],
+        fields: ['apiKey'],
+        url: 'https://platform.01.ai'
+      }
+    ],
+    endpoint: 'https://api.01.ai/v1'
+  },
+
+  zhipu: {
+    id: 'zhipu',
+    name: 'ZHIPU AI (GLM)',
+    description: 'ChatGLM models',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'api_key',
+        label: 'API KEY',
+        description: ['Get key at open.bigmodel.cn', 'ChatGLM-4 models', 'Strong multilingual'],
+        fields: ['apiKey'],
+        url: 'https://open.bigmodel.cn/usercenter/apikeys'
+      }
+    ],
+    endpoint: 'https://open.bigmodel.cn/api/paas/v4'
+  },
+
+  baichuan: {
+    id: 'baichuan',
+    name: 'BAICHUAN',
+    description: 'Multilingual AI model',
+    category: 'direct',
+    models: [],
+    defaultModel: null,
+    options: [
+      {
+        id: 'api_key',
+        label: 'API KEY',
+        description: ['Get key at platform.baichuan-ai.com', 'Strong multilingual support', 'Competitive pricing'],
+        fields: ['apiKey'],
+        url: 'https://platform.baichuan-ai.com/console/apikey'
+      }
+    ],
+    endpoint: 'https://api.baichuan-ai.com/v1'
+  },
+};
+
+module.exports = { DIRECT_PROVIDERS };