mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-09 03:18:44 +00:00
b604e2b20c
About 35,000 fixes ruff marks safe across 32 rules (UP006/UP045/UP007 modern annotations, UP032 f-strings, SIM114/SIM118, RET501, and friends), removal of the 1,296 typing imports the rewrite orphaned, and hand fixes for what the fixers could not see: five star-import freeloaders of typing names, two F823 late-import annotations, the /get/config/list introspection crash on types.UnionType, redundant function-local RoleMappings imports in ui_sso.py that shadowed the module-level name once the annotation lost its quotes, and one FURB168 tautology. B009/B010/PIE804/RUF019 are excluded on purpose: their safe fixes rewrite getattr/setattr/**-splat/key-in-dict escape hatches into forms basedpyright then rejects (283 new errors measured), so their budgets stay at base values. ruff-strict-budget.json drops by 39,579 this commit (39,968 across the branch) with 28 rules at an actual 0 and 9 more sharply down. type-discipline-budget.json ratchets LIT002/LIT006/LIT009 down; LIT001 moves to the now-honest total: the checker matches the spelling `set` but not the alias `Set`, so the 160 typing.Set annotations rewritten to set[...] were always mutable-set annotations and only now count.
116 lines
4.3 KiB
Python
116 lines
4.3 KiB
Python
"""
|
|
MCP server table model.
|
|
|
|
Canonical definition for ``litellm_mcpservertable``. Re-exported from
|
|
``litellm.proxy._types`` for backwards compatibility.
|
|
"""
|
|
|
|
import enum
|
|
from datetime import datetime
|
|
from typing import Literal
|
|
|
|
from pydantic import Field
|
|
|
|
from litellm.types.llms.base import LiteLLMPydanticObjectBase
|
|
from litellm.types.mcp import MCPAuthType, MCPCredentials, MCPTransportType
|
|
from litellm.types.mcp_server.mcp_server_manager import MCPInfo
|
|
|
|
|
|
class MCPEnvVarScope(str, enum.Enum):
|
|
"""Scope for an MCP server environment variable.
|
|
|
|
- ``global``: value is provided by the admin and used for all users.
|
|
- ``user``: each user must provide their own value via the per-user
|
|
env-var endpoint. The admin-supplied ``value`` is treated as a
|
|
placeholder/hint and is not used at request time.
|
|
"""
|
|
|
|
global_ = "global"
|
|
user = "user"
|
|
|
|
|
|
class MCPEnvVar(LiteLLMPydanticObjectBase):
|
|
"""One environment variable for an MCP server.
|
|
|
|
Variables can be interpolated into ``static_headers`` using ``${NAME}``
|
|
syntax. ``scope=global`` values are stored on the server. ``scope=user``
|
|
values are stored per-user in ``LiteLLM_MCPUserEnvVars`` and supplied by
|
|
each user.
|
|
"""
|
|
|
|
name: str
|
|
value: str = ""
|
|
scope: MCPEnvVarScope = MCPEnvVarScope.global_
|
|
description: str | None = None
|
|
|
|
|
|
class LiteLLM_MCPServerTable(LiteLLMPydanticObjectBase):
|
|
"""Represents a LiteLLM_MCPServerTable record"""
|
|
|
|
server_id: str
|
|
server_name: str | None = None
|
|
alias: str | None = None
|
|
description: str | None = None
|
|
url: str | None = None
|
|
spec_path: str | None = None
|
|
transport: MCPTransportType
|
|
auth_type: MCPAuthType | None = None
|
|
credentials: MCPCredentials | None = None
|
|
instructions: str | None = None
|
|
created_at: datetime | None = None
|
|
created_by: str | None = None
|
|
updated_at: datetime | None = None
|
|
updated_by: str | None = None
|
|
teams: list[dict[str, str | None]] = Field(default_factory=list)
|
|
mcp_access_groups: list[str] = Field(default_factory=list)
|
|
allowed_tools: list[str] = Field(default_factory=list)
|
|
tool_name_to_display_name: dict[str, str] | None = None
|
|
tool_name_to_description: dict[str, str] | None = None
|
|
extra_headers: list[str] = Field(default_factory=list)
|
|
mcp_info: MCPInfo | None = None
|
|
static_headers: dict[str, str] | None = None
|
|
env_vars: list[MCPEnvVar] | None = None
|
|
status: Literal["healthy", "unhealthy", "unknown"] | None = Field(
|
|
default="unknown",
|
|
description="Health status: 'healthy', 'unhealthy', 'unknown'",
|
|
)
|
|
last_health_check: datetime | None = None
|
|
health_check_error: str | None = None
|
|
command: str | None = None
|
|
args: list[str] = Field(default_factory=list)
|
|
env: dict[str, str] = Field(default_factory=dict)
|
|
issuer: str | None = None
|
|
authorization_url: str | None = None
|
|
token_url: str | None = None
|
|
registration_url: str | None = None
|
|
oauth2_flow: Literal["client_credentials", "authorization_code"] | None = None
|
|
# Token Exchange (OBO) fields — RFC 8693. ``audience`` is named for the RFC's
|
|
# request parameter (token-exchange only); RFC 8707 resource indicators are a
|
|
# separate concept named ``resource`` in the v2 egress types. A null
|
|
# ``subject_token_type`` means DEFAULT_SUBJECT_TOKEN_TYPE (litellm.types.mcp),
|
|
# applied at the egress build sites.
|
|
token_exchange_endpoint: str | None = None
|
|
audience: str | None = None
|
|
subject_token_type: str | None = None
|
|
token_exchange_profile: str | None = None
|
|
allow_all_keys: bool = False
|
|
available_on_public_internet: bool = True
|
|
delegate_auth_to_upstream: bool = False
|
|
oauth_passthrough: bool = False
|
|
dcr_bridge: bool | None = None
|
|
is_byok: bool = False
|
|
byok_description: list[str] = Field(default_factory=list)
|
|
byok_api_key_help_url: str | None = None
|
|
has_user_credential: bool | None = None
|
|
connected_app_reachable: bool | None = None
|
|
source_url: str | None = None
|
|
timeout: float | None = None
|
|
max_concurrent_requests: int | None = None
|
|
approval_status: str | None = Field(
|
|
default="active",
|
|
description="Approval status: 'pending_review', 'active', 'rejected'",
|
|
)
|
|
submitted_by: str | None = None
|
|
submitted_at: datetime | None = None
|
|
reviewed_at: datetime | None = None
|
|
review_notes: str | None = None
|