diff --git a/lib/completions/endpoints/fake.rb b/lib/completions/endpoints/fake.rb index 70b04543..d307ada5 100644 --- a/lib/completions/endpoints/fake.rb +++ b/lib/completions/endpoints/fake.rb @@ -81,7 +81,7 @@ module DiscourseAi end def self.delays - @delays ||= Array.new(10) { rand * 6 } + @delays ||= Array.new(10) { Rails.env.test? ? 0 : rand(0..5) } end def self.delays=(delays) diff --git a/spec/jobs/scheduled/summaries_backfill_spec.rb b/spec/jobs/scheduled/summaries_backfill_spec.rb index f3757cf5..ccf23b06 100644 --- a/spec/jobs/scheduled/summaries_backfill_spec.rb +++ b/spec/jobs/scheduled/summaries_backfill_spec.rb @@ -7,9 +7,6 @@ RSpec.describe Jobs::SummariesBackfill do let(:limit) { 24 } # guarantee two summaries per batch let(:intervals) { 12 } # budget is split into intervals. Job runs every five minutes. - before { DiscourseAi::Completions::Endpoints::Fake.delays = [] } - after { DiscourseAi::Completions::Endpoints::Fake.reset! } - before do assign_fake_provider_to(:ai_summarization_model) SiteSetting.ai_summarization_enabled = true diff --git a/spec/lib/modules/summarization/fold_content_spec.rb b/spec/lib/modules/summarization/fold_content_spec.rb index 1cb174ec..7f6fafaf 100644 --- a/spec/lib/modules/summarization/fold_content_spec.rb +++ b/spec/lib/modules/summarization/fold_content_spec.rb @@ -10,9 +10,6 @@ RSpec.describe DiscourseAi::Summarization::FoldContent do before { SiteSetting.ai_summarization_enabled = true } - before { DiscourseAi::Completions::Endpoints::Fake.delays = [] } - after { DiscourseAi::Completions::Endpoints::Fake.reset! } - describe "#summarize" do before do # Make sure each content fits in a single chunk.