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: autocomplete

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-11-30 23:46:29 -08:00
parent 2fac9b45cd
commit 746fe9ea16
1 changed files with 35 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -65,8 +65,6 @@ export const AIAutocompletion = Extension.create({
let touchStartX = 0;
let touchStartY = 0;
return [
new Plugin({
key: new PluginKey('aiAutocompletion'),
@ -153,14 +151,18 @@ export const AIAutocompletion = Extension.create({
.generateCompletion(prompt)
.then((suggestion) => {
if (suggestion && suggestion.trim() !== '') {
view.dispatch(
newState.tr.setNodeMarkup(currentPos, null, {
...newNode.attrs,
class: 'ai-autocompletion',
'data-prompt': prompt,
'data-suggestion': suggestion
})
);
if (
view.state.selection.$head.pos === view.state.selection.$head.end()
) {
view.dispatch(
newState.tr.setNodeMarkup(currentPos, null, {
...newNode.attrs,
class: 'ai-autocompletion',
'data-prompt': prompt,
'data-suggestion': suggestion
})
);
}
}
})
.finally(() => {
@ -209,10 +211,32 @@ export const AIAutocompletion = Extension.create({
}
return false;
},
// Add mousedown behavior
mouseup: (view, event) => {
const { state, dispatch } = view;
const { selection } = state;
const { $head } = selection;
const node = $head.parent;
mousedown: () => {
// Reset debounce timer on mouse click
clearTimeout(debounceTimer);
// If a suggestion exists and the cursor moves, remove the suggestion
if (
node.type.name === 'paragraph' &&
node.attrs['data-suggestion'] &&
view.state.selection.$head.pos !== view.state.selection.$head.end()
) {
dispatch(
state.tr.setNodeMarkup($head.before(), null, {
...node.attrs,
class: null,
'data-prompt': null,
'data-suggestion': null
})
);
}
return false;
}
}