diff --git a/enterprise/app/models/enterprise/account.rb b/enterprise/app/models/enterprise/account.rb index c2954a812..869843e3b 100644 --- a/enterprise/app/models/enterprise/account.rb +++ b/enterprise/app/models/enterprise/account.rb @@ -8,7 +8,7 @@ module Enterprise::Account inboxes: get_limits(:inboxes).to_i, captain: { documents: get_captain_limits(:documents), - generated_responses: get_captain_limits(:responses) + responses: get_captain_limits(:responses) } } end diff --git a/enterprise/app/models/enterprise/inbox.rb b/enterprise/app/models/enterprise/inbox.rb index 957118366..4cd627e41 100644 --- a/enterprise/app/models/enterprise/inbox.rb +++ b/enterprise/app/models/enterprise/inbox.rb @@ -16,7 +16,7 @@ module Enterprise::Inbox private def more_responses? - account.usage_limits[:captain][:generated_responses][:current_available].positive? + account.usage_limits[:captain][:responses][:current_available].positive? end def get_agent_ids_over_assignment_limit(limit) diff --git a/spec/enterprise/jobs/captain/conversation/response_builder_job_spec.rb b/spec/enterprise/jobs/captain/conversation/response_builder_job_spec.rb index a5e3ba919..1e4a6e824 100644 --- a/spec/enterprise/jobs/captain/conversation/response_builder_job_spec.rb +++ b/spec/enterprise/jobs/captain/conversation/response_builder_job_spec.rb @@ -28,7 +28,7 @@ RSpec.describe Captain::Conversation::ResponseBuilderJob, type: :job do it 'increments usage response' do described_class.perform_now(conversation, assistant) account.reload - expect(account.usage_limits[:captain][:generated_responses][:consumed]).to eq(1) + expect(account.usage_limits[:captain][:responses][:consumed]).to eq(1) end end end diff --git a/spec/enterprise/models/account_spec.rb b/spec/enterprise/models/account_spec.rb index b063084c1..894fabca6 100644 --- a/spec/enterprise/models/account_spec.rb +++ b/spec/enterprise/models/account_spec.rb @@ -71,7 +71,7 @@ RSpec.describe Account, type: :model do it 'incrementing responses updates usage_limits' do account.increment_response_usage - responses_limits = account.usage_limits[:captain][:generated_responses] + responses_limits = account.usage_limits[:captain][:responses] expect(account.limits['captain_responses']).to eq 1 expect(responses_limits[:consumed]).to eq 1 @@ -82,13 +82,13 @@ RSpec.describe Account, type: :model do account.limits['captain_responses'] = 30 account.save! - responses_limits = account.usage_limits[:captain][:generated_responses] + responses_limits = account.usage_limits[:captain][:responses] expect(responses_limits[:consumed]).to eq 30 expect(responses_limits[:current_available]).to eq captain_limits[:startups][:responses] - 30 account.reset_response_usage - responses_limits = account.usage_limits[:captain][:generated_responses] + responses_limits = account.usage_limits[:captain][:responses] expect(account.limits['captain_responses']).to eq 0 expect(responses_limits[:consumed]).to eq 0 @@ -107,14 +107,14 @@ RSpec.describe Account, type: :model do account.limits['captain_responses'] = 3000 account.save! - responses_limits = account.usage_limits[:captain][:generated_responses] + responses_limits = account.usage_limits[:captain][:responses] expect(responses_limits[:consumed]).to eq 3000 expect(responses_limits[:current_available]).to eq 0 account.limits['captain_responses'] = -100 account.save! - responses_limits = account.usage_limits[:captain][:generated_responses] + responses_limits = account.usage_limits[:captain][:responses] expect(responses_limits[:consumed]).to eq 0 expect(responses_limits[:current_available]).to eq captain_limits[:startups][:responses] end @@ -138,7 +138,7 @@ RSpec.describe Account, type: :model do it 'returns limits based on custom attributes' do usage_limits = account.usage_limits expect(usage_limits[:captain][:documents][:total_count]).to eq(5555) - expect(usage_limits[:captain][:generated_responses][:total_count]).to eq(9999) + expect(usage_limits[:captain][:responses][:total_count]).to eq(9999) end end