From 5da507af70d6373b1f104b0ce2bd456e39910d86 Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Thu, 29 Oct 2026 00:22:53 -0700
Subject: [PATCH] test(responses): read the expected chat message without JSON
 validation so the replay matcher stays

---
 tests/integration/translation/response_scenarios.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/integration/translation/response_scenarios.py b/tests/integration/translation/response_scenarios.py
index 7977c6c5a45..7168ad0ade5 100644
--- a/tests/integration/translation/response_scenarios.py
+++ b/tests/integration/translation/response_scenarios.py
@@ -57,7 +57,10 @@ def two_messages_before_tools(case: TranslationTestCase) -> TranslationTestCase:
 def _messages_then_tools_response(case: TranslationTestCase) -> Mapping[str, JsonValue]:
     expected: Final = case.expected_litellm_response
     if case.litellm_endpoint == "/v1/chat/completions":
-        message: Final = object_value(items(expected["choices"])[0]["message"])
+        choices: Final = expected["choices"]
+        assert isinstance(choices, list) and isinstance(choices[0], dict), choices
+        message: Final = choices[0]["message"]
+        assert isinstance(message, dict), message
         return {
             **expected,
             "choices": [