Improve type hints for the relations datastore.