We have the same concern, mostly. I had TKR 12/24 that was “not right or not complete” as nothing was done for the patella.
Though not knowing it could be a patella problem I did look for a 'second opinion' and is so doing found a GREAT surgeon. He spotted the basis of my problem in 2 minutes just looking at my knee. X-rays confirmed it. So we're considering repairing my patella. Next Sunday I’ll get an MRI so we can plan details as necessary.
And since the patella rides on the femoral surface of your replaced part of your knee (assuming it was the femoral head replaced) a “replacement” patella should be good to consider.
You and my situation got me to searching for good info. I found these sites to be most helpful for me. For your consideration:

Good overview and descriptions -
https://www.physio-pedia.com/Patellofemoral_Joint
Good detail and ‘live action’ photos; and told me I’m a good candidate:
https://www.mayoclinic.org/documents/mc6247-0512-pdf/DOC-20079225
Good luck and God Speed toward resolution and comfort.