On put, validate that key fields on the entity are set to a value

Currently, a seemingly unrelated exception is thrown since the code expects all key fields to be set. We should throw some kind of invalid key exception so that the error is very clear.