def METHOD_NAME(
    self,
    connection: UserOAuthSourceConnection,
    access_token: Optional[str] = None,
) -> UserOAuthSourceConnection:
    """Set the access_token on the connection"""
    connection.access_token = access_token
    return connection