apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-01 11:56:24 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix parse_code bug.

Browse source
This commit is contained in:
刘棒棒 2024-01-31 21:00:02 +08:00
parent 487169ee61
commit b9663cebbd
1 changed files with 14 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

25
metagpt/utils/common.py
View file

@ -265,19 +265,22 @@ class CodeParser:
return block_dict
@classmethod
def parse_code(cls, block: str, text: str, lang: str = "", start_ends: str = r'["\'`]{3}') -> str:
def parse_code(cls, block: str, text: str, lang: str = "") -> str:
if block:
text = cls.parse_block(block, text)
pattern = rf"{start_ends}{lang}.*?\s+(.*?){start_ends}"
match = re.search(pattern, text, re.DOTALL)
if match:
code = match.group(1)
else:
logger.error(f"{pattern} not match following text:")
logger.error(text)
# raise Exception
return text # just assume original text is code
return code
start_ends = ["```", '"""', "'''"]
patterns = []
for start_end in start_ends:
pattern = rf"{start_end}{lang}.*?\s+(.*?){start_end}"
match = re.search(pattern, text, re.DOTALL)
if match:
code = match.group(1)
return code
patterns.append(pattern)
logger.error(f"{patterns} not match following text:")
logger.error(text)
# raise Exception
return text # just assume original text is code
@classmethod
def parse_str(cls, block: str, text: str, lang: str = ""):