You can create an input node inside the function and externally in your material input whatever you want. Something like:
Is this what you mean?
