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
|