mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-07-09 23:53:29 +00:00
Related: https://github.com/discourse/discourse-translator/pull/310 This commit includes all the jobs and event hooks to localize posts, topics, and categories. A few notes: - `feature_name: "translation"` because the site setting is `ai-translation` and module is `Translation` - we will switch to proper ai-feature in the near future, and can consider using the persona_user as `localization.localizer_user_id` - keeping things flat within the module for now as we will be moving to ai-feature soon and have to rearrange - Settings renamed/introduced are: - ai_translation_backfill_rate (0) - ai_translation_backfill_limit_to_public_content (true) - ai_translation_backfill_max_age_days (5) - ai_translation_verbose_logs (false)
61 lines
1.4 KiB
Ruby
61 lines
1.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module DiscourseAi
|
|
module Translation
|
|
class BaseTranslator
|
|
def initialize(text, target_language)
|
|
@text = text
|
|
@target_language = target_language
|
|
end
|
|
|
|
def translate
|
|
prompt =
|
|
DiscourseAi::Completions::Prompt.new(
|
|
prompt_template,
|
|
messages: [{ type: :user, content: formatted_content, id: "user" }],
|
|
)
|
|
|
|
structured_output =
|
|
DiscourseAi::Completions::Llm.proxy(SiteSetting.ai_translation_model).generate(
|
|
prompt,
|
|
user: Discourse.system_user,
|
|
feature_name: "translation",
|
|
response_format: response_format,
|
|
)
|
|
|
|
structured_output&.read_buffered_property(:translation)
|
|
end
|
|
|
|
def formatted_content
|
|
{ content: @text, target_language: @target_language }.to_json
|
|
end
|
|
|
|
def response_format
|
|
{
|
|
type: "json_schema",
|
|
json_schema: {
|
|
name: "reply",
|
|
schema: {
|
|
type: "object",
|
|
properties: {
|
|
translation: {
|
|
type: "string",
|
|
},
|
|
},
|
|
required: ["translation"],
|
|
additionalProperties: false,
|
|
},
|
|
strict: true,
|
|
},
|
|
}
|
|
end
|
|
|
|
private
|
|
|
|
def prompt_template
|
|
raise NotImplementedError
|
|
end
|
|
end
|
|
end
|
|
end
|