bugfix: aofertas

This commit is contained in:
2026-08-21 10:44:34 -03:00
parent e063e4b500
commit a231f063ed
8 changed files with 141 additions and 7 deletions

View File

@@ -359,3 +359,37 @@ def test_finalizacao_informacional_infer_tipo_de_billing_analysis_json():
assert _infer_informational_vas_types(
["YouTube Premium Mensal", "Aluguel de Filme 3"], json.dumps(payload, ensure_ascii=False)
) == {"estrategico", "avulso"}
def test_output_guardrail_context_inclui_pedido_atual_para_aoferta():
from app.workflows.agent_graph import AgentWorkflow
ctx = AgentWorkflow._output_guardrail_context(
{
"context": {},
"history": [
{"role": "assistant", "content": "O TIM Music custa R$ 12,90."},
],
"user_text": "Quero cancelar o TIM Music",
"mcp_results": [],
}
)
assert ctx["conversation_history"][-1] == {
"role": "user",
"content": "Quero cancelar o TIM Music",
}
assert ctx["history_texts"][-1] == "Quero cancelar o TIM Music"
def test_output_guardrail_context_nao_duplica_user_text_ja_no_historico():
from app.workflows.agent_graph import AgentWorkflow
ctx = AgentWorkflow._output_guardrail_context(
{
"context": {},
"history": [{"role": "user", "content": "Quero cancelar o TIM Music"}],
"user_text": "Quero cancelar o TIM Music",
"mcp_results": [],
}
)
assert len(ctx["conversation_history"]) == 1