From 3ef391800a8ce415e6fe07cafb46e9a7c1a9216d Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 13 Feb 2024 21:41:09 -0800
Subject: [PATCH] fix(sagemaker.py): fix token iterator default flag

---
 litellm/llms/sagemaker.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/litellm/llms/sagemaker.py b/litellm/llms/sagemaker.py
index b8ebc05bbc9..aa29b509db8 100644
--- a/litellm/llms/sagemaker.py
+++ b/litellm/llms/sagemaker.py
@@ -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: