Create a configuration for each step and suppress features that occur after the step
Go into the configuration properties
Set the Bill of Materials options to Configuration Name
Go into the custom properties within the configuration properties. Set the property name as part number and assign the part number. You'll need to do this for each step, but it should all populate correctly into a BOM.