mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-08-02 11:23:27 +00:00
d54cd1f602
This commit - normalizes locales like en_GB and variants to en. With this, the feature will not translate en_GB posts to en (or similarly pt_BR to pt_PT) - consolidates whether the feature is enabled in `DiscourseAi::Translation.enabled?` - similarly for backfill in `DiscourseAi::Translation.backfill_enabled?` - turns off backfill if `ai_translation_backfill_max_age_days` is 0 to keep true to what it says. Set it to a high number to backfill everything
69 lines
2.3 KiB
Ruby
69 lines
2.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Jobs
|
|
class LocalizeTopics < ::Jobs::Base
|
|
cluster_concurrency 1
|
|
sidekiq_options retry: false
|
|
|
|
def execute(args)
|
|
limit = args[:limit]
|
|
raise Discourse::InvalidParameters.new(:limit) if limit.blank? || limit <= 0
|
|
|
|
return if !DiscourseAi::Translation.backfill_enabled?
|
|
|
|
locales = SiteSetting.content_localization_supported_locales.split("|")
|
|
locales.each do |locale|
|
|
base_locale = locale.split("_").first
|
|
topics =
|
|
Topic
|
|
.joins(
|
|
"LEFT JOIN topic_localizations tl ON tl.topic_id = topics.id AND tl.locale LIKE '#{base_locale}%'",
|
|
)
|
|
.where(
|
|
"topics.created_at > ?",
|
|
SiteSetting.ai_translation_backfill_max_age_days.days.ago,
|
|
)
|
|
.where(deleted_at: nil)
|
|
.where("topics.user_id > 0")
|
|
.where.not(locale: nil)
|
|
.where("topics.locale NOT LIKE '#{base_locale}%'")
|
|
.where("tl.id IS NULL")
|
|
|
|
if SiteSetting.ai_translation_backfill_limit_to_public_content
|
|
# exclude all PMs
|
|
# and only include posts from public categories
|
|
topics =
|
|
topics
|
|
.where.not(archetype: Archetype.private_message)
|
|
.where(category_id: Category.where(read_restricted: false).select(:id))
|
|
else
|
|
# all regular topics, and group PMs
|
|
topics =
|
|
topics.where(
|
|
"topics.archetype != ? OR topics.id IN (SELECT topic_id FROM topic_allowed_groups)",
|
|
Archetype.private_message,
|
|
)
|
|
end
|
|
|
|
topics = topics.order(updated_at: :desc).limit(limit)
|
|
|
|
next if topics.empty?
|
|
|
|
topics.each do |topic|
|
|
begin
|
|
DiscourseAi::Translation::TopicLocalizer.localize(topic, locale)
|
|
rescue FinalDestination::SSRFDetector::LookupFailedError
|
|
# do nothing, there are too many sporadic lookup failures
|
|
rescue => e
|
|
DiscourseAi::Translation::VerboseLogger.log(
|
|
"Failed to translate topic #{topic.id} to #{locale}: #{e.message}\n\n#{e.backtrace[0..3].join("\n")}",
|
|
)
|
|
end
|
|
end
|
|
|
|
DiscourseAi::Translation::VerboseLogger.log("Translated #{topics.size} topics to #{locale}")
|
|
end
|
|
end
|
|
end
|
|
end
|