Files
agent_contas/scripts/smoke_mcp.py

23 lines
879 B
Python

#!/usr/bin/env python3
import asyncio
from contas_mcp.servers.contas_mcp_server.main import ToolCall, call_tool, health
CASES = [
("consultar_faturas", {"msisdn": "11999999999"}),
("invoice_explanation", {"msisdn": "11999999999"}),
("consultar_vas", {"msisdn": "11999999999"}),
("consultar_historico_vas", {"msisdn": "11999999999"}),
("cancelar_vas_avulso", {"msisdn": "11999999999", "subject": "TIM Fashion Mensal"}),
("contestar_cobranca", {"msisdn": "11999999999", "subject": "Tamboro Mensal", "valor": 14.99, "motivo": "não reconheço"}),
]
async def main():
print(await health())
for name, arguments in CASES:
result = await call_tool(ToolCall(tool_name=name, arguments=arguments))
print(name, "OK" if result.get("ok") else "FAIL")
if not result.get("ok"):
raise SystemExit(result)
asyncio.run(main())