Mark cache as a protocol and update type hints to reflect

Mark cache as a protocol and update type hints to reflect #3915