From 8a9a4c2ed8f77e5ce52eaa76e704b9ebb36d0b4a Mon Sep 17 00:00:00 2001
From: Liz <91279165+lizradway@users.noreply.github.com>
Date: Thu, 21 May 2026 11:19:45 -0400
Subject: [PATCH] fix(tests): update bedrock to accept string or number

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

diff --git a/tests_integ/models/test_model_mantle.py b/tests_integ/models/test_model_mantle.py
index 7cc032146..9a432d993 100644
--- a/tests_integ/models/test_model_mantle.py
+++ b/tests_integ/models/test_model_mantle.py
@@ -40,13 +40,13 @@ def test_chat_completions_agent_invoke(chat_completions_model):
     """OpenAIModel (Chat Completions) reaches Mantle via bedrock_mantle_config."""
     agent = Agent(model=chat_completions_model, system_prompt="Reply in one short sentence.", callback_handler=None)
     result = agent("What is 2+2?")
-    assert "4" in str(result)
+    assert "4" in str(result) or "four" in str(result).lower()
 
 
 def test_agent_invoke(model):
     agent = Agent(model=model, system_prompt="Reply in one short sentence.", callback_handler=None)
     result = agent("What is 2+2?")
-    assert "4" in str(result)
+    assert "4" in str(result) or "four" in str(result).lower()
 
 
 def test_responses_server_side_conversation(stateful_model):