From b9e2af04e50bfcc1125fe49f91d76f774df16d51 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 15 Apr 2025 21:47:29 -0700 Subject: [PATCH] test: fix test --- tests/litellm/integrations/gcs_pubsub/test_pub_sub.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/litellm/integrations/gcs_pubsub/test_pub_sub.py b/tests/litellm/integrations/gcs_pubsub/test_pub_sub.py index e4c9b5ef5e..7ff28bfe83 100644 --- a/tests/litellm/integrations/gcs_pubsub/test_pub_sub.py +++ b/tests/litellm/integrations/gcs_pubsub/test_pub_sub.py @@ -23,6 +23,10 @@ async def test_construct_request_headers_project_id_from_env(monkeypatch): # Set up test environment variable test_project_id = "test-project-123" monkeypatch.setenv("GCS_PUBSUB_PROJECT_ID", test_project_id) + monkeypatch.setattr( + "litellm.proxy.proxy_server.premium_user", + True, + ) try: # Create handler with no project_id