mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-04-25 08:46:48 +02:00
109 lines
3.7 KiB
Python
109 lines
3.7 KiB
Python
"""
|
|
Filename: MetaGPT/examples/debate.py
|
|
Created Date: Tuesday, September 19th 2023, 6:52:25 pm
|
|
Author: garylin2099
|
|
@Modified By: mashenquan, 2023-11-1. In accordance with Chapter 2.1.3 of RFC 116, modify the data type of the `send_to`
|
|
value of the `Message` object; modify the argument type of `get_by_actions`.
|
|
"""
|
|
|
|
import asyncio
|
|
import platform
|
|
from typing import Any
|
|
|
|
import fire
|
|
|
|
from metagpt.actions import Action, UserRequirement
|
|
from metagpt.logs import logger
|
|
from metagpt.roles import Role
|
|
from metagpt.schema import Message
|
|
from metagpt.team import Team
|
|
|
|
|
|
class SpeakAloud(Action):
|
|
"""Action: Speak out aloud in a debate (quarrel)"""
|
|
|
|
PROMPT_TEMPLATE: str = """
|
|
## BACKGROUND
|
|
Suppose you are {name}, you are in a debate with {opponent_name}.
|
|
## DEBATE HISTORY
|
|
Previous rounds:
|
|
{context}
|
|
## YOUR TURN
|
|
Now it's your turn, you should closely respond to your opponent's latest argument, state your position, defend your arguments, and attack your opponent's arguments,
|
|
craft a strong and emotional response in 80 words, in {name}'s rhetoric and viewpoints, your will argue:
|
|
"""
|
|
name: str = "SpeakAloud"
|
|
|
|
async def run(self, context: str, name: str, opponent_name: str):
|
|
prompt = self.PROMPT_TEMPLATE.format(context=context, name=name, opponent_name=opponent_name)
|
|
# logger.info(prompt)
|
|
|
|
rsp = await self._aask(prompt)
|
|
|
|
return rsp
|
|
|
|
|
|
class Debator(Role):
|
|
name: str = ""
|
|
profile: str = ""
|
|
opponent_name: str = ""
|
|
|
|
def __init__(self, **data: Any):
|
|
super().__init__(**data)
|
|
self.set_actions([SpeakAloud])
|
|
self._watch([UserRequirement, SpeakAloud])
|
|
|
|
async def _observe(self) -> int:
|
|
await super()._observe()
|
|
# accept messages sent (from opponent) to self, disregard own messages from the last round
|
|
self.rc.news = [msg for msg in self.rc.news if msg.send_to == {self.name}]
|
|
return len(self.rc.news)
|
|
|
|
async def _act(self) -> Message:
|
|
logger.info(f"{self._setting}: to do {self.rc.todo}({self.rc.todo.name})")
|
|
todo = self.rc.todo # An instance of SpeakAloud
|
|
|
|
memories = self.get_memories()
|
|
context = "\n".join(f"{msg.sent_from}: {msg.content}" for msg in memories)
|
|
# print(context)
|
|
|
|
rsp = await todo.run(context=context, name=self.name, opponent_name=self.opponent_name)
|
|
|
|
msg = Message(
|
|
content=rsp,
|
|
role=self.profile,
|
|
cause_by=type(todo),
|
|
sent_from=self.name,
|
|
send_to=self.opponent_name,
|
|
)
|
|
self.rc.memory.add(msg)
|
|
|
|
return msg
|
|
|
|
|
|
async def debate(idea: str, investment: float = 3.0, n_round: int = 5):
|
|
"""Run a team of presidents and watch they quarrel. :)"""
|
|
Biden = Debator(name="Biden", profile="Democrat", opponent_name="Trump")
|
|
Trump = Debator(name="Trump", profile="Republican", opponent_name="Biden")
|
|
team = Team()
|
|
team.hire([Biden, Trump])
|
|
team.invest(investment)
|
|
team.run_project(idea, send_to="Biden") # send debate topic to Biden and let him speak first
|
|
await team.run(n_round=n_round)
|
|
|
|
|
|
def main(idea: str, investment: float = 3.0, n_round: int = 10):
|
|
"""
|
|
:param idea: Debate topic, such as "Topic: The U.S. should commit more in climate change fighting"
|
|
or "Trump: Climate change is a hoax"
|
|
:param investment: contribute a certain dollar amount to watch the debate
|
|
:param n_round: maximum rounds of the debate
|
|
:return:
|
|
"""
|
|
if platform.system() == "Windows":
|
|
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
|
asyncio.run(debate(idea, investment, n_round))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
fire.Fire(main) # run as python debate.py --idea="TOPIC" --investment=3.0 --n_round=5
|