scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

fix(test): mock headers in test_completion_fine_tuned_model

Browse source
This commit is contained in:
joereyna 2026-04-09 13:18:35 -07:00
parent 3a6db708ce
commit 1571f5e45a
No known key found for this signature in database
GPG key ID: 37E09E2BDB5920E5
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/local_testing/test_amazing_vertex_completion.py
View file

@ -2291,6 +2291,8 @@ def test_prompt_factory_nested():
async def test_completion_fine_tuned_model():
load_vertex_ai_credentials()
mock_response = AsyncMock()
mock_response.headers = {}
mock_response.status_code = 200
def return_val():
return {
@ -2326,7 +2328,6 @@ async def test_completion_fine_tuned_model():
}
mock_response.json = return_val
mock_response.status_code = 200
expected_payload = {
"contents": [