tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages Activity Actions

fix: message after image generation

Browse source
This commit is contained in:
Timothy J. Baek 2024-03-12 13:43:49 -07:00
parent 7f78e58488
commit 5a567ce4d0
2 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/routes/(app)/+page.svelte
View file

@ -308,7 +308,7 @@
.map((file) => file.url.slice(file.url.indexOf(',') + 1)); .map((file) => file.url.slice(file.url.indexOf(',') + 1));
// Add images array only if it contains elements // Add images array only if it contains elements
if (imageUrls && imageUrls.length > 0) { if (imageUrls && imageUrls.length > 0 && message.role === 'user') {
baseMessage.images = imageUrls; baseMessage.images = imageUrls;
} }
@ -532,7 +532,8 @@
.filter((message) => message) .filter((message) => message)
.map((message, idx, arr) => ({ .map((message, idx, arr) => ({
role: message.role, role: message.role,
...(message.files?.filter((file) => file.type === 'image').length > 0 ?? false ...((message.files?.filter((file) => file.type === 'image').length > 0 ?? false) &&
message.role === 'user'
? { ? {
content: [ content: [
{ {

12
src/routes/(app)/c/[id]/+page.svelte
View file

@ -321,7 +321,7 @@
.map((file) => file.url.slice(file.url.indexOf(',') + 1)); .map((file) => file.url.slice(file.url.indexOf(',') + 1));
// Add images array only if it contains elements // Add images array only if it contains elements
if (imageUrls && imageUrls.length > 0) { if (imageUrls && imageUrls.length > 0 && message.role === 'user') {
baseMessage.images = imageUrls; baseMessage.images = imageUrls;
} }
@ -545,7 +545,8 @@
.filter((message) => message) .filter((message) => message)
.map((message, idx, arr) => ({ .map((message, idx, arr) => ({
role: message.role, role: message.role,
...(message.files?.filter((file) => file.type === 'image').length > 0 ?? false ...((message.files?.filter((file) => file.type === 'image').length > 0 ?? false) &&
message.role === 'user'
? { ? {
content: [ content: [
{ {
@ -688,7 +689,12 @@
if (messages.length == 2) { if (messages.length == 2) {
window.history.replaceState(history.state, '', `/c/${_chatId}`); window.history.replaceState(history.state, '', `/c/${_chatId}`);
await setChatTitle(_chatId, userPrompt);
if ($settings?.titleAutoGenerateModel) {
await generateChatTitle(_chatId, userPrompt);
} else {
await setChatTitle(_chatId, userPrompt);
}
} }
}; };