apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-08 15:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

update _process_code.

Browse source
This commit is contained in:
刘棒棒 2024-01-22 18:54:21 +08:00
parent 1bf9e023e4
commit 0ad3de69a1
1 changed files with 3 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

16
metagpt/actions/execute_code.py
View file

@ -165,7 +165,7 @@ class ExecutePyCode(ExecuteCode, Action):
# 如果在Python脚本中运行，__file__ 变量存在
return False
def _process_code(self, code: Union[str, Dict, Message], language: str = None) -> Tuple:
def _process_code(self, code: Union[str, Dict], language: str = None) -> Tuple:
language = language or "python"
if isinstance(code, str) and Path(code).suffix in (".py", ".txt"):
code = Path(code).read_text(encoding="utf-8")
@ -173,20 +173,10 @@ class ExecutePyCode(ExecuteCode, Action):
if isinstance(code, str):
return code, language
if isinstance(code, dict):
assert "code" in code
if "language" not in code:
code["language"] = "python"
code, language = code["code"], code["language"]
elif isinstance(code, Message):
if isinstance(code.content, dict) and "language" not in code.content:
code.content["language"] = "python"
code, language = code.content["code"], code.content["language"]
elif isinstance(code.content, str):
code, language = code.content, language
else:
raise ValueError(f"Not support code type {type(code).__name__}.")
code = code["code"]
return code, language
async def run_cell(self, cell: NotebookNode, cell_index: int) -> Tuple[bool, str]: