From 0db7c3aebfd9ceb942dff2163449b9db1d2363e3 Mon Sep 17 00:00:00 2001 From: Shivam Mishra Date: Wed, 22 Jan 2025 10:07:29 +0530 Subject: [PATCH] feat: use custom error --- .../api/v1/accounts/captain/documents_controller.rb | 2 +- enterprise/app/models/captain/document.rb | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/enterprise/app/controllers/api/v1/accounts/captain/documents_controller.rb b/enterprise/app/controllers/api/v1/accounts/captain/documents_controller.rb index 299d1d97d..594aa0642 100644 --- a/enterprise/app/controllers/api/v1/accounts/captain/documents_controller.rb +++ b/enterprise/app/controllers/api/v1/accounts/captain/documents_controller.rb @@ -23,7 +23,7 @@ class Api::V1::Accounts::Captain::DocumentsController < Api::V1::Accounts::BaseC @document = @assistant.documents.build(document_params) @document.save! - rescue StandardError => e + rescue Captain::Document::LimitExceededError => e render_could_not_create_error(e.message) end diff --git a/enterprise/app/models/captain/document.rb b/enterprise/app/models/captain/document.rb index 3332f863a..8331d291b 100644 --- a/enterprise/app/models/captain/document.rb +++ b/enterprise/app/models/captain/document.rb @@ -20,6 +20,7 @@ # index_captain_documents_on_status (status) # class Captain::Document < ApplicationRecord + class LimitExceededError < StandardError; end self.table_name = 'captain_documents' belongs_to :assistant, class_name: 'Captain::Assistant' @@ -69,6 +70,6 @@ class Captain::Document < ApplicationRecord def ensure_within_plan_limit limits = account.usage_limits[:captain][:documents] - raise 'Document limit exceeded' unless limits[:current_available].positive? + raise LimitExceededError, 'Document limit exceeded' unless limits[:current_available].positive? end end