apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-28 18:36:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix rsp type

Browse source
This commit is contained in:
lidanyang 2023-12-01 13:44:24 +08:00
parent 35e8a501c5
commit cb8a8ffd5c
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
metagpt/actions/write_plan.py
View file

@ -33,7 +33,7 @@ class WritePlan(Action):
```
"""
async def assign_task_type(self, tasks: List[Dict]) -> List[Dict]:
async def assign_task_type(self, tasks: List[Dict]) -> str:
"""Assign task type to each task in tasks
Args:
@ -51,9 +51,9 @@ class WritePlan(Action):
task_type_list = rsp["task_type"]
for task, task_type in zip(tasks, task_type_list):
task["task_type"] = task_type
return tasks
return json.dumps(tasks)
async def run(self, context: List[Message], max_tasks: int = 5) -> List[Dict]:
async def run(self, context: List[Message], max_tasks: int = 5) -> str:
prompt = (
self.PROMPT_TEMPLATE.replace("__context__", "\n".join([str(ct) for ct in context]))
# .replace("__current_plan__", current_plan)
@ -65,6 +65,7 @@ class WritePlan(Action):
return rsp
@staticmethod
def rsp_to_tasks(rsp: List[Dict]) -> List[Task]:
def rsp_to_tasks(rsp: str) -> List[Task]:
rsp = json.loads(rsp)
tasks = [Task(**task_config) for task_config in rsp]
return tasks