Skip to main content

Returns

Returns a context token that must be passed to clear_context() for cleanup.

Example

trace_context = set_context(inference_result_id=inference_result_id)

Parameters

inference_result_id
string
The inference result ID to associate traces with. All subsequent traces will be linked to this ID.
Best Practice: Always use set_context() with a try/finally block to ensure clear_context() is called even if an exception occurs.

How It Works

When you call set_context(), it attaches the provided ID to the current OpenTelemetry context. Any traces created by @trace decorated functions or start_trace() will automatically include this ID as an attribute, enabling:
  1. Trace Correlation - All traces are linked to the same inference result
  2. Automatic Export - Traces are batched and exported to Galtea API