mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-06-09 17:49:42 +08:00
c99e090ea9
新增功能允许用户在 Keys 状态页面点击“详情”按钮，查看指定 API 密钥在过去 24 小时内按模型分类的请求次数统计。
主要变更包括：
后端:
- 新增 `app/router/stats_routes.py`，包含 `/api/key-usage-details/{key}` API 端点用于获取密钥使用详情。
- 重构 `app/service/stats_service.py`，将统计相关函数封装到 `StatsService` 类中，并添加 `get_key_usage_details_last_24h` 方法。
- 在 `app/router/routes.py` 中注册新的 `stats_routes`，并更新对 `stats_service` 的调用方式以使用类实例。
- 更新 `app/log/logger.py` 添加 `get_scheduler_routes` 日志记录器，并在 `app/router/scheduler_routes.py` 中使用它。
前端:
- 在 `app/templates/keys_status.html` 中为每个有效和无效密钥列表项添加“详情”按钮。
- 在 `app/templates/keys_status.html` 中添加用于显示密钥使用详情的模态框 HTML 结构。
- 在 `app/static/js/keys_status.js` 中添加 JavaScript 函数 (`showKeyUsageDetails`, `closeKeyUsageDetailsModal`, `renderKeyUsageDetails`) 来处理按钮点击事件、调用后端 API、控制模态框显示/隐藏以及渲染获取到的统计数据。
60 lines
2.3 KiB
Python
60 lines
2.3 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, Request
|
||
from starlette import status
|
||
from app.core.security import verify_auth_token
|
||
from app.service.stats_service import StatsService
|
||
from app.log.logger import get_stats_logger # 使用路由日志记录器
|
||
|
||
logger = get_stats_logger()
|
||
|
||
|
||
# 认证检查的辅助函数
|
||
async def verify_token(request: Request):
|
||
auth_token = request.cookies.get("auth_token")
|
||
if not auth_token or not verify_auth_token(auth_token):
|
||
logger.warning("Unauthorized access attempt to scheduler API")
|
||
raise HTTPException(
|
||
status_code=status.HTTP_401_UNAUTHORIZED,
|
||
detail="Not authenticated",
|
||
headers={"WWW-Authenticate": "Bearer"},
|
||
)
|
||
|
||
router = APIRouter(
|
||
prefix="/api",
|
||
tags=["stats"],
|
||
dependencies=[Depends(verify_token)] # Assuming API routes need authentication
|
||
)
|
||
|
||
stats_service = StatsService()
|
||
|
||
@router.get("/key-usage-details/{key}",
|
||
summary="获取指定密钥最近24小时的模型调用次数",
|
||
description="根据提供的 API 密钥，返回过去24小时内每个模型被调用的次数统计。")
|
||
async def get_key_usage_details(key: str):
|
||
"""
|
||
Retrieves the model usage count for a specific API key within the last 24 hours.
|
||
|
||
Args:
|
||
key: The API key to get usage details for.
|
||
|
||
Returns:
|
||
A dictionary with model names as keys and their call counts as values.
|
||
Example: {"gemini-pro": 10, "gemini-1.5-pro-latest": 5}
|
||
|
||
Raises:
|
||
HTTPException: If an error occurs during data retrieval.
|
||
"""
|
||
try:
|
||
usage_details = await stats_service.get_key_usage_details_last_24h(key)
|
||
if usage_details is None:
|
||
# Handle case where key might be valid but has no recent usage,
|
||
# or if the service layer explicitly returns None for other reasons.
|
||
# Returning an empty dict is usually fine for the frontend.
|
||
return {}
|
||
return usage_details
|
||
except Exception as e:
|
||
# Log the exception details here if needed
|
||
print(f"Error fetching key usage details for key {key[:4]}...: {e}")
|
||
raise HTTPException(
|
||
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
||
detail=f"获取密钥使用详情时出错: {e}"
|
||
) |