Batch Type Extension Parameters

I am working on a batch type extension for the CUB and I wonder if we have any control over the ID that is used for the Edit Commit portion. Right now it is giving me the primary record id which is not conducive to my application, I would rather have the batch row id. Is there a way to change this or is this determined by the batch type of the CUB which uses the primary record id as the determining field for the add or edit SPROC call?