the code snippet I pasted typically goes in a

beforeSubmit

on a UE script for the originating record (i.e. if you want line keys to flow from Sales Order to Invoice then you'd only need this on the SalesOrder user event)