scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

fix: handle non-string team_alias/key_alias in PolicyMatchContext

Browse source 
Prevent Pydantic validation errors when team_alias or key_alias are not
proper strings (e.g. MagicMock in tests). Only pass values that are
actually strings; default to None otherwise.

Co-authored-by: Ishaan Jaff <ishaan-jaff@users.noreply.github.com>
This commit is contained in:
Cursor Agent 2026-03-07 02:47:58 +00:00
parent dfe2f6f991
commit 6b13ef266c
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/litellm_pre_call_utils.py
View file

@ -1848,9 +1848,11 @@ async def add_guardrails_from_policy_engine(
# Extract tags and build context
all_tags = get_tags_from_request_body(data) or None
_team_alias = user_api_key_dict.team_alias
_key_alias = user_api_key_dict.key_alias
context = PolicyMatchContext(
team_alias=user_api_key_dict.team_alias,
key_alias=user_api_key_dict.key_alias,
team_alias=_team_alias if isinstance(_team_alias, str) else None,
key_alias=_key_alias if isinstance(_key_alias, str) else None,
model=data.get("model"),
tags=all_tags,
)