from langchain_mcp_adapters.client import MultiServerMCPClient

# （1）配置智谱AI、12306等工具的MCP服务器
mcp_zhipu_websearch_config = {
    "url": "https://open.bigmodel.cn/api/mcp/web_search/sse?Authorization=" + ZHIPU_API_KEY,
    'transport': 'sse'
}

mcp_12306_server_config = {
    "url": "https://mcp.api-inference.modelscope.net/74b96a1ea0e547/sse",
    'transport': 'sse'
}

mcp_chart_server_config = {
    "url": "https://mcp.api-inference.modelscope.net/8f29f2b5aced47/sse",
    'transport': 'sse'
}


#（2.1）创建mcp 实例
mcp_client = MultiServerMCPClient(
    {
    'chart_mcp':mcp_chart_server_config,
    '12306_mcp':mcp_12306_server_config,
     'zhipu_mcp':mcp_zhipu_websearch_config,
     })

#(2.2) 另外一种mcp方式

mcp_client = MultiServerMCPClient(
    servers={
        'chart_service': {  
            "url": "https://mcp.api-inference.modelscope.net/8f29f2b5aced47/sse",
            'transport': 'sse'},
        'search_service': {  
            "url": "https://open.bigmodel.cn/api/mcp/web_search/sse?Authorization=" + ZHIPU_API_KEY,
            'transport': 'sse'}      
    }
)


#（3）异步获取mcp工具和绑定到llm
tools = await  mcp_client.get_tools()
llm_with_tools = llm.bind_tools(tools)