diff --git a/.github/workflows/test-litellm.yml b/.github/workflows/test-litellm.yml index b6b77ad326..b7f4a25d59 100644 --- a/.github/workflows/test-litellm.yml +++ b/.github/workflows/test-litellm.yml @@ -40,4 +40,4 @@ jobs: cd .. - name: Run tests run: | - poetry run pytest tests/test_litellm -vv -n 4 + poetry run pytest tests/test_litellm --tb=short -vv --maxfail=10 -n 4 diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/auth/test_user_api_key_auth_mcp.py b/tests/test_litellm/proxy/_experimental/mcp_server/auth/test_user_api_key_auth_mcp.py index fb15dbc14e..afafa510a4 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/auth/test_user_api_key_auth_mcp.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/auth/test_user_api_key_auth_mcp.py @@ -364,6 +364,7 @@ class TestMCPRequestHandler: mcp_auth_header, mcp_servers, mcp_server_auth_headers, + oauth2_headers, ) = await MCPRequestHandler.process_mcp_request(scope) # Assert the results @@ -717,6 +718,7 @@ class TestMCPCustomHeaderName: mcp_auth_header, mcp_servers, mcp_server_auth_headers, + oauth2_headers, ) = await MCPRequestHandler.process_mcp_request(scope) # Assert the results @@ -887,6 +889,7 @@ class TestMCPAccessGroupsE2E: mcp_auth_header, mcp_servers, mcp_server_auth_headers, + oauth2_headers, ) = await MCPRequestHandler.process_mcp_request(scope) # Assert the results @@ -936,6 +939,7 @@ class TestMCPAccessGroupsE2E: mcp_auth_header, mcp_servers, mcp_server_auth_headers, + oauth2_headers, ) = await MCPRequestHandler.process_mcp_request(scope) # Assert the results @@ -964,6 +968,7 @@ def test_mcp_path_based_server_segregation(monkeypatch): mcp_auth_header, mcp_servers, mcp_server_auth_headers, + oauth2_headers, ) = get_auth_context() # Capture the MCP servers for testing