diff --git a/metagpt/roles/di/role_zero.py b/metagpt/roles/di/role_zero.py index 234064088..995e7e5d8 100644 --- a/metagpt/roles/di/role_zero.py +++ b/metagpt/roles/di/role_zero.py @@ -158,7 +158,8 @@ class RoleZero(Role): break context = self.llm.format_msg(memory + [UserMessage(content=prompt)]) # print(*context, sep="\n" + "*" * 5 + "\n") - async with ThoughtReporter(enable_llm_stream=True): + async with ThoughtReporter(enable_llm_stream=True) as reporter: + await reporter.async_report({"type": "react"}) self.command_rsp = await self.llm.aask(context, system_msgs=self.system_msg) self.rc.memory.add(AIMessage(content=self.command_rsp)) @@ -216,7 +217,9 @@ class RoleZero(Role): return rsp_msg context = self.llm.format_msg(self.get_memories(k=4) + [UserMessage(content=QUICK_THINK_PROMPT)]) - rsp = await self.llm.aask(context) + async with ThoughtReporter(enable_llm_stream=True) as reporter: + await reporter.async_report({"type": "quick"}) + rsp = await self.llm.aask(context) pattern = r"#YES#,? ?" if re.search(pattern, rsp): diff --git a/metagpt/utils/report.py b/metagpt/utils/report.py index ed40e5dde..5d1cd93e4 100644 --- a/metagpt/utils/report.py +++ b/metagpt/utils/report.py @@ -266,10 +266,6 @@ class ThoughtReporter(ObjectReporter): block: Literal[BlockType.THOUGHT] = BlockType.THOUGHT - async def __aenter__(self): - await self.async_report({}) - return await super().__aenter__() - class FileReporter(ResourceReporter): """File resource callback for reporting complete file paths.