apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-29 19:06:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

display markdown.

Browse source
This commit is contained in:
刘棒棒 2024-01-16 12:29:52 +08:00
parent 29fd7117ef
commit eef77d1628
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
metagpt/actions/execute_code.py
View file

@ -18,6 +18,8 @@ from nbformat import NotebookNode
from nbformat.v4 import new_code_cell, new_output, new_markdown_cell
from rich.console import Console
from rich.syntax import Syntax
from rich.markdown import Markdown
from metagpt.actions import Action
from metagpt.logs import logger
@ -97,8 +99,12 @@ class ExecutePyCode(ExecuteCode, Action):
def _display(self, code, language: str = "python"):
if language == "python":
code = Syntax(code, "python", theme="paraiso-dark", line_numbers=True)
self.console.print("\n")
self.console.print(code)
elif language == "markdown":
code = Markdown(code, inline_code_theme="paraiso-dark")
self.console.print(code)
else:
raise ValueError(f"Only support for python, markdown, but got {language}")
def add_output_to_cell(self, cell, output):
if "outputs" not in cell:
@ -221,10 +227,12 @@ class ExecutePyCode(ExecuteCode, Action):
# code success
outputs = self.parse_outputs(self.nb.cells[-1].outputs)
return truncate(remove_escape_and_color_codes(outputs), is_success=success)
else:
elif language == 'markdown':
# markdown
self.add_markdown_cell(code)
return code, True
else:
raise ValueError(f"Only support for language: python, markdown, but got {language}, ")
def truncate(result: str, keep_len: int = 2000, is_success: bool = True):