opentelemetry_replicate_instrumentation_quickstart_with_console_export.py

1import replicate
2from opentelemetry import trace
3from opentelemetry.sdk.trace import TracerProvider
4from opentelemetry.sdk.trace.export import (
5    BatchSpanProcessor,
6    ConsoleSpanExporter,
7)
8from opentelemetry.instrumentation.replicate import ReplicateInstrumentor
9
10# Sets up the OpenTelemetry SDK
11provider = TracerProvider()
12processor = BatchSpanProcessor(ConsoleSpanExporter())
13provider.add_span_processor(processor)
14trace.set_tracer_provider(provider)
15
16# Instruments the replicate library
17ReplicateInstrumentor().instrument()
18
19# Now, use the replicate library as usual
20# Ensure REPLICATE_API_TOKEN is set in your environment
21model = replicate.models.get("replicate/hello-world")
22version = model.versions.get("5c0a3665671c5a2c6c4c2ff5827bc932111d14d9e0394747440409a803730e7f")
23prediction = version.predict(topic="OpenTelemetry")
24
25print(prediction)