diff options
Diffstat (limited to 'deckit/modules')
| -rw-r--r-- | deckit/modules/llm.py | 22 |
1 files changed, 15 insertions, 7 deletions
diff --git a/deckit/modules/llm.py b/deckit/modules/llm.py index 2e1cda4..fb2366f 100644 --- a/deckit/modules/llm.py +++ b/deckit/modules/llm.py @@ -29,16 +29,24 @@ def make(cfg: dict) -> Module: except (OSError, ValueError): return {"offline": True} limits = data["/api/claude_limits"] - latest = limits[-1] if isinstance(limits, list) and limits else ( - limits if isinstance(limits, dict) else {}) + if isinstance(limits, dict): + latest = limits.get("latest") or {} + elif isinstance(limits, list) and limits: + latest = limits[-1] + else: + latest = {} return { "offline": False, "claude": { - "context_pct": latest.get("context_pct", data["/api/claude_session"].get("context_used")), - "five_hour": latest.get("five_hour"), - "seven_day": latest.get("seven_day"), - "monthly_used": latest.get("extra_used_credits"), - "monthly_limit": latest.get("extra_monthly_limit"), + "context_pct": latest.get("context_pct"), + "five_hour_pct": latest.get("five_hour_pct"), + "five_hour_resets_at": latest.get("five_hour_resets_at"), + "seven_day_pct": latest.get("seven_day_pct"), + "seven_day_resets_at": latest.get("seven_day_resets_at"), + "extra_used": latest.get("extra_used_credits"), + "extra_limit": latest.get("extra_monthly_limit"), + "extra_utilization": latest.get("extra_utilization"), + "extra_currency": latest.get("extra_currency"), }, "opencode": data["/api/summary"], } |