From b6b5b31ce46641d0764a122e93757ae4f5049b16 Mon Sep 17 00:00:00 2001 From: mateo-berri <277851410+mateo-berri@users.noreply.github.com> Date: Wed, 2 Sep 2026 15:25:05 -0700 Subject: [PATCH 1/2] test(bedrock): accept the router kwarg in the knowledge base search fake --- tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py b/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py index 3f9f2bacdd3..48a92e72b27 100644 --- a/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py +++ b/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py @@ -333,6 +333,7 @@ async def test_bedrock_kb_request_body_has_transformed_filters( timeout=None, client=None, _is_async=False, + router=None, ): litellm_params_dict = ( litellm_params.model_dump(exclude_none=False) From 8e26d13fa0043fbb6f358e171e526685fac037b5 Mon Sep 17 00:00:00 2001 From: mateo-berri <277851410+mateo-berri@users.noreply.github.com> Date: Wed, 2 Sep 2026 16:03:39 -0700 Subject: [PATCH 2/2] test(bedrock): type the router parameter on the knowledge base search fake --- tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py b/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py index 48a92e72b27..0c362db8853 100644 --- a/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py +++ b/tests/logging_callback_tests/test_bedrock_knowledgebase_hook.py @@ -333,7 +333,7 @@ async def test_bedrock_kb_request_body_has_transformed_filters( timeout=None, client=None, _is_async=False, - router=None, + router: "litellm.Router | None" = None, ): litellm_params_dict = ( litellm_params.model_dump(exclude_none=False)