Amplitude

The Amplitude agent connector is a Python package that equips AI agents to interact with Amplitude through strongly typed, well-documented tools. It's ready to use directly in your Python app, in an agent framework, or exposed through an MCP.

Connector for the Amplitude Analytics API. Provides access to core analytics data including event exports, cohort definitions, chart annotations, event type listings, active user counts, and average session length metrics. Authentication uses HTTP Basic with your Amplitude API key and secret key.

Example questions

The Amplitude connector is optimized to handle prompts like these.

List all chart annotations in Amplitude
Show me all cohorts
List all event types
Which cohorts have more than 1000 users?
What are the most popular event types by total count?
Show me annotations created in the last month

Unsupported questions

The Amplitude connector isn't currently able to handle prompts like these.

Create a new annotation
Delete a cohort
Export raw event data

Installation

bash

uv pip install airbyte-agent-amplitude

Usage

Connectors can run in open source or hosted mode.

Open source

In open source mode, you provide API credentials directly to the connector.

python

from airbyte_agent_amplitude import AmplitudeConnector
from airbyte_agent_amplitude.models import AmplitudeAuthConfig

connector = AmplitudeConnector(
    auth_config=AmplitudeAuthConfig(
        api_key="<Your Amplitude project API key. Find it in Settings > Projects in your Amplitude account.
>",
        secret_key="<Your Amplitude project secret key. Find it in Settings > Projects in your Amplitude account.
>"
    )
)

@agent.tool_plain # assumes you're using Pydantic AI
@AmplitudeConnector.tool_utils
async def amplitude_execute(entity: str, action: str, params: dict | None = None):
    return await connector.execute(entity, action, params or {})

Hosted

In hosted mode, API credentials are stored securely in Airbyte Cloud. You provide your Airbyte credentials instead. If your Airbyte client can access multiple organizations, also set organization_id.

This example assumes you've already authenticated your connector with Airbyte. See Authentication to learn more about authenticating. If you need a step-by-step guide, see the hosted execution tutorial.

python

from airbyte_agent_amplitude import AmplitudeConnector, AirbyteAuthConfig

connector = AmplitudeConnector(
    auth_config=AirbyteAuthConfig(
        customer_name="<your_customer_name>",
        organization_id="<your_organization_id>",  # Optional for multi-org clients
        airbyte_client_id="<your-client-id>",
        airbyte_client_secret="<your-client-secret>"
    )
)

@agent.tool_plain # assumes you're using Pydantic AI
@AmplitudeConnector.tool_utils
async def amplitude_execute(entity: str, action: str, params: dict | None = None):
    return await connector.execute(entity, action, params or {})

Full documentation

Entities and actions

This connector supports the following entities and actions. For more details, see this connector's full reference documentation.

Entity	Actions
Annotations	List, Get, Search
Cohorts	List, Get, Search
Events List	List, Search
Active Users	List, Search
Average Session Length	List, Search

Authentication

For all authentication options, see the connector's authentication documentation.

Amplitude API docs

See the official Amplitude API reference.

Version information

Package version: 0.1.6
Connector version: 1.0.2
Generated with Connector SDK commit SHA: b541ca65d697dad0915d1b5b8d8c756cd18299a7
Changelog: View changelog