Fix Token Count functionality for Cost Calculation

'tokens_used = response_raw.usage["total_tokens"]' currently fails with the following error:

```
print("Tokens used for this prompt: ", response_raw.usage["total_tokens"])
AttributeError: 'generator' object has no attribute 'usage'
```