diff --git a/bin/omarchy-agent-usage-claude b/bin/omarchy-agent-usage-claude index c01bab39..7805c4fa 100755 --- a/bin/omarchy-agent-usage-claude +++ b/bin/omarchy-agent-usage-claude @@ -374,8 +374,7 @@ def scan_pi_usage(max_age_seconds: float) -> dict[str, Any] | None: if entry.get("type") != "message" or message.get("role") != "assistant": continue provider = str(message.get("provider") or "") - api = str(message.get("api") or "") - if provider != "anthropic" and not api.startswith("anthropic"): + if provider != "anthropic": continue unique_key = f"{path}:{entry.get('id') or line_number}" if unique_key in seen: diff --git a/test/shell.d/agent-usage-claude-scanner-test.sh b/test/shell.d/agent-usage-claude-scanner-test.sh index 84beb807..e1407f3c 100644 --- a/test/shell.d/agent-usage-claude-scanner-test.sh +++ b/test/shell.d/agent-usage-claude-scanner-test.sh @@ -112,6 +112,7 @@ mkdir -p "$PI_HOME/.pi/agent/sessions/project" "$PI_HOME/.omp/agent/sessions/pro cat >"$PI_HOME/.pi/agent/sessions/project/pi.jsonl" <"$PI_HOME/.omp/agent/sessions/project/omp.jsonl" <