diff --git a/tests/logging_callback_tests/test_otel_logging.py b/tests/logging_callback_tests/test_otel_logging.py index fc10cf63e5..9c19c9d261 100644 --- a/tests/logging_callback_tests/test_otel_logging.py +++ b/tests/logging_callback_tests/test_otel_logging.py @@ -102,6 +102,7 @@ def validate_raw_gen_ai_request_openai_non_streaming(span): "llm.openai.messages", "llm.openai.temperature", "llm.openai.user", + "llm.openai.extra_body", "llm.openai.id", "llm.openai.choices", "llm.openai.created", @@ -125,6 +126,7 @@ def validate_raw_gen_ai_request_openai_streaming(span): "llm.openai.messages", "llm.openai.temperature", "llm.openai.user", + "llm.openai.extra_body", "llm.openai.model", ]