diff --git a/libs/partners/openai/langchain_openai/chat_models/base.py b/libs/partners/openai/langchain_openai/chat_models/base.py index c37625134d7..cb1b68903b1 100644 --- a/libs/partners/openai/langchain_openai/chat_models/base.py +++ b/libs/partners/openai/langchain_openai/chat_models/base.py @@ -2924,9 +2924,9 @@ class OpenAIRefusalError(Exception): def _create_usage_metadata(oai_token_usage: dict) -> UsageMetadata: - input_tokens = oai_token_usage.get("prompt_tokens", 0) - output_tokens = oai_token_usage.get("completion_tokens", 0) - total_tokens = oai_token_usage.get("total_tokens", input_tokens + output_tokens) + input_tokens = oai_token_usage.get("prompt_tokens") or 0 + output_tokens = oai_token_usage.get("completion_tokens") or 0 + total_tokens = oai_token_usage.get("total_tokens") or input_tokens + output_tokens input_token_details: dict = { "audio": (oai_token_usage.get("prompt_tokens_details") or {}).get( "audio_tokens"