mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-07 02:59:05 +00:00
fix: use rsplit to correctly parse OIDC user IDs with colons in spend counter key
This commit is contained in:
parent
fd4e8209d7
commit
36294ffcb2
1 changed files with 3 additions and 3 deletions
|
|
@ -1951,9 +1951,9 @@ async def _reseed_spend_from_db(counter_key: str) -> float:
|
|||
membership_part, model_name = suffix.rsplit(":model:", 1)
|
||||
if ":" not in membership_part:
|
||||
return 0.0
|
||||
member_user_id, member_team_id = membership_part.split(":", 1)
|
||||
member_user_id, member_team_id = membership_part.rsplit(":", 1)
|
||||
model_row = (
|
||||
await prisma_client.db.litellm_teammembermodeispend.find_unique(
|
||||
await prisma_client.db.litellm_teammembermodelspend.find_unique(
|
||||
where={
|
||||
"user_id_team_id_model": {
|
||||
"user_id": member_user_id,
|
||||
|
|
@ -1968,7 +1968,7 @@ async def _reseed_spend_from_db(counter_key: str) -> float:
|
|||
membership_part = suffix
|
||||
if ":" not in membership_part:
|
||||
return 0.0
|
||||
member_user_id, member_team_id = membership_part.split(":", 1)
|
||||
member_user_id, member_team_id = membership_part.rsplit(":", 1)
|
||||
row = await prisma_client.db.litellm_teammembership.find_unique(
|
||||
where={
|
||||
"user_id_team_id": {
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue