Implement partitioning in Upsert / Lookup

[leekeiabstraction]

Search before asking

• I searched in the issues and found nothing similar.

Description

Partitioning in Upsert / Lookup is currently left as unimplemented/todo. We should implement it to achieve parity with Java side

Willingness to contribute

• I'm willing to submit a PR!

[zuston]

Nice to see this!