[FEATURE] Update api to use WalletGenerator class.