add write_review action and its test

metagpt/actions/action_node.py

@@ -41,10 +41,10 @@ Fill in the above nodes based on the format example.
 """
 
 
-def dict_to_markdown(d, prefix="-", postfix="\n"):
+def dict_to_markdown(d, prefix="##", kv_sep="\n", postfix="\n"):
     markdown_str = ""
     for key, value in d.items():
-        markdown_str += f"{prefix} {key}: {value}{postfix}"
+        markdown_str += f"{prefix}{key}{kv_sep}{value}{postfix}"
     return markdown_str
 
 

metagpt/actions/write_review.py

@@ -0,0 +1,40 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Author  : alexanderwu
+@File    : write_review.py
+"""
+from typing import List
+
+from metagpt.actions import Action
+from metagpt.actions.action_node import ActionNode
+
+# from metagpt.llm import LLM
+
+REVIEW = ActionNode(
+    key="Review",
+    expected_type=List[str],
+    instruction="Act as an experienced Reviewer and review the given output. Ask a series of critical questions, "
+    "concisely and clearly, to help the writer improve their work.",
+    example=[
+        "This is a good PRD, but I think it can be improved by adding more details.",
+    ],
+)
+
+LGTM = ActionNode(
+    key="LGTM",
+    expected_type=str,
+    instruction="If the output is good enough, give a LGTM (Looks Good To Me) to the writer, "
+    "else LBTM (Looks Bad To Me).",
+    example="LGTM",
+)
+
+WRITE_REVIEW_NODE = ActionNode.from_children("WRITE_REVIEW_NODE", [REVIEW, LGTM])
+
+
+class WriteReview(Action):
+    """This class allows LLM to further mine noteworthy details based on specific "##TOPIC"(discussion topic) and
+    "##RECORD" (discussion records), thereby deepening the discussion."""
+
+    async def run(self, context):
+        return await WRITE_REVIEW_NODE.fill(context=context, llm=self.llm, schema="markdown")

metagpt/utils/common.py

@@ -18,7 +18,7 @@ import os
 import platform
 import re
 import typing
-from typing import List, Tuple, Union
+from typing import List, Tuple, Union, get_args, get_origin
 
 import aiofiles
 import loguru
@@ -129,8 +129,31 @@ class OutputParser:
             parsed_data[block] = content
         return parsed_data
 
+    @staticmethod
+    def extract_content(text, tag="CONTENT"):
+        # Use regular expression to extract content between [CONTENT] and [/CONTENT]
+        extracted_content = re.search(rf"\[{tag}\](.*?)\[/{tag}\]", text, re.DOTALL)
+
+        if extracted_content:
+            return extracted_content.group(1).strip()
+        else:
+            return "No content found between [CONTENT] and [/CONTENT] tags."
+
+    @staticmethod
+    def is_supported_list_type(i):
+        origin = get_origin(i)
+        if origin is not List:
+            return False
+
+        args = get_args(i)
+        if args == (str,) or args == (Tuple[str, str],) or args == (List[str],):
+            return True
+
+        return False
+
     @classmethod
     def parse_data_with_mapping(cls, data, mapping):
+        data = cls.extract_content(text=data)
         block_dict = cls.parse_blocks(data)
         parsed_data = {}
         for block, content in block_dict.items():