scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

fix(mcp): render registration refusals without recursion

Browse source
This commit is contained in:
Joshua Valluru 2026-09-10 21:54:06 -07:00
parent f04fb748c5
commit bb9b4c4aef
1 changed files with 12 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
litellm/proxy/_experimental/mcp_server/faults/render_oauth.py
View file

@ -43,6 +43,16 @@ def _registration_refused_description(status_code: int) -> str:
) )
def _render_caller_rejected(fault: CallerRejected) -> JSONResponse:
content: Final = {
"error": fault.code,
**({"error_description": fault.description} if fault.description else {}),
**({"error_uri": fault.error_uri} if fault.error_uri else {}),
}
status_code: Final = 401 if fault.code == "invalid_client" else 400
return JSONResponse(status_code=status_code, content=content, headers=TOKEN_NO_CACHE_HEADERS)
def render_token_fault(fault: UpstreamOAuthFault) -> JSONResponse: def render_token_fault(fault: UpstreamOAuthFault) -> JSONResponse:
"""RFC 6749 §5.2 response for a token-endpoint fault. Caller-actionable rejections relay the """RFC 6749 §5.2 response for a token-endpoint fault. Caller-actionable rejections relay the
upstream's code on the status that code implies (401 for invalid_client per §5.2, else 400); upstream's code on the status that code implies (401 for invalid_client per §5.2, else 400);
@ -50,13 +60,7 @@ def render_token_fault(fault: UpstreamOAuthFault) -> JSONResponse:
blamed for, or shown the internals of, a failure only the operator can fix.""" blamed for, or shown the internals of, a failure only the operator can fix."""
match fault.tag: match fault.tag:
case "caller_rejected": case "caller_rejected":
content: Final = { return _render_caller_rejected(fault)
"error": fault.code,
**({"error_description": fault.description} if fault.description else {}),
**({"error_uri": fault.error_uri} if fault.error_uri else {}),
}
status_code = 401 if fault.code == "invalid_client" else 400
return JSONResponse(status_code=status_code, content=content, headers=TOKEN_NO_CACHE_HEADERS)
case "gateway_rejected": case "gateway_rejected":
return JSONResponse( return JSONResponse(
status_code=502, status_code=502,
@ -74,7 +78,7 @@ def render_token_fault(fault: UpstreamOAuthFault) -> JSONResponse:
headers=TOKEN_NO_CACHE_HEADERS, headers=TOKEN_NO_CACHE_HEADERS,
) )
case "upstream_registration_refused": case "upstream_registration_refused":
return render_token_fault( return _render_caller_rejected(
CallerRejected( CallerRejected(
code="unauthorized_client", code="unauthorized_client",
description=_registration_refused_description(fault.status_code), description=_registration_refused_description(fault.status_code),