fix mock_completion

2024-06-07 19:09:17 -07:00 · 2024-06-07 19:09:17 -07:00 · 3ecc077507
commit 3ecc077507
parent 47538135cf
1 changed files with 7 additions and 5 deletions
--- 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: