from aiogram.dispatcher.filters.state import State, StatesGroup
class Form(StatesGroup):
unique_code = State()
await state.update_data(unique_code=int(message.text)) # Это если есть id реферала
unique_code = data['unique_code'] # Тут мы получаем id реферала