Installation & Usage
[01]
Install Package
Using uv or pip
BASH
uv pip install airbyte-agent-amplitude
[02]
Import
Initialize and use
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.
>"
)
)[03]
Tool
Add tools to your agent
python
@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
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
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
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.
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.
Get started in minutes with our open-source connector.