[PR #9088] [CLOSED] **feat** Support CoT models for title generation. #61171

Closed
opened 2026-05-06 04:33:08 -05:00 by GiteaMirror · 0 comments
Owner

📋 Pull Request Information

Original PR: https://github.com/open-webui/open-webui/pull/9088
Author: @joshuawu
Created: 1/29/2025
Status: Closed

Base: devHead: title-generation-cot


📝 Commits (1)

  • 666398f Support CoT models for title generation.

📊 Changes

6 files changed (+62 additions, -24 deletions)

View changed files

📝 backend/open_webui/config.py (+6 -0)
📝 backend/open_webui/main.py (+2 -0)
📝 backend/open_webui/models/chats.py (+3 -1)
📝 backend/open_webui/routers/tasks.py (+14 -7)
📝 backend/open_webui/utils/middleware.py (+29 -16)
📝 src/lib/components/admin/Settings/Interface.svelte (+8 -0)

📄 Description

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 for validating 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

  • Added a new setting to enable title generation by CoT models.
  • Added fallback logic for title generation by CoT models (when the setting is off).

Added

  • Title generation fallback behavior when using a CoT model
  • Interface setting "Title generation model uses chain of thought"

Changed

  • Refactored max_tokens/max_completion_tokens while adding the new behavior
  • Truncate titles to 50 characters immediately before title update
  • Refactored title fallback logic
  • Moved reasoning tags names to a constant

Deprecated

  • None

Removed

  • None

Fixed

  • Fixed titles being updated with CoT e.g. <think>...

Security

  • None

Breaking Changes

  • None

Additional Information

Currently, running chain-of-thought (CoT) models will generate unhelpful titles in the sidebar unless a different model is selected for title generation in the interface settings. For example:
Screenshot 2025-01-30 032209
This PR lets CoT models generate helpful titles by 1) adding a default behavior to fall back to the original assistant message if a CoT is detected and 2) through a new setting, letting a CoT model fully generate a message before extracting just the generated title.

Screenshots or Videos

Example new default behavior running a CoT model:

Screenshot 2025-01-30 042356

New interface setting "Title generation model uses chain of thought"

Screenshot 2025-01-30 043333
Screenshot 2025-01-30 043344

After enabling the setting:

Screenshot 2025-01-30 043547


🔄 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/9088 **Author:** [@joshuawu](https://github.com/joshuawu) **Created:** 1/29/2025 **Status:** ❌ Closed **Base:** `dev` ← **Head:** `title-generation-cot` --- ### 📝 Commits (1) - [`666398f`](https://github.com/open-webui/open-webui/commit/666398f3952fe8e1555e9e10223bc2a595ae110c) Support CoT models for title generation. ### 📊 Changes **6 files changed** (+62 additions, -24 deletions) <details> <summary>View changed files</summary> 📝 `backend/open_webui/config.py` (+6 -0) 📝 `backend/open_webui/main.py` (+2 -0) 📝 `backend/open_webui/models/chats.py` (+3 -1) 📝 `backend/open_webui/routers/tasks.py` (+14 -7) 📝 `backend/open_webui/utils/middleware.py` (+29 -16) 📝 `src/lib/components/admin/Settings/Interface.svelte` (+8 -0) </details> ### 📄 Description # 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. - [ ] **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? - [ ] **Testing:** Have you written and run sufficient tests for validating 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 - Added a new setting to enable title generation by CoT models. - Added fallback logic for title generation by CoT models (when the setting is off). ### Added - Title generation fallback behavior when using a CoT model - Interface setting "Title generation model uses chain of thought" ### Changed - Refactored `max_tokens`/`max_completion_tokens` while adding the new behavior - Truncate titles to 50 characters immediately before title update - Refactored title fallback logic - Moved reasoning tags names to a constant ### Deprecated - None ### Removed - None ### Fixed - Fixed titles being updated with CoT e.g. `<think>...` ### Security - None ### Breaking Changes - None --- ### Additional Information Currently, running chain-of-thought (CoT) models will generate unhelpful titles in the sidebar unless a different model is selected for title generation in the interface settings. For example: ![Screenshot 2025-01-30 032209](https://github.com/user-attachments/assets/59f315a8-8a1e-477c-a425-d38c9c26f388) This PR lets CoT models generate helpful titles by 1) adding a default behavior to fall back to the original assistant message if a CoT is detected and 2) through a new setting, letting a CoT model fully generate a message before extracting just the generated title. ### Screenshots or Videos ## Example new default behavior running a CoT model: ![Screenshot 2025-01-30 042356](https://github.com/user-attachments/assets/c9cd4b36-d758-4b3e-ae55-ee86a10260dd) ## New interface setting "Title generation model uses chain of thought" ![Screenshot 2025-01-30 043333](https://github.com/user-attachments/assets/1db0661a-ae72-413c-b89f-6ec54d1518f1) ![Screenshot 2025-01-30 043344](https://github.com/user-attachments/assets/4b605f55-9a3c-4d25-9907-615e11276aa2) ## After enabling the setting: ![Screenshot 2025-01-30 043547](https://github.com/user-attachments/assets/e0dce91f-7544-44e1-aba6-637fa7caa22f) --- <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 04:33:08 -05:00
Sign in to join this conversation.
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: github-starred/open-webui#61171