diff --git a/litellm/tests/test_logging.py b/litellm/tests/test_logging.py index d0e6ca332..7187193ad 100644 --- a/litellm/tests/test_logging.py +++ b/litellm/tests/test_logging.py @@ -64,30 +64,30 @@ def test_logging_success_completion(): pytest.fail(f"Error occurred: {e}") pass -## test on non-openai completion call -def test_logging_success_completion_non_openai(): - global score - try: - # Redirect stdout - old_stdout = sys.stdout - sys.stdout = new_stdout = io.StringIO() +# ## test on non-openai completion call +# def test_logging_success_completion_non_openai(): +# global score +# try: +# # Redirect stdout +# old_stdout = sys.stdout +# sys.stdout = new_stdout = io.StringIO() - response = completion(model="claude-instant-1", messages=messages) +# response = completion(model="claude-instant-1", messages=messages) - # Restore stdout - sys.stdout = old_stdout - output = new_stdout.getvalue().strip() +# # Restore stdout +# sys.stdout = old_stdout +# output = new_stdout.getvalue().strip() - if "Logging Details Pre-API Call" not in output: - raise Exception("Required log message not found!") - elif "Logging Details Post-API Call" not in output: - raise Exception("Required log message not found!") - elif "Logging Details LiteLLM-Success Call" not in output: - raise Exception("Required log message not found!") - score += 1 - except Exception as e: - pytest.fail(f"Error occurred: {e}") - pass +# if "Logging Details Pre-API Call" not in output: +# raise Exception("Required log message not found!") +# elif "Logging Details Post-API Call" not in output: +# raise Exception("Required log message not found!") +# elif "Logging Details LiteLLM-Success Call" not in output: +# raise Exception("Required log message not found!") +# score += 1 +# except Exception as e: +# pytest.fail(f"Error occurred: {e}") +# pass # # streaming completion # ## test on openai completion call