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)