From b9cd6684f5b265f51763dfab558580267a642ace Mon Sep 17 00:00:00 2001 From: Shay Date: Wed, 13 May 2026 13:17:47 -0700 Subject: [PATCH] Assert session vault contains assistant final state --- tests/test_engine_loop_proof.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/tests/test_engine_loop_proof.py b/tests/test_engine_loop_proof.py index 2c9aad6e..e9311277 100644 --- a/tests/test_engine_loop_proof.py +++ b/tests/test_engine_loop_proof.py @@ -102,10 +102,14 @@ def test_session_context_respond_preserves_and_vaults_final_state() -> None: assert session.state is result.final_state assert not np.array_equal(result.final_state.F, initial.F) - recalled = session.vault.recall(result.final_state.F, top_k=1) - assert recalled[0]["metadata"]["role"] == "assistant" - np.testing.assert_allclose(recalled[0]["versor"], result.final_state.F) - assert not np.array_equal(recalled[0]["versor"], initial.F) + recalled = session.vault.recall(result.final_state.F, top_k=2) + assistant_hits = [ + item for item in recalled + if item["metadata"].get("role") == "assistant" + ] + assert assistant_hits, "Assistant final_state was not present in session vault recall." + np.testing.assert_allclose(assistant_hits[0]["versor"], result.final_state.F) + assert not np.array_equal(assistant_hits[0]["versor"], initial.F) def test_hot_path_modules_route_through_backend_boundary() -> None: