Spaces:
Sleeping
Sleeping
File size: 1,833 Bytes
1c91979 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
import asyncio
import os
import uuid
from dotenv import load_dotenv
from aworld import trace
from examples.debate.agent.debate_agent import DebateAgent
from examples.debate.agent.main import DebateArena
from examples.debate.agent.moderator_agent import ModeratorAgent
from examples.debate.agent.prompts import generate_opinions_prompt
from aworld.config import AgentConfig
from aworld.output import WorkSpace
if __name__ == '__main__':
load_dotenv()
trace.configure()
base_config = {
"llm_provider": "openai",
"llm_model_name": os.environ['LLM_MODEL_NAME'],
"llm_base_url": os.environ['LLM_BASE_URL'],
"llm_api_key": os.environ['LLM_API_KEY'],
}
agentConfig = AgentConfig.model_validate(base_config)
agent1 = DebateAgent(name="affirmativeSpeaker", stance="affirmative", conf=AgentConfig.model_validate(base_config))
agent2 = DebateAgent(name="negativeSpeaker", stance="negative", conf=AgentConfig.model_validate(base_config))
moderator_agent = ModeratorAgent(
conf=AgentConfig.model_validate(base_config | {
"name": "moderator_agent",
"agent_prompt": generate_opinions_prompt
})
)
debate_arena = DebateArena(affirmative_speaker=agent1, negative_speaker=agent2, moderator=moderator_agent,
workspace=WorkSpace.from_local_storages(str(uuid.uuid4())))
async def start_debate(debate_arena, topic, rounds):
speeches = debate_arena.async_run(topic=topic, rounds=rounds)
async for speech in speeches:
if speech.parts:
async for part in speech.parts:
print(part.content, flush=True, end="")
print(f"{speech.name}: {speech.content}")
asyncio.run(start_debate(debate_arena, topic="张居正", rounds=3))
|