ajuste no contas

tests/test_llm_context_compaction.py

@@ -0,0 +1,93 @@
+import pytest
+
+from agent_framework.runtime.agent_runtime import AgentRuntimeMixin
+
+
+class DummyRuntime(AgentRuntimeMixin):
+    def __init__(self, llm=None):
+        self.llm = llm
+        self.cache = None
+        self.telemetry = None
+        self.settings = None
+
+
+class ContextFailThenOkLLM:
+    def __init__(self):
+        self.calls = []
+
+    async def ainvoke(self, messages, **kwargs):
+        self.calls.append((messages, kwargs))
+        if len(self.calls) == 1:
+            raise RuntimeError("Input length (134212) exceeds model's maximum context length (131072)")
+        return "ok"
+
+
+def test_compact_llm_value_drops_recursive_runtime_payload_but_keeps_business_facts():
+    runtime = DummyRuntime()
+    payload = {
+        "result": {
+            "subject": "Tamboro Mensal",
+            "success": True,
+            "service": {"details": {"valor": "14,99"}},
+            "state": {"session": {"secret": "must-not-leak"}, "huge": "x" * 20000},
+            "business_events": [{"payload": "x" * 20000}],
+        }
+    }
+    rendered = runtime._compact_llm_value(payload, max_chars=8000)
+    assert "Tamboro Mensal" in rendered
+    assert "14,99" in rendered
+    assert "must-not-leak" not in rendered
+    assert "business_events" not in rendered
+    assert len(rendered) < 9000
+
+
+def test_build_messages_bounds_mcp_and_transaction_evidence():
+    runtime = DummyRuntime()
+    giant = {
+        "tool_name": "cancelar_vas_avulso",
+        "ok": True,
+        "result": {
+            "subject": "Tamboro Mensal",
+            "validatedAmount": "14,99",
+            "state": {"payload": "x" * 100000},
+            "services": [{"name": f"svc-{i}", "details": {"valor": "1,00"}} for i in range(100)],
+        },
+    }
+    state = {
+        "user_text": "isso mesmo, pode cancelar",
+        "sanitized_input": "isso mesmo, pode cancelar",
+        "intent": "state:WAITING_CONFIRMATION",
+        "route": "contestacao_agent",
+        "business_context": {"customer_key": "11999999999"},
+        "transaction_evidence": [{
+            "transaction_id": "tx-1",
+            "tool_name": "cancelar_vas_avulso",
+            "arguments": {"subject": "Tamboro Mensal"},
+            "status": "COMPLETED",
+            "result": giant,
+        }],
+    }
+    messages = runtime.build_messages(state, system_prompt="system", mcp_results=[giant])
+    total = sum(len(m["content"]) for m in messages)
+    joined = "\n".join(m["content"] for m in messages)
+    assert total < 50000
+    assert "Tamboro Mensal" in joined
+    assert "14,99" in joined
+    assert "x" * 5000 not in joined
+
+
+@pytest.mark.asyncio
+async def test_context_length_error_retries_once_with_compacted_messages():
+    llm = ContextFailThenOkLLM()
+    runtime = DummyRuntime(llm=llm)
+    messages = [
+        {"role": "system", "content": "s" * 30000},
+        {"role": "user", "content": "u" * 120000},
+    ]
+    answer = await runtime._invoke_llm_cached({}, "ContestacaoAgent", messages)
+    assert answer == "ok"
+    assert len(llm.calls) == 2
+    first_chars = sum(len(m["content"]) for m in llm.calls[0][0])
+    second_chars = sum(len(m["content"]) for m in llm.calls[1][0])
+    assert second_chars < first_chars
+    assert second_chars <= 61000