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

test(mcp): use the shared guardrail exception in regression

Browse source
This commit is contained in:
Joshua Valluru 2026-09-17 10:54:28 -07:00
parent 5b91195406
commit f4918e69f4
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/responses/mcp/test_chat_completions_handler.py
View file

@ -1395,7 +1395,7 @@ async def test_acompletion_with_mcp_forwards_unserved_external_mcp_tool_to_the_p
@pytest.mark.parametrize("selection_source", ["metadata", "litellm_metadata", "body"])
@pytest.mark.parametrize("logging_failure", [False, True])
async def test_request_selected_mcp_guardrail_blocks_before_upstream(monkeypatch, selected, stream, selection_source, logging_failure):
from fastapi import HTTPException
from litellm.exceptions import GuardrailRaisedException
from mcp.types import Tool
from litellm.caching.caching import DualCache
from litellm.integrations.custom_guardrail import CustomGuardrail
@ -1410,7 +1410,7 @@ async def test_request_selected_mcp_guardrail_blocks_before_upstream(monkeypatch
class BlockSelected(CustomGuardrail):
async def async_pre_call_hook(self, user_api_key_dict, cache, data, call_type):
if self.should_run_guardrail(data, GuardrailEventHooks.pre_mcp_call):
raise HTTPException(status_code=400, detail="request-selected MCP block")
raise GuardrailRaisedException(message="request-selected MCP block", blocked_content=True)
return data
guardrail = BlockSelected(guardrail_name="block-all", event_hook="pre_mcp_call", default_on=False)