mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-02-05 19:18:11 +00:00
0634b85a81
* UX: Validations to Llm-backed features (except AI Bot) This change is part of an ongoing effort to prevent enabling a broken feature due to lack of configuration. We also want to explicit which provider we are going to use. For example, Claude models are available through AWS Bedrock and Anthropic, but the configuration differs. Validations are: * You must choose a model before enabling the feature. * You must turn off the feature before setting the model to blank. * You must configure each model settings before being able to select it. * Add provider name to summarization options * vLLM can technically support same models as HF * Check we can talk to the selected model * Check for Bedrock instead of anthropic as a site could have both creds setup
90 lines
3.3 KiB
Ruby
90 lines
3.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe DiscourseAi::AiHelper::Painter do
|
|
subject(:painter) { described_class.new }
|
|
|
|
fab!(:user) { Fabricate(:user) }
|
|
|
|
before do
|
|
SiteSetting.ai_helper_model = "fake:fake"
|
|
SiteSetting.ai_stability_api_url = "https://api.stability.dev"
|
|
SiteSetting.ai_stability_api_key = "abc"
|
|
SiteSetting.ai_openai_api_key = "abc"
|
|
SiteSetting.ai_openai_dall_e_3_url = "https://api.openai.com/v1/images/generations"
|
|
end
|
|
|
|
describe "#commission_thumbnails" do
|
|
context "when illustrate post model is stable_diffusion_xl" do
|
|
before { SiteSetting.ai_helper_illustrate_post_model = "stable_diffusion_xl" }
|
|
|
|
let(:artifacts) do
|
|
%w[
|
|
iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQAAAABJRU5ErkJggg==
|
|
iVBORw0KGgoAAAANSUhEUgAAAAoAAAAKCAYAAACNMs+9AAAAFUlEQVR42mP8z8BQz0AEYBxVSF+FABJADveWkH6oAAAAAElFTkSuQmCC
|
|
iVBORw0KGgoAAAANSUhEUgAAAAoAAAAKCAYAAACNMs+9AAAAFUlEQVR42mNk+M9Qz0AEYBxVSF+FAAhKDveksOjmAAAAAElFTkSuQmCC
|
|
iVBORw0KGgoAAAANSUhEUgAAAAoAAAAKCAYAAACNMs+9AAAAFUlEQVR42mNkYPhfz0AEYBxVSF+FAP5FDvcfRYWgAAAAAElFTkSuQmCC
|
|
]
|
|
end
|
|
|
|
let(:raw_content) do
|
|
"Poetry is a form of artistic expression that uses language aesthetically and rhythmically to evoke emotions and ideas."
|
|
end
|
|
|
|
let(:expected_image_prompt) { <<~TEXT.strip }
|
|
Visualize a vibrant scene of an inkwell bursting, spreading colors across a blank canvas,
|
|
embodying words in tangible forms, symbolizing the rhythm and emotion evoked by poetry,
|
|
under the soft glow of a full moon.
|
|
TEXT
|
|
|
|
it "returns 4 samples" do
|
|
StableDiffusionStubs.new.stub_response(expected_image_prompt, artifacts)
|
|
|
|
thumbnails =
|
|
DiscourseAi::Completions::Llm.with_prepared_responses([expected_image_prompt]) do
|
|
thumbnails = subject.commission_thumbnails(raw_content, user)
|
|
end
|
|
|
|
thumbnail_urls = Upload.last(4).map(&:short_url)
|
|
|
|
expect(
|
|
thumbnails.map { |upload_serializer| upload_serializer.short_url },
|
|
).to contain_exactly(*thumbnail_urls)
|
|
end
|
|
end
|
|
|
|
context "when illustrate post model is dall_e_3" do
|
|
before { SiteSetting.ai_helper_illustrate_post_model = "dall_e_3" }
|
|
|
|
let(:artifacts) do
|
|
%w[
|
|
iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQAAAABJRU5ErkJggg==
|
|
]
|
|
end
|
|
|
|
let(:raw_content) do
|
|
"Poetry is a form of artistic expression that uses language aesthetically and rhythmically to evoke emotions and ideas."
|
|
end
|
|
|
|
it "returns an image sample" do
|
|
post = Fabricate(:post)
|
|
|
|
data = [{ b64_json: artifacts.first, revised_prompt: "colors on a canvas" }]
|
|
WebMock
|
|
.stub_request(:post, "https://api.openai.com/v1/images/generations")
|
|
.with do |request|
|
|
json = JSON.parse(request.body, symbolize_names: true)
|
|
true
|
|
end
|
|
.to_return(status: 200, body: { data: data }.to_json)
|
|
|
|
thumbnails = subject.commission_thumbnails(raw_content, user)
|
|
thumbnail_urls = Upload.last(1).map(&:short_url)
|
|
|
|
expect(
|
|
thumbnails.map { |upload_serializer| upload_serializer.short_url },
|
|
).to contain_exactly(*thumbnail_urls)
|
|
end
|
|
end
|
|
end
|
|
end
|