From 5a84969c967b2014bbd94d2bfe044e342f61bf6a Mon Sep 17 00:00:00 2001
From: Keegan George <kgeorge13@gmail.com>
Date: Tue, 19 Dec 2023 12:55:43 -0800
Subject: [PATCH] FIX: Illustrate post icon and translation not appearing
 correctly (#371)

---
 lib/ai_helper/assistant.rb   | 2 +-
 lib/ai_helper/entry_point.rb | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/ai_helper/assistant.rb b/lib/ai_helper/assistant.rb
index 03e6fa54..7c965fa8 100644
--- a/lib/ai_helper/assistant.rb
+++ b/lib/ai_helper/assistant.rb
@@ -141,7 +141,7 @@ module DiscourseAi
           %w[post]
         when "summarize"
           %w[post]
-        when "illustrate_posts"
+        when "illustrate_post"
           %w[composer]
         else
           %w[composer post]
diff --git a/lib/ai_helper/entry_point.rb b/lib/ai_helper/entry_point.rb
index 3ab9daa9..87ebca24 100644
--- a/lib/ai_helper/entry_point.rb
+++ b/lib/ai_helper/entry_point.rb
@@ -7,7 +7,7 @@ module DiscourseAi
           Rails.root.join("plugins", "discourse-ai", "db", "fixtures", "ai_helper"),
         )
 
-        additional_icons = %w[spell-check language]
+        additional_icons = %w[spell-check language images]
         additional_icons.each { |icon| plugin.register_svg_icon(icon) }
 
         plugin.on(:chat_thread_created) do |thread|