Merge pull request #310 from 18620146591/pr

adding an action

metagpt/actions/detail_mining.py

@@ -0,0 +1,52 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2023/9/12 17:45
+@Author  : fisherdeng
+@File    : detail_mining.py
+"""
+from metagpt.actions import Action, ActionOutput
+from metagpt.logs import logger
+
+PROMPT_TEMPLATE = """
+##TOPIC
+{topic}
+
+##RECORD
+{record}
+
+##Format example
+{format_example}
+-----
+
+Task: Refer to the "##TOPIC" (discussion objectives) and "##RECORD" (discussion records) to further inquire about the details that interest you, within a word limit of 150 words.
+Special Note 1: Your intention is solely to ask questions without endorsing or negating any individual's viewpoints.
+Special Note 2: This output should only include the topic "##OUTPUT". Do not add, remove, or modify the topic. Begin the output with '##OUTPUT', followed by an immediate line break, and then proceed to provide the content in the specified format as outlined in the "##Format example" section.
+Special Note 3: The output should be in the same language as the input.
+"""
+FORMAT_EXAMPLE = """
+
+##
+
+##OUTPUT
+...(Please provide the specific details you would like to inquire about here.)
+
+##
+
+##
+"""
+OUTPUT_MAPPING = {
+    "OUTPUT": (str, ...),
+}
+
+
+class DetailMining(Action):
+    """This class allows LLM to further mine noteworthy details based on specific "##TOPIC"(discussion topic) and "##RECORD" (discussion records), thereby deepening the discussion.
+    """
+    def __init__(self, name="", context=None, llm=None):
+        super().__init__(name, context, llm)
+
+    async def run(self, topic, record) -> ActionOutput:
+        prompt = PROMPT_TEMPLATE.format(topic=topic, record=record, format_example=FORMAT_EXAMPLE)
+        rsp = await self._aask_v1(prompt, "detail_mining", OUTPUT_MAPPING)
+        return rsp

tests/metagpt/actions/test_detail_mining.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2023/9/13 00:26
+@Author  : fisherdeng
+@File    : test_detail_mining.py
+"""
+import pytest
+
+from metagpt.actions.detail_mining import DetailMining
+from metagpt.logs import logger
+
+@pytest.mark.asyncio
+async def test_detail_mining():
+    topic = "如何做一个生日蛋糕"
+    record = "我认为应该先准备好材料，然后再开始做蛋糕。"
+    detail_mining = DetailMining("detail_mining")
+    rsp = await detail_mining.run(topic=topic, record=record)
+    logger.info(f"{rsp.content=}")
+    
+    assert '##OUTPUT' in rsp.content
+    assert '蛋糕' in rsp.content
+