mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-11-04 00:18:39 +00:00
f57c1bb0f6
We pass the text to the current LLM and ask them to generate a StableDifussion prompt. We'll use that to generate 4 samples, temporarily creating uploads and returning their short URLs.
54 lines
2.0 KiB
Ruby
54 lines
2.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "../../../support/openai_completions_inference_stubs"
|
|
require_relative "../../../support/stable_difussion_stubs"
|
|
|
|
RSpec.describe DiscourseAi::AiHelper::Painter do
|
|
subject(:painter) { described_class.new }
|
|
|
|
fab!(:user) { Fabricate(:user) }
|
|
|
|
before do
|
|
SiteSetting.ai_stability_api_url = "https://api.stability.dev"
|
|
SiteSetting.ai_stability_api_key = "abc"
|
|
end
|
|
|
|
describe "#commission_thumbnails" do
|
|
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
|
|
expected_prompt = [
|
|
{ role: "system", content: <<~TEXT },
|
|
Provide me a StableDiffusion prompt to generate an image that illustrates the following post in 40 words or less, be creative.
|
|
TEXT
|
|
{ role: "user", content: raw_content },
|
|
]
|
|
|
|
OpenAiCompletionsInferenceStubs.stub_response(expected_prompt, expected_image_prompt)
|
|
StableDiffusionStubs.new.stub_response(expected_image_prompt, artifacts)
|
|
|
|
thumbnails = subject.commission_thumbnails(raw_content, user)
|
|
thumbnail_urls = Upload.last(4).map(&:short_url)
|
|
|
|
expect(thumbnails).to contain_exactly(*thumbnail_urls)
|
|
end
|
|
end
|
|
end
|