pipecat-replicate

Replicate image generation integration for Pipecat — a framework for building voice and multimodal conversational AI applications.

Pipecat Compatibility

Tested with Pipecat v0.0.108

Features

• Text-to-image generation using any Replicate-hosted model
• Official models (owner/name) and versioned community models (owner/name:version)
• Sync prediction requests with automatic polling fallback
• Returns URLImageRawFrame for direct use in Pipecat pipelines
• Configurable via the standard Pipecat Settings dataclass pattern

Installation

Using pip

pip install pipecat-replicate

Using uv

uv add pipecat-replicate

From source

git clone https://github.com/bnovik0v/pipecat-replicate.git
cd pipecat-replicate
pip install -e .

Quick Start

1. Get your Replicate API token at https://replicate.com/account/api-tokens

2. Set the environment variable:

export REPLICATE_API_TOKEN=r8_...

3. Use in a Pipecat pipeline:

import aiohttp
from pipecat.frames.frames import TextFrame
from pipecat.pipeline.pipeline import Pipeline
from pipecat.pipeline.runner import PipelineRunner
from pipecat.pipeline.task import PipelineTask
from pipecat_replicate import ReplicateImageGenService

async with aiohttp.ClientSession() as session:
    imagegen = ReplicateImageGenService(
        aiohttp_session=session,
        settings=ReplicateImageGenService.Settings(
            model="black-forest-labs/flux-schnell",
            aspect_ratio="1:1",
        ),
    )

    pipeline = Pipeline([imagegen, ...])
    task = PipelineTask(pipeline)
    await task.queue_frames([TextFrame("a cat in the style of a screenprint poster")])

    runner = PipelineRunner()
    await runner.run(task)

Configuration

Settings

Constructor Parameters

Examples

See examples/basic_image_gen.py for a complete example that generates an image and displays it in a Tk window.

REPLICATE_API_TOKEN=r8_... python examples/basic_image_gen.py

Running Tests

uv sync --group dev
uv run pytest

License

BSD 2-Clause — see LICENSE.