scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

address greptile review feedback

Browse source
This commit is contained in:
daanhendrio 2026-04-11 12:47:33 +00:00
parent 7c4217a6f5
commit 69c3527bf6
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
litellm/proxy/management_endpoints/team_endpoints.py
View file

@ -112,6 +112,15 @@ from litellm.types.proxy.management_endpoints.team_endpoints import (
router = APIRouter()
def _sanitize_for_log(value: Any) -> str:
"""Strip CR/LF from user-controlled values to prevent log injection."""
try:
text = str(value)
except Exception:
text = repr(value)
return text.replace("\r", "").replace("\n", "")
class TeamMemberBudgetHandler:
"""Helper class to handle team member budget, RPM, and TPM limit operations"""
@ -257,7 +266,7 @@ class TeamMemberBudgetHandler:
@staticmethod
async def backfill_team_member_budget_entries(
team_id: str,
members_with_roles: List[Member],
members_with_roles: List[Union[Member, dict]],
team_member_budget_id: str,
prisma_client: PrismaClient,
) -> None:
@ -305,8 +314,8 @@ class TeamMemberBudgetHandler:
verbose_proxy_logger.info(
"Backfilled %d team_memberships for team %s with budget %s",
len(missing),
team_id,
team_member_budget_id,
_sanitize_for_log(team_id),
_sanitize_for_log(team_member_budget_id),
)