Ma/sora2api
1
0
Fork 0
mirror of https://github.com/TheSmallHanCat/sora2api.git synced 2026-02-14 18:14:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 新增每日用量统计 fix: 修复视频参数

Browse Source
This commit is contained in:
TheSmallHanCat
2025-11-18 09:14:05 +08:00
parent dba97c0fa4
commit 85f5c3620e
6 changed files with 123 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/api/admin.py
View File

@@ -480,24 +480,33 @@ async def get_stats(token: str = Depends(verify_admin_token)):
"""Get system statistics"""
tokens = await token_manager.get_all_tokens()
active_tokens = await token_manager.get_active_tokens()
total_images = 0
total_videos = 0
total_errors = 0
today_images = 0
today_videos = 0
today_errors = 0
for token in tokens:
stats = await db.get_token_stats(token.id)
if stats:
total_images += stats.image_count
total_videos += stats.video_count
total_errors += stats.error_count
today_images += stats.today_image_count
today_videos += stats.today_video_count
today_errors += stats.today_error_count
return {
"total_tokens": len(tokens),
"active_tokens": len(active_tokens),
"total_images": total_images,
"total_videos": total_videos,
"total_errors": total_errors
"today_images": today_images,
"today_videos": today_videos,
"total_errors": total_errors,
"today_errors": today_errors
}
# Sora2 endpoints

6
src/api/routes.py
View File

@@ -111,9 +111,9 @@ async def create_chat_completion(
image_data = url.split("base64,", 1)[1]
else:
image_data = url
elif item.get("type") == "input_video":
# Extract video from input_video
video_url = item.get("videoUrl", {})
elif item.get("type") == "video_url":
# Extract video from video_url
video_url = item.get("video_url", {})
url = video_url.get("url", "")
if url.startswith("data:video") or url.startswith("data:application"):
# Extract base64 data from data URI