Agent Connector

Connect your agents to Amplitude

Production-ready access to your Amplitude data with managed auth, parametric control, and the reliability your agents need to actually work.

Installation & Usage

Get start with the Amplitude connector in minutes.

[01]

Install Package

Using uv or pip

BASH

Copy
uv pip install airbyte-agent-amplitude

[02]

Import

Initialize and use

PYTHON

Copy
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.
>"
    )
)

[03]

Tool

Add tools to your agent

python

Copy
@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 {})

Supported Entities & Actions

Access all your Amplitude data through a unified API

Lorem ipsum

Lorem ipsum

| Entity | Actions |\n|--------|---------|\n| Annotations | [List](./REFERENCE.md#annotations-list), [Get](./REFERENCE.md#annotations-get), [Search](./REFERENCE.md#annotations-search) |\n| Cohorts | [List](./REFERENCE.md#cohorts-list), [Get](./REFERENCE.md#cohorts-get), [Search](./REFERENCE.md#cohorts-search) |\n| Events List | [List](./REFERENCE.md#events-list-list), [Search](./REFERENCE.md#events-list-search) |\n| Active Users | [List](./REFERENCE.md#active-users-list), [Search](./REFERENCE.md#active-users-search) |\n| Average Session Length | [List](./REFERENCE.md#average-session-length-list), [Search](./REFERENCE.md#average-session-length-search) |

Example Prompts

The Amplitude connector is optimized to handle prompts like these

Lorem ipsum

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

Why Airbyte for AI Agents?

Built for production AI workloads with enterprise-grade reliability

Secure Authentication

Built-in OAuth 2.0 handling with automatic token refresh. No hard-coded credentials.

Agent-Native Design
Heading

Structured, LLM-friendly schemas optimized for AI agent consumption with natural language query support.

Production Ready

Battle-tested connectors with comprehensive error handling, logging, and retry logic.

Open Source

Fully open source under the MIT license. Contribute, customize, and extend freely.

Works with your favorite frameworks

Use the Salesforce connector with any AI agent framework.

LangChain

CrewAI

LlamaIndex

AutoGen

OpenAI Agents SDK

Claude Agents SDK

Frequently Asked Questions

Didn't find your answer? Please don't hesitate to reach out.

How do I authenticate with Amplitude?

The Amplitude connector supports HTTP Basic authentication using your Amplitude project API key and secret key. These can be found in Settings > Projects in your Amplitude account. In hosted mode, credentials are stored securely in Airbyte Cloud and you authenticate using your Airbyte client ID and client secret instead.

Can I use this connector with any AI agent framework?

The connector is compatible with any Python-based AI agent framework including LangChain, LlamaIndex, CrewAI, Pydantic AI, and custom implementations.

Does this connector support write operations?

Currently, the Amplitude connector focuses on read operations only. Write operations such as creating annotations, deleting cohorts, or exporting raw event data are not supported. Write support may be added in future versions.

How is this different from the Airbyte data connector?

Agent connectors are specifically designed for AI agents and LLM applications. They provide natural language interfaces, optimized response formats, and seamless integration with agent frameworks, unlike traditional ETL-focused connectors.

Ready to connect your AI agents to Amplitude?

Get started in minutes with our open-source connector.