diff --git a/tests/llm_translation/test_unit_test_bedrock_invoke.py b/tests/llm_translation/test_unit_test_bedrock_invoke.py index da9ad71264b..41380731833 100644 --- a/tests/llm_translation/test_unit_test_bedrock_invoke.py +++ b/tests/llm_translation/test_unit_test_bedrock_invoke.py @@ -28,6 +28,7 @@ def test_get_complete_url_basic(bedrock_transformer): model="anthropic.claude-v2", optional_params={}, stream=False, + litellm_params={}, ) assert ( @@ -43,6 +44,7 @@ def test_get_complete_url_streaming(bedrock_transformer): model="anthropic.claude-v2", optional_params={}, stream=True, + litellm_params={}, ) assert (