I think use "chain_extension" need define custom env is complex, it is easy to use raw "call_chain_extension" function.