rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix crash on poe streaming

Browse Source
This commit is contained in:
SillyLossy
2023-04-13 15:14:43 +03:00
parent 25ae214c12
commit 661b41341e
1 changed files with 42 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
server.js
View File

@@ -353,6 +353,7 @@ app.post("/generate_textgenerationwebui", jsonParser, async function (request, r
const SEND_PARAMS_GRADIO_FN = 29; const SEND_PARAMS_GRADIO_FN = 29;
response_generate.writeHead(200, { response_generate.writeHead(200, {
'Content-Type': 'text/plain;charset=utf-8',
'Transfer-Encoding': 'chunked', 'Transfer-Encoding': 'chunked',
'Cache-Control': 'no-transform', 'Cache-Control': 'no-transform',
}); });
@@ -363,7 +364,7 @@ app.post("/generate_textgenerationwebui", jsonParser, async function (request, r
const websocket = new WebSocket(`ws://${url.host}/queue/join`, { perMessageDeflate: false }); const websocket = new WebSocket(`ws://${url.host}/queue/join`, { perMessageDeflate: false });
let text = ''; let text = '';
websocket.on('open', async function() { websocket.on('open', async function () {
console.log('websocket open'); console.log('websocket open');
}); });
@@ -1551,9 +1552,9 @@ app.post("/exportcharacter", jsonParser, async function (request, response) {
let metadataPath = `./uploads/${Date.now()}_metadata.exif`; let metadataPath = `./uploads/${Date.now()}_metadata.exif`;
let metadata = let metadata =
{ {
"Exif": { "Exif": {
[exif.ExifIFD.UserComment]: json, [exif.ExifIFD.UserComment]: json,
}, },
}; };
const exifString = exif.dump(metadata); const exifString = exif.dump(metadata);
fs.writeFileSync(metadataPath, exifString, 'binary'); fs.writeFileSync(metadataPath, exifString, 'binary');
@@ -1924,11 +1925,20 @@ app.post('/generate_poe', jsonParser, async (request, response) => {
const bot = request.body.bot ?? POE_DEFAULT_BOT; const bot = request.body.bot ?? POE_DEFAULT_BOT;
const streaming = request.body.streaming ?? false; const streaming = request.body.streaming ?? false;
try { let client;
const client = await getPoeClient(token);
if (streaming) { try {
client = await getPoeClient(token);
}
catch (error) {
console.error(error);
return response.sendStatus(500);
}
if (streaming) {
try {
response.writeHead(200, { response.writeHead(200, {
'Content-Type': 'text/plain;charset=utf-8',
'Transfer-Encoding': 'chunked', 'Transfer-Encoding': 'chunked',
'Cache-Control': 'no-transform', 'Cache-Control': 'no-transform',
}); });
@@ -1940,10 +1950,17 @@ app.post('/generate_poe', jsonParser, async (request, response) => {
response.write(newText); response.write(newText);
} }
console.log(reply); console.log(reply);
client.disconnect_ws();
response.end();
} }
else { catch (err) {
console.error(err);
}
finally {
client.disconnect_ws();
return response.end();
}
}
else {
try {
let reply; let reply;
for await (const mes of client.send_message(bot, prompt)) { for await (const mes of client.send_message(bot, prompt)) {
reply = mes.text; reply = mes.text;
@@ -1952,10 +1969,10 @@ app.post('/generate_poe', jsonParser, async (request, response) => {
client.disconnect_ws(); client.disconnect_ws();
return response.send({ 'reply': reply }); return response.send({ 'reply': reply });
} }
} catch {
catch (error) { client.disconnect_ws();
console.error(error); return response.sendStatus(500);
return response.sendStatus(500); }
} }
}); });