From 18db1a3338390b65e9995bd1b368636dcc66be1c Mon Sep 17 00:00:00 2001 From: ShiroKSH Date: Wed, 15 Jul 2026 13:31:51 +0300 Subject: [PATCH] fix(bedrock): forward default model --- instructor/v2/auto_client.py | 1 + tests/coverage/test_auto_client_coverage.py | 7 ++++++- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/instructor/v2/auto_client.py b/instructor/v2/auto_client.py index 54335e960..f925f56a2 100644 --- a/instructor/v2/auto_client.py +++ b/instructor/v2/auto_client.py @@ -1023,6 +1023,7 @@ def _build_bedrock( client, mode=default_mode, async_client=async_client, + model=model_name, **kwargs, ) logger.info( diff --git a/tests/coverage/test_auto_client_coverage.py b/tests/coverage/test_auto_client_coverage.py index 5ad620ec0..86fe9598b 100644 --- a/tests/coverage/test_auto_client_coverage.py +++ b/tests/coverage/test_auto_client_coverage.py @@ -373,7 +373,12 @@ def create_instructor(client: object, **kwargs: Any) -> dict[str, Any]: "aws_session_token": "explicit-token", "region_name": "eu-west-1", } - assert result == {"mode": Mode.JSON, "async_client": True, "max_tokens": 19} + assert result == { + "mode": Mode.JSON, + "async_client": True, + "model": "amazon.titan", + "max_tokens": 19, + } def test_bedrock_reads_all_environment_credentials(