Allow passing prompt to `query`

[tomusher]

Allow the user to override the prompt when calling query so the user can override the default prompt