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

Merge pull request #1933 from BerriAI/litellm_team_specific_models

Browse source 
[FEAT] Proxy - set team specific models
This commit is contained in:
Ishaan Jaff 2024-02-10 21:40:13 -08:00 • committed by GitHub
commit 1cb8d2d3a3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 74 additions and 16 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,