phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update test_sagemaker.py to match changed parameters

Browse source
This commit is contained in:
Andrew Smith 2025-03-18 11:09:52 +11:00 • committed by GitHub
parent 81a7cf0f44
commit d5274efe62
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/local_testing/test_sagemaker.py
View file

@ -265,7 +265,7 @@ async def test_acompletion_sagemaker_non_stream():
# Assert # Assert
mock_post.assert_called_once() mock_post.assert_called_once()
_, kwargs = mock_post.call_args _, kwargs = mock_post.call_args
args_to_sagemaker = kwargs["json"] args_to_sagemaker = json.loads(kwargs["data"])
print("Arguments passed to sagemaker=", args_to_sagemaker) print("Arguments passed to sagemaker=", args_to_sagemaker)
assert args_to_sagemaker == expected_payload assert args_to_sagemaker == expected_payload
assert ( assert (
@ -325,7 +325,7 @@ async def test_completion_sagemaker_non_stream():
# Assert # Assert
mock_post.assert_called_once() mock_post.assert_called_once()
_, kwargs = mock_post.call_args _, kwargs = mock_post.call_args
args_to_sagemaker = kwargs["json"] args_to_sagemaker = json.loads(kwargs["data"])
print("Arguments passed to sagemaker=", args_to_sagemaker) print("Arguments passed to sagemaker=", args_to_sagemaker)
assert args_to_sagemaker == expected_payload assert args_to_sagemaker == expected_payload
assert ( assert (
@ -386,7 +386,7 @@ async def test_completion_sagemaker_prompt_template_non_stream():
# Assert # Assert
mock_post.assert_called_once() mock_post.assert_called_once()
_, kwargs = mock_post.call_args _, kwargs = mock_post.call_args
args_to_sagemaker = kwargs["json"] args_to_sagemaker = json.loads(kwargs["data"])
print("Arguments passed to sagemaker=", args_to_sagemaker) print("Arguments passed to sagemaker=", args_to_sagemaker)
assert args_to_sagemaker == expected_payload assert args_to_sagemaker == expected_payload
@ -445,7 +445,7 @@ async def test_completion_sagemaker_non_stream_with_aws_params():
# Assert # Assert
mock_post.assert_called_once() mock_post.assert_called_once()
_, kwargs = mock_post.call_args _, kwargs = mock_post.call_args
args_to_sagemaker = kwargs["json"] args_to_sagemaker = json.loads(kwargs["data"])
print("Arguments passed to sagemaker=", args_to_sagemaker) print("Arguments passed to sagemaker=", args_to_sagemaker)
assert args_to_sagemaker == expected_payload assert args_to_sagemaker == expected_payload
assert ( assert (