Skip to content

Commit aa65897

Browse files
authored
Apply suggestions from code review
1 parent 8b42a63 commit aa65897

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

instrumentation/elastic-opentelemetry-instrumentation-openai/tests/test_chat_completions.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -561,6 +561,7 @@ def test_chat_tools_with_capture_message_content_integration(trace_exporter, log
561561

562562
address, port = address_and_port(client)
563563
assert dict(span.attributes) == {
564+
GEN_AI_OPENAI_RESPONSE_SERVICE_TIER: "default",
564565
GEN_AI_OPERATION_NAME: "chat",
565566
GEN_AI_REQUEST_MODEL: model,
566567
GEN_AI_SYSTEM: "openai",
@@ -695,6 +696,7 @@ def test_chat_with_capture_message_content_integration(trace_exporter, logs_expo
695696

696697
address, port = address_and_port(client)
697698
assert dict(span.attributes) == {
699+
GEN_AI_OPENAI_RESPONSE_SERVICE_TIER: "default",
698700
GEN_AI_OPERATION_NAME: "chat",
699701
GEN_AI_REQUEST_MODEL: model,
700702
GEN_AI_SYSTEM: "openai",
@@ -1052,6 +1054,7 @@ def test_chat_stream_with_include_usage_option_and_capture_message_content_integ
10521054

10531055
address, port = address_and_port(client)
10541056
assert dict(span.attributes) == {
1057+
GEN_AI_OPENAI_RESPONSE_SERVICE_TIER: "default",
10551058
GEN_AI_OPERATION_NAME: "chat",
10561059
GEN_AI_REQUEST_MODEL: model,
10571060
GEN_AI_SYSTEM: "openai",
@@ -1629,6 +1632,7 @@ async def test_chat_async_with_capture_message_content_integration(trace_exporte
16291632

16301633
address, port = address_and_port(client)
16311634
assert dict(span.attributes) == {
1635+
GEN_AI_OPENAI_RESPONSE_SERVICE_TIER: "default",
16321636
GEN_AI_OPERATION_NAME: "chat",
16331637
GEN_AI_REQUEST_MODEL: model,
16341638
GEN_AI_SYSTEM: "openai",

0 commit comments

Comments
 (0)