增加token计数，增加前端额度查询，更新readme

QUOTA_FEATURE.md

@@ -0,0 +1,122 @@
+# 模型额度管理功能
+
+## 功能说明
+
+新增了模型额度查看功能，可以在前端管理界面查看每个 Token 对应的模型剩余额度和重置时间。
+
+## 实现方案
+
+### 数据存储
+- **accounts.json**: 保持简洁，只存储核心认证信息
+- **data/quotas.json**: 新建文件，专门存储额度信息（轻量级持久化）
+- **内存缓存**: 5分钟缓存，避免频繁请求API
+- **自动清理**: 每小时清理超过1小时未更新的数据
+
+### 核心文件
+
+1. **src/api/client.js**
+   - 新增 `getModelsWithQuotas(token)` 函数
+   - 从 API 响应中提取 `quotaInfo` 字段
+   - 返回简化的额度数据结构
+
+2. **src/auth/quota_manager.js** (新建)
+   - 额度缓存管理
+   - 文件持久化
+   - UTC 时间转北京时间
+   - 自动清理过期数据
+
+3. **src/routes/admin.js**
+   - 新增 `GET /admin/tokens/:refreshToken/quotas` 接口
+   - 支持按需获取指定 Token 的额度信息
+
+4. **public/app.js**
+   - 新增 `toggleQuota()` 函数：展开/收起额度面板
+   - 新增 `loadQuota()` 函数：从API加载额度数据
+   - 新增 `renderQuota()` 函数：渲染进度条和额度信息
+
+5. **public/style.css**
+   - 新增额度展示相关样式
+   - 进度条样式（支持颜色渐变：绿色>50%，黄色20-50%，红色<20%）
+
+## 使用方法
+
+### 前端操作
+
+1. 登录管理界面
+2. 在 Token 卡片中点击 **"📊 查看额度"** 按钮
+3. 系统会自动加载该 Token 的所有模型额度信息
+4. 以进度条形式展示：
+   - 模型名称
+   - 剩余额度百分比（带颜色标识）
+   - 额度重置时间（北京时间）
+
+### 数据格式
+
+#### API 响应示例
+```json
+{
+  "success": true,
+  "data": {
+    "lastUpdated": 1765109350660,
+    "models": {
+      "gemini-2.0-flash-exp": {
+        "remaining": 0.972,
+        "resetTime": "01-07 15:27",
+        "resetTimeRaw": "2025-01-07T07:27:44Z"
+      },
+      "gemini-1.5-pro": {
+        "remaining": 0.85,
+        "resetTime": "01-07 16:15",
+        "resetTimeRaw": "2025-01-07T08:15:30Z"
+      }
+    }
+  }
+}
+```
+
+#### quotas.json 存储格式
+```json
+{
+  "meta": {
+    "lastCleanup": 1765109350660,
+    "ttl": 3600000
+  },
+  "quotas": {
+    "1//0eDtvmkC_KgZv": {
+      "lastUpdated": 1765109350660,
+      "models": {
+        "gemini-2.0-flash-exp": {
+          "r": 0.972,
+          "t": "2025-01-07T07:27:44Z"
+        }
+      }
+    }
+  }
+}
+```
+
+## 特性
+
+✅ **按需加载**: 只在用户点击时才获取额度信息  
+✅ **智能缓存**: 5分钟内重复查看使用缓存，减少API请求  
+✅ **自动清理**: 定期清理过期数据，保持文件轻量  
+✅ **可视化展示**: 进度条直观显示剩余额度  
+✅ **颜色标识**: 绿色(>50%)、黄色(20-50%)、红色(<20%)  
+✅ **时间转换**: 自动将UTC时间转换为北京时间  
+✅ **轻量存储**: 使用字段缩写，只存储有变化的模型  
+
+## 注意事项
+
+1. 首次查看额度时需要调用 Google API，可能需要几秒钟
+2. 额度信息会缓存5分钟，如需最新数据请等待缓存过期后重新查看
+3. quotas.json 文件会自动创建，无需手动配置
+4. 如果 Token 过期或无效，会显示错误提示
+
+## 测试
+
+启动服务后：
+```bash
+npm start
+```
+
+访问管理界面，点击任意 Token 的"查看额度"按钮即可测试功能。

README.md

@@ -79,6 +79,7 @@ npm start
   - 手动填入：直接输入 Access Token 和 Refresh Token
 - 🎯 **Token 管理**：
   - 查看所有 Token 的详细信息（Access Token 后缀、Project ID、过期时间）
+  - 📊 查看模型额度：按类型分组显示（Claude/Gemini/其他），实时查看剩余额度和重置时间
   - 一键启用/禁用 Token
   - 删除无效 Token
   - 实时刷新 Token 列表
@@ -109,6 +110,11 @@ npm start
 
 3. **管理 Token**
    - 查看 Token 卡片显示的状态和信息
+   - 点击「📊 查看额度」按钮查看该账号的模型额度信息
+     - 自动按模型类型分组（Claude/Gemini/其他）
+     - 显示剩余额度百分比和进度条
+     - 显示额度重置时间（北京时间）
+     - 支持「立即刷新」强制更新额度数据
    - 使用「启用/禁用」按钮控制 Token 状态
    - 使用「删除」按钮移除无效 Token
    - 点击「刷新」按钮更新列表

public/app.js

@@ -353,6 +353,7 @@ function renderTokens(tokens) {
                 </div>
             </div>
             <div class="token-actions">
+                <button class="btn btn-info" onclick="showQuotaModal('${token.refresh_token}')">📊 查看额度</button>
                 <button class="btn ${token.enable ? 'btn-warning' : 'btn-success'}" onclick="toggleToken('${token.refresh_token}', ${!token.enable})">
                     ${token.enable ? '⏸️ 禁用' : '▶️ 启用'}
                 </button>
@@ -417,6 +418,146 @@ async function deleteToken(refreshToken) {
     }
 }
 
+async function showQuotaModal(refreshToken) {
+    const modal = document.createElement('div');
+    modal.className = 'modal';
+    modal.innerHTML = `
+        <div class="modal-content" style="max-width: 600px;">
+            <div class="modal-title">📊 模型额度信息</div>
+            <div id="quotaContent" style="max-height: 60vh; overflow-y: auto;">
+                <div class="quota-loading">加载中...</div>
+            </div>
+            <div class="modal-actions">
+                <button class="btn btn-info" onclick="refreshQuotaData('${refreshToken}')">🔄 立即刷新</button>
+                <button class="btn btn-secondary" onclick="this.closest('.modal').remove()">关闭</button>
+            </div>
+        </div>
+    `;
+    document.body.appendChild(modal);
+    modal.onclick = (e) => { if (e.target === modal) modal.remove(); };
+    
+    await loadQuotaData(refreshToken);
+}
+
+async function loadQuotaData(refreshToken, forceRefresh = false) {
+    const quotaContent = document.getElementById('quotaContent');
+    if (!quotaContent) return;
+    
+    const refreshBtn = document.querySelector('.modal-content .btn-info');
+    if (refreshBtn) {
+        refreshBtn.disabled = true;
+        refreshBtn.textContent = '⏳ 加载中...';
+    }
+    
+    quotaContent.innerHTML = '<div class="quota-loading">加载中...</div>';
+    
+    try {
+        const url = `/admin/tokens/${encodeURIComponent(refreshToken)}/quotas${forceRefresh ? '?refresh=true' : ''}`;
+        const response = await fetch(url, {
+            headers: { 'Authorization': `Bearer ${authToken}` }
+        });
+        
+        const data = await response.json();
+        
+        if (data.success) {
+            const quotaData = data.data;
+            const models = quotaData.models;
+            
+            if (Object.keys(models).length === 0) {
+                quotaContent.innerHTML = '<div class="quota-empty">暂无额度信息</div>';
+                return;
+            }
+            
+            const lastUpdated = new Date(quotaData.lastUpdated).toLocaleString('zh-CN', {
+                month: '2-digit', day: '2-digit', hour: '2-digit', minute: '2-digit'
+            });
+            
+            // 按模型类型分组
+            const grouped = { claude: [], gemini: [], other: [] };
+            Object.entries(models).forEach(([modelId, quota]) => {
+                const item = { modelId, quota };
+                if (modelId.toLowerCase().includes('claude')) grouped.claude.push(item);
+                else if (modelId.toLowerCase().includes('gemini')) grouped.gemini.push(item);
+                else grouped.other.push(item);
+            });
+            
+            let html = `<div class="quota-header">更新于 ${lastUpdated}</div>`;
+            
+            // 渲染各组
+            if (grouped.claude.length > 0) {
+                html += '<div class="quota-group-title">🤖 Claude 模型</div>';
+                grouped.claude.forEach(({ modelId, quota }) => {
+                    const percentage = (quota.remaining * 100).toFixed(1);
+                    const barColor = percentage > 50 ? '#10b981' : percentage > 20 ? '#f59e0b' : '#ef4444';
+                    html += `
+                        <div class="quota-item">
+                            <div class="quota-model-name">${modelId}</div>
+                            <div class="quota-bar-container">
+                                <div class="quota-bar" style="width: ${percentage}%; background: ${barColor};"></div>
+                                <span class="quota-percentage">${percentage}%</span>
+                            </div>
+                            <div class="quota-reset">🔄 重置: ${quota.resetTime}</div>
+                        </div>
+                    `;
+                });
+            }
+            
+            if (grouped.gemini.length > 0) {
+                html += '<div class="quota-group-title">💎 Gemini 模型</div>';
+                grouped.gemini.forEach(({ modelId, quota }) => {
+                    const percentage = (quota.remaining * 100).toFixed(1);
+                    const barColor = percentage > 50 ? '#10b981' : percentage > 20 ? '#f59e0b' : '#ef4444';
+                    html += `
+                        <div class="quota-item">
+                            <div class="quota-model-name">${modelId}</div>
+                            <div class="quota-bar-container">
+                                <div class="quota-bar" style="width: ${percentage}%; background: ${barColor};"></div>
+                                <span class="quota-percentage">${percentage}%</span>
+                            </div>
+                            <div class="quota-reset">🔄 重置: ${quota.resetTime}</div>
+                        </div>
+                    `;
+                });
+            }
+            
+            if (grouped.other.length > 0) {
+                html += '<div class="quota-group-title">🔧 其他模型</div>';
+                grouped.other.forEach(({ modelId, quota }) => {
+                    const percentage = (quota.remaining * 100).toFixed(1);
+                    const barColor = percentage > 50 ? '#10b981' : percentage > 20 ? '#f59e0b' : '#ef4444';
+                    html += `
+                        <div class="quota-item">
+                            <div class="quota-model-name">${modelId}</div>
+                            <div class="quota-bar-container">
+                                <div class="quota-bar" style="width: ${percentage}%; background: ${barColor};"></div>
+                                <span class="quota-percentage">${percentage}%</span>
+                            </div>
+                            <div class="quota-reset">🔄 重置: ${quota.resetTime}</div>
+                        </div>
+                    `;
+                });
+            }
+            
+            quotaContent.innerHTML = html;
+        } else {
+            quotaContent.innerHTML = `<div class="quota-error">加载失败: ${data.message}</div>`;
+        }
+    } catch (error) {
+        if (quotaContent) {
+            quotaContent.innerHTML = `<div class="quota-error">加载失败: ${error.message}</div>`;
+        }
+    } finally {
+        if (refreshBtn) {
+            refreshBtn.disabled = false;
+            refreshBtn.textContent = '🔄 立即刷新';
+        }
+    }
+}
+
+async function refreshQuotaData(refreshToken) {
+    await loadQuotaData(refreshToken, true);
+}
+
 async function loadConfig() {
     try {
         const response = await fetch('/admin/config', {

public/style.css

@@ -163,3 +163,24 @@ button.loading::after { content: ''; position: absolute; width: 16px; height: 16
     .toast { right: 10px; left: 10px; min-width: auto; max-width: none; }
     .modal { padding: 10px; }
 }
+
+.btn-info { background: var(--info); color: white; }
+.token-actions { display: grid; grid-template-columns: 1fr 1fr 1fr; gap: 0.5rem; }
+
+.quota-section { margin-top: 1rem; padding-top: 1rem; border-top: 2px solid var(--border); }
+.quota-loading, .quota-error, .quota-empty { text-align: center; padding: 1rem; color: var(--text-light); font-size: 0.875rem; }
+.quota-error { color: var(--danger); }
+.quota-header { font-weight: 600; margin-bottom: 0.75rem; font-size: 0.9rem; color: var(--text); }
+.quota-item { margin-bottom: 1rem; }
+.quota-item:last-child { margin-bottom: 0; }
+.quota-model-name { font-size: 0.8rem; color: var(--text); margin-bottom: 0.5rem; font-family: monospace; font-weight: 500; }
+.quota-bar-container { position: relative; height: 24px; background: var(--border); border-radius: 12px; overflow: hidden; margin-bottom: 0.375rem; }
+.quota-bar { height: 100%; border-radius: 12px; transition: width 0.3s ease; position: relative; }
+.quota-percentage { position: absolute; right: 8px; top: 50%; transform: translateY(-50%); font-size: 0.75rem; font-weight: 600; color: white; text-shadow: 0 1px 2px rgba(0,0,0,0.3); z-index: 1; }
+.quota-reset { font-size: 0.75rem; color: var(--text-light); }
+.quota-group-title { font-size: 1rem; font-weight: 700; color: var(--text); margin: 1.25rem 0 0.75rem 0; padding-bottom: 0.5rem; border-bottom: 2px solid var(--border); }
+.quota-group-title:first-child { margin-top: 0; }
+
+@media (max-width: 768px) {
+    .token-actions { grid-template-columns: 1fr; }
+}

src/api/client.js

@@ -101,6 +101,7 @@ function parseAndEmitStreamChunk(line, state, callback) {
   
   try {
     const data = JSON.parse(line.slice(6));
+    //console.log(JSON.stringify(data));
     const parts = data.response?.candidates?.[0]?.content?.parts;
     
     if (parts) {
@@ -126,14 +127,28 @@ function parseAndEmitStreamChunk(line, state, callback) {
       }
     }
     
-    // 响应结束时发送工具调用
-    if (data.response?.candidates?.[0]?.finishReason && state.toolCalls.length > 0) {
+    // 响应结束时发送工具调用和使用统计
+    if (data.response?.candidates?.[0]?.finishReason) {
       if (state.thinkingStarted) {
         callback({ type: 'thinking', content: '\n</think>\n' });
         state.thinkingStarted = false;
       }
-      callback({ type: 'tool_calls', tool_calls: state.toolCalls });
-      state.toolCalls = [];
+      if (state.toolCalls.length > 0) {
+        callback({ type: 'tool_calls', tool_calls: state.toolCalls });
+        state.toolCalls = [];
+      }
+      // 提取 token 使用统计
+      const usage = data.response?.usageMetadata;
+      if (usage) {
+        callback({ 
+          type: 'usage', 
+          usage: {
+            prompt_tokens: usage.promptTokenCount || 0,
+            completion_tokens: usage.candidatesTokenCount || 0,
+            total_tokens: usage.totalTokenCount || 0
+          }
+        });
+      }
     }
   } catch (e) {
     // 忽略 JSON 解析错误
@@ -205,6 +220,7 @@ export async function getAvailableModels() {
       }
       data = await response.json();
     }
+    //console.log(JSON.stringify(data,null,2));
     const modelList = Object.keys(data.models).map(id => ({
         id,
         object: 'model',
@@ -227,6 +243,38 @@ export async function getAvailableModels() {
   }
 }
 
+export async function getModelsWithQuotas(token) {
+  const headers = buildHeaders(token);
+  
+  try {
+    let data;
+    if (useAxios) {
+      data = (await axios(buildAxiosConfig(config.api.modelsUrl, headers, {}))).data;
+    } else {
+      const response = await requester.antigravity_fetch(config.api.modelsUrl, buildRequesterConfig(headers, {}));
+      if (response.status !== 200) {
+        const errorBody = await response.text();
+        throw { status: response.status, message: errorBody };
+      }
+      data = await response.json();
+    }
+    
+    const quotas = {};
+    Object.entries(data.models || {}).forEach(([modelId, modelData]) => {
+      if (modelData.quotaInfo) {
+        quotas[modelId] = {
+          r: modelData.quotaInfo.remainingFraction,
+          t: modelData.quotaInfo.resetTime
+        };
+      }
+    });
+    
+    return quotas;
+  } catch (error) {
+    await handleApiError(error, token);
+  }
+}
+
 export async function generateAssistantResponseNoStream(requestBody, token) {
   
   const headers = buildHeaders(token);
@@ -246,7 +294,7 @@ export async function generateAssistantResponseNoStream(requestBody, token) {
   } catch (error) {
     await handleApiError(error, token);
   }
-  
+  //console.log(JSON.stringify(data));
   // 解析响应内容
   const parts = data.response?.candidates?.[0]?.content?.parts || [];
   let content = '';
@@ -273,14 +321,22 @@ export async function generateAssistantResponseNoStream(requestBody, token) {
     content = `<think>\n${thinkingContent}\n</think>\n${content}`;
   }
   
+  // 提取 token 使用统计
+  const usage = data.response?.usageMetadata;
+  const usageData = usage ? {
+    prompt_tokens: usage.promptTokenCount || 0,
+    completion_tokens: usage.candidatesTokenCount || 0,
+    total_tokens: usage.totalTokenCount || 0
+  } : null;
+  
   // 生图模型：转换为 markdown 格式
   if (imageUrls.length > 0) {
     let markdown = content ? content + '\n\n' : '';
     markdown += imageUrls.map(url => `![image](${url})`).join('\n\n');
-    return { content: markdown, toolCalls };
+    return { content: markdown, toolCalls, usage: usageData };
   }
   
-  return { content, toolCalls };
+  return { content, toolCalls, usage: usageData };
 }
 
 export function closeRequester() {

src/auth/quota_manager.js

@@ -0,0 +1,117 @@
+import fs from 'fs';
+import path from 'path';
+import { fileURLToPath } from 'url';
+import { log } from '../utils/logger.js';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+
+class QuotaManager {
+  constructor(filePath = path.join(__dirname, '..', '..', 'data', 'quotas.json')) {
+    this.filePath = filePath;
+    this.cache = new Map();
+    this.CACHE_TTL = 5 * 60 * 1000; // 5分钟缓存
+    this.CLEANUP_INTERVAL = 60 * 60 * 1000; // 1小时清理一次
+    this.ensureFileExists();
+    this.loadFromFile();
+    this.startCleanupTimer();
+  }
+
+  ensureFileExists() {
+    const dir = path.dirname(this.filePath);
+    if (!fs.existsSync(dir)) {
+      fs.mkdirSync(dir, { recursive: true });
+    }
+    if (!fs.existsSync(this.filePath)) {
+      fs.writeFileSync(this.filePath, JSON.stringify({ meta: { lastCleanup: Date.now(), ttl: this.CLEANUP_INTERVAL }, quotas: {} }, null, 2), 'utf8');
+    }
+  }
+
+  loadFromFile() {
+    try {
+      const data = fs.readFileSync(this.filePath, 'utf8');
+      const parsed = JSON.parse(data);
+      Object.entries(parsed.quotas || {}).forEach(([key, value]) => {
+        this.cache.set(key, value);
+      });
+    } catch (error) {
+      log.error('加载额度文件失败:', error.message);
+    }
+  }
+
+  saveToFile() {
+    try {
+      const quotas = {};
+      this.cache.forEach((value, key) => {
+        quotas[key] = value;
+      });
+      const data = {
+        meta: { lastCleanup: Date.now(), ttl: this.CLEANUP_INTERVAL },
+        quotas
+      };
+      fs.writeFileSync(this.filePath, JSON.stringify(data, null, 2), 'utf8');
+    } catch (error) {
+      log.error('保存额度文件失败:', error.message);
+    }
+  }
+
+  updateQuota(refreshToken, quotas) {
+    this.cache.set(refreshToken, {
+      lastUpdated: Date.now(),
+      models: quotas
+    });
+    this.saveToFile();
+  }
+
+  getQuota(refreshToken) {
+    const data = this.cache.get(refreshToken);
+    if (!data) return null;
+    
+    // 检查缓存是否过期
+    if (Date.now() - data.lastUpdated > this.CACHE_TTL) {
+      return null;
+    }
+    
+    return data;
+  }
+
+  cleanup() {
+    const now = Date.now();
+    let cleaned = 0;
+    
+    this.cache.forEach((value, key) => {
+      if (now - value.lastUpdated > this.CLEANUP_INTERVAL) {
+        this.cache.delete(key);
+        cleaned++;
+      }
+    });
+    
+    if (cleaned > 0) {
+      log.info(`清理了 ${cleaned} 个过期的额度记录`);
+      this.saveToFile();
+    }
+  }
+
+  startCleanupTimer() {
+    setInterval(() => this.cleanup(), this.CLEANUP_INTERVAL);
+  }
+
+  convertToBeijingTime(utcTimeStr) {
+    if (!utcTimeStr) return 'N/A';
+    try {
+      const utcDate = new Date(utcTimeStr);
+      return utcDate.toLocaleString('zh-CN', {
+        month: '2-digit',
+        day: '2-digit',
+        hour: '2-digit',
+        minute: '2-digit',
+        timeZone: 'Asia/Shanghai'
+      });
+    } catch (error) {
+      return 'N/A';
+    }
+  }
+}
+
+const quotaManager = new QuotaManager();
+export default quotaManager;

src/auth/token_manager.js

@@ -112,25 +112,35 @@ class TokenManager {
       token.access_token = response.data.access_token;
       token.expires_in = response.data.expires_in;
       token.timestamp = Date.now();
-      this.saveToFile();
+      this.saveToFile(token);
       return token;
     } catch (error) {
       throw { statusCode: error.response?.status, message: error.response?.data || error.message };
     }
   }
 
-  saveToFile() {
+  saveToFile(tokenToUpdate = null) {
     try {
       const data = fs.readFileSync(this.filePath, 'utf8');
       const allTokens = JSON.parse(data);
       
-      this.tokens.forEach(memToken => {
-        const index = allTokens.findIndex(t => t.refresh_token === memToken.refresh_token);
+      // 如果指定了要更新的token，直接更新它
+      if (tokenToUpdate) {
+        const index = allTokens.findIndex(t => t.refresh_token === tokenToUpdate.refresh_token);
         if (index !== -1) {
-          const { sessionId, ...tokenToSave } = memToken;
+          const { sessionId, ...tokenToSave } = tokenToUpdate;
           allTokens[index] = tokenToSave;
         }
-      });
+      } else {
+        // 否则更新内存中的所有token
+        this.tokens.forEach(memToken => {
+          const index = allTokens.findIndex(t => t.refresh_token === memToken.refresh_token);
+          if (index !== -1) {
+            const { sessionId, ...tokenToSave } = memToken;
+            allTokens[index] = tokenToSave;
+          }
+        });
+      }
       
       fs.writeFileSync(this.filePath, JSON.stringify(allTokens, null, 2), 'utf8');
     } catch (error) {
@@ -162,7 +172,7 @@ class TokenManager {
         if (!token.projectId) {
           if (config.skipProjectIdFetch) {
             token.projectId = generateProjectId();
-            this.saveToFile();
+            this.saveToFile(token);
             log.info(`...${token.access_token.slice(-8)}: 使用随机生成的projectId: ${token.projectId}`);
           } else {
             try {
@@ -174,7 +184,7 @@ class TokenManager {
                 continue;
               }
               token.projectId = projectId;
-              this.saveToFile();
+              this.saveToFile(token);
             } catch (error) {
               log.error(`...${token.access_token.slice(-8)}: 获取projectId失败:`, error.message);
               this.currentIndex = (this.currentIndex + 1) % this.tokens.length;
@@ -292,6 +302,7 @@ class TokenManager {
       
       return allTokens.map(token => ({
         refresh_token: token.refresh_token,
+        access_token: token.access_token,
         access_token_suffix: token.access_token ? `...${token.access_token.slice(-8)}` : 'N/A',
         expires_in: token.expires_in,
         timestamp: token.timestamp,

src/routes/admin.js

@@ -1,6 +1,7 @@
 import express from 'express';
 import { generateToken, authMiddleware } from '../auth/jwt.js';
 import tokenManager from '../auth/token_manager.js';
+import quotaManager from '../auth/quota_manager.js';
 import config, { getConfigJson, saveConfigJson } from '../config/config.js';
 import logger from '../utils/logger.js';
 import { generateProjectId } from '../utils/idGenerator.js';
@@ -8,6 +9,7 @@ import { parseEnvFile, updateEnvFile } from '../utils/envParser.js';
 import { reloadConfig } from '../utils/configReloader.js';
 import { OAUTH_CONFIG } from '../constants/oauth.js';
 import { deepMerge } from '../utils/deepMerge.js';
+import { getModelsWithQuotas } from '../api/client.js';
 import path from 'path';
 import { fileURLToPath } from 'url';
 import dotenv from 'dotenv';
@@ -170,4 +172,60 @@ router.put('/config', authMiddleware, (req, res) => {
   }
 });
 
+// 获取指定Token的模型额度
+router.get('/tokens/:refreshToken/quotas', authMiddleware, async (req, res) => {
+  try {
+    const { refreshToken } = req.params;
+    const forceRefresh = req.query.refresh === 'true';
+    const tokens = tokenManager.getTokenList();
+    let tokenData = tokens.find(t => t.refresh_token === refreshToken);
+    
+    if (!tokenData) {
+      return res.status(404).json({ success: false, message: 'Token不存在' });
+    }
+    
+    // 检查token是否过期，如果过期则刷新
+    if (tokenManager.isExpired(tokenData)) {
+      try {
+        tokenData = await tokenManager.refreshToken(tokenData);
+      } catch (error) {
+        logger.error('刷新token失败:', error.message);
+        return res.status(401).json({ success: false, message: 'Token已过期且刷新失败' });
+      }
+    }
+    
+    // 先从缓存获取（除非强制刷新）
+    let quotaData = forceRefresh ? null : quotaManager.getQuota(refreshToken);
+    
+    if (!quotaData) {
+      // 缓存未命中或强制刷新，从API获取
+      const token = { access_token: tokenData.access_token, refresh_token: refreshToken };
+      const quotas = await getModelsWithQuotas(token);
+      quotaManager.updateQuota(refreshToken, quotas);
+      quotaData = { lastUpdated: Date.now(), models: quotas };
+    }
+    
+    // 转换时间为北京时间
+    const modelsWithBeijingTime = {};
+    Object.entries(quotaData.models).forEach(([modelId, quota]) => {
+      modelsWithBeijingTime[modelId] = {
+        remaining: quota.r,
+        resetTime: quotaManager.convertToBeijingTime(quota.t),
+        resetTimeRaw: quota.t
+      };
+    });
+    
+    res.json({ 
+      success: true, 
+      data: { 
+        lastUpdated: quotaData.lastUpdated,
+        models: modelsWithBeijingTime 
+      } 
+    });
+  } catch (error) {
+    logger.error('获取额度失败:', error.message);
+    res.status(500).json({ success: false, message: error.message });
+  }
+});
+
 export default router;

src/server/index.js

@@ -41,8 +41,7 @@ const writeStreamData = (res, data) => {
 };
 
 // 工具函数：结束流式响应
-const endStream = (res, id, created, model, finish_reason) => {
-  writeStreamData(res, createStreamChunk(id, created, model, {}, finish_reason));
+const endStream = (res) => {
   res.write('data: [DONE]\n\n');
   res.end();
 };
@@ -134,22 +133,29 @@ app.post('/v1/chat/completions', async (req, res) => {
       setStreamHeaders(res);
       
       if (isImageModel) {
-        const { content } = await generateAssistantResponseNoStream(requestBody, token);
+        const { content, usage } = await generateAssistantResponseNoStream(requestBody, token);
         writeStreamData(res, createStreamChunk(id, created, model, { content }));
-        endStream(res, id, created, model, 'stop');
+        writeStreamData(res, { ...createStreamChunk(id, created, model, {}, 'stop'), usage });
+        endStream(res);
       } else {
         let hasToolCall = false;
+        let usageData = null;
         await generateAssistantResponse(requestBody, token, (data) => {
-          const delta = data.type === 'tool_calls' 
-            ? { tool_calls: data.tool_calls } 
-            : { content: data.content };
-          if (data.type === 'tool_calls') hasToolCall = true;
-          writeStreamData(res, createStreamChunk(id, created, model, delta));
+          if (data.type === 'usage') {
+            usageData = data.usage;
+          } else {
+            const delta = data.type === 'tool_calls' 
+              ? { tool_calls: data.tool_calls } 
+              : { content: data.content };
+            if (data.type === 'tool_calls') hasToolCall = true;
+            writeStreamData(res, createStreamChunk(id, created, model, delta));
+          }
         });
-        endStream(res, id, created, model, hasToolCall ? 'tool_calls' : 'stop');
+        writeStreamData(res, { ...createStreamChunk(id, created, model, {}, hasToolCall ? 'tool_calls' : 'stop'), usage: usageData });
+        endStream(res);
       }
     } else {
-      const { content, toolCalls } = await generateAssistantResponseNoStream(requestBody, token);
+      const { content, toolCalls, usage } = await generateAssistantResponseNoStream(requestBody, token);
       const message = { role: 'assistant', content };
       if (toolCalls.length > 0) message.tool_calls = toolCalls;
       
@@ -162,7 +168,8 @@ app.post('/v1/chat/completions', async (req, res) => {
           index: 0,
           message,
           finish_reason: toolCalls.length > 0 ? 'tool_calls' : 'stop'
-        }]
+        }],
+        usage
       });
     }
   } catch (error) {
@@ -174,7 +181,8 @@ app.post('/v1/chat/completions', async (req, res) => {
       if (stream) {
         setStreamHeaders(res);
         writeStreamData(res, createStreamChunk(id, created, model, { content: errorContent }));
-        endStream(res, id, created, model, 'stop');
+        writeStreamData(res, createStreamChunk(id, created, model, {}, 'stop'));
+        endStream(res);
       } else {
         res.json({
           id,

src/utils/utils.js

@@ -69,7 +69,7 @@ function handleAssistantMessage(message, antigravityMessages){
     lastMessage.parts.push(...antigravityTools)
   }else{
     const parts = [];
-    if (hasContent) parts.push({ text: message.content });
+    if (hasContent) parts.push({ text: message.content.trimEnd() });
     parts.push(...antigravityTools);
     
     antigravityMessages.push({