diff --git a/Appraisals b/Appraisals
index 6fc69a7..e1d18f1 100644
--- a/Appraisals
+++ b/Appraisals
@@ -23,3 +23,7 @@ end
 appraise "ruby_llm-current" do
   gem "ruby_llm"
 end
+
+appraise "ruby_llm-main" do
+  gem "ruby_llm", github: "crmne/ruby_llm", branch: "main"
+end
diff --git a/gemfiles/ruby_llm_1.10.gemfile b/gemfiles/ruby_llm_1.10.gemfile
index 17d757e..b2d5ac6 100644
--- a/gemfiles/ruby_llm_1.10.gemfile
+++ b/gemfiles/ruby_llm_1.10.gemfile
@@ -11,6 +11,7 @@ group :development, :test do
   gem "rake", "~> 13.0"
   gem "rubocop", "~> 1.86"
   gem "rubocop-minitest", "~> 0.39"
+  gem "ruby-lsp", "~> 0.26"
   gem "simplecov", "~> 0.22"
 end
 
diff --git a/gemfiles/ruby_llm_1.13.gemfile b/gemfiles/ruby_llm_1.13.gemfile
index 3b2b964..a985670 100644
--- a/gemfiles/ruby_llm_1.13.gemfile
+++ b/gemfiles/ruby_llm_1.13.gemfile
@@ -11,6 +11,7 @@ group :development, :test do
   gem "rake", "~> 13.0"
   gem "rubocop", "~> 1.86"
   gem "rubocop-minitest", "~> 0.39"
+  gem "ruby-lsp", "~> 0.26"
   gem "simplecov", "~> 0.22"
 end
 
diff --git a/gemfiles/ruby_llm_1.14.gemfile b/gemfiles/ruby_llm_1.14.gemfile
index 43d02b8..e3659b7 100644
--- a/gemfiles/ruby_llm_1.14.gemfile
+++ b/gemfiles/ruby_llm_1.14.gemfile
@@ -11,6 +11,7 @@ group :development, :test do
   gem "rake", "~> 13.0"
   gem "rubocop", "~> 1.86"
   gem "rubocop-minitest", "~> 0.39"
+  gem "ruby-lsp", "~> 0.26"
   gem "simplecov", "~> 0.22"
 end
 
diff --git a/gemfiles/ruby_llm_1.5.gemfile b/gemfiles/ruby_llm_1.5.gemfile
index 2abcb89..51a70b8 100644
--- a/gemfiles/ruby_llm_1.5.gemfile
+++ b/gemfiles/ruby_llm_1.5.gemfile
@@ -11,6 +11,7 @@ group :development, :test do
   gem "rake", "~> 13.0"
   gem "rubocop", "~> 1.86"
   gem "rubocop-minitest", "~> 0.39"
+  gem "ruby-lsp", "~> 0.26"
   gem "simplecov", "~> 0.22"
 end
 
diff --git a/gemfiles/ruby_llm_1.6.gemfile b/gemfiles/ruby_llm_1.6.gemfile
index 3b9c757..d5b0e73 100644
--- a/gemfiles/ruby_llm_1.6.gemfile
+++ b/gemfiles/ruby_llm_1.6.gemfile
@@ -11,6 +11,7 @@ group :development, :test do
   gem "rake", "~> 13.0"
   gem "rubocop", "~> 1.86"
   gem "rubocop-minitest", "~> 0.39"
+  gem "ruby-lsp", "~> 0.26"
   gem "simplecov", "~> 0.22"
 end
 
diff --git a/gemfiles/ruby_llm_current.gemfile b/gemfiles/ruby_llm_current.gemfile
index 3c45106..ce31b8d 100644
--- a/gemfiles/ruby_llm_current.gemfile
+++ b/gemfiles/ruby_llm_current.gemfile
@@ -11,6 +11,7 @@ group :development, :test do
   gem "rake", "~> 13.0"
   gem "rubocop", "~> 1.86"
   gem "rubocop-minitest", "~> 0.39"
+  gem "ruby-lsp", "~> 0.26"
   gem "simplecov", "~> 0.22"
 end
 
diff --git a/gemfiles/ruby_llm_main.gemfile b/gemfiles/ruby_llm_main.gemfile
new file mode 100644
index 0000000..93e8a0f
--- /dev/null
+++ b/gemfiles/ruby_llm_main.gemfile
@@ -0,0 +1,18 @@
+# This file was generated by Appraisal
+
+source "https://gem.coop"
+
+gem "ruby_llm", branch: "main", git: "https://github.com/crmne/ruby_llm"
+
+group :development, :test do
+  gem "appraisal", "~> 2.5"
+  gem "minitest", "~> 6.0"
+  gem "minitest-mock", "~> 5.27"
+  gem "rake", "~> 13.0"
+  gem "rubocop", "~> 1.86"
+  gem "rubocop-minitest", "~> 0.39"
+  gem "ruby-lsp", "~> 0.26"
+  gem "simplecov", "~> 0.22"
+end
+
+gemspec path: "../"
diff --git a/lib/ruby_llm/test/test_provider.rb b/lib/ruby_llm/test/test_provider.rb
index 6207117..23db4fa 100644
--- a/lib/ruby_llm/test/test_provider.rb
+++ b/lib/ruby_llm/test/test_provider.rb
@@ -11,6 +11,12 @@ def initialize(provider, test_harness)
         @test_harness = test_harness
       end
 
+      # There are a number of cases that access `provider.class.display_name`, which avoids the delegation. Adding
+      # dummy stub for this. TBD if this needs to be more sophisticated in the future.
+      def self.display_name
+        "TestProvider"
+      end
+
       def complete(...)
         parameters = CompleteParameters.capture_from(__getobj__, ...)
         @test_harness.record_request(parameters)