|
1 | 1 | # frozen_string_literal: true |
2 | 2 |
|
3 | | - class ::Jobs::BackfillChatbotQuotas < ::Jobs::Onceoff |
4 | | - def execute_onceoff(args) |
5 | | - return unless SiteSetting.chatbot_enabled |
| 3 | +class ::Jobs::BackfillChatbotQuotas < ::Jobs::Onceoff |
| 4 | + def execute_onceoff(args) |
| 5 | + return unless SiteSetting.chatbot_enabled |
6 | 6 |
|
7 | | - # Initialize Chatbot Quotas for all users as required |
8 | | - user_count = User.count |
9 | | - queries_field_count = UserCustomField.where(name: ::DiscourseChatbot::CHATBOT_REMAINING_QUOTA_QUERIES_CUSTOM_FIELD).count |
10 | | - token_field_count = UserCustomField.where(name: ::DiscourseChatbot::CHATBOT_REMAINING_QUOTA_TOKENS_CUSTOM_FIELD).count |
11 | | - Rails.logger.info "CHATBOT: Checked presence of Chatbot Custom Fields" |
12 | | - if user_count > queries_field_count * 2 || user_count > token_field_count * 2 |
13 | | - ::DiscourseChatbot::Bot.new.reset_all_quotas |
14 | | - Rails.logger.info "CHATBOT: Resetted Chatbot Quotas for all users as many users without required Chatbot Custom Fields" |
15 | | - end |
| 7 | + # Initialize Chatbot Quotas for all users as required |
| 8 | + user_count = User.count |
| 9 | + queries_field_count = UserCustomField.where(name: ::DiscourseChatbot::CHATBOT_REMAINING_QUOTA_QUERIES_CUSTOM_FIELD).count |
| 10 | + token_field_count = UserCustomField.where(name: ::DiscourseChatbot::CHATBOT_REMAINING_QUOTA_TOKENS_CUSTOM_FIELD).count |
| 11 | + Rails.logger.info "CHATBOT: Checked presence of Chatbot Custom Fields" |
| 12 | + if user_count > queries_field_count * 2 || user_count > token_field_count * 2 |
| 13 | + ::DiscourseChatbot::Bot.new.reset_all_quotas |
| 14 | + Rails.logger.info "CHATBOT: Resetted Chatbot Quotas for all users as many users without required Chatbot Custom Fields" |
16 | 15 | end |
17 | 16 | end |
18 | 17 | end |
0 commit comments