Skip to content

Commit f02d02a

Browse files
committed
fix: follow redirects for OAuth endpoint requests
1 parent a692308 commit f02d02a

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

e2e_tests/python/server_clients/automated_oauth.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ async def perform_client_credentials_flow(self) -> None:
128128
async with httpx.AsyncClient() as client:
129129
base_url = self.authorization_server_url.rstrip("/")
130130
metadata_url = f"{base_url}/.well-known/oauth-authorization-server"
131-
response = await client.get(metadata_url)
131+
response = await client.get(metadata_url, follow_redirects=True)
132132

133133
if response.status_code != 200:
134134
raise RuntimeError(
@@ -330,7 +330,9 @@ async def _discover_scope_and_auth_server(self, server_url: str) -> tuple[str, s
330330

331331
async with httpx.AsyncClient() as client:
332332
headers = {MCP_PROTOCOL_VERSION: LATEST_PROTOCOL_VERSION}
333-
response = await client.get(server_url, headers=headers)
333+
response = await client.get(
334+
server_url, headers=headers, follow_redirects=True
335+
)
334336

335337
if response.status_code != 401:
336338
raise RuntimeError(
@@ -364,7 +366,9 @@ async def _discover_scope_and_auth_server(self, server_url: str) -> tuple[str, s
364366

365367
# Fetch protected resource metadata
366368
logging.debug("Fetching OAuth protected resource metadata...")
367-
metadata_response = await client.get(resource_metadata_url, headers=headers)
369+
metadata_response = await client.get(
370+
resource_metadata_url, headers=headers, follow_redirects=True
371+
)
368372

369373
if metadata_response.status_code != 200:
370374
raise RuntimeError(

examples/chatbots/python/server_clients/interactive_oauth.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -396,7 +396,9 @@ async def discover_scope(
396396

397397
async with httpx.AsyncClient() as client:
398398
headers = {MCP_PROTOCOL_VERSION: LATEST_PROTOCOL_VERSION}
399-
response = await client.get(server_url, headers=headers)
399+
response = await client.get(
400+
server_url, headers=headers, follow_redirects=True
401+
)
400402

401403
discovery_request = await oauthProvider._discover_protected_resource(
402404
response

0 commit comments

Comments
 (0)