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

test(otel): project routing survives a sibling backend destination

Browse source
This commit is contained in:
Yucheng He 2026-09-05 05:57:27 -07:00
parent 7b8e233975
commit 8fc2ff1f27
1 changed files with 15 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
tests/test_litellm/integrations/otel/test_otel_v2_destinations.py
View file

@ -920,19 +920,27 @@ class TestRouting:
assert route.detached is False
assert route.provider is None
def test_a_backend_with_its_own_credentials_still_routes_next_to_another_backend_destination(self):
"""Credentials name the tenant's own account for this backend, which the other
backend's destination cannot stand in for."""
@pytest.mark.parametrize(
("owner", "params", "auth_metadata"),
[
(ExporterOwner.ARIZE_AX, {"arize_space_key": "space", "arize_api_key": "key"}, {}),
(ExporterOwner.ARIZE_PHOENIX, None, {"phoenix_project_name": "team-project"}),
],
)
def test_a_backend_pointed_at_its_own_account_still_routes_next_to_another_backend_destination(
self, owner, params, auth_metadata
):
"""Credentials or a project name the tenant's own account for this backend, which
the other backend's destination cannot stand in for."""
config = OpenTelemetryV2Config(
exporters=[ExporterSpec(kind="otlp_http", endpoint="http://op.local", owner=ExporterOwner.ARIZE_AX)]
exporters=[ExporterSpec(kind="otlp_http", endpoint="http://op.local", owner=owner)]
)
cache = TenantTracerCache(config, "arize", "litellm")
cache = TenantTracerCache(config, owner.value, "litellm")
default = get_tracer(TracerProvider(), "litellm")
params = {"arize_space_key": "space", "arize_api_key": "key"}
def run():
set_request_destinations((LANGFUSE_DEST,))
return cache.route_for(default, params, {"otel_service_name": "team-checkout"})
return cache.route_for(default, params, {"otel_service_name": "team-checkout", **auth_metadata})
route = in_fresh_context(run)
assert route.tracer is not default