UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-12-28 06:42:47 +00:00
Code Issues Actions 13 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
if (event.key === 'Tab') {
if (!node.attrs['data-suggestion']) {
// Generate completion
if (loading) return true
loading = true
const prompt = node.textContent
this.options.generateCompletion(prompt).then(suggestion => {
if (suggestion && suggestion.trim() !== '') {
dispatch(state.tr.setNodeMarkup($head.before(), null, {
...node.attrs,
class: 'ai-autocompletion',
'data-prompt': prompt,
'data-suggestion': suggestion,
}))
}
// If suggestion is empty or null, do nothing
}).finally(() => {
loading = false
})
} else {
// if (!node.attrs['data-suggestion']) {
// // Generate completion
// if (loading) return true
// loading = true
// const prompt = node.textContent
// this.options.generateCompletion(prompt).then(suggestion => {
// if (suggestion && suggestion.trim() !== '') {
// dispatch(state.tr.setNodeMarkup($head.before(), null, {
// ...node.attrs,
// class: 'ai-autocompletion',
// 'data-prompt': prompt,
// 'data-suggestion': suggestion,
// }))
// }
// // If suggestion is empty or null, do nothing
// }).finally(() => {
// loading = false
// })
// }
if (node.attrs['data-suggestion']) {
// Accept suggestion
const suggestion = node.attrs['data-suggestion']
dispatch(state.tr
@ -96,8 +98,8 @@ export const AIAutocompletion = Extension.create({
'data-suggestion': null,
})
)
return true
}
return true
} else {
if (node.attrs['data-suggestion']) {