mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-06-19 05:59:26 +08:00
5a00b47523
Cached topic summaries that became outdated after post edits could remain visible longer than intended, and the 1-hour throttle could delay corrective regeneration. This update makes edit-driven staleness a special case: users who can regenerate summaries now bypass the throttle and refresh immediately, while users who cannot regenerate are not served edit-outdated cached content at all. In practice, stale summaries from edited content are no longer a fallback path, but normal throttle behavior is preserved for non-edit staleness such as newly added posts. --- **Security Advisory:** https://github.com/discourse/discourse/security/advisories/GHSA-pr9m-5hpq-wc57
77 lines
2.2 KiB
Ruby
Vendored
77 lines
2.2 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
module Jobs
|
|
class StreamTopicAiSummary < ::Jobs::Base
|
|
sidekiq_options retry: false
|
|
|
|
def execute(args)
|
|
return unless topic = Topic.find_by(id: args[:topic_id])
|
|
return unless user = User.find_by(id: args[:user_id])
|
|
|
|
strategy = DiscourseAi::Summarization.topic_summary(topic)
|
|
return if strategy.nil?
|
|
|
|
summarization_service = DiscourseAi::TopicSummarization.new(strategy, user)
|
|
cached_summary = summarization_service.cached_summary
|
|
|
|
guardian = Guardian.new(user)
|
|
return if !guardian.can_see_summary?(topic, cached_summary: cached_summary)
|
|
|
|
return unless guardian.can_see?(topic)
|
|
|
|
skip_age_check = !!args[:skip_age_check]
|
|
|
|
streamed_summary = +""
|
|
start = Time.now
|
|
|
|
begin
|
|
summary =
|
|
summarization_service.summarize(skip_age_check: skip_age_check) do |partial_summary|
|
|
streamed_summary << partial_summary
|
|
|
|
# Throttle updates.
|
|
if (Time.now - start > 0.3) || Rails.env.test?
|
|
payload = { done: false, ai_topic_summary: { summarized_text: streamed_summary } }
|
|
|
|
publish_update(topic, user, payload)
|
|
start = Time.now
|
|
end
|
|
end
|
|
|
|
publish_update(
|
|
topic,
|
|
user,
|
|
AiTopicSummarySerializer.new(summary, { scope: guardian }).as_json.merge(done: true),
|
|
)
|
|
rescue LlmCreditAllocation::CreditLimitExceeded => e
|
|
publish_error_update(topic, user, e)
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def publish_update(topic, user, payload)
|
|
MessageBus.publish("/discourse-ai/summaries/topic/#{topic.id}", payload, user_ids: [user.id])
|
|
end
|
|
|
|
def publish_error_update(topic, user, exception)
|
|
allocation = exception.allocation
|
|
|
|
details = {}
|
|
if allocation
|
|
details[:reset_time_relative] = allocation.relative_reset_time
|
|
details[:reset_time_absolute] = allocation.formatted_reset_time
|
|
end
|
|
|
|
payload = {
|
|
error: true,
|
|
error_type: "credit_limit_exceeded",
|
|
message: exception.message,
|
|
details: details,
|
|
done: true,
|
|
}
|
|
|
|
MessageBus.publish("/discourse-ai/summaries/topic/#{topic.id}", payload, user_ids: [user.id])
|
|
end
|
|
end
|
|
end
|
