diff --git a/litellm/main.py b/litellm/main.py
index 8748cb1f5e..a025cf72d1 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -285,7 +285,7 @@ def completion(
 
     # Optional liteLLM function params
     **kwargs,
-) -> ModelResponse:
+) -> Union[ModelResponse, CustomStreamWrapper]:
     """
     Perform a completion() using any of litellm supported llms (example gpt-4, gpt-3.5-turbo, claude-2, command-nightly)
     Parameters: