add thought reporter

metagpt/roles/di/data_analyst.py

@@ -20,6 +20,7 @@ from metagpt.strategy.thinking_command import (
 )
 from metagpt.tools.tool_recommend import BM25ToolRecommender
 from metagpt.utils.common import CodeParser
+from metagpt.utils.report import ThoughtReporter
 
 
 class DataAnalyst(DataInterpreter):
@@ -82,8 +83,8 @@ class DataAnalyst(DataInterpreter):
             available_commands=prepare_command_prompt(self.available_commands),
         )
         context = self.llm.format_msg(self.working_memory.get() + [Message(content=prompt, role="user")])
-
-        rsp = await self.llm.aask(context)
+        async with ThoughtReporter():
+            rsp = await self.llm.aask(context)
         self.commands = json.loads(CodeParser.parse_code(block=None, text=rsp))
         self.rc.memory.add(Message(content=rsp, role="assistant"))
 

metagpt/roles/di/data_interpreter.py

@@ -15,6 +15,7 @@ from metagpt.schema import Message, Task, TaskResult
 from metagpt.strategy.task_type import TaskType
 from metagpt.tools.tool_recommend import BM25ToolRecommender, ToolRecommender
 from metagpt.utils.common import CodeParser
+from metagpt.utils.report import ThoughtReporter
 
 REACT_THINK_PROMPT = """
 # User Requirement
@@ -73,7 +74,8 @@ class DataInterpreter(Role):
             return True
 
         prompt = REACT_THINK_PROMPT.format(user_requirement=self.user_requirement, context=context)
-        rsp = await self.llm.aask(prompt)
+        async with ThoughtReporter():
+            rsp = await self.llm.aask(prompt)
         rsp_dict = json.loads(CodeParser.parse_code(text=rsp))
         self.working_memory.add(Message(content=rsp_dict["thoughts"], role="assistant"))
         need_action = rsp_dict["state"]

metagpt/roles/di/team_leader.py

@@ -20,6 +20,7 @@ from metagpt.strategy.thinking_command import (
     run_commands,
 )
 from metagpt.utils.common import CodeParser
+from metagpt.utils.report import ThoughtReporter
 
 
 class TeamLeader(Role):
@@ -69,7 +70,8 @@ class TeamLeader(Role):
         )
         context = self.llm.format_msg(self.get_memories(k=10) + [Message(content=prompt, role="user")])
 
-        rsp = await self.llm.aask(context, system_msgs=[SYSTEM_PROMPT])
+        async with ThoughtReporter():
+            rsp = await self.llm.aask(context, system_msgs=[SYSTEM_PROMPT])
         self.commands = json.loads(CodeParser.parse_code(text=rsp))
         self.rc.memory.add(Message(content=rsp, role="assistant"))
 

metagpt/roles/role.py

@@ -47,6 +47,7 @@ from metagpt.strategy.planner import Planner
 from metagpt.utils.common import any_to_name, any_to_str, role_raise_decorator
 from metagpt.utils.project_repo import ProjectRepo
 from metagpt.utils.repair_llm_raw_output import extract_state_value_from_output
+from metagpt.utils.report import ThoughtReporter
 
 if TYPE_CHECKING:
     from metagpt.environment import Environment  # noqa: F401
@@ -381,8 +382,8 @@ class Role(SerializationMixin, ContextMixin, BaseModel):
             n_states=len(self.states) - 1,
             previous_state=self.rc.state,
         )
-
-        next_state = await self.llm.aask(prompt)
+        async with ThoughtReporter():
+            next_state = await self.llm.aask(prompt)
         next_state = extract_state_value_from_output(next_state)
         logger.debug(f"{prompt=}")
 

metagpt/utils/common.py

@@ -646,7 +646,7 @@ def role_raise_decorator(func):
             raise Exception(format_trackback_info(limit=None))
         except Exception as e:
             if self.latest_observed_msg:
-                logger.warning(
+                logger.exception(
                     "There is a exception in role's execution, in order to resume, "
                     "we delete the newest role communication message in the role's memory."
                 )

metagpt/utils/report.py

@@ -39,6 +39,7 @@ class BlockType(str, Enum):
     GALLERY = "Gallery"
     NOTEBOOK = "Notebook"
     DOCS = "Docs"
+    THOUGHT = "Thought"
 
 
 END_MARKER_NAME = "end_marker"
@@ -259,6 +260,16 @@ class TaskReporter(ObjectReporter):
     block: Literal[BlockType.TASK] = BlockType.TASK
 
 
+class ThoughtReporter(ObjectReporter):
+    """Reporter for object resources to Task Block."""
+
+    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.