Projeto do Agent Contas ORACLE
This commit is contained in:
780
contas_mcp/servers/contas_mcp_server/main.py
Normal file
780
contas_mcp/servers/contas_mcp_server/main.py
Normal file
@@ -0,0 +1,780 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import sys
|
||||
from pathlib import Path
|
||||
from typing import Any, Callable
|
||||
|
||||
from dotenv import load_dotenv
|
||||
from fastapi import FastAPI
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
PROJECT_ROOT = Path(__file__).resolve().parents[3]
|
||||
FRAMEWORK_SRC = PROJECT_ROOT / "agent_framework_oci" / "libs" / "agent_framework" / "src"
|
||||
for entry in (PROJECT_ROOT, FRAMEWORK_SRC):
|
||||
if str(entry) not in sys.path:
|
||||
sys.path.insert(0, str(entry))
|
||||
load_dotenv(PROJECT_ROOT / ".env", override=False)
|
||||
|
||||
from agent_framework.checkpoints.langgraph_saver import create_langgraph_checkpointer
|
||||
from agent_framework.idempotency import create_idempotency_store
|
||||
from agent_framework.cache.cache import create_cache
|
||||
from agent_framework.config.settings import get_settings
|
||||
from agent_framework.workflows import FileWorkflowRepository, WorkflowRuntime
|
||||
from app.domain.contas import ContasDomainService
|
||||
from app.domain.contas.workflow_actions import build_contas_workflow_actions
|
||||
from app.domain.contas.invoice_resolver import InvoiceResolver
|
||||
from app.domain.contas.invoice_context import InvoiceContextService
|
||||
from app.domain.contas.item_matcher import SimilarityItemMatcher
|
||||
from app.domain.contas.vas_cancellation_message import compose_vas_cancellation_message
|
||||
|
||||
app = FastAPI(title="TIM Contas MCP - Framework Native")
|
||||
service = ContasDomainService()
|
||||
invoice_resolver = InvoiceResolver(matcher=SimilarityItemMatcher())
|
||||
settings = get_settings()
|
||||
_workflow_runtime: WorkflowRuntime | None = None
|
||||
_invoice_context_service: InvoiceContextService | None = None
|
||||
|
||||
|
||||
def get_invoice_context_service() -> InvoiceContextService:
|
||||
global _invoice_context_service
|
||||
if _invoice_context_service is None:
|
||||
_invoice_context_service = InvoiceContextService(service.client, create_cache(settings))
|
||||
return _invoice_context_service
|
||||
|
||||
|
||||
def get_workflow_runtime() -> WorkflowRuntime:
|
||||
"""Inicializa WorkflowRuntime/checkpointer/idempotência somente no primeiro uso.
|
||||
|
||||
Evita abrir Oracle/Redis durante import, health ou tools/list e garante que
|
||||
transações usem o IdempotencyStore selecionado pelo framework, não memória local.
|
||||
"""
|
||||
global _workflow_runtime
|
||||
if _workflow_runtime is None:
|
||||
idempotency_store = create_idempotency_store(
|
||||
settings, namespace="contas", require_durable=False
|
||||
)
|
||||
_workflow_runtime = WorkflowRuntime(
|
||||
FileWorkflowRepository(PROJECT_ROOT / "workflows"),
|
||||
actions=build_contas_workflow_actions(
|
||||
service, idempotency_store=idempotency_store
|
||||
),
|
||||
checkpointer=create_langgraph_checkpointer(settings),
|
||||
)
|
||||
return _workflow_runtime
|
||||
|
||||
|
||||
class ToolCall(BaseModel):
|
||||
tool_name: str
|
||||
arguments: dict[str, Any] = Field(default_factory=dict)
|
||||
|
||||
|
||||
TOOLS: dict[str, dict[str, Any]] = {
|
||||
"consultar_faturas": {"description": "Consulta faturas do cliente.", "input_schema": {"msisdn": "string"}},
|
||||
"invoice_explanation": {"description": "Executa o workflow de explicação de fatura com pause/resume pelo WorkflowRuntime do framework.", "input_schema": {"msisdn": "string"}},
|
||||
"consultar_vas": {"description": "Consulta VAS ativos.", "input_schema": {"msisdn": "string"}},
|
||||
"consultar_historico_vas": {"description": "Consulta histórico de VAS.", "input_schema": {"msisdn": "string"}},
|
||||
"cancelar_vas_avulso": {"description": "Executa workflow de cancelamento VAS após confirmação transacional do framework.", "input_schema": {"msisdn": "string", "subject": "string", "items": "array"}},
|
||||
"tratar_vas_estrategico": {"description": "Executa workflow conversacional de VAS estratégico/bundle.", "input_schema": {"msisdn": "string", "subject": "string", "items": "array"}},
|
||||
"contestar_cobranca": {"description": "Executa workflow completo de contestação/Conta Certa.", "input_schema": {"msisdn": "string", "subject": "string", "valor": "number"}},
|
||||
"pro_rata": {"description": "Executa workflow conversacional de pró-rata.", "input_schema": {"msisdn": "string", "planos": "array", "has_plano_controle": "boolean"}},
|
||||
"termino_desconto": {"description": "Formata capability de término de desconto.", "input_schema": {"msisdn": "string", "nome_plano": "string"}},
|
||||
"valor_divergente": {"description": "Executa capability de valor divergente.", "input_schema": {"msisdn": "string"}},
|
||||
"retomar_workflow": {"description": "Retoma um workflow pausado pelo mesmo execution_id.", "input_schema": {"workflow_name": "string", "execution_id": "string", "resposta_usuario": "string"}},
|
||||
"consultar_status_solicitacao": {"description": "Consulta/atualiza status técnico de solicitação/protocolo TIM.", "input_schema": {"msisdn": "string", "protocol": "string"}},
|
||||
"enviar_sms": {"description": "Envia SMS por integração TIM.", "input_schema": {"msisdn": "string", "message": "string"}},
|
||||
"recuperar_fatura_pdf": {"description": "Recupera SecurePDF usando contrato criptografado TIM.", "input_schema": {"msisdn": "string", "invoice_id": "string", "customer_id": "string"}},
|
||||
"finalizar_atendimento": {"description": "Executa workflow de finalização e seus efeitos de domínio.", "input_schema": {"status": "string", "summary": "string", "msisdn": "string"}},
|
||||
}
|
||||
|
||||
|
||||
def _require(args: dict[str, Any], *names: str) -> None:
|
||||
missing = [n for n in names if args.get(n) in (None, "")]
|
||||
if missing:
|
||||
raise ValueError("Parâmetros obrigatórios ausentes: " + ", ".join(missing))
|
||||
|
||||
|
||||
|
||||
|
||||
def _clarification_option(item: Any) -> dict[str, Any]:
|
||||
name = str(getattr(item, "canonical_name", "") or "").strip()
|
||||
msisdn = str(getattr(item, "msisdn", "") or "").strip()
|
||||
charge_date = str(getattr(item, "charge_date", "") or "").strip()
|
||||
value = getattr(item, "value", None)
|
||||
details = []
|
||||
if msisdn:
|
||||
details.append(f"linha final {msisdn[-4:]}")
|
||||
if charge_date:
|
||||
details.append(f"data {charge_date}")
|
||||
if value is not None:
|
||||
details.append(f"R$ {value}")
|
||||
label = name + ((" — " + ", ".join(details)) if details else "")
|
||||
return {"label": label, "value": name, "msisdn": msisdn, "charge_date": charge_date}
|
||||
|
||||
|
||||
def _preflight_subject(name: str, args: dict[str, Any]) -> dict[str, Any] | None:
|
||||
"""Resolve/desambigua o item contra a evidência da fatura antes do workflow.
|
||||
|
||||
A regra é de domínio; a persistência e a retomada da clarificação ficam no
|
||||
AgentRuntimeMixin. Quando o cliente já escolheu uma opção, não reabre o gate.
|
||||
"""
|
||||
if name not in {"cancelar_vas_avulso", "tratar_vas_estrategico", "contestar_cobranca"}:
|
||||
return None
|
||||
if bool(args.get("clarification_resolved")):
|
||||
return None
|
||||
msisdn = str(args.get("msisdn") or "").strip()
|
||||
subject = str(args.get("subject") or "").strip()
|
||||
if not msisdn or not subject:
|
||||
return None
|
||||
try:
|
||||
invoice_detail = args.get("billing_analysis")
|
||||
if not isinstance(invoice_detail, dict):
|
||||
invoice_detail = service.client.billing_analysis(msisdn)
|
||||
if not isinstance(invoice_detail, dict):
|
||||
return None
|
||||
outcome = invoice_resolver.resolve([subject], invoice_detail)
|
||||
except Exception:
|
||||
# Fail-safe de disponibilidade: a própria operação ainda valida o serviço
|
||||
# no backend TIM. O gate de similaridade nunca transforma falha de leitura
|
||||
# em autorização para um item diferente.
|
||||
return None
|
||||
if outcome.ambiguous:
|
||||
matches = outcome.ambiguous[0].matches
|
||||
options = [_clarification_option(item) for item in matches]
|
||||
return {
|
||||
"status": "NEEDS_CLARIFICATION",
|
||||
"parameter": "subject",
|
||||
"question": f"Encontrei mais de uma cobrança parecida com '{subject}'. Qual delas você quis dizer?",
|
||||
"options": options,
|
||||
}
|
||||
if len(outcome.resolved) == 1:
|
||||
resolved = outcome.resolved[0]
|
||||
args["subject"] = resolved.canonical_name
|
||||
if resolved.msisdn:
|
||||
args.setdefault("item_msisdn", resolved.msisdn)
|
||||
if resolved.charge_date:
|
||||
args.setdefault("charge_date", resolved.charge_date)
|
||||
if getattr(resolved, "value", None) is not None:
|
||||
args.setdefault("resolved_value", str(resolved.value))
|
||||
# Paridade do backend original: se o cliente pediu cancelamento avulso,
|
||||
# mas a própria fatura classifica o item como estratégico/bundle, não
|
||||
# executamos a operação errada. A resolução determinística do domínio
|
||||
# redireciona a mesma solicitação para o workflow VAS estratégico.
|
||||
if name == "cancelar_vas_avulso" and resolved.tool_category == "vas_estrategico":
|
||||
args["_domain_redirect"] = "tratar_vas_estrategico"
|
||||
args["type"] = resolved.item_type
|
||||
args["items"] = [{
|
||||
"type": resolved.item_type,
|
||||
"msisdn": resolved.msisdn or msisdn,
|
||||
"name": resolved.canonical_name,
|
||||
}]
|
||||
if outcome.out_of_scope and not outcome.resolved:
|
||||
return {
|
||||
"status": "OUT_OF_SCOPE",
|
||||
"success": False,
|
||||
"error": f"O item '{subject}' existe na fatura, mas não pertence a uma categoria tratável por esta operação.",
|
||||
}
|
||||
return None
|
||||
|
||||
async def _enrich_invoice_context(name: str, args: dict[str, Any]) -> None:
|
||||
if name not in {"invoice_explanation", "cancelar_vas_avulso", "tratar_vas_estrategico", "contestar_cobranca"}:
|
||||
return
|
||||
msisdn = str(args.get("msisdn") or "").strip()
|
||||
if not msisdn:
|
||||
return
|
||||
# If all required evidence is already present, preserve it exactly.
|
||||
if isinstance(args.get("complete_invoices_payload"), dict) and isinstance(args.get("billing_analysis"), dict):
|
||||
return
|
||||
session_id = str(args.get("session_id") or args.get("original_session_id") or args.get("conversation_key") or "").strip()
|
||||
invoice_id = str(args.get("invoice_id") or args.get("current_invoice_number") or "").strip()
|
||||
include_detail = name in {"cancelar_vas_avulso", "contestar_cobranca"} and not isinstance(args.get("invoice_detail"), dict)
|
||||
try:
|
||||
ctx = await get_invoice_context_service().get(
|
||||
session_id=session_id,
|
||||
msisdn=msisdn,
|
||||
invoice_id=invoice_id,
|
||||
use_cache=bool(args.get("use_invoice_context_cache", True)),
|
||||
include_detail=include_detail,
|
||||
message_id=str(args.get("message_id") or args.get("interaction_key") or ""),
|
||||
)
|
||||
except Exception:
|
||||
return
|
||||
values = ctx.as_dict()
|
||||
if ctx.business_events:
|
||||
args["_prefetch_business_events"] = list(ctx.business_events)
|
||||
for key, value in values.items():
|
||||
if value not in (None, ""):
|
||||
args.setdefault(key, value)
|
||||
if ctx.invoice_id:
|
||||
args.setdefault("invoice_id", ctx.invoice_id)
|
||||
args.setdefault("current_invoice_number", ctx.invoice_id)
|
||||
if ctx.customer_id:
|
||||
args.setdefault("customer_id", ctx.customer_id)
|
||||
|
||||
|
||||
def _with_prefetch_events(result: Any, args: dict[str, Any]) -> Any:
|
||||
events = args.get("_prefetch_business_events") if isinstance(args.get("_prefetch_business_events"), list) else []
|
||||
if not events or not isinstance(result, dict):
|
||||
return result
|
||||
merged = dict(result)
|
||||
existing = merged.get("business_events") if isinstance(merged.get("business_events"), list) else []
|
||||
merged["business_events"] = [*events, *existing]
|
||||
return merged
|
||||
|
||||
|
||||
def _workflow_payload(name: str, args: dict[str, Any]) -> dict[str, Any]:
|
||||
payload = dict(args)
|
||||
msisdn = str(args.get("msisdn") or "")
|
||||
subject = str(args.get("subject") or "")
|
||||
if name == "cancelamento_vas_avulso" and not isinstance(payload.get("items"), list):
|
||||
payload["items"] = [{"msisdn": msisdn, "name": subject, "type": "avulso"}]
|
||||
elif name == "vas_estrategico" and not isinstance(payload.get("items"), list):
|
||||
payload["items"] = [{"msisdn": msisdn, "name": subject, "type": str(args.get("type") or "estrategico")}]
|
||||
elif name == "contestacao_tool":
|
||||
# Paridade do backend original: o workflow usa tipo_atendimento para
|
||||
# selecionar branches/contratos de contestação. O wrapper antigo sempre
|
||||
# fixava esse valor quando a tool contestar_cobranca era chamada.
|
||||
payload.setdefault("tipo_atendimento", "contestacao")
|
||||
payload.setdefault("servico", subject)
|
||||
payload.setdefault("descricao", args.get("motivo") or "")
|
||||
payload.setdefault("items", [{"itemName": subject, "claimedAmount": args.get("valor"), "validatedAmount": args.get("valor")}])
|
||||
return payload
|
||||
|
||||
|
||||
def _result_payload(result: Any, *, workflow_name: str) -> dict[str, Any]:
|
||||
data = result.model_dump() if hasattr(result, "model_dump") else dict(result)
|
||||
trace = data.get("trace") if isinstance(data.get("trace"), list) else []
|
||||
last_node = ""
|
||||
for row in reversed(trace):
|
||||
if isinstance(row, dict) and row.get("node"):
|
||||
last_node = str(row.get("node"))
|
||||
break
|
||||
if not last_node and isinstance(data.get("state"), dict):
|
||||
last_node = str(data["state"].get("current_node") or "")
|
||||
metadata = {
|
||||
"workflow_name": workflow_name,
|
||||
"workflow_execution_id": data.get("execution_id"),
|
||||
"workflow_status": data.get("status"),
|
||||
"workflow_last_node": last_node or None,
|
||||
"resume_tool": "retomar_workflow" if data.get("status") == "PAUSED" else None,
|
||||
}
|
||||
payload = {**data, "metadata": metadata}
|
||||
# Compatibilidade funcional do antigo backend, agora derivada apenas do
|
||||
# branch determinístico do WorkflowRuntime do framework.
|
||||
if data.get("status") == "COMPLETED":
|
||||
if workflow_name == "invoice_explanation":
|
||||
if last_node == "registrar_protocolo_aceite":
|
||||
payload["recomenda_finalizacao"] = True
|
||||
payload["status_finalizacao_sugerido"] = "resolvido"
|
||||
elif last_node == "finalizar_nao_resolvido":
|
||||
payload["recomenda_finalizacao"] = True
|
||||
payload["status_finalizacao_sugerido"] = "nao_resolvido"
|
||||
elif last_node == "resposta_falha_servico":
|
||||
payload["success"] = False
|
||||
payload["service_failed"] = True
|
||||
payload["auto_finalize_on_failure"] = False
|
||||
payload["mensagem"] = "Para continuar com a sua solicitação, aguarde um instante."
|
||||
elif workflow_name == "pro_rata" and last_node in {"registrar_aceitou", "registrar_nao_controle"}:
|
||||
payload["recomenda_finalizacao"] = True
|
||||
payload["status_finalizacao_sugerido"] = "resolvido"
|
||||
return payload
|
||||
|
||||
|
||||
async def _run_workflow(workflow_name: str, args: dict[str, Any]) -> dict[str, Any]:
|
||||
result = await get_workflow_runtime().arun(
|
||||
workflow_name,
|
||||
_workflow_payload(workflow_name, args),
|
||||
execution_id=args.get("workflow_execution_id"),
|
||||
)
|
||||
payload = _result_payload(result, workflow_name=workflow_name)
|
||||
if workflow_name == "contestacao_tool" and payload.get("status") == "FAILED":
|
||||
details = payload.get("error_details") if isinstance(payload.get("error_details"), dict) else {}
|
||||
body = details.get("body")
|
||||
provider_message = ""
|
||||
if isinstance(body, dict):
|
||||
provider = body.get("provider") if isinstance(body.get("provider"), dict) else {}
|
||||
provider_message = str(provider.get("errorMessage") or body.get("description") or "").strip()
|
||||
protocol_node = (payload.get("output") or {}).get("registrar_protocolo") if isinstance(payload.get("output"), dict) else {}
|
||||
protocol = str((protocol_node or {}).get("protocolo_id") or (protocol_node or {}).get("protocol_number") or "") if isinstance(protocol_node, dict) else ""
|
||||
message = provider_message or str(payload.get("error") or "Falha ao executar contestação")
|
||||
payload["success"] = False
|
||||
payload["mensagem"] = message
|
||||
if provider_message:
|
||||
payload["contestation_error_description"] = provider_message
|
||||
if protocol:
|
||||
payload["contestacao_protocol"] = protocol
|
||||
payload["erro_sistemico"] = not bool(provider_message)
|
||||
payload["erro"] = "contestacao_nao_realizada" if provider_message else "erro_falha_sistema"
|
||||
if workflow_name == "contestacao_tool" and payload.get("status") == "COMPLETED":
|
||||
output = payload.get("output") if isinstance(payload.get("output"), dict) else {}
|
||||
contest = output.get("abrir_contestacao_cliente") if isinstance(output.get("abrir_contestacao_cliente"), dict) else {}
|
||||
if contest and contest.get("success") is False:
|
||||
reason = str(contest.get("guardrail_reason") or contest.get("contestation_error_description") or contest.get("error") or "Contestação não executada")
|
||||
payload["success"] = False
|
||||
payload["error"] = reason
|
||||
payload["erro"] = "erro_falha_sistema" if contest.get("blocked") else "contestacao_nao_realizada"
|
||||
payload["erro_sistemico"] = bool(contest.get("blocked"))
|
||||
payload["mensagem"] = reason
|
||||
if contest.get("contestation_error_description"):
|
||||
payload["contestation_error_description"] = contest.get("contestation_error_description")
|
||||
return _with_prefetch_events(payload, args)
|
||||
|
||||
|
||||
|
||||
|
||||
def _workflow_result_dict(result: Any) -> dict[str, Any]:
|
||||
return result.model_dump() if hasattr(result, "model_dump") else dict(result)
|
||||
|
||||
|
||||
def _node_output(result: Any, node_id: str) -> dict[str, Any]:
|
||||
data = _workflow_result_dict(result)
|
||||
output = data.get("output") if isinstance(data.get("output"), dict) else {}
|
||||
node = output.get(node_id)
|
||||
return dict(node) if isinstance(node, dict) else {}
|
||||
|
||||
|
||||
def _first_invoice_context(invoices: Any) -> tuple[dict[str, Any], dict[str, Any]]:
|
||||
if not isinstance(invoices, dict):
|
||||
return {}, {}
|
||||
billing = invoices.get("billingProfile") or invoices.get("billing_profile") or {}
|
||||
customer = billing.get("customer") if isinstance(billing, dict) and isinstance(billing.get("customer"), dict) else {}
|
||||
items = invoices.get("paymentItems") or invoices.get("payment_items") or []
|
||||
invoice = next((x for x in items if isinstance(x, dict)), {}) if isinstance(items, list) else {}
|
||||
return dict(customer), dict(invoice)
|
||||
|
||||
|
||||
def _value_for_subject(args: dict[str, Any], subject: str) -> Any:
|
||||
for item in args.get("items") or []:
|
||||
if not isinstance(item, dict):
|
||||
continue
|
||||
name = str(item.get("name") or item.get("service") or item.get("subject") or item.get("desc") or "").strip()
|
||||
if name.casefold() == subject.casefold():
|
||||
return item.get("value") if item.get("value") is not None else item.get("valor")
|
||||
if str(args.get("subject") or "").strip().casefold() == subject.casefold():
|
||||
return args.get("resolved_value") if args.get("resolved_value") is not None else args.get("valor")
|
||||
return None
|
||||
|
||||
|
||||
|
||||
|
||||
def _invoice_detail_msisdn(invoice_detail: Any, subject: str) -> str:
|
||||
"""Resolve the billed line for an item from invoice detail.
|
||||
|
||||
The legacy behavior used the invoice evidence to fix LLM/tool arguments that
|
||||
accidentally attached a dependent item to the holder line. This remains a
|
||||
pure deterministic domain normalization before the workflows are invoked.
|
||||
"""
|
||||
needle = str(subject or "").strip().casefold()
|
||||
if not needle or not isinstance(invoice_detail, dict):
|
||||
return ""
|
||||
for line_key, sections in invoice_detail.items():
|
||||
if not isinstance(sections, dict):
|
||||
continue
|
||||
for rows in sections.values():
|
||||
if not isinstance(rows, list):
|
||||
continue
|
||||
for row in rows:
|
||||
if not isinstance(row, dict):
|
||||
continue
|
||||
desc = str(row.get("desc") or row.get("name") or row.get("description") or "").strip().casefold()
|
||||
if desc == needle:
|
||||
return str(row.get("msisdn") or line_key or "").strip()
|
||||
return ""
|
||||
|
||||
|
||||
def _normalize_cancel_items(args: dict[str, Any], holder_msisdn: str) -> list[dict[str, Any]]:
|
||||
items = [dict(x) for x in (args.get("items") or []) if isinstance(x, dict)]
|
||||
if not items and args.get("subject"):
|
||||
items = [{
|
||||
"msisdn": args.get("item_msisdn") or holder_msisdn,
|
||||
"name": args.get("subject"),
|
||||
"value": args.get("resolved_value") if args.get("resolved_value") is not None else args.get("valor"),
|
||||
}]
|
||||
invoice_detail = args.get("invoice_detail")
|
||||
for item in items:
|
||||
subject = str(item.get("name") or item.get("service") or item.get("subject") or item.get("desc") or "").strip()
|
||||
resolved_line = _invoice_detail_msisdn(invoice_detail, subject)
|
||||
if resolved_line:
|
||||
item["msisdn"] = resolved_line
|
||||
elif not item.get("msisdn"):
|
||||
item["msisdn"] = holder_msisdn
|
||||
return items
|
||||
|
||||
def _digits_only(value: Any) -> str:
|
||||
return "".join(ch for ch in str(value or "") if ch.isdigit())
|
||||
|
||||
|
||||
def _money_ptbr(value: Any) -> str:
|
||||
text = str(value if value is not None else "0").strip()
|
||||
try:
|
||||
if "," in text:
|
||||
numeric = float(text.replace(".", "").replace(",", "."))
|
||||
else:
|
||||
numeric = float(text)
|
||||
except (TypeError, ValueError):
|
||||
return text
|
||||
return f"{numeric:.2f}".replace(".", ",")
|
||||
|
||||
|
||||
def _nonzero_money_or(value: Any, fallback: Any) -> Any:
|
||||
try:
|
||||
text = str(value if value is not None else "").strip()
|
||||
if not text:
|
||||
return fallback
|
||||
numeric = float(text.replace(".", "").replace(",", ".") if "," in text else text)
|
||||
return value if abs(numeric) > 1e-9 else fallback
|
||||
except (TypeError, ValueError):
|
||||
return value if value not in (None, "") else fallback
|
||||
|
||||
|
||||
async def _run_cancelamento_com_contestacao(args: dict[str, Any]) -> dict[str, Any]:
|
||||
# The session/customer MSISDN is always the holder for the contestation.
|
||||
# Individual cancellation items may belong to dependent lines.
|
||||
holder_msisdn = str(args.get("holder_msisdn") or args.get("session_msisdn") or args.get("msisdn") or "")
|
||||
normalized_args = dict(args)
|
||||
normalized_args["msisdn"] = holder_msisdn
|
||||
# Paridade do backend original: CPF é alias aceito para social_sec_no e
|
||||
# deve chegar ao workflow sem máscara.
|
||||
social_sec_no = _digits_only(args.get("social_sec_no") or args.get("cpf"))
|
||||
if social_sec_no:
|
||||
normalized_args["social_sec_no"] = social_sec_no
|
||||
normalized_args["items"] = _normalize_cancel_items(args, holder_msisdn)
|
||||
cancel_result = await get_workflow_runtime().arun(
|
||||
"cancelamento_vas_avulso",
|
||||
_workflow_payload("cancelamento_vas_avulso", normalized_args),
|
||||
execution_id=args.get("workflow_execution_id"),
|
||||
)
|
||||
cancel_data = _workflow_result_dict(cancel_result)
|
||||
if cancel_data.get("status") != "COMPLETED":
|
||||
return _result_payload(cancel_result, workflow_name="cancelamento_vas_avulso")
|
||||
|
||||
cancel_node = _node_output(cancel_result, "cancelar_vas_avulso")
|
||||
candidate_rows = cancel_node.get("contestation_candidates")
|
||||
if not isinstance(candidate_rows, list):
|
||||
candidate_rows = cancel_node.get("itens_para_contestacao")
|
||||
candidates = [x for x in (candidate_rows or []) if isinstance(x, dict)]
|
||||
raw_results = [x for x in (cancel_node.get("results") or []) if isinstance(x, dict)]
|
||||
successful = [x for x in raw_results if x.get("success")]
|
||||
failed = [x for x in raw_results if not x.get("success")]
|
||||
# Compatibilidade do wrapper original: versões/actions podem reportar o
|
||||
# resultado agregado em cancelados/nao_cancelados sem repetir `success` em
|
||||
# results[]. O wrapper deve preservar esse outcome, não transformá-lo em falha.
|
||||
if not successful:
|
||||
successful = [
|
||||
{**dict(x), "subject": x.get("subject") or x.get("servico") or x.get("name"), "success": True}
|
||||
for x in (cancel_node.get("cancelados") or []) if isinstance(x, dict)
|
||||
]
|
||||
if not failed:
|
||||
failed = [
|
||||
{**dict(x), "subject": x.get("subject") or x.get("servico") or x.get("name"), "success": False}
|
||||
for x in (cancel_node.get("nao_cancelados") or []) if isinstance(x, dict)
|
||||
]
|
||||
|
||||
if not candidates:
|
||||
line_protocols = [
|
||||
str(row.get("protocolo_id") or row.get("protocol") or "").strip()
|
||||
for row in (cancel_node.get("protocolos_por_linha") or [])
|
||||
if isinstance(row, dict) and str(row.get("protocolo_id") or row.get("protocol") or "").strip()
|
||||
]
|
||||
message = compose_vas_cancellation_message({
|
||||
"success": bool(successful),
|
||||
"cancelados": [{"servico": x.get("subject")} for x in successful],
|
||||
"nao_encontrados": [{"servico": x.get("subject")} for x in failed if "não encontrado" in str(x.get("error") or "").casefold()],
|
||||
"protocols_for_response": line_protocols,
|
||||
})
|
||||
cancel_data.setdefault("output", {})["resposta_cancelamento"] = {"mensagem": message}
|
||||
cancel_data.update({
|
||||
"success": bool(successful),
|
||||
"mensagem": message,
|
||||
"recomenda_finalizacao": bool(successful),
|
||||
"status_finalizacao_sugerido": "resolvido" if successful else "nao_resolvido",
|
||||
"cancelamento_vas_protocol": line_protocols[0] if line_protocols else None,
|
||||
"protocols_for_response": line_protocols,
|
||||
"requires_protocol_in_response": bool(line_protocols),
|
||||
"auto_finalize_on_failure": False,
|
||||
})
|
||||
cancel_data["metadata"] = {
|
||||
"workflow_name": "cancelamento_vas_avulso",
|
||||
"workflow_execution_id": cancel_data.get("execution_id"),
|
||||
"workflow_status": cancel_data.get("status"),
|
||||
"composite_workflows": ["cancelamento_vas_avulso"],
|
||||
}
|
||||
return cancel_data
|
||||
|
||||
msisdn = holder_msisdn
|
||||
invoices = args.get("complete_invoices_payload")
|
||||
if not isinstance(invoices, dict):
|
||||
invoices = service.consultar_faturas(msisdn=msisdn)
|
||||
customer, invoice = _first_invoice_context(invoices)
|
||||
items: list[dict[str, Any]] = []
|
||||
total = 0.0
|
||||
for candidate in candidates:
|
||||
subject = str(candidate.get("subject") or (candidate.get("service") or {}).get("name") or "").strip()
|
||||
value = _value_for_subject(normalized_args, subject)
|
||||
try:
|
||||
numeric = float(str(value).replace(".", "").replace(",", ".")) if isinstance(value, str) and "," in value else float(value or 0)
|
||||
except Exception:
|
||||
numeric = 0.0
|
||||
total += numeric
|
||||
money = _money_ptbr(value if value is not None else numeric)
|
||||
items.append({
|
||||
"itemName": subject,
|
||||
"item_name": subject,
|
||||
"claimedAmount": money,
|
||||
"validatedAmount": money,
|
||||
"claimed_amount": money,
|
||||
"validated_amount": money,
|
||||
})
|
||||
|
||||
contest_payload = {
|
||||
**dict(normalized_args),
|
||||
"msisdn": msisdn,
|
||||
"social_sec_no": normalized_args.get("social_sec_no") or _digits_only(customer.get("document") or customer.get("socialSecNo")) or "",
|
||||
"customer_id": args.get("customer_id") or customer.get("customerId") or customer.get("id") or "",
|
||||
"current_invoice_number": args.get("current_invoice_number") or invoice.get("invoiceId") or invoice.get("invoiceNumber") or "",
|
||||
"invoice_id": args.get("invoice_id") or invoice.get("invoiceId") or "",
|
||||
"current_invoice_due_date": args.get("current_invoice_due_date") or invoice.get("dueDate") or "",
|
||||
"complete_invoices_payload": invoices,
|
||||
"servico": items[0]["itemName"] if items else str(args.get("subject") or ""),
|
||||
"valor": args.get("valor") if args.get("valor") is not None else total,
|
||||
"items": items,
|
||||
"item_msisdn": msisdn,
|
||||
"dependent_invoice_item": any(str(x.get("msisdn") or msisdn) != msisdn for x in candidates),
|
||||
}
|
||||
contest_result = await get_workflow_runtime().arun("contestacao_tool", contest_payload)
|
||||
contest_data = _workflow_result_dict(contest_result)
|
||||
contest_node_raw = _node_output(contest_result, "abrir_contestacao_cliente")
|
||||
protocol_node = _node_output(contest_result, "registrar_protocolo")
|
||||
cancellation_protocols = [
|
||||
str(row.get("protocolo_id") or row.get("protocol") or "").strip()
|
||||
for row in (cancel_node.get("protocolos_por_linha") or [])
|
||||
if isinstance(row, dict) and str(row.get("protocolo_id") or row.get("protocol") or "").strip()
|
||||
]
|
||||
partial_protocol = str(
|
||||
protocol_node.get("protocolo_id") or protocol_node.get("protocol_number")
|
||||
or contest_data.get("protocol_number") or ""
|
||||
).strip()
|
||||
if contest_data.get("status") != "COMPLETED" or (contest_node_raw and contest_node_raw.get("success") is False):
|
||||
details = contest_data.get("error_details") if isinstance(contest_data.get("error_details"), dict) else {}
|
||||
provider_body = details.get("body") if isinstance(details, dict) else None
|
||||
provider_text = str(provider_body or "")
|
||||
contest_error = str(
|
||||
contest_node_raw.get("guardrail_reason")
|
||||
or contest_node_raw.get("contestation_error_description")
|
||||
or contest_data.get("contestation_error_description")
|
||||
or contest_data.get("error")
|
||||
or "Contestação não executada"
|
||||
)
|
||||
normalized_error = f"{contest_error} {provider_text}".casefold()
|
||||
already_contested = any(token in normalized_error for token in (
|
||||
"já contestado", "ja contestado", "itens já contestados",
|
||||
"itens ja contestados", "item_ja_contestado", "conflito de itens",
|
||||
))
|
||||
response_protocols = list(dict.fromkeys([*cancellation_protocols, *([partial_protocol] if partial_protocol else [])]))
|
||||
if already_contested and successful:
|
||||
message = compose_vas_cancellation_message({
|
||||
"success": True,
|
||||
"cancelados": [{"servico": x.get("subject")} for x in successful],
|
||||
"contestation_error_description": contest_error,
|
||||
"contestacao_protocol": partial_protocol,
|
||||
"protocols_for_response": response_protocols,
|
||||
})
|
||||
return {
|
||||
**cancel_data,
|
||||
"status": "COMPLETED",
|
||||
"success": True,
|
||||
"error": None,
|
||||
"mensagem": message,
|
||||
"erro_sistemico": False,
|
||||
"auto_finalize_on_failure": False,
|
||||
"protocol_closed": False,
|
||||
"contestacao_protocol": partial_protocol or None,
|
||||
"cancelamento_vas_protocol": cancellation_protocols[0] if cancellation_protocols else None,
|
||||
"protocols_for_response": response_protocols,
|
||||
"requires_protocol_in_response": bool(response_protocols),
|
||||
"contestation_error_description": contest_error,
|
||||
"contestacao_workflow": contest_data,
|
||||
"metadata": {
|
||||
"workflow_name": "cancelamento_vas_avulso",
|
||||
"workflow_execution_id": cancel_data.get("execution_id"),
|
||||
"workflow_status": "COMPLETED",
|
||||
"composite_workflows": ["cancelamento_vas_avulso", "contestacao_tool"],
|
||||
},
|
||||
}
|
||||
return {
|
||||
**cancel_data,
|
||||
"status": contest_data.get("status") or "FAILED",
|
||||
"error": contest_error,
|
||||
"success": False,
|
||||
"mensagem": contest_error,
|
||||
"erro_sistemico": True,
|
||||
"auto_finalize_on_failure": True,
|
||||
"contestacao_protocol": partial_protocol or None,
|
||||
"cancelamento_vas_protocol": cancellation_protocols[0] if cancellation_protocols else None,
|
||||
"protocols_for_response": response_protocols,
|
||||
"requires_protocol_in_response": bool(response_protocols),
|
||||
"contestation_error_description": contest_node_raw.get("contestation_error_description") or contest_data.get("contestation_error_description") or None,
|
||||
"contestacao_workflow": contest_data,
|
||||
"metadata": {
|
||||
"workflow_name": "cancelamento_vas_avulso",
|
||||
"workflow_execution_id": cancel_data.get("execution_id"),
|
||||
"workflow_status": contest_data.get("status"),
|
||||
"composite_workflows": ["cancelamento_vas_avulso", "contestacao_tool"],
|
||||
},
|
||||
}
|
||||
|
||||
contest_node = _node_output(contest_result, "abrir_contestacao_cliente")
|
||||
sms_node = _node_output(contest_result, "enviar_sms")
|
||||
protocol = str(protocol_node.get("protocolo_id") or protocol_node.get("protocol_number") or "")
|
||||
response_protocols = list(dict.fromkeys([*cancellation_protocols, *([protocol] if protocol else [])]))
|
||||
items_response = contest_node.get("items_response") if isinstance(contest_node.get("items_response"), list) else []
|
||||
normalized_contested = contest_node.get("contested_items")
|
||||
normalized_not_contested = contest_node.get("not_contested_items")
|
||||
contested_items = [x for x in normalized_contested if isinstance(x, dict)] if isinstance(normalized_contested, list) else [x for x in items_response if isinstance(x, dict) and str(x.get("status") or "").upper() in {"CRIAR", "INICIADA", "ENVIADA", "SUCCESS", "SUCESSO"}]
|
||||
not_contested_items = [x for x in normalized_not_contested if isinstance(x, dict)] if isinstance(normalized_not_contested, list) else [x for x in items_response if isinstance(x, dict) and x not in contested_items]
|
||||
already_contested = contest_node.get("itens_ja_contestados")
|
||||
if not isinstance(already_contested, list):
|
||||
already_contested = contest_node.get("already_contested_items")
|
||||
if not isinstance(already_contested, list):
|
||||
already_contested = contest_data.get("itens_ja_contestados")
|
||||
if not isinstance(already_contested, list):
|
||||
already_contested = []
|
||||
# Ausência de itemsResponse só implica usar os itens solicitados como
|
||||
# contestados quando o provider também não informou explicitamente que eles
|
||||
# já haviam sido contestados. Essa distinção evita afirmar um novo crédito
|
||||
# quando a operação foi recusada por conflito de negócio.
|
||||
if not items_response and not already_contested:
|
||||
contested_items = items
|
||||
|
||||
message = compose_vas_cancellation_message({
|
||||
"success": True,
|
||||
"cancelados": [{"servico": x.get("subject")} for x in successful],
|
||||
"nao_encontrados": [{"servico": x.get("subject")} for x in failed if "não encontrado" in str(x.get("error") or "").casefold()],
|
||||
"contested_items": contested_items,
|
||||
"not_contested_items": not_contested_items,
|
||||
"itens_ja_contestados": already_contested,
|
||||
"contested_invoice_amount_open": _money_ptbr(_nonzero_money_or(contest_node.get("contested_invoice_amount_open"), total)),
|
||||
"sms_sent": bool(sms_node),
|
||||
"sms_not_send_error": bool(sms_node and not sms_node.get("success", True)),
|
||||
"contestacao_protocol": protocol,
|
||||
"contestation_error_description": contest_node.get("contestation_error_description") or "",
|
||||
"protocols_for_response": response_protocols,
|
||||
})
|
||||
|
||||
combined_output = dict(cancel_data.get("output") or {})
|
||||
combined_output["contestacao_tool"] = contest_data.get("output") or {}
|
||||
combined_output["resposta_cancelamento"] = {"mensagem": message}
|
||||
return {
|
||||
**cancel_data,
|
||||
"output": combined_output,
|
||||
"contestacao_workflow": contest_data,
|
||||
"success": bool(successful) or bool(contested_items),
|
||||
"contestacao_protocol": protocol,
|
||||
"cancelamento_vas_protocol": cancellation_protocols[0] if cancellation_protocols else None,
|
||||
"protocols_for_response": response_protocols,
|
||||
"requires_protocol_in_response": bool(response_protocols),
|
||||
"recomenda_finalizacao": bool(successful) or bool(contested_items),
|
||||
"status_finalizacao_sugerido": "resolvido" if (successful or contested_items) else "nao_resolvido",
|
||||
"protocol_closed": bool(contest_node.get("protocol_closed") or protocol_node.get("protocol_closed")),
|
||||
"auto_finalize_on_failure": False,
|
||||
"sms_not_send_error": bool(sms_node and not sms_node.get("success", True)),
|
||||
"contestation_error_description": contest_node.get("contestation_error_description") or "",
|
||||
"mensagem": message,
|
||||
"metadata": {
|
||||
"workflow_name": "cancelamento_vas_avulso",
|
||||
"workflow_execution_id": cancel_data.get("execution_id"),
|
||||
"workflow_status": "COMPLETED",
|
||||
"composite_workflows": ["cancelamento_vas_avulso", "contestacao_tool"],
|
||||
},
|
||||
}
|
||||
|
||||
async def _invoke(name: str, args: dict[str, Any]) -> Any:
|
||||
requirements = {
|
||||
"consultar_faturas": ("msisdn",),
|
||||
"invoice_explanation": ("msisdn",),
|
||||
"consultar_vas": ("msisdn",),
|
||||
"consultar_historico_vas": ("msisdn",),
|
||||
"cancelar_vas_avulso": ("msisdn", "subject"),
|
||||
"tratar_vas_estrategico": ("msisdn", "subject"),
|
||||
"contestar_cobranca": ("msisdn", "subject", "valor"),
|
||||
"pro_rata": ("msisdn",),
|
||||
"termino_desconto": ("msisdn",),
|
||||
"valor_divergente": ("msisdn",),
|
||||
"enviar_sms": ("msisdn", "message"),
|
||||
"recuperar_fatura_pdf": ("msisdn", "invoice_id"),
|
||||
"retomar_workflow": ("workflow_name", "execution_id", "resposta_usuario"),
|
||||
}
|
||||
_require(args, *requirements.get(name, ()))
|
||||
|
||||
await _enrich_invoice_context(name, args)
|
||||
preflight = _preflight_subject(name, args)
|
||||
if preflight is not None:
|
||||
return _with_prefetch_events(preflight, args)
|
||||
domain_redirect = str(args.pop("_domain_redirect", "") or "")
|
||||
if domain_redirect == "tratar_vas_estrategico":
|
||||
redirected = await _run_workflow("vas_estrategico", args)
|
||||
redirected.setdefault("metadata", {})["domain_redirect_from"] = name
|
||||
redirected["metadata"]["domain_redirect_to"] = domain_redirect
|
||||
return _with_prefetch_events(redirected, args)
|
||||
|
||||
workflow_map = {
|
||||
"invoice_explanation": "invoice_explanation",
|
||||
"cancelar_vas_avulso": "cancelamento_vas_avulso",
|
||||
"tratar_vas_estrategico": "vas_estrategico",
|
||||
"contestar_cobranca": "contestacao_tool",
|
||||
"pro_rata": "pro_rata",
|
||||
"termino_desconto": "termino_desconto",
|
||||
"valor_divergente": "valor_divergente",
|
||||
"finalizar_atendimento": "finalizar_atendimento",
|
||||
}
|
||||
if name == "cancelar_vas_avulso":
|
||||
return _with_prefetch_events(await _run_cancelamento_com_contestacao(args), args)
|
||||
if name in workflow_map:
|
||||
return await _run_workflow(workflow_map[name], args)
|
||||
if name == "retomar_workflow":
|
||||
result = await get_workflow_runtime().aresume(
|
||||
str(args["workflow_name"]),
|
||||
str(args["execution_id"]),
|
||||
{"resposta_usuario": args["resposta_usuario"]},
|
||||
)
|
||||
return _with_prefetch_events(_result_payload(result, workflow_name=str(args["workflow_name"])), args)
|
||||
|
||||
fn: Callable[..., Any] = getattr(service, name)
|
||||
call_args = {k: v for k, v in args.items() if not str(k).startswith("_")}
|
||||
return _with_prefetch_events(fn(**call_args), args)
|
||||
|
||||
|
||||
@app.get("/health")
|
||||
async def health() -> dict[str, Any]:
|
||||
return {
|
||||
"status": "ok",
|
||||
"architecture": "framework-native",
|
||||
"legacy_dependency": False,
|
||||
"workflow_engine": "agent_framework.workflows.WorkflowRuntime",
|
||||
"langgraph_direct_import": False,
|
||||
"checkpoint_provider": getattr(settings, "CHECKPOINT_REPOSITORY_PROVIDER", "memory"),
|
||||
"gateway_mode": "mock" if service.client.mock else "real",
|
||||
"tools": len(TOOLS),
|
||||
"env_file": str(PROJECT_ROOT / ".env"),
|
||||
}
|
||||
|
||||
|
||||
@app.get("/mcp/tools/list")
|
||||
async def list_tools() -> dict[str, Any]:
|
||||
return {"tools": [{"name": name, **definition} for name, definition in TOOLS.items()]}
|
||||
|
||||
|
||||
@app.post("/mcp/tools/call")
|
||||
async def call_tool(call: ToolCall) -> dict[str, Any]:
|
||||
if call.tool_name not in TOOLS:
|
||||
return {"ok": False, "error": f"Tool não encontrada: {call.tool_name}"}
|
||||
try:
|
||||
result = await _invoke(call.tool_name, dict(call.arguments or {}))
|
||||
failed = isinstance(result, dict) and (result.get("success") is False or result.get("status") == "FAILED")
|
||||
metadata = {"server": "contas", "tool": call.tool_name, "framework_native": True}
|
||||
if isinstance(result, dict) and isinstance(result.get("metadata"), dict):
|
||||
metadata.update(result["metadata"])
|
||||
if failed:
|
||||
return {"ok": False, "error": result.get("error") or "Falha de domínio", "result": result, "metadata": metadata}
|
||||
return {"ok": True, "result": result, "metadata": metadata}
|
||||
except Exception as exc:
|
||||
return {"ok": False, "error": f"{type(exc).__name__}: {exc}", "metadata": {"server": "contas", "tool": call.tool_name, "framework_native": True}}
|
||||
Reference in New Issue
Block a user