Hi Stepan,
Just such a situation does sound like a valuable one to have the ability to store tokens in Axon Server; thanks for sharing that with us.
In the mean time you'll indeed have to configure a TokenStore implementation alongside the Neo4J Query Model you're updating.
To that end, you can use the 'InMemoryTokenStore', 'JdbcTokenStore', 'JpaTokenStore' and 'MongoTokenStore'.
If anything in this area is still unclear, feel free to reach out on this forum.
And concluding, I hope this helps you out Stepan!
Cheers,
Steven