1

video-gen-api/app/admin_api/api_keys/routes.py

@@ -261,6 +261,8 @@ async def save_upscale_config(
 async def get_usage(
     key_id: str = Path(...),
     days: int = Query(30, ge=1, le=365),
+    page: int = Query(1, ge=1),
+    page_size: int = Query(20, ge=1, le=100),
     admin: User = Depends(get_admin_user),
     db: AsyncSession = Depends(get_db),
 ) -> ApiUsageSummaryResponse:
@@ -271,7 +273,7 @@ async def get_usage(
         raise HTTPException(status_code=404, detail="API Key 不存在")
 
     summary = await usage_log_service.get_usage_summary(db, api_key_id=key_id, days=days)
-    total, logs = await usage_log_service.list_usage_logs(db, api_key_id=key_id, limit=100)
+    total, logs = await usage_log_service.list_usage_logs(db, api_key_id=key_id, limit=page_size, skip=(page - 1) * page_size)
 
     return ApiUsageSummaryResponse(
         total_requests=summary["total_requests"],
@@ -280,6 +282,9 @@ async def get_usage(
         success_count=summary["success_count"],
         failed_count=summary["failed_count"],
         avg_duration_ms=summary["avg_duration_ms"],
+        total=total,
+        page=page,
+        page_size=page_size,
         items=[
             ApiUsageLogResponse(
                 id=log.id,
@@ -312,6 +317,7 @@ async def list_all_usage(
     status_filter: str | None = Query(None, alias="status"),
     start_date: datetime | None = None,
     end_date: datetime | None = None,
+    search: str | None = None,
     admin: User = Depends(get_admin_user),
     db: AsyncSession = Depends(get_db),
 ) -> dict:
@@ -336,6 +342,12 @@ async def list_all_usage(
         filters.append(ApiUsageLog.created_at >= start_date)
     if end_date:
         filters.append(ApiUsageLog.created_at <= end_date)
+    if search:
+        search_pattern = f"%{search}%"
+        filters.append(
+            (ApiKey.company_name.ilike(search_pattern))
+            | (ApiKey.api_key_prefix.ilike(search_pattern))
+        )
 
     for f in filters:
         query = query.where(f)