Enable retry for POST requests in streaming error handling

lib/ruby_llm/connection.rb

@@ -77,7 +77,8 @@ def setup_retry(faraday)
         interval_randomness: @config.retry_interval_randomness,
         backoff_factor: @config.retry_backoff_factor,
         exceptions: retry_exceptions,
-        retry_statuses: [429, 500, 502, 503, 504, 529]
+        retry_statuses: [429, 500, 502, 503, 504, 529],
+        methods: %i[delete get head options patch post put]
       }
     end
 

spec/ruby_llm/chat_error_spec.rb

@@ -29,6 +29,12 @@
 RSpec.describe RubyLLM::Chat do
   include_context 'with configured RubyLLM'
 
+  before do
+    RubyLLM.configure do |config|
+      config.max_retries = 0
+    end
+  end
+
   describe 'error handling' do
     CHAT_MODELS.each do |model_info|
       model = model_info[:model]

spec/ruby_llm/chat_streaming_spec.rb

@@ -88,6 +88,18 @@
               end
             end.to raise_error(expected_error_for(provider))
           end
+
+          it 'retries the request' do
+            stub_error_response(provider, :chunk)
+
+            expect do
+              chat.ask('Count from 1 to 3') do |chunk|
+                chunks << chunk
+              end
+            end.to raise_error(expected_error_for(provider))
+
+            expect(WebMock).to have_requested(:post, expected_url_for(provider)).times(3)
+          end
         end
 
         describe 'Faraday version 2' do # rubocop:disable RSpec/NestedGroups
@@ -124,6 +136,18 @@
               end
             end.to raise_error(expected_error_for(provider))
           end
+
+          it 'retries the request' do
+            stub_error_response(provider, :chunk)
+
+            expect do
+              chat.ask('Count from 1 to 3') do |chunk|
+                chunks << chunk
+              end
+            end.to raise_error(expected_error_for(provider))
+
+            expect(WebMock).to have_requested(:post, expected_url_for(provider)).times(3)
+          end
         end
       end
     end

spec/support/rubyllm_configuration.rb

@@ -28,10 +28,10 @@
       config.vertexai_location = ENV.fetch('GOOGLE_CLOUD_LOCATION', 'us-central1')
 
       config.request_timeout = 240
-      config.max_retries = 10
-      config.retry_interval = 1
-      config.retry_backoff_factor = 3
-      config.retry_interval_randomness = 0.5
+      config.max_retries = 2
+      config.retry_interval = 0.01
+      config.retry_backoff_factor = 0.1
+      config.retry_interval_randomness = 0.01
 
       config.model_registry_class = 'Model'
     end

spec/support/streaming_error_helpers.rb

@@ -154,6 +154,10 @@ def expected_error_for(provider)
     ERROR_HANDLING_CONFIGS[provider][:expected_error]
   end
 
+  def expected_url_for(provider)
+    ERROR_HANDLING_CONFIGS[provider][:url]
+  end
+
   def stub_error_response(provider, type)
     config = ERROR_HANDLING_CONFIGS[provider]
     return unless config