FEATURE: migrate to new open ai image apis

Author: SamSaffron

config/locales/server.en.yml

@@ -55,7 +55,9 @@ en:
     ai_nsfw_flag_threshold_sexy: "Threshold for an image classified as sexy to be considered NSFW."
     ai_nsfw_models: "Models to use for NSFW inference."
 
-    ai_openai_api_key: "API key for OpenAI API. ONLY used for Dall-E. For GPT use the LLM config tab"
+    ai_openai_api_key: "API key for OpenAI API. ONLY used for Image creation and edits. For GPT use the LLM config tab"
+    ai_openai_image_generation_url: "URL for OpenAI image generation API"
+    ai_openai_image_edit_url: "URL for OpenAI image edit API"
 
     ai_helper_enabled: "Enable the AI helper."
     composer_ai_helper_allowed_groups: "Users on these groups will see the AI helper button in the composer."

config/settings.yml

@@ -26,7 +26,8 @@ discourse_ai:
     default: 60
     hidden: true
 
-  ai_openai_dall_e_3_url: "https://api.openai.com/v1/images/generations"
+  ai_openai_image_generation_url: "https://api.openai.com/v1/images/generations"
+  ai_openai_image_edit_url: "https://api.openai.com/v1/images/edits"
   ai_openai_embeddings_url:
     hidden: true
     default: "https://api.openai.com/v1/embeddings"

lib/ai_helper/painter.rb

@@ -21,17 +21,18 @@ def commission_thumbnails(input, user)
 
           base64_to_image(artifacts, user.id)
         elsif model == "dall_e_3"
-          api_key = SiteSetting.ai_openai_api_key
-          api_url = SiteSetting.ai_openai_dall_e_3_url
-
-          artifacts =
-            DiscourseAi::Inference::OpenAiImageGenerator
-              .perform!(input, api_key: api_key, api_url: api_url)
-              .dig(:data)
-              .to_a
-              .map { |art| art[:b64_json] }
-
-          base64_to_image(artifacts, user.id)
+          attribution =
+            I18n.t(
+              "discourse_ai.ai_helper.painter.attribution.#{SiteSetting.ai_helper_illustrate_post_model}",
+            )
+          results =
+            DiscourseAi::Inference::OpenAiImageGenerator.create_uploads!(
+              input,
+              model: "dall-e-3",
+              user_id: user.id,
+              title: attribution,
+            )
+          results.map { |result| UploadSerializer.new(result[:upload], root: false) }
         end
       end