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

Merge branch 'fix-parse_code' into 'code_intepreter'

Browse source 
fix CodeParser.parse_code bug.

See merge request agents/data_agents_opt!69
This commit is contained in:
林义章 2024-01-31 13:24:34 +00:00
commit df58d55635
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 = ""):