mcp_server_demo.py

from dotenv import load_dotenv
from langchain_mcp_adapters.client import MultiServerMCPClient
from langgraph.prebuilt import create_react_agent

load_dotenv()


async def main():
    client = MultiServerMCPClient(
        {
            "mcp-server-chart": {
                "command": "npx",
                "args": ["-y", "@antv/mcp-server-chart"],
                "transport": "stdio",
            },
        }
    )
    tools = await client.get_tools()
    agent = create_react_agent("deepseek:deepseek-chat", tools)
    chart_response = await agent.ainvoke(
        {
            "messages": [
                {"role": "user", "content": "Please generate a simple bar chart."
                "There are 3 categories: A, B, and C."
                "The values are 10, 20, and 30 respectively."}
            ]
        }
    )
    print(chart_response)


if __name__ == "__main__":
    import asyncio

    asyncio.run(main())