Hi Abhishek,
This is quite possible with the customization itself.
You need to update your specific partner determination procedure at item level also with your custom partner function.
Let me know if still not working after adding in partner procedure determination.
Best Regards,
Dharmakasi.