scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-08-28 05:27:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2026-08-25 14:39:25 -04:00
parent 8be4c5fa6a
commit 3374b21a7d
4 changed files with 79 additions and 54 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

122
src/lib/components/chat/Chat.svelte
View file

@ -312,9 +312,9 @@
$: contextUsage = getContextUsage() ?? (contextCompactionEnabled ? serverContextUsage : null);
$: embeddedHeaderTitle = embeddedTitle || $chatTitle || $i18n.t('Chat');
let selectedToolIds = [];
let selectedSkillIds = [];
let selectedFilterIds = [];
let selectedToolIds: string[] = [];
let selectedSkillIds: string[] = [];
let selectedFilterIds: string[] = [];
let pendingOAuthTools = [];
let imageGenerationEnabled = false;
@ -408,7 +408,7 @@
// Chat Input
let prompt = '';
let chatFiles = [];
let files = [];
let files: any[] = [];
let params = {};
let chatVariables = {};
let showChatVariablesModal = false;
@ -737,6 +737,31 @@
});
};
const restoreChatInput = async (storageChatInput: string | null) => {
if (!storageChatInput || $temporaryChatEnabled) {
return false;
}
try {
const input = JSON.parse(storageChatInput);
prompt = input.prompt ?? '';
messageInput?.setText(prompt);
files = input.files ?? [];
selectedToolIds = input.selectedToolIds ?? [];
selectedSkillIds = input.selectedSkillIds ?? [];
selectedFilterIds = input.selectedFilterIds ?? [];
webSearchEnabled = input.webSearchEnabled ?? false;
imageGenerationEnabled = input.imageGenerationEnabled ?? false;
codeInterpreterEnabled = input.codeInterpreterEnabled ?? false;
if (input.toolApprovalMode) {
await handleToolApprovalModeChange(input.toolApprovalMode);
}
return true;
} catch (e) {
return false;
}
};
const withSelectedText = (text: string) =>
embedded && selectedText?.trim()
? `${text}\n\nSelected note text for replace_note_content operations:\n${selectedText.trim()}`
@ -820,25 +845,7 @@
await processNextInQueue(chatIdProp);
}
if (storageChatInput) {
try {
const input = JSON.parse(storageChatInput);
if (!$temporaryChatEnabled) {
messageInput?.setText(input.prompt);
files = input.files;
selectedToolIds = input.selectedToolIds;
selectedSkillIds = input.selectedSkillIds ?? [];
selectedFilterIds = input.selectedFilterIds;
webSearchEnabled = input.webSearchEnabled;
imageGenerationEnabled = input.imageGenerationEnabled;
codeInterpreterEnabled = input.codeInterpreterEnabled;
if (input.toolApprovalMode) {
handleToolApprovalModeChange(input.toolApprovalMode);
}
}
} catch (e) {}
} else {
if (!(await restoreChatInput(storageChatInput))) {
await setDefaults();
}
@ -941,9 +948,7 @@
return;
}
saveSessionSelectedModels();
await tick();
initiateOAuthRedirect(nextTool);
await oauthRedirectHandler(nextTool);
};
const resetInput = async () => {
@ -1590,23 +1595,7 @@
imageGenerationEnabled = false;
codeInterpreterEnabled = false;
try {
const input = JSON.parse(storageChatInput);
if (!$temporaryChatEnabled) {
messageInput?.setText(input.prompt);
files = input.files;
selectedToolIds = input.selectedToolIds;
selectedSkillIds = input.selectedSkillIds ?? [];
selectedFilterIds = input.selectedFilterIds;
webSearchEnabled = input.webSearchEnabled;
imageGenerationEnabled = input.imageGenerationEnabled;
codeInterpreterEnabled = input.codeInterpreterEnabled;
if (input.toolApprovalMode) {
handleToolApprovalModeChange(input.toolApprovalMode);
}
}
} catch (e) {}
await restoreChatInput(storageChatInput);
}
messageInput?.focus({ preventScroll: true });
@ -2125,6 +2114,7 @@
autoScroll = true;
await resetInput();
await restoreChatInput(sessionStorage.getItem('chat-input'));
await chatId.set('');
await chatTitle.set('');
@ -4007,23 +3997,52 @@
const MAX_DRAFT_LENGTH = 5000;
let saveDraftTimeout: ReturnType<typeof setTimeout> | null = null;
const saveDraft = async (draft: any, chatId: string | null = null) => {
const getChatInputDraft = () => ({
prompt,
files: files
.filter((file) => file.type !== 'image')
.map((file) => ({
...file,
user: undefined,
access_grants: undefined
})),
selectedToolIds,
selectedSkillIds,
selectedFilterIds,
imageGenerationEnabled,
webSearchEnabled,
codeInterpreterEnabled,
toolApprovalMode
});
const saveDraft = async (draft: any, chatId: string | null = null, debounce = true) => {
if (saveDraftTimeout) {
clearTimeout(saveDraftTimeout);
}
if (draft.prompt !== null && draft.prompt.length < MAX_DRAFT_LENGTH) {
saveDraftTimeout = setTimeout(async () => {
await sessionStorage.setItem(
`chat-input${chatId ? `-${chatId}` : ''}`,
JSON.stringify(draft)
);
}, 500);
const key = `chat-input${chatId ? `-${chatId}` : ''}`;
const write = () => sessionStorage.setItem(key, JSON.stringify(draft));
if (debounce) {
saveDraftTimeout = setTimeout(write, 500);
} else {
write();
}
} else {
sessionStorage.removeItem(`chat-input${chatId ? `-${chatId}` : ''}`);
}
};
const oauthRedirectHandler = async (tool: {
id: string;
serverId: string;
authType?: string | null;
}) => {
saveSessionSelectedModels();
await saveDraft(getChatInputDraft(), null, false);
initiateOAuthRedirect(tool);
};
const clearDraft = async (chatId: string | null = null) => {
if (saveDraftTimeout) {
clearTimeout(saveDraftTimeout);
@ -4406,6 +4425,7 @@
bind:imageGenerationEnabled
bind:codeInterpreterEnabled
{pendingOAuthTools}
{oauthRedirectHandler}
bind:webSearchEnabled
bind:atSelectedModel
bind:showCommands
@ -4497,6 +4517,7 @@
bind:imageGenerationEnabled
bind:codeInterpreterEnabled
{pendingOAuthTools}
{oauthRedirectHandler}
bind:webSearchEnabled
bind:atSelectedModel
bind:showCommands
@ -4557,6 +4578,7 @@
{toolApprovalMode}
onToolApprovalModeChange={handleToolApprovalModeChange}
{pendingOAuthTools}
{oauthRedirectHandler}
{stopResponse}
{createMessagePair}
{onSelect}

5
src/lib/components/chat/MessageInput.svelte
View file

@ -67,7 +67,6 @@
import { getSessionUser } from '$lib/apis/auths';
import { WEBUI_BASE_URL, WEBUI_API_BASE_URL, PASTED_TEXT_CHARACTER_LIMIT } from '$lib/constants';
import { initiateOAuthRedirect } from '$lib/apis/configs';
import { matchKeybinding, Shortcut } from '$lib/shortcuts';
import { createNoteHandler } from '../notes/utils';
@ -193,6 +192,7 @@
export let onToolApprovalModeChange: Function = () => {};
export let pendingOAuthTools = [];
export let oauthRedirectHandler: Function = () => {};
let showTerminalMenu = false;
@ -2259,6 +2259,7 @@
bind:webSearchEnabled
bind:imageGenerationEnabled
bind:codeInterpreterEnabled
{oauthRedirectHandler}
{onWebSearchToggle}
closeOnOutsideClick={integrationsMenuCloseOnOutsideClick}
onShowValves={(e) => {
@ -2478,7 +2479,7 @@
<Tooltip content={$i18n.t('Click to connect')} placement="top">
<button
on:click|preventDefault={() => {
initiateOAuthRedirect(pendingTool);
oauthRedirectHandler(pendingTool);
}}
type="button"
class="group px-2 py-[0.3125rem] flex gap-1.5 items-center text-xs rounded-full transition-colors duration-300 focus:outline-hidden max-w-full overflow-hidden

4
src/lib/components/chat/MessageInput/IntegrationsMenu.svelte
View file

@ -4,7 +4,6 @@
import { user, tools as _tools, skills as _skills, toolServers } from '$lib/stores';
import { initiateOAuthRedirect } from '$lib/apis/configs';
import { deleteOAuthSession } from '$lib/apis/auths';
import { getTools } from '$lib/apis/tools';
import { getSkills } from '$lib/apis/skills';
@ -46,6 +45,7 @@
export let selectedModels: string[] = [];
export let fileUploadCapableModels: string[] = [];
export let oauthRedirectHandler: Function = () => {};
export let toggleFilters: {
id: string;
@ -217,7 +217,7 @@
e.preventDefault();
const parts = toolId.split(':');
initiateOAuthRedirect({
oauthRedirectHandler({
id: toolId,
serverId: parts.at(-1) ?? toolId,
authType: parts.length > 1 ? (parts[0] === 'server' ? parts[1] : parts[0]) : null

2
src/lib/components/chat/Placeholder.svelte
View file

@ -56,6 +56,7 @@
export let webSearchEnabled = false;
export let toolApprovalMode = 'full';
export let onToolApprovalModeChange: Function = () => {};
export let oauthRedirectHandler: Function = () => {};
export let onUpload: Function = (e) => {};
export let onUpdate: (data?: { file?: any }) => void = () => {};
@ -246,6 +247,7 @@
bind:showCommands
bind:dragged
{pendingOAuthTools}
{oauthRedirectHandler}
{toolApprovalMode}
{onToolApprovalModeChange}
{stopResponse}