mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
from fastapi import Request
|
|
from pydantic import SecretStr
|
|
|
|
from openhands.integrations.provider import PROVIDER_TOKEN_TYPE, ProviderType
|
|
|
|
|
|
def get_provider_tokens(request: Request) -> PROVIDER_TOKEN_TYPE | None:
|
|
"""Get GitHub token from request state. For backward compatibility."""
|
|
return getattr(request.state, 'provider_tokens', None)
|
|
|
|
|
|
def get_access_token(request: Request) -> SecretStr | None:
|
|
return getattr(request.state, 'access_token', None)
|
|
|
|
|
|
def get_user_id(request: Request) -> str | None:
|
|
return getattr(request.state, 'user_id', None)
|
|
|
|
|
|
def get_github_token(request: Request) -> SecretStr | None:
|
|
provider_tokens = get_provider_tokens(request)
|
|
|
|
if provider_tokens and ProviderType.GITHUB in provider_tokens:
|
|
return provider_tokens[ProviderType.GITHUB].token
|
|
|
|
return None
|
|
|
|
|
|
def get_github_user_id(request: Request) -> str | None:
|
|
provider_tokens = get_provider_tokens(request)
|
|
|
|
if provider_tokens and ProviderType.GITHUB in provider_tokens:
|
|
return provider_tokens[ProviderType.GITHUB].user_id
|
|
|
|
return None
|