计算成本
在大规模上线 AI 功能前,估算成本至关重要。成本取决于使用的输入与输出 token 数量以及所选模型。
您的任务是计算对客户聊天记录进行摘要的成本。
OpenAI 的 client、prompt 和 max_tokens 已为您预先加载。
本练习是课程的一部分
在 Python 中使用 DeepSeek
练习说明
- 从
response中提取输入 token 的用量。 - 完成成本计算,将输出 token 的成本加入。
交互式实操练习
通过完成这段示例代码来试试这个练习。
client = OpenAI(api_key="", base_url="https://api.together.xyz/v1")
response = client.chat.completions.create(
model="deepseek-ai/DeepSeek-V4-Pro",
messages=[{"role": "user", "content": prompt}],
max_tokens=max_tokens
)
input_token_price = 2.1 / 1_000_000
output_token_price = 4.4 / 1_000_000
# Extract token usage
input_tokens = response.____.____
output_tokens = max_tokens
# Calculate cost
cost = (input_tokens * input_token_price + ____ * ____)
print(f"Estimated cost: ${cost}")