opentelemetry_instrumentation_google_genai_tracing_quickstart.py

1import os
2import google.generativeai as genai
3from opentelemetry import trace
4from opentelemetry.sdk.trace import TracerProvider
5from opentelemetry.sdk.trace.export import (
6    BatchSpanProcessor,
7    ConsoleSpanExporter,
8)
9from opentelemetry.instrumentation.google_genai import GoogleGenAIInstrumentor
10
11# Initialize OpenTelemetry
12provider = TracerProvider()
13processor = BatchSpanProcessor(ConsoleSpanExporter())
14provider.add_span_processor(processor)
15trace.set_tracer_provider(provider)
16
17# Instrument the Google GenAI library
18GoogleGenAIInstrumentor().instrument()
19
20# Configure the Google GenAI SDK
21genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
22
23# Use the library as usual
24model = genai.GenerativeModel("gemini-1.5-flash")
25response = model.generate_content("Write a poem about OpenTelemetry.")
26
27print(response.text)