bhetherman/open-webui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-08-24 05:40:53 -04:00
parent 16c2a9eda4
commit 978d257214
3 changed files with 34 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lib/components/chat/Chat.svelte
+5 -7
View File
@@ -61,7 +61,8 @@
processDetails,
removeAllDetails,
getCodeBlockContents,
displayFileHandler
displayFileHandler,
getUsageTokenCount
} from '$lib/utils';
import { AudioQueue } from '$lib/utils/audio';
import { createTemporaryChatId, isTemporaryChatId } from '$lib/utils/chatId';
@@ -287,13 +288,10 @@
for (let idx = activeMessages.length - 1; idx >= 0; idx -= 1) {
const usage = activeMessages[idx]?.usage ?? activeMessages[idx]?.info?.usage;
const inputTokens = usage?.input_tokens ?? usage?.prompt_tokens;
if (inputTokens) {
const usageTokens = getUsageTokenCount(usage);
if (usageTokens) {
hasUsageCheckpoint = true;
estimatedTokens =
Number(inputTokens || 0) +
Number(usage.output_tokens ?? usage.completion_tokens ?? 0) +
estimateMessagesTokens(activeMessages.slice(idx + 1));
estimatedTokens = usageTokens + estimateMessagesTokens(activeMessages.slice(idx + 1));
break;
}
}
src/lib/components/chat/MessageInput.svelte
+4 -6
View File
@@ -52,6 +52,7 @@
getCurrentDateTime,
getFormattedDate,
getFormattedTime,
getUsageTokenCount,
getUserPosition,
getUserTimezone,
getWeekday
@@ -482,13 +483,10 @@
for (let idx = activeMessages.length - 1; idx >= 0; idx -= 1) {
const usage = activeMessages[idx]?.usage ?? activeMessages[idx]?.info?.usage;
const inputTokens = usage?.input_tokens ?? usage?.prompt_tokens;
if (inputTokens) {
const usageTokens = getUsageTokenCount(usage);
if (usageTokens) {
hasUsageCheckpoint = true;
estimatedTokens =
Number(inputTokens || 0) +
Number(usage.output_tokens ?? usage.completion_tokens ?? 0) +
estimateMessagesTokens(activeMessages.slice(idx + 1));
estimatedTokens = usageTokens + estimateMessagesTokens(activeMessages.slice(idx + 1));
break;
}
}