Skip to content

Commit 42a538d

Browse files
committed
fix: Use new discovery URLs method
1 parent c4745c2 commit 42a538d

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

e2e_tests/python/automated_oauth.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,11 @@
1212

1313
import httpx
1414

15-
from mcp.client.auth import OAuthClientProvider, TokenStorage
15+
from mcp.client.auth import (
16+
OAuthClientProvider,
17+
TokenStorage,
18+
build_oauth_authorization_server_metadata_discovery_urls,
19+
)
1620
from mcp.client.streamable_http import streamablehttp_client, MCP_PROTOCOL_VERSION
1721
from mcp.shared.auth import (
1822
OAuthClientInformationFull,
@@ -190,8 +194,10 @@ async def perform_client_credentials_flow(self) -> None:
190194

191195
async def _discover_oauth_metadata(self) -> None:
192196
"""Discover OAuth metadata using upstream MCP SDK discovery logic."""
193-
# Use the inherited _get_discovery_urls method from parent class
194-
discovery_urls = self._get_discovery_urls()
197+
198+
discovery_urls = build_oauth_authorization_server_metadata_discovery_urls(
199+
self.context.auth_server_url, self.context.server_url
200+
)
195201

196202
async with httpx.AsyncClient() as client:
197203
for metadata_url in discovery_urls:

0 commit comments

Comments
 (0)