core.telemetry_integration

OpenTelemetry Integration Module

Provides comprehensive observability integration using OpenTelemetry for distributed tracing, metrics collection, and logging correlation.

Functions

count_function_calls([metric_name])	Decorator to count function calls
get_rag_instrumentation()	Get RAG-specific instrumentation
get_telemetry()	Get global telemetry integration instance
initialize_telemetry([config])	Initialize global telemetry integration
time_function_execution([metric_name])	Decorator to time function execution
trace_async_function([operation_name, ...])	Decorator to trace async function execution
trace_function([operation_name, attributes])	Decorator to trace function execution

Classes

RAGTelemetryInstrumentation(telemetry)	RAG-specific telemetry instrumentation
TelemetryConfig([config])	Configuration for OpenTelemetry integration
TelemetryIntegration([config])	Main OpenTelemetry integration class

class core.telemetry_integration.TelemetryConfig(config=None)

Bases: object

Configuration for OpenTelemetry integration

__init__(config=None)

class core.telemetry_integration.TelemetryIntegration(config=None)

Bases: object

Main OpenTelemetry integration class

__init__(config=None)

initialize()

Initialize OpenTelemetry configuration

shutdown()

Release this instance’s SDK providers, including partial initialization.

trace_span(name, attributes=None)

Create a trace span context manager

create_counter(name, description='', unit='1')

Create a metrics counter

create_histogram(name, description='', unit='1')

Create a metrics histogram

create_gauge(name, description='', unit='1')

Create a metrics gauge

class core.telemetry_integration.RAGTelemetryInstrumentation(telemetry)

Bases: object

RAG-specific telemetry instrumentation

__init__(telemetry)

trace_document_processing(document_id)

Trace document processing operation

trace_rag_query(query, user_id=None)

Trace RAG query operation

trace_vector_search(query_embedding_size, top_k)

Trace vector similarity search

record_document_processed(processing_time, success=True)

Record document processing metrics

record_query_executed(query_time, retrieval_count, success=True)

Record query execution metrics

core.telemetry_integration.initialize_telemetry(config=None)

Initialize global telemetry integration

core.telemetry_integration.get_telemetry()

Get global telemetry integration instance

core.telemetry_integration.get_rag_instrumentation()

Get RAG-specific instrumentation

core.telemetry_integration.trace_function(operation_name=None, attributes=None)

Decorator to trace function execution

core.telemetry_integration.trace_async_function(operation_name=None, attributes=None)

Decorator to trace async function execution

core.telemetry_integration.count_function_calls(metric_name=None)

Decorator to count function calls

core.telemetry_integration.time_function_execution(metric_name=None)

Decorator to time function execution