From 6803cb4f3426e3a6642e6880b95514853e89aa62 Mon Sep 17 00:00:00 2001 From: yahya-mouman <103438582+yahya-mouman@users.noreply.github.com> Date: Mon, 28 Oct 2024 20:49:43 +0100 Subject: [PATCH] openai[patch]: add check for none values when summing token usage (#27585) **Description:** Fixes None addition issues when an empty value is passed on If no one reviews your PR within a few days, please @-mention one of baskaryan, efriis, eyurtsev, ccurme, vbarda, hwchase17. --- libs/partners/openai/langchain_openai/chat_models/base.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/libs/partners/openai/langchain_openai/chat_models/base.py b/libs/partners/openai/langchain_openai/chat_models/base.py index 0d991cbe4dd..7ad85862960 100644 --- a/libs/partners/openai/langchain_openai/chat_models/base.py +++ b/libs/partners/openai/langchain_openai/chat_models/base.py @@ -605,6 +605,8 @@ class BaseChatOpenAI(BaseChatModel): token_usage = output["token_usage"] if token_usage is not None: for k, v in token_usage.items(): + if v is None: + continue if k in overall_token_usage: overall_token_usage[k] = _update_token_usage( overall_token_usage[k], v