diff --git a/litellm/main.py b/litellm/main.py
index bf35d2a27d..8848f40f92 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -284,7 +284,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: