scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

fix(tool-registry): pass naive UTC datetime to fix PostgreSQL timestamp type error

Browse source 
`datetime.now(timezone.utc).isoformat()` produces a timezone-aware string
("2026-03-12T13:34:30+00:00") that PostgreSQL rejects when binding to a
`timestamp without time zone` column via execute_raw, logging:

  ERROR: column "created_at" is of type timestamp without time zone
  but expression is of type text

Replacing with `.replace(tzinfo=None)` produces a timezone-naive UTC datetime
that Prisma passes natively without needing ::timestamp casts. The stored
value is correct because the source is always UTC.

Fixes the ~185 errors/hour reported against v1.82.0-stable with PostgreSQL.
Applies to both batch_upsert_tools and update_tool_policy call sites.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
joereyna 2026-03-12 12:42:31 -07:00
parent 97947c2542
commit a73b1cad1e
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/db/tool_registry_writer.py
View file

@ -62,7 +62,7 @@ async def batch_upsert_tools(
key_hash = item.get("key_hash")
team_id = item.get("team_id")
key_alias = item.get("key_alias")
now = datetime.now(timezone.utc).isoformat()
now = datetime.now(timezone.utc).replace(tzinfo=None)
await prisma_client.db.execute_raw(
'INSERT INTO "LiteLLM_ToolTable" '
"(tool_id, tool_name, origin, call_policy, call_count, created_by, updated_by, key_hash, team_id, key_alias, created_at, updated_at) "
@ -140,7 +140,7 @@ async def update_tool_policy(
"""Update the call_policy for a tool. Upserts the row if it does not exist yet."""
try:
_updated_by = updated_by or "system"
now = datetime.now(timezone.utc).isoformat()
now = datetime.now(timezone.utc).replace(tzinfo=None)
await prisma_client.db.execute_raw(
'INSERT INTO "LiteLLM_ToolTable" (tool_id, tool_name, call_policy, created_by, updated_by, created_at, updated_at) '
"VALUES ($4, $1, $2, $3, $3, $5, $5) "