scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-10 08:28:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test(proxy): cover proxy metadata assembly in the preview metadata regression test

Browse source
This commit is contained in:
shrey kharbanda 2026-10-27 22:18:43 +00:00
parent 0cd4ddca01
commit cb9e6d11f6
1 changed files with 38 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
tests/unit/proxy/test_litellm_pre_call_utils.py
View file

@ -59,6 +59,7 @@ from litellm.proxy.litellm_pre_call_utils import (
)
from litellm.proxy.spend_tracking.spend_tracking_utils import get_logging_payload
from litellm.types.utils import CredentialItem
from litellm.utils import get_requester_metadata
def test_check_if_token_is_service_account():
@ -266,6 +267,43 @@ async def test_add_litellm_data_to_request_keeps_requester_ip_address_for_guardr
assert updated_data["metadata"]["requester_ip_address"] == "203.0.113.9"
@pytest.mark.asyncio
@pytest.mark.parametrize(
"data, expected",
(
({"model": "gpt-4o"}, None),
({"model": "gpt-4o", "metadata": {"team": "search"}}, {"team": "search"}),
),
)
async def test_proxy_forwards_only_caller_metadata_to_openai(
monkeypatch: pytest.MonkeyPatch, data: dict[str, object], expected: dict[str, str] | None
) -> None:
from litellm.proxy import proxy_server
settings: Final = {}
monkeypatch.setattr(proxy_server, "general_settings", settings)
request_mock: Final = MagicMock(spec=Request)
request_mock.url = MagicMock()
request_mock.url.path = "/v1/chat/completions"
request_mock.url.__str__.return_value = "http://localhost/v1/chat/completions"
request_mock.method = "POST"
request_mock.query_params = {}
request_mock.headers = {"Content-Type": "application/json", "user-agent": "test-agent"}
request_mock.client = MagicMock()
request_mock.client.host = "127.0.0.1"
updated_data: Final = await add_litellm_data_to_request(
data=data,
request=request_mock,
user_api_key_dict=UserAPIKeyAuth(api_key="hashed-key", user_id="u1"),
proxy_config=MagicMock(),
general_settings=settings,
version="test-version",
)
assert get_requester_metadata(updated_data["metadata"]) == expected, f"metadata={updated_data['metadata']!r}"
@pytest.mark.asyncio
async def test_key_otel_service_name_outranks_team_metadata_merge():
from litellm.proxy.litellm_pre_call_utils import add_litellm_data_to_request