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

fix(lint): widen user_fields helpers to accept LiteLLM_MCPServerTable

Browse source 
The helpers use getattr(server, 'user_fields', None) and work with any
server-shaped object. Widen the type so the management endpoint callers
that pass LiteLLM_MCPServerTable type-check cleanly.
This commit is contained in:
mateo-berri 2026-05-19 06:43:22 +00:00
parent 8255b00062
commit adbc4c277f
No known key found for this signature in database
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
litellm/proxy/_experimental/mcp_server/user_fields.py
View file

@ -13,13 +13,20 @@ resolution / injection logic.
from __future__ import annotations
from typing import Any, Dict, List, Optional
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
from litellm._logging import verbose_logger
from litellm.types.mcp_server.mcp_server_manager import MCPServer
if TYPE_CHECKING:
from litellm.proxy._types import LiteLLM_MCPServerTable
def coerce_user_fields(server: MCPServer) -> List[Dict[str, Any]]:
UserFieldServer = Union[MCPServer, LiteLLM_MCPServerTable]
else:
UserFieldServer = MCPServer
def coerce_user_fields(server: UserFieldServer) -> List[Dict[str, Any]]:
"""Return the server's declared user fields as a list of plain dicts.
The column is stored as JSONB but Prisma sometimes hands it back as a
@ -44,13 +51,13 @@ def coerce_user_fields(server: MCPServer) -> List[Dict[str, Any]]:
return []
def server_has_user_fields(server: MCPServer) -> bool:
def server_has_user_fields(server: UserFieldServer) -> bool:
"""True iff the server declares any user fields at all."""
return bool(coerce_user_fields(server))
def compute_missing_user_fields(
server: MCPServer, stored_values: Optional[Dict[str, str]]
server: UserFieldServer, stored_values: Optional[Dict[str, str]]
) -> List[Dict[str, Any]]:
"""Return the declared field definitions the user has yet to fill in.