Skip to content

Commit 365af51

Browse files
committed
minor change
1 parent 7740f3e commit 365af51

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/observability/core/test_agent365_exporter.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -720,7 +720,7 @@ def test_export_includes_inference_operation_type_chat_spans(self):
720720
spans_out = request_data["resourceSpans"][0]["scopeSpans"][0]["spans"]
721721
self.assertEqual(len(spans_out), 1)
722722
# Value is preserved as-is; no normalization
723-
self.assertEqual(spans_out[0]["attributes"]["gen_ai.operation.name"], "Chat")
723+
self.assertEqual(spans_out[0]["attributes"][GEN_AI_OPERATION_NAME_KEY], "Chat")
724724

725725
def test_export_filters_out_unsupported_inference_operation_types(self):
726726
"""Spans with TextCompletion / GenerateContent are filtered out."""
@@ -754,7 +754,7 @@ def test_export_does_not_normalize_canonical_operation_names(self):
754754
_, body, _ = mock_post.call_args[0]
755755
request_data = json.loads(body)
756756
span_out = request_data["resourceSpans"][0]["scopeSpans"][0]["spans"][0]
757-
self.assertEqual(span_out["attributes"]["gen_ai.operation.name"], op)
757+
self.assertEqual(span_out["attributes"][GEN_AI_OPERATION_NAME_KEY], op)
758758

759759

760760
if __name__ == "__main__":

0 commit comments

Comments
 (0)