Function _collect_bearer_token_callbacks

python/copilot/client.py:182–204

Collect per-provider ``bearer_token_provider`` callbacks keyed by provider name. The singular, whole-session ``provider`` uses the implicit ``_DEFAULT_BEARER_TOKEN_PROVIDER_NAME``; ``providers`` entries use their own ``name``. The callbacks are never serialized — the wire conversion emi

(
    provider: ProviderConfig | None,
    providers: list[NamedProviderConfig] | None,
)

def _collect_bearer_token_callbacks(
    provider: ProviderConfig | None,
    providers: list[NamedProviderConfig] | None,
) -> dict[str, BearerTokenProvider]:
    """Collect per-provider ``bearer_token_provider`` callbacks keyed by provider name.

    The singular, whole-session ``provider`` uses the implicit
    ``_DEFAULT_BEARER_TOKEN_PROVIDER_NAME``; ``providers`` entries use their own
    ``name``. The callbacks are never serialized — the wire conversion emits
    ``hasBearerTokenProvider: true`` instead and the runtime calls back over
    ``providerToken.getToken``.
    """
    callbacks: dict[str, BearerTokenProvider] = {}
    if provider is not None:
        singular = provider.get("bearer_token_provider")
        if singular is not None:
            callbacks[_DEFAULT_BEARER_TOKEN_PROVIDER_NAME] = singular
    if providers:
        for named in providers:
            callback = named.get("bearer_token_provider")
            if callback is not None:
                callbacks[named["name"]] = callback
    return callbacks


def _validate_session_fs_config(config: SessionFsConfig) -> None: