Google ADK

Integrate AgentSuite with the Google Agent Development Kit (ADK) via the Virtue Gateway for full protection: MCP Guard, AgentGuard, and Access Control — with full session tracing and observability in the dashboard.

Installation

pip install agentsuite-sdk[adk]

How It Works

• adapter.toolset() — returns an MCPToolset to pass into Agent(tools=[...]).
• adapter.before_model_callback, adapter.before_tool_callback, adapter.after_tool_callback — wire into Agent(...) for session tracking.

Quickstart

from google.adk.agents import Agent
from agentsuite import GatewayClient

client = GatewayClient(url="...", api_key="sk-vai-...")
adapter = client.adk()

agent = Agent(
    name="my_agent",
    model="gemini-2.0-flash",
    tools=[adapter.toolset()],
    before_model_callback=adapter.before_model_callback,
    before_tool_callback=adapter.before_tool_callback,
    after_tool_callback=adapter.after_tool_callback,
)

Full runnable example: demo_adk.py

Example Output

The agent responds to the query and prints the session ID:

View the full session trace in the VirtueAgent dashboard (Observability → Sessions):

Environment Variables

Variable	Description
VIRTUE_GATEWAY_URL	Gateway MCP endpoint URL
VIRTUE_API_KEY	VirtueAI API key
GOOGLE_API_KEY	Google API key for Gemini models
ADK_MODEL	Model name (default: gemini-2.0-flash)