a64378956a
Some checks failed
Pre-commit / run (ubuntu-latest) (push) Has been cancelled
Deploy Sphinx documentation to Pages / build_en (ubuntu-latest, 3.10) (push) Has been cancelled
Deploy Sphinx documentation to Pages / build_zh (ubuntu-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (macos-15, 3.12) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (ubuntu-latest, 3.12) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.10) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.11) (push) Has been cancelled
Python Unittest Coverage / test (windows-latest, 3.12) (push) Has been cancelled
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""The planner agent example."""
|
|
import asyncio
|
|
import os
|
|
|
|
from tool import create_worker
|
|
|
|
from agentscope.agent import ReActAgent, UserAgent
|
|
from agentscope.formatter import DashScopeChatFormatter
|
|
from agentscope.model import DashScopeChatModel
|
|
from agentscope.plan import PlanNotebook
|
|
from agentscope.tool import Toolkit
|
|
|
|
|
|
async def main() -> None:
|
|
"""The main function."""
|
|
# Connect to the studio for better visualization (optional)
|
|
# import agentscope
|
|
# agentscope.init(
|
|
# project="meta_planner_agent",
|
|
# studio_url="http://localhost:3000",
|
|
# )
|
|
|
|
toolkit = Toolkit()
|
|
toolkit.register_tool_function(create_worker)
|
|
|
|
planner = ReActAgent(
|
|
name="Friday",
|
|
# pylint: disable=C0301
|
|
sys_prompt="""You are Friday, a multifunctional agent that can help people solving different complex tasks. You act like a meta planner to solve complicated tasks by decomposing the task and building/orchestrating different worker agents to finish the sub-tasks.
|
|
|
|
## Core Mission
|
|
Your primary purpose is to break down complicated tasks into manageable subtasks (a plan), create worker agents to finish the subtask, and coordinate their execution to achieve the user's goal efficiently.
|
|
|
|
### Important Constraints
|
|
1. DO NOT TRY TO SOLVE THE SUBTASKS DIRECTLY yourself.
|
|
2. Always follow the plan sequence.
|
|
3. DO NOT finish the plan until all subtasks are finished.
|
|
""", # noqa: E501
|
|
model=DashScopeChatModel(
|
|
model_name="qwen3-max",
|
|
api_key=os.environ["DASHSCOPE_API_KEY"],
|
|
),
|
|
formatter=DashScopeChatFormatter(),
|
|
plan_notebook=PlanNotebook(),
|
|
toolkit=toolkit,
|
|
max_iters=20,
|
|
)
|
|
user = UserAgent(name="user")
|
|
|
|
msg = None
|
|
while True:
|
|
msg = await planner(msg)
|
|
msg = await user(msg)
|
|
if msg.get_text_content() == "exit":
|
|
break
|
|
|
|
|
|
asyncio.run(main())
|