diff --git a/lib/ruby_llm.rb b/lib/ruby_llm.rb
index dfe1c5cd..3e5c17a3 100644
--- a/lib/ruby_llm.rb
+++ b/lib/ruby_llm.rb
@@ -29,6 +29,7 @@
 )
 loader.ignore("#{__dir__}/tasks")
 loader.ignore("#{__dir__}/generators")
+loader.ignore("#{__dir__}/ruby_llm/railtie.rb")
 loader.setup
 
 # A delightful Ruby interface to modern AI language models.
diff --git a/lib/ruby_llm/railtie.rb b/lib/ruby_llm/railtie.rb
index 14db08bf..97a368b0 100644
--- a/lib/ruby_llm/railtie.rb
+++ b/lib/ruby_llm/railtie.rb
@@ -1,33 +1,35 @@
 # frozen_string_literal: true
 
-module RubyLLM
-  # Rails integration for RubyLLM
-  class Railtie < Rails::Railtie
-    initializer 'ruby_llm.inflections' do
-      ActiveSupport::Inflector.inflections(:en) do |inflect|
-        inflect.acronym 'RubyLLM'
+if defined?(Rails::Railtie)
+  module RubyLLM
+    # Rails integration for RubyLLM
+    class Railtie < Rails::Railtie
+      initializer 'ruby_llm.inflections' do
+        ActiveSupport::Inflector.inflections(:en) do |inflect|
+          inflect.acronym 'RubyLLM'
+        end
       end
-    end
 
-    initializer 'ruby_llm.active_record' do
-      ActiveSupport.on_load :active_record do
-        if RubyLLM.config.use_new_acts_as
-          require 'ruby_llm/active_record/acts_as'
-          ::ActiveRecord::Base.include RubyLLM::ActiveRecord::ActsAs
-        else
-          require 'ruby_llm/active_record/acts_as_legacy'
-          ::ActiveRecord::Base.include RubyLLM::ActiveRecord::ActsAsLegacy
+      initializer 'ruby_llm.active_record' do
+        ActiveSupport.on_load :active_record do
+          if RubyLLM.config.use_new_acts_as
+            require 'ruby_llm/active_record/acts_as'
+            ::ActiveRecord::Base.include RubyLLM::ActiveRecord::ActsAs
+          else
+            require 'ruby_llm/active_record/acts_as_legacy'
+            ::ActiveRecord::Base.include RubyLLM::ActiveRecord::ActsAsLegacy
 
-          Rails.logger.warn(
-            "\n!!! RubyLLM's legacy acts_as API is deprecated and will be removed in RubyLLM 2.0.0. " \
-            "Please consult the migration guide at https://rubyllm.com/upgrading-to-1-7/\n"
-          )
+            Rails.logger.warn(
+              "\n!!! RubyLLM's legacy acts_as API is deprecated and will be removed in RubyLLM 2.0.0. " \
+              "Please consult the migration guide at https://rubyllm.com/upgrading-to-1-7/\n"
+            )
+          end
         end
       end
-    end
 
-    rake_tasks do
-      load 'tasks/ruby_llm.rake'
+      rake_tasks do
+        load 'tasks/ruby_llm.rake'
+      end
     end
   end
 end