diff --git a/README.md b/README.md index 2bff88b55d..508b86bb1e 100644 --- a/README.md +++ b/README.md @@ -67,7 +67,7 @@ python setup.py install - don't forget to the configuration for mmdc in config.yml ```yml - PUPPETEER_CONFIG: "./puppeteer-config.json" + PUPPETEER_CONFIG: "./config/puppeteer-config.json" MMDC: "./node_modules/.bin/mmdc" ``` diff --git a/config/config.yaml b/config/config.yaml index 4abec8f252..1ca86ee427 100644 --- a/config/config.yaml +++ b/config/config.yaml @@ -57,7 +57,6 @@ SD_T2I_API: "/sdapi/v1/txt2img" #LONG_TERM_MEMORY: false #### for Mermaid CLI -# adds support for installing mmdc (Mermaid CLI) locally on the user's machine. -# PUPPETEER_SKIP_CHROMIUM_DOWNLOAD=true npm install @mermaid-js/mermaid-cli -#PUPPETEER_CONFIG: "./puppeteer-config.json" +## If you installed mmdc (Mermaid CLI) only for metagpt then enable the following configuration. +#PUPPETEER_CONFIG: "./config/puppeteer-config.json" #MMDC: "./node_modules/.bin/mmdc" \ No newline at end of file diff --git a/puppeteer-config.json b/config/puppeteer-config.json similarity index 100% rename from puppeteer-config.json rename to config/puppeteer-config.json diff --git a/metagpt/utils/mermaid.py b/metagpt/utils/mermaid.py index d1bbb85f7b..3850faae3a 100644 --- a/metagpt/utils/mermaid.py +++ b/metagpt/utils/mermaid.py @@ -16,8 +16,6 @@ IS_DOCKER = os.environ.get('AM_I_IN_A_DOCKER_CONTAINER', 'false').lower() - - def mermaid_to_file(mermaid_code, output_file_without_suffix, width=2048, height=2048) -> int: """suffix: png/svg/pdf @@ -41,15 +39,15 @@ def mermaid_to_file(mermaid_code, output_file_without_suffix, width=2048, height # Call the `mmdc` command to convert the Mermaid code to a PNG logger.info(f"Generating {output_file}..") if IS_DOCKER == 'true': - subprocess.run(['mmdc', '-p', '/app/metagpt/puppeteer-config.json', '-i', + subprocess.run(['mmdc', '-p', '/app/metagpt/config/puppeteer-config.json', '-i', str(tmp), '-o', output_file, '-w', str(width), '-H', str(height)]) else: if CONFIG.puppeteer_config: - subprocess.run([CONFIG.mmdc,'-p',CONFIG.puppeteer_config, '-i', str(tmp), '-o', - output_file, '-w', str(width), '-H', str(height)]) + subprocess.run([CONFIG.mmdc, '-p', CONFIG.puppeteer_config, '-i', str(tmp), '-o', + output_file, '-w', str(width), '-H', str(height)]) else: - subprocess.run( [CONFIG.mmdc, '-i', str(tmp), '-o', - output_file, '-w', str(width), '-H', str(height)]) + subprocess.run([CONFIG.mmdc, '-i', str(tmp), '-o', + output_file, '-w', str(width), '-H', str(height)]) return 0