Add type annotations to the in-mem storage

so we can start bikeshedding on it before using type annotation elsewhere

---

Migrated from T1343 (view on Phabricator)