bugfix: #1547 Before calling serializeToString in mmdc_playwright.py,…

… make sure the SVG element exists.
geekan · Nov 1, 2024 · 0443865 · 0443865
1 parent b5bb4d7
commit 0443865
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/metagpt/utils/mmdc_playwright.py b/metagpt/utils/mmdc_playwright.py
@@ -81,10 +81,13 @@ async def console_message(msg):
             if "svg" in suffixes:
                 svg_xml = await page.evaluate(
                     """() => {
-                    const svg = document.querySelector('svg');
-                    const xmlSerializer = new XMLSerializer();
-                    return xmlSerializer.serializeToString(svg);
-                }"""
+                        const svg = document.querySelector('svg');
+                        if (!svg) {
+                            throw new Error('SVG element not found');
+                        }
+                        const xmlSerializer = new XMLSerializer();
+                        return xmlSerializer.serializeToString(svg);
+                    }"""
                 )
                 logger.info(f"Generating {output_file_without_suffix}.svg..")
                 with open(f"{output_file_without_suffix}.svg", "wb") as f: