from agentscope.agents import ReActAgentV2, UserAgent
from agentscope.service import ServiceToolkit
import agentscope
# 加载模型配置
agentscope.init(
    model_configs={
        "model_config": "my_config",
        "model_type": "dashscope_chat",
        "model_name": "qwen-max",
    }
)
# 初始化 toolkit，并连接 MCP 服务器
toolkit = ServiceToolkit()
toolkit.add_mcp_servers(
    {
        "mcpServers": {
            "amap-amap-sse": {
                "url": "https://mcp.amap.com/sse?key=xxx"
            }
        },
    }
)
# 创建 ReAct 智能体
agent = ReActAgentV2(
    name="Friday",
    model_config_name="my_config",
    service_toolkit=toolkit,
    max_iters=20,
    sys_prompt="""You're a helpful assistant named {{name}}. 
# Target
Your target is to finish the given task with the provided tools."""
)
user_agent = UserAgent(name="user")
# 显式地构建工作流程/对话
x = None
while True:
    x = agent(x)
    x = user_agent(x)
    if x.content == "exit":
        break

"""You're a helpful assistant named {{name}}. 
# Target
Your target is to finish the given task with the provided tools.
# Note
## When Using Gaode API Tools
1. The input locations maybe not specific, so you MUST first use the `maps_text_search` function to get the complete and accurate address. After that, you can use the `maps_geo` function to get the latitude and longitude of the location
2. DON'T make any assumptions! All the locations (include latitude and longitude) you use should be obtained from the tools
3. Sometimes, there maybe multiple locations for the same name, once you feel the search result is not accurate, you should research with different keywords or use `generate_response` function to ask for more information
4. You can use Gaode API tools to obtain POI related pictures"""

① 先调用 maps_text_search 函数，将模糊的地点描述转化为详细的结构化地址
② 再调用 maps_geo 函数，获取该地点的经纬度坐标
③ 最后调用 maps_around_search 函数，搜索出 radius 半径范围的 POI 地点信息

# ...
class ReActAgentV2(AgentBase):
    # ...
    def reset_equipped_tools(
        self,
        gaode_map: bool = False,        # 高德 MCP Server
        webpage_fetch: bool = False,    # 网页内容抓取 MCP Server
        webpage_deploy: bool = False,   # 网页部署 MCP Server
        multimodal: bool = False,       # 多模态 MCP Server
    ) -> ServiceResponse:
        """Reset your equipped tool functions. Use this function when your current tools are not enough to solve the task. Note this function is not incremental, it will first disable all the current tools, and then enable the new tools.
        Args:
            gaode_map (`bool`, defaults to `False`):
                The gaode map related tools for geography related tasks
            webpage_fetch (`bool`, defaults to `False`):
                The webpage fetching tools
            webpage_deploy (`bool`, defaults to `False`):
                The webpage deploying tools, which can be used to deploy the given webpage
            multimodal (`bool`, defaults to `False`):
                The multimodal tools, including text to image/audio tools
        """
        # Clear the current tools
        self.service_toolkit.service_funcs = {}
        self.service_toolkit.add(self.reset_equipped_tools)
        self.service_toolkit.add(self.generate_response)
        selected_mcps = {}
        notes = []
        if gaode_map:
            selected_mcps = {
                **selected_mcps,
                **MCP_CONFIGS[MCP_NAMES.GAODE_MAP]
            }
            notes.append(MCP_NOTES[MCP_NAMES.GAODE_MAP])
        if webpage_fetch:
            selected_mcps = {
                **selected_mcps,
                **MCP_CONFIGS[MCP_NAMES.WEBPAGE_FETCH]
            }
            notes.append(MCP_NOTES[MCP_NAMES.WEBPAGE_FETCH])
        if webpage_deploy:
            selected_mcps = {
                **selected_mcps,
                **MCP_CONFIGS[MCP_NAMES.WEBPAGE_DEPLOY]
            }
            notes.append(MCP_NOTES[MCP_NAMES.WEBPAGE_DEPLOY])
        if multimodal:
            selected_mcps = {
                **selected_mcps,
                **MCP_CONFIGS[MCP_NAMES.MULTIMODAL]
            }
            notes.append(MCP_NOTES[MCP_NAMES.MULTIMODAL])
        self.service_toolkit.add_mcp_servers({"mcpServers": selected_mcps})
        # Delete useless tools
        if "fetch_html" in self.service_toolkit.service_funcs:
            self.service_toolkit.service_funcs.pop("fetch_html")
        if "fetch_txt" in self.service_toolkit.service_funcs:
            self.service_toolkit.service_funcs.pop("fetch_txt")
        # Reset the system prompt
        self.sys_prompt = "\n\n".join([SYSTEM_PROMPT_BASE] + notes).format(
            name=self.name)
        print("重置之后", self.sys_prompt)
        return ServiceResponse(
            status=ServiceExecStatus.SUCCESS,
            content="Success: Tools reset successfully",
        )

① 根据参数改变当前 ServiceToolkit 对象中的工具
② 根据参数动态改变智能体的系统提示，从而指导智能体使用不同的 MCP Server