20 lines
645 B
Python
20 lines
645 B
Python
from decimal import Decimal
|
|
|
|
def calc_cost(tokens: tuple) -> Decimal :
|
|
"""计算费用"""
|
|
input_tokens, output_tokens = tokens
|
|
if input_tokens <= 32000:
|
|
input_price = Decimal("0.001")
|
|
output_price = Decimal("0.01")
|
|
elif input_tokens <= 128000:
|
|
input_price = Decimal("0.0015")
|
|
output_price = Decimal("0.015")
|
|
else:
|
|
input_price = Decimal("0.003")
|
|
output_price = Decimal("0.03")
|
|
|
|
# 计算费用
|
|
input_cost = (Decimal(input_tokens) / Decimal(1000)) * input_price
|
|
output_cost = (Decimal(output_tokens) / Decimal(1000)) * output_price
|
|
|
|
return input_cost + output_cost |