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

Poe API fixed

Browse Source
This commit is contained in:
SillyLossy
2023-04-05 12:46:35 +03:00
parent 4cfad2029c
commit ce1f33679e
1 changed files with 32 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
poe.js
View File

@@ -252,24 +252,44 @@ class Client {
}
}
on_message(ws, msg) {
const data = JSON.parse(msg);
const message = JSON.parse(data["messages"][0])["payload"]["data"]["messageAdded"];
async on_message(ws, msg) {
try {
const data = JSON.parse(msg);
const copiedDict = Object.assign({}, this.active_messages);
for (const [key, value] of Object.entries(copiedDict)) {
//add the message to the appropriate queue
if (value === message["messageId"] && key in this.message_queues) {
this.message_queues[key].push(message);
if (!('messages' in data)) {
return;
}
//indicate that the response id is tied to the human message id
else if (key !== "pending" && value === null && message["state"] !== "complete") {
this.active_messages[key] = message["messageId"];
this.message_queues[key].push(message);
for (const message_str of data["messages"]) {
const message_data = JSON.parse(message_str);
if (message_data["message_type"] != "subscriptionUpdate"){
continue;
}
const message = message_data["payload"]["data"]["messageAdded"]
const copiedDict = Object.assign({}, this.active_messages);
for (const [key, value] of Object.entries(copiedDict)) {
//add the message to the appropriate queue
if (value === message["messageId"] && key in this.message_queues) {
this.message_queues[key].push(message);
return;
}
//indicate that the response id is tied to the human message id
else if (key !== "pending" && value === null && message["state"] !== "complete") {
this.active_messages[key] = message["messageId"];
this.message_queues[key].push(message);
}
}
}
}
catch (err) {
console.log('Error occurred in onMessage', err);
this.disconnect_ws();
await this.connect_ws();
}
}
async *send_message(chatbot, message, with_chat_break = false, timeout = 20) {