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

Download all Poe bots.

Browse Source
This commit is contained in:
Cohee
2023-07-09 23:07:54 +03:00
parent e883a37953
commit d35ed21f48
2 changed files with 106 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/poe-client.js
View File

@@ -428,13 +428,13 @@ class Client {
};
this.session.defaults.headers.common = this.headers;
[this.next_data, this.channel] = await Promise.all([this.get_next_data(), this.get_channel_data()]);
this.bots = await this.get_bots();
this.bot_names = this.get_bot_names();
this.gql_headers = {
"poe-formkey": this.formkey,
"poe-tchannel": this.channel["channel"],
...this.headers,
};
this.bots = await this.get_bots();
this.bot_names = this.get_bot_names();
if (this.device_id === null) {
this.device_id = this.get_device_id();
}
@@ -519,6 +519,20 @@ class Client {
throw new Error('Invalid token.');
}
const botList = viewer.availableBotsConnection.edges.map(x => x.node);
try {
const botsQuery = await this.send_query('BotSwitcherModalQuery', {});
botsQuery.data.viewer.availableBotsConnection.edges.forEach(edge => {
const bot = edge.node;
if (botList.findIndex(x => x.id === bot.id) === -1) {
botList.push(bot);
}
});
} catch (e) {
console.log(e);
}
const retries = 2;
const bots = {};
const promises = [];