Google ADK Integration

THSP-based safety guardrails for Google ADK agents and multi-agent systems.

Installation

pip install google-adk sentinelseed
export GOOGLE_API_KEY="your-api-key"

Approaches

1. Plugin-based (Recommended): Global guardrails for all agents in a Runner

2. Callback-based: Add guardrails to specific agents

Quick Start

Plugin-based (Multi-Agent)

from google.adk.runners import Runner
from google.adk.agents import LlmAgent
from google.adk.sessions import InMemorySessionService
from sentinelseed.integrations.google_adk import SentinelPlugin

agent = LlmAgent(
    name="Assistant",
    model="gemini-2.0-flash",
    instruction="You are a helpful assistant.",
)

plugin = SentinelPlugin(
    seed_level="standard",
    block_on_failure=True,
)

session_service = InMemorySessionService()
runner = Runner(
    app_name="my_app",
    agent=agent,
    plugins=[plugin],
    session_service=session_service,
)

response = await runner.run("Hello, how can you help?")

Callback-based (Individual Agents)

from google.adk.agents import LlmAgent
from sentinelseed.integrations.google_adk import create_sentinel_callbacks

callbacks = create_sentinel_callbacks(seed_level="standard")

agent = LlmAgent(
    name="SafeAgent",
    model="gemini-2.0-flash",
    instruction="You are a helpful assistant.",
    **callbacks,
)

Configuration

SentinelPlugin

SentinelPlugin(
    seed_level="standard",
    block_on_failure=True,
    fail_closed=False,
    validate_inputs=True,
    validate_outputs=True,
    validate_tools=True,
    max_text_size=100000,
    validation_timeout=5.0,
    log_violations=True,
)

Validation Points

Monitoring

stats = plugin.get_stats()
print(f"Total: {stats['total_validations']}")
print(f"Blocked: {stats['blocked_count']}")

violations = plugin.get_violations()
for v in violations:
    print(f"[{v['risk_level']}] {v['concerns']}")

Multi-Agent Systems

plugin = SentinelPlugin(seed_level="standard")

agent1 = LlmAgent(name="Agent1", model="gemini-2.0-flash")
agent2 = LlmAgent(name="Agent2", model="gemini-2.0-flash")

workflow = SequentialAgent(name="Workflow", sub_agents=[agent1, agent2])

runner = Runner(
    app_name="my_app",
    agent=workflow,
    plugins=[plugin],  # Applies to all agents
    session_service=session_service,
)

Links

• Google ADK Documentation