添加获取token数和计费

controllers/calculate_cost.py

@@ -0,0 +1,20 @@
+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