LeadMagnet/leadchat
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

fix: Use current_available instead available to compute the document limit (#10776)

Browse Source 
The available limit for documents is now stored in
captain.documents.current_available. This PR fixes the limit sent on
crawler job.
This commit is contained in:
Pranav
2025-01-27 21:25:46 -08:00
committed by GitHub
parent cb42be8e65
commit cf0975ad94
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
enterprise/app/jobs/captain/documents/crawl_job.rb
View File

@@ -32,7 +32,7 @@ class Captain::Documents::CrawlJob < ApplicationJob
def perform_firecrawl_crawl(document)
captain_usage_limits = document.account.usage_limits[:captain] || {}
document_limit = captain_usage_limits[:documents] || {}
crawl_limit = [document_limit[:available] || 10, 500].min
crawl_limit = [document_limit[:current_available] || 10, 500].min
Captain::Tools::FirecrawlService
.new

4
spec/enterprise/jobs/captain/documents/crawl_job_spec.rb
View File

@@ -19,7 +19,7 @@ RSpec.describe Captain::Documents::CrawlJob, type: :job do
context 'with account usage limits' do
before do
allow(account).to receive(:usage_limits).and_return({ captain: { documents: { available: 20 } } })
allow(account).to receive(:usage_limits).and_return({ captain: { documents: { current_available: 20 } } })
end
it 'uses FirecrawlService with the correct crawl limit' do
@@ -35,7 +35,7 @@ RSpec.describe Captain::Documents::CrawlJob, type: :job do
context 'when crawl limit exceeds maximum' do
before do
allow(account).to receive(:usage_limits).and_return({ captain: { documents: { available: 1000 } } })
allow(account).to receive(:usage_limits).and_return({ captain: { documents: { current_available: 1000 } } })
end
it 'caps the crawl limit at 500' do