scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

fix(lens): bind nullable due timestamps as text for Prisma

Browse source
This commit is contained in:
moe-berri 2026-10-07 12:45:44 -07:00
parent 22e94fb29a
commit 25a531aaaa
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/lens/repository.py
View file

@ -239,7 +239,7 @@ class LensRepository:
async def create(self, lens: Lens) -> Lens:
await self.db.execute_raw(
"""INSERT INTO "LiteLLM_Lens" (id, version, data, due_at)
VALUES ($1,0,$2::jsonb,($3::timestamptz AT TIME ZONE 'UTC'))""",
VALUES ($1,0,$2::jsonb,($3::text::timestamptz AT TIME ZONE 'UTC'))""",
lens.id,
lens.model_dump_json(),
scheduled_at.isoformat() if (scheduled_at := due_at(lens)) else None,
@ -249,9 +249,9 @@ class LensRepository:
async def sync_due(self, lens: Lens) -> None:
await self.db.execute_raw(
"""UPDATE "LiteLLM_Lens"
SET due_at=($3::timestamptz AT TIME ZONE 'UTC')
SET due_at=($3::text::timestamptz AT TIME ZONE 'UTC')
WHERE id=$1 AND version=$2
AND due_at IS DISTINCT FROM ($3::timestamptz AT TIME ZONE 'UTC')""",
AND due_at IS DISTINCT FROM ($3::text::timestamptz AT TIME ZONE 'UTC')""",
lens.id,
lens.version,
scheduled_at.isoformat() if (scheduled_at := due_at(lens)) else None,
@ -288,7 +288,7 @@ class LensRepository:
SELECT data FROM "LiteLLM_Lens" WHERE id=$2 AND version=$3 FOR UPDATE
), updated AS (
UPDATE "LiteLLM_Lens" SET data=$1::jsonb, version=version+1,
due_at=($4::timestamptz AT TIME ZONE 'UTC')
due_at=($4::text::timestamptz AT TIME ZONE 'UTC')
WHERE id=$2 AND version=$3 AND EXISTS (SELECT 1 FROM previous) RETURNING id
)
, archived AS (INSERT INTO "LiteLLM_LensRun" (id, lens_id, created_at, data)