add_action( 'login_init', function () {

    if ( ($_GET['action'] ?? '') !== 'rp' ) {
        return;
    }

    if ( empty($_GET['key']) || empty($_GET['login']) ) {
        return;
    }

    wp_redirect(
        add_query_arg(
            [
                'key'   => sanitize_text_field( $_GET['key'] ),
                'login' => sanitize_text_field( $_GET['login'] ),
            ],
            site_url('/moje-konto/lost-password/')
        )
    );
    exit;
});