From 2062ceb6167436b81431359fab5c8bf8ad0cf001 Mon Sep 17 00:00:00 2001 From: aniket-kardile Date: Fri, 21 Aug 2026 17:00:35 +0530 Subject: [PATCH] chore(guardrails): fix linting --- .../guardrails/guardrail_hooks/test_singulr.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_singulr.py b/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_singulr.py index 9eb1fa0d9e1..4996fe38dba 100644 --- a/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_singulr.py +++ b/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_singulr.py @@ -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: