Skip to main content
ReplicateTools enables an Agent to generate media using the Replicate platform.

​
Prerequisites

export REPLICATE_API_KEY=***
export REPLICATE_API_TOKEN=***
Set both variables to the same value. The toolkit reads REPLICATE_API_KEY while the replicate client authenticates with REPLICATE_API_TOKEN. The following example requires the replicate and openai libraries. 
uv pip install -U replicate openai

​
Example

The following agent will use Replicate to generate images or videos requested by the user.
cookbook/91_tools/replicate_tools.py
from agno.agent import Agent
from agno.models.openai import OpenAIChat
from agno.tools.replicate import ReplicateTools

"""Create an agent specialized for Replicate AI content generation"""

image_agent = Agent(
    name="Image Generator Agent",
    model=OpenAIChat(id="gpt-4o"),
    tools=[ReplicateTools(model="luma/photon-flash", enable_generate_media=True)],
    description="You are an AI agent that can generate images using the Replicate API.",
    instructions=[
        "When the user asks you to create an image, use the `generate_media` tool to create the image.",
        "Return the URL as raw to the user.",
        "Don't convert image URL to markdown or anything else.",
    ],
    markdown=True,
)

image_agent.print_response("Generate an image of a horse in the dessert.")

​
Toolkit Params

ParameterTypeDefaultDescription
api_keystrNoneIf you want to manually supply the Replicate API key.
modelstrminimax/video-01The replicate model to use. Find out more on the Replicate platform.
enable_generate_mediaboolTrueEnable the generate_media functionality.
allboolFalseEnable all functionality.

​
Toolkit Functions

FunctionDescription
generate_mediaGenerate either an image or a video from a prompt. The output depends on the model.

​
Developer Resources