PDIO-342 Support storing prediction keys that come with queries