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

Fix Codex CLI responses websocket warmup and routing

Browse source
This commit is contained in:
Mark Ferraz 2026-04-01 22:25:16 -05:00
parent af5c2581d0
commit a5af86d578
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/proxy/response_api_endpoints/endpoints.py
View file

@ -939,8 +939,8 @@ async def cancel_response(
@router.websocket("/responses")
async def responses_websocket_endpoint(
websocket: WebSocket,
model: str = fastapi.Query(
..., description="The model to use for the responses WebSocket session."
model: Optional[str] = fastapi.Query(
None, description="The model to use for the responses WebSocket session."
),
user_api_key_dict=Depends(user_api_key_auth_websocket),
):
@ -981,6 +981,8 @@ async def responses_websocket_endpoint(
"model": model,
"websocket": websocket,
}
if not model:
data["llm_router"] = llm_router
# Construct a synthetic Request for pre-call processing
headers_list = list(websocket.scope.get("headers") or [])
@ -994,7 +996,9 @@ async def responses_websocket_endpoint(
request._url = websocket.url
async def return_body():
return f'{{"model": "{model}"}}'.encode()
if model:
return f'{{"model": "{model}"}}'.encode()
return b"{}"
request.body = return_body # type: ignore