Files
wehub-resource-sync 4b6817381b
CI (OpenClaw E2E) / openclaw test (push) Has been cancelled
CI / coverage-report (push) Has been cancelled
CI / test-kubernetes (push) Has been cancelled
CI / should-run-thorough (push) Has been cancelled
CI / test-thorough (cloudwatch-demo) (push) Has been cancelled
CI / test-thorough (flink-ecs) (push) Has been cancelled
CI / test-thorough (upstream-lambda) (push) Has been cancelled
CI / test-thorough (prefect-ecs-fargate) (push) Has been cancelled
Release / build-binaries (zip, opensre.exe, onefile, windows-latest, windows-x64) (push) Has been cancelled
Benchmark image — build + push to ECR (any adapter) / build + push (push) Has been cancelled
CI / quality (ubuntu-latest) (push) Has been cancelled
CI / test (tools-runtime) (push) Has been cancelled
CI / test (e2e-general) (push) Has been cancelled
CI / test (cli-runtime) (push) Has been cancelled
CI / test (e2e-provider-and-openclaw) (push) Has been cancelled
CI / test (integrations-and-misc) (push) Has been cancelled
Release / verify (push) Has been cancelled
Release / build-python-dist (push) Has been cancelled
Release / build-binaries (tar.gz, opensre, onedir, macos-15-intel, darwin-x64) (push) Has been cancelled
Release / build-binaries (tar.gz, opensre, onedir, macos-latest, darwin-arm64) (push) Has been cancelled
Release / build-binaries (tar.gz, opensre, onedir, ubuntu-22.04, linux-x64) (push) Has been cancelled
Release / publish-release (push) Has been cancelled
Release / publish-main-release (push) Has been cancelled
Interactive Shell Live (PR + post-merge) / turn-checks (no-LLM) (push) Has been cancelled
CodeQL / Analyze (python) (push) Has been cancelled
Interactive Shell Live (PR + post-merge) / turn-live shard ${{ matrix.shard_index }} (push) Has been cancelled
Release / prepare (push) Has been cancelled
Release / build-binaries (tar.gz, opensre, onedir, ubuntu-22.04-arm, linux-arm64) (push) Has been cancelled
Synthetic Deterministic Tests / Synthetic offline (deterministic) (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 13:10:45 +08:00

36 lines
1.1 KiB
Python

"""Provider-name → :class:`TokenSource` lookup table.
Mirrors :mod:`tools.system.fleet_monitoring.meters.registry`. Unknown names fall back
to :data:`null_token_source` so a new provider on the developer's
machine cannot crash the dashboard.
"""
from __future__ import annotations
from tools.system.fleet_monitoring.token_sources import (
NullTokenSource,
TokenSource,
null_token_source,
)
from tools.system.fleet_monitoring.token_sources.claude_code import ClaudeCodeJsonlSource
from tools.system.fleet_monitoring.token_sources.codex import CodexRolloutSource
TOKEN_SOURCE_REGISTRY: dict[str, TokenSource] = {
"claude-code": ClaudeCodeJsonlSource(),
"codex": CodexRolloutSource(),
"cursor": NullTokenSource(),
"aider": NullTokenSource(),
"gemini-cli": NullTokenSource(),
"antigravity-cli": NullTokenSource(),
"opencode": NullTokenSource(),
"kimi": NullTokenSource(),
"copilot": NullTokenSource(),
}
def get_token_source(provider: str) -> TokenSource:
return TOKEN_SOURCE_REGISTRY.get(provider, null_token_source)
__all__ = ["TOKEN_SOURCE_REGISTRY", "get_token_source"]