How to integrate Amplitude MCP with Google ADK

This guide walks you through connecting Amplitude to Google ADK using the Composio tool router. By the end, you'll have a working Amplitude agent that can get daily active users for last month, generate funnel analysis for onboarding flow, list top events for premium users through natural language commands. This guide will help you understand how to give your Google ADK agent real control over a Amplitude account through Composio's Amplitude MCP server. Before we dive in, let's take a quick look at the key ideas and tools involved.

Amplitude logoAmplitude
Api Key

Amplitude is a digital analytics platform for product and behavioral data insights. It helps teams analyze user journeys and make data-driven decisions quickly.

54 Tools

Introduction

This guide walks you through connecting Amplitude to Google ADK using the Composio tool router. By the end, you'll have a working Amplitude agent that can get daily active users for last month, generate funnel analysis for onboarding flow, list top events for premium users through natural language commands.

This guide will help you understand how to give your Google ADK agent real control over a Amplitude account through Composio's Amplitude MCP server.

Before we dive in, let's take a quick look at the key ideas and tools involved.

Also integrate Amplitude with

TL;DR

Here's what you'll learn:
  • Get a Amplitude account set up and connected to Composio
  • Install the Google ADK and Composio packages
  • Create a Composio Tool Router session for Amplitude
  • Build an agent that connects to Amplitude through MCP
  • Interact with Amplitude using natural language

What is Google ADK?

Google ADK (Agents Development Kit) is Google's framework for building AI agents powered by Gemini models. It provides tools for creating agents that can use external services through the Model Context Protocol.

Key features include:

  • Gemini Integration: Native support for Google's Gemini models
  • MCP Toolset: Built-in support for Model Context Protocol tools
  • Streamable HTTP: Connect to external services through streamable HTTP
  • CLI and Web UI: Run agents via command line or web interface

What is the Amplitude MCP server, and what's possible with it?

The Amplitude MCP server is an implementation of the Model Context Protocol that connects your AI agent and assistants like Claude, Cursor, etc directly to your Amplitude account. It provides structured and secure access to your analytics platform, so your agent can perform actions like managing event types, organizing cohorts, updating user properties, and tracking event categories on your behalf.

  • Cohort and user management: Ask your agent to request, download, and check the status of specific user cohorts for advanced segmentation or analysis.
  • Event type and category administration: Effortlessly create, update, or delete event types and categories, keeping your analytics taxonomy organized and up to date.
  • User property updates: Direct your agent to set or modify user properties—like device information or location—without sending new events, making user profile management a breeze.
  • Comprehensive analytics lookup: Retrieve detailed information about event types and categories, enabling your agent to provide insights or answer analytics questions in real time.

What is the Composio tool router, and how does it fit here?

What is Composio SDK?

Composio's Composio SDK helps agents find the right tools for a task at runtime. You can plug in multiple toolkits (like Gmail, HubSpot, and GitHub), and the agent will identify the relevant app and action to complete multi-step workflows. This can reduce token usage and improve the reliability of tool calls. Read more here: Getting started with Composio SDK

The tool router generates a secure MCP URL that your agents can access to perform actions.

How the Composio SDK works

The Composio SDK follows a three-phase workflow:

  1. Discovery: Searches for tools matching your task and returns relevant toolkits with their details.
  2. Authentication: Checks for active connections. If missing, creates an auth config and returns a connection URL via Auth Link.
  3. Execution: Executes the action using the authenticated connection.

Step-by-step Guide

Step by step09 STEPS
1

Prerequisites

Before starting, make sure you have:
  • A Google API key for Gemini models
  • A Composio account and API key
  • Python 3.9 or later installed
  • Basic familiarity with Python
2

Getting API Keys for Google and Composio

Google API Key
  • Go to Google AI Studio and create an API key.
  • Copy the key and keep it safe. You will put this in GOOGLE_API_KEY.
Composio API Key and User ID
  • Log in to the Composio dashboard.
  • Go to Settings → API Keys and copy your Composio API key. Use this for COMPOSIO_API_KEY.
  • Decide on a stable user identifier to scope sessions, often your email or a user ID. Use this for COMPOSIO_USER_ID.
3

Install dependencies

bash
pip install google-adk composio python-dotenv

Inside your virtual environment, install the required packages.

What's happening:

  • google-adk is Google's Agents Development Kit
  • composio connects your agent to Amplitude via MCP
  • python-dotenv loads environment variables
4

Set up ADK project

bash
adk create my_agent

Set up a new Google ADK project.

What's happening:

  • This creates an agent folder with a root agent file and .env file
5

Set environment variables

bash
GOOGLE_API_KEY=your-google-api-key
COMPOSIO_API_KEY=your-composio-api-key
COMPOSIO_USER_ID=your-user-id-or-email

Save all your credentials in the .env file.

What's happening:

  • GOOGLE_API_KEY authenticates with Google's Gemini models
  • COMPOSIO_API_KEY authenticates with Composio
  • COMPOSIO_USER_ID identifies the user for session management
6

Import modules and validate environment

python
import os
import warnings

from composio import Composio
from dotenv import load_dotenv
from google.adk.agents.llm_agent import Agent
from google.adk.tools.mcp_tool.mcp_session_manager import StreamableHTTPConnectionParams
from google.adk.tools.mcp_tool.mcp_toolset import McpToolset

load_dotenv()

warnings.filterwarnings("ignore", message=".*BaseAuthenticatedTool.*")

GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
COMPOSIO_API_KEY = os.getenv("COMPOSIO_API_KEY")
COMPOSIO_USER_ID = os.getenv("COMPOSIO_USER_ID")

if not GOOGLE_API_KEY:
    raise ValueError("GOOGLE_API_KEY is not set in the environment.")
if not COMPOSIO_API_KEY:
    raise ValueError("COMPOSIO_API_KEY is not set in the environment.")
if not COMPOSIO_USER_ID:
    raise ValueError("COMPOSIO_USER_ID is not set in the environment.")
What's happening:
  • os reads environment variables
  • Composio is the main Composio SDK client
  • GoogleProvider declares that you are using Google ADK as the agent runtime
  • Agent is the Google ADK LLM agent class
  • McpToolset lets the ADK agent call MCP tools over HTTP
7

Create Composio client and Tool Router session

python
composio_client = Composio(api_key=COMPOSIO_API_KEY)

composio_session = composio_client.create(
    user_id=COMPOSIO_USER_ID,
    toolkits=["amplitude"],
)

COMPOSIO_MCP_URL = composio_session.mcp.url,
print(f"Composio MCP URL: {COMPOSIO_MCP_URL}")
What's happening:
  • Authenticates to Composio with your API key
  • Declares Google ADK as the provider
  • Spins up a short-lived MCP endpoint for your user and selected toolkit
  • Stores the MCP HTTP URL for the ADK MCP integration
8

Set up the McpToolset and create the Agent

python
composio_toolset = McpToolset(
    connection_params=StreamableHTTPConnectionParams(
        url=COMPOSIO_MCP_URL,
        headers={"x-api-key": COMPOSIO_API_KEY}
    )
)

root_agent = Agent(
    model="gemini-2.5-flash",
    name="composio_agent",
    description="An agent that uses Composio tools to perform actions.",
    instruction=(
        "You are a helpful assistant connected to Composio. "
        "You have the following tools available: "
        "COMPOSIO_SEARCH_TOOLS, COMPOSIO_MULTI_EXECUTE_TOOL, "
        "COMPOSIO_MANAGE_CONNECTIONS, COMPOSIO_REMOTE_BASH_TOOL, COMPOSIO_REMOTE_WORKBENCH. "
        "Use these tools to help users with Amplitude operations."
    ),
    tools=[composio_toolset],
)

print("\nAgent setup complete. You can now run this agent directly ;)")
What's happening:
  • Connects the ADK agent to the Composio MCP endpoint through McpToolset
  • Uses Gemini as the model powering the agent
  • Lists exact tool names in instruction to reduce misnamed tool calls
9

Run the agent

bash
# Run in CLI mode
adk run my_agent

# Or run in web UI mode
adk web

Execute the agent from the project root. The web command opens a web portal where you can chat with the agent.

What's happening:

  • adk run runs the agent in CLI mode
  • adk web . opens a web UI for interactive testing

Complete Code

Here's the complete code to get you started with Amplitude and Google ADK:

python
import os
import warnings

from composio import Composio
from composio_google import GoogleProvider
from dotenv import load_dotenv
from google.adk.agents.llm_agent import Agent
from google.adk.tools.mcp_tool.mcp_session_manager import StreamableHTTPConnectionParams
from google.adk.tools.mcp_tool.mcp_toolset import McpToolset

load_dotenv()
warnings.filterwarnings("ignore", message=".*BaseAuthenticatedTool.*")

GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
COMPOSIO_API_KEY = os.getenv("COMPOSIO_API_KEY")
COMPOSIO_USER_ID = os.getenv("COMPOSIO_USER_ID")

if not GOOGLE_API_KEY:
    raise ValueError("GOOGLE_API_KEY is not set in the environment.")
if not COMPOSIO_API_KEY:
    raise ValueError("COMPOSIO_API_KEY is not set in the environment.")
if not COMPOSIO_USER_ID:
    raise ValueError("COMPOSIO_USER_ID is not set in the environment.")

composio_client = Composio(api_key=COMPOSIO_API_KEY, provider=GoogleProvider())

composio_session = composio_client.create(
    user_id=COMPOSIO_USER_ID,
    toolkits=["amplitude"],
)

COMPOSIO_MCP_URL = composio_session.mcp.url


composio_toolset = McpToolset(
    connection_params=StreamableHTTPConnectionParams(
        url=COMPOSIO_MCP_URL,
        headers={"x-api-key": COMPOSIO_API_KEY}
    )
)

root_agent = Agent(
    model="gemini-2.5-flash",
    name="composio_agent",
    description="An agent that uses Composio tools to perform actions.",
    instruction=(
        "You are a helpful assistant connected to Composio. "
        "You have the following tools available: "
        "COMPOSIO_SEARCH_TOOLS, COMPOSIO_MULTI_EXECUTE_TOOL, "
        "COMPOSIO_MANAGE_CONNECTIONS, COMPOSIO_REMOTE_BASH_TOOL, COMPOSIO_REMOTE_WORKBENCH. "
        "Use these tools to help users with Amplitude operations."
    ),  
    tools=[composio_toolset],
)

print("\nAgent setup complete. You can now run this agent directly ;)")

Conclusion

You've successfully integrated Amplitude with the Google ADK through Composio's MCP Tool Router. Your agent can now interact with Amplitude using natural language commands.

Key takeaways:

  • The Tool Router approach dynamically routes requests to the appropriate Amplitude tools
  • Environment variables keep your credentials secure and separate from code
  • Clear agent instructions reduce tool calling errors
  • The ADK web UI provides an interactive interface for testing and development

You can extend this setup by adding more toolkits to the toolkits array in your session configuration.

TOOLS

Supported Tools

Every Amplitude action and event your agent gets out of the box.

Bulk Assign Annotations to Category

Tool to bulk assign multiple annotations to a category in Amplitude.

Cancel User Deletion

Cancel a pending user deletion request in Amplitude.

Check Amplitude Cohort Status

Check the status of a cohort export request.

Create Chart Annotation in Amplitude

Create a chart annotation in Amplitude to mark important dates.

Create Annotation Category

Tool to create an annotation category in Amplitude to organize annotations.

Create Amplitude Event Category

Create a new event category in Amplitude.

Create Amplitude Event Type

Create a new event type in Amplitude.

Create Amplitude Release

Create a release to document product changes.

Delete Amplitude Chart Annotation

Delete a chart annotation from Amplitude.

Delete Amplitude Annotation Category

Delete an annotation category from Amplitude.

Delete Amplitude Event Category

Delete an event category from Amplitude.

Delete Amplitude Event Type

Delete an event type from Amplitude.

Delete Amplitude Users

Submit user deletion requests for GDPR/CCPA compliance.

Download Amplitude Cohort File

Download the cohort file after request is complete.

Search Amplitude User

Search for users in Amplitude by canonical identifier (Amplitude ID, device ID, user ID, or user ID prefix).

Get Active or New Users

Get the number of active or new users for a date range with optional segmentation.

Get Amplitude Annotation

Get a single chart annotation by ID from Amplitude.

Get Amplitude Annotation Category

Get a single annotation category by ID from Amplitude.

Request Amplitude Cohort

Get a single cohort by ID and initiate download.

Get User Deletion Requests

Get the status of user deletion requests within a date range.

Get Amplitude Event Categories

Get event categories from Amplitude.

Get Amplitude Event Property

Get a specific event property from Amplitude taxonomy.

Get Event Segmentation Data

Get event segmentation data from Amplitude Analytics API.

Get Amplitude Event Type

Get a specific event type from Amplitude by name.

Get Amplitude Event Types

Get all event types from Amplitude.

Get Funnel Analysis Data

Get funnel analysis data showing user conversion through a sequence of events.

Get Real-time Active Users

Get real-time active users count from Amplitude.

Get User Retention Analysis

Get user retention analysis showing how users return over time after a starting action.

Get Revenue LTV Metrics

Get revenue lifetime value (LTV) metrics including ARPU, ARPPU, and total revenue.

Get Session Average Length

Get average session length (in seconds) for a specified date range from Amplitude.

Get Session Length Distribution

Tool to retrieve session length distribution data for a specified date range from Amplitude.

Get Sessions Per User from Amplitude

Tool to get average number of sessions per user for each day in a date range from Amplitude.

Get User Activity from Amplitude

Fetch a single user's profile summary and event stream by Amplitude ID.

Get User Composition by Property

Tool to get user composition breakdown by property (platform, version, country, etc.

Get User Mappings

Get the list of user mappings for provided user IDs.

Get Amplitude User Property

Get a specific user property from Amplitude taxonomy.

Update User Properties in Amplitude

Update user properties using Amplitude's Identify API.

List Amplitude Annotation Categories

List all annotation categories from Amplitude.

List Chart Annotations

Tool to get all chart annotations with optional filtering by category, chart, and date range.

List Amplitude Cohorts

List all discoverable cohorts for an Amplitude project.

List Amplitude Event Properties

Get all event properties from Amplitude, optionally filtered by event type or property name.

List Amplitude Events

Tool to get a list of all event types in your Amplitude project with current week's statistics.

List Amplitude User Properties

Tool to get all user properties in your Amplitude project.

Map Users in Amplitude

Map users with different user IDs together (alias/merge users) in Amplitude.

Restore Amplitude Event Type

Restore a deleted event type in Amplitude.

Send Events to Amplitude

Send events to Amplitude using the HTTP V2 API.

Set Group Properties in Amplitude

Set group properties for account-level reporting without sending an event.

Update Amplitude Chart Annotation

Tool to update an existing chart annotation in Amplitude.

Update Amplitude Annotation Category

Tool to update an annotation category in Amplitude.

Update Amplitude Cohort Membership

Incrementally update cohort membership by adding or removing IDs.

Update Amplitude Event Category

Update an existing event category in Amplitude.

Update Amplitude Event Type

Update an existing event type in Amplitude.

Batch Upload Events to Amplitude

Bulk upload events to Amplitude using the Batch Event Upload API.

Upload Amplitude Cohort

Generate a new cohort or update an existing cohort by uploading user IDs or Amplitude IDs.

FAQ

Frequently asked questions

With a standalone Amplitude MCP server, the agents and LLMs can only access a fixed set of Amplitude tools tied to that server. However, with the Composio Tool Router, agents can dynamically load tools from Amplitude and many other apps based on the task at hand, all through a single MCP endpoint.

Yes, you can. Google ADK fully supports MCP integration. You get structured tool calling, message history handling, and model orchestration while Tool Router takes care of discovering and serving the right Amplitude tools.

Yes, absolutely. You can configure which Amplitude scopes and actions are allowed when connecting your account to Composio. You can also bring your own OAuth credentials or API configuration so you keep full control over what the agent can do.

All sensitive data such as tokens, keys, and configuration is fully encrypted at rest and in transit. Composio is SOC 2 Type 2 compliant and follows strict security practices so your Amplitude data and credentials are handled as safely as possible.

Start with Amplitude.It takes 30 seconds.

Managed auth, hosted MCP servers, and every Amplitude tool your agent needs.Free to start.

Start building