diff --git a/litellm/utils.py b/litellm/utils.py index 1932b8af6..09f1bb2e2 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -11102,7 +11102,7 @@ class CustomStreamWrapper: class TextCompletionStreamWrapper: - def __init__(self, completion_stream, model, stream_options): + def __init__(self, completion_stream, model, stream_options: Optional[dict] = None): self.completion_stream = completion_stream self.model = model self.stream_options = stream_options