UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 05:24:02 +00:00
Code Issues Actions 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>
<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}
{#if groupedMessagesIdx[model] !== undefined && groupedMessages[model].messages.length > 0}
<!-- svelte-ignore a11y-no-static-element-interactions -->
<!-- svelte-ignore a11y-click-events-have-key-events -->
<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
].model === model
? 'outline-gray-200 dark:outline-gray-700 outline-2'
@ -133,8 +136,8 @@
{rateMessage}
{copyToClipboard}
{continueGeneration}
regenerateResponse={async (model) => {
regenerateResponse(model);
regenerateResponse={async (message) => {
regenerateResponse(message);
await tick();
groupedMessagesIdx[model] = groupedMessages[model].messages.length - 1;
}}

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

@ -895,7 +895,7 @@
? '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"
on:click={() => {
regenerateResponse(message.model);
regenerateResponse(message);
}}
>
<svg

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

@ -759,16 +759,14 @@
console.log('stopResponse');
};
const regenerateResponse = async (modelId) => {
const regenerateResponse = async (message) => {
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;
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');
};
const regenerateResponse = async (modelId = null) => {
const regenerateResponse = async (message) => {
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;
await sendPrompt(userPrompt, userMessage.id, modelId);
await sendPrompt(userPrompt, userMessage.id, message.model);
}
};