tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-03 02:20:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Krrish Dholakia
2024-02-16 11:54:12 -08:00
committed by ishaan-jaff
parent 936e460a30
commit 01077b20b9
2 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/model_prices_and_context_window_backup.json
+4 -4
View File
@@ -965,15 +965,15 @@
},
"dolphin": {
"max_tokens": 4096,
"input_cost_per_token": 0.00002,
"output_cost_per_token": 0.00002,
"input_cost_per_token": 0.0000005,
"output_cost_per_token": 0.0000005,
"litellm_provider": "nlp_cloud",
"mode": "completion"
},
"chatdolphin": {
"max_tokens": 4096,
"input_cost_per_token": 0.00002,
"output_cost_per_token": 0.00002,
"input_cost_per_token": 0.0000005,
"output_cost_per_token": 0.0000005,
"litellm_provider": "nlp_cloud",
"mode": "chat"
},
litellm/proxy/proxy_server.py
+6 -1
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 ""