UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-12-29 07:12:07 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

refac: autocompletion behaviour

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-11-30 13:10:47 -08:00
parent 51e344f5b2
commit 5b879a2121
1 changed files with 22 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/lib/components/common/RichTextInput/AutoCompletion.js
View File

@ -66,25 +66,27 @@ export const AIAutocompletion = Extension.create({
const node = $head.parent const node = $head.parent
if (event.key === 'Tab') { if (event.key === 'Tab') {
if (!node.attrs['data-suggestion']) { // if (!node.attrs['data-suggestion']) {
// Generate completion // // Generate completion
if (loading) return true // if (loading) return true
loading = true // loading = true
const prompt = node.textContent // const prompt = node.textContent
this.options.generateCompletion(prompt).then(suggestion => { // this.options.generateCompletion(prompt).then(suggestion => {
if (suggestion && suggestion.trim() !== '') { // if (suggestion && suggestion.trim() !== '') {
dispatch(state.tr.setNodeMarkup($head.before(), null, { // dispatch(state.tr.setNodeMarkup($head.before(), null, {
...node.attrs, // ...node.attrs,
class: 'ai-autocompletion', // class: 'ai-autocompletion',
'data-prompt': prompt, // 'data-prompt': prompt,
'data-suggestion': suggestion, // 'data-suggestion': suggestion,
})) // }))
} // }
// If suggestion is empty or null, do nothing // // If suggestion is empty or null, do nothing
}).finally(() => { // }).finally(() => {
loading = false // loading = false
}) // })
} else { // }
if (node.attrs['data-suggestion']) {
// Accept suggestion // Accept suggestion
const suggestion = node.attrs['data-suggestion'] const suggestion = node.attrs['data-suggestion']
dispatch(state.tr dispatch(state.tr
@ -96,8 +98,8 @@ export const AIAutocompletion = Extension.create({
'data-suggestion': null, 'data-suggestion': null,
}) })
) )
}
return true return true
}
} else { } else {
if (node.attrs['data-suggestion']) { if (node.attrs['data-suggestion']) {