From 88811f97bf876e77fcaab7a84cbe098ea2613de6 Mon Sep 17 00:00:00 2001
From: Keegan George <kgeorge13@gmail.com>
Date: Tue, 15 Jul 2025 10:23:58 -0700
Subject: [PATCH] fixes

---
 lib/ai_helper/painter.rb                           | 2 +-
 lib/configuration/llm_dependency_validator.rb      | 3 ++-
 spec/requests/admin/ai_features_controller_spec.rb | 1 +
 spec/requests/admin/ai_llms_controller_spec.rb     | 2 +-
 4 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/lib/ai_helper/painter.rb b/lib/ai_helper/painter.rb
index 9be8b95a..3e51534f 100644
--- a/lib/ai_helper/painter.rb
+++ b/lib/ai_helper/painter.rb
@@ -66,7 +66,7 @@ module DiscourseAi
             messages: [{ type: :user, content: text, id: user.username }],
           )
 
-        DiscourseAi::Completions::Llm.proxy(SiteSetting.ai_helper_model).generate(
+        DiscourseAi::Completions::Llm.proxy(SiteSetting.ai_default_llm_model).generate(
           prompt,
           user: user,
           feature_name: "illustrate_post",
diff --git a/lib/configuration/llm_dependency_validator.rb b/lib/configuration/llm_dependency_validator.rb
index 0cf715fe..bf6a5c12 100644
--- a/lib/configuration/llm_dependency_validator.rb
+++ b/lib/configuration/llm_dependency_validator.rb
@@ -10,7 +10,8 @@ module DiscourseAi
       def valid_value?(val)
         return true if val == "f"
 
-        if @opts[:name] == :ai_summarization_enabled || @opts[:name] == :ai_helper_enabled
+        if @opts[:name] == :ai_summarization_enabled || @opts[:name] == :ai_helper_enabled ||
+             @opts[:name] == :ai_embeddings_semantic_search_enabled
           has_llms = LlmModel.count > 0
           @no_llms_configured = !has_llms
           has_llms
diff --git a/spec/requests/admin/ai_features_controller_spec.rb b/spec/requests/admin/ai_features_controller_spec.rb
index f826e0bc..e7b2cc84 100644
--- a/spec/requests/admin/ai_features_controller_spec.rb
+++ b/spec/requests/admin/ai_features_controller_spec.rb
@@ -10,6 +10,7 @@ RSpec.describe DiscourseAi::Admin::AiFeaturesController do
 
   before do
     sign_in(admin)
+    assign_fake_provider_to(:ai_default_llm_model)
     SiteSetting.ai_bot_enabled = true
     SiteSetting.discourse_ai_enabled = true
   end
diff --git a/spec/requests/admin/ai_llms_controller_spec.rb b/spec/requests/admin/ai_llms_controller_spec.rb
index e0aaa27e..e0444e65 100644
--- a/spec/requests/admin/ai_llms_controller_spec.rb
+++ b/spec/requests/admin/ai_llms_controller_spec.rb
@@ -515,7 +515,7 @@ RSpec.describe DiscourseAi::Admin::AiLlmsController do
     end
 
     it "validates the model is not in use" do
-      fake_llm = assign_fake_provider_to(:ai_default_llm_model)
+      fake_llm = assign_fake_provider_to(:ai_helper_model)
 
       delete "/admin/plugins/discourse-ai/ai-llms/#{fake_llm.id}.json"