refactor: rename async_put_message to put_message

This commit is contained in:
莘权 马 2023-11-02 10:40:26 +08:00
parent bfaeda0a90
commit bc1a757293
6 changed files with 11 additions and 11 deletions

View file

@ -26,7 +26,7 @@ async def test_action_planner():
role.import_skill(TimeSkill(), "time")
role.import_skill(TextSkill(), "text")
task = "What is the sum of 110 and 990?"
role.async_put_message(Message(content=task, cause_by=BossRequirement.get_class_name()))
role.put_message(Message(content=task, cause_by=BossRequirement.get_class_name()))
await role._observe()
await role._think() # it will choose mathskill.Add
assert "1100" == (await role._act()).content

View file

@ -29,7 +29,7 @@ async def test_basic_planner():
role.import_semantic_skill_from_directory(SKILL_DIRECTORY, "WriterSkill")
role.import_skill(TextSkill(), "TextSkill")
# using BasicPlanner
role.async_put_message(Message(content=task, cause_by=BossRequirement.get_class_name()))
role.put_message(Message(content=task, cause_by=BossRequirement.get_class_name()))
await role._observe()
await role._think()
# assuming sk_agent will think he needs WriterSkill.Brainstorm and WriterSkill.Translate

View file

@ -16,7 +16,7 @@ from tests.metagpt.roles.mock import MockMessages
@pytest.mark.asyncio
async def test_architect():
role = Architect()
role.async_put_message(MockMessages.req)
role.put_message(MockMessages.req)
rsp = await role.run(MockMessages.prd)
logger.info(rsp)
assert len(rsp.content) > 0

View file

@ -23,9 +23,9 @@ from tests.metagpt.roles.mock import (
async def test_engineer():
engineer = Engineer()
engineer.async_put_message(MockMessages.req)
engineer.async_put_message(MockMessages.prd)
engineer.async_put_message(MockMessages.system_design)
engineer.put_message(MockMessages.req)
engineer.put_message(MockMessages.prd)
engineer.put_message(MockMessages.system_design)
rsp = await engineer.run(MockMessages.tasks)
logger.info(rsp)