phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(proxy_server.py): restrict model access for /v1/completions endpoint

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-16 11:49:59 -08:00
parent 691542ddf0
commit d97369dcbe
2 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/proxy_server.py
View file

@ -2259,8 +2259,13 @@ async def completion(
response = await llm_router.atext_completion(
**data, specific_deployment=True
)
else: # router is not set
elif user_model is not None: # `litellm --model <your-model-name>`
response = await litellm.atext_completion(**data)
else:
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail={"error": "Invalid model name passed in"},
)
if hasattr(response, "_hidden_params"):
model_id = response._hidden_params.get("model_id", None) or ""