phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix test_get_gcs_logging_config_without_service_account

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-06 20:49:44 -08:00
parent e3519aa5ae
commit c6da749997
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/local_testing/test_gcs_bucket.py
View file

@ -531,6 +531,9 @@ async def test_get_gcs_logging_config_without_service_account():
_old_gcs_bucket_name = os.environ.get("GCS_BUCKET_NAME") _old_gcs_bucket_name = os.environ.get("GCS_BUCKET_NAME")
os.environ.pop("GCS_BUCKET_NAME") os.environ.pop("GCS_BUCKET_NAME")
_old_gcs_service_acct = os.environ.get("GCS_PATH_SERVICE_ACCOUNT")
os.environ.pop("GCS_PATH_SERVICE_ACCOUNT")
# Mock the load_auth function to avoid credential loading issues # Mock the load_auth function to avoid credential loading issues
# Test 1: With standard_callback_dynamic_params (with service account) # Test 1: With standard_callback_dynamic_params (with service account)
gcs_logger = GCSBucketLogger() gcs_logger = GCSBucketLogger()
@ -566,3 +569,6 @@ async def test_get_gcs_logging_config_without_service_account():
if _old_gcs_bucket_name is not None: if _old_gcs_bucket_name is not None:
os.environ["GCS_BUCKET_NAME"] = _old_gcs_bucket_name os.environ["GCS_BUCKET_NAME"] = _old_gcs_bucket_name
if _old_gcs_service_acct is not None:
os.environ["GCS_PATH_SERVICE_ACCOUNT"] = _old_gcs_service_acct