feat: Add route for retrieving current usage statistics

This commit is contained in:
Will Miao
2025-04-22 09:39:00 +08:00
parent 85f987d15c
commit 8cb78ad931

View File

@@ -11,6 +11,7 @@ class UsageStatsRoutes:
def setup_routes(app):
"""Register usage stats routes"""
app.router.add_post('/loras/api/update-usage-stats', UsageStatsRoutes.update_usage_stats)
app.router.add_get('/loras/api/get-usage-stats', UsageStatsRoutes.get_usage_stats)
@staticmethod
async def update_usage_stats(request):
@@ -47,3 +48,22 @@ class UsageStatsRoutes:
'success': False,
'error': str(e)
}, status=500)
@staticmethod
async def get_usage_stats(request):
"""Get current usage statistics"""
try:
usage_stats = UsageStats()
stats = await usage_stats.get_stats()
return web.json_response({
'success': True,
'data': stats
})
except Exception as e:
logger.error(f"Failed to get usage stats: {e}", exc_info=True)
return web.json_response({
'success': False,
'error': str(e)
}, status=500)