scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-30 01:52:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

fix(mcp): wrap json CAS filters in prisma Json sentinel

Browse source 
Co-Authored-By: bot_apk <apk@cognition.ai>
This commit is contained in:
Devin AI 2026-09-25 02:26:07 +00:00
parent 989ac9263d
commit 94362273fe
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
litellm/proxy/_experimental/mcp_server/tool_permission_backfill.py
View file

@ -16,6 +16,7 @@ from dataclasses import dataclass
from types import MappingProxyType
from typing import TYPE_CHECKING, Final, TypeAlias
from prisma.fields import Json
from pydantic import TypeAdapter
from typing_extensions import ReadOnly, TypedDict
@ -266,14 +267,14 @@ async def _convert_one_row(
if isinstance(conversion, Unavailable):
return conversion.server_ids
stored_fields: Final = (
("mcp_servers", raw_row.mcp_servers),
("mcp_access_groups", raw_row.mcp_access_groups),
("mcp_toolsets", raw_row.mcp_toolsets),
("mcp_tool_permissions", raw_row.mcp_tool_permissions),
("mcp_servers", raw_row.mcp_servers, False),
("mcp_access_groups", raw_row.mcp_access_groups, False),
("mcp_toolsets", raw_row.mcp_toolsets, False),
("mcp_tool_permissions", raw_row.mcp_tool_permissions, True),
)
equals_filters: Final = { # mutable-ok: prisma where kwarg requires a JSON-serializable dict
field: {"equals": value} # mutable-ok: prisma where filter shape
for field, value in stored_fields
field: {"equals": Json(value) if is_json else value} # mutable-ok: prisma where filter shape
for field, value, is_json in stored_fields
if value is not None
}
where: Final[dict[str, object]] = { # mutable-ok: prisma where kwarg requires a JSON-serializable dict