From 4ef886b1d840ecec750143ebc71b466b73efe080 Mon Sep 17 00:00:00 2001 From: Jeffrey Morgan Date: Tue, 27 Jun 2023 22:08:53 -0400 Subject: [PATCH] desktop: capture text from last chunk of response --- desktop/src/app.tsx | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/desktop/src/app.tsx b/desktop/src/app.tsx index 0a5e5bc7..380709f9 100644 --- a/desktop/src/app.tsx +++ b/desktop/src/app.tsx @@ -46,11 +46,12 @@ async function generate(prompt: string, model: string, callback: (res: string) = for (const message of messages) { const choice = message.choices[0] + + callback(choice.text) + if (choice.finish_reason === 'stop') { break } - - callback(choice.text) } }