Replicate Automation

Automate your Replicate AI model workflows -- run predictions on any public model (image generation, LLMs, audio, video), upload input files, inspect model schemas and documentation, list model versions, and track prediction history.

Toolkit docs: composio.dev/toolkits/replicate

---

Setup

1. Add the Composio MCP server to your client: https://rube.app/mcp
2. Connect your Replicate account when prompted (API token authentication)
3. Start using the workflows below

---

Core Workflows

1. Get Model Details and Schema

Use REPLICATE_MODELS_GET to inspect a model's input/output schema before running predictions.

Tool: REPLICATE_MODELS_GET
Inputs:
  - model_owner: string (required) -- e.g., "meta", "black-forest-labs", "stability-ai"
  - model_name: string (required) -- e.g., "meta-llama-3-8b-instruct", "flux-1.1-pro"

Important: Each model has unique input keys and types. Always check the openapi_schema from this response before constructing prediction inputs.

2. Run a Prediction

Use REPLICATE_MODELS_PREDICTIONS_CREATE to run inference on any model with optional synchronous waiting and webhooks.

Tool: REPLICATE_MODELS_PREDICTIONS_CREATE
Inputs:
  - model_owner: string (required) -- e.g., "meta", "black-forest-labs"
  - model_name: string (required) -- e.g., "flux-1.1-pro", "sdxl"
  - input: object (required) -- model-specific inputs, e.g., { "prompt": "A sunset over mountains" }
  - wait_for: integer (1-60 seconds, optional) -- synchronous wait for completion
  - cancel_after: string (optional) -- max execution time, e.g., "300s", "5m"
  - webhook: string (optional) -- HTTPS URL for async completion notifications
  - webhook_events_filter: array (optional) -- ["start", "output", "logs", "completed"]