test: fix test

This commit is contained in:
Krrish Dholakia 2025-04-15 21:47:29 -07:00
parent 30cce8c26b
commit b9e2af04e5

View file

@ -23,6 +23,10 @@ async def test_construct_request_headers_project_id_from_env(monkeypatch):
# Set up test environment variable # Set up test environment variable
test_project_id = "test-project-123" test_project_id = "test-project-123"
monkeypatch.setenv("GCS_PUBSUB_PROJECT_ID", test_project_id) monkeypatch.setenv("GCS_PUBSUB_PROJECT_ID", test_project_id)
monkeypatch.setattr(
"litellm.proxy.proxy_server.premium_user",
True,
)
try: try:
# Create handler with no project_id # Create handler with no project_id