Limitar o número de tokens de IA que um usuário pode usar por dia

Eu meio que preciso de limites de taxa por usuário (não por grupo) agora, então estou tentando criar minha própria solução provisória. Limitar o número de prompts em vez de tokens seria bom. Estou pensando em um webhook em post_event que diga se é uma mensagem privada e um usuário está postando para um bot, para incrementar um campo personalizado ‘ai_query_count’ no usuário. Acho que essa parte funcionaria.

Então, o que fazer se a contagem ficar muito alta? Tentei um pouco de JavaScript em admin > customize > head que lê a contagem do usuário e tenta desabilitar o botão “Responder” se a contagem for muito alta, mas não consigo encontrar um termo seletor para obter o botão.

Ou talvez haja uma abordagem melhor. Qualquer ideia é bem-vinda!