diff --git a/litellm/__pycache__/main.cpython-311.pyc b/litellm/__pycache__/main.cpython-311.pyc index 7bed614a6ce..593e38f0d53 100644 Binary files a/litellm/__pycache__/main.cpython-311.pyc and b/litellm/__pycache__/main.cpython-311.pyc differ diff --git a/litellm/main.py b/litellm/main.py index d099c1c177c..60c4d802dd0 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -571,7 +571,7 @@ def completion( if "stream" in optional_params and optional_params["stream"] == True: # don't try to access stream object, - response = CustomStreamWrapper(model_response, model, logging_obj=logging) + response = CustomStreamWrapper(model_response, model, custom_llm_provider="cohere", logging_obj=logging) return response response = model_response elif ( diff --git a/pyproject.toml b/pyproject.toml index 812377984bc..843bdea0182 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "litellm" -version = "0.1.684" +version = "0.1.685" description = "Library to easily interface with LLM API providers" authors = ["BerriAI"] license = "MIT License"