From 3bcfbcc87147143dd27efeb18b848eabeb4c7545 Mon Sep 17 00:00:00 2001 From: ccurme Date: Wed, 24 Apr 2024 12:50:09 -0400 Subject: [PATCH] groq: handle null queue_time (#20839) --- libs/partners/groq/langchain_groq/chat_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/partners/groq/langchain_groq/chat_models.py b/libs/partners/groq/langchain_groq/chat_models.py index 61fecd7b461..2650f7608c2 100644 --- a/libs/partners/groq/langchain_groq/chat_models.py +++ b/libs/partners/groq/langchain_groq/chat_models.py @@ -469,7 +469,7 @@ class ChatGroq(BaseChatModel): token_usage = output["token_usage"] if token_usage is not None: for k, v in token_usage.items(): - if k in overall_token_usage: + if k in overall_token_usage and v is not None: overall_token_usage[k] += v else: overall_token_usage[k] = v