init

conversational_agents/multiagent_conversation/main.py

@@ -0,0 +1,80 @@
+# -*- coding: utf-8 -*-
+"""The example of how to construct multi-agent conversation with MsgHub and
+pipeline in AgentScope."""
+import asyncio
+import os
+
+from agentscope.agent import ReActAgent
+from agentscope.formatter import DashScopeMultiAgentFormatter
+from agentscope.message import Msg
+from agentscope.model import DashScopeChatModel
+from agentscope.pipeline import MsgHub, sequential_pipeline
+
+
+def create_participant_agent(
+    name: str,
+    age: int,
+    career: str,
+    character: str,
+) -> ReActAgent:
+    """Create a participant agent with a specific name, age, and character."""
+    return ReActAgent(
+        name=name,
+        sys_prompt=(
+            f"You're a {age}-year-old {career} named {name} and you're "
+            f"a {character} person."
+        ),
+        model=DashScopeChatModel(
+            model_name="qwen-max",
+            api_key=os.environ["DASHSCOPE_API_KEY"],
+            stream=True,
+        ),
+        # Use multiagent formatter because the multiple entities will
+        # occur in the prompt of the LLM API call
+        formatter=DashScopeMultiAgentFormatter(),
+    )
+
+
+async def main() -> None:
+    """Run a multi-agent conversation workflow."""
+
+    # Create multiple participant agents with different characteristics
+    alice = create_participant_agent("Alice", 30, "teacher", "friendly")
+    bob = create_participant_agent("Bob", 14, "student", "rebellious")
+    charlie = create_participant_agent("Charlie", 28, "doctor", "thoughtful")
+
+    # Create a conversation where participants introduce themselves within
+    # a message hub
+    async with MsgHub(
+        participants=[alice, bob, charlie],
+        # The greeting message will be sent to all participants at the start
+        announcement=Msg(
+            "system",
+            "Now you meet each other with a brief self-introduction.",
+            "system",
+        ),
+    ) as hub:
+        # Quick construct a pipeline to run the conversation
+        await sequential_pipeline([alice, bob, charlie])
+        # Or by the following way:
+        # await alice()
+        # await bob()
+        # await charlie()
+
+        # Delete a participant agent from the hub and fake a broadcast message
+        print("##### We fake Bob's departure #####")
+        hub.delete(bob)
+        await hub.broadcast(
+            Msg(
+                "bob",
+                "I have to start my homework now, see you later!",
+                "assistant",
+            ),
+        )
+        await alice()
+        await charlie()
+
+        # ...
+
+
+asyncio.run(main())