scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 114

test: update test

Browse source
This commit is contained in:
Krrish Dholakia 2025-02-10 20:40:56 -08:00
parent 197b1db6ec
commit a552da5eef
2 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/types/guardrails.py
View file

@ -156,7 +156,7 @@ class GuardrailLiteLLMParamsResponse(BaseModel):
class GuardrailInfoResponse(BaseModel):
guardrail_name: Optional[str]
guardrail_name: str
litellm_params: GuardrailLiteLLMParamsResponse
guardrail_info: Optional[Dict]

13
tests/logging_callback_tests/test_custom_guardrail.py
View file

@ -156,6 +156,12 @@ def test_get_guardrails_list_response():
sample_config = [
{
"guardrail_name": "test-guard",
"litellm_params": {
"guardrail": "test-guard",
"mode": "pre_call",
"api_key": "test-api-key",
"api_base": "test-api-base",
},
"guardrail_info": {
"params": [
{
@ -188,7 +194,12 @@ def test_get_guardrails_list_response():
assert len(empty_response.guardrails) == 0
# Test case 3: Missing optional fields
minimal_config = [{"guardrail_name": "minimal-guard"}]
minimal_config = [
{
"guardrail_name": "minimal-guard",
"litellm_params": {"guardrail": "minimal-guard", "mode": "pre_call"},
}
]
minimal_response = _get_guardrails_list_response(minimal_config)
assert isinstance(minimal_response, ListGuardrailsResponse)
assert len(minimal_response.guardrails) == 1