tokencost_prompt_completion_cost_calculation_for_gpt_model.py

1from tokencost import calculate_prompt_cost, calculate_completion_cost
2
3model = "gpt-3.5-turbo"
4prompt = "Hello, how are you?"
5completion = "I'm doing well, thank you!"
6
7# Calculate prompt cost
8prompt_cost = calculate_prompt_cost(prompt, model)
9print(f"Prompt cost: ${prompt_cost:.6f}")
10
11# Calculate completion cost
12completion_cost = calculate_completion_cost(completion, model)
13print(f"Completion cost: ${completion_cost:.6f}")
14
15# Total cost
16total_cost = prompt_cost + completion_cost
17print(f"Total cost: ${total_cost:.6f}")