mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-09-05 07:26:39 +08:00
snaily
2072f54ca1
主要变更： - 新增 `app/handler/error_handler.py`，引入 `handle_route_errors` 异步上下文管理器，用于统一处理路由中的错误和日志记录。 - 在 `openai_routes` 和 `openai_compatiable_routes` 中应用 `handle_route_errors`，移除冗余的 try-except 块，简化路由逻辑。 - 将 `OpenAICompatiableService` 移动到 `app/service/openai_compatiable/` 目录下。 - 将 `StatsService` 移动到 `app/service/stats/` 目录下，并更新相关导入路径。 - 修复 `response_handler` 中处理 Gemini API 响应时 `inlineData` 字段的错误（原为 `inline_data`）。 - 修复 `openai_routes` 和 `openai_compatiable_routes` 中处理图像生成聊天（如 imagen3-chat）时未正确使用付费 API key 的问题。 - 在 `requirements.txt` 中将 `httpx` 更改为 `httpx[socks]`，以增加 SOCKS 代理支持。
58 lines
2.1 KiB
Python
58 lines
2.1 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.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)]
|
|
)
|
|
|
|
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:
|
|
logger.error(f"Error fetching key usage details for key {key[:4]}...: {e}")
|
|
raise HTTPException(
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
detail=f"获取密钥使用详情时出错: {e}"
|
|
) |