Dayanand,
Try maintaining all the different Trading partners in the customer master, that way system will let you the option of manually choosing the right partner based on the company code that is being traded with.
Wouldn't this work?
Thanks,
Ravin