diff --git a/litellm/llms/sagemaker.py b/litellm/llms/sagemaker.py index dab8cb1a30..e3a58a7675 100644 --- a/litellm/llms/sagemaker.py +++ b/litellm/llms/sagemaker.py @@ -387,7 +387,7 @@ class SagemakerLLM(BaseAWSLLM): original_response=completion_response, additional_args={"complete_input_dict": _data}, ) - print_verbose(f"raw model_response: {response}") + print_verbose(f"raw model_response: {completion_response}") ## RESPONSE OBJECT try: if isinstance(completion_response, list): diff --git a/litellm/tests/test_sagemaker.py b/litellm/tests/test_sagemaker.py index 155b07be06..a06a238a38 100644 --- a/litellm/tests/test_sagemaker.py +++ b/litellm/tests/test_sagemaker.py @@ -84,7 +84,7 @@ async def test_completion_sagemaker(sync_mode): @pytest.mark.asyncio() -@pytest.mark.parametrize("sync_mode", [True]) +@pytest.mark.parametrize("sync_mode", [False, True]) async def test_completion_sagemaker_stream(sync_mode): try: litellm.set_verbose = False