forked from phoenix/litellm-mirror
34 lines
1 KiB
Python
34 lines
1 KiB
Python
from prisma import Prisma
|
|
from litellm._logging import verbose_logger
|
|
|
|
|
|
async def apply_db_fixes(db: Prisma):
|
|
"""
|
|
Do Not Run this in production, only use it as a one-time fix
|
|
"""
|
|
verbose_logger.warning(
|
|
"DO NOT run this in Production....Running update_unassigned_teams"
|
|
)
|
|
try:
|
|
sql_query = """
|
|
UPDATE "LiteLLM_SpendLogs"
|
|
SET team_id = (
|
|
SELECT vt.team_id
|
|
FROM "LiteLLM_VerificationToken" vt
|
|
WHERE vt.token = "LiteLLM_SpendLogs".api_key
|
|
)
|
|
WHERE team_id IS NULL
|
|
AND EXISTS (
|
|
SELECT 1
|
|
FROM "LiteLLM_VerificationToken" vt
|
|
WHERE vt.token = "LiteLLM_SpendLogs".api_key
|
|
);
|
|
"""
|
|
response = await db.query_raw(sql_query)
|
|
print(
|
|
"Updated unassigned teams, Response=%s",
|
|
response,
|
|
)
|
|
except Exception as e:
|
|
raise Exception(f"Error apply_db_fixes: {str(e)}")
|
|
return
|