From 644ea8822c7831e04c4125d648fbbb4e77b42956 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 8 Aug 2023 13:58:58 -0700
Subject: [PATCH] fix model response

---
 litellm/main.py | 1 +
 pyproject.toml  | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

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"