[PR #14466] [CLOSED] feat: Add Replicate image generation integration with 14+ models #62332

New Issue

GiteaMirror · 2026-05-06T06:26:01-05:00

GiteaMirror commented

2026-05-06 06:26:01 -05:00

📋 Pull Request Information

Original PR: https://github.com/open-webui/open-webui/pull/14466
Author: @Legorobotdude
Created: 5/29/2025
Status: ❌ Closed

Base: dev ← Head: feature/replicate-integration

📝 Commits (3)

761746f feat: Add Replicate image generation integration with 14+ models, hybrid loading, and robust error handling
92dc646 test: Add focused unit tests for Replicate image generation - Tests model fetching, validation, and fallback behavior
57ff7d7 docs: Update README to include Replicate in image generation providers

📊 Changes

8 files changed (+722 additions, -359 deletions)

View changed files

📝 README.md (+1 -1)
📝 backend/open_webui/config.py (+10 -0)
📝 backend/open_webui/main.py (+3 -0)
📝 backend/open_webui/routers/images.py (+563 -330)
➕ backend/test_replicate_unit.py (+89 -0)
📝 pyproject.toml (+1 -0)
📝 src/lib/components/admin/Settings/Images.svelte (+54 -27)
📝 src/lib/components/common/Image.svelte (+1 -1)

📄 Description

PR Description:
This pull request adds comprehensive support for Replicate as an image generation provider in Open WebUI, expanding the platform's image generation capabilities alongside existing providers like OpenAI DALL-E, Automatic1111, ComfyUI, and Gemini.
Key Features:
✅ Replicate API Integration: Full support for Replicate's image generation models with API token authentication
✅ 14+ Pre-loaded Models: Includes popular models like FLUX 1.1 Pro Ultra, FLUX Schnell, Stable Diffusion 3.5, and more
✅ Robust Fallback System: Gracefully falls back to cached models when API is unavailable
✅ Admin UI Integration: Complete settings interface in the Admin panel for model selection and configuration
✅ Error Handling: Comprehensive error handling with user-friendly messages
✅ Unit Tests: Full test coverage for model fetching and validation

Technical Implementation:
Backend integration in routers/images.py with dedicated Replicate handling
Frontend UI updates in Images.svelte with Replicate-specific controls
Configuration management through the existing config system
Middleware updates for proper URL handling

Pull Request Checklist

Note to first-time contributors: Please open a discussion post in Discussions and describe your changes before submitting a pull request.

Before submitting, make sure you've checked the following:

Target branch: Please verify that the pull request targets the dev branch.
Description: Provide a concise description of the changes made in this pull request.
Changelog: Ensure a changelog entry following the format of Keep a Changelog is added at the bottom of the PR description.
Documentation: Have you updated relevant documentation Open WebUI Docs, or other documentation sources?
Dependencies: Are there any new dependencies? Have you updated the dependency versions in the documentation?
Testing: Have you written and run sufficient tests to validate the changes?
Code review: Have you performed a self-review of your code, addressing any coding standard issues and ensuring adherence to the project's coding standards?
Prefix: To clearly categorize this pull request, prefix the pull request title using one of the following:
- BREAKING CHANGE: Significant changes that may affect compatibility
- build: Changes that affect the build system or external dependencies
- ci: Changes to our continuous integration processes or workflows
- chore: Refactor, cleanup, or other non-functional code changes
- docs: Documentation update or addition
- feat: Introduces a new feature or enhancement to the codebase
- fix: Bug fix or error correction
- i18n: Internationalization or localization changes
- perf: Performance improvement
- refactor: Code restructuring for better maintainability, readability, or scalability
- style: Changes that do not affect the meaning of the code (white space, formatting, missing semi-colons, etc.)
- test: Adding missing tests or correcting existing tests
- WIP: Work in progress, a temporary label for incomplete or ongoing work

Changelog Entry

Description

[Concisely describe the changes made in this pull request, including any relevant motivation and impact (e.g., fixing a bug, adding a feature, or improving performance)]

Added

[List any new features, functionalities, or additions]

Changed

[List any changes, updates, refactorings, or optimizations]

Deprecated

[List any deprecated functionality or features that have been removed]

Removed

[List any removed features, files, or functionalities]

Fixed

[List any fixes, corrections, or bug fixes]

Security

[List any new or updated security-related changes, including vulnerability fixes]

Breaking Changes

BREAKING CHANGE: [List any breaking changes affecting compatibility or functionality]

Additional Information

[Insert any additional context, notes, or explanations for the changes]
- [Reference any related issues, commits, or other relevant information]

Screenshots or Videos

[Attach any relevant screenshots or videos demonstrating the changes]

Contributor License Agreement

By submitting this pull request, I confirm that I have read and fully agree to the Contributor License Agreement (CLA), and I am providing my contributions under its terms.

_{🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.}

## 📋 Pull Request Information **Original PR:** https://github.com/open-webui/open-webui/pull/14466 **Author:** [@Legorobotdude](https://github.com/Legorobotdude) **Created:** 5/29/2025 **Status:** ❌ Closed **Base:** `dev` ← **Head:** `feature/replicate-integration` --- ### 📝 Commits (3) - [`761746f`](https://github.com/open-webui/open-webui/commit/761746f1cbea6c170887e607a0138553c6a0eda5) feat: Add Replicate image generation integration with 14+ models, hybrid loading, and robust error handling - [`92dc646`](https://github.com/open-webui/open-webui/commit/92dc6460a92cf89d234643e9bb8ec98f17fd96a8) test: Add focused unit tests for Replicate image generation - Tests model fetching, validation, and fallback behavior - [`57ff7d7`](https://github.com/open-webui/open-webui/commit/57ff7d72df6608fbc83d334a69585e931e53a3ac) docs: Update README to include Replicate in image generation providers ### 📊 Changes **8 files changed** (+722 additions, -359 deletions) <details> <summary>View changed files</summary> 📝 `README.md` (+1 -1) 📝 `backend/open_webui/config.py` (+10 -0) 📝 `backend/open_webui/main.py` (+3 -0) 📝 `backend/open_webui/routers/images.py` (+563 -330) ➕ `backend/test_replicate_unit.py` (+89 -0) 📝 `pyproject.toml` (+1 -0) 📝 `src/lib/components/admin/Settings/Images.svelte` (+54 -27) 📝 `src/lib/components/common/Image.svelte` (+1 -1) </details> ### 📄 Description PR Description: This pull request adds comprehensive support for Replicate as an image generation provider in Open WebUI, expanding the platform's image generation capabilities alongside existing providers like OpenAI DALL-E, Automatic1111, ComfyUI, and Gemini. Key Features: ✅ Replicate API Integration: Full support for Replicate's image generation models with API token authentication ✅ 14+ Pre-loaded Models: Includes popular models like FLUX 1.1 Pro Ultra, FLUX Schnell, Stable Diffusion 3.5, and more ✅ Robust Fallback System: Gracefully falls back to cached models when API is unavailable ✅ Admin UI Integration: Complete settings interface in the Admin panel for model selection and configuration ✅ Error Handling: Comprehensive error handling with user-friendly messages ✅ Unit Tests: Full test coverage for model fetching and validation Technical Implementation: Backend integration in routers/images.py with dedicated Replicate handling Frontend UI updates in Images.svelte with Replicate-specific controls Configuration management through the existing config system Middleware updates for proper URL handling # Pull Request Checklist ### Note to first-time contributors: Please open a discussion post in [Discussions](https://github.com/open-webui/open-webui/discussions) and describe your changes before submitting a pull request. **Before submitting, make sure you've checked the following:** - [x] **Target branch:** Please verify that the pull request targets the `dev` branch. - [x] **Description:** Provide a concise description of the changes made in this pull request. - [x] **Changelog:** Ensure a changelog entry following the format of [Keep a Changelog](https://keepachangelog.com/) is added at the bottom of the PR description. - [x] **Documentation:** Have you updated relevant documentation [Open WebUI Docs](https://github.com/open-webui/docs), or other documentation sources? - [x] **Dependencies:** Are there any new dependencies? Have you updated the dependency versions in the documentation? - [x] **Testing:** Have you written and run sufficient tests to validate the changes? - [x] **Code review:** Have you performed a self-review of your code, addressing any coding standard issues and ensuring adherence to the project's coding standards? - [x] **Prefix:** To clearly categorize this pull request, prefix the pull request title using one of the following: - **BREAKING CHANGE**: Significant changes that may affect compatibility - **build**: Changes that affect the build system or external dependencies - **ci**: Changes to our continuous integration processes or workflows - **chore**: Refactor, cleanup, or other non-functional code changes - **docs**: Documentation update or addition - **feat**: Introduces a new feature or enhancement to the codebase - **fix**: Bug fix or error correction - **i18n**: Internationalization or localization changes - **perf**: Performance improvement - **refactor**: Code restructuring for better maintainability, readability, or scalability - **style**: Changes that do not affect the meaning of the code (white space, formatting, missing semi-colons, etc.) - **test**: Adding missing tests or correcting existing tests - **WIP**: Work in progress, a temporary label for incomplete or ongoing work # Changelog Entry ### Description - [Concisely describe the changes made in this pull request, including any relevant motivation and impact (e.g., fixing a bug, adding a feature, or improving performance)] ### Added - [List any new features, functionalities, or additions] ### Changed - [List any changes, updates, refactorings, or optimizations] ### Deprecated - [List any deprecated functionality or features that have been removed] ### Removed - [List any removed features, files, or functionalities] ### Fixed - [List any fixes, corrections, or bug fixes] ### Security - [List any new or updated security-related changes, including vulnerability fixes] ### Breaking Changes - **BREAKING CHANGE**: [List any breaking changes affecting compatibility or functionality] --- ### Additional Information - [Insert any additional context, notes, or explanations for the changes] - [Reference any related issues, commits, or other relevant information] ### Screenshots or Videos - [Attach any relevant screenshots or videos demonstrating the changes] ### Contributor License Agreement By submitting this pull request, I confirm that I have read and fully agree to the [Contributor License Agreement (CLA)](/CONTRIBUTOR_LICENSE_AGREEMENT), and I am providing my contributions under its terms. --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>

GiteaMirror added the pull-request label 2026-05-06 06:26:01 -05:00

GiteaMirror closed this issue

2026-05-06 06:26:08 -05:00

Sign in to join this conversation.

Branches Tags

1 Participants

Notifications

Due Date

No due date set.

Dependencies

No dependencies set.

Reference: github-starred/open-webui#62332