feat(cosmos): support custom document IDs and partition keys (#8699)