Add token model to enable support for schema generation