Module: Tracebook::Pricing::Calculator

Extended by:
Calculator
Included in:
Calculator
Defined in:
lib/tracebook/pricing/calculator.rb

Instance Method Summary collapse

Instance Method Details

#call(provider:, model:, input_tokens:, output_tokens:, occurred_at: Time.current) ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/tracebook/pricing/calculator.rb', line 10

def call(provider:, model:, input_tokens:, output_tokens:, occurred_at: Time.current)
  rule = matching_rule(provider, model, occurred_at)
  return CostBreakdown.new(input_cents: 0, output_cents: 0, total_cents: 0, currency: Tracebook.config.default_currency) unless rule

  input_cents = cost_for(rule.input_cents_per_unit, input_tokens)
  output_cents = cost_for(rule.output_cents_per_unit, output_tokens)
  CostBreakdown.new(
    input_cents: input_cents,
    output_cents: output_cents,
    total_cents: input_cents + output_cents,
    currency: rule.currency
  )
end

#cost_for(cents_per_million, tokens) ⇒ Object

Calculates cost in cents. cents_per_unit is cents per 1M tokens (matching the pricing rule storage). Example: $2.50/1M = 250 cents/1M. 1000 tokens => 1000 * 250 / 1_000_000 = 0.25 cents.



37
38
39
40
41
 
# File 'lib/tracebook/pricing/calculator.rb', line 37

def cost_for(cents_per_million, tokens)
  return 0 if cents_per_million.to_d <= 0 || tokens.to_i <= 0

  (tokens.to_i * cents_per_million.to_d / 1_000_000).round(4)
end

#glob_specificity(glob) ⇒ Object 



30
31
32
 
# File 'lib/tracebook/pricing/calculator.rb', line 30

def glob_specificity(glob)
  glob.delete("*?").length
end

#matching_rule(provider, model, occurred_at) ⇒ Object 



24
25
26
27
28
 
# File 'lib/tracebook/pricing/calculator.rb', line 24

def matching_rule(provider, model, occurred_at)
  Tracebook::PricingRule.where(provider: provider).select do |rule|
    rule.matches_model?(model) && rule.active_on?(occurred_at.to_date)
  end.max_by { |rule| [ glob_specificity(rule.model_glob), rule.effective_from ] }
end