OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-11-28 23:24:16 -08:00
parent 95000c7b15
commit fa5e1f7452
2 changed files with 40 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/lib/components/common/RichTextInput.svelte
View File

@ -153,10 +153,10 @@
Placeholder.configure({ placeholder }), Placeholder.configure({ placeholder }),
AIAutocompletion.configure({ AIAutocompletion.configure({
generateCompletion: async (text) => { generateCompletion: async (text) => {
// Implement your AI text generation logic here if (text.trim().length === 0) {
// This should return a Promise that resolves to the suggested text return null;
}
console.log(text);
return 'AI-generated suggestion'; return 'AI-generated suggestion';
} }
}) })

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

@ -50,8 +50,6 @@ export const AIAutocompletion = Extension.create({
key: new PluginKey('aiAutocompletion'), key: new PluginKey('aiAutocompletion'),
props: { props: {
handleKeyDown: (view, event) => { handleKeyDown: (view, event) => {
if (event.key !== 'Tab') return false
const { state, dispatch } = view const { state, dispatch } = view
const { selection } = state const { selection } = state
const { $head } = selection const { $head } = selection
@ -59,10 +57,11 @@ export const AIAutocompletion = Extension.create({
if ($head.parent.type.name !== 'paragraph') return false if ($head.parent.type.name !== 'paragraph') return false
const node = $head.parent const node = $head.parent
const prompt = node.textContent
if (event.key === 'Tab') {
if (!node.attrs['data-suggestion']) { if (!node.attrs['data-suggestion']) {
// Generate completion // Generate completion
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, {
@ -72,6 +71,7 @@ export const AIAutocompletion = Extension.create({
'data-suggestion': suggestion, 'data-suggestion': suggestion,
})) }))
} }
// If suggestion is empty or null, do nothing
}) })
} else { } else {
// Accept suggestion // Accept suggestion
@ -86,8 +86,18 @@ export const AIAutocompletion = Extension.create({
}) })
) )
} }
return true return true
} else if (node.attrs['data-suggestion']) {
// Reset suggestion on any other key press
dispatch(state.tr.setNodeMarkup($head.before(), null, {
...node.attrs,
class: null,
'data-prompt': null,
'data-suggestion': null,
}))
}
return false
}, },
}, },
}), }),