Function withCopilotOtelContext

apps/sim/lib/copilot/request/otel.ts:545–589

(
  scope: CopilotOtelScope,
  fn: (otelContext: Context) => Promise<T>
)

}

export async function withCopilotOtelContext<T>(
  scope: CopilotOtelScope,
  fn: (otelContext: Context) => Promise<T>
): Promise<T> {
  const parentContext = context.active()
  // Same trace-id-derives-requestId dance as startCopilotOtelRoot — see
  // that function for the rationale. Stamp a placeholder, read the real
  // trace ID off the span, then overwrite.
  const span = getTracer().startSpan(
    TraceSpan.GenAiAgentExecute,
    { attributes: buildAgentSpanAttributes({ ...scope, requestId: scope.requestId ?? '' }) },
    parentContext
  )
  const carrierSpan = isValidSpanContext(span.spanContext())
    ? span
    : trace.wrapSpanContext(createFallbackSpanContext())
  const spanContext = carrierSpan.spanContext()
  const resolvedRequestId =
    scope.requestId ??
    (spanContext.traceId && spanContext.traceId.length === 32 ? spanContext.traceId : '')
  if (resolvedRequestId) {
    span.setAttribute(TraceAttr.RequestId, resolvedRequestId)
    span.setAttribute(TraceAttr.SimRequestId, resolvedRequestId)
  }
  const otelContext = trace.setSpan(parentContext, carrierSpan)
  let terminalStatusSet = false

  try {
    const result = await context.with(otelContext, () => fn(otelContext))
    span.setStatus({ code: SpanStatusCode.OK })
    terminalStatusSet = true
    return result
  } catch (error) {
    markSpanForError(span, error)
    terminalStatusSet = true
    throw error
  } finally {
    if (!terminalStatusSet) {
      // Extremely defensive: should be unreachable, but avoids leaking
      // an unset span status if some future refactor breaks both arms.
      span.setStatus({ code: SpanStatusCode.OK })
    }
    span.end()
  }
}