mirror of
https://ghfast.top/https://github.com/discourse/discourse-translator.git
synced 2026-07-17 11:56:50 +08:00
7c2fe57ab6
This commit adds DiscourseAi as a provider for translation. It defines two DiscourseAi::Completions::Prompt, in translator and language_detector, that uses the set ai_helper_model. Validations are in place when this new provider is selected but its requirements are not set up (SiteSetting.ai_helper_model, SiteSetting.ai_enabled)
67 lines
2.6 KiB
Ruby
67 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "rails_helper"
|
|
|
|
describe DiscourseAi::Translator do
|
|
before do
|
|
Fabricate(:fake_model).tap do |fake_llm|
|
|
SiteSetting.public_send("ai_helper_model=", "custom:#{fake_llm.id}")
|
|
end
|
|
SiteSetting.ai_helper_enabled = true
|
|
end
|
|
|
|
describe ".translate" do
|
|
let(:text_to_translate) { "cats are great" }
|
|
let(:target_language) { "de" }
|
|
|
|
it "creates the correct prompt" do
|
|
allow(DiscourseAi::Completions::Prompt).to receive(:new).with(
|
|
<<~TEXT,
|
|
You are a highly skilled linguist and web programmer, with expertise in many languages, and very well versed in HTML.
|
|
Your task is to identify the language of the text I provide and accurately translate it into this language locale "de" while preserving the meaning, tone, and nuance of the original text.
|
|
The text will contain html tags, which must absolutely be preserved in the translation.
|
|
Maintain proper grammar, spelling, and punctuation in the translated version.
|
|
Wrap the translated text in a <translation> tag.
|
|
TEXT
|
|
messages: [{ type: :user, content: text_to_translate, id: "user" }],
|
|
).and_call_original
|
|
|
|
described_class.new(text_to_translate, target_language).translate
|
|
end
|
|
|
|
it "sends the translation prompt to the selected ai helper model" do
|
|
mock_prompt = instance_double(DiscourseAi::Completions::Prompt)
|
|
mock_llm = instance_double(DiscourseAi::Completions::Llm)
|
|
|
|
allow(DiscourseAi::Completions::Prompt).to receive(:new).and_return(mock_prompt)
|
|
allow(DiscourseAi::Completions::Llm).to receive(:proxy).with(
|
|
SiteSetting.ai_helper_model,
|
|
).and_return(mock_llm)
|
|
allow(mock_llm).to receive(:generate).with(
|
|
mock_prompt,
|
|
user: Discourse.system_user,
|
|
feature_name: "translator-translate",
|
|
)
|
|
|
|
described_class.new(text_to_translate, target_language).translate
|
|
end
|
|
|
|
it "returns the translation from the llm's response in the translation tag" do
|
|
DiscourseAi::Completions::Llm.with_prepared_responses(
|
|
["<translation>hur dur hur dur!</translation>"],
|
|
) do
|
|
expect(
|
|
described_class.new(text_to_translate, target_language).translate,
|
|
).to eq "hur dur hur dur!"
|
|
end
|
|
end
|
|
|
|
it "returns the raw response if the translation tag is not present" do
|
|
DiscourseAi::Completions::Llm.with_prepared_responses(["raw response."]) do
|
|
expect(
|
|
described_class.new(text_to_translate, target_language).translate,
|
|
).to eq "raw response."
|
|
end
|
|
end
|
|
end
|
|
end
|
