(model, input_tokens, output_tokens)
| 89 | } |
| 90 | |
| 91 | def calculate_cost(model, input_tokens, output_tokens): |
| 92 | base_model = 'turbo' |
| 93 | if model.find('gpt-4') != -1: |
| 94 | base_model = 'gpt-4' |
| 95 | cost = costs[base_model]['input'] * input_tokens + costs[base_model]['output'] * output_tokens |
| 96 | return cost |
| 97 | |
| 98 | |
| 99 | @failsoft |