All easy:

def callback_code(task):
    ctx = task.get_context()
    code = ctx.get(msg_code)
    print(f'Code: {code}')

We can get context straight from Task object!