Class: TranslationAPI::Provider::Gemini::Cost

Inherits:
Object
  • Object
show all
Defined in:
lib/translation_api/provider/gemini/cost.rb

Constant Summary collapse

ONE_MILLION =
1_000_000

Instance Method Summary collapse

Constructor Details

#initialize(model) ⇒ Cost

Returns a new instance of Cost.



9
10
11
# File 'lib/translation_api/provider/gemini/cost.rb', line 9

def initialize(model)
  @model = model
end

Instance Method Details

#input_cost(used_tokens) ⇒ Object



13
14
15
# File 'lib/translation_api/provider/gemini/cost.rb', line 13

def input_cost(used_tokens)
  calculate_cost(used_tokens, :input)
end

#output_cost(used_tokens) ⇒ Object



17
18
19
# File 'lib/translation_api/provider/gemini/cost.rb', line 17

def output_cost(used_tokens)
  calculate_cost(used_tokens, :output)
end