apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-23 15:48:11 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fixbug: di run

Browse source
This commit is contained in:
莘权 马 2024-03-30 20:19:46 +08:00
parent eedba3038d
commit a27fcd7e52
2 changed files with 17 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
metagpt/roles/di/data_interpreter.py
View file

@ -14,7 +14,7 @@ from metagpt.roles import Role
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.common import CodeParser, role_raise_decorator
REACT_THINK_PROMPT = """
# User Requirement
@ -182,3 +182,11 @@ class DataInterpreter(Role):
print(result)
data_info = DATA_INFO.format(info=result)
self.working_memory.add(Message(content=data_info, role="user", cause_by=CheckData))
@role_raise_decorator
async def run(self, with_message=None) -> Message | None:
if not self.rc.todo:
self.set_actions([WriteAnalysisCode])
self._set_state(0)
return await super().run(with_message)