mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-06-19 02:05:37 +08:00
3332e9f4c3
`.annotaterb.yml` has carried `classified_sort: true` since the project
switched from `annotate` to `annotaterb` (commit 0eab7daea4, July 2025),
but annotaterb's default behaviour is to compare the existing schema
block against what it would generate and skip the rewrite when the
column list matches — even when the *ordering* of those columns differs.
The result is that models which haven't had a schema change since the
config landed never get reordered, and `classified_sort` drift
accumulates indefinitely.
`--force` makes annotaterb always rewrite, so a single `bin/rake
annotate:clean` run brings every model into the canonical format and
keeps them there. Every schema block is now grouped primary-key →
regular columns → timestamps → foreign keys (alphabetical within each
group). Pure annotation comment change — no code modifications.
Also cleans up the rake task to avoid string interpolation for `system`
calls.
85 lines
2.3 KiB
Ruby
Vendored
85 lines
2.3 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
class LlmQuota < ActiveRecord::Base
|
|
self.table_name = "llm_quotas"
|
|
|
|
belongs_to :group
|
|
belongs_to :llm_model
|
|
has_many :llm_quota_usages
|
|
|
|
validates :group_id, presence: true
|
|
# we can not validate on create cause it breaks build
|
|
validates :llm_model_id, presence: true, on: :update
|
|
validates :duration_seconds, presence: true, numericality: { greater_than: 0 }
|
|
validates :max_tokens, numericality: { only_integer: true, greater_than: 0, allow_nil: true }
|
|
validates :max_usages, numericality: { greater_than: 0, allow_nil: true }
|
|
|
|
validate :at_least_one_limit
|
|
|
|
def self.check_quotas!(llm, user)
|
|
return true if user.blank?
|
|
quotas = joins(:group).where(llm_model: llm).where(group: user.groups)
|
|
|
|
return true if quotas.empty?
|
|
errors =
|
|
quotas.map do |quota|
|
|
usage = LlmQuotaUsage.find_or_create_for(user: user, llm_quota: quota)
|
|
begin
|
|
usage.check_quota!
|
|
nil
|
|
rescue LlmQuotaUsage::QuotaExceededError => e
|
|
e
|
|
end
|
|
end
|
|
|
|
return if errors.include?(nil)
|
|
|
|
raise errors.first
|
|
end
|
|
|
|
def self.log_usage(llm, user, input_tokens, output_tokens)
|
|
return if user.blank?
|
|
|
|
quotas = joins(:group).where(llm_model: llm).where(group: user.groups)
|
|
|
|
quotas.each do |quota|
|
|
usage = LlmQuotaUsage.find_or_create_for(user: user, llm_quota: quota)
|
|
usage.increment_usage!(input_tokens: input_tokens, output_tokens: output_tokens)
|
|
end
|
|
end
|
|
|
|
def available_tokens
|
|
max_tokens
|
|
end
|
|
|
|
def available_usages
|
|
max_usages
|
|
end
|
|
|
|
private
|
|
|
|
def at_least_one_limit
|
|
if max_tokens.nil? && max_usages.nil?
|
|
errors.add(:base, I18n.t("discourse_ai.errors.quota_required"))
|
|
end
|
|
end
|
|
end
|
|
|
|
# == Schema Information
|
|
#
|
|
# Table name: llm_quotas
|
|
#
|
|
# id :bigint not null, primary key
|
|
# duration_seconds :integer not null
|
|
# max_tokens :integer
|
|
# max_usages :integer
|
|
# created_at :datetime not null
|
|
# updated_at :datetime not null
|
|
# group_id :bigint not null
|
|
# llm_model_id :bigint not null
|
|
#
|
|
# Indexes
|
|
#
|
|
# index_llm_quotas_on_group_id_and_llm_model_id (group_id,llm_model_id) UNIQUE
|
|
# index_llm_quotas_on_llm_model_id (llm_model_id)
|
|
#
|
