From cb42f77ed29339dd6c28cb39b2bf4996508ea377 Mon Sep 17 00:00:00 2001 From: Stef Dunlap Date: Tue, 7 Mar 2023 21:30:57 -0500 Subject: [PATCH] Make sure a line doesn't get cut off --- index.js | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/index.js b/index.js index 97cfb2b..63045c4 100644 --- a/index.js +++ b/index.js @@ -39,6 +39,7 @@ async function chatgpt(query, callback) { let parts = data.split('\n'); // parse if starts with data: for(part of parts) { + console.log(part); if(part === 'data: [DONE]') { callback(line); line = ''; @@ -47,7 +48,11 @@ async function chatgpt(query, callback) { try { let json = JSON.parse(jsonString); let chunk = json.choices[0].delta.content; - if(!chunk) { return; } + if(!chunk) { + callback(line); + line = ''; + continue; + } //split the chunk into lines leaving the delimiter in the array const lines = chunk.split(/\r?\n/); // split by new lines @@ -70,6 +75,11 @@ async function chatgpt(query, callback) { callback(line); line = ''; } + + if(line.length > 400) { + callback(line); + line = ''; + } } catch (e) { console.log(e); console.log(part);