mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 11:43:54 +00:00
fix(sagemaker.py): fix token iterator default flag
This commit is contained in:
parent
a3212a83cb
commit
0f1fdef7d5
1 changed files with 1 additions and 1 deletions
|
@ -30,7 +30,7 @@ import json
|
|||
|
||||
|
||||
class TokenIterator:
|
||||
def __init__(self, stream, acompletion: bool):
|
||||
def __init__(self, stream, acompletion: bool = False):
|
||||
if acompletion == False:
|
||||
self.byte_iterator = iter(stream)
|
||||
elif acompletion == True:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue