opentelemetry_google_generativeai_sdk_tracing_instrumentation_quickstart.py

1import google.generativeai as genai
2from opentelemetry import trace
3from opentelemetry.sdk.trace import TracerProvider
4from opentelemetry.sdk.trace.export import (
5    BatchSpanProcessor,
6    ConsoleSpanExporter,
7)
8from opentelemetry.instrumentation.google_generativeai import GoogleGenerativeAIInstrumentor
9
10# Initialize tracing
11provider = TracerProvider()
12processor = BatchSpanProcessor(ConsoleSpanExporter())
13provider.add_span_processor(processor)
14trace.set_tracer_provider(provider)
15
16# Instrument Google Generative AI
17GoogleGenerativeAIInstrumentor().instrument()
18
19# Standard Google Generative AI usage
20genai.configure(api_key="YOUR_GOOGLE_API_KEY")
21model = genai.GenerativeModel('gemini-pro')
22response = model.generate_content("Write a poem about open telemetry.")
23
24print(response.text)