Skip to content

Consolidate LLM I/O Events into Single Event Per Span [WIP] #375

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 2 commits into
base: genesis-dev-v2
Choose a base branch
from
Open

Consolidate LLM I/O Events into Single Event Per Span [WIP] #375

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
1f13a82
impl: add unsampled span pipeline for agent observability enabled
yiyuan-he Jun 4, 2025
1fd5e98
consolidate gen ai events
yiyuan-he May 28, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 27 additions & 0 deletions aws-opentelemetry-distro/src/amazon/opentelemetry/distro/aws_opentelemetry_configurator.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -288,6 +288,30 @@ def _export_unsampled_span_for_lambda(trace_provider: TracerProvider, resource:
)


def _export_unsampled_span_for_agent_observability(trace_provider: TracerProvider, resource: Resource = None):
if not is_agent_observability_enabled():
return

# Get the traces endpoint from environment
traces_endpoint = os.environ.get(OTEL_EXPORTER_OTLP_TRACES_ENDPOINT)

if not traces_endpoint:
# No traces endpoint configured, skip unsampled span export
_logger.warning("No traces endpoint configured for agent observability unsampled spans")
return

# Create the appropriate span exporter based on the endpoint
if _is_aws_otlp_endpoint(traces_endpoint, "xray"):
span_exporter = OTLPAwsSpanExporter(endpoint=traces_endpoint, logger_provider=get_logger_provider())
else:
span_exporter = OTLPSpanExporter(endpoint=traces_endpoint)

# Add the unsampled span processor
trace_provider.add_span_processor(
BatchUnsampledSpanProcessor(span_exporter=span_exporter)
)


def _is_defer_to_workers_enabled():
return os.environ.get(OTEL_AWS_PYTHON_DEFER_TO_WORKERS_ENABLED_CONFIG, "false").strip().lower() == "true"

Expand Down Expand Up @@ -426,6 +450,9 @@ def _customize_span_processors(provider: TracerProvider, resource: Resource) ->
if _is_lambda_environment():
provider.add_span_processor(AwsLambdaSpanProcessor())

# Add unsampled span export for agent observability
_export_unsampled_span_for_agent_observability(provider, resource)

if not _is_application_signals_enabled():
return

Expand Down
Loading