tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages Activity Actions

fix: text completion replaced with chat completion

Browse source
This commit is contained in:
Timothy J. Baek 2024-03-02 13:14:14 -08:00
parent 2379e94cbf
commit 06ce5aed4a
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/routes/(app)/playground/+page.svelte
View file

@ -8,7 +8,7 @@
import { WEBUI_API_BASE_URL } from '$lib/constants'; import { WEBUI_API_BASE_URL } from '$lib/constants';
import { WEBUI_NAME, config, user, models } from '$lib/stores'; import { WEBUI_NAME, config, user, models } from '$lib/stores';
import { cancelChatCompletion, generateTextCompletion } from '$lib/apis/ollama'; import { cancelChatCompletion, generateChatCompletion } from '$lib/apis/ollama';
import { splitStream } from '$lib/utils'; import { splitStream } from '$lib/utils';
let mode = 'complete'; let mode = 'complete';
@ -44,7 +44,15 @@
if (selectedModel) { if (selectedModel) {
loading = true; loading = true;
const res = await generateTextCompletion(localStorage.token, selectedModel, text); const [res, controller] = await generateChatCompletion(localStorage.token, {
model: selectedModel,
messages: [
{
role: 'assistant',
content: text
}
]
});
if (res && res.ok) { if (res && res.ok) {
const reader = res.body const reader = res.body
@ -80,7 +88,7 @@
currentRequestId = data.id; currentRequestId = data.id;
} else { } else {
if (data.done == false) { if (data.done == false) {
text += data.response; text += data.message.content;
} else { } else {
console.log('done'); console.log('done');
} }