UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-20 12:00:22 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

fix: chat pdf, txt export issue

Browse Source
This commit is contained in:
Timothy J. Baek 2024-09-26 20:59:25 +02:00
parent 15bd5ebd7b
commit 1d225dd804
6 changed files with 48 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/lib/apis/utils/index.ts
View File

@ -55,7 +55,7 @@ export const formatPythonCode = async (code: string) => {
return res;
};
export const downloadChatAsPDF = async (chat: object) => {
export const downloadChatAsPDF = async (title: string, messages: object[]) => {
let error = null;
const blob = await fetch(`${WEBUI_API_BASE_URL}/utils/pdf`, {
@ -64,8 +64,8 @@ export const downloadChatAsPDF = async (chat: object) => {
'Content-Type': 'application/json'
},
body: JSON.stringify({
title: chat.title,
messages: chat.messages
title: title,
messages: messages
})
})
.then(async (res) => {

2
src/lib/components/chat/Chat.svelte
View File

@ -493,6 +493,7 @@
if (!$temporaryChatEnabled) {
chat = await updateChatById(localStorage.token, chatId, {
models: selectedModels,
messages: messages,
history: history,
params: params,
files: chatFiles
@ -1808,6 +1809,7 @@
chat = await updateChatById(localStorage.token, _chatId, {
models: selectedModels,
history: history,
messages: createMessagesList(history.currentId),
params: params,
files: chatFiles
});

17
src/lib/components/chat/Messages.svelte
View File

@ -80,7 +80,8 @@
const updateChatHistory = async () => {
await tick();
await updateChatById(localStorage.token, chatId, {
history: history
history: history,
messages: messages
});
currentChatPage.set(1);
@ -380,9 +381,19 @@
{regenerateResponse}
{continueResponse}
{mergeResponses}
{updateChatHistory}
{chatActionHandler}
{readOnly}
on:action={async (e) => {
const message = history.messages[message.id];
if (typeof e.detail === 'string') {
await chatActionHandler(chatId, e.detail, message.model, message.id);
} else {
const { id, event } = e.detail;
await chatActionHandler(chatId, id, message.model, message.id, event);
}
}}
on:update={() => {
updateChatHistory();
}}
on:scroll={() => {
if (autoScroll) {
const element = document.getElementById('messages-container');

50
src/lib/components/chat/Messages/Message.svelte
View File

@ -11,7 +11,6 @@
import MultiResponseMessages from './MultiResponseMessages.svelte';
import ResponseMessage from './ResponseMessage.svelte';
import UserMessage from './UserMessage.svelte';
import { updateChatById } from '$lib/apis/chats';
export let chatId;
export let idx = 0;
@ -21,9 +20,6 @@
export let user;
export let updateChatHistory;
export let chatActionHandler;
export let showPreviousMessage;
export let showNextMessage;
@ -81,18 +77,10 @@
{continueResponse}
{regenerateResponse}
on:action={async (e) => {
console.log('action', e);
const message = history.messages[messageId];
if (typeof e.detail === 'string') {
await chatActionHandler(chatId, e.detail, message.model, message.id);
} else {
const { id, event } = e.detail;
await chatActionHandler(chatId, id, message.model, message.id, event);
}
dispatch('action', e.detail);
}}
on:update={async (e) => {
console.log('update', e);
updateChatHistory();
dispatch('update');
}}
on:save={async (e) => {
console.log('save', e);
@ -100,13 +88,9 @@
const message = e.detail;
if (message) {
history.messages[message.id] = message;
await updateChatById(localStorage.token, chatId, {
history: history
});
dispatch('update');
} else {
await updateChatById(localStorage.token, chatId, {
history: history
});
dispatch('update');
}
}}
{readOnly}
@ -123,40 +107,24 @@
{regenerateResponse}
{mergeResponses}
on:action={async (e) => {
console.log('action', e);
const message = history.messages[messageId];
if (typeof e.detail === 'string') {
await chatActionHandler(chatId, e.detail, message.model, message.id);
} else {
const { id, event } = e.detail;
await chatActionHandler(chatId, id, message.model, message.id, event);
}
dispatch('action', e.detail);
}}
on:update={async (e) => {
console.log('update', e);
updateChatHistory();
dispatch('update');
}}
on:save={async (e) => {
console.log('save', e);
const message = e.detail;
if (message) {
history.messages[message.id] = message;
await updateChatById(localStorage.token, chatId, {
history: history
});
dispatch('update');
} else {
await updateChatById(localStorage.token, chatId, {
history: history
});
dispatch('update');
}
}}
on:change={async () => {
await tick();
await updateChatById(localStorage.token, chatId, {
history: history
});
dispatch('update');
dispatch('scroll');
}}
{readOnly}

10
src/lib/components/layout/Navbar/Menu.svelte
View File

@ -6,7 +6,7 @@
const { saveAs } = fileSaver;
import { downloadChatAsPDF } from '$lib/apis/utils';
import { copyToClipboard } from '$lib/utils';
import { copyToClipboard, createMessagesList } from '$lib/utils';
import { showOverview, showControls, mobile } from '$lib/stores';
import { flyAndScale } from '$lib/utils/transitions';
@ -32,7 +32,9 @@
const getChatAsText = async () => {
const _chat = chat.chat;
const chatText = _chat.messages.reduce((a, message, i, arr) => {
const messages = createMessagesList(_chat.history, _chat.history.currentId);
const chatText = messages.reduce((a, message, i, arr) => {
return `${a}### ${message.role.toUpperCase()}\n${message.content}\n\n`;
}, '');
@ -51,9 +53,11 @@
const downloadPdf = async () => {
const _chat = chat.chat;
const messages = createMessagesList(_chat.history, _chat.history.currentId);
console.log('download', chat);
const blob = await downloadChatAsPDF(_chat);
const blob = await downloadChatAsPDF(_chat.title, messages);
// Create a URL for the blob
const url = window.URL.createObjectURL(blob);

13
src/lib/utils/index.ts
View File

@ -860,3 +860,16 @@ export const getWeekday = () => {
const weekdays = ['Sunday', 'Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday'];
return weekdays[date.getDay()];
};
export const createMessagesList = (history, messageId) => {
if (messageId === null) {
return [];
}
const message = history.messages[messageId];
if (message?.parentId) {
return [...createMessagesList(history, message.parentId), message];
} else {
return [message];
}
};