tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-11 07:05:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix linting

Browse Source
This commit is contained in:
Ishaan Jaffer
2026-01-23 13:45:50 -08:00
parent 37b7dff194
commit 5b341ee842
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/proxy/policy_engine/policy_registry.py
+2 -5
View File
@@ -207,8 +207,6 @@ class PolicyRegistry:
PolicyDBResponse with the created policy
"""
try:
from prisma import Json
# Build data dict, only include condition if it's set
data: Dict[str, Any] = {
"policy_name": policy_request.policy_name,
@@ -227,7 +225,7 @@ class PolicyRegistry:
data["created_by"] = created_by
data["updated_by"] = created_by
if policy_request.condition is not None:
data["condition"] = Json(policy_request.condition.model_dump())
data["condition"] = policy_request.condition.model_dump()
created_policy = await prisma_client.db.litellm_policytable.create(
data=data
@@ -304,8 +302,7 @@ class PolicyRegistry:
if policy_request.guardrails_remove is not None:
update_data["guardrails_remove"] = policy_request.guardrails_remove
if policy_request.condition is not None:
from prisma import Json
update_data["condition"] = Json(policy_request.condition.model_dump())
update_data["condition"] = policy_request.condition.model_dump()
updated_policy = await prisma_client.db.litellm_policytable.update(
where={"policy_id": policy_id},