openinference-instrumentation-google-genai

import os from openinference.instrumentation.google_genai import GoogleGenAIInstrumentor from arize.otel import register from google import genai # Configure your Google GenAI API key # It's recommended to set this as an environment variable # os.environ["GEMINI_API_KEY"] = "your_gemini_api_key" # Setup OpenTelemetry via Arize's convenience function # Replace with your actual space_id, api_key, and project_name if using Arize tracer_provider = register( space_id=os.environ.get('ARIZE_SPACE_ID', 'YOUR_ARIZE_SPACE_ID'), api_key=os.environ.get('ARIZE_API_KEY', 'YOUR_ARIZE_API_KEY'), project_name="my-genai-app", ) # Instrument the Google GenAI client GoogleGenAIInstrumentor().instrument(tracer_provider=tracer_provider) # Ensure GEMINI_API_KEY is set for google-genai to function if not os.environ.get("GEMINI_API_KEY"): print("Warning: GEMINI_API_KEY environment variable not set. Using a placeholder.") # This placeholder will likely cause authentication errors for actual API calls os.environ["GEMINI_API_KEY"] = "sk-your-gemini-api-key" def send_message_multi_turn() -> tuple[str, str]: client = genai.Client(api_key=os.environ["GEMINI_API_KEY"]) chat = client.chats.create(model="gemini-1.5-flash-001") response1 = chat.send_message("What is the capital of France?") response2 = chat.send_message("Why is the sky blue?") return response1.text or "", response2.text or "" if __name__ == "__main__": print("Starting GenAI chat...") resp1_text, resp2_text = send_message_multi_turn() print(f"Response 1: {resp1_text}") print(f"Response 2: {resp2_text}") print("Traces should now be available in your configured OpenTelemetry collector (e.g., Arize Phoenix).")