From 37a69bddc90efdda1e5211cf6bcac767b46d5048 Mon Sep 17 00:00:00 2001 From: ryan Date: Thu, 5 Nov 2026 04:54:36 +0000 Subject: [PATCH] test: route Bedrock preamble checks through adaptive thinking Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/e2e/llm_translation/test_bedrock_native_e2e.py | 1 + .../unit/llms/bedrock/chat/test_converse_transformation.py | 6 ++++++ 2 files changed, 7 insertions(+) diff --git a/tests/e2e/llm_translation/test_bedrock_native_e2e.py b/tests/e2e/llm_translation/test_bedrock_native_e2e.py index 95720bcace2..6df4ae1816e 100644 --- a/tests/e2e/llm_translation/test_bedrock_native_e2e.py +++ b/tests/e2e/llm_translation/test_bedrock_native_e2e.py @@ -230,6 +230,7 @@ class TestBedrockNative: model=model, messages=[{"role": "user", "content": WAREHOUSE_PROMPT}], response_format=WAREHOUSE_RESPONSE_FORMAT, + extra_body={"thinking": {"type": "adaptive"}}, max_tokens=1500, ) .choices[0] diff --git a/tests/unit/llms/bedrock/chat/test_converse_transformation.py b/tests/unit/llms/bedrock/chat/test_converse_transformation.py index b8c39d01e6e..168386b2110 100644 --- a/tests/unit/llms/bedrock/chat/test_converse_transformation.py +++ b/tests/unit/llms/bedrock/chat/test_converse_transformation.py @@ -7137,6 +7137,10 @@ def test_bedrock_json_mode_fake_stream_preserves_preamble_in_chunks() -> None: } def respond(request: httpx.Request) -> httpx.Response: + request_body: Final = cast(RequestObject, json.loads(request.content)) + tool_config: Final = request_body.get("toolConfig") + assert tool_config is not None + assert [tool["toolSpec"]["name"] for tool in tool_config["tools"]] == ["json_tool_call"] return httpx.Response(200, json=response_json, request=request) model: Final = "bedrock/us.anthropic.claude-opus-5-5" @@ -7156,6 +7160,8 @@ def test_bedrock_json_mode_fake_stream_preserves_preamble_in_chunks() -> None: }, }, }, + thinking={"type": "adaptive"}, + max_tokens=1500, stream=True, aws_access_key_id="fake", aws_secret_access_key="fake",