Google Agent Development Kit (ADK)

# 1. Set the environment variables
import os

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

# 2. Call `init` to auto-instrument the installed LLM SDKs BEFORE creating agents
from openlayer.lib import init

init()

# 3. Create your agents with tools and sub-agents
from google.adk.agents import Agent
from google.adk.runners import Runner

def get_weather(location: str) -> str:
    """Get weather for a location."""
    return f"Sunny and 72°F in {location}"

weather_agent = Agent(
    name="weather_agent",
    model="gemini-2.0-flash-exp",
    description="Provides weather information",
    instructions="You help users get weather information for any location.",
    tools=[get_weather]
)

main_agent = Agent(
    name="main_agent",
    model="gemini-2.0-flash-exp",
    instructions="You are a helpful assistant that can check the weather.",
    sub_agents=[weather_agent]
)

# 4. Run conversations with automatic tracing
async def run_conversation(user_input: str):
    runner = Runner(main_agent)

    async for event in runner.run_async(
        session_id="session-123",
        user_id="user-456",
        message=user_input
    ):
        if event.is_final_response():
            print(event.content)

# From now on, all agent conversations, handoffs, and tool calls
# are automatically traced by Openlayer
await run_conversation("What's the weather in San Francisco?")

from google.adk.telemetry.setup import maybe_set_otel_providers
from opentelemetry.instrumentation.google_genai import GoogleGenAiSdkInstrumentor

# Environment variables above must already be set.
maybe_set_otel_providers()
GoogleGenAiSdkInstrumentor().instrument()

from google.adk.agents import Agent
from google.adk.runners import Runner

def get_weather(location: str) -> str:
    """Get weather for a location."""
    return f"Sunny and 72°F in {location}"

agent = Agent(
    name="weather_agent",
    model="gemini-2.0-flash-exp",
    description="Provides weather information",
    instructions="You help users get weather information for any location.",
    tools=[get_weather],
)

async def run_conversation(user_input: str):
    runner = Runner(agent)
    async for event in runner.run_async(
        session_id="session-123",
        user_id="user-456",
        message=user_input,
    ):
        if event.is_final_response():
            print(event.content)

await run_conversation("What's the weather in San Francisco?")