Stanko K.R.
36 lines
1.1 KiB
Ruby
36 lines
1.1 KiB
Ruby
require "test_helper"
|
|
|
|
class Conversation::Message::ResponseGenerator::ResponseTest < ActiveSupport::TestCase
|
|
test "price calculations" do
|
|
response = Conversation::Message::ResponseGenerator::Response.new(
|
|
answer: "Hi!",
|
|
input_tokens: 198,
|
|
output_tokens: 2,
|
|
model_id: "gpt-4"
|
|
)
|
|
|
|
# The price of an input token is 30 USD per million tokens
|
|
# and 60 USD per million output tokens
|
|
# That's 0.00003 cents per input token and 0.00006 cents
|
|
# per output token
|
|
# Which is 3 micro-cents per input token and 6 micro-cents
|
|
# per output token
|
|
assert_equal "3.0".to_d, response.input_token_price_microcents
|
|
assert_equal "6.0".to_d, response.output_token_price_microcents
|
|
|
|
# We've got 198 input tokens, so that's
|
|
# 193 * 3 = 594
|
|
assert_equal 594, response.input_cost_microcents
|
|
|
|
# We've got 2 output tokens, so that's
|
|
# 2 * 6 = 12
|
|
assert_equal 12, response.output_cost_microcents
|
|
|
|
# So the total is 594 + 12 micro-cents
|
|
assert_equal 606, response.cost_microcents
|
|
|
|
# If we convert that to a decimal value we get 0.00606 cents
|
|
assert_equal "0.00606".to_d, response.cost
|
|
end
|
|
end
|