mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-05 08:07:05 +00:00
test: give the mocked prepared request real headers for the masked debug log
This commit is contained in:
parent
e601114383
commit
76cfa6339b
1 changed files with 3 additions and 1 deletions
|
|
@ -917,7 +917,9 @@ async def test_bedrock_guardrail_make_api_request_passes_api_key():
|
|||
"Content-Type": "application/json",
|
||||
"Authorization": "Bearer test-api-key-789",
|
||||
}
|
||||
mock_request_instance.prepare.return_value = Mock()
|
||||
mock_request_instance.prepare.return_value = Mock(
|
||||
headers=mock_request_instance.headers
|
||||
)
|
||||
mock_aws_request.return_value = mock_request_instance
|
||||
|
||||
await guardrail_hook.make_bedrock_api_request(
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue