phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +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", call_type="completion",
) )
except Exception as e: 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", call_type="completion",
) )
pytest.fail(f"Should have failed - {str(e)}") pytest.fail(f"Should have failed - {str(e)}")