UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-24 21:13:59 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

refac: feedback optimisation

Browse Source
This commit is contained in:
Timothy J. Baek 2024-10-23 20:27:32 -07:00
parent b5d7e57d3c
commit ec37d801be
2 changed files with 55 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
src/lib/components/chat/Messages/RateComment.svelte
View File

@ -12,34 +12,28 @@
export let message; export let message;
export let show = false; export let show = false;
let LIKE_REASONS = []; let LIKE_REASONS = [
let DISLIKE_REASONS = []; 'accurate_information',
'followed_instructions_perfectly',
'showcased_creativity',
'positive_attitude',
'attention_to_detail',
'thorough_explanation',
'other'
];
let DISLIKE_REASONS = [
'dont_like_the_style',
'too_verbose',
'not_helpful',
'not_factually_correct',
'didnt_fully_follow_instructions',
'refused_when_it_shouldnt_have',
'being_lazy',
'other'
];
let tags = []; let tags = [];
function loadReasons() {
LIKE_REASONS = [
'accurate_information',
'followed_instructions_perfectly',
'showcased_creativity',
'positive_attitude',
'attention_to_detail',
'thorough_explanation',
'other'
];
DISLIKE_REASONS = [
'dont_like_the_style',
'too_verbose',
'not_helpful',
'not_factually_correct',
'didnt_fully_follow_instructions',
'refused_when_it_shouldnt_have',
'being_lazy',
'other'
];
}
let reasons = []; let reasons = [];
let selectedReason = null; let selectedReason = null;
let comment = ''; let comment = '';
@ -52,13 +46,19 @@
reasons = DISLIKE_REASONS; reasons = DISLIKE_REASONS;
} }
onMount(() => { $: if (message) {
init();
}
const init = () => {
selectedReason = message?.annotation?.reason ?? ''; selectedReason = message?.annotation?.reason ?? '';
comment = message?.annotation?.comment ?? ''; comment = message?.annotation?.comment ?? '';
tags = (message?.annotation?.tags ?? []).map((tag) => ({ tags = (message?.annotation?.tags ?? []).map((tag) => ({
name: tag name: tag
})); }));
};
onMount(() => {
if (message?.arena) { if (message?.arena) {
selectedModel = $models.find((m) => m.id === message.selectedModelId); selectedModel = $models.find((m) => m.id === message.selectedModelId);
toast.success( toast.success(
@ -67,8 +67,6 @@
}) })
); );
} }
loadReasons();
}); });
const saveHandler = () => { const saveHandler = () => {

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

@ -335,10 +335,13 @@
generatingImage = false; generatingImage = false;
}; };
let feedbackLoading = false;
const feedbackHandler = async ( const feedbackHandler = async (
rating: number | null = null, rating: number | null = null,
annotation: object | null = null annotation: object | null = null
) => { ) => {
feedbackLoading = true;
console.log('Feedback', rating, annotation); console.log('Feedback', rating, annotation);
const updatedMessage = { const updatedMessage = {
@ -357,24 +360,6 @@
return; return;
} }
if (!annotation) {
const messages = createMessagesList(history, message.id);
const tags = await generateTags(
localStorage.token,
message?.selectedModelId ?? message.model,
messages,
chatId
).catch((error) => {
console.error(error);
return [];
});
console.log(tags);
if (tags) {
updatedMessage.annotation.tags = tags;
}
}
let feedbackItem = { let feedbackItem = {
type: 'rating', type: 'rating',
data: { data: {
@ -425,7 +410,29 @@
if (!annotation) { if (!annotation) {
showRateComment = true; showRateComment = true;
if (!updatedMessage.annotation?.tags) {
// attempt to generate tags
const messages = createMessagesList(history, message.id);
const tags = await generateTags(
localStorage.token,
message?.selectedModelId ?? message.model,
messages,
chatId
).catch((error) => {
console.error(error);
return [];
});
console.log(tags);
if (tags) {
updatedMessage.annotation.tags = tags;
dispatch('save', updatedMessage);
}
}
} }
feedbackLoading = false;
}; };
$: if (!edit) { $: if (!edit) {
@ -982,7 +989,8 @@
message?.annotation?.rating ?? '' message?.annotation?.rating ?? ''
).toString() === '1' ).toString() === '1'
? 'bg-gray-100 dark:bg-gray-800' ? 'bg-gray-100 dark:bg-gray-800'
: ''} dark:hover:text-white hover:text-black transition" : ''} dark:hover:text-white hover:text-black transition disabled:cursor-progress disabled:hover:bg-transparent"
disabled={feedbackLoading}
on:click={async () => { on:click={async () => {
await feedbackHandler(1); await feedbackHandler(1);
@ -1032,7 +1040,8 @@
message?.annotation?.rating ?? '' message?.annotation?.rating ?? ''
).toString() === '-1' ).toString() === '-1'
? 'bg-gray-100 dark:bg-gray-800' ? 'bg-gray-100 dark:bg-gray-800'
: ''} dark:hover:text-white hover:text-black transition" : ''} dark:hover:text-white hover:text-black transition disabled:cursor-progress disabled:hover:bg-transparent"
disabled={feedbackLoading}
on:click={async () => { on:click={async () => {
await feedbackHandler(-1); await feedbackHandler(-1);