From 3ecc077507e152f6f7d24b1c04c8165d0253f0df Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Fri, 7 Jun 2024 19:09:17 -0700
Subject: [PATCH] fix mock_completion

---
 litellm/main.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/litellm/main.py b/litellm/main.py
index b62cb78cfd3..dd1fdb9f9f1 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -475,11 +475,13 @@ def mock_completion(
         except Exception:
             # dont let setting a hidden param block a mock_respose
             pass
-        logging.post_call(
-            input=messages,
-            api_key="my-secret-key",
-            original_response="my-original-response",
-        )
+
+        if logging is not None:
+            logging.post_call(
+                input=messages,
+                api_key="my-secret-key",
+                original_response="my-original-response",
+            )
         return model_response
 
     except Exception as e: