From 0a401ee468de6a13c905c8b41c66a73b95d5b1c5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 25 Mar 2025 19:27:15 -0700 Subject: [PATCH] test_litellm_proxy_server_config_no_general_settings --- .../test_using_litellm_db_outage.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/proxy_reliability_tests/test_using_litellm_db_outage.py b/tests/proxy_reliability_tests/test_using_litellm_db_outage.py index 52d3ec62220..ea9c7e3f3ab 100644 --- a/tests/proxy_reliability_tests/test_using_litellm_db_outage.py +++ b/tests/proxy_reliability_tests/test_using_litellm_db_outage.py @@ -19,18 +19,20 @@ import time import pytest import requests +TEST_MASTER_KEY = "sk-1234" + def test_litellm_proxy_server_config_no_general_settings(): # Install the litellm[proxy] package # Start the server try: subprocess.run(["pip", "install", "litellm[proxy]"]) - subprocess.run(["pip", "install", "litellm[extra_proxy]"]) filepath = os.path.dirname(os.path.abspath(__file__)) config_fp = f"{filepath}/test_configs/test_config.yaml" # Set DATABASE_URL environment variable os.environ["DATABASE_URL"] = os.getenv("TOXI_PROXY_DATABASE_URL") + os.environ["LITELLM_MASTER_KEY"] = TEST_MASTER_KEY server_process = subprocess.Popen( [ @@ -72,6 +74,3 @@ def test_litellm_proxy_server_config_no_general_settings(): # Shut down the server server_process.terminate() server_process.wait() - - # Additional assertions can be added here - assert True