laantungir/routstr-core
1
0
Fork 0
mirror of https://github.com/Routstr/routstr-core.git synced 2026-10-06 12:38:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: Add type hints and improve sorting in get_revenue_by_model

Browse Source 
Co-authored-by: db2002dominic <db2002dominic@gmail.com>
This commit is contained in:
Cursor Agent
2025-11-16 19:53:03 +00:00
co-authored by db2002dominic
parent 757637d86e
commit d7e78b66d5
1 changed files with 20 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
routstr/core/admin.py
+20 -16
View File
@@ -3421,7 +3421,7 @@ async def get_revenue_by_model(
except Exception:
continue
models = []
models: list[dict[str, str | int | float]] = []
total_revenue = 0.0
for model, stats in model_stats.items():
@@ -3451,22 +3451,26 @@ async def get_revenue_by_model(
assert isinstance(failed_raw, int)
failed_val = failed_raw
models.append(
{
"model": model,
"revenue_sats": revenue_sats,
"refunds_sats": refunds_sats,
"net_revenue_sats": net_revenue_sats,
"requests": requests_val,
"successful": successful_val,
"failed": failed_val,
"avg_revenue_per_request": (
revenue_sats / successful_val if successful_val > 0 else 0
),
}
)
model_data: dict[str, str | int | float] = {
"model": model,
"revenue_sats": revenue_sats,
"refunds_sats": refunds_sats,
"net_revenue_sats": net_revenue_sats,
"requests": requests_val,
"successful": successful_val,
"failed": failed_val,
"avg_revenue_per_request": (
revenue_sats / successful_val if successful_val > 0 else 0
),
}
models.append(model_data)
models.sort(key=lambda x: float(x["net_revenue_sats"]), reverse=True)
def _sort_key(x: dict[str, str | int | float]) -> float:
val = x["net_revenue_sats"]
assert isinstance(val, (int, float))
return float(val)
models.sort(key=_sort_key, reverse=True)
return {
"models": models[:limit],