fix bug with openai token count (langchain-ai#1806)

apify · Mar 20, 2023 · f6d24d5 · f6d24d5
1 parent b1c4480
commit f6d24d5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/langchain/chat_models/openai.py b/langchain/chat_models/openai.py
@@ -229,7 +229,8 @@ def _combine_llm_outputs(self, llm_outputs: List[Optional[dict]]) -> dict:
         overall_token_usage: dict = {}
         for output in llm_outputs:
             if output is None:
-                raise ValueError("Should always be something for OpenAI.")
+                # Happens in streaming
+                continue
             token_usage = output["token_usage"]
             for k, v in token_usage.items():
                 if k in overall_token_usage: