O LangFlow permite criar componentes customizados em Python.
- Arraste um Custom Component.
- Edite o código Python.
- Use o código abaixo para conectar ao servidor MCP:
from langflow import CustomComponent
from mcp import ClientSession
from mcp.client.streamable_http import streamablehttp_client
import asyncio
class MCPToolComponent(CustomComponent):
display_name = "APIBrasil MCP"
description = "Executa ferramentas via MCP"
def build_config(self):
return {
"tool_name": {"display_name": "Nome da Tool", "value": "cep_lookup"},
"arguments": {"display_name": "Argumentos (JSON)", "value": "{}"},
}
def build(self, tool_name: str, arguments: dict):
async def run_mcp():
url = "https://mcp.apibrasil.cloud/mcp"
async with streamablehttp_client(url) as (read, write, _):
async with ClientSession(read, write) as session:
await session.initialize()
result = await session.call_tool(tool_name, arguments)
return str(result)
# Nota: LangFlow roda em loop assíncrono, pode precisar de ajustes
return asyncio.run(run_mcp())