scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 170

chore(guardrails): fix linting

Browse source
This commit is contained in:
aniket-kardile 2026-08-21 17:00:35 +05:30
parent 7732eaa3b4
commit 2062ceb616
No known key found for this signature in database
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/test_litellm/proxy/guardrails/guardrail_hooks/test_singulr.py
View file

@ -187,7 +187,9 @@ class TestSingulrRequestPayload:
request_data={},
input_type="request",
)
mock_post.assert_called_once()
sent_payload = mock_post.call_args.kwargs["json"]
for key, value in extra_inputs.items():
assert sent_payload[key] == value
@pytest.mark.asyncio
async def test_tools_are_forwarded(self, singulr_guardrail):
@ -557,8 +559,13 @@ class TestSingulrLoggingHook:
@pytest.mark.asyncio
async def test_no_messages_and_no_result_skips_both_api_calls(self, singulr_guardrail):
with patch.object(singulr_guardrail.async_handler, "post") as mock_post:
await singulr_guardrail.async_logging_hook(kwargs={}, result=None, call_type="acompletion")
returned_kwargs, returned_result = await singulr_guardrail.async_logging_hook(
kwargs={}, result=None, call_type="acompletion"
)
mock_post.assert_not_called()
assert returned_result is None
guardrail_information = returned_kwargs["standard_logging_object"]["guardrail_information"]
assert guardrail_information[0]["guardrail_status"] == "success"
@pytest.mark.asyncio
async def test_records_standard_logging_guardrail_information(self, singulr_guardrail):
@ -629,8 +636,9 @@ class TestSingulrRequestWiring:
request_data={},
input_type="request",
)
assert mock_post.call_args.kwargs["timeout"] == 5.0
assert mock_post.call_args.kwargs["url"] == "https://api.test.singulr.ai/api/v1/ai-gateway/litellm"
call_kwargs = mock_post.call_args.kwargs
assert call_kwargs["timeout"] == 5.0
assert call_kwargs["url"] == "https://api.test.singulr.ai/api/v1/ai-gateway/litellm"
class TestSingulrBuildHeaders: