Merge branch 'main' into add_open_interpreter

tests/metagpt/actions/test_write_tutorial.py

@@ -0,0 +1,40 @@
+#!/usr/bin/env python3
+# _*_ coding: utf-8 _*_
+"""
+@Time    : 2023/9/6 21:41:34
+@Author  : Stitch-z
+@File    : test_write_tutorial.py
+"""
+from typing import Dict
+
+import pytest
+
+from metagpt.actions.write_tutorial import WriteDirectory, WriteContent
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    ("language", "topic"),
+    [("English", "Write a tutorial about Python")]
+)
+async def test_write_directory(language: str, topic: str):
+    ret = await WriteDirectory(language=language).run(topic=topic)
+    assert isinstance(ret, dict)
+    assert "title" in ret
+    assert "directory" in ret
+    assert isinstance(ret["directory"], list)
+    assert len(ret["directory"])
+    assert isinstance(ret["directory"][0], dict)
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    ("language", "topic", "directory"),
+    [("English", "Write a tutorial about Python", {"Introduction": ["What is Python?", "Why learn Python?"]})]
+)
+async def test_write_content(language: str, topic: str, directory: Dict):
+    ret = await WriteContent(language=language, directory=directory).run(topic=topic)
+    assert isinstance(ret, str)
+    assert list(directory.keys())[0] in ret
+    for value in list(directory.values())[0]:
+        assert value in ret

tests/metagpt/planner/__init__.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2023/9/16 20:03
+@Author  : femto Zheng
+@File    : __init__.py
+"""

tests/metagpt/planner/test_action_planner.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2023/9/16 20:03
+@Author  : femto Zheng
+@File    : test_basic_planner.py
+"""
+import pytest
+from semantic_kernel.core_skills import FileIOSkill, MathSkill, TextSkill, TimeSkill
+from semantic_kernel.planning.action_planner.action_planner import ActionPlanner
+
+from metagpt.actions import BossRequirement
+from metagpt.roles.sk_agent import SkAgent
+from metagpt.schema import Message
+
+
+@pytest.mark.asyncio
+async def test_action_planner():
+    role = SkAgent(planner_cls=ActionPlanner)
+    # let's give the agent 4 skills
+    role.import_skill(MathSkill(), "math")
+    role.import_skill(FileIOSkill(), "fileIO")
+    role.import_skill(TimeSkill(), "time")
+    role.import_skill(TextSkill(), "text")
+    task = "What is the sum of 110 and 990?"
+    role.recv(Message(content=task, cause_by=BossRequirement))
+
+    await role._think()  # it will choose mathskill.Add
+    assert "1100" == (await role._act()).content

tests/metagpt/planner/test_basic_planner.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+@Time    : 2023/9/16 20:03
+@Author  : femto Zheng
+@File    : test_basic_planner.py
+"""
+import pytest
+from semantic_kernel.core_skills import TextSkill
+
+from metagpt.actions import BossRequirement
+from metagpt.const import SKILL_DIRECTORY
+from metagpt.roles.sk_agent import SkAgent
+from metagpt.schema import Message
+
+
+@pytest.mark.asyncio
+async def test_basic_planner():
+    task = """
+        Tomorrow is Valentine's day. I need to come up with a few date ideas. She speaks French so write it in French.
+        Convert the text to uppercase"""
+    role = SkAgent()
+
+    # let's give the agent some skills
+    role.import_semantic_skill_from_directory(SKILL_DIRECTORY, "SummarizeSkill")
+    role.import_semantic_skill_from_directory(SKILL_DIRECTORY, "WriterSkill")
+    role.import_skill(TextSkill(), "TextSkill")
+    # using BasicPlanner
+    role.recv(Message(content=task, cause_by=BossRequirement))
+    await role._think()
+    # assuming sk_agent will think he needs WriterSkill.Brainstorm and WriterSkill.Translate
+    assert "WriterSkill.Brainstorm" in role.plan.generated_plan.result
+    assert "WriterSkill.Translate" in role.plan.generated_plan.result
+    # assert "SALUT" in (await role._act()).content #content will be some French

tests/metagpt/roles/test_tutorial_assistant.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+# _*_ coding: utf-8 _*_
+"""
+@Time    : 2023/9/6 23:11:27
+@Author  : Stitch-z
+@File    : test_tutorial_assistant.py
+"""
+import aiofiles
+import pytest
+
+from metagpt.roles.tutorial_assistant import TutorialAssistant
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    ("language", "topic"),
+    [("Chinese", "Write a tutorial about Python")]
+)
+async def test_tutorial_assistant(language: str, topic: str):
+    topic = "Write a tutorial about MySQL"
+    role = TutorialAssistant(language=language)
+    msg = await role.run(topic)
+    filename = msg.content
+    title = filename.split("/")[-1].split(".")[0]
+    async with aiofiles.open(filename, mode="r") as reader:
+        content = await reader.read()
+        assert content.startswith(f"# {title}")

tests/metagpt/utils/test_file.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+# _*_ coding: utf-8 _*_
+"""
+@Time    : 2023/9/4 15:40:40
+@Author  : Stitch-z
+@File    : test_file.py
+"""
+from pathlib import Path
+
+import aiofiles
+import pytest
+
+from metagpt.utils.file import File
+
+
+@pytest.mark.asyncio
+@pytest.mark.parametrize(
+    ("root_path", "filename", "content"),
+    [(Path("/code/MetaGPT/data/tutorial_docx/2023-09-07_17-05-20"), "test.md", "Hello World!")]
+)
+async def test_write_file(root_path: Path, filename: str, content: bytes):
+    full_file_name = await File.write(root_path=root_path, filename=filename, content=content.encode('utf-8'))
+    assert isinstance(full_file_name, Path)
+    assert root_path / filename == full_file_name
+    async with aiofiles.open(full_file_name, mode="r") as reader:
+        body = await reader.read()
+        assert body == content