From 832494706e7f93407b43780daae7f0a6bf96d981 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= Date: Tue, 10 Sep 2024 16:24:04 +0800 Subject: [PATCH] feat: mermaid + font --- metagpt/utils/mermaid.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/metagpt/utils/mermaid.py b/metagpt/utils/mermaid.py index 008bb849d..88e58ae44 100644 --- a/metagpt/utils/mermaid.py +++ b/metagpt/utils/mermaid.py @@ -7,6 +7,7 @@ """ import asyncio import os +import re from pathlib import Path from typing import List, Optional @@ -38,6 +39,9 @@ async def mermaid_to_file( Returns: int: 0 if the conversion is successful, -1 if the conversion fails. """ + file_head = "%%{init: {'theme': 'default', 'themeVariables': { 'fontFamily': 'Inter' }}}%%\n" + if not re.match(r"^%%\{.+", mermaid_code): + mermaid_code = file_head + mermaid_code suffixes = suffixes or ["svg"] # Write the Mermaid code to a temporary file config = config if config else Config.default()