Docker/Maybe
Docker/Maybe
1
0
Fork 0
mirror of https://github.com/maybe-finance/maybe.git synced 2025-08-08 06:55:21 +02:00
Code Issues Releases Activity Actions

Encapsulate enrichment actions, add tests

Browse source
This commit is contained in:
Zach Gollwitzer 2025-05-20 11:33:35 -04:00
parent dd605a577e
commit 94a807c3c9
9 changed files with 149 additions and 71 deletions
Download patch file Download diff file Expand all files Collapse all files

18
app/models/family/auto_categorizer.rb
View file

@ -27,23 +27,19 @@ class Family::AutoCategorizer
end
scope.each do |transaction|
transaction.lock_attr!(:category_id)
auto_categorization = result.data.find { |c| c.transaction_id == transaction.id }
category_id = user_categories_input.find { |c| c[:name] == auto_categorization&.category_name }&.dig(:id)
if category_id.present?
Family.transaction do
transaction.log_enrichment!(
attribute_name: "category_id",
attribute_value: category_id,
source: "ai",
)
transaction.update!(category_id: category_id)
end
transaction.enrich_attribute(
:category_id,
category_id,
source: "ai"
)
end
transaction.lock_attr!(:category_id)
end
end

18
app/models/family/auto_merchant_detector.rb
View file

@ -27,8 +27,6 @@ class Family::AutoMerchantDetector
end
scope.each do |transaction|
transaction.lock_attr!(:merchant_id)
auto_detection = result.data.find { |c| c.transaction_id == transaction.id }
merchant_id = user_merchants_input.find { |m| m[:name] == auto_detection&.business_name }&.dig(:id)
@ -46,16 +44,16 @@ class Family::AutoMerchantDetector
merchant_id = merchant_id || ai_provider_merchant&.id
if merchant_id.present?
Family.transaction do
transaction.log_enrichment!(
attribute_name: "merchant_id",
attribute_value: merchant_id,
source: "ai",
)
transaction.enrich_attribute(
:merchant_id,
merchant_id,
source: "ai"
)
transaction.update!(merchant_id: merchant_id)
end
end
# We lock the attribute so that this Rule doesn't try to run again
transaction.lock_attr!(:merchant_id)
end
end