Web3Watch-HK/ai.js at master · Beltran12138/Web3Watch-HK · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
'use strict';
/**
 * ai.js — 向后兼容的薄封装层
 *
 * 所有 AI 处理逻辑已迁移至 ai-enhanced.js（多提供商降级版本）。
 * 本文件仅保留 aiCostTracker（成本追踪，ai-enhanced 未包含）并
 * 重新导出 ai-enhanced.js 的全部接口，确保现有 require('./ai') 的调用方无需改动。
 */

const { AI_COST } = require('./config');

// ── AI 成本追踪（仅此文件独有）────────────────────────────────────────────────
const aiCostTracker = {
  dailyTokens: 0,
  dailyCostUSD: 0,
  lastResetDate: new Date().toDateString(),
  totalCalls: 0,
  alertSent: false,

  reset() {
    const today = new Date().toDateString();
    if (today !== this.lastResetDate) {
      this.dailyTokens = 0;
      this.dailyCostUSD = 0;
      this.lastResetDate = today;
      this.alertSent = false;
    }
  },

  track(tokens) {
    this.reset();
    this.dailyTokens += tokens;
    this.dailyCostUSD = (this.dailyTokens / 1000) * (AI_COST?.COST_PER_1K_TOKENS || 0.001);
    this.totalCalls++;

    // Alert if approaching budget
    const budget = AI_COST?.DAILY_BUDGET_USD || 10;
    const threshold = AI_COST?.ALERT_THRESHOLD || 0.8;
    if (!this.alertSent && this.dailyCostUSD >= budget * threshold) {
      console.warn(`[AI COST] WARNING: Daily cost $${this.dailyCostUSD.toFixed(4)} approaching budget $${budget} (${(this.dailyCostUSD / budget * 100).toFixed(1)}%)`);
      this.alertSent = true;
    }
  },

  isOverBudget() {
    this.reset();
    const budget = AI_COST?.DAILY_BUDGET_USD || 10;
    return this.dailyCostUSD >= budget;
  },

  getStatus() {
    this.reset();
    const budget = AI_COST?.DAILY_BUDGET_USD || 10;
    return {
      dailyTokens: this.dailyTokens,
      dailyCostUSD: this.dailyCostUSD.toFixed(4),
      budgetUSD: budget,
      budgetUsedPercent: (this.dailyCostUSD / budget * 100).toFixed(1),
      totalCalls: this.totalCalls,
      isOverBudget: this.dailyCostUSD >= budget,
    };
  },
};

// ── 从 ai-enhanced.js 重新导出所有接口 ───────────────────────────────────────
const enhanced = require('./ai-enhanced');

module.exports = {
  // ai-enhanced.js 的全部导出
  processWithAI:        enhanced.processWithAI,
  batchClassify:        enhanced.batchClassify,
  generateDailySummary: enhanced.generateDailySummary,
  generateWeeklySummary: enhanced.generateWeeklySummary,
  getAIStatus:          enhanced.getAIStatus,
  ruleEngine:           enhanced.ruleEngine,
  callAI:               enhanced.callAI,

  // 本文件独有
  aiCostTracker,
};