opentelemetry_cohere_sdk_instrumentation_quickstart_with_console_export.py

1import cohere
2from opentelemetry.instrumentation.cohere import CohereInstrumentor
3from opentelemetry import trace
4from opentelemetry.sdk.trace import TracerProvider
5from opentelemetry.sdk.trace.export import (
6    BatchSpanProcessor,
7    ConsoleSpanExporter,
8)
9
10# Initialize OpenTelemetry Tracer
11trace.set_tracer_provider(TracerProvider())
12tracer_provider = trace.get_tracer_provider()
13tracer_provider.add_span_processor(BatchSpanProcessor(ConsoleSpanExporter()))
14
15# Instrument Cohere
16CohereInstrumentor().instrument()
17
18# Now use the Cohere SDK as usual
19co = cohere.Client("YOUR_API_KEY")
20
21response = co.chat(
22    message="What is the capital of France?",
23    model="command"
24)
25
26print(response.text)