From 20ac04d6ccfd824b0263b9af3849285701f27a64 Mon Sep 17 00:00:00 2001
From: oandreeva-nv <oandreeva@nvidia.com>
Date: Tue, 5 Dec 2023 17:24:45 -0800
Subject: [PATCH] Added request id to span attributes

---
 src/tracer.cc | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/tracer.cc b/src/tracer.cc
index d9562f0b82..c64d10ee10 100644
--- a/src/tracer.cc
+++ b/src/tracer.cc
@@ -477,13 +477,20 @@ TraceManager::Trace::StartSpan(
 
   if (activity == TRITONSERVER_TRACE_REQUEST_START) {
     int64_t model_version;
+    const char* request_id;
     LOG_TRITONSERVER_ERROR(
         TRITONSERVER_InferenceTraceModelVersion(trace, &model_version),
         "getting model version");
+    LOG_TRITONSERVER_ERROR(
+        TRITONSERVER_InferenceTraceRequestId(trace, &request_id),
+        "getting request id");
     span->SetAttribute("triton.model_name", model_name);
     span->SetAttribute("triton.model_version", model_version);
     span->SetAttribute("triton.trace_id", trace_id);
     span->SetAttribute("triton.trace_parent_id", parent_id);
+    if (std::string(request_id) != "") {
+      span->SetAttribute("triton.request_id", request_id);
+    }
   }
 
   otel_context_ = otel_context_.SetValue(span_key, span);