scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

test(grayswan): capture monitor calls immutably in the unit test client

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-30 19:57:44 +00:00
parent 98888713e4
commit b53a8623a0
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/test_litellm/proxy/guardrails/guardrail_hooks/test_grayswan.py
View file

@ -1,3 +1,6 @@
from collections.abc import Mapping
from types import MappingProxyType
import pytest
from fastapi import HTTPException
@ -595,12 +598,15 @@ def test_ensure_litellm_metadata_noop_when_already_present() -> None:
class _CapturingClient:
def __init__(self, payload: dict | None = None):
def __init__(self, payload: dict[str, float] | None = None):
self.payload = payload or {"violation": 0.0}
self.calls: list[dict] = []
self.calls: tuple[Mapping[str, object], ...] = ()
async def post(self, *, url: str, headers: dict, json: dict, timeout: float):
self.calls.append({"url": url, "headers": headers, "json": json, "timeout": timeout})
self.calls = (
*self.calls,
MappingProxyType({"url": url, "headers": headers, "json": json, "timeout": timeout}),
)
return _DummyResponse(self.payload)