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

remove "[Message] from A to B:'"in quick answer

Browse source
This commit is contained in:
黄伟韬 2024-08-20 11:58:10 +08:00
parent 6b4c00f286
commit f3d9b8e767
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/roles/di/role_zero.py
View file

@ -295,13 +295,13 @@ class RoleZero(Role):
if "QUICK" in intent_result or "AMBIGUOUS" in intent_result: # llm call with the original context
async with ThoughtReporter(enable_llm_stream=True) as reporter:
await reporter.async_report({"type": "quick"})
# remove "[Message] from A to B :" tag.
pattern = r"\[Message\] from .* to .*:\s*"
memory[-1].content = re.sub(pattern, "", memory[-1].content, count=1)
answer = await self.llm.aask(
self.llm.format_msg(memory),
system_msgs=[QUICK_RESPONSE_SYSTEM_PROMPT.format(role_info=self._get_prefix())],
)
# If the answer contains the substring '[Message] from A to B:', remove it.
pattern = r"\[Message\] from .* to .*:\s*"
answer = re.sub(pattern, "", answer, count=1)
elif "SEARCH" in intent_result:
query = "\n".join(str(msg) for msg in memory)
answer = await SearchEnhancedQA().run(query)