diff --git a/litellm/main.py b/litellm/main.py
index 8d8c78e256..0eac877247 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -239,7 +239,7 @@ def completion(
             max_tokens_to_sample=max_tokens_to_sample,
             **optional_params
         )
-      if optional_params['stream'] == True:
+      if 'stream' in optional_params and optional_params['stream'] == True:
         # don't try to access stream object,
         response = CustomStreamWrapper(completion)
         return response