blob42/langchain
1
0
Fork 0
forked from Archives/langchain
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Empty check before pop (#4929)

Browse Source 
# Check whether 'other' is empty before popping

This PR could fix a potential 'popping empty set' error.

Co-authored-by: Junlin Zhou <jlzhou@zjuici.com>
This commit is contained in:
Junlin Zhou 2023-05-24 07:46:50 +08:00 committed by GitHub
parent de6e6c764e
commit 9242998db1
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/callbacks/streaming_aiter.py
View File

@ -58,7 +58,8 @@ class AsyncIteratorCallbackHandler(AsyncCallbackHandler):
) )
# Cancel the other task # Cancel the other task
other.pop().cancel() if other:
other.pop().cancel()
# Extract the value of the first completed task # Extract the value of the first completed task
token_or_done = cast(Union[str, Literal[True]], done.pop().result()) token_or_done = cast(Union[str, Literal[True]], done.pop().result())