diff --git a/litellm/main.py b/litellm/main.py index 4fc3bcb05e..8f7873099c 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -300,6 +300,7 @@ def completion( "completion_tokens": completion_tokens, "total_tokens": prompt_tokens + completion_tokens } + response = model_response else: ## LOGGING logging(model=model, input=messages, azure=azure, logger_fn=logger_fn) diff --git a/pyproject.toml b/pyproject.toml index d260539972..20bc61f467 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "litellm" -version = "0.1.361" +version = "0.1.362" description = "Library to easily interface with LLM API providers" authors = ["BerriAI"] license = "MIT License"