github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-23 03:22:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bug with openai token count (#1806)

Browse Source
This commit is contained in:
Harrison Chase 2023-03-20 07:51:18 -07:00 committed by GitHub
parent b1c4480d7c
commit f6d24d5740
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
langchain/chat_models/openai.py
View File

@ -229,7 +229,8 @@ class ChatOpenAI(BaseChatModel, BaseModel):
overall_token_usage: dict = {} overall_token_usage: dict = {}
for output in llm_outputs: for output in llm_outputs:
if output is None: if output is None:
raise ValueError("Should always be something for OpenAI.") # Happens in streaming
continue
token_usage = output["token_usage"] token_usage = output["token_usage"]
for k, v in token_usage.items(): for k, v in token_usage.items():
if k in overall_token_usage: if k in overall_token_usage: