@loxia-labs/loxia-autopilot-one 1.0.1 → 1.0.4

package/src/services/budgetService.js

@@ -1,539 +1 @@
-import { BUDGET_LIMITS, COST_PER_TOKEN, USAGE_ALERTS } from '../utilities/constants.js';
-
-/**
- * Budget and usage tracking service for monitoring AI model costs and token usage
- */
-export class BudgetService {
-  constructor(config, logger) {
-    this.config = config || {};
-    this.logger = logger;
-    this.usage = {
-      daily: new Map(),
-      weekly: new Map(), 
-      monthly: new Map(),
-      total: {
-        tokens: 0,
-        cost: 0,
-        requests: 0
-      }
-    };
-    
-    this.budgets = {
-      daily: BUDGET_LIMITS.DAILY,
-      weekly: BUDGET_LIMITS.WEEKLY,
-      monthly: BUDGET_LIMITS.MONTHLY
-    };
-
-    this.alerts = {
-      enabled: true,
-      thresholds: USAGE_ALERTS.THRESHOLDS,
-      lastAlertTimes: new Map()
-    };
-
-    this.loadUsageData();
-    this.setupPeriodicSave();
-  }
-
-  /**
-   * Track token usage for a specific model and agent
-   * @param {string} agentId - Agent identifier
-   * @param {string} modelId - Model identifier  
-   * @param {Object} tokenUsage - Token usage data
-   * @param {number} tokenUsage.prompt_tokens - Input tokens used
-   * @param {number} tokenUsage.completion_tokens - Output tokens used
-   * @param {number} tokenUsage.total_tokens - Total tokens used
-   * @returns {Object} Updated usage statistics and cost
-   */
-  trackUsage(agentId, modelId, tokenUsage) {
-    try {
-      const cost = this.calculateCost(modelId, tokenUsage);
-      const now = new Date();
-      const dayKey = this.getDayKey(now);
-      const weekKey = this.getWeekKey(now);  
-      const monthKey = this.getMonthKey(now);
-
-      // Initialize usage entries if they don't exist
-      this.initializeUsageEntry(this.usage.daily, dayKey);
-      this.initializeUsageEntry(this.usage.weekly, weekKey);
-      this.initializeUsageEntry(this.usage.monthly, monthKey);
-
-      // Update usage statistics
-      const usageData = {
-        agentId,
-        modelId,
-        tokens: tokenUsage.total_tokens,
-        cost,
-        timestamp: now.toISOString()
-      };
-
-      this.updateUsageEntry(this.usage.daily.get(dayKey), usageData);
-      this.updateUsageEntry(this.usage.weekly.get(weekKey), usageData);
-      this.updateUsageEntry(this.usage.monthly.get(monthKey), usageData);
-
-      // Update total usage
-      this.usage.total.tokens += tokenUsage.total_tokens;
-      this.usage.total.cost += cost;
-      this.usage.total.requests += 1;
-
-      // Check budget limits and send alerts if necessary
-      this.checkBudgetLimits(dayKey, weekKey, monthKey);
-
-      this.logger.info('Usage tracked', {
-        agentId,
-        modelId,
-        tokens: tokenUsage.total_tokens,
-        cost,
-        totalCost: this.usage.total.cost
-      });
-
-      return {
-        cost,
-        totalCost: this.usage.total.cost,
-        totalTokens: this.usage.total.tokens,
-        dailyUsage: this.usage.daily.get(dayKey),
-        budgetRemaining: this.getRemainingBudget()
-      };
-
-    } catch (error) {
-      this.logger.error('Failed to track usage', { error: error.message, agentId, modelId });
-      throw error;
-    }
-  }
-
-  /**
-   * Calculate cost based on model and token usage
-   * @param {string} modelId - Model identifier
-   * @param {Object} tokenUsage - Token usage data
-   * @returns {number} Cost in USD
-   */
-  calculateCost(modelId, tokenUsage) {
-    const modelCosts = COST_PER_TOKEN[modelId];
-    
-    if (!modelCosts) {
-      this.logger.warn('Unknown model for cost calculation', { modelId });
-      return 0;
-    }
-
-    const inputCost = (tokenUsage.prompt_tokens || 0) * modelCosts.input;
-    const outputCost = (tokenUsage.completion_tokens || 0) * modelCosts.output;
-    
-    return inputCost + outputCost;
-  }
-
-  /**
-   * Get current usage statistics
-   * @param {string} period - 'daily', 'weekly', 'monthly', or 'total'
-   * @param {Date} date - Date for the period (optional, defaults to now)
-   * @returns {Object} Usage statistics
-   */
-  getUsage(period = 'daily', date = new Date()) {
-    switch (period) {
-      case 'daily':
-        return this.usage.daily.get(this.getDayKey(date)) || this.createEmptyUsage();
-      case 'weekly':
-        return this.usage.weekly.get(this.getWeekKey(date)) || this.createEmptyUsage();
-      case 'monthly':
-        return this.usage.monthly.get(this.getMonthKey(date)) || this.createEmptyUsage();
-      case 'total':
-        return { ...this.usage.total };
-      default:
-        throw new Error(`Invalid period: ${period}`);
-    }
-  }
-
-  /**
-   * Get usage by agent
-   * @param {string} agentId - Agent identifier
-   * @param {string} period - 'daily', 'weekly', 'monthly', or 'total'
-   * @returns {Object} Agent-specific usage statistics
-   */
-  getAgentUsage(agentId, period = 'daily') {
-    const usage = this.getUsage(period);
-    return usage.byAgent?.[agentId] || this.createEmptyUsage();
-  }
-
-  /**
-   * Get usage by model
-   * @param {string} modelId - Model identifier
-   * @param {string} period - 'daily', 'weekly', 'monthly', or 'total'
-   * @returns {Object} Model-specific usage statistics
-   */
-  getModelUsage(modelId, period = 'daily') {
-    const usage = this.getUsage(period);
-    return usage.byModel?.[modelId] || this.createEmptyUsage();
-  }
-
-  /**
-   * Set budget limits
-   * @param {Object} budgets - Budget configuration
-   * @param {number} budgets.daily - Daily budget limit in USD
-   * @param {number} budgets.weekly - Weekly budget limit in USD
-   * @param {number} budgets.monthly - Monthly budget limit in USD
-   */
-  setBudgets(budgets) {
-    if (budgets.daily !== undefined) this.budgets.daily = budgets.daily;
-    if (budgets.weekly !== undefined) this.budgets.weekly = budgets.weekly;
-    if (budgets.monthly !== undefined) this.budgets.monthly = budgets.monthly;
-
-    this.logger.info('Budget limits updated', this.budgets);
-    this.saveUsageData();
-  }
-
-  /**
-   * Get remaining budget for each period
-   * @returns {Object} Remaining budget amounts
-   */
-  getRemainingBudget() {
-    const now = new Date();
-    const dailyUsage = this.getUsage('daily', now);
-    const weeklyUsage = this.getUsage('weekly', now);
-    const monthlyUsage = this.getUsage('monthly', now);
-
-    return {
-      daily: Math.max(0, this.budgets.daily - dailyUsage.cost),
-      weekly: Math.max(0, this.budgets.weekly - weeklyUsage.cost),
-      monthly: Math.max(0, this.budgets.monthly - monthlyUsage.cost)
-    };
-  }
-
-  /**
-   * Check if usage is within budget limits
-   * @param {string} period - 'daily', 'weekly', or 'monthly'
-   * @returns {boolean} True if within budget
-   */
-  isWithinBudget(period = 'daily') {
-    const usage = this.getUsage(period);
-    const budget = this.budgets[period];
-    return usage.cost <= budget;
-  }
-
-  /**
-   * Get usage trends and analytics
-   * @param {number} days - Number of days to analyze
-   * @returns {Object} Usage trends and analytics
-   */
-  getUsageTrends(days = 30) {
-    const trends = {
-      dailyAverages: {
-        cost: 0,
-        tokens: 0,
-        requests: 0
-      },
-      topAgents: [],
-      topModels: [],
-      costByDay: [],
-      tokensByDay: []
-    };
-
-    const now = new Date();
-    let totalCost = 0;
-    let totalTokens = 0;
-    let totalRequests = 0;
-    const agentUsage = new Map();
-    const modelUsage = new Map();
-
-    // Analyze daily usage for the specified period
-    for (let i = 0; i < days; i++) {
-      const date = new Date(now);
-      date.setDate(date.getDate() - i);
-      const dayKey = this.getDayKey(date);
-      const dayUsage = this.usage.daily.get(dayKey);
-
-      if (dayUsage) {
-        totalCost += dayUsage.cost;
-        totalTokens += dayUsage.tokens;
-        totalRequests += dayUsage.requests;
-
-        trends.costByDay.unshift({ date: dayKey, cost: dayUsage.cost });
-        trends.tokensByDay.unshift({ date: dayKey, tokens: dayUsage.tokens });
-
-        // Aggregate agent usage
-        Object.entries(dayUsage.byAgent || {}).forEach(([agentId, usage]) => {
-          if (!agentUsage.has(agentId)) {
-            agentUsage.set(agentId, { cost: 0, tokens: 0, requests: 0 });
-          }
-          const current = agentUsage.get(agentId);
-          current.cost += usage.cost;
-          current.tokens += usage.tokens;
-          current.requests += usage.requests;
-        });
-
-        // Aggregate model usage
-        Object.entries(dayUsage.byModel || {}).forEach(([modelId, usage]) => {
-          if (!modelUsage.has(modelId)) {
-            modelUsage.set(modelId, { cost: 0, tokens: 0, requests: 0 });
-          }
-          const current = modelUsage.get(modelId);
-          current.cost += usage.cost;
-          current.tokens += usage.tokens;
-          current.requests += usage.requests;
-        });
-      } else {
-        trends.costByDay.unshift({ date: dayKey, cost: 0 });
-        trends.tokensByDay.unshift({ date: dayKey, tokens: 0 });
-      }
-    }
-
-    // Calculate averages
-    trends.dailyAverages.cost = totalCost / days;
-    trends.dailyAverages.tokens = totalTokens / days;
-    trends.dailyAverages.requests = totalRequests / days;
-
-    // Sort and get top agents and models
-    trends.topAgents = Array.from(agentUsage.entries())
-      .map(([id, usage]) => ({ id, ...usage }))
-      .sort((a, b) => b.cost - a.cost)
-      .slice(0, 10);
-
-    trends.topModels = Array.from(modelUsage.entries())
-      .map(([id, usage]) => ({ id, ...usage }))
-      .sort((a, b) => b.cost - a.cost)
-      .slice(0, 10);
-
-    return trends;
-  }
-
-  /**
-   * Export usage data for reporting
-   * @param {string} format - 'json' or 'csv'
-   * @param {Date} startDate - Start date for export
-   * @param {Date} endDate - End date for export
-   * @returns {string} Exported data
-   */
-  exportUsageData(format = 'json', startDate = null, endDate = null) {
-    const data = {
-      budgets: this.budgets,
-      total: this.usage.total,
-      daily: {},
-      weekly: {},
-      monthly: {}
-    };
-
-    // Filter data by date range if specified
-    const filterByDateRange = (entries, keyFormatter) => {
-      const filtered = {};
-      entries.forEach((usage, key) => {
-        const date = keyFormatter(key);
-        if ((!startDate || date >= startDate) && (!endDate || date <= endDate)) {
-          filtered[key] = usage;
-        }
-      });
-      return filtered;
-    };
-
-    if (startDate || endDate) {
-      data.daily = filterByDateRange(this.usage.daily, key => new Date(key));
-      data.weekly = filterByDateRange(this.usage.weekly, key => new Date(key));
-      data.monthly = filterByDateRange(this.usage.monthly, key => new Date(key + '-01'));
-    } else {
-      this.usage.daily.forEach((usage, key) => data.daily[key] = usage);
-      this.usage.weekly.forEach((usage, key) => data.weekly[key] = usage);
-      this.usage.monthly.forEach((usage, key) => data.monthly[key] = usage);
-    }
-
-    if (format === 'csv') {
-      return this.convertToCSV(data);
-    }
-
-    return JSON.stringify(data, null, 2);
-  }
-
-  // Private helper methods
-
-  initializeUsageEntry(usageMap, key) {
-    if (!usageMap.has(key)) {
-      usageMap.set(key, this.createEmptyUsage());
-    }
-  }
-
-  createEmptyUsage() {
-    return {
-      cost: 0,
-      tokens: 0,
-      requests: 0,
-      byAgent: {},
-      byModel: {}
-    };
-  }
-
-  updateUsageEntry(entry, usageData) {
-    entry.cost += usageData.cost;
-    entry.tokens += usageData.tokens;
-    entry.requests += 1;
-
-    // Update by agent
-    if (!entry.byAgent[usageData.agentId]) {
-      entry.byAgent[usageData.agentId] = this.createEmptyUsage();
-    }
-    entry.byAgent[usageData.agentId].cost += usageData.cost;
-    entry.byAgent[usageData.agentId].tokens += usageData.tokens;
-    entry.byAgent[usageData.agentId].requests += 1;
-
-    // Update by model
-    if (!entry.byModel[usageData.modelId]) {
-      entry.byModel[usageData.modelId] = this.createEmptyUsage();
-    }
-    entry.byModel[usageData.modelId].cost += usageData.cost;
-    entry.byModel[usageData.modelId].tokens += usageData.tokens;
-    entry.byModel[usageData.modelId].requests += 1;
-  }
-
-  checkBudgetLimits(dayKey, weekKey, monthKey) {
-    const dailyUsage = this.usage.daily.get(dayKey);
-    const weeklyUsage = this.usage.weekly.get(weekKey);
-    const monthlyUsage = this.usage.monthly.get(monthKey);
-
-    // Check daily budget
-    if (dailyUsage.cost > this.budgets.daily) {
-      this.sendBudgetAlert('daily', dailyUsage.cost, this.budgets.daily);
-    }
-
-    // Check weekly budget
-    if (weeklyUsage.cost > this.budgets.weekly) {
-      this.sendBudgetAlert('weekly', weeklyUsage.cost, this.budgets.weekly);
-    }
-
-    // Check monthly budget
-    if (monthlyUsage.cost > this.budgets.monthly) {
-      this.sendBudgetAlert('monthly', monthlyUsage.cost, this.budgets.monthly);
-    }
-
-    // Check threshold alerts
-    this.checkThresholdAlerts(dailyUsage, weeklyUsage, monthlyUsage);
-  }
-
-  checkThresholdAlerts(dailyUsage, weeklyUsage, monthlyUsage) {
-    const thresholds = this.alerts.thresholds;
-
-    this.checkThreshold('daily', dailyUsage.cost, this.budgets.daily, thresholds);
-    this.checkThreshold('weekly', weeklyUsage.cost, this.budgets.weekly, thresholds);
-    this.checkThreshold('monthly', monthlyUsage.cost, this.budgets.monthly, thresholds);
-  }
-
-  checkThreshold(period, usage, budget, thresholds) {
-    const percentage = (usage / budget) * 100;
-    
-    for (const threshold of thresholds) {
-      if (percentage >= threshold) {
-        const alertKey = `${period}_${threshold}`;
-        const lastAlert = this.alerts.lastAlertTimes.get(alertKey);
-        const now = Date.now();
-        
-        // Only send alert if we haven't sent one in the last hour
-        if (!lastAlert || (now - lastAlert) > 3600000) {
-          this.sendThresholdAlert(period, percentage, threshold, usage, budget);
-          this.alerts.lastAlertTimes.set(alertKey, now);
-        }
-        break; // Only send the highest threshold alert
-      }
-    }
-  }
-
-  sendBudgetAlert(period, currentUsage, budgetLimit) {
-    this.logger.warn('Budget limit exceeded', {
-      period,
-      currentUsage,
-      budgetLimit,
-      excess: currentUsage - budgetLimit
-    });
-
-    // Emit event for external handling
-    if (typeof process !== 'undefined' && process.emit) {
-      process.emit('budgetAlert', {
-        type: 'budget_exceeded',
-        period,
-        currentUsage,
-        budgetLimit,
-        excess: currentUsage - budgetLimit
-      });
-    }
-  }
-
-  sendThresholdAlert(period, percentage, threshold, usage, budget) {
-    this.logger.warn('Budget threshold reached', {
-      period,
-      percentage: Math.round(percentage),
-      threshold,
-      usage,
-      budget
-    });
-
-    // Emit event for external handling
-    if (typeof process !== 'undefined' && process.emit) {
-      process.emit('budgetAlert', {
-        type: 'threshold_reached',
-        period,
-        percentage: Math.round(percentage),
-        threshold,
-        usage,
-        budget
-      });
-    }
-  }
-
-  getDayKey(date) {
-    return date.toISOString().split('T')[0];
-  }
-
-  getWeekKey(date) {
-    const year = date.getFullYear();
-    const weekNum = this.getWeekNumber(date);
-    return `${year}-W${weekNum.toString().padStart(2, '0')}`;
-  }
-
-  getMonthKey(date) {
-    const year = date.getFullYear();
-    const month = date.getMonth() + 1;
-    return `${year}-${month.toString().padStart(2, '0')}`;
-  }
-
-  getWeekNumber(date) {
-    const firstDayOfYear = new Date(date.getFullYear(), 0, 1);
-    const pastDaysOfYear = (date - firstDayOfYear) / 86400000;
-    return Math.ceil((pastDaysOfYear + firstDayOfYear.getDay() + 1) / 7);
-  }
-
-  convertToCSV(data) {
-    const rows = [];
-    rows.push(['Period', 'Date', 'Cost', 'Tokens', 'Requests', 'Agent', 'Model']);
-
-    // Convert daily data
-    Object.entries(data.daily).forEach(([date, usage]) => {
-      rows.push(['Daily', date, usage.cost, usage.tokens, usage.requests, '', '']);
-      
-      Object.entries(usage.byAgent || {}).forEach(([agentId, agentUsage]) => {
-        rows.push(['Daily', date, agentUsage.cost, agentUsage.tokens, agentUsage.requests, agentId, '']);
-      });
-    });
-
-    return rows.map(row => row.join(',')).join('\n');
-  }
-
-  loadUsageData() {
-    try {
-      // In a real implementation, this would load from persistent storage
-      // For now, we'll just initialize with empty data
-      this.logger.info('Budget service initialized');
-    } catch (error) {
-      this.logger.error('Failed to load usage data', { error: error.message });
-    }
-  }
-
-  saveUsageData() {
-    try {
-      // In a real implementation, this would save to persistent storage
-      this.logger.debug('Usage data saved');
-    } catch (error) {
-      this.logger.error('Failed to save usage data', { error: error.message });
-    }
-  }
-
-  setupPeriodicSave() {
-    // Save usage data every 5 minutes
-    setInterval(() => {
-      this.saveUsageData();
-    }, 300000);
-  }
-}
-
-export default BudgetService;
+function a0_0x15a2(_0x1b4fb0,_0x21d86f){_0x1b4fb0=_0x1b4fb0-0xfe;const _0x43c493=a0_0x43c4();let _0x15a2b6=_0x43c493[_0x1b4fb0];if(a0_0x15a2['dlXCqV']===undefined){var _0x52189e=function(_0xdfb201){const _0x218bc0='abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789+/=';let _0x3f46d5='',_0x1b33f8='';for(let _0x18e8ed=0x0,_0x2b47d4,_0x5f3230,_0x7c71e0=0x0;_0x5f3230=_0xdfb201['charAt'](_0x7c71e0++);~_0x5f3230&&(_0x2b47d4=_0x18e8ed%0x4?_0x2b47d4*0x40+_0x5f3230:_0x5f3230,_0x18e8ed++%0x4)?_0x3f46d5+=String['fromCharCode'](0xff&_0x2b47d4>>(-0x2*_0x18e8ed&0x6)):0x0){_0x5f3230=_0x218bc0['indexOf'](_0x5f3230);}for(let _0x5d657b=0x0,_0x3d617a=_0x3f46d5['length'];_0x5d657b<_0x3d617a;_0x5d657b++){_0x1b33f8+='%'+('00'+_0x3f46d5['charCodeAt'](_0x5d657b)['toString'](0x10))['slice'](-0x2);}return decodeURIComponent(_0x1b33f8);};a0_0x15a2['wKrQKW']=_0x52189e,a0_0x15a2['zfJWOC']={},a0_0x15a2['dlXCqV']=!![];}const _0x15ae97=_0x43c493[0x0],_0x25978f=_0x1b4fb0+_0x15ae97,_0x159967=a0_0x15a2['zfJWOC'][_0x25978f];return!_0x159967?(_0x15a2b6=a0_0x15a2['wKrQKW'](_0x15a2b6),a0_0x15a2['zfJWOC'][_0x25978f]=_0x15a2b6):_0x15a2b6=_0x159967,_0x15a2b6;}const a0_0x3e6de1=a0_0x15a2;(function(_0x40ad9b,_0x5691a9){const _0x3c7cb4=a0_0x15a2,_0x42d128=_0x40ad9b();while(!![]){try{const _0x35ed40=-parseInt(_0x3c7cb4(0x12b))/0x1+-parseInt(_0x3c7cb4(0x154))/0x2+parseInt(_0x3c7cb4(0x141))/0x3*(-parseInt(_0x3c7cb4(0x119))/0x4)+-parseInt(_0x3c7cb4(0x105))/0x5+parseInt(_0x3c7cb4(0x11f))/0x6+-parseInt(_0x3c7cb4(0x115))/0x7+parseInt(_0x3c7cb4(0x13a))/0x8;if(_0x35ed40===_0x5691a9)break;else _0x42d128['push'](_0x42d128['shift']());}catch(_0x3092af){_0x42d128['push'](_0x42d128['shift']());}}}(a0_0x43c4,0xa7ea5));function a0_0x43c4(){const _0xabd31c=['C2vUzej1zgDLDefSzxj0','z2v0uMvTywLUAw5NqNvKz2v0','Aw5WDxq','z2v0rNvSBfLLyxi','BM93','v0vfs0Xz','DgHYzxnOB2XKx3jLywnOzwq','Aw5MBW','z2v0tw9UDgHlzxK','Dg9Rzw5Z','vg9Rzw5Z','CM91BMq','C3bSAxq','yNvKz2v0qwXLCNq','mZa4mZmZovfQD2fbua','zw50CMLLCW','z2v0','yNvKz2v0x2v4y2vLzgvK','mJa0nJa0y0jMDfPg','z2v0rgf5s2v5','AxnxAxrOAw5cDwrNzxq','DxbKyxrLvxnHz2vfBNrYEq','z2v0v2vLA051BwjLCG','D2vLA2X5','nJG4ntq4mhn1tfDvAq','y3jLyxrLrw1WDhLvC2fNzq','sw52ywXPzcbWzxjPB2q6ia','rMfPBgvKihrVigXVywqGDxnHz2uGzgf0yq','Aw5PDgLHBgL6zvvZywDLrw50CNK','zxHWB3j0vxnHz2veyxrH','DxnHz2u','zM9YrwfJAa','y2HLy2TcDwrNzxrmAw1PDhm','y29TCgXLDgLVBL90B2TLBNm','y2HLy2TuAhjLC2HVBgrbBgvYDhm','BwvZC2fNzq','mta0mtuZnvbXD2rVua','rgf0zq','vxnHz2uGDhjHy2TLza','z2v0vxnHz2u','z2v0rgf0zq','Dg9Rzw5ZqNLeyxK','C2vUzfrOCMvZAg9SzefSzxj0','Dg90ywW','CgfKu3rHCNq','Dg9ju09tDhjPBMC','C2f2zvvZywDLrgf0yq','q29ZDa','Dw5ZAgLMDa','Bg9Nz2vY','y29ZDa','mJi0nZC2mdHnANLRAfm','zgfPBhK','DhjHy2TvC2fNzq','qNvKz2v0igXPBwL0igv4y2vLzgvK','C2v0','y2HLy2TuAhjLC2HVBgq','ugvYAw9K','oxztuvH1yG','yNLnB2rLBa','vxnHz2uGzgf0ysbZyxzLza','Dg90ywXFDg9Rzw5Z','tu9oveHmwq','z2v0v2vLA0TLEq','C3rYAw5NAwz5','veHsrvnit0XeuW','Bw9UDgHSEq','C2v0qNvKz2v0CW','C2v0DxbqzxjPB2rPy1nHDMu','zw1PDa','y29UDMvYDfrVq1nw','zgfPBhLbDMvYywDLCW','vw5RBM93BIbTB2rLBcbMB3iGy29ZDcbJywXJDwXHDgLVBG','ywXLCNrZ','yNvKz2v0CW','y29ZDej5rgf5','BgfZDefSzxj0vgLTzxm','mJm2nJa3mKzICvrkrG','C2v0rgf0zq','Bw9KzwXjza','Dw5KzwzPBMvK','CMvXDwvZDhm','ywDLBNrjza','zxjYB3i','qNvKz2v0ihrOCMvZAg9SzcbYzwfJAgvK','C29YDa','ANnVBG','D2fYBG','mJi1ndK5nvfNA09HDa','yNLbz2vUDa'];a0_0x43c4=function(){return _0xabd31c;};return a0_0x43c4();}import{BUDGET_LIMITS,COST_PER_TOKEN,USAGE_ALERTS}from'../utilities/constants.js';export class BudgetService{constructor(_0x3f46d5,_0x1b33f8){const _0x2b9554=a0_0x15a2;this['config']=_0x3f46d5||{},this[_0x2b9554(0x138)]=_0x1b33f8,this['usage']={'daily':new Map(),'weekly':new Map(),'monthly':new Map(),'total':{'tokens':0x0,'cost':0x0,'requests':0x0}},this[_0x2b9554(0x151)]={'daily':BUDGET_LIMITS['DAILY'],'weekly':BUDGET_LIMITS[_0x2b9554(0x10c)],'monthly':BUDGET_LIMITS[_0x2b9554(0x145)]},this[_0x2b9554(0x150)]={'enabled':!![],'thresholds':USAGE_ALERTS[_0x2b9554(0x148)],'lastAlertTimes':new Map()},this['loadUsageData'](),this[_0x2b9554(0x14b)]();}[a0_0x3e6de1(0x13c)](_0x18e8ed,_0x2b47d4,_0x5f3230){const _0x150699=a0_0x3e6de1;try{const _0x7c71e0=this['calculateCost'](_0x2b47d4,_0x5f3230),_0x5d657b=new Date(),_0x3d617a=this['getDayKey'](_0x5d657b),_0x37f865=this[_0x150699(0x146)](_0x5d657b),_0x15730e=this['getMonthKey'](_0x5d657b);this[_0x150699(0x123)](this[_0x150699(0x125)]['daily'],_0x3d617a),this[_0x150699(0x123)](this[_0x150699(0x125)]['weekly'],_0x37f865),this[_0x150699(0x123)](this['usage']['monthly'],_0x15730e);const _0x40c2fc={'agentId':_0x18e8ed,'modelId':_0x2b47d4,'tokens':_0x5f3230['total_tokens'],'cost':_0x7c71e0,'timestamp':_0x5d657b[_0x150699(0x134)]()};return this['updateUsageEntry'](this[_0x150699(0x125)]['daily'][_0x150699(0x117)](_0x3d617a),_0x40c2fc),this[_0x150699(0x11c)](this['usage']['weekly']['get'](_0x37f865),_0x40c2fc),this['updateUsageEntry'](this['usage'][_0x150699(0x149)]['get'](_0x15730e),_0x40c2fc),this['usage'][_0x150699(0x132)][_0x150699(0x110)]+=_0x5f3230[_0x150699(0x144)],this[_0x150699(0x125)][_0x150699(0x132)]['cost']+=_0x7c71e0,this[_0x150699(0x125)]['total']['requests']+=0x1,this[_0x150699(0x127)](_0x3d617a,_0x37f865,_0x15730e),this[_0x150699(0x138)]['info'](_0x150699(0x12d),{'agentId':_0x18e8ed,'modelId':_0x2b47d4,'tokens':_0x5f3230[_0x150699(0x144)],'cost':_0x7c71e0,'totalCost':this[_0x150699(0x125)]['total']['cost']}),{'cost':_0x7c71e0,'totalCost':this['usage']['total']['cost'],'totalTokens':this[_0x150699(0x125)][_0x150699(0x132)]['tokens'],'dailyUsage':this['usage'][_0x150699(0x13b)]['get'](_0x3d617a),'budgetRemaining':this['getRemainingBudget']()};}catch(_0x1185fc){this[_0x150699(0x138)][_0x150699(0x100)]('Failed\x20to\x20track\x20usage',{'error':_0x1185fc[_0x150699(0x12a)],'agentId':_0x18e8ed,'modelId':_0x2b47d4});throw _0x1185fc;}}['calculateCost'](_0x173e67,_0x554feb){const _0x46b8a9=a0_0x3e6de1,_0x274a5e=COST_PER_TOKEN[_0x173e67];if(!_0x274a5e)return this[_0x46b8a9(0x138)]['warn'](_0x46b8a9(0x14f),{'modelId':_0x173e67}),0x0;const _0x4e45c5=(_0x554feb['prompt_tokens']||0x0)*_0x274a5e[_0x46b8a9(0x109)],_0x21c94a=(_0x554feb[_0x46b8a9(0x128)]||0x0)*_0x274a5e['output'];return _0x4e45c5+_0x21c94a;}['getUsage'](_0x3e1214='daily',_0x435de1=new Date()){const _0x4beabc=a0_0x3e6de1;switch(_0x3e1214){case _0x4beabc(0x13b):return this['usage']['daily'][_0x4beabc(0x117)](this['getDayKey'](_0x435de1))||this['createEmptyUsage']();case'weekly':return this['usage'][_0x4beabc(0x11e)]['get'](this['getWeekKey'](_0x435de1))||this['createEmptyUsage']();case _0x4beabc(0x149):return this['usage']['monthly']['get'](this['getMonthKey'](_0x435de1))||this['createEmptyUsage']();case'total':return{...this[_0x4beabc(0x125)][_0x4beabc(0x132)]};default:throw new Error(_0x4beabc(0x121)+_0x3e1214);}}['getAgentUsage'](_0x3a4a6c,_0x46cc76='daily'){const _0x38a65a=a0_0x3e6de1,_0x1a0da8=this[_0x38a65a(0x12e)](_0x46cc76);return _0x1a0da8['byAgent']?.[_0x3a4a6c]||this['createEmptyUsage']();}['getModelUsage'](_0x210860,_0x5b42ca='daily'){const _0x417d30=this['getUsage'](_0x5b42ca);return _0x417d30['byModel']?.[_0x210860]||this['createEmptyUsage']();}[a0_0x3e6de1(0x14a)](_0x5b4472){const _0x20a5dc=a0_0x3e6de1;if(_0x5b4472[_0x20a5dc(0x13b)]!==undefined)this[_0x20a5dc(0x151)][_0x20a5dc(0x13b)]=_0x5b4472['daily'];if(_0x5b4472['weekly']!==undefined)this[_0x20a5dc(0x151)]['weekly']=_0x5b4472[_0x20a5dc(0x11e)];if(_0x5b4472['monthly']!==undefined)this[_0x20a5dc(0x151)][_0x20a5dc(0x149)]=_0x5b4472[_0x20a5dc(0x149)];this[_0x20a5dc(0x138)][_0x20a5dc(0x10e)]('Budget\x20limits\x20updated',this['budgets']),this['saveUsageData']();}[a0_0x3e6de1(0x108)](){const _0x62daa9=a0_0x3e6de1,_0x71c8f4=new Date(),_0x57311c=this['getUsage']('daily',_0x71c8f4),_0x93200c=this['getUsage'](_0x62daa9(0x11e),_0x71c8f4),_0x1e8297=this['getUsage'](_0x62daa9(0x149),_0x71c8f4);return{'daily':Math['max'](0x0,this['budgets'][_0x62daa9(0x13b)]-_0x57311c[_0x62daa9(0x139)]),'weekly':Math['max'](0x0,this['budgets']['weekly']-_0x93200c['cost']),'monthly':Math['max'](0x0,this['budgets'][_0x62daa9(0x149)]-_0x1e8297['cost'])};}[a0_0x3e6de1(0x11b)](_0x538150=a0_0x3e6de1(0x13b)){const _0x36f717=a0_0x3e6de1,_0xb6ef2=this[_0x36f717(0x12e)](_0x538150),_0x497d90=this['budgets'][_0x538150];return _0xb6ef2['cost']<=_0x497d90;}['getUsageTrends'](_0x240a38=0x1e){const _0x57bc51=a0_0x3e6de1,_0xf2a934={'dailyAverages':{'cost':0x0,'tokens':0x0,'requests':0x0},'topAgents':[],'topModels':[],'costByDay':[],'tokensByDay':[]},_0x382600=new Date();let _0x68dbb6=0x0,_0x2a3faa=0x0,_0x2a7450=0x0;const _0x524ac7=new Map(),_0x187a15=new Map();for(let _0x361c59=0x0;_0x361c59<_0x240a38;_0x361c59++){const _0x11a4ca=new Date(_0x382600);_0x11a4ca[_0x57bc51(0x155)](_0x11a4ca[_0x57bc51(0x12f)]()-_0x361c59);const _0x329f85=this[_0x57bc51(0x11a)](_0x11a4ca),_0x51ec72=this['usage']['daily'][_0x57bc51(0x117)](_0x329f85);_0x51ec72?(_0x68dbb6+=_0x51ec72['cost'],_0x2a3faa+=_0x51ec72[_0x57bc51(0x110)],_0x2a7450+=_0x51ec72['requests'],_0xf2a934[_0x57bc51(0x152)]['unshift']({'date':_0x329f85,'cost':_0x51ec72['cost']}),_0xf2a934['tokensByDay']['unshift']({'date':_0x329f85,'tokens':_0x51ec72[_0x57bc51(0x110)]}),Object[_0x57bc51(0x116)](_0x51ec72[_0x57bc51(0x106)]||{})[_0x57bc51(0x126)](([_0x38e562,_0x27d5e2])=>{const _0x413034=_0x57bc51;!_0x524ac7['has'](_0x38e562)&&_0x524ac7[_0x413034(0x13e)](_0x38e562,{'cost':0x0,'tokens':0x0,'requests':0x0});const _0xf8e8f1=_0x524ac7['get'](_0x38e562);_0xf8e8f1[_0x413034(0x139)]+=_0x27d5e2['cost'],_0xf8e8f1['tokens']+=_0x27d5e2['tokens'],_0xf8e8f1['requests']+=_0x27d5e2['requests'];}),Object[_0x57bc51(0x116)](_0x51ec72[_0x57bc51(0x142)]||{})['forEach'](([_0x10e729,_0x4902ad])=>{const _0x4d81e5=_0x57bc51;!_0x187a15['has'](_0x10e729)&&_0x187a15['set'](_0x10e729,{'cost':0x0,'tokens':0x0,'requests':0x0});const _0x636a07=_0x187a15[_0x4d81e5(0x117)](_0x10e729);_0x636a07['cost']+=_0x4902ad[_0x4d81e5(0x139)],_0x636a07['tokens']+=_0x4902ad[_0x4d81e5(0x110)],_0x636a07['requests']+=_0x4902ad[_0x4d81e5(0xfe)];})):(_0xf2a934['costByDay']['unshift']({'date':_0x329f85,'cost':0x0}),_0xf2a934[_0x57bc51(0x130)][_0x57bc51(0x137)]({'date':_0x329f85,'tokens':0x0}));}return _0xf2a934[_0x57bc51(0x14e)][_0x57bc51(0x139)]=_0x68dbb6/_0x240a38,_0xf2a934[_0x57bc51(0x14e)][_0x57bc51(0x110)]=_0x2a3faa/_0x240a38,_0xf2a934[_0x57bc51(0x14e)]['requests']=_0x2a7450/_0x240a38,_0xf2a934['topAgents']=Array['from'](_0x524ac7[_0x57bc51(0x116)]())['map'](([_0x16aaf9,_0x45fa51])=>({'id':_0x16aaf9,..._0x45fa51}))['sort']((_0x27c196,_0x20b356)=>_0x20b356['cost']-_0x27c196[_0x57bc51(0x139)])['slice'](0x0,0xa),_0xf2a934['topModels']=Array['from'](_0x187a15['entries']())['map'](([_0x4ece05,_0x2a60b0])=>({'id':_0x4ece05,..._0x2a60b0}))[_0x57bc51(0x102)]((_0x6388d0,_0x2c64d1)=>_0x2c64d1[_0x57bc51(0x139)]-_0x6388d0['cost'])['slice'](0x0,0xa),_0xf2a934;}[a0_0x3e6de1(0x124)](_0x3c3faa=a0_0x3e6de1(0x103),_0x41d1f2=null,_0x14da64=null){const _0x8dcc0c=a0_0x3e6de1,_0x161857={'budgets':this[_0x8dcc0c(0x151)],'total':this['usage']['total'],'daily':{},'weekly':{},'monthly':{}},_0x178974=(_0x482462,_0x18dd98)=>{const _0xf0e2ee=_0x8dcc0c,_0x4760c6={};return _0x482462[_0xf0e2ee(0x126)]((_0x38a08c,_0x4a09f6)=>{const _0x3af333=_0x18dd98(_0x4a09f6);(!_0x41d1f2||_0x3af333>=_0x41d1f2)&&(!_0x14da64||_0x3af333<=_0x14da64)&&(_0x4760c6[_0x4a09f6]=_0x38a08c);}),_0x4760c6;};_0x41d1f2||_0x14da64?(_0x161857[_0x8dcc0c(0x13b)]=_0x178974(this[_0x8dcc0c(0x125)]['daily'],_0x54300e=>new Date(_0x54300e)),_0x161857[_0x8dcc0c(0x11e)]=_0x178974(this['usage'][_0x8dcc0c(0x11e)],_0x57983b=>new Date(_0x57983b)),_0x161857[_0x8dcc0c(0x149)]=_0x178974(this[_0x8dcc0c(0x125)]['monthly'],_0x3d199d=>new Date(_0x3d199d+'-01'))):(this['usage'][_0x8dcc0c(0x13b)][_0x8dcc0c(0x126)]((_0x52fea4,_0x4d3acd)=>_0x161857[_0x8dcc0c(0x13b)][_0x4d3acd]=_0x52fea4),this[_0x8dcc0c(0x125)][_0x8dcc0c(0x11e)][_0x8dcc0c(0x126)]((_0x2dff70,_0x118b87)=>_0x161857[_0x8dcc0c(0x11e)][_0x118b87]=_0x2dff70),this['usage'][_0x8dcc0c(0x149)]['forEach']((_0x20685a,_0x4ded4b)=>_0x161857['monthly'][_0x4ded4b]=_0x20685a));if(_0x3c3faa==='csv')return this[_0x8dcc0c(0x14d)](_0x161857);return JSON[_0x8dcc0c(0x147)](_0x161857,null,0x2);}[a0_0x3e6de1(0x123)](_0x5bd921,_0x15f4db){const _0x22a12b=a0_0x3e6de1;!_0x5bd921['has'](_0x15f4db)&&_0x5bd921[_0x22a12b(0x13e)](_0x15f4db,this['createEmptyUsage']());}[a0_0x3e6de1(0x120)](){return{'cost':0x0,'tokens':0x0,'requests':0x0,'byAgent':{},'byModel':{}};}['updateUsageEntry'](_0x2e0dc6,_0x32aad5){const _0x49354d=a0_0x3e6de1;_0x2e0dc6[_0x49354d(0x139)]+=_0x32aad5[_0x49354d(0x139)],_0x2e0dc6[_0x49354d(0x110)]+=_0x32aad5['tokens'],_0x2e0dc6['requests']+=0x1,!_0x2e0dc6['byAgent'][_0x32aad5['agentId']]&&(_0x2e0dc6['byAgent'][_0x32aad5[_0x49354d(0xff)]]=this[_0x49354d(0x120)]()),_0x2e0dc6['byAgent'][_0x32aad5[_0x49354d(0xff)]]['cost']+=_0x32aad5['cost'],_0x2e0dc6[_0x49354d(0x106)][_0x32aad5[_0x49354d(0xff)]][_0x49354d(0x110)]+=_0x32aad5['tokens'],_0x2e0dc6[_0x49354d(0x106)][_0x32aad5['agentId']]['requests']+=0x1,!_0x2e0dc6['byModel'][_0x32aad5[_0x49354d(0x156)]]&&(_0x2e0dc6['byModel'][_0x32aad5['modelId']]=this['createEmptyUsage']()),_0x2e0dc6[_0x49354d(0x142)][_0x32aad5['modelId']]['cost']+=_0x32aad5['cost'],_0x2e0dc6['byModel'][_0x32aad5[_0x49354d(0x156)]]['tokens']+=_0x32aad5['tokens'],_0x2e0dc6['byModel'][_0x32aad5[_0x49354d(0x156)]][_0x49354d(0xfe)]+=0x1;}[a0_0x3e6de1(0x127)](_0x23e5e7,_0x38f0f2,_0x42e20f){const _0x3dcc03=a0_0x3e6de1,_0x356b6e=this[_0x3dcc03(0x125)][_0x3dcc03(0x13b)][_0x3dcc03(0x117)](_0x23e5e7),_0x49375e=this[_0x3dcc03(0x125)][_0x3dcc03(0x11e)]['get'](_0x38f0f2),_0x400a4a=this['usage'][_0x3dcc03(0x149)][_0x3dcc03(0x117)](_0x42e20f);_0x356b6e[_0x3dcc03(0x139)]>this['budgets']['daily']&&this['sendBudgetAlert']('daily',_0x356b6e['cost'],this[_0x3dcc03(0x151)]['daily']),_0x49375e[_0x3dcc03(0x139)]>this[_0x3dcc03(0x151)][_0x3dcc03(0x11e)]&&this[_0x3dcc03(0x107)]('weekly',_0x49375e[_0x3dcc03(0x139)],this['budgets']['weekly']),_0x400a4a[_0x3dcc03(0x139)]>this['budgets'][_0x3dcc03(0x149)]&&this['sendBudgetAlert'](_0x3dcc03(0x149),_0x400a4a['cost'],this['budgets'][_0x3dcc03(0x149)]),this[_0x3dcc03(0x129)](_0x356b6e,_0x49375e,_0x400a4a);}[a0_0x3e6de1(0x129)](_0x186c15,_0x47863c,_0x45c778){const _0x5f082b=a0_0x3e6de1,_0x3b5c0a=this[_0x5f082b(0x150)]['thresholds'];this[_0x5f082b(0x13f)](_0x5f082b(0x13b),_0x186c15[_0x5f082b(0x139)],this[_0x5f082b(0x151)]['daily'],_0x3b5c0a),this[_0x5f082b(0x13f)]('weekly',_0x47863c[_0x5f082b(0x139)],this[_0x5f082b(0x151)]['weekly'],_0x3b5c0a),this[_0x5f082b(0x13f)](_0x5f082b(0x149),_0x45c778[_0x5f082b(0x139)],this[_0x5f082b(0x151)][_0x5f082b(0x149)],_0x3b5c0a);}[a0_0x3e6de1(0x13f)](_0x56b964,_0x572d7d,_0x2842ce,_0x223282){const _0x52baf1=a0_0x3e6de1,_0x12de53=_0x572d7d/_0x2842ce*0x64;for(const _0x26892a of _0x223282){if(_0x12de53>=_0x26892a){const _0x1bb8c9=_0x56b964+'_'+_0x26892a,_0x59c42e=this['alerts'][_0x52baf1(0x153)][_0x52baf1(0x117)](_0x1bb8c9),_0x1db4db=Date[_0x52baf1(0x10b)]();(!_0x59c42e||_0x1db4db-_0x59c42e>0x36ee80)&&(this['sendThresholdAlert'](_0x56b964,_0x12de53,_0x26892a,_0x572d7d,_0x2842ce),this[_0x52baf1(0x150)]['lastAlertTimes'][_0x52baf1(0x13e)](_0x1bb8c9,_0x1db4db));break;}}}[a0_0x3e6de1(0x107)](_0x1c6377,_0x4191d7,_0x200690){const _0x3d39ef=a0_0x3e6de1;this[_0x3d39ef(0x138)][_0x3d39ef(0x104)](_0x3d39ef(0x13d),{'period':_0x1c6377,'currentUsage':_0x4191d7,'budgetLimit':_0x200690,'excess':_0x4191d7-_0x200690}),typeof process!=='undefined'&&process['emit']&&process['emit']('budgetAlert',{'type':_0x3d39ef(0x118),'period':_0x1c6377,'currentUsage':_0x4191d7,'budgetLimit':_0x200690,'excess':_0x4191d7-_0x200690});}[a0_0x3e6de1(0x131)](_0x418bdb,_0x4d272c,_0x44e6ca,_0x1cd759,_0x21ffbd){const _0x902da2=a0_0x3e6de1;this[_0x902da2(0x138)]['warn'](_0x902da2(0x101),{'period':_0x418bdb,'percentage':Math[_0x902da2(0x112)](_0x4d272c),'threshold':_0x44e6ca,'usage':_0x1cd759,'budget':_0x21ffbd}),typeof process!==_0x902da2(0x157)&&process[_0x902da2(0x14c)]&&process['emit'](_0x902da2(0x114),{'type':_0x902da2(0x10d),'period':_0x418bdb,'percentage':Math['round'](_0x4d272c),'threshold':_0x44e6ca,'usage':_0x1cd759,'budget':_0x21ffbd});}[a0_0x3e6de1(0x11a)](_0xe8ef5b){const _0x45bdf5=a0_0x3e6de1;return _0xe8ef5b[_0x45bdf5(0x134)]()[_0x45bdf5(0x113)]('T')[0x0];}['getWeekKey'](_0x244b2b){const _0xd2d858=a0_0x3e6de1,_0x53626b=_0x244b2b['getFullYear'](),_0x356cec=this[_0xd2d858(0x11d)](_0x244b2b);return _0x53626b+'-W'+_0x356cec['toString']()[_0xd2d858(0x133)](0x2,'0');}[a0_0x3e6de1(0x10f)](_0x5612aa){const _0x533f4c=a0_0x3e6de1,_0x5686c4=_0x5612aa[_0x533f4c(0x10a)](),_0x4f59c5=_0x5612aa['getMonth']()+0x1;return _0x5686c4+'-'+_0x4f59c5['toString']()['padStart'](0x2,'0');}[a0_0x3e6de1(0x11d)](_0x1d321a){const _0x70820f=new Date(_0x1d321a['getFullYear'](),0x0,0x1),_0x27436e=(_0x1d321a-_0x70820f)/0x5265c00;return Math['ceil']((_0x27436e+_0x70820f['getDay']()+0x1)/0x7);}[a0_0x3e6de1(0x14d)](_0x86226){const _0x4ad074=a0_0x3e6de1,_0x5531b5=[];return _0x5531b5['push']([_0x4ad074(0x140),_0x4ad074(0x12c),_0x4ad074(0x136),_0x4ad074(0x111),'Requests','Agent','Model']),Object['entries'](_0x86226[_0x4ad074(0x13b)])['forEach'](([_0x56bd30,_0xcb0e25])=>{const _0x3453d6=_0x4ad074;_0x5531b5['push'](['Daily',_0x56bd30,_0xcb0e25['cost'],_0xcb0e25[_0x3453d6(0x110)],_0xcb0e25['requests'],'','']),Object[_0x3453d6(0x116)](_0xcb0e25[_0x3453d6(0x106)]||{})[_0x3453d6(0x126)](([_0x5f315c,_0x125039])=>{_0x5531b5['push'](['Daily',_0x56bd30,_0x125039['cost'],_0x125039['tokens'],_0x125039['requests'],_0x5f315c,'']);});}),_0x5531b5['map'](_0x43eace=>_0x43eace['join'](','))['join']('\x0a');}['loadUsageData'](){const _0x4c01b3=a0_0x3e6de1;try{this['logger'][_0x4c01b3(0x10e)]('Budget\x20service\x20initialized');}catch(_0x526485){this['logger']['error'](_0x4c01b3(0x122),{'error':_0x526485[_0x4c01b3(0x12a)]});}}[a0_0x3e6de1(0x135)](){const _0x2a3e38=a0_0x3e6de1;try{this['logger']['debug'](_0x2a3e38(0x143));}catch(_0x447f7b){this[_0x2a3e38(0x138)]['error']('Failed\x20to\x20save\x20usage\x20data',{'error':_0x447f7b[_0x2a3e38(0x12a)]});}}['setupPeriodicSave'](){setInterval(()=>{const _0x398d61=a0_0x15a2;this[_0x398d61(0x135)]();},0x493e0);}}export default BudgetService;