[PR #22166] [MERGED] fix: persist token usage data for non-streaming chat responses #65382

New Issue

GiteaMirror · 2026-05-06T11:10:41-05:00

GiteaMirror commented

2026-05-06 11:10:41 -05:00

📋 Pull Request Information

Original PR: https://github.com/open-webui/open-webui/pull/22166
Author: @Classic298
Created: 3/2/2026
Status: ✅ Merged
Merged: 3/7/2026
Merged by: @tjbck

Base: dev ← Head: fix/non-streaming-save-usage

📝 Commits (1)

da85e74 fix: persist token usage data for non-streaming chat responses

📊 Changes

1 file changed (+4 additions, -0 deletions)

View changed files

📝 backend/open_webui/utils/middleware.py (+4 -0)

📄 Description

The non-streaming response handler was saving assistant messages without their usage/token data. While the streaming handler correctly extracted and saved usage information, the non-streaming path discarded it entirely.

This caused assistant messages from non-streaming completions to have NULL usage in the chat_message table, making them invisible to the analytics token aggregation queries and contributing to the '0 tokens' display in Admin Panel Analytics.

Extract and normalize the usage data from the API response and include it in the database upsert, matching the pattern already used by the streaming handler.

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.

Note

Deleting the CLA section will lead to immediate closure of your PR and it will not be merged in.

_{🔄 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/22166 **Author:** [@Classic298](https://github.com/Classic298) **Created:** 3/2/2026 **Status:** ✅ Merged **Merged:** 3/7/2026 **Merged by:** [@tjbck](https://github.com/tjbck) **Base:** `dev` ← **Head:** `fix/non-streaming-save-usage` --- ### 📝 Commits (1) - [`da85e74`](https://github.com/open-webui/open-webui/commit/da85e740ce2faca86a72f7d83a38cd21dedfd6d8) fix: persist token usage data for non-streaming chat responses ### 📊 Changes **1 file changed** (+4 additions, -0 deletions) <details> <summary>View changed files</summary> 📝 `backend/open_webui/utils/middleware.py` (+4 -0) </details> ### 📄 Description The non-streaming response handler was saving assistant messages without their usage/token data. While the streaming handler correctly extracted and saved usage information, the non-streaming path discarded it entirely. This caused assistant messages from non-streaming completions to have NULL usage in the chat_message table, making them invisible to the analytics token aggregation queries and contributing to the '0 tokens' display in Admin Panel Analytics. Extract and normalize the usage data from the API response and include it in the database upsert, matching the pattern already used by the streaming handler. ### Contributor License Agreement  By submitting this pull request, I confirm that I have read and fully agree to the [Contributor License Agreement (CLA)](https://github.com/open-webui/open-webui/blob/main/CONTRIBUTOR_LICENSE_AGREEMENT), and I am providing my contributions under its terms. > [!NOTE] > Deleting the CLA section will lead to immediate closure of your PR and it will not be merged in. --- <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 11:10:41 -05:00

GiteaMirror closed this issue

2026-05-06 11:10:46 -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#65382