fix for openrouter which doesn't return completion prices

This commit is contained in:
Joshua Vial 2023-12-19 23:31:37 +13:00
parent 179b648864
commit 48c680ffb6

View file

@ -693,7 +693,7 @@ class Coder:
raise Exception("No data found in openai response!") raise Exception("No data found in openai response!")
tokens = None tokens = None
if hasattr(completion, "usage"): if hasattr(completion, "usage") and completion.usage is not None:
prompt_tokens = completion.usage.prompt_tokens prompt_tokens = completion.usage.prompt_tokens
completion_tokens = completion.usage.completion_tokens completion_tokens = completion.usage.completion_tokens