weaviate_opentelemetry_instrumentation_quickstart_console_tracing.py

1import weaviate
2from opentelemetry import trace
3from opentelemetry.sdk.trace import TracerProvider
4from opentelemetry.sdk.trace.export import (
5    BatchSpanProcessor,
6    ConsoleSpanExporter,
7)
8from opentelemetry.instrumentation.weaviate import WeaviateInstrumentor
9
10# 1. Setup OpenTelemetry SDK (Tracing)
11provider = TracerProvider()
12processor = BatchSpanProcessor(ConsoleSpanExporter())
13provider.add_span_processor(processor)
14trace.set_tracer_provider(provider)
15
16# 2. Instrument Weaviate
17# This will automatically wrap the Weaviate client methods with tracing spans
18WeaviateInstrumentor().instrument()
19
20# 3. Use the Weaviate client as normal
21client = weaviate.Client("http://localhost:8080")
22
23# Example operation that will now generate spans
24try:
25    meta_info = client.get_meta()
26    print(meta_info)
27except Exception as e:
28    print(f"Error: {e}")