scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

refactor(mcp): hoist repeated bridge relay check and align raise detail shapes

Browse source
This commit is contained in:
Tin Chi Lo 2026-07-10 00:52:42 -07:00
parent aa4f585e4c
commit c79dcef812
1 changed files with 7 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
litellm/proxy/_experimental/mcp_server/discoverable_endpoints.py
View file

@ -539,12 +539,10 @@ def _require_s256_pkce(
return code_challenge, code_challenge_method
raise HTTPException(
status_code=400,
detail={
"error": (
"This server requires PKCE: send code_challenge with "
"code_challenge_method=S256 on the authorization request"
)
},
detail=(
"This server requires PKCE: send code_challenge with "
"code_challenge_method=S256 on the authorization request"
),
)
@ -701,7 +699,8 @@ async def exchange_token_with_server(
status_code=400,
detail="code is required for authorization_code grant",
)
if _dcr_bridge_relays_client_registration(mcp_server) and not redirect_uri:
bridge_token_relay = _dcr_bridge_relays_client_registration(mcp_server)
if bridge_token_relay and not redirect_uri:
raise HTTPException(
status_code=400,
detail=(
@ -710,9 +709,7 @@ async def exchange_token_with_server(
),
)
proxy_base_url = get_request_base_url(request)
resolved_redirect_uri = (
redirect_uri if _dcr_bridge_relays_client_registration(mcp_server) else f"{proxy_base_url}/callback"
)
resolved_redirect_uri = redirect_uri if bridge_token_relay else f"{proxy_base_url}/callback"
token_data = {
"grant_type": "authorization_code",
"code": code,