mirror of
https://github.com/discourse/discourse.git
synced 2026-08-08 17:53:55 +08:00
86680f6b0dFollow up to https://github.com/discourse/discourse/pull/41586 1. **Bulk of the PR diff** -- there's a mix of usages for "description" and "caption", we'll stick to V1 terminology and keep "AI caption" instead of bringing in "description". 2. Move image caption out of AI helper, into its own dedicated "AI Feature" for consolidated settings 3. Update default values for the existing image caption agent - vision_enabled was default off, now on - update description 4. Validations and problem checks - via Site Setting validation - agent / llm - when enabling the feature - or when selecting invalid agent - inform admin when they've enabled the setting but the agent is misconfigured via ProblemCheck
60 lines
1.9 KiB
Ruby
Vendored
60 lines
1.9 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
describe "AI post image captions in RSS" do
|
|
fab!(:upload) do
|
|
UploadCreator.new(
|
|
file_from_fixtures(
|
|
"100x100.jpg",
|
|
"images",
|
|
Rails.root.join("plugins/discourse-ai/spec/fixtures").to_s,
|
|
),
|
|
"caption-image.jpg",
|
|
).create_for(Discourse.system_user.id)
|
|
end
|
|
|
|
fab!(:post) { Fabricate(:post, raw: "") }
|
|
|
|
before do
|
|
enable_current_plugin
|
|
configure_valid_caption_agent
|
|
SiteSetting.ai_post_image_captions_enabled = true
|
|
post.update_column(:cooked, post.cook(post.raw, topic_id: post.topic_id))
|
|
post.link_post_uploads
|
|
end
|
|
|
|
def configure_valid_caption_agent
|
|
llm_model = assign_fake_provider_to(:ai_default_llm_model)
|
|
llm_model.update!(vision_enabled: true)
|
|
caption_agent =
|
|
AiAgent.find_by(id: SiteSetting.ai_image_caption_agent.to_i) ||
|
|
Fabricate(:ai_agent, id: SiteSetting.ai_image_caption_agent.to_i)
|
|
caption_agent.update!(enabled: true, vision_enabled: true, default_llm_id: llm_model.id)
|
|
end
|
|
|
|
it "omits generated descriptions from topic feeds", :aggregate_failures do
|
|
description = "A generated RSS-only 字 description"
|
|
AiPostImageCaption.upsert_all(
|
|
[
|
|
{
|
|
post_id: post.id,
|
|
upload_id: upload.id,
|
|
base62_sha1: upload.base62_sha1,
|
|
locale: SiteSetting.default_locale,
|
|
description: description,
|
|
attempts: 0,
|
|
},
|
|
],
|
|
unique_by: DiscourseAi::PostImageCaptions::LOOKUP_INDEX,
|
|
)
|
|
|
|
processor = CookedPostProcessor.new(post)
|
|
processor.post_process
|
|
post.update_column(:cooked, processor.html)
|
|
|
|
get "/t/#{post.topic.slug}/#{post.topic.id}.rss"
|
|
|
|
expect(response.status).to eq(200)
|
|
expect(response.body).not_to include(description)
|
|
expect(response.body).not_to include("aria-description")
|
|
end
|
|
end
|
