phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(feat)team-specific models

Browse source
This commit is contained in:
ishaan-jaff 2024-02-10 21:13:39 -08:00
parent afd0d7cb41
commit 5ac5f6c53c
2 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View file

@ -92,6 +92,7 @@ from litellm.proxy.utils import (
hash_token,
html_form,
_read_request_body,
_is_valid_team_configs,
)
from litellm.proxy.secret_managers.google_kms import load_google_kms
import pydantic
@ -2294,6 +2295,9 @@ async def chat_completion(
pass
else:
team_id = team_config.pop("team_id", None)
_is_valid_team_configs(
team_id=team_id, team_config=team_config, request_data=data
)
data["metadata"]["team_id"] = team_id
data = {
**team_config,