How to integrate Replicate MCP with Atomic Agent

Connect Atomic Agent to Replicate MCP. Run stable diffusion to generate an image, list all your uploaded files on replicate, and more using natural language, with authentication handled for you.

Replicate logoReplicate
Api Key

Replicate is a cloud API platform for running AI models without managing infrastructure. Get scalable, production-ready access to the latest machine learning models in minutes.

31 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 Replicate.

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 Replicate account. See the Atomic Agent documentation for installation instructions.

Also integrate Replicate with

How Atomic Agent connects to Replicate

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 Replicate 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 Replicate. It searches Composio for the right tool.
  4. On first use, open the sign-in link returned in the chat. Connect your Replicate 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 Replicate MCP server, and what's possible with it?

The Replicate MCP server is an implementation of the Model Context Protocol that connects your AI agent and assistants like Claude, Cursor, etc directly to your Replicate account. It provides structured and secure access to your Replicate resources, so your agent can perform actions like running AI model predictions, managing files, browsing model collections, and retrieving model documentation on your behalf.

  • Run and manage AI model predictions: Easily instruct your agent to create, monitor, and manage predictions on any deployed Replicate model using custom input parameters.
  • Browse and discover model collections: Ask your agent to fetch and list available model collections or retrieve example predictions to explore what’s possible on Replicate.
  • Upload and organize files: Let your agent upload new files, list all stored files, or inspect file details to streamline your model workflows.
  • Access model metadata and documentation: Retrieve full model details, schemas, and markdown README docs for any model to help you choose and utilize the right model for your tasks.

Use Replicate from Atomic Agent

With your account connected, ask Atomic Agent to work with Replicate 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 Replicate action and event your agent gets out of the box.

Get Account Information

Tool to get authenticated account information.

Cancel Prediction

Tool to cancel a prediction that is still running.

Get model collection

Tool to get a specific collection of models by its slug.

List model collections

Tool to list all collections of models.

Create Model

Tool to create a new Replicate model with specified owner, name, visibility, and hardware.

Create Prediction

Tool to create a prediction for a Replicate Deployment.

Create Deployment

Tool to create a new deployment with specified model, version, hardware, and scaling parameters.

Delete Deployment

Tool to delete a deployment from your account.

Get Deployment Details

Tool to get deployment details by owner and name.

List deployments

Tool to list all deployments associated with the account.

Create File

Tool to create or upload a file to Replicate.

Delete File

Tool to delete a file by its ID.

Get File Details

Tool to get details of a file by its ID.

List Files

Tool to retrieve a paginated list of uploaded files.

Get Prediction

Tool to get the status and output of a prediction by its ID.

List Available Hardware

Tool to list available hardware SKUs for models and deployments.

List model examples

Tool to list example predictions for a specific model.

Get Model Details

Tool to get details of a specific model by owner and name.

List Public Models

Tool to list public models with pagination and sorting.

Create Model Prediction

Tool to create a prediction using an official Replicate model.

Get Model README

Tool to get the README content for a model in Markdown format.

Get Model Version

Tool to get a specific version of a model.

List Model Versions

Tool to list all versions of a specific model.

Create Prediction

Tool to create a prediction to run a model by version ID.

List All Predictions

Tool to list all predictions for the authenticated user or organization with pagination.

Search Models and Collections

Tool to search for models, collections, and docs using text queries (beta).

Cancel Training

Tool to cancel an ongoing training operation in Replicate.

Create Training Job

Tool to create a training job for a specific model version.

List Training Jobs

Tool to list all training jobs for the authenticated user or organization with pagination.

Update Model Metadata

Tool to update metadata for a model including description, URLs, and README.

Get Webhook Signing Secret

Tool to get the signing secret for the default webhook.

FAQ

Frequently asked questions

With a standalone Replicate MCP server, the agents and LLMs can only access a fixed set of Replicate tools tied to that server. However, with the Composio Tool Router, agents can dynamically load tools from Replicate 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 Replicate tools.

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

Start with Replicate.It takes 30 seconds.

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

Start building