fix: 🐛 Skip updating estimates if event isn't a prompt nor message

This commit is contained in:
André Ferreira
2024-06-18 16:22:42 +02:00
parent e1d52dbaa3
commit 1c5dae9b6c

View File

@@ -39,6 +39,8 @@ class TokenCostHandler(BaseCallbackHandler):
estimates = calculate_all_costs_and_tokens(
messages_str, response, self.model
)
else:
return
self.prompt_cost += estimates["prompt_cost"]
self.completion_cost += estimates["completion_cost"]