From 771f3778d0fbfc671621cfaf45e59296ab6c9d5b Mon Sep 17 00:00:00 2001 From: Sisyphus Date: Sun, 4 Oct 2026 18:08:21 +0800 Subject: [PATCH] fix(mcp): satisfy strict lint for safe YAML loader --- .../_experimental/mcp_server/openapi_to_mcp_generator.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/_experimental/mcp_server/openapi_to_mcp_generator.py b/litellm/proxy/_experimental/mcp_server/openapi_to_mcp_generator.py index 60fa218d2da..0ce437bd7b3 100644 --- a/litellm/proxy/_experimental/mcp_server/openapi_to_mcp_generator.py +++ b/litellm/proxy/_experimental/mcp_server/openapi_to_mcp_generator.py @@ -183,13 +183,14 @@ def _parse_openapi_spec(text: str) -> Mapping[str, Any]: import yaml class _NoMergeSafeLoader(yaml.SafeLoader): - def compose_node(self, parent: Any, index: Any) -> Any: + def compose_node(self, parent: object, index: object) -> object: if self.check_event(yaml.events.AliasEvent): raise yaml.YAMLError("YAML aliases are not supported") return super().compose_node(parent, index) - def flatten_mapping(self, node: Any) -> None: - if any(key_node.tag == "tag:yaml.org,2002:merge" for key_node, _ in node.value): + def flatten_mapping(self, node: object) -> None: + node_values = getattr(node, "value", ()) + if any(key_node.tag == "tag:yaml.org,2002:merge" for key_node, _ in node_values): raise yaml.YAMLError("YAML merge keys are not supported") super().flatten_mapping(node)