From bfc7d6b134b8c5dbf0ce6ede6a00032092137276 Mon Sep 17 00:00:00 2001 From: femto Date: Wed, 7 Aug 2024 13:13:34 +0800 Subject: [PATCH] rm sk agent --- tests/metagpt/planner/test_action_planner.py | 32 ----------------- tests/metagpt/planner/test_basic_planner.py | 37 -------------------- 2 files changed, 69 deletions(-) delete mode 100644 tests/metagpt/planner/test_action_planner.py delete mode 100644 tests/metagpt/planner/test_basic_planner.py diff --git a/tests/metagpt/planner/test_action_planner.py b/tests/metagpt/planner/test_action_planner.py deleted file mode 100644 index 1bc451db8..000000000 --- a/tests/metagpt/planner/test_action_planner.py +++ /dev/null @@ -1,32 +0,0 @@ -#!/usr/bin/env python -# -*- coding: utf-8 -*- -""" -@Time : 2023/9/16 20:03 -@Author : femto Zheng -@File : test_basic_planner.py -@Modified By: mashenquan, 2023-11-1. In accordance with Chapter 2.2.1 and 2.2.2 of RFC 116, utilize the new message - distribution feature for message handling. -""" -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 UserRequirement -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.put_message(Message(content=task, cause_by=UserRequirement)) - await role._observe() - await role._think() # it will choose mathskill.Add - assert "1100" == (await role._act()).content diff --git a/tests/metagpt/planner/test_basic_planner.py b/tests/metagpt/planner/test_basic_planner.py deleted file mode 100644 index f406143ee..000000000 --- a/tests/metagpt/planner/test_basic_planner.py +++ /dev/null @@ -1,37 +0,0 @@ -#!/usr/bin/env python -# -*- coding: utf-8 -*- -""" -@Time : 2023/9/16 20:03 -@Author : femto Zheng -@File : test_basic_planner.py -@Modified By: mashenquan, 2023-11-1. In accordance with Chapter 2.2.1 and 2.2.2 of RFC 116, utilize the new message - distribution feature for message handling. -""" -import pytest -from semantic_kernel.core_skills import TextSkill - -from metagpt.actions import UserRequirement -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.put_message(Message(content=task, cause_by=UserRequirement)) - await role._observe() - 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