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

refine code

Browse source
This commit is contained in:
lidanyang 2024-07-12 16:44:05 +08:00
parent df340cb330
commit df29e16e2b
5 changed files with 23 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

10
metagpt/actions/di/write_analysis_code.py
View file

@ -16,7 +16,6 @@ from metagpt.prompts.di.write_analysis_code import (
REFLECTION_PROMPT,
REFLECTION_SYSTEM_MSG,
STRUCTUAL_PROMPT,
BROWSER_INFO,
)
from metagpt.schema import Message, Plan
from metagpt.utils.common import CodeParser, remove_comments
@ -49,14 +48,9 @@ class WriteAnalysisCode(Action):
plan_status=plan_status,
tool_info=tool_info,
)
message = [Message(content=structual_prompt, role="user")]
browser_actions = [msg for msg in working_memory if msg.cause_by == "browser"]
if browser_actions:
browser_prompt = BROWSER_INFO.format(browser_actions=browser_actions)
message = [Message(content=browser_prompt, role="user")] + message
working_memory = [msg for msg in working_memory if msg.cause_by != "browser"] if use_reflection else []
context = self.llm.format_msg(message + working_memory)
working_memory = working_memory or []
context = self.llm.format_msg([Message(content=structual_prompt, role="user")] + working_memory)
# LLM call
if use_reflection: