baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

desktop: capture text from last chunk of response

Browse source
This commit is contained in:
Jeffrey Morgan 2023-06-27 22:08:53 -04:00
parent ab828602f2
commit 4ef886b1d8
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
desktop/src/app.tsx
View file

@ -46,11 +46,12 @@ async function generate(prompt: string, model: string, callback: (res: string) =
for (const message of messages) { for (const message of messages) {
const choice = message.choices[0] const choice = message.choices[0]
callback(choice.text)
if (choice.finish_reason === 'stop') { if (choice.finish_reason === 'stop') {
break break
} }
callback(choice.text)
} }
} }