UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-05 05:10:46 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: many model regeneration

Browse Source
This commit is contained in:
Timothy J. Baek 2024-05-19 05:01:29 -07:00
parent 676a4dffd0
commit 68104358d0
4 changed files with 16 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/lib/components/chat/Messages/CompareMessages.svelte
View File

@ -58,14 +58,17 @@
</script> </script>
<div> <div>
<div class="flex snap-x snap-mandatory overflow-x-auto scrollbar-none"> <div
class="flex snap-x snap-mandatory overflow-x-auto scrollbar-none"
id="responses-container-{parentMessage.id}"
>
{#each Object.keys(groupedMessages) as model} {#each Object.keys(groupedMessages) as model}
{#if groupedMessagesIdx[model] !== undefined && groupedMessages[model].messages.length > 0} {#if groupedMessagesIdx[model] !== undefined && groupedMessages[model].messages.length > 0}
<!-- svelte-ignore a11y-no-static-element-interactions --> <!-- svelte-ignore a11y-no-static-element-interactions -->
<!-- svelte-ignore a11y-click-events-have-key-events --> <!-- svelte-ignore a11y-click-events-have-key-events -->
<div <div
class=" snap-center min-w-96 w-full max-w-full m-1 outline outline-1 {history.messages[ class=" snap-center min-w-80 w-full max-w-full m-1 outline outline-1 {history.messages[
currentMessageId currentMessageId
].model === model ].model === model
? 'outline-gray-200 dark:outline-gray-700 outline-2' ? 'outline-gray-200 dark:outline-gray-700 outline-2'
@ -133,8 +136,8 @@
{rateMessage} {rateMessage}
{copyToClipboard} {copyToClipboard}
{continueGeneration} {continueGeneration}
regenerateResponse={async (model) => { regenerateResponse={async (message) => {
regenerateResponse(model); regenerateResponse(message);
await tick(); await tick();
groupedMessagesIdx[model] = groupedMessages[model].messages.length - 1; groupedMessagesIdx[model] = groupedMessages[model].messages.length - 1;
}} }}

2
src/lib/components/chat/Messages/ResponseMessage.svelte
View File

@ -895,7 +895,7 @@
? 'visible' ? 'visible'
: 'invisible group-hover:visible'} p-1.5 hover:bg-black/5 dark:hover:bg-white/5 rounded-lg dark:hover:text-white hover:text-black transition regenerate-response-button" : 'invisible group-hover:visible'} p-1.5 hover:bg-black/5 dark:hover:bg-white/5 rounded-lg dark:hover:text-white hover:text-black transition regenerate-response-button"
on:click={() => { on:click={() => {
regenerateResponse(message.model); regenerateResponse(message);
}} }}
> >
<svg <svg

10
src/routes/(app)/+page.svelte
View File

@ -759,16 +759,14 @@
console.log('stopResponse'); console.log('stopResponse');
}; };
const regenerateResponse = async (modelId) => { const regenerateResponse = async (message) => {
console.log('regenerateResponse'); console.log('regenerateResponse');
if (messages.length != 0 && messages.at(-1).done == true) {
messages.splice(messages.length - 1, 1);
messages = messages;
let userMessage = messages.at(-1); if (messages.length != 0) {
let userMessage = history.messages[message.parentId];
let userPrompt = userMessage.content; let userPrompt = userMessage.content;
await sendPrompt(userPrompt, userMessage.id, modelId); await sendPrompt(userPrompt, userMessage.id, message.model);
} }
}; };

10
src/routes/(app)/c/[id]/+page.svelte
View File

@ -764,16 +764,14 @@
console.log('stopResponse'); console.log('stopResponse');
}; };
const regenerateResponse = async (modelId = null) => { const regenerateResponse = async (message) => {
console.log('regenerateResponse'); console.log('regenerateResponse');
if (messages.length != 0 && messages.at(-1).done == true) {
messages.splice(messages.length - 1, 1);
messages = messages;
let userMessage = messages.at(-1); if (messages.length != 0) {
let userMessage = history.messages[message.parentId];
let userPrompt = userMessage.content; let userPrompt = userMessage.content;
await sendPrompt(userPrompt, userMessage.id, modelId); await sendPrompt(userPrompt, userMessage.id, message.model);
} }
}; };