hedgequantx 2.6.163 → 2.7.1

package/src/services/ai/index.js

@@ -1,384 +0,0 @@
-/**
- * AI Service Manager
- * Manages multiple AI provider connections
- */
-
-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;
-
-/**
- * Get AI settings from storage
- */
-const getAISettings = () => {
-  try {
-    const sessions = storage.load();
-    const aiSettings = sessions.find(s => s.type === 'ai') || { type: 'ai', agents: [] };
-    
-    // Migrate old single-agent format to multi-agent
-    if (aiSettings.provider && !aiSettings.agents) {
-      return {
-        type: 'ai',
-        agents: [{
-          id: generateAgentId(),
-          provider: aiSettings.provider,
-          option: aiSettings.option,
-          credentials: aiSettings.credentials,
-          model: aiSettings.model,
-          name: getProvider(aiSettings.provider)?.name || 'AI Agent',
-          createdAt: Date.now()
-        }],
-        activeAgentId: null
-      };
-    }
-    return aiSettings;
-  } catch {
-    return { type: 'ai', agents: [] };
-  }
-};
-
-/**
- * Save AI settings to storage
- */
-const saveAISettings = (aiSettings) => {
-  try {
-    const sessions = storage.load();
-    const otherSessions = sessions.filter(s => s.type !== 'ai');
-    
-    aiSettings.type = 'ai';
-    otherSessions.push(aiSettings);
-    
-    storage.save(otherSessions);
-    connectionsCache = null; // Invalidate cache
-  } catch (e) {
-    // Silent fail
-  }
-};
-
-/**
- * Generate unique agent ID
- */
-const generateAgentId = () => {
-  return 'agent_' + Date.now().toString(36) + Math.random().toString(36).substr(2, 5);
-};
-
-/**
- * Get all connected agents
- * IMPORTANT: Includes credentials for API calls in StrategySupervisor
- */
-const getAgents = () => {
-  const aiSettings = getAISettings();
-  const agents = aiSettings.agents || [];
-  
-  return agents.map(agent => {
-    const provider = getProvider(agent.provider);
-    return {
-      id: agent.id,
-      name: agent.name || provider?.name || 'Unknown',
-      provider: provider,
-      providerId: agent.provider,
-      option: agent.option,
-      model: agent.model || provider?.defaultModel,
-      credentials: agent.credentials, // CRITICAL: Required for API calls
-      createdAt: agent.createdAt,
-      isActive: agent.id === aiSettings.activeAgentId
-    };
-  }).filter(a => a.provider); // Filter out invalid providers
-};
-
-/**
- * Get agent count
- */
-const getAgentCount = () => {
-  const aiSettings = getAISettings();
-  return (aiSettings.agents || []).length;
-};
-
-/**
- * Check if any AI is connected
- */
-const isConnected = () => {
-  return getAgentCount() > 0;
-};
-
-/**
- * Get active agent (or first agent if none active)
- */
-const getActiveAgent = () => {
-  const aiSettings = getAISettings();
-  const agents = aiSettings.agents || [];
-  
-  if (agents.length === 0) return null;
-  
-  // Find active agent or use first one
-  const activeId = aiSettings.activeAgentId;
-  const agent = activeId 
-    ? agents.find(a => a.id === activeId) 
-    : agents[0];
-  
-  if (!agent) return null;
-  
-  const provider = getProvider(agent.provider);
-  if (!provider) return null;
-  
-  return {
-    id: agent.id,
-    name: agent.name || provider.name,
-    provider: provider,
-    providerId: agent.provider,
-    option: agent.option,
-    model: agent.model || provider.defaultModel,
-    credentials: agent.credentials,
-    connected: true
-  };
-};
-
-/**
- * Get agent by ID
- */
-const getAgent = (agentId) => {
-  const aiSettings = getAISettings();
-  const agents = aiSettings.agents || [];
-  const agent = agents.find(a => a.id === agentId);
-  
-  if (!agent) return null;
-  
-  const provider = getProvider(agent.provider);
-  if (!provider) return null;
-  
-  return {
-    id: agent.id,
-    name: agent.name || provider.name,
-    provider: provider,
-    providerId: agent.provider,
-    option: agent.option,
-    model: agent.model || provider.defaultModel,
-    credentials: agent.credentials,
-    connected: true
-  };
-};
-
-/**
- * Get agent credentials by ID
- * @param {string} agentId - Agent ID
- * @returns {Object|null} Credentials object or null
- */
-const getAgentCredentials = (agentId) => {
-  const aiSettings = getAISettings();
-  const agents = aiSettings.agents || [];
-  const agent = agents.find(a => a.id === agentId);
-  
-  if (!agent) return null;
-  return agent.credentials || null;
-};
-
-/**
- * Set active agent
- */
-const setActiveAgent = (agentId) => {
-  const aiSettings = getAISettings();
-  const agents = aiSettings.agents || [];
-  
-  if (!agents.find(a => a.id === agentId)) {
-    throw new Error('Agent not found');
-  }
-  
-  aiSettings.activeAgentId = agentId;
-  saveAISettings(aiSettings);
-};
-
-/**
- * Add a new agent (connect to provider)
- */
-const addAgent = async (providerId, optionId, credentials, model = null, customName = null) => {
-  const provider = getProvider(providerId);
-  if (!provider) {
-    throw new Error('Invalid provider');
-  }
-  
-  const option = provider.options.find(o => o.id === optionId);
-  if (!option) {
-    throw new Error('Invalid option');
-  }
-  
-  const aiSettings = getAISettings();
-  if (!aiSettings.agents) {
-    aiSettings.agents = [];
-  }
-  
-  // Create new agent
-  const agentId = generateAgentId();
-  const newAgent = {
-    id: agentId,
-    provider: providerId,
-    option: optionId,
-    credentials: credentials,
-    model: model || provider.defaultModel,
-    name: customName || provider.name,
-    createdAt: Date.now()
-  };
-  
-  aiSettings.agents.push(newAgent);
-  
-  // Set as active if first agent
-  if (aiSettings.agents.length === 1) {
-    aiSettings.activeAgentId = agentId;
-  }
-  
-  saveAISettings(aiSettings);
-  
-  // Get the full agent object
-  const agent = getAgent(agentId);
-  
-  // Notify StrategySupervisor if algo is running
-  // This ensures new agents are immediately connected to live trading
-  try {
-    StrategySupervisor.addAgent(agent);
-  } catch (e) {
-    // Supervisor might not be active - that's OK
-  }
-  
-  return agent;
-};
-
-/**
- * Remove an agent
- */
-const removeAgent = (agentId) => {
-  const aiSettings = getAISettings();
-  const agents = aiSettings.agents || [];
-  
-  const index = agents.findIndex(a => a.id === agentId);
-  if (index === -1) {
-    throw new Error('Agent not found');
-  }
-  
-  agents.splice(index, 1);
-  aiSettings.agents = agents;
-  
-  // Stop AI supervision for this agent
-  AISupervisor.stop(agentId);
-  
-  // Notify StrategySupervisor to remove agent from live trading
-  try {
-    StrategySupervisor.removeAgent(agentId);
-  } catch (e) {
-    // Supervisor might not be active - that's OK
-  }
-  
-  // If removed agent was active, set new active
-  if (aiSettings.activeAgentId === agentId) {
-    aiSettings.activeAgentId = agents.length > 0 ? agents[0].id : null;
-  }
-  
-  saveAISettings(aiSettings);
-};
-
-/**
- * Update agent settings
- */
-const updateAgent = (agentId, updates) => {
-  const aiSettings = getAISettings();
-  const agents = aiSettings.agents || [];
-  
-  const agent = agents.find(a => a.id === agentId);
-  if (!agent) {
-    throw new Error('Agent not found');
-  }
-  
-  // Apply updates
-  if (updates.name) agent.name = updates.name;
-  if (updates.model) agent.model = updates.model;
-  if (updates.credentials) agent.credentials = updates.credentials;
-  
-  saveAISettings(aiSettings);
-  return getAgent(agentId);
-};
-
-/**
- * Disconnect all agents
- */
-const disconnectAll = () => {
-  // Stop all AI supervision sessions
-  AISupervisor.stopAll();
-  
-  // Refresh StrategySupervisor to clear agents
-  try {
-    StrategySupervisor.refreshAgents();
-  } catch (e) {
-    // Supervisor might not be active
-  }
-  
-  saveAISettings({ agents: [] });
-};
-
-/**
- * Legacy: Get current connection (returns active agent)
- * @deprecated Use getActiveAgent() instead
- */
-const getConnection = () => {
-  return getActiveAgent();
-};
-
-/**
- * Legacy: Connect to a provider (adds new agent)
- * @deprecated Use addAgent() instead
- */
-const connect = async (providerId, optionId, credentials, model = null) => {
-  return addAgent(providerId, optionId, credentials, model);
-};
-
-/**
- * Legacy: Disconnect (removes all agents)
- * @deprecated Use removeAgent() or disconnectAll() instead
- */
-const disconnect = () => {
-  disconnectAll();
-};
-
-/**
- * Get credentials for active agent
- */
-const getCredentials = () => {
-  const agent = getActiveAgent();
-  return agent?.credentials || null;
-};
-
-// validateConnection imported from ./validation
-
-module.exports = {
-  // Provider info
-  getProviders,
-  getProvider,
-  
-  // Multi-agent API
-  getAgents,
-  getAgentCount,
-  getAgent,
-  getAgentCredentials,
-  getActiveAgent,
-  setActiveAgent,
-  addAgent,
-  removeAgent,
-  updateAgent,
-  disconnectAll,
-  
-  // Legacy API (for backwards compatibility)
-  isConnected,
-  getConnection,
-  connect,
-  disconnect,
-  getCredentials,
-  
-  // Validation
-  validateConnection,
-  
-  // Settings
-  getAISettings,
-  saveAISettings
-};

package/src/services/ai/oauth-anthropic.js

@@ -1,265 +0,0 @@
-/**
- * Anthropic OAuth Authentication
- * 
- * Implements OAuth 2.0 with PKCE for Anthropic Claude Pro/Max plans.
- * Based on the public OAuth flow used by OpenCode.
- * 
- * Data source: Anthropic OAuth API (https://console.anthropic.com/v1/oauth/token)
- */
-
-const crypto = require('crypto');
-const https = require('https');
-
-// Public OAuth Client ID (same as OpenCode - registered with Anthropic)
-const CLIENT_ID = '9d1c250a-e61b-44d9-88ed-5944d1962f5e';
-const REDIRECT_URI = 'https://console.anthropic.com/oauth/code/callback';
-
-/**
- * Generate PKCE code verifier and challenge
- * @returns {Object} { verifier: string, challenge: string }
- */
-const generatePKCE = () => {
-  // Generate a random 32-byte code verifier (base64url encoded)
-  const verifier = crypto.randomBytes(32)
-    .toString('base64')
-    .replace(/\+/g, '-')
-    .replace(/\//g, '_')
-    .replace(/=/g, '');
-  
-  // Generate SHA256 hash of verifier, then base64url encode it
-  const challenge = crypto.createHash('sha256')
-    .update(verifier)
-    .digest('base64')
-    .replace(/\+/g, '-')
-    .replace(/\//g, '_')
-    .replace(/=/g, '');
-  
-  return { verifier, challenge };
-};
-
-/**
- * Generate OAuth authorization URL
- * @param {'max' | 'console'} mode - 'max' for Claude Pro/Max, 'console' for API key creation
- * @returns {Object} { url: string, verifier: string }
- */
-const authorize = (mode = 'max') => {
-  const pkce = generatePKCE();
-  
-  const baseUrl = mode === 'max' 
-    ? 'https://claude.ai/oauth/authorize'
-    : 'https://console.anthropic.com/oauth/authorize';
-  
-  const url = new URL(baseUrl);
-  url.searchParams.set('code', 'true');
-  url.searchParams.set('client_id', CLIENT_ID);
-  url.searchParams.set('response_type', 'code');
-  url.searchParams.set('redirect_uri', REDIRECT_URI);
-  url.searchParams.set('scope', 'org:create_api_key user:profile user:inference');
-  url.searchParams.set('code_challenge', pkce.challenge);
-  url.searchParams.set('code_challenge_method', 'S256');
-  url.searchParams.set('state', pkce.verifier);
-  
-  return {
-    url: url.toString(),
-    verifier: pkce.verifier
-  };
-};
-
-/**
- * Make HTTPS request
- * @param {string} url - Full URL
- * @param {Object} options - Request options
- * @returns {Promise<Object>} Response JSON
- */
-const makeRequest = (url, options) => {
-  return new Promise((resolve, reject) => {
-    const req = https.request(url, {
-      method: options.method || 'POST',
-      headers: options.headers || {}
-    }, (res) => {
-      let data = '';
-      res.on('data', chunk => data += chunk);
-      res.on('end', () => {
-        try {
-          const json = JSON.parse(data);
-          if (res.statusCode >= 200 && res.statusCode < 300) {
-            resolve(json);
-          } else {
-            reject(new Error(json.error?.message || `HTTP ${res.statusCode}: ${data}`));
-          }
-        } catch (e) {
-          reject(new Error(`Invalid JSON response: ${data.substring(0, 200)}`));
-        }
-      });
-    });
-    
-    req.on('error', reject);
-    
-    if (options.body) {
-      req.write(JSON.stringify(options.body));
-    }
-    req.end();
-  });
-};
-
-/**
- * Exchange authorization code for tokens
- * @param {string} code - Authorization code from callback (format: code#state)
- * @param {string} verifier - PKCE code verifier
- * @returns {Promise<Object>} { type: 'success', access: string, refresh: string, expires: number } or { type: 'failed' }
- * 
- * Data source: https://console.anthropic.com/v1/oauth/token (POST)
- */
-const exchange = async (code, verifier) => {
-  try {
-    // Code format from callback: "authorization_code#state"
-    const splits = code.split('#');
-    const authCode = splits[0];
-    const state = splits[1] || '';
-    
-    const response = await makeRequest('https://console.anthropic.com/v1/oauth/token', {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json'
-      },
-      body: {
-        code: authCode,
-        state: state,
-        grant_type: 'authorization_code',
-        client_id: CLIENT_ID,
-        redirect_uri: REDIRECT_URI,
-        code_verifier: verifier
-      }
-    });
-    
-    return {
-      type: 'success',
-      access: response.access_token,
-      refresh: response.refresh_token,
-      expires: Date.now() + (response.expires_in * 1000)
-    };
-  } catch (error) {
-    return {
-      type: 'failed',
-      error: error.message
-    };
-  }
-};
-
-/**
- * Refresh access token using refresh token
- * @param {string} refreshToken - The refresh token
- * @returns {Promise<Object>} { type: 'success', access: string, refresh: string, expires: number } or { type: 'failed' }
- * 
- * Data source: https://console.anthropic.com/v1/oauth/token (POST)
- */
-const refreshToken = async (refreshToken) => {
-  try {
-    const response = await makeRequest('https://console.anthropic.com/v1/oauth/token', {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json'
-      },
-      body: {
-        grant_type: 'refresh_token',
-        refresh_token: refreshToken,
-        client_id: CLIENT_ID
-      }
-    });
-    
-    return {
-      type: 'success',
-      access: response.access_token,
-      refresh: response.refresh_token,
-      expires: Date.now() + (response.expires_in * 1000)
-    };
-  } catch (error) {
-    return {
-      type: 'failed',
-      error: error.message
-    };
-  }
-};
-
-/**
- * Create an API key using OAuth token (for console mode)
- * @param {string} accessToken - The access token
- * @returns {Promise<Object>} { type: 'success', key: string } or { type: 'failed' }
- * 
- * Data source: https://api.anthropic.com/api/oauth/claude_cli/create_api_key (POST)
- */
-const createApiKey = async (accessToken) => {
-  try {
-    const response = await makeRequest('https://api.anthropic.com/api/oauth/claude_cli/create_api_key', {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-        'Authorization': `Bearer ${accessToken}`
-      }
-    });
-    
-    return {
-      type: 'success',
-      key: response.raw_key
-    };
-  } catch (error) {
-    return {
-      type: 'failed',
-      error: error.message
-    };
-  }
-};
-
-/**
- * Get valid access token (refresh if expired)
- * @param {Object} oauthData - OAuth data { access, refresh, expires }
- * @returns {Promise<Object>} { access: string, refresh: string, expires: number, refreshed: boolean }
- */
-const getValidToken = async (oauthData) => {
-  if (!oauthData || !oauthData.refresh) {
-    return null;
-  }
-  
-  // Check if token is expired or will expire in the next 5 minutes
-  const expirationBuffer = 5 * 60 * 1000; // 5 minutes
-  if (oauthData.expires && oauthData.expires > Date.now() + expirationBuffer) {
-    return {
-      ...oauthData,
-      refreshed: false
-    };
-  }
-  
-  // Token expired or about to expire, refresh it
-  const result = await refreshToken(oauthData.refresh);
-  if (result.type === 'success') {
-    return {
-      access: result.access,
-      refresh: result.refresh,
-      expires: result.expires,
-      refreshed: true
-    };
-  }
-  
-  return null;
-};
-
-/**
- * Check if credentials are OAuth tokens
- * @param {Object} credentials - Agent credentials
- * @returns {boolean}
- */
-const isOAuthCredentials = (credentials) => {
-  return credentials && credentials.oauth && credentials.oauth.refresh;
-};
-
-module.exports = {
-  CLIENT_ID,
-  REDIRECT_URI,
-  generatePKCE,
-  authorize,
-  exchange,
-  refreshToken,
-  createApiKey,
-  getValidToken,
-  isOAuthCredentials
-};