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