From a0c4e4684a8dc9d0fb57b9dbcdc09f36e228391c Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Tue, 14 Jul 2026 17:12:15 -0700
Subject: [PATCH] test(e2e): scope virtual keys to the deployment under test

---
 .../test_messages_mid_conversation_system_e2e.py              | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/e2e/llm_translation/test_messages_mid_conversation_system_e2e.py b/tests/e2e/llm_translation/test_messages_mid_conversation_system_e2e.py
index f5da3744fab..1025d603fca 100644
--- a/tests/e2e/llm_translation/test_messages_mid_conversation_system_e2e.py
+++ b/tests/e2e/llm_translation/test_messages_mid_conversation_system_e2e.py
@@ -156,7 +156,7 @@ class TestBedrockInvokeMidConversationSystem:
         model = _register_invoke_deployment(
             endpoints_client, resources, FLAGGED_INVOKE_MODEL
         )
-        key = resources.key()
+        key = resources.key(models=[model])
         system_block = _cacheable_system_block(unique_marker())
 
         primed = _prime_prompt_cache(endpoints_client, key, model, system_block)
@@ -196,7 +196,7 @@ class TestBedrockInvokeMidConversationSystem:
         model = _register_invoke_deployment(
             endpoints_client, resources, UNFLAGGED_INVOKE_MODEL
         )
-        key = resources.key()
+        key = resources.key(models=[model])
 
         body = RichMessagesRequest(
             model=model,