agent-sdk 0.9.2

Rust Agent SDK for building LLM agents
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

//! OpenTelemetry context propagation helpers for async boundaries.
//!
//! The SDK uses `tokio::spawn` in `run()`. Those spawned futures are wrapped
//! with `FutureExt::with_context()` so the caller's `OTel` context is
//! re-attached on every poll, surviving task migration across tokio worker
//! threads. `run_turn()` executes directly in the caller task and therefore
//! does not need an extra spawn propagation hop.

use opentelemetry::Context;
use opentelemetry::trace::{SpanContext, TraceContextExt};

/// Capture the current `OTel` context for propagation into a spawned task.
///
/// Call this at the public API boundary (before `tokio::spawn`) and pass
/// the returned `Context` via `FutureExt::with_context()`.
#[must_use]
pub fn capture_context() -> Context {
    Context::current()
}

/// Capture the current `OTel` context and replace the active span.
///
/// This is used after starting a span that should become the parent for work
/// performed inside an async sub-future while preserving any existing baggage
/// from the caller context.
#[must_use]
pub fn current_with_span_context(span_context: SpanContext) -> Context {
    Context::current().with_remote_span_context(span_context)
}