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

Merge pull request #23481 from joereyna/fix/stable-tool-registry-timestamp

Browse source 
fix(tool-registry): pass naive UTC datetime to fix PostgreSQL timestamp error on v1.82.0-stable
This commit is contained in:
Joe Reyna 2026-03-13 19:51:43 -07:00 • committed by GitHub
commit 8eb2daff75
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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) "