From 89632ca1bee86d562fc72d92a483da7460b9c697 Mon Sep 17 00:00:00 2001
From: "devin-ai-integration[bot]"
 <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Fri, 13 Nov 2026 05:11:21 +0000
Subject: [PATCH] test(observability): classify post-auth datastore spans by
 identity (#51554)

Co-authored-by: mateo <277851410+mateo-berri@users.noreply.github.com>
Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 .../observability/test_otel_excluded_services_matrix.py    | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/integration/observability/test_otel_excluded_services_matrix.py b/tests/integration/observability/test_otel_excluded_services_matrix.py
index 8a7236f8626..cc839c0d4c2 100644
--- a/tests/integration/observability/test_otel_excluded_services_matrix.py
+++ b/tests/integration/observability/test_otel_excluded_services_matrix.py
@@ -346,8 +346,11 @@ def _db_systems(spans: tuple[Span, ...]) -> set[str]:
 
 
 def _post_auth_datastore_spans(spans: tuple[Span, ...]) -> tuple[Span, ...]:
-    auth_ids: Final = frozenset(span["span_id"] for span in spans if span["name"].startswith("auth "))
-    return tuple(span for span in spans if _db_systems((span,)) and span["parent_span_id"] not in auth_ids)
+    return tuple(
+        span
+        for span in spans
+        if _db_systems((span,)) and span["attributes"].get("litellm.team.id") is not None
+    )
 
 
 def _names(spans: tuple[Span, ...]) -> list[str]: