scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-16 04:29:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 225

fix(mcp): satisfy strict lint for safe YAML loader

Browse source
This commit is contained in:
Sisyphus 2026-10-04 18:08:21 +08:00
parent 45a3cbc33d
commit 771f3778d0
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/_experimental/mcp_server/openapi_to_mcp_generator.py
View file

@ -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)