kova/youchat_proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

optimize custom Model name records

Browse Source
This commit is contained in:
被遗忘的记忆
2024-10-13 13:51:17 +08:00
committed by GitHub
parent 463b331194
commit 50c0d47a40
1 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
provider.mjs
+18 -4
View File
@@ -693,7 +693,20 @@ class YouProvider {
if (!this.config.sessions[session.configIndex].user_chat_mode_id) { if (!this.config.sessions[session.configIndex].user_chat_mode_id) {
this.config.sessions[session.configIndex].user_chat_mode_id = {}; this.config.sessions[session.configIndex].user_chat_mode_id = {};
} }
if (!this.config.sessions[session.configIndex].user_chat_mode_id[proxyModel]) {
// 检查是否存在与当前用户名匹配的记录
let existingUserRecord = Object.keys(this.config.sessions[session.configIndex].user_chat_mode_id).find(key => key === username);
if (!existingUserRecord) {
// 为当前用户创建新的记录
this.config.sessions[session.configIndex].user_chat_mode_id[username] = {};
// 写回config
fs.writeFileSync("./config.mjs", "export const config = " + JSON.stringify(this.config, null, 4));
console.log(`Created new record for user: ${username}`);
}
// 检查是否存在对应模型的记录
if (!this.config.sessions[session.configIndex].user_chat_mode_id[username][proxyModel]) {
// 创建新的user chat mode // 创建新的user chat mode
let userChatMode = await page.evaluate( let userChatMode = await page.evaluate(
async (proxyModel, proxyModelName) => { async (proxyModel, proxyModelName) => {
@@ -720,15 +733,16 @@ class YouProvider {
uuidV4().substring(0, 4) uuidV4().substring(0, 4)
); );
if (userChatMode.chat_mode_id) { if (userChatMode.chat_mode_id) {
this.config.sessions[session.configIndex].user_chat_mode_id[proxyModel] = userChatMode.chat_mode_id; this.config.sessions[session.configIndex].user_chat_mode_id[username][proxyModel] = userChatMode.chat_mode_id;
// 写回config // 写回config
fs.writeFileSync("./config.mjs", "export const config = " + JSON.stringify(this.config, null, 4)); fs.writeFileSync("./config.mjs", "export const config = " + JSON.stringify(this.config, null, 4));
console.log(`Created new chat mode for user ${username} and model ${proxyModel}`);
} else { } else {
if (userChatMode.error) console.log(userChatMode.error) if (userChatMode.error) console.log(userChatMode.error);
console.log("Failed to create user chat mode, will use default mode instead."); console.log("Failed to create user chat mode, will use default mode instead.");
} }
} }
userChatModeId = this.config.sessions[session.configIndex].user_chat_mode_id[proxyModel]; userChatModeId = this.config.sessions[session.configIndex].user_chat_mode_id[username][proxyModel];
} else { } else {
console.log("Custom mode is disabled, using default mode."); console.log("Custom mode is disabled, using default mode.");
} }