DEV: Force default to be set if it was found not to be set!

Author: keegangeorge

lib/ai_helper/assistant.rb

@@ -320,6 +320,13 @@ def self.find_ai_helper_model(helper_mode, persona_klass)
         if model_id.present?
           LlmModel.find_by(id: model_id)
         else
+          last_model_id = LlmModel.last&.id
+
+          # SiteSetting.ai_default_llm_model shouldn't be empty, but if it is, we set it to the last model.
+          if last_model_id.present? && SiteSetting.ai_default_llm_model.empty?
+            SiteSetting.set_and_log("ai_default_llm_model", "custom:#{last_model_id}", Discourse.system_user) # Remove legacy custom provider.
+          end
+
           LlmModel.last
         end
       end

lib/embeddings/semantic_search.rb

@@ -215,9 +215,19 @@ def find_ai_hyde_model(persona_klass)
         model_id =
           persona_klass.default_llm_id || SiteSetting.ai_default_llm_model&.split(":")&.last # Remove legacy custom provider.
 
-        return if model_id.blank?
+        if model_id.present?
+          LlmModel.find_by(id: model_id)
+        else
+          last_model_id = LlmModel.last&.id
+
+          # SiteSetting.ai_default_llm_model shouldn't be empty, but if it is, we set it to the last model.
+          if last_model_id.present? && SiteSetting.ai_default_llm_model.empty?
+            SiteSetting.set_and_log("ai_default_llm_model", "custom:#{last_model_id}", Discourse.system_user) # Remove legacy custom provider.
+          end
+
+          LlmModel.last
+        end
 
-        LlmModel.find_by(id: model_id)
       end
 
       private

lib/summarization.rb

@@ -62,6 +62,13 @@ def find_summarization_model(persona_klass)
         if model_id.present?
           LlmModel.find_by(id: model_id)
         else
+          last_model_id = LlmModel.last&.id
+
+          # SiteSetting.ai_default_llm_model shouldn't be empty, but if it is, we set it to the last model.
+          if last_model_id.present? && SiteSetting.ai_default_llm_model.empty?
+            SiteSetting.set_and_log("ai_default_llm_model", "custom:#{last_model_id}", Discourse.system_user) # Remove legacy custom provider.
+          end
+
           LlmModel.last
         end
       end