fix(plugin): fix footnote plugins when rendering ast

lepture · lepture · commit a72895289234 · 2025-02-19T12:37:50.000+09:00
diff --git a/src/mistune/markdown.py b/src/mistune/markdown.py
@@ -40,7 +40,7 @@ def __init__(
         self.before_parse_hooks: List[Callable[["Markdown", BlockState], None]] = []
         self.before_render_hooks: List[Callable[["Markdown", BlockState], Any]] = []
         self.after_render_hooks: List[
-            Callable[["Markdown", Union[str, List[Dict[str, Any]]], BlockState], str]
+            Callable[["Markdown", Union[str, List[Dict[str, Any]]], BlockState], Union[str, List[Dict[str, Any]]]]
         ] = []
 
         if plugins:
diff --git a/src/mistune/plugins/footnotes.py b/src/mistune/plugins/footnotes.py
@@ -94,8 +94,7 @@ def parse_footnote_item(
 
 def md_footnotes_hook(
     md: "Markdown", result: Union[str, List[Dict[str, Any]]], state: BlockState
-) -> str:
-    assert isinstance(result, str)
+) -> Union[str, List[Dict[str, Any]]]:
     notes = state.env.get("footnotes")
     if not notes:
         return result
@@ -107,8 +106,7 @@ def md_footnotes_hook(
     state = BlockState()
     state.tokens = [{'type': 'footnotes', 'children': children}]
     output = md.render_state(state)
-    assert isinstance(output, str)
-    return result + output
+    return result + output  # type: ignore[operator]
 
 
 def render_footnote_ref(renderer: "BaseRenderer", key: str, index: int) -> str:
