bugfix: aofertas
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user