LlamaIndex Workflows

pip install llama-index-tools-mcp

from llama_index.tools.mcp import BasicMCPClient, McpToolSpec

mcp_client = BasicMCPClient(
    "https://playground.sanfoundation.com/mcp",
    headers={"x-api-key": "sk_..."},
)

tools = await McpToolSpec(client=mcp_client).to_tool_list_async()

import asyncio
from llama_index.core.agent.workflow import FunctionAgent
from llama_index.llms.openai import OpenAI
from llama_index.tools.mcp import BasicMCPClient, McpToolSpec

async def main() -> None:
    mcp_client = BasicMCPClient(
        "https://playground.sanfoundation.com/mcp",
        headers={"x-api-key": "sk_..."},
    )
    tools = await McpToolSpec(client=mcp_client).to_tool_list_async()

    agent = FunctionAgent(
        tools=tools,
        llm=OpenAI(model="gpt-4o"),
        system_prompt="Use the SAN tools to answer the question.",
    )
    print(await agent.run("Which SAN agents are available?"))

asyncio.run(main())