Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

handle missing finish_reason (#7296)

Browse Source 
In some cases, the OpenAI response is missing the `finish_reason`
attribute. It seems to happen when using Ada or Babbage and
`stream=true`, but I can't always reproduce it. This change just
gracefully handles the missing key.
This commit is contained in:
Bruno Bornsztein 2023-07-06 14:13:51 -05:00 committed by GitHub
parent 6ff9e9b34a
commit 1a4ca3eff9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
langchain/llms/openai.py
View File

@ -48,9 +48,9 @@ def update_token_usage(
def _update_response(response: Dict[str, Any], stream_response: Dict[str, Any]) -> None: def _update_response(response: Dict[str, Any], stream_response: Dict[str, Any]) -> None:
"""Update response from the stream response.""" """Update response from the stream response."""
response["choices"][0]["text"] += stream_response["choices"][0]["text"] response["choices"][0]["text"] += stream_response["choices"][0]["text"]
response["choices"][0]["finish_reason"] = stream_response["choices"][0][ response["choices"][0]["finish_reason"] = stream_response["choices"][0].get(
"finish_reason" "finish_reason", None
] )
response["choices"][0]["logprobs"] = stream_response["choices"][0]["logprobs"] response["choices"][0]["logprobs"] = stream_response["choices"][0]["logprobs"]