init
This commit is contained in:
101
functionality/plan/main_manual_plan.py
Normal file
101
functionality/plan/main_manual_plan.py
Normal file
@@ -0,0 +1,101 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""Manual specification plan example."""
|
||||
import asyncio
|
||||
import os
|
||||
|
||||
from agentscope.agent import ReActAgent, UserAgent
|
||||
from agentscope.formatter import DashScopeChatFormatter
|
||||
from agentscope.message import Msg
|
||||
from agentscope.model import DashScopeChatModel
|
||||
from agentscope.plan import PlanNotebook, SubTask
|
||||
from agentscope.tool import (
|
||||
Toolkit,
|
||||
execute_python_code,
|
||||
execute_shell_command,
|
||||
insert_text_file,
|
||||
view_text_file,
|
||||
write_text_file,
|
||||
)
|
||||
|
||||
plan_notebook = PlanNotebook()
|
||||
|
||||
|
||||
async def main() -> None:
|
||||
"""The main entry point for the manual plan example."""
|
||||
|
||||
# Create the plan manually
|
||||
await plan_notebook.create_plan(
|
||||
name="Comprehensive Report on AgentScope",
|
||||
description="Study the code of AgentScope and write a comprehensive "
|
||||
"report about this framework.",
|
||||
expected_outcome="A markdown format report summarizing the features, "
|
||||
"architecture, advantages/disadvantages, and "
|
||||
"potential improvements of AgentScope.",
|
||||
subtasks=[
|
||||
SubTask(
|
||||
name="Clone the repository",
|
||||
description="Clone the AgentScope GitHub repository from "
|
||||
"agentscope-ai/agentscope, and ensure it's the "
|
||||
"latest version.",
|
||||
expected_outcome="A local copy of the AgentScope repository.",
|
||||
),
|
||||
SubTask(
|
||||
name="View the documentation",
|
||||
description="View the documentation of AgentScope in the "
|
||||
"repository.",
|
||||
expected_outcome="A comprehensive understanding of the "
|
||||
"features and usage of AgentScope.",
|
||||
),
|
||||
SubTask(
|
||||
name="Study the code",
|
||||
description="Study the code of AgentScope, focusing on the "
|
||||
"core modules and their interactions.",
|
||||
expected_outcome="A deep understanding of the architecture "
|
||||
"and implementation of AgentScope.",
|
||||
),
|
||||
SubTask(
|
||||
name="Summarize the findings",
|
||||
description="Summarize the findings from the documentation "
|
||||
"and code study, and write a comprehensive report "
|
||||
"in markdown format.",
|
||||
expected_outcome="A markdown format report",
|
||||
),
|
||||
],
|
||||
)
|
||||
|
||||
# Add basic tools
|
||||
toolkit = Toolkit()
|
||||
toolkit.register_tool_function(execute_shell_command)
|
||||
toolkit.register_tool_function(execute_python_code)
|
||||
toolkit.register_tool_function(write_text_file)
|
||||
toolkit.register_tool_function(insert_text_file)
|
||||
toolkit.register_tool_function(view_text_file)
|
||||
|
||||
# Create the agent
|
||||
agent = ReActAgent(
|
||||
name="Friday",
|
||||
sys_prompt="You're a helpful assistant named Friday. Your target is "
|
||||
"to finish the given task with careful planning.",
|
||||
model=DashScopeChatModel(
|
||||
model_name="qwen3-max-preview",
|
||||
api_key=os.environ["DASHSCOPE_API_KEY"],
|
||||
),
|
||||
formatter=DashScopeChatFormatter(),
|
||||
toolkit=toolkit,
|
||||
plan_notebook=plan_notebook,
|
||||
)
|
||||
user = UserAgent(name="user")
|
||||
|
||||
msg = Msg(
|
||||
"user",
|
||||
"Now start to finish the task by the given plan",
|
||||
"user",
|
||||
)
|
||||
while True:
|
||||
msg = await agent(msg)
|
||||
msg = await user(msg)
|
||||
if msg.get_text_content() == "exit":
|
||||
break
|
||||
|
||||
|
||||
asyncio.run(main())
|
||||
Reference in New Issue
Block a user