Google ADK Tracing

pip install openinference-instrumentation-google-adk google-adk arize-phoenix-otel

export GOOGLE_API_KEY=[your_key_here]

from phoenix.otel import register

# Configure the Phoenix tracer
tracer_provider = register(
  project_name="my-llm-app", # Default is 'default'
  auto_instrument=True # Auto-instrument your app based on installed OI dependencies
)

import asyncio

from google.adk.agents import Agent
from google.adk.runners import InMemoryRunner
from google.genai import types

def get_weather(city: str) -> dict:
    """Retrieves the current weather report for a specified city.

    Args:
        city (str): The name of the city for which to retrieve the weather report.

    Returns:
        dict: status and result or error msg.
    """
    if city.lower() == "new york":
        return {
            "status": "success",
            "report": (
                "The weather in New York is sunny with a temperature of 25 degrees"
                " Celsius (77 degrees Fahrenheit)."
            ),
        }
    else:
        return {
            "status": "error",
            "error_message": f"Weather information for '{city}' is not available.",
        }

agent = Agent(
   name="test_agent",
   model="gemini-2.0-flash-exp",
   description="Agent to answer questions using tools.",
   instruction="You must use the available tools to find an answer.",
   tools=[get_weather]
)

async def main():
    app_name = "test_instrumentation"
    user_id = "test_user"
    session_id = "test_session"
    runner = InMemoryRunner(agent=agent, app_name=app_name)
    session_service = runner.session_service
    await session_service.create_session(
        app_name=app_name,
        user_id=user_id,
        session_id=session_id
    )
    async for event in runner.run_async(
        user_id=user_id,
        session_id=session_id,
        new_message=types.Content(role="user", parts=[
            types.Part(text="What is the weather in New York?")]
        )
    ):
        if event.is_final_response():
            print(event.content.parts[0].text.strip())

if __name__ == "__main__":
    asyncio.run(main())

from vertexai import agent_engines

remote_agent = agent_engines.create(
    agent_engine=ModuleAgent(module_name="adk_agent", agent_name="app"),
    requirements=[
        "google-cloud-aiplatform[agent_engines,adk]",
        "arize-phoenix-otel",
        "openinference-instrumentation-google-adk",
    ],
    extra_packages=["adk_agent.py"],
    env_vars={
        "PHOENIX_COLLECTOR_ENDPOINT": "https://app.phoenix.arize.com/s/<handle>/v1/traces",  # Or your self-hosted Phoenix URL
        "PHOENIX_API_KEY": "<your-phoenix-api-key>",
    },
)

from phoenix.otel import register
from openinference.instrumentation.google_adk import GoogleADKInstrumentor

tracer_provider = register(
    project_name="adk-agent",
    batch=False,  # Use sync export because Agent Engine pauses CPU after requests
    set_global_tracer_provider=False,  # Required: avoids conflict with Agent Engine's global provider
)
GoogleADKInstrumentor().instrument(tracer_provider=tracer_provider)

# Your agent code here...