Google Gemini

# 1. Install required packages
# !pip install google-genai openlayer

# 2. Set the environment variables
import os

os.environ["GOOGLE_AI_API_KEY"] = "YOUR_GOOGLE_AI_API_KEY_HERE"
os.environ["OPENLAYER_API_KEY"] = "YOUR_OPENLAYER_API_KEY_HERE"
os.environ["OPENLAYER_INFERENCE_PIPELINE_ID"] = "YOUR_OPENLAYER_INFERENCE_PIPELINE_ID_HERE"

# 3. Call `init` to auto-instrument the installed LLM SDKs (Google Gen AI, etc.)
from openlayer.lib import init

init()

# 4. Create the client. Build it after `init()`, which is what makes it traced
from google import genai

client = genai.Client(api_key=os.environ["GOOGLE_AI_API_KEY"])  # auto-traced by Openlayer

# 5. From now on, every generation call with
# the `client` is traced by Openlayer. E.g.,
response = client.models.generate_content(
    model="gemini-2.5-flash",
    contents="How are you doing today?",
)

// 1. Install required packages
//   npm install @google/genai openlayer

// 2. Set the environment variables:
//   GOOGLE_AI_API_KEY
//   OPENLAYER_API_KEY
//   OPENLAYER_INFERENCE_PIPELINE_ID

// 3. Wrap the client with `traceGoogleGenAI`. It patches the client in place
// and hands it back, so there is no separate `init()` step in TypeScript.
import { GoogleGenAI } from "@google/genai";
import { traceGoogleGenAI } from "openlayer/lib/integrations/googleGenAiTracer";

const client = traceGoogleGenAI(
  new GoogleGenAI({ apiKey: process.env["GOOGLE_AI_API_KEY"] ?? "" }),
);

// 4. From now on, every generation call with
// the `client` is traced by Openlayer. E.g.,
const response = await client.models.generateContent({
  model: "gemini-2.5-flash",
  contents: "How are you doing today?",
});

client = genai.Client(  # auto-traced by Openlayer
    vertexai=True,
    project="YOUR_GCP_PROJECT",
    location="us-central1",
)

const client = traceGoogleGenAI(
  new GoogleGenAI({
    vertexai: true,
    project: "YOUR_GCP_PROJECT",
    location: "us-central1",
  }),
);