Add sharding key logic

Why

We need to set the sharding key in the model since it validates on the key existing

What

• Add method to set the sharding key
• Add specs

Notes

See Investigate: Handle setting sharding key (#536752 - closed)