How to integrate Arize AX MCP with Atomic Agent

Connect Atomic Agent to Arize AX MCP. Summarize failed spans from latest traces, compare experiment results across prompt versions, and more using natural language, with authentication handled for you.

Arize AX logoArize AX
Api Key

Arize AX is an AI engineering platform for tracing, evaluating, and improving AI applications. Use it to debug LLM behavior, compare experiments, and improve production AI quality.

24 Tools

Introduction

Atomic Agent is an open-source AI agent that runs on your machine and supports local models through llama.cpp. Connect it to 1,500+ apps through Composio, including Arize AX.

This guide uses Atomic Agent's built-in Composio integration. You need Atomic Agent v0.5.6 or later, your own Composio API key, and a Arize AX account. See the Atomic Agent documentation for installation instructions.

Also integrate Arize AX with

How Atomic Agent connects to Arize AX

Atomic Agent connects to Composio's Tool Router over Streamable HTTP MCP. It discovers the tools needed for your request, and Composio handles account authentication.

Execution and account-connection tools follow Atomic Agent's configured approval policy. Review any approval prompt before allowing an action on your connected account.

Connect Arize AX to Atomic Agent

  1. Sign up or sign in to the Composio dashboard. Copy your API key.
  2. In Atomic Agent, open the Integrations tab and select Composio. Select API key, press e, paste the key, and press Enter. The tools become available immediately, without a restart.
  3. Ask Atomic Agent to do something in Arize AX. It searches Composio for the right tool.
  4. On first use, open the sign-in link returned in the chat. Connect your Arize AX account, then return to Atomic Agent to continue.

Headless setup

Add the following line to ~/.atomic-agent/.env, replacing YOUR_COMPOSIO_API_KEY with your key. Start Atomic Agent after saving the file.

bash
COMPOSIO_API_KEY=YOUR_COMPOSIO_API_KEY

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

The Arize AX MCP server is an implementation of the Model Context Protocol that connects your AI agent and assistants like Claude, Cursor, etc directly to your Arize AX account. It provides structured and secure access so your agent can perform Arize AX operations on your behalf.

Use Arize AX from Atomic Agent

With your account connected, ask Atomic Agent to work with Arize AX in natural language. You can connect more apps through the same Composio integration as you need them.

Visit the Atomic Agent website or inspect the Atomic Agent source code.

Troubleshooting

Why is my API key rejected?

Copy the key directly from the Composio dashboard. Atomic Agent requires ASCII characters because it sends the key in an HTTP header. Remove any extra characters introduced when copying.

Why are Composio tools unavailable after saving a key?

Check that you use Atomic Agent v0.5.6 or later and that Composio is enabled in your configuration. A "composio": { "enabled": false } setting disables the integration while keeping the saved key on disk.

Does Composio have a free plan?

The Hobby plan includes 100,000 tool calls per month. Composio-managed OAuth apps have a lower included allowance. See Composio pricing for the current limits that apply to your connection.

TOOLS

Supported Tools

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

Add Dataset Examples

Append arbitrary-field examples to an Arize dataset or a specified dataset version.

Add Experiment Runs

Append one or more runs to an existing Arize experiment.

Create Dataset

Create an Arize dataset in a space with an explicit initial array of examples.

Create Experiment

Create an Arize experiment with initial runs, associated with exactly one dataset or space.

Create Prompt

Create an Arize prompt and its initial version in a space.

Create Prompt Version

Create a new immutable version of an existing Arize prompt.

Get Dataset

Get one Arize dataset and its versions by dataset ID.

Get Experiment

Get one Arize experiment by its ID, without its experiment runs.

Get Project

Get one accessible Arize project by its ID.

Get Prompt

Get an Arize prompt with its current, explicitly selected, or labeled version.

Get Space

Get one accessible Arize space by its ID.

List Dataset Examples

List one page of examples from an Arize dataset or a specific dataset version, preserving user-defined example fields.

List Datasets

List one page of Arize datasets, optionally filtered by space or dataset name.

List Experiment Runs

List one page of runs and outputs for an Arize experiment, preserving additional run fields.

List Experiments

List one page of Arize experiments, optionally filtered by dataset, space, or name.

List Projects

List one page of accessible Arize projects, optionally filtered by space or project name.

List Prompts

List one page of Arize prompts, optionally filtered by space or prompt name.

List Prompt Versions

List one page of versions for an Arize prompt, newest first.

List Spaces

List one page of Arize spaces accessible to the connected API key, optionally filtered by organization or name.

List Spans

Read one bounded page of spans for an Arize project using optional time and filter constraints.

List Traces

Read one bounded page of traces and nested spans for an Arize project using optional time and filter constraints; nested spans may be truncated per trace.

Update Dataset

Rename an existing Arize dataset.

Update Dataset Examples

Update existing Arize dataset examples by ID, optionally creating a named dataset version.

Update Prompt

Update or clear an Arize prompt's description.

FAQ

Frequently asked questions

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

Yes, you can. Atomic Agent 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 Arize AX tools.

Yes, absolutely. You can configure which Arize AX 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 Arize AX data and credentials are handled as safely as possible.

Start with Arize AX.It takes 30 seconds.

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

Start building