The UK implementation of ePrivacy makes this quite clear and the regulator uses this as an example [1] in their guidance.

If you are setting the cookie in direct response to a user initiated action, for functionality, that doesn't require consent.

[1] https://ico.org.uk/for-organisations/guide-to-pecr/guidance-...