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

test(test_llm_guard.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-03-26 18:37:27 -07:00
parent f62f642393
commit 4488480188
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_llm_guard.py
View file

@ -53,6 +53,7 @@ async def test_llm_guard_valid_response():
}
]
},
user_api_key_dict=user_api_key_dict,
call_type="completion",
)
except Exception as e:
@ -88,6 +89,7 @@ async def test_llm_guard_error_raising():
}
]
},
user_api_key_dict=user_api_key_dict,
call_type="completion",
)
pytest.fail(f"Should have failed - {str(e)}")